Compare commits
4 Commits
Author | SHA1 | Date | |
---|---|---|---|
7d5e1514a1 | |||
cd886522ec | |||
44ef95b440 | |||
b5d96c21a9 |
@ -1,7 +1,7 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "adept"
|
name = "adept"
|
||||||
description = "Bit Goblin automated video transcoding service."
|
description = "Bit Goblin automated video transcoding service."
|
||||||
version = "0.1.1"
|
version = "0.1.3"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
license = "BSD 2-Clause"
|
license = "BSD 2-Clause"
|
||||||
|
@ -1,11 +1,11 @@
|
|||||||
use log4rs;
|
use log4rs;
|
||||||
use config::Config;
|
use config::Config;
|
||||||
use repository::Repository;
|
use transcoder::repository::Repository;
|
||||||
use transcoder::Transcoder;
|
use transcoder::transcoder::Transcoder;
|
||||||
|
|
||||||
mod config;
|
mod config;
|
||||||
mod repository;
|
|
||||||
mod transcoder;
|
mod transcoder;
|
||||||
|
mod util;
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
// initialize the log4rs logger
|
// initialize the log4rs logger
|
||||||
|
2
src/transcoder/mod.rs
Normal file
2
src/transcoder/mod.rs
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
pub mod repository;
|
||||||
|
pub mod transcoder;
|
@ -4,7 +4,8 @@ use std::{thread, time};
|
|||||||
use log::{info};
|
use log::{info};
|
||||||
|
|
||||||
use crate::config::Config;
|
use crate::config::Config;
|
||||||
use crate::repository::Repository;
|
use crate::util::io;
|
||||||
|
use super::repository::Repository;
|
||||||
|
|
||||||
pub struct Transcoder {
|
pub struct Transcoder {
|
||||||
config: Config,
|
config: Config,
|
||||||
@ -31,6 +32,12 @@ impl Transcoder {
|
|||||||
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 {
|
||||||
for i in ingest_files {
|
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
|
// copy the file to the archive
|
||||||
self.repository.archive_file(&i);
|
self.repository.archive_file(&i);
|
||||||
|
|
61
src/util/io.rs
Normal file
61
src/util/io.rs
Normal 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
1
src/util/mod.rs
Normal file
@ -0,0 +1 @@
|
|||||||
|
pub mod io;
|
Reference in New Issue
Block a user