fix: Add display for unknown battery state (#316)

1 files changed, 11 insertions, 0 deletions

diff --git a/src/modules/battery.rs b/src/modules/battery.rs
index 881893940..3ecc651f3 100644
--- a/src/modules/battery.rs
+++ b/src/modules/battery.rs
@@ -45,6 +45,17 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
             battery::State::Discharging => {
                 module.new_segment("discharging_symbol", BATTERY_DISCHARGING);
             }
+            battery::State::Unknown => {
+                log::debug!("Unknown detected");
+                module.new_segment_if_config_exists("unknown_symbol")?;
+            }
+            battery::State::Empty => {
+                module.new_segment_if_config_exists("empty_symbol")?;
+            }
+            _ => {
+                log::debug!("Unhandled battery state `{}`", state);
+                return None;
+            }
             _ => return None,
         }