From 688e41e401fc3b3b59b1a609a04e59dc507487f6 Mon Sep 17 00:00:00 2001
From: Ben S <ogham@bsago.me>
Date: Sat, 3 May 2014 14:26:49 +0100
Subject: [PATCH] Add file size column

---
 exa.rs | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/exa.rs b/exa.rs
index da6a7e77..9fd30f50 100644
--- a/exa.rs
+++ b/exa.rs
@@ -22,6 +22,10 @@ enum FileName {
     FileName,
 }
 
+struct FileSize {
+    useSIPrefixes: bool,
+}
+
 trait Column {
     fn display(&self, stat: &io::FileStat, filename: &str) -> ~str;
 }
@@ -50,6 +54,31 @@ impl Column for Permissions {
     }
 }
 
+impl Column for FileSize {
+    fn display(&self, stat: &io::FileStat, filename: &str) -> ~str {
+        let sizeStr = if self.useSIPrefixes {
+            formatBytes(stat.size, 1024, ~[ "B  ", "KiB", "MiB", "GiB", "TiB" ])
+        } else {
+            formatBytes(stat.size, 1000, ~[ "B ", "KB", "MB", "GB", "TB" ])
+        };
+
+        return if stat.kind == io::TypeDirectory {
+            Green.normal()
+        } else {
+            Green.bold()
+        }.paint(sizeStr);
+    }
+}
+
+fn formatBytes(mut amount: u64, kilo: u64, prefixes: ~[&str]) -> ~str {
+    let mut prefix = 0;
+    while amount > kilo {
+        amount /= kilo;
+        prefix += 1;
+    }
+    return format!("{:4}{}", amount, prefixes[prefix]);
+}
+
 fn list(path: Path) {
     let mut files = match fs::readdir(&path) {
         Ok(files) => files,
@@ -68,7 +97,12 @@ fn list(path: Path) {
             Err(e) => fail!("Couldn't stat {}: {}", filename, e),
         };
 
-        let columns = ~[~Permissions as ~Column, ~FileName as ~Column];
+        let columns = ~[
+            ~Permissions as ~Column,
+            ~FileSize { useSIPrefixes: false } as ~Column,
+            ~FileName as ~Column
+        ];
+
         let mut cells = columns.iter().map(|c| c.display(&stat, filename));
 
         let mut first = true;