From 0098031da192aebd436c1b11e35b4d37004f1ba6 Mon Sep 17 00:00:00 2001
From: Baoshuo <i@baoshuo.ren>
Date: Sun, 12 Nov 2023 15:38:45 +0800
Subject: [PATCH] fix(remote_judger/codeforces): fetch submission

---
 remote_judger/src/providers/codeforces.ts | 24 +++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/remote_judger/src/providers/codeforces.ts b/remote_judger/src/providers/codeforces.ts
index bacd14f..9dde778 100644
--- a/remote_judger/src/providers/codeforces.ts
+++ b/remote_judger/src/providers/codeforces.ts
@@ -341,9 +341,12 @@ export default class CodeforcesProvider implements IBasicProvider {
     this.csrf = document
       .querySelector('meta[name="X-Csrf-Token"]')
       .getAttribute('content');
-    return document
+
+    const submission = document
       .querySelector('[data-submission-id]')
       .getAttribute('data-submission-id');
+
+    return type !== 'GYM' ? submission : `${contestId}#${submission}`;
   }
 
   async ensureIsOwnSubmission(id: string) {
@@ -354,21 +357,30 @@ export default class CodeforcesProvider implements IBasicProvider {
     let count = 0;
     let fail = 0;
 
+    const contestId = id.includes('#') ? id.split('#')[0] : null;
+    const submissionId = id.includes('#') ? id.split('#')[1] : id;
+
     while (count < 360 && fail < 60) {
       count++;
       await sleep(500);
 
       try {
         const { body } = await this.post('/data/submitSource')
+          .set(
+            'referer',
+            contestId
+              ? `https://codeforces.com/gym/${contestId}/my`
+              : 'https://codeforces.com/problemset/status?my=on'
+          )
           .send({
             csrf_token: this.csrf,
-            submissionId: id,
+            submissionId: submissionId,
           })
           .retry(3);
 
         if (body.compilationError === 'true') {
           return await end({
-            id,
+            id: submissionId,
             error: true,
             status: 'Compile Error',
             message: crlf(body['checkerStdoutAndStderr#1'], LF),
@@ -439,7 +451,7 @@ export default class CodeforcesProvider implements IBasicProvider {
           Object.entries({
             比赛: stripHtml(body.contestName).result,
             题目: stripHtml(body.problemName).result,
-            提交记录: `<a href="https://codeforces.com${body.href}">${id}</a>`,
+            提交记录: `<a href="https://codeforces.com${body.href}">${submissionId}</a>`,
             账号: `<a href="https://codeforces.com/profile/${remote_handle}">${remote_handle}</a>`,
             状态: stripHtml(body.verdict).result,
           })
@@ -453,7 +465,7 @@ export default class CodeforcesProvider implements IBasicProvider {
           '</div>';
 
         return await end({
-          id,
+          id: submissionId,
           status,
           score: status === 'Accepted' ? 100 : 0,
           time,
@@ -469,7 +481,7 @@ export default class CodeforcesProvider implements IBasicProvider {
     }
 
     return await end({
-      id,
+      id: submissionId,
       error: true,
       status: 'Judgment Failed',
       message: 'Failed to fetch submission details.',