From 53e288a6f7e6339c59789e15e55aa25daa5e6672 Mon Sep 17 00:00:00 2001
From: Adam <24621027+adoyle0@users.noreply.github.com>
Date: Sat, 4 May 2024 02:46:10 -0400
Subject: [PATCH] oops

---
 src/api.rs                 | 20 ++++++++++++++++++--
 src/api/message_handler.rs | 15 ---------------
 2 files changed, 18 insertions(+), 17 deletions(-)

diff --git a/src/api.rs b/src/api.rs
index 3934555..804fbf1 100644
--- a/src/api.rs
+++ b/src/api.rs
@@ -1,6 +1,12 @@
 use crate::AppState;
-use axum::extract::ws::{Message, WebSocket};
-use futures::{sink::SinkExt, stream::StreamExt};
+use axum::{
+    extract::{
+        ws::{Message, WebSocket},
+        ConnectInfo, State, WebSocketUpgrade,
+    },
+    response::IntoResponse,
+};
+use futures::{SinkExt, StreamExt};
 use std::{net::SocketAddr, sync::Arc};
 pub mod message_handler;
 use crate::message_handler::*;
@@ -14,6 +20,16 @@ fn greeting(state: &Arc<AppState>) -> String {
     )
 }
 
+pub async fn websocket_handler(
+    ws: WebSocketUpgrade,
+    // user_agent: Option<TypedHeader<headers::UserAgent>>,
+    ConnectInfo(addr): ConnectInfo<SocketAddr>,
+    State(state): State<Arc<AppState>>,
+) -> impl IntoResponse {
+    tracing::debug!("New connection from {addr}");
+    ws.on_upgrade(move |socket| websocket(socket, state, addr))
+}
+
 pub async fn websocket(stream: WebSocket, state: Arc<AppState>, who: SocketAddr) {
     // By splitting, we can send and receive at the same time.
     let (mut sender, mut receiver) = stream.split();
diff --git a/src/api/message_handler.rs b/src/api/message_handler.rs
index 98de9a8..a03e234 100644
--- a/src/api/message_handler.rs
+++ b/src/api/message_handler.rs
@@ -1,13 +1,8 @@
 use crate::api::{greeting, Message, SocketAddr};
-use crate::websocket;
 use crate::AppState;
 use crate::Arc;
 use crate::CAHGame;
 use crate::CAHPlayer;
-use axum::{
-    extract::{ConnectInfo, State, WebSocketUpgrade},
-    response::IntoResponse,
-};
 use serde::Deserialize;
 
 /// New game request structure
@@ -31,16 +26,6 @@ pub struct GameJoinRequest {
     pub player: CAHPlayer,
 }
 
-pub async fn websocket_handler(
-    ws: WebSocketUpgrade,
-    // user_agent: Option<TypedHeader<headers::UserAgent>>,
-    ConnectInfo(addr): ConnectInfo<SocketAddr>,
-    State(state): State<Arc<AppState>>,
-) -> impl IntoResponse {
-    tracing::debug!("New connection from {addr}");
-    ws.on_upgrade(move |socket| websocket(socket, state, addr))
-}
-
 pub async fn message_handler(message: Message, state: &Arc<AppState>, who: SocketAddr) {
     let tx = &state.tx;