Refactor AIS server to use Axum framework with shared stream manager and state handling. Fix metadata key mismatch in frontend vessel mapper.

crates/ais/Cargo.toml

@@ -14,6 +14,7 @@ axum = { version = "0.7", features = ["ws"] }
 tower = "0.4"
 tower-http = { version = "0.5", features = ["cors"] }
 base64 = "0.22.1"
+tokio-util = "0.7.15"
 
 [dev-dependencies]
 tokio-test = "0.4"

crates/ais/src/main.rs

@@ -1,10 +1,36 @@
-use crate::ais::start_ais_server;
+use std::sync::Arc;
+use axum::Router;
+use axum::routing::get;
+use tower_http::cors::CorsLayer;
+use crate::ais::{AisStreamManager, AppState};
 
 mod ais;
 
 #[tokio::main]
-async fn main() {
-    if let Err(e) = start_ais_server().await {
-        eprintln!("Server error: {:?}", e);
-    }
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    // Create the shared state with the AIS stream manager
+    let state = AppState {
+        ais_stream_manager: Arc::new(AisStreamManager::new()),
+    };
+
+    // Create and start the Axum HTTP server
+    let app = create_router(state);
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await?;
+
+    println!("AIS server running on http://0.0.0.0:3000");
+
+    axum::serve(listener, app)
+        .with_graceful_shutdown(ais::shutdown_signal())
+        .await?;
+
+    Ok(())
+}
+
+// Create the Axum router
+fn create_router(state: AppState) -> Router {
+    Router::new()
+        .route("/ais", get(crate::ais::get_ais_data))
+        .route("/ws", get(crate::ais::websocket_handler))
+        .layer(CorsLayer::permissive())
+        .with_state(state)
 }