From 0ab8ad96608b8960f5b78ec69ef165b6fb270184 Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Mon, 23 Dec 2024 19:55:32 -0600
Subject: [PATCH] Copy send_message result for send_state

---
 src/main.rs   | 8 ++++++--
 src/models.rs | 4 ++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 58aecc3..7e43c3c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -105,7 +105,9 @@ async fn handle_socket(session_id: u32, websocket: WebSocket) {
         .expect("Unable to get session");
     session.tx = Some(tx_channel);
 
-    session.send_state();
+    session
+        .send_state()
+        .expect("Failed to buffer state message");
     session
         .send_message(executable_message)
         .expect("Failed to buffer executables message");
@@ -196,7 +198,9 @@ pub async fn download(req: &mut Request, res: &mut Response, depot: &mut Depot)
 
     // Don't try to send state if somehow the session has not connected
     if session.tx.is_some() {
-        session.send_state();
+        session
+            .send_state()
+            .expect("Failed to buffer state message");
     } else {
         tracing::warn!("Download being made without any connection websocket");
     }
diff --git a/src/models.rs b/src/models.rs
index e527884..108086d 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -67,12 +67,12 @@ impl Session {
         }
     }
 
-    pub fn send_state(&mut self) {
+    pub fn send_state(&mut self) -> Result<(), anyhow::Error> {
         let message = OutgoingMessage::State {
             session: self.clone(),
         };
 
-        self.send_message(message);
+        self.send_message(message)
     }
 }