baywatch 0.0.3

Watch files, execute commands on changes.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use anyhow::Result;
use notify::{Config, RecommendedWatcher, RecursiveMode, Watcher};
use std::collections::HashSet;
use std::path::PathBuf;
use std::sync::mpsc;

use crate::filters::PathFilter;

#[derive(Debug, Clone, Copy)]
pub struct ChangeEvent;

pub trait PathWatcher {
    fn watch(&self, changes: mpsc::SyncSender<ChangeEvent>) -> Result<()>;
}

pub struct NotifyWatcher<F: PathFilter> {
    paths: Vec<PathBuf>,
    filter: F,
}

impl<F: PathFilter> PathWatcher for NotifyWatcher<F> {
    fn watch(&self, changes: mpsc::SyncSender<ChangeEvent>) -> Result<()> {
        let (tx, rx) = mpsc::channel();

        let mut watcher = RecommendedWatcher::new(tx, Config::default())?;

        for path in &self.paths {
            watcher.watch(path.as_ref(), RecursiveMode::Recursive)?;
        }

        let change = ChangeEvent;
        let filtered_paths: HashSet<PathBuf> = HashSet::from_iter(self.filter.paths());

        for res in rx {
            match res {
                Ok(event) => {
                    log::debug!("Change in: {:?}", event.paths);
                    for path in event.paths {
                        if !filtered_paths.contains(&path) {
                            log::debug!("ignored");
                            continue;
                        };

                        if changes.try_send(change).is_err() {
                            log::debug!("buffer full, skipping event");
                        };
                    }
                }
                Err(error) => log::error!("Error: {error:?}"),
            }
        }

        Ok(())
    }
}

impl<F: PathFilter> NotifyWatcher<F> {
    pub fn new(paths: Vec<PathBuf>, filter: F) -> Self {
        NotifyWatcher { paths, filter }
    }
}