rust/bilibili-merge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: rust:69ca7c217ec203c7d67ba32e2c40ae3de3dfd354
Branches Tags
rust:master
..
compare: rust:58fd589b5bf13f604e46f7be5c58c427bec52aa1
Branches Tags
rust:master

No commits in common. "69ca7c217ec203c7d67ba32e2c40ae3de3dfd354" and "58fd589b5bf13f604e46f7be5c58c427bec52aa1" have entirely different histories.
69ca7c217e ... 58fd589b5b

1 changed files with 5 additions and 61 deletions
Show Stats Expand all files Collapse all files

66
src/merge.rs
View File

@ -1,7 +1,6 @@
use std::fs;
use std::io::Result;
use std::path::{Path, PathBuf};
use std::process::Command;
use std::{fs, io};
fn guess_video_path(path: &Path) -> Result<PathBuf> {
let mut largest_size = 0;
@ -13,7 +12,7 @@ fn guess_video_path(path: &Path) -> Result<PathBuf> {
}
let size = entry_path.metadata()?.len();
if size >= largest_size {
if size > largest_size {
largest_size = size;
largest_file = entry_path.into();
}
@ -22,65 +21,10 @@ fn guess_video_path(path: &Path) -> Result<PathBuf> {
Ok(largest_file)
}
fn convert_audio_to_aac(path: &Path) -> Result<()> {
let save_path = path.with_extension("aac");
let output = Command::new("ffmpeg")
.arg("-y")
.arg("-hide_banner")
.arg("-i").arg(path)
.arg("-c").arg("copy")
.arg(save_path)
.spawn()?.wait_with_output();
match output {
Ok(res) => {
if !res.status.success() {
return Err(io::Error::new(io::ErrorKind::Other, format!("ffmpeg exited with status {}", res.status)));
}
}
Err(e) => return Err(e)
}
Ok(())
}
fn merge(video_path: &Path, audio_path: &Path) -> Result<()> {
let origin_video_extension = video_path.extension().unwrap().to_str().unwrap();
let origin_filename = format!("original.{}", &origin_video_extension);
let origin_path = video_path.parent().unwrap().join(origin_filename);
fs::rename(video_path, &origin_path)?;
let output = Command::new("ffmpeg")
.arg("-y")
.arg("-hide_banner")
.arg("-i").arg(origin_path)
.arg("-i").arg(audio_path)
.arg("-c").arg("copy")
.arg(video_path)
.spawn()?.wait_with_output();
match output {
Ok(res) => {
if !res.status.success() {
return Err(io::Error::new(io::ErrorKind::Other, format!("ffmpeg exited with status {}", res.status)));
}
}
Err(e) => return Err(e)
}
Ok(())
}
pub fn merge_video_from_path(path: &Path) -> Result<()> {
let video_path = guess_video_path(path)?;
let video_name = video_path.file_name().unwrap().to_str().unwrap();
let video_extension = video_path.extension().unwrap();
let video_name = video_path.file_name().unwrap();
println!("Merging video {:?}", video_name);
let audio_name = format!("{}.m4a", &video_name[..video_name.len() - video_extension.len() - 1]);
let audio_path = video_path.parent().unwrap().join(audio_name);
if !audio_path.exists() {
return Err(io::Error::new(io::ErrorKind::NotFound, "Audio file does not exist"));
}
convert_audio_to_aac(&audio_path).expect("Audio file corrupted.");
merge(&video_path, &audio_path)
Ok(())
}