Calculate sizes for artifacts, use progress bar in main

Author: EncodePanda

Cargo.lock

@@ -6,8 +6,10 @@ edition = "2024"
 [dependencies]
 clap = { version = "4.4", features = ["derive"] }
 console = "0.15"
+dialoguer = "0.11.0"
 globset = "0.4"
+indicatif = "0.17.11"
 walkdir = "2.4"
 
 [dev-dependencies]
-assert_fs = "1"
+assert_fs = "1"

Cargo.toml

@@ -2,3 +2,4 @@ pub mod cli;
 pub mod estimator;
 pub mod fetcher;
 pub mod types;
+pub mod ui;

src/crufty.rs

@@ -0,0 +1,14 @@
+use indicatif::{ProgressBar, ProgressStyle};
+
+/// Creates and returns a configured progress bar for use in the application.
+pub fn create_progress_bar(total: u64) -> ProgressBar {
+  let pb = ProgressBar::new(total);
+  let template =
+	"{spinner:.green} [{elapsed_precise}] [{bar:40.cyan/blue}] {pos}/{len} {msg}";
+  let style = ProgressStyle::default_bar()
+    .template(template)
+    .unwrap()
+    .progress_chars("#>-");
+  pb.set_style(style);
+  pb
+}

src/crufty/ui.rs

@@ -1,8 +1,10 @@
 use clap::Parser;
 use console::{style, Term};
 use crufty::cli::{Cli, Commands};
+use crufty::estimator::estimate;
 use crufty::fetcher::fetch_artifacts;
 use crufty::types::Size;
+use crufty::ui;
 use std::env;
 use std::io;
 use std::process;
@@ -26,16 +28,26 @@ fn scan() -> io::Result<()> {
   term
     .write_line(&format!("[+] Scanning: {}", style(path.display()).bold()))?;
   term.write_line("")?;
-  let artifacts = fetch_artifacts(&path);
+
+  // Fetch artifacts
+  let mut artifacts = fetch_artifacts(&path);
 
   if artifacts.is_empty() {
     term.write_line("No significant build artifacts found.")
   } else {
+    let pb = ui::create_progress_bar(artifacts.len() as u64);
+
+    for artifact in artifacts.iter_mut() {
+      estimate(artifact);
+      pb.inc(1);
+    }
+    pb.finish_and_clear();
+
     for (i, artifact) in artifacts.iter().enumerate() {
       let rel_path =
         artifact.path.strip_prefix(&path).unwrap_or(&artifact.path);
       let size_display = match &artifact.size {
-        Size::UnknownSize => style(format!("unknown")).yellow(),
+        Size::UnknownSize => style(format!("{}", artifact.size)).yellow(),
         Size::KnownSize(_) => style(format!("{}", artifact.size)).green(),
       };
 
@@ -46,6 +58,13 @@ fn scan() -> io::Result<()> {
         size_display
       ))?;
     }
+
+    term.write_line("")?;
+    term.write_line(&format!(
+      "{} Scan completed successfully",
+      style("✓").green()
+    ))?;
+
     Ok(())
   }
 }