feat(client-cli): implement directory copy functionality

Author: dlachaume

mithril-client-cli/src/utils/fs.rs

@@ -0,0 +1,109 @@
+use std::{
+    fs,
+    path::{Path, PathBuf},
+};
+
+use anyhow::{anyhow, Context};
+
+use mithril_client::MithrilResult;
+
+/// Copies a directory and its contents to a new location.
+pub fn copy_dir(source_dir: &Path, target_dir: &Path) -> MithrilResult<PathBuf> {
+    let source_dir_name = source_dir
+        .file_name()
+        .ok_or_else(|| anyhow!("Invalid source directory: {}", source_dir.display()))?;
+    let destination_path = target_dir.join(source_dir_name);
+    copy_dir_contents(source_dir, &destination_path)?;
+
+    Ok(destination_path)
+}
+
+fn copy_dir_contents(source_dir: &Path, target_dir: &Path) -> MithrilResult<()> {
+    fs::create_dir_all(target_dir).with_context(|| {
+        format!(
+            "Failed to create target directory: {}",
+            target_dir.display()
+        )
+    })?;
+
+    for entry in fs::read_dir(source_dir)? {
+        let entry = entry?;
+        let file_type = entry.file_type()?;
+        let src_path = entry.path();
+        let dst_path = target_dir.join(entry.file_name());
+
+        if file_type.is_dir() {
+            copy_dir_contents(&src_path, &dst_path)?;
+        } else {
+            fs::copy(&src_path, &dst_path).with_context(|| {
+                format!(
+                    "Failed to copy file '{}' to '{}'",
+                    src_path.display(),
+                    dst_path.display()
+                )
+            })?;
+        }
+    }
+
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use std::fs::File;
+
+    use mithril_common::{assert_dir_eq, temp_dir_create};
+
+    use super::*;
+
+    #[test]
+    fn fails_if_source_does_not_exist() {
+        let temp_dir = temp_dir_create!();
+        let dir_not_exist = PathBuf::from("dir_not_exist");
+
+        copy_dir(&dir_not_exist, &temp_dir)
+            .expect_err("Expected error when source directory does not exist");
+    }
+
+    #[test]
+    fn returns_copied_directory_path() {
+        let temp_dir = temp_dir_create!();
+        let src = temp_dir.join("dir_to_copy");
+        fs::create_dir(&src).unwrap();
+        let dst = temp_dir.join("dst");
+
+        let copied_dir_path = copy_dir(&src, &dst).unwrap();
+
+        assert_eq!(copied_dir_path, dst.join("dir_to_copy"));
+    }
+
+    #[test]
+    fn copies_nested_directories_and_files() {
+        let temp_dir = temp_dir_create!();
+        let src = temp_dir.join("dir_to_copy");
+        fs::create_dir(&src).unwrap();
+        File::create(src.join("root.txt")).unwrap();
+
+        let sub_dir1 = src.join("subdir1");
+        fs::create_dir(&sub_dir1).unwrap();
+        File::create(sub_dir1.join("subdir1.txt")).unwrap();
+
+        let sub_dir2 = src.join("subdir2");
+        fs::create_dir(&sub_dir2).unwrap();
+        File::create(sub_dir2.join("subdir2.txt")).unwrap();
+
+        let dst = temp_dir.join("dst");
+
+        copy_dir(&src, &dst).unwrap();
+
+        assert_dir_eq!(
+            &dst,
+            "* dir_to_copy/
+            ** subdir1/
+            *** subdir1.txt
+            ** subdir2/
+            *** subdir2.txt
+            ** root.txt"
+        );
+    }
+}

mithril-client-cli/src/utils/mod.rs

@@ -6,6 +6,7 @@ mod cardano_db;
 mod cardano_db_download_checker;
 mod expander;
 mod feedback_receiver;
+mod fs;
 mod github_release_retriever;
 mod http_downloader;
 mod multi_download_progress_reporter;
@@ -16,6 +17,7 @@ pub use cardano_db::*;
 pub use cardano_db_download_checker::*;
 pub use expander::*;
 pub use feedback_receiver::*;
+pub use fs::*;
 pub use github_release_retriever::*;
 pub use http_downloader::*;
 pub use multi_download_progress_reporter::*;