atcoder-workspace 1.1.0-beta.1

package/src/atcoder/submit.ts

@@ -0,0 +1,170 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import * as cheerio from 'cheerio';
+import { createAtCoderClient } from './client';
+import { loadConfig } from '../config/config-store';
+import { detectCodeFile } from '../test-runner/runner';
+import { AtcError } from '../utils/errors';
+import { getLanguage, t } from '../utils/i18n';
+
+export interface SubmissionDetails {
+  submissionId: string;
+  url: string;
+}
+
+/**
+ * Submits the code file for a task to AtCoder.
+ * Returns the submission ID and submission URL.
+ */
+export async function submitTask(
+  workspaceRoot: string,
+  contestId: string,
+  taskId: string,
+  taskLabel: string,
+  fileArg?: string
+): Promise<SubmissionDetails> {
+  const config = loadConfig(workspaceRoot);
+  const lang = getLanguage(workspaceRoot);
+  const contestParentDir = config.contestDir ? path.join(workspaceRoot, config.contestDir) : workspaceRoot;
+  const taskDir = path.join(contestParentDir, contestId, taskLabel);
+
+  if (!fs.existsSync(taskDir)) {
+    throw new AtcError(`Task directory "${taskLabel}" not found in contest "${contestId}".`);
+  }
+
+  const { codeFile, langConfig } = detectCodeFile(workspaceRoot, taskDir, config, fileArg);
+  const codePath = path.join(taskDir, codeFile);
+  const codeContent = fs.readFileSync(codePath, 'utf8');
+
+  const client = createAtCoderClient(workspaceRoot);
+
+  let submitPageHtml = '';
+  try {
+    const res = await client.get(`/contests/${contestId}/submit?taskScreenName=${taskId}`);
+    submitPageHtml = res.data;
+  } catch (err: any) {
+    throw new AtcError(`Failed to access AtCoder submit page: ${err.message}`);
+  }
+
+  const $ = cheerio.load(submitPageHtml);
+  
+  const csrfToken = $('input[name="csrf_token"]').val();
+  if (!csrfToken) {
+    throw new AtcError('Could not find CSRF token. Make sure you are logged in (run "atc login").');
+  }
+
+  let selectElement = $(`#select-lang-${taskId} select`);
+  if (selectElement.length === 0) {
+    selectElement = $('div#select-lang select');
+  }
+  if (selectElement.length === 0) {
+    selectElement = $('select[name="data.LanguageId"]');
+  }
+
+  if (selectElement.length === 0) {
+    if ($('input[name="username"]').length > 0 || $('input[name="password"]').length > 0 || submitPageHtml.includes('/login')) {
+      throw new AtcError(t('submitSessionExpired', lang));
+    }
+    throw new AtcError(t('submitLangSelectNotFound', lang));
+  }
+
+  const options: { id: string; name: string }[] = [];
+  selectElement.find('option').each((_, opt) => {
+    const id = $(opt).val();
+    const name = $(opt).text().trim();
+    if (id && name) {
+      options.push({ id: id.toString(), name });
+    }
+  });
+
+  let selectedLangId = '';
+  // Custom user regex in language config
+  const userRegexStr = (langConfig as any).atcoderLanguageIdRegex;
+  if (userRegexStr) {
+    const userRegex = new RegExp(userRegexStr, 'i');
+    const matched = options.find(opt => userRegex.test(opt.name));
+    if (matched) {
+      selectedLangId = matched.id;
+    }
+  }
+
+  if (!selectedLangId) {
+    if (langConfig.extension === 'cpp') {
+      const matched = options.find(opt => /C\+\+/i.test(opt.name) && !/Clang/i.test(opt.name)) || 
+                      options.find(opt => /C\+\+/i.test(opt.name));
+      if (matched) selectedLangId = matched.id;
+    } else if (langConfig.extension === 'py') {
+      const matched = options.find(opt => /PyPy3/i.test(opt.name)) || 
+                      options.find(opt => /Python3/i.test(opt.name)) || 
+                      options.find(opt => /Python/i.test(opt.name));
+      if (matched) selectedLangId = matched.id;
+    }
+  }
+
+  if (!selectedLangId) {
+    const extRegex = new RegExp(langConfig.extension, 'i');
+    const matched = options.find(opt => extRegex.test(opt.name));
+    if (matched) {
+      selectedLangId = matched.id;
+    } else if (options.length > 0) {
+      selectedLangId = options[0].id;
+    } else {
+      throw new AtcError('No language options available on AtCoder submit page.');
+    }
+  }
+
+  const postData = new URLSearchParams();
+  postData.append('csrf_token', csrfToken.toString());
+  postData.append('data.TaskScreenName', taskId);
+  postData.append('data.LanguageId', selectedLangId);
+  postData.append('sourceCode', codeContent);
+
+  try {
+    const postRes = await client.post(`/contests/${contestId}/submit`, postData.toString(), {
+      headers: {
+        'Content-Type': 'application/x-www-form-urlencoded',
+        'Referer': `https://atcoder.jp/contests/${contestId}/submit?taskScreenName=${taskId}`
+      },
+      maxRedirects: 0,
+      validateStatus: (status) => status >= 200 && status < 400
+    });
+
+    if (postRes.status !== 302) {
+      const $post = cheerio.load(postRes.data);
+      let alertText = $post('.alert-danger, .alert-warning').text().trim();
+      if (!alertText) {
+        if ($post('.cf-challenge').length > 0) {
+          throw new AtcError(t('submitTurnstileDetected', lang));
+        }
+        throw new AtcError(t('submitRejected', lang));
+      }
+      alertText = alertText.replace(/^×\s*/, '').trim();
+      throw new AtcError(alertText);
+    }
+
+    const meRes = await client.get(`/contests/${contestId}/submissions/me`);
+    const $me = cheerio.load(meRes.data);
+
+    let submissionId = '';
+    $me('table tbody tr').first().find('a').each((_, a) => {
+      const href = $me(a).attr('href');
+      if (href) {
+        const match = href.match(/\/contests\/[^/]+\/submissions\/(\d+)/);
+        if (match && match[1]) {
+          submissionId = match[1];
+        }
+      }
+    });
+
+    if (!submissionId) {
+      throw new AtcError('Submission succeeded but could not retrieve the submission ID.');
+    }
+
+    return {
+      submissionId,
+      url: `/contests/${contestId}/submissions/${submissionId}`
+    };
+  } catch (err: any) {
+    throw new AtcError(`Failed to submit code: ${err.message}`);
+  }
+}