From 26f09218f4b307a188b0148820c059948e0428f7 Mon Sep 17 00:00:00 2001
From: Patryk Hegenberg <patrykhegenberg@gmail.com>
Date: Sun, 28 Apr 2024 10:03:22 +0200
Subject: [PATCH] fix(shuttle): fix imports after code refactoring

---
 src/shuttle.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/shuttle.rs b/src/shuttle.rs
index d9ee51d..c0cc0c0 100644
--- a/src/shuttle.rs
+++ b/src/shuttle.rs
@@ -1,6 +1,7 @@
+use crate::relay::appstate::AppState;
+use crate::relay::server::ws_handler;
 use axum::{routing::get, Router};
 use axum_client_ip::SecureClientIpSource;
-use relay::server;
 use shuttle_axum::ShuttleAxum;
 
 pub mod receiver;
@@ -11,11 +12,11 @@ pub mod shared;
 #[shuttle_runtime::main]
 async fn axum() -> ShuttleAxum {
     // Create a new server data structure.
-    let appstate = server::AppState::new();
+    let appstate = AppState::new();
 
     // Set up the application routes.
     let app = Router::new()
-        .route("/ws", get(relay::ws_handler))
+        .route("/ws", get(ws_handler))
         .with_state(appstate)
         .layer(SecureClientIpSource::ConnectInfo.into_extension());