core: import: support using extracted folder

packages/hydrooj/src/model/problem.ts

@@ -407,19 +407,26 @@ export class ProblemModel {
     }
 
     static async import(domainId: string, filepath: string, operator: number) {
-        const tmpdir = path.join(os.tmpdir(), 'hydro', `${Math.random()}.import`);
+        let tmpdir = '';
-        let zip: AdmZip;
+        if (filepath.endsWith('.zip')) {
-        try {
+            tmpdir = path.join(os.tmpdir(), 'hydro', `${Math.random()}.import`);
-            zip = new AdmZip(filepath);
+            let zip: AdmZip;
-        } catch (e) {
+            try {
-            throw new ValidationError('zip', null, e.message);
+                zip = new AdmZip(filepath);
-        }
+            } catch (e) {
-        await new Promise((resolve, reject) => {
+                throw new ValidationError('zip', null, e.message);
-            zip.extractAllToAsync(tmpdir, true, (err) => {
+            }
-                if (err) reject(err);
+            await new Promise((resolve, reject) => {
-                resolve(null);
+                zip.extractAllToAsync(tmpdir, true, (err) => {
+                    if (err) reject(err);
+                    resolve(null);
+                });
             });
-        });
+        } else if (fs.statSync(filepath).isDirectory()) {
+            tmpdir = filepath;
+        } else {
+            throw new ValidationError('file', null, 'Invalid file');
+        }
         try {
             const problems = await fs.readdir(tmpdir, { withFileTypes: true });
             for (const p of problems) {