From fde8bf62d451c621fb1bc05f9564140ffdc1672a Mon Sep 17 00:00:00 2001
From: Roman Perepelitsa <roman.perepelitsa@gmail.com>
Date: Fri, 10 Dec 2021 23:34:10 +0100
Subject: [PATCH] fix prefix-based formatting of numbers that are slightly
 below the boundary (#1664)

---
 internal/p10k.zsh | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/internal/p10k.zsh b/internal/p10k.zsh
index dfd1c1b..754c312 100644
--- a/internal/p10k.zsh
+++ b/internal/p10k.zsh
@@ -317,16 +317,24 @@ function _p9k_prompt_length() {
 
 typeset -gr __p9k_byte_suffix=('B' 'K' 'M' 'G' 'T' 'P' 'E' 'Z' 'Y')
 
-# 42 => 42B
-# 1536 => 1.5K
+# 512 => 512B
+# 1800 => 1.76K
+# 18000 => 17.6K
 function _p9k_human_readable_bytes() {
-  typeset -F 2 n=$1
+  typeset -F n=$1
   local suf
   for suf in $__p9k_byte_suffix; do
-    (( n < 100 )) && break
+    (( n < 1024 )) && break
     (( n /= 1024 ))
   done
-  _p9k__ret=${${n%%0#}%.}$suf
+  if (( n >= 100 )); then
+    printf -v _p9k__ret '%.0f.' $n
+  elif (( n >= 10 )); then
+    printf -v _p9k__ret '%.1f' $n
+  else
+    printf -v _p9k__ret '%.2f' $n
+  fi
+  _p9k__ret=${${_p9k__ret%%0#}%.}$suf
 }
 
 if is-at-least 5.4; then