1 files changed, 66 insertions, 0 deletions
diff --git a/src/room/websocket.rs b/src/room/websocket.rs
new file mode 100644
index 0000000..50a4537
--- /dev/null
+++ b/src/room/websocket.rs
@@ -0,0 +1,66 @@
+use std::time::Duration;
+
+use axum::extract::ws::{WebSocket, Message};
+use tokio::sync::mpsc;
+
+use super::RoomMessage;
+use super::messages::ServerMessage;
+
+// set up some senders and receivers so that the websocket can receive messages from the task, send messages to the task, and notify the task when it closes
+pub async fn start_ws(mut ws: WebSocket, id: usize, tx: mpsc::Sender<RoomMessage>, mut rx: mpsc::Receiver<ServerMessage>) {
+	loop {
+		tokio::select! {
+			m = ws.recv() => { // receive from the websocket and send it to `tx`
+				if let Some(Ok(msg)) = m {
+					// get the string contents
+					let optionstring = match msg {
+						Message::Text(s) => {
+							Some(s)
+						},
+						Message::Binary(bin) => {
+							String::from_utf8(bin).ok()
+						},
+						Message::Close(_) => { // quit the whole loop on disconnect
+							break;
+						},
+						_ => None
+					};
+					
+					// ignore things that aren't strings
+					let Some(s) = optionstring else {
+						continue;
+					};
+					
+					// decode and send to the room
+					match serde_json::from_str(&s) {
+						Ok(message) => {
+							tx.send_timeout(RoomMessage::WsMessage(id, message), Duration::from_secs(1)).await.ok();
+						},
+						Err(e) => { // let the client know if they sent a bad message
+							if let Ok(text) = serde_json::to_string(&ServerMessage::Error{
+								error: format!("Failed to decode JSON message: {}: {}", e, s),
+							}) {
+								ws.send(Message::Text(text)).await.ok();
+							}
+						}
+					}
+				} else { // websocket error
+					break;
+				}
+			}
+			s = rx.recv() => { // receive from `rx` and send it to the websocket
+				if let Some(msg) = s {
+					if let Ok(string) = serde_json::to_string(&msg) {
+						ws.send(Message::Text(string)).await.ok();
+					}
+				} else { // shouldn't happen but this is if the room drops the sender, it should close the websocket anyways
+					break;
+				}
+			}
+		}
+	}
+	
+	// websocket disconnect due to either error or normal disconnect
+	// notify the room that the socket should be removed
+	tx.send_timeout(RoomMessage::Remove(id), Duration::from_secs(1)).await.ok();
+}