don't show crap for directory size

1 files changed, 8 insertions, 0 deletions

diff --git a/src/files.rs b/src/files.rs
index 6cb3611..4bd0f45 100644
--- a/src/files.rs
+++ b/src/files.rs
@@ -273,6 +273,14 @@ impl File {
     }
 
     pub fn calculate_size(&self) -> (usize, String) {
+        if self.is_dir() {
+            let dir_iterator = std::fs::read_dir(&self.path);
+            match dir_iterator {
+                Ok(dir_iterator) => return (dir_iterator.count(), "".to_string()),
+                Err(_) => return (0, "".to_string())
+            }
+        }
+
         let mut unit = 0;
         let mut size = self.size.unwrap();
         while size > 1024 {