jbilcke-hf's picture
jbilcke-hf HF staff
update dependencies
051e9e4
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);
}
)
}