//SVR.JS LOG VIEWER

const fs = require("fs");
const readline = require("readline");

const args = process.argv;
for (
  let i =
    process.argv[0].indexOf("node") > -1 || process.argv[0].indexOf("bun") > -1
      ? 2
      : 1;
  i < args.length;
  i++
) {
  if (
    args[i] == "-h" ||
    args[i] == "--help" ||
    args[i] == "-?" ||
    args[i] == "/h" ||
    args[i] == "/?"
  ) {
    console.log("SVR.JS log viewer usage:");
    console.log("node logviewer.js [-h] [--help] [-?] [/h] [/?]");
    console.log("-h -? /h /? --help    -- Displays help");
    process.exit(0);
  } else {
    console.log("Unrecognized argument: " + args[i]);
    console.log("SVR.JS log viewer usage:");
    console.log("node logviewer.js [-h] [--help] [-?] [/h] [/?]");
    console.log("-h -? /h /? --help    -- Displays help");
    process.exit(1);
  }
}

const logo = [
  "",
  "",
  "\x1b[0m                                    \x1b[38;5;255m@@@@@@@@@\x1b[0m",
  "\x1b[0m                               ,\x1b[38;5;255m@@@\x1b[38;5;254m&\x1b[38;5;041m/////////\x1b[38;5;254m@\x1b[38;5;255m@@@\x1b[0m,",
  "\x1b[0m                           &\x1b[38;5;255m@@@\x1b[38;5;078m#\x1b[38;5;041m/////////////////\x1b[38;5;115m#\x1b[38;5;255m@@@\x1b[0m%",
  "\x1b[0m                       \x1b[38;5;255m@@@@\x1b[38;5;041m///////////////////////////\x1b[38;5;255m@@@@\x1b[0m",
  "\x1b[0m                  *\x1b[38;5;255m@@@\x1b[38;5;254m&\x1b[38;5;041m///////////////////////////////////\x1b[38;5;255m@@@@\x1b[0m*",
  "\x1b[0m              &\x1b[38;5;255m@@@\x1b[38;5;078m#\x1b[38;5;041m///////////////////////////////////////////\x1b[38;5;115m#\x1b[38;5;255m@@@\x1b[0m%",
  "\x1b[0m          \x1b[38;5;255m@@@@\x1b[38;5;041m/////////////////////////////////////////////////////\x1b[38;5;255m@@@@\x1b[0m",
  "\x1b[0m     /\x1b[38;5;255m@@@\x1b[38;5;254m@\x1b[38;5;041m/////////////////////////////////////////////////////////////\x1b[38;5;255m@@@@\x1b[0m,",
  "\x1b[0m &\x1b[38;5;255m@@@\x1b[38;5;114m#\x1b[38;5;041m/////////////////////////////////////////////////////////////////////\x1b[38;5;115m#\x1b[38;5;255m@@@\x1b[0m%",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m.\x1b[38;5;255m@@@@\x1b[38;5;041m/////////////////////////////////////////////////////////////////\x1b[38;5;255m@@@@\x1b[38;5;235m.\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m.....\x1b[38;5;151m%\x1b[38;5;255m@@@\x1b[38;5;115m#\x1b[38;5;041m///////////////////////////////////////////////////////\x1b[38;5;151m%\x1b[38;5;255m@@@\x1b[38;5;109m#\x1b[38;5;235m.....\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m.........\x1b[38;5;238m,\x1b[38;5;255m@@@@\x1b[38;5;041m///////////////////////////////////////////////\x1b[38;5;255m@@@@\x1b[38;5;236m,\x1b[38;5;235m.........\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m..............\x1b[38;5;255m@@@@\x1b[38;5;041m//////////////////////////////////////\x1b[38;5;077m/\x1b[38;5;255m@@@@\x1b[38;5;235m..............\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m..................\x1b[38;5;151m%\x1b[38;5;255m@@@\x1b[38;5;115m#\x1b[38;5;041m/////////////////////////////\x1b[38;5;151m%\x1b[38;5;255m@@@\x1b[38;5;248m#\x1b[38;5;235m..................\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m......................\x1b[38;5;238m,\x1b[38;5;255m@@@\x1b[38;5;254m@\x1b[38;5;041m/////////////////////\x1b[38;5;255m@@@@\x1b[38;5;237m,\x1b[38;5;235m.............\x1b[38;5;041m///\x1b[38;5;236m.\x1b[38;5;235m.....\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;254m&\x1b[38;5;235m...........................\x1b[38;5;255m@@@@\x1b[38;5;041m////////////\x1b[38;5;077m/\x1b[38;5;255m@@@@\x1b[38;5;235m................\x1b[38;5;041m/////\x1b[38;5;035m*\x1b[38;5;235m.....\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m...............................\x1b[38;5;007m%\x1b[38;5;255m@@@\x1b[38;5;114m#\x1b[38;5;041m///\x1b[38;5;115m#\x1b[38;5;255m@@@\x1b[38;5;249m#\x1b[38;5;235m...................\x1b[38;5;041m/////\x1b[38;5;035m/\x1b[38;5;235m......\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m#\x1b[38;5;255m@\x1b[38;5;188m&\x1b[38;5;235m...................................\x1b[38;5;237m,\x1b[38;5;255m@@@\x1b[38;5;237m,\x1b[38;5;235m.......................\x1b[38;5;041m///\x1b[38;5;035m/\x1b[38;5;235m........\x1b[38;5;188m&\x1b[38;5;255m@",
  "\x1b[0m,\x1b[38;5;255m@@@\x1b[38;5;251m%\x1b[38;5;235m..................................\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m.........\x1b[38;5;035m*\x1b[38;5;041m///\x1b[38;5;035m/\x1b[38;5;235m....................\x1b[38;5;253m&\x1b[38;5;255m@@@",
  "\x1b[0m     \x1b[38;5;255m@@@@\x1b[38;5;235m..............................\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m........\x1b[38;5;041m/////\x1b[38;5;029m*\x1b[38;5;235m................\x1b[38;5;255m@@@@\x1b[0m",
  "\x1b[0m         \x1b[38;5;254m@\x1b[38;5;255m@@@\x1b[38;5;238m,\x1b[38;5;235m.........................\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m.......\x1b[38;5;041m/////\x1b[38;5;029m*\x1b[38;5;235m............\x1b[38;5;240m*\x1b[38;5;255m@@@\x1b[0m&",
  "\x1b[0m             *\x1b[38;5;255m@@@\x1b[38;5;251m%\x1b[38;5;235m.....................\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m.......\x1b[38;5;029m*\x1b[38;5;041m//\x1b[38;5;236m.\x1b[38;5;235m..........\x1b[38;5;252m&\x1b[38;5;255m@@@\x1b[0m/",
  "\x1b[0m                //\x1b[38;5;255m@@@@\x1b[38;5;235m.................\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m.................\x1b[38;5;255m@@@@\x1b[0m//",
  "\x1b[0m            //////////&\x1b[38;5;255m@@@\x1b[38;5;238m,\x1b[38;5;235m............\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m............\x1b[38;5;239m,\x1b[38;5;255m@@@\x1b[0m&//////////",
  "\x1b[0m       .//////////////////(\x1b[38;5;255m@@@\x1b[38;5;251m%\x1b[38;5;235m........\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m........\x1b[38;5;253m&\x1b[38;5;255m@@@\x1b[0m(//////////////////,",
  "\x1b[0m    ///////////////////////////\x1b[38;5;255m@@@@\x1b[38;5;235m....\x1b[38;5;251m%\x1b[38;5;255m@\x1b[38;5;242m/\x1b[38;5;235m....\x1b[38;5;255m@@@@\x1b[0m///////////////////////////",
  "\x1b[0m    ///////////////////////////////&\x1b[38;5;255m@@@\x1b[38;5;252m&\x1b[38;5;255m@\x1b[38;5;246m(\x1b[38;5;255m@@@\x1b[0m&///////////////////////////////",
  "\x1b[0m      /////////////////////////////////////////////////////////////////////",
  "\x1b[0m          .///////////////////////////////////////////////////////////.",
  "\x1b[0m               ///////////////////////////////////////////////////",
  "\x1b[0m                   ///////////////////////////////////////////",
  "\x1b[0m                       ./////////////////////////////////,",
  "\x1b[0m                            /////////////////////////",
  "\x1b[0m                                /////////////////",
  "\x1b[0m                                    ,///////,",
  "",
  "",
  "\x1b[0m",
];

for (let i = 0; i < logo.length; i++) {
  console.log(logo[i]);
}
console.log("Welcome to SVR.JS log viewer");

if (!fs.existsSync(__dirname + "/log")) {
  console.log("No log directory, exiting...");
  process.exit(1);
}

function prompt(options) {
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
    prompt: "logviewer> ",
  });
  console.log("Options:");
  for (let i = 0; i < options.length; i++) {
    console.log("[" + i + "] - " + options[i].name);
  }
  rl.prompt();
  rl.on("line", (line) => {
    const op = line.trim();
    if (op == "") {
      rl.prompt();
      return;
    }
    rl.close();
    if (options[op]) {
      options[op].callback();
    } else {
      console.log("Invalid option.");
      prompt(options);
    }
  });
}

function viewLog(log) {
  if (log[log.length - 1] == "") log.pop();
  if (log[0] == "") log.shift();
  for (let i = 0; i < log.length; i++) {
    if (log[i].indexOf("SERVER REQUEST MESSAGE") != -1) {
      log[i] =
        log[i].replace(
          "SERVER REQUEST MESSAGE",
          "\x1b[34m\x1b[1mSERVER REQUEST MESSAGE\x1b[22m",
        ) + "\x1b[37m\x1b[0m";
    } else if (log[i].indexOf("SERVER RESPONSE MESSAGE") != -1) {
      log[i] =
        log[i].replace(
          "SERVER RESPONSE MESSAGE",
          "\x1b[32m\x1b[1mSERVER RESPONSE MESSAGE\x1b[22m",
        ) + "\x1b[37m\x1b[0m";
    } else if (log[i].indexOf("SERVER RESPONSE ERROR MESSAGE") != -1) {
      log[i] =
        log[i].replace(
          "SERVER RESPONSE ERROR MESSAGE",
          "\x1b[31m\x1b[1mSERVER RESPONSE ERROR MESSAGE\x1b[22m",
        ) + "\x1b[37m\x1b[0m";
    } else if (log[i].indexOf("SERVER ERROR MESSAGE") != -1) {
      log[i] =
        log[i].replace(
          "SERVER ERROR MESSAGE",
          "\x1b[41m\x1b[1mSERVER ERROR MESSAGE\x1b[22m",
        ) + "\x1b[40m\x1b[0m";
    } else if (log[i].indexOf("SERVER WARNING MESSAGE") != -1) {
      log[i] =
        log[i].replace(
          "SERVER WARNING MESSAGE",
          "\x1b[43m\x1b[1mSERVER WARNING MESSAGE\x1b[22m",
        ) + "\x1b[40m\x1b[0m";
    } else if (log[i].indexOf("SERVER MESSAGE") != -1) {
      log[i] = log[i].replace(
        "SERVER MESSAGE",
        "\x1b[1mSERVER MESSAGE\x1b[22m",
      );
    }
    console.log(log[i]);
  }
}

function viewMasterLogs() {
  const logList = fs.readdirSync(__dirname + "/log");
  let masterLogs = [];
  for (var i = 0; i < logList.length; i++) {
    if (logList[i].match(/^master-[0-9]+\.log$/)) {
      masterLogs.push(logList[i]);
    }
  }
  if (masterLogs.length == 0) {
    console.log("No master log.");
    return;
  }
  const latestLogFileName = masterLogs.sort().reverse()[0];
  viewLog(
    fs
      .readFileSync(__dirname + "/log/" + latestLogFileName)
      .toString()
      .split("\n"),
  );
  prompt(mainOptions);
}

function viewWorkerLogs() {
  const logList = fs.readdirSync("log");
  let masterLogs = [];
  for (var i = 0; i < logList.length; i++) {
    if (logList[i].match(/^worker-[0-9]+\.log$/)) {
      masterLogs.push(logList[i]);
    }
  }
  if (masterLogs.length == 0) {
    console.log("No worker logs.");
    return;
  }
  const latestLogFileNames = masterLogs.sort().reverse().slice(0, 5).reverse();
  let log = [];
  for (let i = 0; i < latestLogFileNames.length; i++) {
    let rlog = fs
      .readFileSync("log/" + latestLogFileNames[i])
      .toString()
      .split("\n");
    if (rlog[rlog.length - 1] == "") rlog.pop();
    if (rlog[0] == "") rlog.shift();
    for (let j = 0; j < rlog.length; j++) {
      log.push(rlog[j]);
    }
  }
  log = log.sort();
  viewLog(log);
  prompt(mainOptions);
}

function viewFilteredWorkerLogs(filter) {
  const logList = fs.readdirSync("log");
  let masterLogs = [];
  for (var i = 0; i < logList.length; i++) {
    if (logList[i].match(/^worker-[0-9]+\.log$/)) {
      masterLogs.push(logList[i]);
    }
  }
  if (masterLogs.length == 0) {
    console.log("No worker logs.");
    return;
  }
  const latestLogFileNames = masterLogs.sort().reverse().slice(0, 20).reverse();
  let log = [];
  for (let i = 0; i < latestLogFileNames.length; i++) {
    let rlog = fs
      .readFileSync("log/" + latestLogFileNames[i])
      .toString()
      .split("\n");
    if (rlog[rlog.length - 1] == "") rlog.pop();
    if (rlog[0] == "") rlog.shift();
    for (let j = 0; j < rlog.length; j++) {
      if (rlog[j].indexOf(filter) != -1) log.push(rlog[j]);
    }
  }
  log = log.sort();
  viewLog(log);
  prompt(mainOptions);
}

function viewFilteredWorkerLogsPrompt() {
  var rl2 = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
    prompt: "filter> ",
  });
  console.log("Input filter:");
  rl2.prompt();
  rl2.on("line", (line) => {
    rl2.close();
    viewFilteredWorkerLogs(line);
  });
}

var mainOptions = [
  { name: "View latest master log", callback: viewMasterLogs },
  { name: "View 5 latest worker logs", callback: viewWorkerLogs },
  {
    name: "View filtered worker logs (latest 20 logs)",
    callback: viewFilteredWorkerLogsPrompt,
  },
  {
    name: "Exit log viewer",
    callback: () => {
      console.log("Bye!");
      process.exit(0);
    },
  },
];

prompt(mainOptions);