4 Commits

Author SHA1 Message Date
7d5e1514a1 Version bump to v0.1.3
All checks were successful
ci/woodpecker/tag/woodpecker Pipeline was successful
2022-09-01 23:44:59 -04:00
cd886522ec Added unit tests for the file locking function 2022-09-01 23:44:37 -04:00
44ef95b440 Added a basic check to make sure we don't try to transcode a partially written file 2022-09-01 22:15:46 -04:00
b5d96c21a9 Version bump to v0.1.2 2022-09-01 18:25:18 -04:00
7 changed files with 76 additions and 5 deletions

View File

@ -1,7 +1,7 @@
[package]
name = "adept"
description = "Bit Goblin automated video transcoding service."
version = "0.1.1"
version = "0.1.3"
edition = "2021"
readme = "README.md"
license = "BSD 2-Clause"

View File

@ -1,11 +1,11 @@
use log4rs;
use config::Config;
use repository::Repository;
use transcoder::Transcoder;
use transcoder::repository::Repository;
use transcoder::transcoder::Transcoder;
mod config;
mod repository;
mod transcoder;
mod util;
fn main() {
// initialize the log4rs logger

2
src/transcoder/mod.rs Normal file
View File

@ -0,0 +1,2 @@
pub mod repository;
pub mod transcoder;

View File

@ -4,7 +4,8 @@ use std::{thread, time};
use log::{info};
use crate::config::Config;
use crate::repository::Repository;
use crate::util::io;
use super::repository::Repository;
pub struct Transcoder {
config: Config,
@ -31,6 +32,12 @@ impl Transcoder {
info!("There were no files found in ingest to transcode; skipping run.");
} else {
for i in ingest_files {
let ingest_path = Path::new(&self.repository.ingest_dir).join(&i);
if io::is_file_locked(&ingest_path.to_str().unwrap()) {
info!("{} is currently open in another program; skipping it for this run.", &i);
continue;
}
// copy the file to the archive
self.repository.archive_file(&i);

61
src/util/io.rs Normal file
View File

@ -0,0 +1,61 @@
use std::process;
// checks whether a file is currently open in another program (e.g. still being written)
pub fn is_file_locked(filepath: &str) -> bool {
let cmd = process::Command::new("/usr/bin/lsof")
.arg(filepath)
.output()
.expect("Failed to execute command");
let results = &String::from_utf8_lossy(&cmd.stdout);
println!("{}", results);
if results.contains(filepath) {
return true;
}
return false;
}
#[cfg(test)]
mod tests {
use std::fs;
use std::mem::drop;
use std::path::Path;
use super::*;
#[test]
fn file_should_not_be_locked() {
setup("test-not-locked.txt");
assert!(!is_file_locked("test-not-locked.txt"));
teardown("test-not-locked.txt");
}
#[test]
fn file_should_be_locked() {
setup("test-locked.txt");
let path = Path::new("test-locked.txt");
match fs::File::open(&path) {
Err(e) => panic!("couldn't open {}: {}", "test-locked.txt", e),
Ok(_) => assert!(is_file_locked("test-locked.txt")),
};
teardown("test-locked.txt");
}
// get things ready for our tests
fn setup(test_file: &str) {
let f = match fs::File::create(test_file) {
Ok(file) => file,
Err(e) => panic!("{:?}", e),
};
drop(f);
}
// clean up after running tests
fn teardown(test_file: &str) {
match fs::remove_file(test_file) {
Ok(_) => {},
Err(e) => panic!("{:?}", e),
};
}
}

1
src/util/mod.rs Normal file
View File

@ -0,0 +1 @@
pub mod io;