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);