Initial commit
This commit is contained in:
commit
42406483ac
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
/target
|
||||||
253
Cargo.lock
generated
Normal file
253
Cargo.lock
generated
Normal file
|
|
@ -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"
|
||||||
8
Cargo.toml
Normal file
8
Cargo.toml
Normal file
|
|
@ -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"
|
||||||
188
src/main.rs
Normal file
188
src/main.rs
Normal file
|
|
@ -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<PathBuf>,
|
||||||
|
#[arg(long)]
|
||||||
|
fonts_dir: Option<PathBuf>,
|
||||||
|
output: PathBuf,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Parser)]
|
||||||
|
struct Dir {
|
||||||
|
input_dir: PathBuf,
|
||||||
|
output_dir: PathBuf,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn find_subs(root: &Path, match_str: &OsStr) -> io::Result<Option<PathBuf>> {
|
||||||
|
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<Item = io::Result<PathBuf>> {
|
||||||
|
struct FindFonts {
|
||||||
|
read_dirs: Vec<ReadDir>,
|
||||||
|
extension: OsString,
|
||||||
|
}
|
||||||
|
impl Iterator for FindFonts {
|
||||||
|
type Item = io::Result<PathBuf>;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
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<Command> {
|
||||||
|
let subtitles = if let Some(subtitles) = subtitles {
|
||||||
|
subtitles
|
||||||
|
} else {
|
||||||
|
find_subs(&source.parent().unwrap(), source.file_stem().unwrap())?.unwrap()
|
||||||
|
};
|
||||||
|
let fonts: Vec<PathBuf> = if let Some(fonts_dir) = fonts_dir {
|
||||||
|
fonts_dir
|
||||||
|
.read_dir()?
|
||||||
|
.map(|r| r.map(|entry| entry.path()))
|
||||||
|
.collect::<io::Result<_>>()?
|
||||||
|
} else {
|
||||||
|
find_files(source.parent().unwrap(), OsStr::new("ttf")).collect::<io::Result<_>>()?
|
||||||
|
};
|
||||||
|
|
||||||
|
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<ExitCode> {
|
||||||
|
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))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue