import { existsSync } from "node:fs"
import path from "node:path"

import { UUID } from "@aitube/clap"
import ffmpeg, { FfmpegCommand } from "fluent-ffmpeg"
import { getRandomDirectory } from "@aitube/io"

import { getMediaInfo } from "../analyze/getMediaInfo"

export type ConcatenateVideoOutput = {
  filepath: string;
  durationInSec: number;
}

export async function concatenateVideos({
  output,
  videoFilePaths = [],
}: {
  output?: string;

  // those are videos PATHs, not base64 strings!
  videoFilePaths: string[];
}): Promise<ConcatenateVideoOutput> {
  if (!Array.isArray(videoFilePaths)) {
    throw new Error("Videos must be provided in an array")
  }

  videoFilePaths = videoFilePaths.filter((videoPath) => existsSync(videoPath))

  // Create a temporary working directory
  const tempDir = await getRandomDirectory()

  const filePath = output ? output : path.join(tempDir, `${UUID()}.mp4`)

  if (!filePath) {
    throw new Error("Failed to generate a valid temporary file path")
  }

  let cmd: FfmpegCommand = ffmpeg()

  videoFilePaths.forEach((video) => {
    cmd = cmd.addInput(video)
  })

  return new Promise<{ filepath: string; durationInSec: number }>(
    (resolve, reject) => {
      cmd
        .on('error', reject)
        .on('end', async () => {
          try {
            const { durationInSec } = await getMediaInfo(filePath);
            resolve({ filepath: filePath, durationInSec });
          } catch (err) {
            reject(err);
          }
        })
        .mergeToFile(filePath, tempDir);
    }
  )
}