add_problems.ts

import {
  createWriteStream,
  mkdirSync,
  readdirSync,
  readFileSync,
  writeFileSync,
} from 'fs';
import * as prettier from 'prettier';
import { ProblemData } from './ProblemData';

process.chdir(__dirname); // aligns require paths and write paths
const ids = readFileSync('../ids.log').toString().split('\n');
const EXTRAPROBLEMS_PATH = '../content/extraProblems.json';
const extraProblems = require(EXTRAPROBLEMS_PATH);
const DIVTOPROBS_PATH =
  '../src/components/markdown/ProblemsList/DivisionList/div_to_probs.json';
const div_to_probs = require(DIVTOPROBS_PATH) as { [key: string]: string[][] };
const IDTOSOL_PATH =
  '../src/components/markdown/ProblemsList/DivisionList/id_to_sol.json';
const id_to_sol = require(IDTOSOL_PATH) as { [key: string]: string };
const problems = require('./problems.json') as { [key: string]: ProblemData };
if (!readdirSync('.').find(x => x == 'out')) mkdirSync('out');
const report = createWriteStream('out/report.txt');
report.write('added problems:\n```\n');
for (const id in problems) {
  if (
    id == '742' ||
    problems[id].source.year < 2015 ||
    (problems[id].source.year == 2015 &&
      problems[id].source.contest != 'December')
  )
    continue; // id 742: modern art :(
  if (!ids.find(x => x == `usaco-${id}`)) {
    extraProblems.EXTRA_PROBLEMS.push({
      uniqueId: `usaco-${id}`,
      name: problems[id].title.name,
      url: problems[id].url,
      source: problems[id].source.division,
      difficulty: 'N/A',
      isStarred: false,
      tags: [],
      solutionMetadata: {
        kind: 'USACO',
        usacoId: id,
      },
    });
    report.write(
      // example: 'id 1355: Haybale Distribution (#3 from 2023 December Gold)'
      `id ${id}: ${problems[id].title.name} (#${problems[id].title.place} from ${problems[id].source.sourceString})\n`
    );
  }
  if (!div_to_probs[problems[id].source.division].find(x => x[0] == id)) {
    //example: ["1352", "2023 December", "Target Practice"]
    div_to_probs[problems[id].source.division].push([
      id,
      `${problems[id].source.year} ${problems[id].source.contest}`,
      problems[id].title.name,
    ]);
  }
  const mon = // dec, jan, feb, open
    problems[id].source.contest === 'US Open'
      ? 'open'
      : problems[id].source.contest.substring(0, 3).toLowerCase();
  if (!Object.keys(id_to_sol).find(x => x == id)) {
    id_to_sol[id] = `sol_prob${problems[id].title.place}_${problems[
      id
    ].source.division.toLowerCase()}_${mon}${problems[id].source.year
      .toString()
      .substring(2)}.html`;
  }
}
report.write('```\n');
for (const div in div_to_probs) {
  // sort by id
  div_to_probs[div].sort((a, b) => {
    return Number(a[0]) - Number(b[0]);
  });
}
(async () => {
  writeFileSync(
    EXTRAPROBLEMS_PATH,
    await prettier.format(JSON.stringify(extraProblems, null, 2), {
      parser: 'json',
      trailingComma: 'es5',
    })
  );
  writeFileSync(
    DIVTOPROBS_PATH,
    await prettier.format(JSON.stringify(div_to_probs, null, 2), {
      parser: 'json',
      trailingComma: 'es5',
    })
  );
  writeFileSync(
    IDTOSOL_PATH,
    await prettier.format(JSON.stringify(id_to_sol, null, 2), {
      parser: 'json',
      trailingComma: 'es5',
    })
  );
})();