forked from kemitix/git-next
70 lines
1.9 KiB
Rust
70 lines
1.9 KiB
Rust
use actix::prelude::*;
|
|
|
|
use actix::Recipient;
|
|
use inotify::{EventMask, Inotify, WatchMask};
|
|
use std::{path::PathBuf, time::Duration};
|
|
use tracing::{debug, info};
|
|
|
|
const CHECK_INTERVAL: Duration = Duration::from_secs(1);
|
|
|
|
#[derive(Debug, Clone, Message)]
|
|
#[rtype(result = "()")]
|
|
pub struct WatchFile;
|
|
|
|
#[derive(Debug, Message)]
|
|
#[rtype(result = "()")]
|
|
pub struct FileUpdated;
|
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
pub enum Error {
|
|
#[error("io")]
|
|
Io(#[from] std::io::Error),
|
|
}
|
|
|
|
pub struct FileWatcher {
|
|
path: PathBuf,
|
|
inotify: Inotify,
|
|
recipient: Recipient<FileUpdated>,
|
|
run_interval: Option<SpawnHandle>,
|
|
}
|
|
impl FileWatcher {
|
|
pub fn new(path: PathBuf, recipient: Recipient<FileUpdated>) -> Result<Self, Error> {
|
|
let inotify = Inotify::init()?;
|
|
inotify.watches().add(
|
|
path.clone(),
|
|
WatchMask::MODIFY | WatchMask::CREATE | WatchMask::DELETE | WatchMask::ATTRIB,
|
|
)?;
|
|
Ok(Self {
|
|
path,
|
|
inotify,
|
|
recipient,
|
|
run_interval: None,
|
|
})
|
|
}
|
|
}
|
|
impl Actor for FileWatcher {
|
|
type Context = Context<Self>;
|
|
|
|
fn started(&mut self, ctx: &mut Self::Context) {
|
|
info!("Starting file watcher actor");
|
|
self.run_interval
|
|
.replace(ctx.run_interval(CHECK_INTERVAL, |_act, ctx| {
|
|
ctx.notify(WatchFile);
|
|
}));
|
|
}
|
|
}
|
|
|
|
impl Handler<WatchFile> for FileWatcher {
|
|
type Result = ();
|
|
|
|
fn handle(&mut self, _msg: WatchFile, _ctx: &mut Self::Context) -> Self::Result {
|
|
debug!("Watching {} for activity...", self.path.display());
|
|
let mut buffer = [0u8; 4096];
|
|
if let Ok(mut events) = self.inotify.read_events(&mut buffer) {
|
|
if events.any(|event| event.mask.contains(EventMask::MODIFY)) {
|
|
info!("File modified");
|
|
self.recipient.do_send(FileUpdated);
|
|
};
|
|
}
|
|
}
|
|
}
|