added flag for clearing log files at restart

Signed-off-by: Aminu 'Seun Joshua <[email protected]>

Author: seun-ja

crates/runtime-config/src/lib.rs

@@ -99,6 +99,7 @@ where
         local_app_dir: Option<PathBuf>,
         provided_state_dir: UserProvidedPath,
         provided_log_dir: UserProvidedPath,
+        refresh_logs: bool,
     ) -> anyhow::Result<Self> {
         let toml = match runtime_config_path {
             Some(runtime_config_path) => {
@@ -117,8 +118,13 @@ where
             }
             None => Default::default(),
         };
-        let toml_resolver =
-            TomlResolver::new(&toml, local_app_dir, provided_state_dir, provided_log_dir);
+        let toml_resolver = TomlResolver::new(
+            &toml,
+            local_app_dir,
+            provided_state_dir,
+            provided_log_dir,
+            refresh_logs,
+        );
 
         Self::new(toml_resolver, runtime_config_path)
     }
@@ -192,6 +198,8 @@ pub struct TomlResolver<'a> {
     state_dir: UserProvidedPath,
     /// Explicitly provided log directory.
     log_dir: UserProvidedPath,
+    /// Whether to refresh logs when restarted.
+    refresh_logs: bool,
 }
 
 impl<'a> TomlResolver<'a> {
@@ -201,12 +209,14 @@ impl<'a> TomlResolver<'a> {
         local_app_dir: Option<PathBuf>,
         state_dir: UserProvidedPath,
         log_dir: UserProvidedPath,
+        refresh_logs: bool,
     ) -> Self {
         Self {
             table: TomlKeyTracker::new(table),
             local_app_dir,
             state_dir,
             log_dir,
+            refresh_logs,
         }
     }
 
@@ -268,7 +278,17 @@ impl<'a> TomlResolver<'a> {
         }
 
         match log_dir {
+            UserProvidedPath::Provided(p) if self.refresh_logs => Ok(Some({
+                let _ = std::fs::remove_dir_all(&p);
+
+                std::path::absolute(p)?
+            })),
             UserProvidedPath::Provided(p) => Ok(Some(std::path::absolute(p)?)),
+            UserProvidedPath::Default if self.refresh_logs => Ok(self.state_dir()?.map(|p| {
+                let _ = std::fs::remove_dir_all(&p);
+
+                p.join("logs")
+            })),
             UserProvidedPath::Default => Ok(self.state_dir()?.map(|p| p.join("logs"))),
             UserProvidedPath::Unset => Ok(None),
         }
@@ -619,6 +639,7 @@ mod tests {
             None,
             UserProvidedPath::Default,
             UserProvidedPath::Default,
+            false,
         )
     }
 

crates/runtime-factors/src/build.rs

@@ -28,6 +28,7 @@ impl RuntimeFactorsBuilder for FactorsBuilder {
             config.local_app_dir.clone().map(PathBuf::from),
             config.state_dir.clone(),
             config.log_dir.clone(),
+            config.refresh_logs,
         )?;
 
         runtime_config.summarize(config.runtime_config_file.as_deref());

crates/trigger/src/cli.rs

@@ -27,6 +27,7 @@ pub use stdio::StdioLoggingExecutorHooks;
 pub use summary::{KeyValueDefaultStoreSummaryHook, SqliteDefaultStoreSummaryHook};
 
 pub const APP_LOG_DIR: &str = "APP_LOG_DIR";
+pub const REFRESH_LOGS: &str = "REFRESH_LOGS";
 pub const DISABLE_WASMTIME_CACHE: &str = "DISABLE_WASMTIME_CACHE";
 pub const FOLLOW_LOG_OPT: &str = "FOLLOW_ID";
 pub const WASMTIME_CACHE_FILE: &str = "WASMTIME_CACHE_FILE";
@@ -54,6 +55,14 @@ pub struct FactorsTriggerCommand<T: Trigger<B::Factors>, B: RuntimeFactorsBuilde
     )]
     pub log: Option<PathBuf>,
 
+    /// Whether to refresh logs when restarted.
+    #[clap(
+        name = REFRESH_LOGS,
+        short = 'r',
+        long = "refresh-logs",
+    )]
+    pub refresh_logs: bool,
+
     /// Disable Wasmtime cache.
     #[clap(
         name = DISABLE_WASMTIME_CACHE,
@@ -139,6 +148,8 @@ pub struct FactorsConfig {
     pub follow_components: FollowComponents,
     /// Log directory for component stdout/stderr.
     pub log_dir: UserProvidedPath,
+    /// Whether to refresh logs when restarted.
+    pub refresh_logs: bool,
 }
 
 /// An empty implementation of clap::Args to be used as TriggerExecutor::RunConfig
@@ -220,6 +231,7 @@ impl<T: Trigger<B::Factors>, B: RuntimeFactorsBuilder> FactorsTriggerCommand<T,
             local_app_dir: local_app_dir.clone(),
             follow_components,
             log_dir,
+            refresh_logs: self.refresh_logs,
         };
 
         let run_fut = builder