diff --git a/src/watchman.rs b/src/watchman.rs index 474bec2..908deb8 100644 --- a/src/watchman.rs +++ b/src/watchman.rs @@ -14,6 +14,7 @@ use std::time::{Duration, SystemTime}; use crate::arguments::Arguments; use crate::config::setup_service; +use crate::monitors::network::NETMon; use crate::util::{get_cache_dir, get_config_dir, write_default_blacklist, write_default_config}; pub trait EventMonitor { @@ -23,7 +24,7 @@ pub trait EventMonitor { #[derive(Debug)] pub enum Monitors { USBMon(USBMon), - // NetMon(NETMon), + NetMon(NETMon), SSHBurnMon(SSHBurnMon), FileChanges(FileChanges), NewClients(NewClientMon), @@ -190,6 +191,9 @@ impl AppState { Monitors::NewClients(e) => { e.check().await; } + Monitors::NetMon(e) => { + e.check().await + } } tokio::time::sleep(Duration::new(n_tick, 0)).await; }