deadman's switch

bertiqwerty · Oct 12, 2024 · df18212 · df18212
1 parent 2c9332c
commit df18212
Show file tree

Hide file tree

Showing 6 changed files with 206 additions and 32 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/rvimage/Cargo.toml b/rvimage/Cargo.toml
@@ -59,7 +59,7 @@ serde_json = "1.0.128"
 ssh2 = { version = "0.9.4", features = ["vendored-openssl"] }
 toml = "0.8.19"
 walkdir = "2.5"
-tokio = { version = "1.40.0", optional = true, features = ["rt-multi-thread"]}
+tokio = { version = "1.40.0", optional = true, features = ["rt-multi-thread"] }
 uuid = { version = "1.10.0", features = ["v4", "fast-rng"] }
 exmex = "0.20.3"
 tracing = "0.1.40"
@@ -69,3 +69,5 @@ backtrace = "0.3.74"
 zip = "2.2.0"
 rvimage-domain = "0.4.5"
 clap = { version = "4.5.19", features = ["derive"] }
+chrono = { version = "0.4.38", features = ["serde"] }
+whoami = "1.5.2"
diff --git a/rvimage/src/rvlib/cfg.rs b/rvimage/src/rvlib/cfg.rs
@@ -4,6 +4,7 @@ use crate::{
     sort_params::SortParams,
     ssh,
 };
+use chrono::{DateTime, Utc};
 use rvimage_domain::{rverr, to_rv, RvError, RvResult};
 use serde::{de::DeserializeOwned, Deserialize, Serialize};
 use std::{
@@ -76,6 +77,7 @@ impl CfgLegacy {
                 prefix: ab.prefix,
             }),
             sort_params: SortParams::default(),
+            deadmansswitch: Utc::now(),
         };
         Cfg { usr, prj }
     }
@@ -339,6 +341,8 @@ pub struct CfgPrj {
     pub azure_blob: Option<AzureBlobCfgPrj>,
     #[serde(default)]
     pub sort_params: SortParams,
+    #[serde(default = "Utc::now")]
+    pub deadmansswitch: DateTime<Utc>,
 }
 #[derive(Deserialize, Serialize, Debug, Clone, PartialEq, Eq, Default)]
 pub struct Cfg {
@@ -370,6 +374,10 @@ impl Cfg {
         }
     }
 
+    pub fn push_deadmansswitch(&mut self) {
+        self.prj.deadmansswitch = Utc::now();
+    }
+
     pub fn tmpdir(&self) -> &str {
         match &self.usr.tmpdir {
             Some(td) => td.as_str(),