Add byte formatting

Byron · Jun 1, 2019 · 6db07e2 · 6db07e2
1 parent a19e3d7
commit 6db07e2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -10,6 +10,7 @@ failure = "0.1.1"
 failure-tools = "4.0.2"
 structopt = "0.2.14"
 jwalk = "0.4.0"
+byte-unit = "2.1.0"
 
 [[bin]]
 name="dua"

diff --git a/src/lib.rs b/src/lib.rs
@@ -6,6 +6,12 @@ pub struct WalkOptions {
 }
 
 impl WalkOptions {
+    pub fn format_bytes(&self, b: u64) -> String {
+        use byte_unit::Byte;
+        Byte::from_bytes(b as u128)
+            .get_appropriate_unit(false)
+            .format(2)
+    }
     pub fn iter_from_path(&self, path: &Path) -> WalkDir {
         WalkDir::new(path)
             .preload_metadata(true)
@@ -57,7 +63,12 @@ mod aggregate {
                 }
             }
 
-            writeln!(out, "{}\t{}", num_bytes, path.as_ref().display())?;
+            writeln!(
+                out,
+                "{}\t{}",
+                options.format_bytes(num_bytes),
+                path.as_ref().display()
+            )?;
         }
         Ok(res)
     }

diff --git a/tests/snapshots/success-no-arguments b/tests/snapshots/success-no-arguments
@@ -1 +1 @@
-1258947	.
+1.26 MB	.