feat: added flag for notification

client/src/main.rs

@@ -13,8 +13,13 @@ use tracing_subscriber::{EnvFilter, FmtSubscriber};
 use uuid::Uuid;
 use zerotier_api::Identity;
 mod validation;
+use std::fs::OpenOptions;
+use std::io::Read;
 use validation::init_validation;
 
+const SUCCESS_FLAG: &str = "SUCCESS";
+const FAIL_FLAG: &str = "FAIL";
+
 fn get_holoport_url(id: VerifyingKey) -> String {
     if let Ok(network) = env::var("HOLO_NETWORK") {
         if network == "devNet" {
@@ -24,6 +29,13 @@ fn get_holoport_url(id: VerifyingKey) -> String {
     format!("https://{}.holohost.net", public_key::to_base36_id(&id))
 }
 
+fn holo_auth_flag_file() -> String {
+    match env::var("HOLO_AUTH_FLAG_FILE") {
+        Ok(file) => file,
+        _ => "/var/lib/holo-auth/holo-auth-status".to_string(),
+    }
+}
+
 fn mem_proof_server_url() -> String {
     match env::var("MEM_PROOF_SERVER_URL") {
         Ok(url) => url,
@@ -52,6 +64,31 @@ fn device_bundle_password() -> Option<String> {
     }
 }
 
+fn read_flag_file() -> Option<String> {
+    let file_path = holo_auth_flag_file();
+    if Path::new(&file_path).exists() {
+        let mut file = File::open(file_path).expect("Failed to open flag file");
+        let mut contents = String::new();
+        file.read_to_string(&mut contents)
+            .expect("Failed to read flag file");
+        Some(contents)
+    } else {
+        None
+    }
+}
+
+fn write_flag_file(flag: &str) {
+    let file_path = holo_auth_flag_file();
+    let mut file = OpenOptions::new()
+        .write(true)
+        .create(true)
+        .truncate(true) // Overwrite the file
+        .open(file_path)
+        .expect("Failed to open flag file");
+    file.write_all(flag.as_bytes())
+        .expect("Failed to write flag file");
+}
+
 lazy_static! {
     static ref CLIENT: Client = Client::new();
 }
@@ -158,6 +195,11 @@ struct NotifyPayload {
 }
 async fn send_failure_email(email: String, data: String) -> Fallible<()> {
     info!("Sending Failure Email to: {:?}", email);
+    let flag = read_flag_file();
+    write_flag_file(FAIL_FLAG);
+    if flag != Some(SUCCESS_FLAG.to_string()) {
+        return Ok(());
+    }
     send_email(email, data, false).await
 }
 async fn send_email(email: String, data: String, success: bool) -> Fallible<()> {
@@ -299,5 +341,6 @@ async fn main() -> Fallible<()> {
         thread::sleep(backoff);
         backoff += backoff;
     }
+    write_flag_file(SUCCESS_FLAG);
     Ok(())
 }