Add Size with a Display impl

Author: EncodePanda

src/crufty.rs

@@ -1,20 +1,49 @@
 #![allow(dead_code)]
+use std::fmt;
 use std::path::PathBuf;
 
 use globset::{Glob, GlobSetBuilder};
 use ignore::WalkBuilder;
 
 pub mod cli;
 
+#[derive(Debug, PartialEq, Eq, Clone)]
+pub enum Size {
+  UnknownSize,
+  KnownSize(u64),
+}
+
+impl fmt::Display for Size {
+  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+    match self {
+      Size::UnknownSize => write!(f, "unknown"),
+      Size::KnownSize(bytes) => {
+        if *bytes < 1024 {
+          write!(f, "{} B", bytes)
+        } else if *bytes < 1024 * 1024 {
+          write!(f, "{:.1} KB", *bytes as f64 / 1024.0)
+        } else if *bytes < 1024 * 1024 * 1024 {
+          write!(f, "{:.1} MB", *bytes as f64 / (1024.0 * 1024.0))
+        } else {
+          write!(f, "{:.1} GB", *bytes as f64 / (1024.0 * 1024.0 * 1024.0))
+        }
+      }
+    }
+  }
+}
+
 #[derive(Debug, PartialEq, Eq)]
 pub struct ArtifactCandidate {
   pub path: PathBuf,
-  size: Option<u64>,
+  pub size: Size,
 }
 
 impl ArtifactCandidate {
   fn new(path: PathBuf) -> Self {
-    ArtifactCandidate { path, size: None }
+    ArtifactCandidate {
+      path,
+      size: Size::UnknownSize,
+    }
   }
 }
 

src/main.rs

@@ -1,7 +1,7 @@
 use clap::Parser;
 use console::{style, Term};
 use crufty::cli::{Cli, Commands};
-use crufty::fetch_artifacts;
+use crufty::{fetch_artifacts, Size};
 use std::env;
 use std::io;
 use std::process;
@@ -33,11 +33,16 @@ fn scan() -> io::Result<()> {
     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::KnownSize(_) => style(format!("{}", artifact.size)).green(),
+      };
+
       term.write_line(&format!(
         "[{}] {:<30} {}",
         i + 1,
         style(format!("./{}", rel_path.display())).bold(),
-        style(format!("unknown size")).yellow()
+        size_display
       ))?;
     }
     Ok(())