handle overfull battery properly

1 files changed, 9 insertions, 7 deletions

diff --git a/src/prompt.rs b/src/prompt.rs
index 98d2e7b..8c8eece 100644
--- a/src/prompt.rs
+++ b/src/prompt.rs
@@ -146,15 +146,17 @@ impl Prompt {
             let charging = self.data.power_info.charging();
             let color = battery_discharge_color(battery_usage, charging);
             let filled = (battery_usage * (len as f64)).ceil() as usize;
-            let unfilled = len - filled;
-            if unfilled > 0 {
+            if len > filled {
+                let unfilled = len - filled;
                 self.colors.print(color, &"-".repeat(unfilled));
             }
-            if charging {
-                self.colors.print("battery_charging", "<");
-            }
-            else {
-                self.colors.print(color, ">");
+            if len >= filled {
+                if charging {
+                    self.colors.print("battery_charging", "<");
+                }
+                else {
+                    self.colors.print(color, ">");
+                }
             }
             if filled > 1 {
                 self.colors