Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 09:46:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

core: add preferredPrefix option for problem_import

Browse Source
This commit is contained in:
undefined committed 2023-06-04 16:55:28 +08:00
1 parent 98a15069eb
commit b8bbd504b0
2 files changed
+19 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/model/problem.ts
+13 -3
View File
@@ -407,7 +407,7 @@ export class ProblemModel {
return true;
}
static async import(domainId: string, filepath: string, operator = 1) {
static async import(domainId: string, filepath: string, operator = 1, preferredPrefix?: string) {
let tmpdir = '';
let del = false;
if (filepath.endsWith('.zip')) {
@@ -441,9 +441,19 @@ export class ProblemModel {
const pdoc: ProblemDoc = yaml.load(content) as any;
if (!pdoc) continue;
let pid = pdoc.pid;
const isValidPid = async (id: string) => {
if (!(/^[A-Za-z]+[0-9A-Za-z]*$/.test(id))) return false;
if (await ProblemModel.get(domainId, id)) return false;
return true;
};
if (pid) {
if (!(/^[A-Za-z]+[0-9A-Za-z]*$/.test(pid))) pid = undefined;
else if (await ProblemModel.get(domainId, pid)) pid = undefined;
if (preferredPrefix) {
const newPid = pid.replace(/^[A-Za-z]+/, preferredPrefix);
if (isValidPid(newPid)) pid = newPid;
}
if (!isValidPid(pid)) pid = undefined;
}
const overrideContent = findOverrideContent(path.join(tmpdir, i));
const docId = await ProblemModel.add(