feat: optimize main function; proper error handling

src/main.rs

@@ -12,10 +12,11 @@ fn main() {
     env_logger::init();
 
     // start tray service
-    start_tray_service();
+    match start_tray_service() {
+        Ok(_) => println!("Tray service started successfully."),
+        Err(e) => eprintln!("Failed to start the tray service: {}", e),
+    }
 
     // keep the main thread alive
-    loop {
-        park();
-    }
+    park();
 }

src/tray/utils.rs

@@ -1,13 +1,15 @@
 use crate::tailscale;
 use crate::tray::menu::SysTray;
+use std::error::Error;
 
-pub fn start_tray_service() {
-    // start the tray service
-    let _handle = ksni::spawn(SysTray {
-        ctx: tailscale::status::get_current_status()
-            .expect("Failed to update Tailscale status! Is Tailscale daemon running?"),
-    })
-    .unwrap_or_else(|e| {
-        panic!("Failed to start the tray service: {}", e);
-    });
+type TrayServiceError = Box<dyn Error>;
+
+pub fn start_tray_service() -> Result<(), TrayServiceError> {
+    let status = tailscale::status::get_current_status()
+        .map_err(|e| format!("Failed to update Tailscale status: {}", e))?;
+
+    let _handle = ksni::spawn(SysTray { ctx: status })
+        .map_err(|e| format!("Failed to spawn Tray implementation: {}", e))?;
+
+    Ok(())
 }