115 lines
2.8 KiB
Rust
115 lines
2.8 KiB
Rust
#![feature(rust_2018_preview, iterator_find_map, use_extern_macros)]
|
||
|
||
#[macro_use] extern crate serde_derive;
|
||
|
||
mod config;
|
||
mod state;
|
||
|
||
use crate::{
|
||
config::Config,
|
||
state::State,
|
||
};
|
||
|
||
use chrono::{DateTime, Duration, Local, TimeZone, Utc};
|
||
|
||
use failure::Error;
|
||
|
||
use notify::{DebouncedEvent, RecursiveMode, Watcher};
|
||
|
||
use std::{fs, path::PathBuf, sync::mpsc};
|
||
|
||
pub type Result<T> = std::result::Result<T, Error>;
|
||
|
||
fn main() -> Result<()> {
|
||
let config_path = match std::env::args().nth(1) {
|
||
Some(x) => x,
|
||
None => "config.json".into(),
|
||
};
|
||
|
||
let f = fs::File::open(config_path)?;
|
||
let config: Config = serde_json::from_reader(f)?;
|
||
|
||
let screenshots_dir = config.options.screenshots_dir.canonicalize()?;
|
||
|
||
let (tx, rx) = mpsc::channel();
|
||
|
||
for entry in std::fs::read_dir(&screenshots_dir)? {
|
||
let entry = entry?;
|
||
if let Err(e) = handle(&config, entry.path()) {
|
||
eprintln!("{}", e);
|
||
}
|
||
}
|
||
|
||
let mut watcher = notify::watcher(
|
||
tx,
|
||
Duration::milliseconds(config.options.event_delay).to_std().unwrap(),
|
||
)?;
|
||
|
||
|
||
watcher.watch(
|
||
screenshots_dir.to_string_lossy().to_string(),
|
||
RecursiveMode::NonRecursive
|
||
)?;
|
||
|
||
loop {
|
||
match rx.recv() {
|
||
Ok(DebouncedEvent::Create(p)) => if let Err(e) = handle(&config, p) {
|
||
eprintln!("{}", e);
|
||
},
|
||
Ok(_) => {},
|
||
Err(e) => eprintln!("{:#?}", e),
|
||
}
|
||
}
|
||
}
|
||
|
||
fn handle(config: &Config, p: PathBuf) -> Result<()> {
|
||
let screenshots_dir = config.options.screenshots_dir.canonicalize()?;
|
||
|
||
// if the path doesn't exist, ignore
|
||
// no need to print out an error – usually a result of duplicate events
|
||
if !p.exists() {
|
||
return Ok(());
|
||
}
|
||
|
||
// if the path isn't at the top level, ignore
|
||
if p.strip_prefix(&screenshots_dir).unwrap().components().count() != 1 {
|
||
return Ok(());
|
||
}
|
||
|
||
// ignore paths without file names
|
||
let file_name: String = match p.file_name() {
|
||
Some(f) => f.to_string_lossy().to_string(),
|
||
None => return Ok(()),
|
||
};
|
||
|
||
// ignore paths that don't match
|
||
let time = match parse_screenshot_name(&config, &file_name) {
|
||
Some(t) => t,
|
||
None => return Ok(()),
|
||
};
|
||
|
||
// execute the jobs in the pipeline
|
||
let mut state = State::new(p, time);
|
||
for job in &config.pipeline {
|
||
job.execute(&config, &mut state)?;
|
||
}
|
||
|
||
Ok(())
|
||
}
|
||
|
||
fn parse_screenshot_name(config: &Config, s: &str) -> Option<DateTime<Utc>> {
|
||
let caps = config.options.patterns.iter().find_map(|p| p.captures(s))?;
|
||
let dt = Local
|
||
.ymd(
|
||
caps.name("year")?.as_str().parse().ok()?,
|
||
caps.name("month")?.as_str().parse().ok()?,
|
||
caps.name("day")?.as_str().parse().ok()?,
|
||
)
|
||
.and_hms(
|
||
caps.name("hour")?.as_str().parse().ok()?,
|
||
caps.name("minute")?.as_str().parse().ok()?,
|
||
caps.name("second")?.as_str().parse().ok()?,
|
||
);
|
||
Some(dt.with_timezone(&Utc))
|
||
}
|