From 459956049f804b047e933a273859c8b4f4860021 Mon Sep 17 00:00:00 2001 From: Baoshuo Date: Tue, 24 Jan 2023 16:45:22 +0800 Subject: [PATCH] feat(remote_judger): add loj --- remote_judger/src/providers/loj.ts | 308 ++++++++++++++++++++++++++++ remote_judger/src/vjudge.ts | 1 + web/app/models/UOJRemoteProblem.php | 2 +- 3 files changed, 310 insertions(+), 1 deletion(-) create mode 100644 remote_judger/src/providers/loj.ts diff --git a/remote_judger/src/providers/loj.ts b/remote_judger/src/providers/loj.ts new file mode 100644 index 0000000..9b3ba20 --- /dev/null +++ b/remote_judger/src/providers/loj.ts @@ -0,0 +1,308 @@ +import superagent from 'superagent'; +import proxy from 'superagent-proxy'; +import sleep from '../utils/sleep'; +import { IBasicProvider, RemoteAccount, USER_AGENT } from '../interface'; +import Logger from '../utils/logger'; + +proxy(superagent); +const logger = new Logger('remote/loj'); + +const langs_map = { + C: { + name: 'C (gcc, c11, O2, m64)', + info: { + language: 'c', + compileAndRunOptions: { + compiler: 'gcc', + O: '2', + m: '64', + std: 'c11', + }, + }, + comment: '//', + }, + 'C++03': { + name: 'C++ (g++, c++03, O2, m64)', + info: { + language: 'cpp', + compileAndRunOptions: { + compiler: 'g++', + std: 'c++03', + O: '2', + m: '64', + }, + }, + comment: '//', + }, + 'C++11': { + name: 'C++ (g++, c++11, O2, m64)', + info: { + language: 'cpp', + compileAndRunOptions: { + compiler: 'g++', + std: 'c++11', + O: '2', + m: '64', + }, + }, + comment: '//', + }, + 'C++': { + name: 'C++ (g++, c++14, O2, m64)', + info: { + language: 'cpp', + compileAndRunOptions: { + compiler: 'g++', + std: 'c++14', + O: '2', + m: '64', + }, + }, + comment: '//', + }, + 'C++17': { + name: 'C++ (g++, c++17, O2, m64)', + info: { + language: 'cpp', + compileAndRunOptions: { + compiler: 'g++', + std: 'c++17', + O: '2', + m: '64', + }, + }, + comment: '//', + }, + 'C++20': { + name: 'C++ (g++, c++20, O2, m64)', + info: { + language: 'cpp', + compileAndRunOptions: { + compiler: 'g++', + std: 'c++20', + O: '2', + m: '64', + }, + }, + comment: '//', + }, + 'Python2.7': { + name: 'Python (2.7)', + info: { + language: 'python', + compileAndRunOptions: { + version: '2.7', + }, + }, + comment: '#', + }, + Python3: { + name: 'Python (3.10)', + info: { + language: 'python', + compileAndRunOptions: { + version: '3.10', + }, + }, + comment: '#', + }, + Java17: { + name: 'Java', + info: { + language: 'java', + compileAndRunOptions: {}, + }, + comment: '//', + }, + Pascal: { + name: 'Pascal', + info: { + language: 'pascal', + compileAndRunOptions: { + O: '2', + }, + }, + comment: '//', + }, +}; + +export function getAccountInfoFromEnv(): RemoteAccount | null { + const { + LOJ_HANDLE, + LOJ_TOKEN, + LOJ_ENDPOINT = 'https://api.loj.ac.cn/api', + LOJ_PROXY, + } = process.env; + + if (!LOJ_TOKEN) return null; + + const account: RemoteAccount = { + type: 'loj', + handle: LOJ_HANDLE, + password: LOJ_TOKEN, + endpoint: LOJ_ENDPOINT, + }; + + if (LOJ_PROXY) account.proxy = LOJ_PROXY; + + return account; +} + +export default class LibreojProvider implements IBasicProvider { + constructor(public account: RemoteAccount) { + this.account.endpoint ||= 'https://api.loj.ac.cn/api'; + } + + get(url: string) { + logger.debug('get', url); + if (!url.includes('//')) url = `${this.account.endpoint}${url}`; + const req = superagent + .get(url) + .auth(this.account.password, { type: 'bearer' }) + .set('User-Agent', USER_AGENT); + if (this.account.proxy) return req.proxy(this.account.proxy); + return req; + } + + post(url: string) { + logger.debug('post', url); + if (!url.includes('//')) url = `${this.account.endpoint}${url}`; + const req = superagent + .post(url) + .type('json') + .auth(this.account.password, { type: 'bearer' }) + .set('User-Agent', USER_AGENT) + .set('x-recaptcha-token', 'skip'); + if (this.account.proxy) return req.proxy(this.account.proxy); + return req; + } + + get loggedIn() { + return this.get('/auth/getSessionInfo?token=' + this.account.password).then( + res => + res.body.userMeta && res.body.userMeta.username === this.account.handle + ); + } + + async ensureLogin() { + if (await this.loggedIn) return true; + logger.info('retry login'); + // TODO: login + return false; + } + + async getProblemId(displayId: number) { + const { body } = await this.post('/problem/getProblem').send({ displayId }); + + return body.meta.id; + } + + async submitProblem( + displayId: string, + lang: string, + code: string, + submissionId: number, + next, + end + ) { + const programType = langs_map[lang] || langs_map['C++']; + const comment = programType.comment; + + if (comment) { + const msg = `S2OJ Submission #${submissionId} @ ${new Date().getTime()}`; + if (typeof comment === 'string') code = `${comment} ${msg}\n${code}`; + else if (comment instanceof Array) + code = `${comment[0]} ${msg} ${comment[1]}\n${code}`; + } + + const id = await this.getProblemId(parseInt(displayId)); + + logger.debug( + 'Submitting', + id, + `(displayId: ${displayId})`, + programType, + lang, + `(S2OJ Submission #${submissionId})` + ); + + const { body, error } = await this.post('/submission/submit').send({ + problemId: id, + content: { + code, + ...programType.info, + }, + uploadInfo: null, + }); + + if (error) { + await end({ + error: true, + status: 'Judgment Failed', + message: 'Failed to submit code.', + }); + + return null; + } + + return body.submissionId; + } + + async waitForSubmission(problem_id: string, id: string, next, end) { + let i = 0; + + while (true) { + if (++i > 60) { + return await end({ + id, + error: true, + status: 'Judgment Failed', + message: 'Failed to fetch submission details.', + }); + } + + await sleep(2000); + const { body, error } = await this.post('/submission/getSubmissionDetail') + .send({ submissionId: String(id), locale: 'zh_CN' }) + .retry(3); + + if (error) continue; + + if (body.progress.progressType !== 'Finished') { + await next({ + status: `${body.progress.progressType}: ${body.progress.status}`, + }); + + continue; + } + + if (body.meta.status === 'CompilationError') { + await end({ + error: true, + id, + status: 'Compile Error', + }); + } + + if ( + ['SystemError', 'JudgementFailed', 'ConfigurationError'].includes( + body.meta.status + ) + ) { + await end({ + error: true, + id, + status: 'Judgment Failed', + }); + } + + return await end({ + id, + status: body.meta.status, + score: body.meta.score, + time: body.meta.timeUsed, + memory: body.meta.memoryUsed, + }); + } + } +} diff --git a/remote_judger/src/vjudge.ts b/remote_judger/src/vjudge.ts index b18e6b0..387053c 100644 --- a/remote_judger/src/vjudge.ts +++ b/remote_judger/src/vjudge.ts @@ -154,6 +154,7 @@ export async function apply(request: any) { await vjudge.addProvider('codeforces'); await vjudge.addProvider('atcoder'); await vjudge.addProvider('uoj'); + await vjudge.addProvider('loj'); return vjudge; } diff --git a/web/app/models/UOJRemoteProblem.php b/web/app/models/UOJRemoteProblem.php index c172efa..b5f0ac9 100644 --- a/web/app/models/UOJRemoteProblem.php +++ b/web/app/models/UOJRemoteProblem.php @@ -36,7 +36,7 @@ class UOJRemoteProblem { 'name' => 'LibreOJ', 'short_name' => 'LOJ', 'url' => 'https://loj.ac', - 'languages' => ['C', 'C++03', 'C++11', 'C++', 'C++17', 'C++20', 'Python3', 'Python2.7', 'Java8', 'Java11', 'Java17', 'Pascal'], + 'languages' => ['C', 'C++03', 'C++11', 'C++', 'C++17', 'C++20', 'Python3', 'Python2.7', 'Java17', 'Pascal'], ], ];