89 lines
2.5 KiB
Rust
89 lines
2.5 KiB
Rust
use std::fs;
|
|
use std::path::Path;
|
|
use log::{error, info};
|
|
|
|
pub struct Repository {
|
|
pub base_dir: String,
|
|
pub ingest_dir: String,
|
|
pub archive_dir: String,
|
|
pub output_dir: String,
|
|
}
|
|
|
|
impl Repository {
|
|
pub fn new(base_path: &str) -> Repository {
|
|
// create the base directory path
|
|
create_directory(base_path);
|
|
|
|
// create the needed sub-directories
|
|
let ingest_path = Path::new(base_path).join("ingest");
|
|
create_directory(ingest_path.to_str().unwrap());
|
|
let archive_path = Path::new(base_path).join("archive");
|
|
create_directory(archive_path.to_str().unwrap());
|
|
let output_path = Path::new(base_path).join("output");
|
|
create_directory(output_path.to_str().unwrap());
|
|
|
|
return Repository {
|
|
base_dir: String::from(base_path),
|
|
ingest_dir: String::from(ingest_path.to_str().unwrap()),
|
|
archive_dir: String::from(archive_path.to_str().unwrap()),
|
|
output_dir: String::from(output_path.to_str().unwrap()),
|
|
};
|
|
}
|
|
|
|
pub fn search_ingest(&self) -> Vec<String> {
|
|
// read file entries from ingest
|
|
let files = fs::read_dir(&self.ingest_dir).unwrap();
|
|
|
|
// create vec object and loop through entries to find what we want
|
|
let mut ingest_files: Vec<String> = vec![];
|
|
for f in files {
|
|
let f = f.unwrap();
|
|
let path = f.path();
|
|
if path.is_file() {
|
|
let file_path = path.file_name().unwrap().to_str();
|
|
ingest_files.push(String::from(file_path.unwrap()));
|
|
}
|
|
}
|
|
|
|
return ingest_files;
|
|
}
|
|
|
|
pub fn archive_file(&self, file: &str) {
|
|
let ingest_file = Path::new(&self.ingest_dir).join(file);
|
|
let archive_file = Path::new(&self.archive_dir).join(file);
|
|
|
|
match fs::copy(&ingest_file, &archive_file) {
|
|
Ok(_) => {
|
|
info!("Archiving video file {}.", ingest_file.to_str().unwrap());
|
|
},
|
|
Err(e) => {
|
|
error!("Error archiving file {}: {}", ingest_file.to_str().unwrap(), e);
|
|
std::process::exit(1);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn cleanup_file(&self, file: &str) {
|
|
let ingest_file = Path::new(&self.ingest_dir).join(file);
|
|
fs::remove_file(&ingest_file)
|
|
.expect("File deletion failed.");
|
|
}
|
|
}
|
|
|
|
fn create_directory(path: &str) {
|
|
let d = Path::new(path);
|
|
if d.is_dir() {
|
|
info!("Directory {} already exists.", path);
|
|
} else {
|
|
match fs::create_dir(path) {
|
|
Ok(_) => {
|
|
info!("Creating directory {}.", path);
|
|
},
|
|
Err(e) => {
|
|
error!("Error creating {}: {}", path, e);
|
|
std::process::exit(1);
|
|
}
|
|
}
|
|
}
|
|
}
|