From 7101a3096cc0ba3343ed148a7e7252e657107cab Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Wed, 3 Aug 2022 18:41:48 +0800
Subject: [PATCH] vjudge: luogu: o2

---
 packages/vjudge/package.json           |  2 +-
 packages/vjudge/src/providers/luogu.ts | 12 ++++++++++--
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/packages/vjudge/package.json b/packages/vjudge/package.json
index e8537ed75..23229ed40 100644
--- a/packages/vjudge/package.json
+++ b/packages/vjudge/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@hydrooj/vjudge",
-    "version": "1.5.0-next.5",
+    "version": "1.5.0-next.6",
     "description": "Submit problems to remote oj",
     "main": "package.json",
     "repository": "https://github.com/hydro-dev/Hydro.git",
diff --git a/packages/vjudge/src/providers/luogu.ts b/packages/vjudge/src/providers/luogu.ts
index 3b8fb18d3..5a1751586 100644
--- a/packages/vjudge/src/providers/luogu.ts
+++ b/packages/vjudge/src/providers/luogu.ts
@@ -105,17 +105,25 @@ export default class LuoguProvider implements IBasicProvider {
     }
 
     async submitProblem(id: string, lang: string, code: string, info) {
+        let enableO2 = 0;
         const comment = setting.langs[lang]?.comment;
         if (comment) {
             const msg = `Hydro submission #${info.rid}@${new Date().toLocaleString()}`;
             if (typeof comment === 'string') code = `${comment} ${msg}\n${code}`;
             else if (comment instanceof Array) code = `${comment[0]} ${msg} ${comment[1]}\n${code}`;
         }
+        if (lang.endsWith('o2')) {
+            enableO2 = 1;
+            lang = lang.slice(0, -2);
+        }
         lang = lang.includes('luogu.') ? lang.split('luogu.')[1] : '0';
         const result = await this.post(`/fe/api/problem/submit/${id}${this.account.query || ''}`)
             .set('referer', `https://www.luogu.com.cn/problem/${id}`)
-            // TODO O2
-            .send({ code, lang: +lang, enableO2: 0 });
+            .send({
+                code,
+                lang: +lang,
+                enableO2,
+            });
         logger.info('RecordID:', result.body.rid);
         return result.body.rid;
     }