main.js

#!/usr/bin/env node

const readline = require("readline");
const yargs = require("yargs");
const fs = require("fs");
const util = require("util");
const Puppets = require("./puppets");
const logger = require("./log");

const argv = yargs.argv;
const readFile = util.promisify(fs.readFile);

const parseArgs = () => {
  if (!argv.site || !argv["rate-limit"] || !argv["max-tabs"]) {
    logger.info(
      'Usage:\nnode main.js --site="localhost:3000" --rate-limit=60 --max-tabs=10'
    );
    process.exit(0);
  }
  return argv;
};

const readFileAsString = async (filePath) => {
  try {
    const data = await readFile(filePath, "utf8");
    return data;
  } catch (error) {
    logger.error(`Error reading file: ${error.message}`);
    throw error;
  }
};

const getRegistrationPageInnerText = async () => {
  if (argv.test && argv.test !== "false") {
    return await readFileAsString("resources/test.txt");
  } else {
    return await readFileAsString("resources/live.txt");
  }
};

const setupKeyPressHandler = (tabs) => {
  readline.emitKeypressEvents(process.stdin);
  process.stdin.on("keypress", (str, key) => {
    if (key.ctrl && key.name === "c") {
      tabs
        .closeTabs()
        .then(() => process.exit(0))
        .catch((error) => {
          logger.error(`Error closing tabs: ${error.message}`);
          process.exit(1);
        });
    } else if (key.name === "enter") {
      tabs.setPaused(!tabs.getPaused());
    }
  });
};

const run = async () => {
  parseArgs();
  const registrationPageInnerText = await getRegistrationPageInnerText();
  const tabs = new Puppets(
    argv.site,
    argv["rate-limit"],
    registrationPageInnerText
  );

  setupKeyPressHandler(tabs);

  await tabs.initializeTabs(argv["max-tabs"]);
  await tabs.loadPagesAtRate();
};

run();