import ffmpeg from "fluent-ffmpeg";
import ffmpegInstaller from "@ffmpeg-installer/ffmpeg";
import ffprobeInstaller from "@ffprobe-installer/ffprobe";
import which from "which";

const getExecutablePath = (
  envVar: string,
  executable: string,
  installer: { path: string },
) => {
  if (process.env[envVar]) {
    return process.env[envVar];
  }

  try {
    return which.sync(executable);
  } catch (error) {
    return installer.path;
  }
};

const ffmpegPath = getExecutablePath(
  "EB_FFMPEG_PATH",
  "ffmpeg",
  ffmpegInstaller,
);
const ffprobePath = getExecutablePath(
  "EB_FFPROBE_PATH",
  "ffprobe",
  ffprobeInstaller,
);

console.log(`Using ffmpeg from path: ${ffmpegPath}`);
console.log(`Using ffprobe from path: ${ffprobePath}`);

ffmpeg.setFfmpegPath(ffmpegPath!);
ffmpeg.setFfprobePath(ffprobePath!);

export enum EncodingType {
  CPU = "libx264",
  NVIDIA = "h264_nvenc",
  AMD = "h264_amf",
  INTEL = "h264_qsv",
  APPLE = "h264_videotoolbox",
}

export const generateTestVideo = async (
  encodingType: EncodingType,
): Promise<void> => {
  console.log(`Generating test video using ${encodingType}...`);

  const startTime = Date.now();
  let totalFrames = 0;

  const outputOptions = [
    "-vf",
    "scale=-2:720",
    "-vcodec",
    encodingType,
    "-c:a",
    "copy",
    "-b:v",
    "5000k",
    "-pix_fmt",
    "yuv420p",
  ];

  // Adjust output options based on encoder for maximum quality
  switch (encodingType) {
    case EncodingType.CPU:
      //outputOptions.push('-crf', '0');
      break;
    case EncodingType.NVIDIA:
      outputOptions.push("-rc", "cqp", "-qp", "0");
      break;
    case EncodingType.AMD:
      outputOptions.push("-qp_i", "0", "-qp_p", "0", "-qp_b", "0");
      break;
    case EncodingType.INTEL:
      outputOptions.push("-global_quality", "1"); // Intel QSV specific setting for high quality
      break;
    case EncodingType.APPLE:
      outputOptions.push("-global_quality", "1");
      break;
  }

  return new Promise<void>((resolve, reject) => {
    ffmpeg()
      .input("unknown_replay_2023.10.29-22.57-00.00.38.103-00.01.00.016.mp4")
      .inputFormat("mp4")
      .outputOptions(outputOptions)
      .output(`720p-test-${encodingType}.mp4`)
      .on("progress", (progress) => {
        totalFrames = progress.frames;
      })
      .on("end", () => {
        const elapsedTime = (Date.now() - startTime) / 1000; // Convert to seconds
        const avgFps = totalFrames / elapsedTime;

        console.log(
          `720p copy complete using ${encodingType}, took ${
            Date.now() - startTime
          }ms to complete`,
        );
        console.log(`Average FPS for the entire process: ${avgFps.toFixed(2)}`);

        resolve();
      })
      .on("error", (e) => reject(new Error(e)))
      .run();
  });
};

// Test commands (uncomment to use)
// generateTestVideo(EncodingType.CPU).catch(console.error);
// generateTestVideo(EncodingType.AMD).catch(console.error);
// generateTestVideo(EncodingType.INTEL).catch(console.error);
// generateTestVideo(EncodingType.NVIDIA).catch(console.error);
// generateTestVideo(EncodingType.APPLE).catch(console.error);