online-judge-supporter/src/parser/atcoder.js

A tool to support competitve programmer

import * as cheerio from "cheerio";
import { getHtmlWithLogin } from "./fetcher.js";
import { TestCase } from "../test/testcase.js";
import { ProblemData } from "../test/problem_data.js";
import { Exception } from "../error_handler/error.js";
import { connect } from "puppeteer-real-browser";
import { timeoutDuration } from "../config/load_config.js";

const atcoder_sample_input_regex = /^Sample Input [0-9]+/;
const atcoder_sample_output_regex = /^Sample Output [0-9]+/;
const atcoder_tl_ml_regex = /\s*Time Limit:\s*([\d.]+)\s*sec\s*\/\s*Memory Limit:\s*(\d+)/;
class Atcoder {
  constructor() {
    this.name = "atcoder";
    this.baseUrl = "https://atcoder.jp";
    this.homePage = "https://atcoder.jp/home";
    this.loginUrl = "https://atcoder.jp/login";
    this.submitLoginButton = "#submit"; // submit button's id
  }
  /**
   * create a full url to task (https://atcoder.jp/contests/abc397/tasks/abc397_e)
   *
   * @param {String} contest_id
   * @param {String} task_id
   */
  convertToUrl(contest_id, task_id) {
    return `https://atcoder.jp/contests/${contest_id}/tasks/${task_id}`;
  }
  getProblem(contest_id, task_id) {
    return this.getProblemFromUrl(this.convertToUrl(contest_id, task_id));
  }
  /**
   * fetching problem date from url, including testcases
   *
   * @param {String} problem_url
   * @return {ProblemData} problem_data with testcases
   */
  getProblemFromUrl(url) {
    const call_back = (html) => {
      const $ = cheerio.load(html);
      let test_data = new ProblemData();
      let current_test = new TestCase();
      $(".lang-en")
        .find(".part")
        .filter((_, el) => {
          let name = $(el).find("h3").text();
          if (atcoder_sample_input_regex.test(name)) {
            current_test.input = $(el).find("pre").text();
          } else if (atcoder_sample_output_regex.test(name)) {
            current_test.output = $(el).find("pre").text();
            test_data.addTestCase(current_test);
            current_test = new TestCase();
          }
        });
      const tl_ml = $("#task-statement")
        .prev()
        .text()
        .replace(/[\t\n]/g, "");
      const [, timeLimit, memoryLimit] = tl_ml.match(atcoder_tl_ml_regex);
      test_data.timeLimit = parseFloat(timeLimit);
      test_data.memoryLimit = parseInt(memoryLimit);
      const name = $("#contest-nav-tabs")
        .next()
        .find("span")
        .contents()
        .first()
        .text();
      test_data.name = name.replace(/[\t\n]/g, "");
      return test_data;
    };
    return new Promise((resolve, reject) => {
      getHtmlWithLogin(url, this)
        .then(call_back)
        .then((obj) => {
          obj.url = url;
          resolve(obj);
        })
        .catch((err) => reject(err));
    });
  }

  async login() {
    try {
      const { page, browser } = await connect({
        headless: false,
      });
      let res;
      try {
        await page.goto(this.loginUrl, { waitUntil: 'networkidle2' });
        res = await page.waitForNavigation({ timeout: timeoutDuration });
      } catch (err) {
        browser.close();
        throw err;
      }
      if (res.url() !== this.homePage) {
        await browser.close();
        throw Exception.loginFailed(this.homePage);
      }
      const cookies = await browser.cookies();
      await browser.close();
      return cookies;
    }
    catch (error) {
      throw error;
    }
  }
}
export { Atcoder };
export default {
  Atcoder
};