From 6869b08a772a5aab1d303120d8e49faf35ca0b7f Mon Sep 17 00:00:00 2001
From: Andrew Phillips <andrew.phillips2@canada.ca>
Date: Tue, 12 Aug 2025 15:01:38 -0300
Subject: [PATCH] refactor: restructure server routing and add documentation
 routes

---
 src/modes/server.rs      | 2 +-
 src/modes/server/docs.rs | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/modes/server.rs b/src/modes/server.rs
index 1128e36..aaa9a4a 100644
--- a/src/modes/server.rs
+++ b/src/modes/server.rs
@@ -59,7 +59,7 @@ async fn run_server(
         args: Arc::new(args.clone()),
     };
     
-    let app = api::api_router()
+    let app = Router::new()
         .layer(axum::middleware::from_fn(logging_middleware))
         .layer(
             ServiceBuilder::new()
diff --git a/src/modes/server/docs.rs b/src/modes/server/docs.rs
index 04f4511..cdc04e9 100644
--- a/src/modes/server/docs.rs
+++ b/src/modes/server/docs.rs
@@ -81,3 +81,10 @@ pub async fn handle_swagger_ui() -> Html<&'static str> {
     
     Html(html)
 }
+
+pub fn add_routes(router: Router<AppState>) -> Router<AppState> {
+    router
+        // Documentation endpoints
+        .route("/openapi.json", get(docs::handle_openapi))
+        .route("/swagger/", get(docs::handle_swagger_ui))
+}