feat(client): add a temporary folder provider in utils

Author: jpraynaud

mithril-client/src/utils/mod.rs

@@ -2,19 +2,21 @@
 //! This module contains tools needed mostly for the snapshot download and unpack.
 
 cfg_fs! {
-    pub const ANCILLARIES_NOT_SIGNED_BY_MITHRIL:&str = "Ancillary verification does not use the Mithril certification: as a mitigation, IOG owned keys are used to sign these files.";
+    pub const ANCILLARIES_NOT_SIGNED_BY_MITHRIL: &str = "Ancillary verification does not use the Mithril certification: as a mitigation, IOG owned keys are used to sign these files.";
 
     mod ancillary_verifier;
+    mod bootstrap_files;
     mod fs;
     mod stream_reader;
-    mod bootstrap_files;
+    mod temp_dir_provider;
     mod unexpected_downloaded_file_verifier;
     mod vec_deque_extensions;
 
-    pub use fs::*;
-    pub use vec_deque_extensions::VecDequeExtensions;
     pub use ancillary_verifier::AncillaryVerifier;
-    pub(crate) use unexpected_downloaded_file_verifier::*;
-    pub use stream_reader::*;
     pub use bootstrap_files::*;
+    pub use fs::*;
+    pub use stream_reader::*;
+    pub(crate) use temp_dir_provider::{TempDirectoryProvider, TimestampTempDirectoryProvider};
+    pub(crate) use unexpected_downloaded_file_verifier::*;
+    pub use vec_deque_extensions::VecDequeExtensions;
 }

mithril-client/src/utils/temp_dir_provider.rs

@@ -0,0 +1,31 @@
+use chrono::Utc;
+use std::path::PathBuf;
+
+/// Trait to provide a temporary directory path
+#[cfg_attr(test, mockall::automock)]
+pub trait TempDirectoryProvider: Send + Sync {
+    /// Provides a temporary directory path
+    fn temp_dir(&self) -> PathBuf;
+}
+
+/// Timestamp based temporary directory provider implementation
+pub struct TimestampTempDirectoryProvider {
+    temp_dir: PathBuf,
+}
+
+impl TimestampTempDirectoryProvider {
+    /// Constructs a new `TimestampTempDirectoryProvider`.
+    pub fn new(prefix: &str) -> Self {
+        let temp_dir = std::env::temp_dir().join(format!(
+            "mithril_client_{prefix}_{}",
+            Utc::now().timestamp_micros()
+        ));
+        Self { temp_dir }
+    }
+}
+
+impl TempDirectoryProvider for TimestampTempDirectoryProvider {
+    fn temp_dir(&self) -> PathBuf {
+        self.temp_dir.clone()
+    }
+}