feat: add notifications when errors

src/main.rs

@@ -1,6 +1,7 @@
 use std::error::Error;
 
 mod websocket;
+mod notifications;
 
 fn main() -> Result<(), Box<dyn Error>> {
     dotenvy::dotenv()?;

src/notifications.rs

@@ -0,0 +1,11 @@
+use notify_rust::Notification;
+
+pub fn send_error_notification(body: &str) {
+    Notification::new()
+        .summary("Computer Statistics Error")
+        .body(body)
+        .icon("dialog-error")
+        .timeout(0)
+        .show()
+        .unwrap();
+}

src/websocket.rs

@@ -2,11 +2,15 @@ use std::{env, net::TcpStream};
 
 use tungstenite::{stream::MaybeTlsStream, Message, WebSocket};
 
+use crate::notifications;
+
 pub fn connect() -> Result<WebSocket<MaybeTlsStream<TcpStream>>, tungstenite::Error> {
     let (socket, _) = match tungstenite::connect(env::var("WEBSOCKET_URL").unwrap()) {
         Ok(ws) => ws,
         Err(err) => {
-            eprintln!("Unable to connect to WebSocket");
+            eprintln!("Unable to connect to WebSocket: {}", err);
+            notifications::send_error_notification("Unable to connect to WebSocket");
+
             return Err(err);
         }
     };