feat(node-sub): improve support for series
All checks were successful
Discord / discord commits (push) Has been skipped
All checks were successful
Discord / discord commits (push) Has been skipped
This commit is contained in:
parent
d625b740be
commit
b7b25ea3cc
1 changed files with 105 additions and 99 deletions
|
|
@ -21,11 +21,12 @@ const SPAWN_OPTS = {
|
||||||
* @param languages a comma-separated list of languages (3 letters) to sync the subtitles
|
* @param languages a comma-separated list of languages (3 letters) to sync the subtitles
|
||||||
*/
|
*/
|
||||||
const DIR = process.argv[2];
|
const DIR = process.argv[2];
|
||||||
let langs = process.argv[3].split(',');
|
const LANGS = process.argv[3]?.split(',');
|
||||||
|
let langs: string[];
|
||||||
|
|
||||||
|
|
||||||
// Check if there are 2 params
|
// Check if there are 2 params
|
||||||
if (DIR && langs) {
|
if (DIR && LANGS) {
|
||||||
main();
|
main();
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
|
|
@ -33,9 +34,9 @@ else {
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
const escapePath = (p: string) => p.replaceAll("'", "'\\''");
|
const escapePath = (p: string): string => p.replaceAll("'", "'\\''");
|
||||||
|
|
||||||
function getVideoPath(files: string[]) {
|
function getVideoPath(files: string[]): string[] {
|
||||||
const fileName = DIR.split('/').at(-1) ?? '';
|
const fileName = DIR.split('/').at(-1) ?? '';
|
||||||
|
|
||||||
const videoFiles = files.filter((f) =>
|
const videoFiles = files.filter((f) =>
|
||||||
|
|
@ -46,10 +47,10 @@ function getVideoPath(files: string[]) {
|
||||||
process.exit(0);
|
process.exit(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
return `${DIR}/${videoFiles[0]}`;
|
return videoFiles.map((file) => `${DIR}/${file}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
async function backupSubs(files: string[]) {
|
async function backupSubs(files: string[], base: string) {
|
||||||
// Check if backup folder already exists and create it if not
|
// Check if backup folder already exists and create it if not
|
||||||
if (!files.some((f) => f.endsWith('.srt.bak'))) {
|
if (!files.some((f) => f.endsWith('.srt.bak'))) {
|
||||||
await mkdir(`${DIR}/.srt.bak`);
|
await mkdir(`${DIR}/.srt.bak`);
|
||||||
|
|
@ -60,18 +61,19 @@ async function backupSubs(files: string[]) {
|
||||||
|
|
||||||
// Remove synced subtitles from the list to sync
|
// Remove synced subtitles from the list to sync
|
||||||
// langs - backups
|
// langs - backups
|
||||||
langs = langs.filter((n) => !backups
|
langs = langs
|
||||||
.some((s) => {
|
.filter((n) => !backups
|
||||||
const l2 = s.split('.').at(-2) ?? '';
|
.filter((l) => l.includes(base))
|
||||||
const l3 = ISO6391To3.get(l2);
|
.some((s) => {
|
||||||
|
const l2 = s.split('.').at(-2) ?? '';
|
||||||
|
const l3 = ISO6391To3.get(l2);
|
||||||
|
|
||||||
return n === l3;
|
return n === l3;
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (langs.length === 0) {
|
if (langs.length === 0) {
|
||||||
console.warn('Subtitles have already been synced');
|
console.warn(`Subtitles have already been synced for ${base}`);
|
||||||
process.exit(0);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -93,97 +95,101 @@ async function runSubSync(
|
||||||
async function main() {
|
async function main() {
|
||||||
const files = await readDir(DIR);
|
const files = await readDir(DIR);
|
||||||
|
|
||||||
const VIDEO = getVideoPath(files);
|
const VIDEO_FILES = getVideoPath(files);
|
||||||
const BASE_NAME = VIDEO.split('/').at(-1)?.replace(/\.[^.]*$/, '');
|
|
||||||
|
|
||||||
backupSubs(files);
|
VIDEO_FILES.forEach((VIDEO) => {
|
||||||
|
langs = LANGS;
|
||||||
|
const BASE_NAME = VIDEO.split('/').at(-1)!.replace(/\.[^.]*$/, '');
|
||||||
|
|
||||||
// ffprobe the video file to see available audio tracks
|
backupSubs(files, BASE_NAME);
|
||||||
ffProbe(VIDEO, (_e, data) => {
|
|
||||||
if (!data?.streams) {
|
|
||||||
console.error('Couldn\'t find streams in video file');
|
|
||||||
process.exit(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
const AVAIL_LANGS = data.streams
|
// ffprobe the video file to see available audio tracks
|
||||||
.filter((s) => s.codec_type === 'audio')
|
ffProbe(VIDEO, (_e, data) => {
|
||||||
.map((s) => s['tags'] && s['tags']['language']);
|
if (!data?.streams) {
|
||||||
|
console.error('Couldn\'t find streams in video file');
|
||||||
// Sync subtitles one by one
|
|
||||||
langs.forEach(async(lang) => {
|
|
||||||
const FILE_NAME = `${BASE_NAME}.${ISO6393To1.get(lang)}.srt`;
|
|
||||||
const IN_FILE = `${DIR}/.srt.bak/${FILE_NAME}`;
|
|
||||||
const OUT_FILE = `${DIR}/${FILE_NAME}`;
|
|
||||||
|
|
||||||
const cmd = [
|
|
||||||
'--cli sync',
|
|
||||||
`--sub-lang ${lang}`,
|
|
||||||
|
|
||||||
`--ref-stream-by-lang ${AVAIL_LANGS.includes(lang) ?
|
|
||||||
lang :
|
|
||||||
AVAIL_LANGS[0]}`,
|
|
||||||
'--ref-stream-by-type "audio"',
|
|
||||||
|
|
||||||
`--sub '${escapePath(IN_FILE)}'`,
|
|
||||||
`--out '${escapePath(OUT_FILE)}'`,
|
|
||||||
`--ref '${escapePath(VIDEO)}'`,
|
|
||||||
];
|
|
||||||
|
|
||||||
if (files.includes(FILE_NAME)) {
|
|
||||||
await mv(OUT_FILE, IN_FILE);
|
|
||||||
|
|
||||||
runSubSync(cmd, IN_FILE, OUT_FILE);
|
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
let subs = data.streams.filter((s) => {
|
const AVAIL_LANGS = data.streams
|
||||||
return s['tags'] &&
|
.filter((s) => s.codec_type === 'audio')
|
||||||
s['tags']['language'] &&
|
.map((s) => s['tags'] && s['tags']['language']);
|
||||||
s['tags']['language'] === lang &&
|
|
||||||
s.codec_type === 'subtitle';
|
// Sync subtitles one by one
|
||||||
|
langs.forEach(async(lang) => {
|
||||||
|
const FILE_NAME = `${BASE_NAME}.${ISO6393To1.get(lang)}.srt`;
|
||||||
|
const IN_FILE = `${DIR}/.srt.bak/${FILE_NAME}`;
|
||||||
|
const OUT_FILE = `${DIR}/${FILE_NAME}`;
|
||||||
|
|
||||||
|
const cmd = [
|
||||||
|
'--cli sync',
|
||||||
|
`--sub-lang ${lang}`,
|
||||||
|
|
||||||
|
`--ref-stream-by-lang ${AVAIL_LANGS.includes(lang) ?
|
||||||
|
lang :
|
||||||
|
AVAIL_LANGS[0]}`,
|
||||||
|
'--ref-stream-by-type "audio"',
|
||||||
|
|
||||||
|
`--sub '${escapePath(IN_FILE)}'`,
|
||||||
|
`--out '${escapePath(OUT_FILE)}'`,
|
||||||
|
`--ref '${escapePath(VIDEO)}'`,
|
||||||
|
];
|
||||||
|
|
||||||
|
if (files.includes(FILE_NAME)) {
|
||||||
|
await mv(OUT_FILE, IN_FILE);
|
||||||
|
|
||||||
|
runSubSync(cmd, IN_FILE, OUT_FILE);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
let subs = data.streams.filter((s) => {
|
||||||
|
return s['tags'] &&
|
||||||
|
s['tags']['language'] &&
|
||||||
|
s['tags']['language'] === lang &&
|
||||||
|
s.codec_type === 'subtitle';
|
||||||
|
});
|
||||||
|
|
||||||
|
const pgs = subs.filter((s) => s.codec_name === 'hdmv_pgs_subtitle');
|
||||||
|
|
||||||
|
// If we only have PGS subs, warn user
|
||||||
|
if (pgs.length === subs.length) {
|
||||||
|
console.warn(`No SRT subtitle tracks were found for ${lang}`);
|
||||||
|
}
|
||||||
|
// Remove PGS streams from subs
|
||||||
|
subs = subs.filter((s) => s.codec_name !== 'hdmv_pgs_subtitle');
|
||||||
|
|
||||||
|
// Prefer normal subs
|
||||||
|
if (subs.length !== 1) {
|
||||||
|
subs = subs.filter((s) => s.disposition?.forced === 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (subs.length === 0) {
|
||||||
|
console.warn(`No subtitle tracks were found for ${lang}`);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
// Extract subtitle
|
||||||
|
spawn('ffmpeg', [
|
||||||
|
'-i', `'${escapePath(VIDEO)}'`,
|
||||||
|
'-map', `"0:${subs[0].index}"`, `'${escapePath(IN_FILE)}'`,
|
||||||
|
], SPAWN_OPTS);
|
||||||
|
|
||||||
|
// Delete subtitle from video
|
||||||
|
spawn('mv', [
|
||||||
|
`'${escapePath(VIDEO)}'`,
|
||||||
|
`'${escapePath(VIDEO)}.bak'`,
|
||||||
|
], SPAWN_OPTS);
|
||||||
|
|
||||||
|
spawn('ffmpeg', [
|
||||||
|
'-i', `'${escapePath(VIDEO)}.bak'`,
|
||||||
|
'-map', '0',
|
||||||
|
'-map', `-0:${subs[0].index}`,
|
||||||
|
'-c', 'copy', `'${escapePath(VIDEO)}'`,
|
||||||
|
], SPAWN_OPTS);
|
||||||
|
|
||||||
|
spawn('rm', [`'${escapePath(VIDEO)}.bak'`], SPAWN_OPTS);
|
||||||
|
|
||||||
|
// Sync extracted subtitle
|
||||||
|
runSubSync(cmd, IN_FILE, OUT_FILE);
|
||||||
|
}
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
const pgs = subs.filter((s) => s.codec_name === 'hdmv_pgs_subtitle');
|
|
||||||
|
|
||||||
// If we only have PGS subs, warn user
|
|
||||||
if (pgs.length === subs.length) {
|
|
||||||
console.warn(`No SRT subtitle tracks were found for ${lang}`);
|
|
||||||
}
|
|
||||||
// Remove PGS streams from subs
|
|
||||||
subs = subs.filter((s) => s.codec_name !== 'hdmv_pgs_subtitle');
|
|
||||||
|
|
||||||
// Prefer normal subs
|
|
||||||
if (subs.length !== 1) {
|
|
||||||
subs = subs.filter((s) => s.disposition?.forced === 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (subs.length === 0) {
|
|
||||||
console.warn(`No subtitle tracks were found for ${lang}`);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
// Extract subtitle
|
|
||||||
spawn('ffmpeg', [
|
|
||||||
'-i', `'${escapePath(VIDEO)}'`,
|
|
||||||
'-map', `"0:${subs[0].index}"`, `'${escapePath(IN_FILE)}'`,
|
|
||||||
], SPAWN_OPTS);
|
|
||||||
|
|
||||||
// Delete subtitle from video
|
|
||||||
spawn('mv', [
|
|
||||||
`'${escapePath(VIDEO)}'`,
|
|
||||||
`'${escapePath(VIDEO)}.bak'`,
|
|
||||||
], SPAWN_OPTS);
|
|
||||||
|
|
||||||
spawn('ffmpeg', [
|
|
||||||
'-i', `'${escapePath(VIDEO)}.bak'`,
|
|
||||||
'-map', '0',
|
|
||||||
'-map', `-0:${subs[0].index}`,
|
|
||||||
'-c', 'copy', `'${escapePath(VIDEO)}'`,
|
|
||||||
], SPAWN_OPTS);
|
|
||||||
|
|
||||||
spawn('rm', [`'${escapePath(VIDEO)}.bak'`], SPAWN_OPTS);
|
|
||||||
|
|
||||||
// Sync extracted subtitle
|
|
||||||
runSubSync(cmd, IN_FILE, OUT_FILE);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue