core: 支持限制题目的提交语言

packages/hydrooj/package.json

@@ -1,6 +1,6 @@
 {
     "name": "hydrooj",
-    "version": "2.26.12",
+    "version": "2.26.13",
     "bin": "bin/hydrooj.js",
     "main": "dist/loader.js",
     "typings": "dist/loader.d.ts",

packages/hydrooj/src/handler/contest.ts

@@ -4,7 +4,7 @@ import AdmZip from 'adm-zip';
 import { isSafeInteger } from 'lodash';
 import {
     ContestNotLiveError, ValidationError, ProblemNotFoundError,
-    ContestNotAttendedError, PermissionError,
+    ContestNotAttendedError, PermissionError, BadRequestError,
 } from '../error';
 import { ProblemDoc, Tdoc, User } from '../interface';
 import paginate from '../lib/paginate';
@@ -344,6 +344,9 @@ class ContestDetailProblemSubmitHandler extends ContestProblemHandler {
     @param('lang', Types.Name)
     @param('code', Types.Content)
     async post(domainId: string, tid: ObjectID, lang: string, code: string) {
+        if (this.response.body.pdoc.config.langs && !this.response.body.pdoc.config.langs.includes('lang')) {
+            throw new BadRequestError('Language not allowed.');
+        }
         await this.limitRate('add_record', 60, 10);
         const rid = await record.add(domainId, this.pdoc.docId, this.user._id, lang, code, true, {
             type: document.TYPE_CONTEST,

packages/hydrooj/src/handler/problem.ts

@@ -264,6 +264,9 @@ export class ProblemSubmitHandler extends ProblemDetailHandler {
     @param('lang', Types.Name)
     @param('code', Types.Content)
     async post(domainId: string, lang: string, code: string) {
+        if (this.response.body.pdoc.config.langs && !this.response.body.pdoc.config.langs.includes('lang')) {
+            throw new BadRequestError('Language not allowed.');
+        }
         await this.limitRate('add_record', 60, 5);
         const rid = await record.add(domainId, this.pdoc.docId, this.user._id, lang, code, true);
         const [rdoc] = await Promise.all([

packages/hydrooj/src/interface.ts

@@ -143,19 +143,20 @@ export interface SubtaskConfig {
 }
 
 export interface ProblemConfig {
-    type?: ProblemType,
-    score?: number,
-    time?: string,
-    memory?: string,
-    filename?: string,
-    checker_type?: string,
-    checker?: string,
-    interactor?: string,
-    user_extra_files?: string[],
-    judge_extra_files?: string[],
-    outputs?: [string, number][],
-    cases?: TestCaseConfig[],
-    subtasks?: SubtaskConfig[],
+    type?: ProblemType;
+    score?: number;
+    time?: string;
+    memory?: string;
+    filename?: string;
+    checker_type?: string;
+    checker?: string;
+    interactor?: string;
+    user_extra_files?: string[];
+    judge_extra_files?: string[];
+    outputs?: [string, number][];
+    cases?: TestCaseConfig[];
+    subtasks?: SubtaskConfig[];
+    langs?: string[];
 }
 
 export interface PlainContentNode {

packages/hydrooj/src/lib/testdataConfig.ts

@@ -3,11 +3,12 @@ import { load } from 'js-yaml';
 import type { ProblemConfig } from '../interface';
 
 interface ParseResult {
-    count: number,
-    memoryMax: number,
-    memoryMin: number,
-    timeMax: number,
-    timeMin: number,
+    count: number;
+    memoryMax: number;
+    memoryMin: number;
+    timeMax: number;
+    timeMin: number;
+    langs?: string[];
 }
 
 export async function parseConfig(config: string | ProblemConfig = {}) {
@@ -40,6 +41,7 @@ export async function parseConfig(config: string | ProblemConfig = {}) {
     if (result.timeMax < result.timeMin) {
         result.timeMax = result.timeMin = 1000;
     }
+    if (cfg.langs) result.langs = cfg.langs;
     return result;
 }
 

packages/hydrooj/src/model/discussion.ts

@@ -284,7 +284,8 @@ const t = Math.exp(-0.15);
 
 async function updateSort() {
     const cursor = document.coll.find({ docType: document.TYPE_DISCUSSION });
-    while (cursor.hasNext()) {
+    // eslint-disable-next-line no-await-in-loop
+    while (await cursor.hasNext()) {
         // eslint-disable-next-line no-await-in-loop
         const data = await cursor.next();
         // eslint-disable-next-line no-await-in-loop