core: allow run script via cli

packages/hydrooj/src/entry/cli.ts

@@ -3,10 +3,11 @@
 import { argv } from 'yargs';
 import { ObjectID } from 'mongodb';
 import {
-    lib, service, model,
-    builtinLib, builtinModel,
+    lib, service, model, script,
+    builtinLib, builtinModel, builtinScript,
 } from './common';
 import options from '../options';
+import { validate } from '../lib/validator';
 import * as bus from '../service/bus';
 import db from '../service/db';
 
@@ -21,8 +22,23 @@ function parseParameters(fn: Function) {
     return result === null ? [] : result;
 }
 
+async function runScript(name: string, arg: any) {
+    if (!global.Hydro.script[name]) return console.error('Script %s not found.', name);
+    validate(global.Hydro.script[name].validate, arg);
+    return await global.Hydro.script[name].run(arg, console.info);
+}
+
 async function cli() {
     const [, modelName, func, ...args] = argv._ as [string, string, string, ...any[]];
+    if (modelName === 'script') {
+        let arg: any;
+        try {
+            arg = JSON.parse(args.join(' '));
+        } catch (e) {
+            return console.error('Invalid argument');
+        }
+        return await runScript(func, arg);
+    }
     if (!global.Hydro.model[modelName]) {
         return console.error(`Model ${modelName} doesn't exist.`);
     }
@@ -65,22 +81,18 @@ export async function load() {
     require('../options');
     const opts = options();
     await db.start(opts);
-    const modelSystem = require('../model/system');
-    const [endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge] = modelSystem.getMany([
-        'file.endPoint', 'file.accessKey', 'file.secretKey', 'file.bucket', 'file.region',
-        'file.endPointForUser', 'file.endPointForJudge',
-    ]);
-    const sopts = {
-        endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge,
-    };
     const storage = require('../service/storage');
-    await storage.start(sopts);
+    await storage.start();
     for (const i of builtinLib) require(`../lib/${i}`);
     await lib(pending, fail);
+    const systemModel = require('../model/system');
+    await systemModel.runConfig();
     require('../service/gridfs');
     await service(pending, fail);
     for (const i of builtinModel) require(`../model/${i}`);
     await model(pending, fail);
+    for (const i of builtinScript) require(`../script/${i}`);
+    await script(pending, fail, []);
     await bus.parallel('app/started');
     await cli();
 }

packages/hydrooj/src/entry/master.ts

@@ -57,15 +57,8 @@ export async function load(call: Entry) {
         await modelSystem.set('file.secretKey', process.env.MINIO_SECRET_KEY);
         await modelSystem.set('file.endPoint', 'http://localhost:9000/');
     }
-    const [endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge] = modelSystem.getMany([
-        'file.endPoint', 'file.accessKey', 'file.secretKey', 'file.bucket', 'file.region',
-        'file.endPointForUser', 'file.endPointForJudge',
-    ]);
-    const sopts = {
-        endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge,
-    };
     const storage = require('../service/storage');
-    await storage.start(sopts);
+    await storage.start();
     require('../service/monitor');
     for (const i of builtinModel) require(`../model/${i}`);
     const scripts = require('../upgrade').default;

packages/hydrooj/src/entry/worker.ts

@@ -34,15 +34,8 @@ export async function load() {
     const modelSystem = require('../model/system');
     await modelSystem.runConfig();
     if (argv.loaderDetail) logger.info('finish: config');
-    const [endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge] = modelSystem.getMany([
-        'file.endPoint', 'file.accessKey', 'file.secretKey', 'file.bucket', 'file.region',
-        'file.endPointForUser', 'file.endPointForJudge',
-    ]);
-    const sopts = {
-        endPoint, accessKey, secretKey, bucket, region, endPointForUser, endPointForJudge,
-    };
     const storage = require('../service/storage');
-    storage.start(sopts);
+    await storage.start();
     if (argv.loaderDetail) logger.info('finish: storage.connect');
     for (const i of builtinLib) {
         let t;