pdf.js.mirror/test/add_test.mjs

import { calculateMD5 } from "./downloadutils.mjs";
import { execSync } from "child_process";
import fs from "fs";

const testManifest = "test/test_manifest.json";
const pdfFolder = "test/pdfs/";
const gitIgnore = "test/pdfs/.gitignore";

if (process.argv.length < 3) {
  console.log("\nUsage: node add_test.js FILE\n");
  console.log(
    `  Add a PDF as a reference test. FILE must be located in ${pdfFolder}`
  );
  process.exit(1);
}

const file = process.argv[2];
if (!file.startsWith(pdfFolder)) {
  throw new Error(`PDF file must be in '${pdfFolder}' directory.`);
}
if (!fs.existsSync(file)) {
  throw new Error(`PDF file does not exist '${file}'.`);
}

function getRandomArbitrary(min, max) {
  return Math.floor(Math.random() * (max - min) + min);
}

const md5 = await calculateMD5(file);

let contents = fs.readFileSync(gitIgnore, "utf8").split("\n");
const randomLine = getRandomArbitrary(10, contents.length - 2);
contents.splice(randomLine, 0, "!" + file.substring(file.lastIndexOf("/") + 1));
fs.writeFileSync("test/pdfs/.gitignore", contents.join("\n"));

contents = fs.readFileSync(testManifest, "utf8");
const pdf = file.substring(file.lastIndexOf("/") + 1, file.length - 4);
const randomPoint = getRandomArbitrary(100, contents.length - 20);
const bracket = contents.indexOf("},\n", randomPoint);
const out =
  contents.substring(0, bracket) +
  "},\n" +
  `    {  "id": "${pdf}",\n` +
  `       "file": "pdfs/${pdf}.pdf",\n` +
  `       "md5": "${md5}",\n` +
  '       "rounds": 1,\n' +
  '       "type": "eq"\n' +
  "    " +
  contents.substring(bracket);
fs.writeFileSync("test/test_manifest.json", out);
execSync(`git add ${testManifest} ${gitIgnore}`);
execSync(`git add ${file}`);