Show total cleaned/to clean amount

fixes holmgr#43
birkenfeld · Dec 3, 2022 · 4d0a722 · 4d0a722
1 parent 95fce4a
commit 4d0a722
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -219,6 +219,8 @@ fn main() -> anyhow::Result<()> {
 
         let dry_run = matches.is_present("dry-run");
 
+        let mut total_cleaned = 0;
+
         // Default to current invocation path.
         let path = match matches.value_of("path") {
             Some(p) => PathBuf::from(p),
@@ -251,9 +253,13 @@ fn main() -> anyhow::Result<()> {
             for project_path in &paths {
                 match remove_not_built_with(project_path, matches.value_of("toolchains"), dry_run) {
                     Ok(cleaned_amount) if dry_run => {
+                        total_cleaned += cleaned_amount;
                         info!("Would clean: {}", format_bytes(cleaned_amount))
                     }
-                    Ok(cleaned_amount) => info!("Cleaned {}", format_bytes(cleaned_amount)),
+                    Ok(cleaned_amount) => {
+                        total_cleaned += cleaned_amount;
+                        info!("Cleaned {}", format_bytes(cleaned_amount))
+                    }
                     Err(e) => error!(
                         "{:?}",
                         e.context(format!("Failed to clean {:?}", project_path))
@@ -275,9 +281,13 @@ fn main() -> anyhow::Result<()> {
             for project_path in &paths {
                 match remove_older_until_fits(project_path, size, dry_run) {
                     Ok(cleaned_amount) if dry_run => {
+                        total_cleaned += cleaned_amount;
                         info!("Would clean: {}", format_bytes(cleaned_amount))
                     }
-                    Ok(cleaned_amount) => info!("Cleaned {}", format_bytes(cleaned_amount)),
+                    Ok(cleaned_amount) => {
+                        total_cleaned += cleaned_amount;
+                        info!("Cleaned {}", format_bytes(cleaned_amount))
+                    }
                     Err(e) => error!("Failed to clean {:?}: {:?}", project_path, e),
                 };
             }
@@ -297,13 +307,19 @@ fn main() -> anyhow::Result<()> {
             for project_path in &paths {
                 match remove_older_than(project_path, &keep_duration, dry_run) {
                     Ok(cleaned_amount) if dry_run => {
+                        total_cleaned += cleaned_amount;
                         info!("Would clean: {}", format_bytes(cleaned_amount))
                     }
-                    Ok(cleaned_amount) => info!("Cleaned {}", format_bytes(cleaned_amount)),
+                    Ok(cleaned_amount) => {
+                        total_cleaned += cleaned_amount;
+                        info!("Cleaned {}", format_bytes(cleaned_amount))
+                    }
                     Err(e) => error!("Failed to clean {:?}: {:?}", project_path, e),
                 };
             }
         }
+
+        info!("Total amount: {}", format_bytes(total_cleaned));
     }
 
     Ok(())