Mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/wukko/cobalt.git synced 2025-05-24 18:42:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

web/queue: decompose ffmpeg construction in createSavePipeline

Browse Source
This commit is contained in:
jj 2025-05-16 14:46:42 +00:00
parent d6ad74d429
commit 3d92a85ba2
No known key found for this signature in database
1 changed files with 78 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

129
web/src/lib/task-manager/queue.ts
View File

@ -59,6 +59,56 @@ const mediaIcons: { [key: string]: CobaltPipelineResultFileType } = {
gif: "image"
}
const makeRemuxArgs = (info: CobaltLocalProcessingResponse) => {
const ffargs = ["-c:v", "copy"];
if (info.type === "merge") {
ffargs.push("-c:a", "copy");
} else if (info.type === "mute") {
ffargs.push("-an");
}
ffargs.push(
...(info.output.metadata ? ffmpegMetadataArgs(info.output.metadata) : [])
);
return ffargs;
}
const makeAudioArgs = (info: CobaltLocalProcessingResponse) => {
if (!info.audio) {
return;
}
const ffargs = [
"-vn",
...(info.audio.copy ? ["-c:a", "copy"] : ["-b:a", `${info.audio.bitrate}k`]),
...(info.output.metadata ? ffmpegMetadataArgs(info.output.metadata) : [])
];
if (info.audio.format === "mp3" && info.audio.bitrate === "8") {
ffargs.push("-ar", "12000");
}
if (info.audio.format === "opus") {
ffargs.push("-vbr", "off")
}
const outFormat = info.audio.format === "m4a" ? "ipod" : info.audio.format;
ffargs.push('-f', outFormat);
return ffargs;
}
const makeGifArgs = () => {
return [
"-vf",
"scale=-1:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse",
"-loop", "0",
"-f", "gif"
];
}
export const createSavePipeline = (info: CobaltLocalProcessingResponse, request: CobaltSaveRequestBody) => {
// this is a pre-queue part of processing,
// so errors have to be returned via a regular dialog
@ -100,57 +150,30 @@ export const createSavePipeline = (info: CobaltLocalProcessingResponse, request:
});
}
let ffargs: string[];
let workerType: 'encode' | 'remux';
if (["merge", "mute"].includes(info.type)) {
const ffargs = ["-c:v", "copy"];
if (info.type === "merge") {
ffargs.push("-c:a", "copy");
} else if (info.type === "mute") {
ffargs.push("-an");
}
ffargs.push(
...(info.output.metadata ? ffmpegMetadataArgs(info.output.metadata) : [])
);
pipeline.push({
worker: "remux",
workerId: crypto.randomUUID(),
parentId,
workerArgs: {
files: [],
ffargs,
output: {
type: info.output.type,
format: info.output.filename.split(".").pop(),
},
},
});
workerType = "remux";
ffargs = makeRemuxArgs(info);
} else if (info.type === "audio") {
if (!info.audio) {
const args = makeAudioArgs(info);
if (!args) {
return error("pipeline.missing_response_data");
}
const ffargs = [
"-vn",
...(info.audio.copy ? ["-c:a", "copy"] : ["-b:a", `${info.audio.bitrate}k`]),
...(info.output.metadata ? ffmpegMetadataArgs(info.output.metadata) : [])
];
if (info.audio.format === "mp3" && info.audio.bitrate === "8") {
ffargs.push("-ar", "12000");
workerType = "encode";
ffargs = args;
} else if (info.type === "gif") {
workerType = "encode";
ffargs = makeGifArgs();
} else {
throw new Error("unknown work type: " + info.type);
}
if (info.audio.format === "opus") {
ffargs.push("-vbr", "off")
}
const outFormat = info.audio.format === "m4a" ? "ipod" : info.audio.format;
ffargs.push('-f', outFormat);
pipeline.push({
worker: "encode",
worker: workerType,
workerId: crypto.randomUUID(),
parentId,
workerArgs: {
@ -162,26 +185,6 @@ export const createSavePipeline = (info: CobaltLocalProcessingResponse, request:
},
},
});
} else if (info.type === "gif") {
pipeline.push({
worker: "encode",
workerId: crypto.randomUUID(),
parentId,
workerArgs: {
files: [],
ffargs: [
"-vf",
"scale=-1:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse",
"-loop", "0",
"-f", "gif"
],
output: {
type: info.output.type,
format: info.output.filename.split(".").pop(),
},
},
});
}
addItem({
id: parentId,