import readline from "readline";
import { generateTestVideo, EncodingType } from "./ffmpeg";

const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

const questionAsync = (query: string) => {
  return new Promise<string>((resolve) => {
    rl.question(query, resolve);
  });
};

const main = async () => {
  console.log("Testing software encoder: ");
  await generateTestVideo(EncodingType.CPU).catch(console.error);

  const answer = await questionAsync(
    "Would you like to test other hardware encoders? (yes/no): ",
  );

  if (answer.toLowerCase() === "yes") {
    const encoder = await questionAsync(
      "Which hardware encoder would you like to test? (INTEL/NVIDIA/AMD/APPLE): ",
    );
    let selectedEncoder: EncodingType;

    switch (encoder.toUpperCase()) {
      case "INTEL":
        selectedEncoder = EncodingType.INTEL;
        break;
      case "NVIDIA":
        selectedEncoder = EncodingType.NVIDIA;
        break;
      case "AMD":
        selectedEncoder = EncodingType.AMD;
        break;
      case "APPLE":
        selectedEncoder = EncodingType.APPLE;
        break;
      default:
        console.log("Invalid choice. Exiting.");
        rl.close();
        return;
    }

    console.log(`Testing ${selectedEncoder} encoder:`);
    await generateTestVideo(selectedEncoder).catch(console.error);
  } else {
    console.log("Exiting.");
  }

  rl.close();
};

main().catch((err) => {
  console.error("An error occurred:", err);
  rl.close();
});