From 42406483acaf5aa53b265ed0b01b0b0dd976a5ed Mon Sep 17 00:00:00 2001 From: Skye Date: Mon, 11 Nov 2024 18:59:04 -0500 Subject: [PATCH] Initial commit --- .gitignore | 1 + Cargo.lock | 253 ++++++++++++++++++++++++++++++++++++++++++++++++++++ Cargo.toml | 8 ++ src/main.rs | 188 ++++++++++++++++++++++++++++++++++++++ 4 files changed, 450 insertions(+) create mode 100644 .gitignore create mode 100644 Cargo.lock create mode 100644 Cargo.toml create mode 100644 src/main.rs diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..c67305e --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,253 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "anstream" +version = "0.6.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8acc5369981196006228e28809f761875c0327210a891e941f4c683b3a99529b" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "55cc3b69f167a1ef2e161439aa98aed94e6028e5f9a59be9a6ffb47aef1651f9" + +[[package]] +name = "anstyle-parse" +version = "0.2.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b2d16507662817a6a20a9ea92df6652ee4f94f914589377d69f3b21bc5798a9" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "79947af37f4177cfead1110013d678905c37501914fba0efea834c3fe9a8d60c" +dependencies = [ + "windows-sys", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2109dbce0e72be3ec00bed26e6a7479ca384ad226efdd66db8fa2e3a38c83125" +dependencies = [ + "anstyle", + "windows-sys", +] + +[[package]] +name = "clap" +version = "4.5.20" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b97f376d85a664d5837dbae44bf546e6477a679ff6610010f17276f686d867e8" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.20" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "19bc80abd44e4bed93ca373a0704ccbd1b710dc5749406201bb018272808dc54" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4ac6a0c7b1a9e9a5186361f67dfa1b88213572f427fb9ab038efb2bd8c582dab" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1462739cb27611015575c0c11df5df7601141071f07518d56fcc1be504cbec97" + +[[package]] +name = "colorchoice" +version = "1.0.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990" + +[[package]] +name = "either" +version = "1.13.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "60b1af1c220855b6ceac025d3f6ecdd2b7c4894bfe9cd9bda4fbb4bc7c0d4cf0" + +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + +[[package]] +name = "itertools" +version = "0.13.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "413ee7dfc52ee1a4949ceeb7dbc8a33f2d6c088194d9f922fb8318faf1f01186" +dependencies = [ + "either", +] + +[[package]] +name = "proc-macro2" +version = "1.0.89" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f139b0662de085916d1fb67d2b4169d1addddda1919e696f3252b740b629986e" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.37" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + +[[package]] +name = "subtitle-merge" +version = "0.1.0" +dependencies = [ + "clap", + "itertools", +] + +[[package]] +name = "syn" +version = "2.0.87" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "25aa4ce346d03a6dcd68dd8b4010bcb74e54e62c90c573f394c46eae99aba32d" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "unicode-ident" +version = "1.0.13" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e91b56cd4cadaeb79bbf1a5645f6b4f8dc5bde8834ad5894a8db35fda9efa1fe" + +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + +[[package]] +name = "windows-sys" +version = "0.59.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..98de9bd --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "subtitle-merge" +version = "0.1.0" +edition = "2021" + +[dependencies] +clap = { version = "4.5.20", features = ["derive"] } +itertools = "0.13.0" diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..033710c --- /dev/null +++ b/src/main.rs @@ -0,0 +1,188 @@ +use std::{ + ffi::{OsStr, OsString}, + fs::ReadDir, + io, + os::unix::{ffi::OsStrExt, process::CommandExt}, + path::{Path, PathBuf}, + process::{Command, ExitCode, ExitStatus}, +}; + +use clap::Parser; +use itertools::Either; + +#[derive(Parser)] +enum Args { + Single(Single), + Dir(Dir), +} + +#[derive(Parser)] +struct Single { + source: PathBuf, + #[arg(long)] + subtitles: Option, + #[arg(long)] + fonts_dir: Option, + output: PathBuf, +} + +#[derive(Parser)] +struct Dir { + input_dir: PathBuf, + output_dir: PathBuf, +} + +fn find_subs(root: &Path, match_str: &OsStr) -> io::Result> { + for entry in root.read_dir()? { + let entry = entry?; + let type_ = entry.file_type()?; + if type_.is_dir() { + let recursed = find_subs(&entry.path(), match_str)?; + if recursed.is_some() { + return Ok(recursed); + } + } else { + // Assume found a file + let path = entry.path(); + if path + .file_stem() + .unwrap() + .as_bytes() + .windows(match_str.as_bytes().len()) + .any(|subslice| subslice == match_str.as_bytes()) + && path.extension() == Some(OsStr::new("ass")) + { + return Ok(Some(entry.path())); + } + } + } + Ok(None) +} + +fn find_files(root: &Path, extension: &OsStr) -> impl Iterator> { + struct FindFonts { + read_dirs: Vec, + extension: OsString, + } + impl Iterator for FindFonts { + type Item = io::Result; + + fn next(&mut self) -> Option { + loop { + let Some(next) = self.read_dirs.last_mut()?.next() else { + self.read_dirs.pop(); + continue; + }; + let entry = match next { + Err(e) => return Some(Err(e)), + Ok(entry) => entry, + }; + let file_type = entry.file_type(); + let file_type = match file_type { + Err(e) => return Some(Err(e)), + Ok(file_type) => file_type, + }; + if file_type.is_dir() { + match entry.path().read_dir() { + Err(e) => return Some(Err(e)), + Ok(read_dir) => { + self.read_dirs.push(read_dir); + continue; + } + } + } else { + let path = entry.path(); + if path.extension().is_some_and(|ex| ex == self.extension) { + return Some(Ok(path)); + } else { + continue; + } + } + } + } + } + + match root.read_dir() { + Ok(read_dir) => Either::Left(FindFonts { + read_dirs: vec![read_dir], + extension: extension.to_owned(), + }), + Err(e) => Either::Right(std::iter::once(Err(e))), + } +} + +fn convert_single( + Single { + source, + subtitles, + fonts_dir, + output, + }: Single, +) -> io::Result { + let subtitles = if let Some(subtitles) = subtitles { + subtitles + } else { + find_subs(&source.parent().unwrap(), source.file_stem().unwrap())?.unwrap() + }; + let fonts: Vec = if let Some(fonts_dir) = fonts_dir { + fonts_dir + .read_dir()? + .map(|r| r.map(|entry| entry.path())) + .collect::>()? + } else { + find_files(source.parent().unwrap(), OsStr::new("ttf")).collect::>()? + }; + + let mut command = Command::new("mkvmerge"); + command.arg("--output").args([ + output.into_os_string(), + source.into_os_string(), + subtitles.into_os_string(), + ]); + for font in fonts { + command + .args(&["--attachment-mime-type", "font/ttf", "--attach-file"]) + .arg(font.into_os_string()); + } + Ok(command) +} + +fn main() -> io::Result { + let args = Args::parse(); + match args { + Args::Single(single) => { + let mut command = convert_single(single)?; + Err(command.exec()) + } + Args::Dir(dir) => { + let files = find_files(&dir.input_dir, OsStr::new("mkv")); + let mut commands = Vec::new(); + for file in files { + let input_file = file?; + let output_file = dir + .output_dir + .join(input_file.strip_prefix(&dir.input_dir).unwrap()); + let subtitles = find_subs(&dir.input_dir, input_file.file_stem().unwrap())?; + if subtitles.is_none() { + continue; + } + let mut command = convert_single(Single { + source: input_file, + subtitles, + fonts_dir: None, + output: output_file.clone(), + })?; + std::fs::create_dir_all(output_file.parent().unwrap())?; + commands.push(command.spawn()?); + } + let mut status = 0; + for mut child in commands { + let child_status = child.wait()?; + if status == 0 && !child_status.success() { + status = child_status.code().unwrap_or(1); + } + } + Ok(ExitCode::from(status as u8)) + } + } +}