import { CheckBoxChoice, CheckBoxOptions } from "types";
import chalk from "chalk";
import readline from "readline";
import { PromptBase } from "../input/PromptBase";

const renderChoice = (choice: string, value: boolean, newline = true) => {
  process.stdout.write(
    ` ${chalk.greenBright(value == true ? "◉" : "○")} ` +
      choice +
      (newline ? "\n" : "")
  );
  process.stdout.cursorTo(0);
};

const CheckBox = (options: CheckBoxOptions): Promise<CheckBoxChoice[]> => {
  return new Promise((res, rej) => {
    const choices: CheckBoxChoice[] = options.choices.map((choice) => {
      if ((choice as CheckBoxChoice).name) return choice;

      return { name: choice, value: false };
    }) as CheckBoxChoice[];
    PromptBase({ prompt: options.prompt });
    process.stdout.write("\n\n");
    let newlines = 0;
    options.choices.forEach((option, index) => {
      const choice = (option as CheckBoxChoice).name || (option as string);
      renderChoice(
        index == 0 ? chalk.cyan(choice) : choice,
        (option as CheckBoxChoice).value
          ? (option as CheckBoxChoice).value
          : false
      );
      newlines++;
    });
    process.stdout.write(chalk.green("   done\n"));
    newlines++;

    process.stdout.write(
      `\n\n${chalk.dim(
        "   use arrow keys to select option, enter to finalise\n"
      )}`
    );
    newlines += 3;

    process.stdout.moveCursor(0, -newlines);
    process.stdout.cursorTo(0);

    process.stdin.setRawMode(true);
    readline.emitKeypressEvents(process.stdin);

    let currentY = 0;
    const callback = (data: Buffer, key: any) => {
      if (key && key.name == "c" && key.ctrl) {
        process.exit(0);
      } else if (key.name == "return") {
        if (currentY == choices.length) {
          process.stdout.moveCursor(0, -newlines - currentY + 5);
          for (let i = 0; i < newlines; i++) {
            process.stdout.clearLine(0);
            process.stdout.moveCursor(0, 1);
          }
          process.stdout.moveCursor(0, -newlines - 1);

          process.stdin.setRawMode(false);
          process.stdin.removeListener("keypress", callback);

          res(choices);
        } else {
          choices[currentY].value = !choices[currentY].value;
          process.stdout.clearLine(0);
          renderChoice(choices[currentY].name, choices[currentY].value, false);
        }
      } else {
        let yBefore = currentY;
        if (key.name == "up") {
          currentY = Math.max(0, currentY - 1);
        } else if (key.name == "down") {
          currentY = Math.min(currentY + 1, choices.length);
        }

        process.stdout.clearLine(0);
        if (yBefore != choices.length) {
          renderChoice(choices[yBefore].name, choices[yBefore].value, false);
        } else {
          process.stdout.write(chalk.green("   done"));
          process.stdout.cursorTo(0);
        }

        process.stdout.moveCursor(0, currentY - yBefore);

        process.stdout.clearLine(0);
        if (currentY != choices.length) {
          renderChoice(
            chalk.cyan(choices[currentY].name),
            choices[currentY].value,
            false
          );
        } else {
          process.stdout.write("   " + chalk.greenBright.underline("done"));
          process.stdout.cursorTo(0);
        }
      }
    };
    process.stdin.on("keypress", callback);
  });
};

(globalThis as any).CheckBox = CheckBox;

export { CheckBox };