Added some basic file archiving functionality
This commit is contained in:
parent
0fa83eff42
commit
b979850582
@ -2,7 +2,10 @@ use std::fs;
|
|||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
|
|
||||||
pub struct Repository {
|
pub struct Repository {
|
||||||
base_path: String,
|
base_dir: String,
|
||||||
|
ingest_dir: String,
|
||||||
|
archive_dir: String,
|
||||||
|
output_dir: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Repository {
|
impl Repository {
|
||||||
@ -10,15 +13,54 @@ impl Repository {
|
|||||||
// create the base directory path
|
// create the base directory path
|
||||||
create_directory(base_path);
|
create_directory(base_path);
|
||||||
|
|
||||||
let sub_dirs: Vec<&str> = vec!["ingest", "archive", "output"];
|
// create the needed sub-directories
|
||||||
for s in sub_dirs {
|
let ingest_path = Path::new(base_path).join("ingest");
|
||||||
create_directory(Path::new(base_path).join(s).to_str().unwrap());
|
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 {
|
return Repository {
|
||||||
base_path: String::from(base_path),
|
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: String) {
|
||||||
|
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(_) => {
|
||||||
|
println!("Archiving video file {}.", ingest_file.to_str().unwrap());
|
||||||
|
},
|
||||||
|
Err(e) => {
|
||||||
|
eprintln!("Error archiving file {}: {}", ingest_file.to_str().unwrap(), e);
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn create_directory(path: &str) {
|
fn create_directory(path: &str) {
|
||||||
|
@ -16,5 +16,13 @@ impl Transcoder {
|
|||||||
|
|
||||||
pub fn start(self) {
|
pub fn start(self) {
|
||||||
println!("Starting transcoder...");
|
println!("Starting transcoder...");
|
||||||
|
|
||||||
|
// search for files in ingest
|
||||||
|
let ingest_files = self.repository.search_ingest();
|
||||||
|
|
||||||
|
for i in ingest_files {
|
||||||
|
// copy the file to the archive
|
||||||
|
self.repository.archive_file(i);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user