mirror of
https://gitcode.com/gh_mirrors/hy/Hydro.git
synced 2026-10-17 17:36:08 +08:00
119 lines
4.4 KiB
TypeScript
119 lines
4.4 KiB
TypeScript
/* eslint-disable no-tabs */
|
||
/* eslint-disable no-await-in-loop */
|
||
import mysql from 'mysql';
|
||
import 'hydrooj';
|
||
import { buildContent } from 'hydrooj/dist/utils';
|
||
|
||
const {
|
||
user, problem, builtin: { PRIV },
|
||
} = global.Hydro.model;
|
||
|
||
export async function run({
|
||
host = 'localhost', port = 27017, name = 'vijos4', username, password,
|
||
}, report: Function) {
|
||
const src = mysql.createConnection({
|
||
host,
|
||
port,
|
||
user: username,
|
||
password,
|
||
database: name,
|
||
});
|
||
await new Promise((resolve) => src.connect(resolve));
|
||
const query = (q: string | mysql.Query): Promise<[values: any[], fields: mysql.FieldInfo[]]> => new Promise((res, rej) => {
|
||
src.query(q, (err, val, fields) => {
|
||
if (err) rej(err);
|
||
res([val, fields]);
|
||
});
|
||
});
|
||
report({ message: 'Connected to database' });
|
||
/*
|
||
user_id varchar 20 N 用户id（主键）
|
||
email varchar 100 Y 用户E-mail
|
||
submit int 11 Y 用户提交次数
|
||
solved int 11 Y 成功次数
|
||
defunct char 1 N 是否屏蔽（Y/N）
|
||
ip varchar 20 N 用户注册ip
|
||
accesstime datetime Y 用户注册时间
|
||
volume int 11 N 页码（表示用户上次看到第几页）
|
||
language int 11 N 语言
|
||
password varchar 32 Y 密码（加密）
|
||
reg_time datetime Y 用户注册时间
|
||
nick varchar 100 N 昵称
|
||
school varchar 100 N 用户所在学校
|
||
*/
|
||
const uidMap: Record<string, number> = {};
|
||
|
||
const [udocs] = await query('SELECT * FROM `users`');
|
||
for (const udoc of udocs) {
|
||
const uid = await user.create(udoc.email, udoc.nick, '', null, udoc.ip, udoc.defunct === 'Y' ? 0 : PRIV.PRIV_DEFAULT);
|
||
uidMap[udoc.uid] = uid;
|
||
await user.setById(uid, {
|
||
nSubmit: udoc.submit,
|
||
nSolve: udoc.solved,
|
||
loginat: udoc.accesstime,
|
||
regat: udoc.reg_time,
|
||
hash: udoc.password,
|
||
salt: udoc.password,
|
||
hashType: 'hust',
|
||
});
|
||
}
|
||
|
||
const [admins] = await query("SELECT * FROM 'privilege' WHERE rightstr = 'administrator'");
|
||
for (const admin of admins) await user.setSuperAdmin(uidMap[admin.user_id]);
|
||
|
||
/*
|
||
problem_id int 11 N 题目编号，主键
|
||
title varchar 200 N 标题
|
||
description text Y 题目描述
|
||
inupt text Y 输入说明
|
||
output text Y 输出说明
|
||
sample_input text Y 输入参照
|
||
sample_output text Y 输出参照
|
||
spj char 1 N 是否为特别题目
|
||
hint text Y 暗示
|
||
source varchar 100 Y 来源
|
||
in_date datetime Y 加入时间
|
||
time_limit int 11 N 限时（秒）
|
||
memory_limit int 11 N 空间限制(MByte)
|
||
defunct char 1 N 是否屏蔽（Y/N）
|
||
accepted int 11 Y 总ac次数
|
||
submit int 11 Y 总提交次数
|
||
solved int 11 Y 解答（未用）
|
||
*/
|
||
const [[pcount]] = await query('SELECT count(*) FROM `problem`');
|
||
const step = 50;
|
||
const pageCount = Math.ceil(pcount / step);
|
||
for (let pageId = 0; pageId < pageCount; pageId++) {
|
||
const [pdocs] = await query(`SELECT * FROM \`problem\` LIMIT ${pageId * step}, ${step}`);
|
||
for (const pdoc of pdocs) {
|
||
const pid = await problem.add(
|
||
'system', `P${pdoc.problem_id}`,
|
||
pdoc.title, buildContent({
|
||
description: pdoc.description,
|
||
input: pdoc.input,
|
||
output: pdoc.output,
|
||
samples: [[pdoc.sample_input, pdoc.sample_output]],
|
||
hint: pdoc.hint,
|
||
source: pdoc.source,
|
||
}, 'html'),
|
||
1, pdoc.source.split(' '), undefined, undefined, pdoc.defunct === 'Y',
|
||
);
|
||
const [cdoc] = await query(`SELECT * FROM 'privilege' WHERE rightstr = 'p${pdoc.problem_id}'`);
|
||
await problem.edit('system', pid, {
|
||
nAccept: pdoc.accepted,
|
||
nSubmit: pdoc.submit,
|
||
config: { time: `${pdoc.time_limit}s`, memory: `${pdoc.memory_limit}m` },
|
||
owner: uidMap[cdoc[0].user_id],
|
||
});
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
export const description = 'migrate from hustoj';
|
||
export const validate = {
|
||
host: 'string', port: 'number', name: 'string', username: 'string', password: 'string',
|
||
};
|
||
|
||
global.Hydro.script.migrateHustoj = { run, description, validate };
|