Compare commits
8 Commits
Author | SHA1 | Date | |
---|---|---|---|
d56f7f2f0f | |||
1cc8306fec | |||
597b64c9c5 | |||
e479959ca8 | |||
ef492f6008 | |||
6371417234 | |||
a7cc2bc771 | |||
ed4173f3dc |
@ -1,13 +1,13 @@
|
|||||||
pipeline:
|
pipeline:
|
||||||
tests:
|
tests:
|
||||||
image: rust:1.63
|
image: rust:1.75
|
||||||
commands:
|
commands:
|
||||||
- "apt update"
|
- "apt update"
|
||||||
- "apt install -y lsof"
|
- "apt install -y lsof"
|
||||||
- "cargo test"
|
- "cargo test"
|
||||||
|
|
||||||
build_release:
|
build_release:
|
||||||
image: rust:1.63
|
image: rust:1.75
|
||||||
commands:
|
commands:
|
||||||
- "cargo install cargo-deb cargo-generate-rpm"
|
- "cargo install cargo-deb cargo-generate-rpm"
|
||||||
- "cargo build --release"
|
- "cargo build --release"
|
||||||
|
@ -1,11 +1,12 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "zealot"
|
name = "zealot"
|
||||||
description = "Bit Goblin automated video transcoding service."
|
description = "Bit Goblin automated video transcoding service."
|
||||||
version = "0.2.3"
|
version = "0.2.7"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
license = "BSD 2-Clause"
|
license = "BSD 2-Clause"
|
||||||
authors = ["Gregory Ballantine <gballantine@bitgoblin.tech>"]
|
authors = ["Gregory Ballantine <gballantine@bitgoblin.tech>"]
|
||||||
|
repository = "https://git.metaunix.net/BitGoblin/zealot"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
clap = { version = "3.2", features = ['derive'] }
|
clap = { version = "3.2", features = ['derive'] }
|
||||||
|
@ -18,3 +18,21 @@ pub fn setup_command() {
|
|||||||
// initialize the video repository
|
// initialize the video repository
|
||||||
r.initialize();
|
r.initialize();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn version_command() {
|
||||||
|
println!("===================================================================================");
|
||||||
|
|
||||||
|
// output the program and version
|
||||||
|
println!("Zealot version v{}", env!("CARGO_PKG_VERSION"));
|
||||||
|
|
||||||
|
// output project description
|
||||||
|
println!("{}", env!("CARGO_PKG_DESCRIPTION"));
|
||||||
|
|
||||||
|
// display license info
|
||||||
|
println!("\nThis code is licensed under the {} license.", env!("CARGO_PKG_LICENSE"));
|
||||||
|
|
||||||
|
// output the project source URL
|
||||||
|
println!("View the source code or file issues at {}.", env!("CARGO_PKG_REPOSITORY"));
|
||||||
|
|
||||||
|
println!("===================================================================================");
|
||||||
|
}
|
||||||
|
@ -22,8 +22,11 @@ struct Cli {
|
|||||||
|
|
||||||
#[derive(Subcommand)]
|
#[derive(Subcommand)]
|
||||||
enum Commands {
|
enum Commands {
|
||||||
/// displays version info about this program
|
// Runs the setup command
|
||||||
Setup {},
|
Setup {},
|
||||||
|
|
||||||
|
// Displays version info about the program
|
||||||
|
Version {},
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
@ -37,6 +40,7 @@ fn main() {
|
|||||||
match &cli.command {
|
match &cli.command {
|
||||||
// sub-commands will be handled here
|
// sub-commands will be handled here
|
||||||
Some(Commands::Setup {}) => cmd::core::setup_command(),
|
Some(Commands::Setup {}) => cmd::core::setup_command(),
|
||||||
|
Some(Commands::Version {}) => cmd::core::version_command(),
|
||||||
|
|
||||||
// run the main program without any commands
|
// run the main program without any commands
|
||||||
None => {
|
None => {
|
||||||
|
@ -60,7 +60,7 @@ impl Repository {
|
|||||||
match fs::copy(&ingest_file, &archive_file) {
|
match fs::copy(&ingest_file, &archive_file) {
|
||||||
Ok(_) => {
|
Ok(_) => {
|
||||||
info!("Archiving video file {}.", ingest_file.to_str().unwrap());
|
info!("Archiving video file {}.", ingest_file.to_str().unwrap());
|
||||||
copy_file_times(ingest_file.to_str().unwrap(), archive_file.to_str().unwrap());
|
copy_file_metadata(ingest_file.to_str().unwrap(), archive_file.to_str().unwrap());
|
||||||
},
|
},
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
error!("Error archiving file {}: {}", ingest_file.to_str().unwrap(), e);
|
error!("Error archiving file {}: {}", ingest_file.to_str().unwrap(), e);
|
||||||
@ -93,13 +93,15 @@ fn create_directory(path: &str) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn copy_file_times(old_file: &str, new_file: &str) {
|
fn copy_file_metadata(old_file: &str, new_file: &str) {
|
||||||
// grab old file metadata
|
// grab old file metadata
|
||||||
let old_metadata = fs::metadata(old_file).unwrap();
|
let old_metadata = fs::metadata(old_file).unwrap();
|
||||||
|
let old_permissions = old_metadata.permissions();
|
||||||
let mtime = FileTime::from_last_modification_time(&old_metadata);
|
let mtime = FileTime::from_last_modification_time(&old_metadata);
|
||||||
let atime = FileTime::from_last_access_time(&old_metadata);
|
let atime = FileTime::from_last_access_time(&old_metadata);
|
||||||
|
|
||||||
// set metadata on the new file
|
// set metadata on the new file
|
||||||
let new_path = Path::new(new_file);
|
let new_path = Path::new(new_file);
|
||||||
|
fs::set_permissions(new_file, old_permissions).expect("Failed to set new file permissions.");
|
||||||
filetime::set_file_times(new_path, atime, mtime).expect("Failed to set atime and mtime.");
|
filetime::set_file_times(new_path, atime, mtime).expect("Failed to set atime and mtime.");
|
||||||
}
|
}
|
||||||
|
@ -31,6 +31,10 @@ impl Transcoder {
|
|||||||
if ingest_files.len() < 1 {
|
if ingest_files.len() < 1 {
|
||||||
info!("There were no files found in ingest to transcode; skipping run.");
|
info!("There were no files found in ingest to transcode; skipping run.");
|
||||||
} else {
|
} else {
|
||||||
|
// log that the transcoder is starting up
|
||||||
|
info!("Found {} files in ingest to transcode. Standby...", ingest_files.len());
|
||||||
|
|
||||||
|
// loop through each file found in ingest
|
||||||
for i in ingest_files {
|
for i in ingest_files {
|
||||||
let ingest_path = Path::new(&self.repository.ingest_dir).join(&i);
|
let ingest_path = Path::new(&self.repository.ingest_dir).join(&i);
|
||||||
if io::is_file_locked(&ingest_path.to_str().unwrap()) {
|
if io::is_file_locked(&ingest_path.to_str().unwrap()) {
|
||||||
@ -47,6 +51,9 @@ impl Transcoder {
|
|||||||
// remove the source file
|
// remove the source file
|
||||||
self.repository.cleanup_file(&i);
|
self.repository.cleanup_file(&i);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// let the user know the transcode has finished
|
||||||
|
info!("Finished transcoding. Sleeping...");
|
||||||
}
|
}
|
||||||
|
|
||||||
// put the loop to sleep for X minutes
|
// put the loop to sleep for X minutes
|
||||||
@ -66,17 +73,28 @@ impl Transcoder {
|
|||||||
let video_codec = &self.config.get_string("transcoder.video_codec").unwrap();
|
let video_codec = &self.config.get_string("transcoder.video_codec").unwrap();
|
||||||
info!("Transcoding {} to {} with the {} encoder.", ingest_file.display(), output_file.display(), video_codec);
|
info!("Transcoding {} to {} with the {} encoder.", ingest_file.display(), output_file.display(), video_codec);
|
||||||
|
|
||||||
let cmd_output = process::Command::new("/usr/bin/ffmpeg")
|
let binding = process::Command::new("/usr/bin/ffmpeg");
|
||||||
.arg("-i") .arg(&*ingest_file.to_string_lossy())
|
let mut cmd = binding;
|
||||||
|
|
||||||
|
// start building the command
|
||||||
|
cmd.arg("-i") .arg(&*ingest_file.to_string_lossy())
|
||||||
.arg("-y")
|
.arg("-y")
|
||||||
.arg("-f") .arg(&video_format)
|
.arg("-f") .arg(&video_format)
|
||||||
.arg("-c:v") .arg(&video_codec)
|
.arg("-c:v") .arg(&video_codec);
|
||||||
.arg("-s") .arg(&self.config.get_string("transcoder.video_resolution").unwrap())
|
|
||||||
.arg("-r") .arg(format!("{}", self.config.get_string("transcoder.video_framerate").unwrap()))
|
// add video resolution if it's available
|
||||||
|
if self.config.get_string("transcoder.video_resolution").is_ok() {
|
||||||
|
cmd.arg("-s").arg(&self.config.get_string("transcoder.video_resolution").unwrap());
|
||||||
|
};
|
||||||
|
|
||||||
|
// finish out command
|
||||||
|
cmd.arg("-r").arg(format!("{}", self.config.get_string("transcoder.video_framerate").unwrap()))
|
||||||
.arg("-vf") .arg(format!("format={}", &self.config.get_string("transcoder.video_color").unwrap()))
|
.arg("-vf") .arg(format!("format={}", &self.config.get_string("transcoder.video_color").unwrap()))
|
||||||
.arg("-profile:v").arg(&self.config.get_string("transcoder.video_profile").unwrap())
|
.arg("-profile:v").arg(&self.config.get_string("transcoder.video_profile").unwrap())
|
||||||
.arg("-c:a") .arg(&self.config.get_string("transcoder.audio_codec").unwrap())
|
.arg("-c:a") .arg(&self.config.get_string("transcoder.audio_codec").unwrap());
|
||||||
.arg(&*output_file.to_string_lossy())
|
|
||||||
|
// finish the command and run it
|
||||||
|
let cmd_output = cmd.arg(&*output_file.to_string_lossy())
|
||||||
.output()
|
.output()
|
||||||
.expect("Failed to execute command");
|
.expect("Failed to execute command");
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user