first commit

src/monitors/ssh_burn_file.rs

@@ -0,0 +1,84 @@
+use crate::fortiwatch::EventMonitor;
+use crate::monitors::actions::unmount_encrypted_volumes;
+use chrono::{DateTime, Utc};
+use config::Config;
+use std::process::Command;
+
+#[allow(unused)]
+#[derive(Debug)]
+pub struct SSHBurnMon {
+    triggered: bool,
+    settings_map: Config,
+    last_check: DateTime<Utc>,
+}
+
+impl SSHBurnMon {
+    pub fn new(settings_map: Config) -> Self {
+        SSHBurnMon {
+            triggered: false,
+            settings_map,
+            last_check: Utc::now(),
+        }
+    }
+    async fn ssh_burn_triggered(&self) {
+        if self
+            .settings_map
+            .get::<String>("unmount_crypt_on_file_burn")
+            .unwrap()
+            .eq_ignore_ascii_case("true")
+        {
+            unmount_encrypted_volumes().await;
+        }
+    }
+}
+
+impl EventMonitor for SSHBurnMon {
+    async fn check(&mut self) {
+        let ssh_check_burn_check_interval = self
+            .settings_map
+            .get::<String>("ssh_check_burn_check_interval")
+            .unwrap()
+            .parse::<i64>()
+            .unwrap();
+        if Utc::now()
+            .signed_duration_since(self.last_check)
+            .num_seconds()
+            > ssh_check_burn_check_interval
+        {
+            let ssh_check_burn_host = self
+                .settings_map
+                .get::<String>("ssh_check_burn_host")
+                .unwrap();
+            let ssh_check_burn_user = self
+                .settings_map
+                .get::<String>("ssh_check_burn_user")
+                .unwrap();
+            let ssh_check_burn_key = self
+                .settings_map
+                .get::<String>("ssh_check_burn_key")
+                .unwrap();
+            let ssh_check_burn_path = self
+                .settings_map
+                .get::<String>("ssh_check_burn_path")
+                .unwrap();
+            let addr = format!("{}@{}", ssh_check_burn_user, ssh_check_burn_host);
+            let command_str = format!(
+                "if [ -f {} ]; then cat {}; fi",
+                ssh_check_burn_path, ssh_check_burn_path
+            );
+            if let Ok(result) = Command::new("ssh")
+                .arg("-i")
+                .arg(ssh_check_burn_key)
+                .arg(addr)
+                .arg(command_str)
+                .output()
+            {
+                let burn_contents = String::from_utf8(result.stdout).unwrap();
+                if burn_contents.eq_ignore_ascii_case("burn") {
+                    self.ssh_burn_triggered().await;
+                }
+            }
+            self.last_check = Utc::now();
+        }
+    }
+}