From 4c09716ad8e5d913c7e4ee8264573dd1329e57df Mon Sep 17 00:00:00 2001
From: Saeed Rasooli <saeed.gnu@gmail.com>
Date: Sun, 4 Apr 2021 04:38:17 +0430
Subject: [PATCH] add TiB and PiB (#3257)

---
 crates/nu-parser/src/parse.rs | 22 ++++++++++++++++++++++
 crates/nu-protocol/src/hir.rs |  8 ++++++++
 2 files changed, 30 insertions(+)

diff --git a/crates/nu-parser/src/parse.rs b/crates/nu-parser/src/parse.rs
index f1fcca5200..7a2f7667e6 100644
--- a/crates/nu-parser/src/parse.rs
+++ b/crates/nu-parser/src/parse.rs
@@ -341,6 +341,8 @@ fn parse_unit(lite_arg: &Spanned<String>) -> (SpannedExpression, Option<ParseErr
         (Unit::Kibibyte, "KIB", Some((Unit::Byte, 1024))),
         (Unit::Mebibyte, "MIB", Some((Unit::Kibibyte, 1024))),
         (Unit::Gibibyte, "GIB", Some((Unit::Mebibyte, 1024))),
+        (Unit::Tebibyte, "TIB", Some((Unit::Gibibyte, 1024))),
+        (Unit::Pebibyte, "PIB", Some((Unit::Tebibyte, 1024))),
         (Unit::Byte, "B", None),
         (Unit::Nanosecond, "NS", None),
         (Unit::Microsecond, "US", Some((Unit::Nanosecond, 1000))),
@@ -2179,6 +2181,26 @@ fn unit_parse_byte_units() {
             value: 123,
             unit: Unit::Gibibyte,
         },
+        TestCase {
+            string: String::from("10tib"),
+            value: 10,
+            unit: Unit::Tebibyte,
+        },
+        TestCase {
+            string: String::from("123TiB"),
+            value: 123,
+            unit: Unit::Tebibyte,
+        },
+        TestCase {
+            string: String::from("10pib"),
+            value: 10,
+            unit: Unit::Pebibyte,
+        },
+        TestCase {
+            string: String::from("123PiB"),
+            value: 123,
+            unit: Unit::Pebibyte,
+        },
     ];
 
     for case in cases.iter() {
diff --git a/crates/nu-protocol/src/hir.rs b/crates/nu-protocol/src/hir.rs
index 1229131039..f58750d70e 100644
--- a/crates/nu-protocol/src/hir.rs
+++ b/crates/nu-protocol/src/hir.rs
@@ -356,6 +356,8 @@ pub enum Unit {
     Kibibyte,
     Mebibyte,
     Gibibyte,
+    Tebibyte,
+    Pebibyte,
 
     // Duration units
     Nanosecond,
@@ -546,6 +548,8 @@ impl Unit {
             Unit::Kibibyte => "KiB",
             Unit::Mebibyte => "MiB",
             Unit::Gibibyte => "GiB",
+            Unit::Tebibyte => "TiB",
+            Unit::Pebibyte => "PiB",
             Unit::Nanosecond => "ns",
             Unit::Microsecond => "us",
             Unit::Millisecond => "ms",
@@ -575,6 +579,10 @@ impl Unit {
             Unit::Kibibyte => filesize(convert_number_to_u64(&size) * 1024),
             Unit::Mebibyte => filesize(convert_number_to_u64(&size) * 1024 * 1024),
             Unit::Gibibyte => filesize(convert_number_to_u64(&size) * 1024 * 1024 * 1024),
+            Unit::Tebibyte => filesize(convert_number_to_u64(&size) * 1024 * 1024 * 1024 * 1024),
+            Unit::Pebibyte => {
+                filesize(convert_number_to_u64(&size) * 1024 * 1024 * 1024 * 1024 * 1024)
+            }
 
             Unit::Nanosecond => duration(size.to_bigint().expect("Conversion should never fail.")),
             Unit::Microsecond => {