add chat function

#

src/main.rs

@@ -0,0 +1,46 @@
+mod state;
+mod handlers;
+mod models;
+
+use axum::{
+    routing::{get, post},
+    Router,
+};
+use std::sync::Arc;
+use tower_http::services::ServeDir;
+use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
+
+use state::AppState;
+
+#[tokio::main]
+async fn main() {
+    tracing_subscriber::registry()
+        .with(tracing_subscriber::EnvFilter::new(
+            std::env::var("RUST_LOG").unwrap_or_else(|_| "chat_service=debug,info".into()),
+        ))
+        .with(tracing_subscriber::fmt::layer())
+        .init();
+
+    let state = Arc::new(AppState::new());
+
+    let app = Router::new()
+        .route("/", get(handlers::pages::customer_page))
+        .route("/agent", get(handlers::pages::agent_page))
+        .route("/ws/customer/:session_id", get(handlers::ws::customer_ws))
+        .route("/ws/agent/:agent_id", get(handlers::ws::agent_ws))
+        .route("/messages/:session_id", get(handlers::chat::get_messages))
+        .route("/send/:session_id", post(handlers::chat::send_message))
+        .route("/agent/send/:session_id", post(handlers::chat::agent_send_message))
+        .route("/sessions", get(handlers::chat::get_sessions))
+        .route("/sessions/new", post(handlers::chat::new_session))
+        .nest_service("/static", ServeDir::new("static"))
+        .with_state(state);
+
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
+    tracing::info!("Chat service running — http://0.0.0.0:3000");
+    axum::Server::from_tcp(listener.into_std().unwrap())
+        .unwrap()
+        .serve(app.into_make_service())
+        .await
+        .unwrap();
+}