From 35eb20ccdee9a48ff70b23101062bcdc7b94ca0e Mon Sep 17 00:00:00 2001
From: lennard <lennardkittner@icloud.com>
Date: Sat, 10 Jun 2023 12:23:46 +0200
Subject: [PATCH] Even more error handling

---
 src/main.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index a84cfdb..64c55d3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -36,7 +36,12 @@ fn main() {
                 continue;
             }
             Err(DeviceError::NoDeviceFound()) => {
-                eprintln!("Error: {}", DeviceError::NoDeviceFound());
+                eprintln!("{}", DeviceError::NoDeviceFound());
+                handle.update(|tray: &mut BatteryTray| { tray.no_device_found(); });
+                continue;
+            }
+            Err(DeviceError::HeadSetOff()) => {
+                eprintln!("{}", DeviceError::HeadSetOff());
                 handle.update(|tray: &mut BatteryTray| { tray.no_device_found(); });
                 continue;
             }