From ac245642127721477090e88c5e68f0ba4e222d1a Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Wed, 24 Jan 2024 12:15:21 +0800
Subject: [PATCH] migrate: hustoj: add ybtbas support

---
 packages/migrate/scripts/hustoj.ts | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/packages/migrate/scripts/hustoj.ts b/packages/migrate/scripts/hustoj.ts
index 4f837e437..a5a615080 100644
--- a/packages/migrate/scripts/hustoj.ts
+++ b/packages/migrate/scripts/hustoj.ts
@@ -4,7 +4,7 @@ import path from 'path';
 import mariadb from 'mariadb';
 import TurndownService from 'turndown';
 import {
-    _, buildContent, ContestModel, DomainModel, fs, noop, NotFoundError, ObjectId, postJudge, ProblemModel,
+    _, buildContent, ContestModel, DomainModel, fs, MessageModel, noop, NotFoundError, ObjectId, postJudge, ProblemModel,
     RecordDoc, RecordModel, SolutionModel, STATUS, StorageModel, SystemModel, Time, UserModel,
 } from 'hydrooj';
 
@@ -65,6 +65,7 @@ export async function run({
     username, password, domainId, contestType = 'oi',
     dataDir, uploadDir = '/home/judge/src/web/upload/', rerun = true, randomMail = false,
 }, report: Function) {
+    let remoteUsed = false;
     const src = await mariadb.createConnection({
         host,
         port,
@@ -199,17 +200,31 @@ export async function run({
             await ProblemModel.edit(domainId, pidMap[pdoc.problem_id], {
                 nAccept: 0,
                 nSubmit: pdoc.submit,
-                config: `time: ${pdoc.time_limit}s\nmemory: ${pdoc.memory_limit}m`,
+                config: `time: ${pdoc.time_limit}s
+memory: ${pdoc.memory_limit}m
+${pdoc.remote_oj === 'bas' ? `type: remote_judge
+subType: ybtbas
+target: ybtbas/${+pdoc.id - 3000}
+` : ''}`,
                 owner: uidMap[cdoc[0]?.user_id] || 1,
                 maintainer,
                 html: true,
             });
+            if (pdoc.remote_oj === 'bas') remoteUsed = true;
             if (pdoc.solution) {
                 const md = turndown.turndown(pdoc.solution);
                 await SolutionModel.add(domainId, pidMap[pdoc.problem_id], 1, md);
             }
         }
     }
+    if (remoteUsed) {
+        MessageModel.sendNotification(`您导入的数据中使用了一本通编程启蒙远端测试题目。
+请在迁移脚本运行完成后，在终端中运行
+hydrooj install https://hydro.ac/hydroac-client.zip
+安装远端评测所需要的插件。插件安装完成后，请重启 hydrooj，待一分钟后，再次重启 hydrooj。
+在此之后，远端评测应当能够正常进行。
+（注：您无需再使用申请的评测账号）`);
+    }
     report({ message: 'problem finished' });
 
     /*