From 4a7ed2d4c887ba7a528bce5e4b1b38d4c31fb2fe Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 21 Jul 2019 13:15:29 +0800
Subject: [PATCH] add TB

---
 src/lib.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 1b15781..63a7975 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -70,7 +70,10 @@ pub fn get_display_size(size: i64) -> String {
     if size < SIZE_PB {
         return format!("{:.*}GB", 2, (size as f64) / 1024. / 1024. / 1024.);
     }
-    return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
+    if size < SIZE_TB {
+        return format!("{:.*}PB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024.);
+    }
+    return format!("{:.*}TB", 2, (size as f64) / 1024. / 1024. / 1024. / 1024. / 1024.);
 }
 
 pub fn run_command_and_wait(cmd: &mut Command) -> io::Result<()> {