Add total_size function to estimator module

Author: EncodePanda

src/crufty/estimator.rs

@@ -4,6 +4,15 @@ use walkdir::WalkDir;
 
 use super::types::{ArtifactCandidate, Size};
 
+pub fn total_size(artifacts: &[ArtifactCandidate]) -> u64 {
+  artifacts.iter().fold(0, |acc, artifact| {
+    match &artifact.size {
+      Size::UnknownSize => acc,
+      Size::KnownSize(size) => acc + size,
+    }
+  })
+}
+
 pub fn estimate(artifact: &mut ArtifactCandidate) -> &ArtifactCandidate {
   let size = estimate_path(&artifact.path);
   artifact.size = size;
@@ -18,7 +27,7 @@ fn estimate_path(path: &PathBuf) -> Size {
 }
 
 fn calculate_dir_size(path: &PathBuf) -> std::io::Result<u64> {
-  let mut total_size = 0;
+  let mut total = 0;
 
   for entry in WalkDir::new(path)
     .follow_links(false)
@@ -27,11 +36,11 @@ fn calculate_dir_size(path: &PathBuf) -> std::io::Result<u64> {
     .filter(|e| e.file_type().is_file())
   {
     if let Ok(metadata) = fs::metadata(entry.path()) {
-      total_size += metadata.len();
+      total += metadata.len();
     }
   }
 
-  Ok(total_size)
+  Ok(total)
 }
 
 #[cfg(test)]