cleaup code

server/src/websocket/mod.rs

@@ -0,0 +1,23 @@
+use warp::filters::ws::{Message, WebSocket};
+use futures_util::{SinkExt, StreamExt};
+use tokio::sync::mpsc::channel;
+use std::sync::Arc;
+
+use crate::bridge::ClientManager;
+
+pub mod server;
+
+pub async fn handle_ws_connection(websocket: WebSocket, clients: Arc<ClientManager>) {
+    let (mut ws_tx, _) = websocket.split();
+    let (tx, mut rx) = channel::<String>(100);
+
+    clients.add_client(tx).await;
+
+    tokio::spawn(async move {
+        while let Some(msg) = rx.recv().await {
+            if ws_tx.send(Message::text(msg)).await.is_err() {
+                break;
+            }
+        }
+    });
+}