refactor: split subtitle processing into discrete functions with error handling

This commit is contained in:
Esa Kataja
2025-08-14 16:39:34 +03:00
parent 17c038aa08
commit 73fab48815
+58 -28
View File
@@ -19,21 +19,62 @@ const lang_codes = {
"tur": "tr", "tur": "tr",
}; };
function main() { function subtitlesLookup(file: string): videoFile {
console.log("Looking up subtitles from " + file);
const command = new Deno.Command("mkvmerge", { const command = new Deno.Command("mkvmerge", {
args: ["-J", input], args: ["-J", file],
}); });
const { code, stderr, stdout } = command.outputSync();
const { code, stdout, stderr } = command.outputSync();
const mkvmerge_output = JSON.parse(
new TextDecoder().decode(stdout),
) as videoFile;
const error = new TextDecoder().decode(stderr); const error = new TextDecoder().decode(stderr);
if (code !== 0) { if (code !== 0) {
console.error(error); console.error(error);
return; throw new Error(error);
} }
return JSON.parse(new TextDecoder().decode(stdout)) as videoFile;
}
function ripSubtitles(file: string, languages: Record<string, string>) {
console.log("Ripping subtitles from " + file);
const command = new Deno.Command("mkvextract", {
args: [
"tracks",
file,
...Object.entries(languages).map(([id, lang]) => `${id}:${lang}`),
],
});
const { code, stderr } = command.outputSync();
const error = new TextDecoder().decode(stderr);
if (code !== 0) {
console.error(error);
throw new Error(error);
}
}
async function convertSubtitles(languages: Record<string, string>) {
console.log("Converting subtitles");
const pool: Deno.ChildProcess[] = [];
for (const [_, lang] of Object.entries(languages)) {
const child = new Deno.Command("vobsub2srt", {
args: ["-l", lang, "--blacklist", "|", lang],
}).spawn();
pool.push(child);
}
for (const child of pool) {
await child.status;
}
}
function cleanUp(languages: Record<string, string>) {
console.log("Cleaning up");
for (const [_, lang] of Object.entries(languages)) {
Deno.remove(lang + ".srt");
Deno.remove(lang + ".idx");
Deno.remove(lang + ".sub");
}
}
async function main() {
const mkvmerge_output = subtitlesLookup(input);
const languages_in_file: Record<string, string> = {}; const languages_in_file: Record<string, string> = {};
for (const track of mkvmerge_output.tracks) { for (const track of mkvmerge_output.tracks) {
if (!(track.type === "subtitles")) { if (!(track.type === "subtitles")) {
@@ -47,21 +88,9 @@ function main() {
} }
languages_in_file[track.id] = lang; languages_in_file[track.id] = lang;
} }
const formatted_language_string = Object.entries(languages_in_file) ripSubtitles(input, languages_in_file);
.map(([id, lang]) => `${id}:${lang}`);
const rip_cmd = new Deno.Command("mkvextract", {
args: ["tracks", input, ...formatted_language_string],
});
rip_cmd.outputSync();
for (const [_, lang] of Object.entries(languages_in_file)) { await convertSubtitles(languages_in_file);
const convert_cmd = new Deno.Command("vobsub2srt", {
args: ["-l", lang, "--blacklist", "|", lang],
});
convert_cmd.outputSync();
Deno.remove(lang + ".idx");
Deno.remove(lang + ".sub");
}
const ffmpeg_cmd_raw = []; const ffmpeg_cmd_raw = [];
const subTracksById = new Map( const subTracksById = new Map(
@@ -104,13 +133,14 @@ function main() {
const ffmpeg_error = new TextDecoder().decode(ffmpeg_stderr); const ffmpeg_error = new TextDecoder().decode(ffmpeg_stderr);
if (ffmpeg_code !== 0) { if (ffmpeg_code !== 0) {
console.error(ffmpeg_error); console.error(ffmpeg_error);
return; throw new Error(ffmpeg_error);
}
for (const [_, lang] of Object.entries(languages_in_file)) {
Deno.remove(lang + ".srt");
} }
cleanUp(languages_in_file);
} }
if (import.meta.main) { if (import.meta.main) {
main(); main().catch((e) => {
console.error(e);
Deno.exit(1);
});
} }