mod handlers;
mod models;
mod state;

use axum::{
    Router,
    routing::{get, post},
};
use std::{net::SocketAddr, sync::Arc};
use tower_http::services::ServeDir;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

use state::{AppState, CHAT_RESET_INTERVAL};

#[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 reset_state = state.clone();
    tokio::spawn(async move {
        loop {
            tokio::time::sleep(CHAT_RESET_INTERVAL).await;
            reset_state.clear_everything();
            tracing::info!(
                "Cleared all in-memory chat sessions, messages, notifiers, and rate limits"
            );
        }
    });

    let app = Router::new()
        .route("/", get(handlers::pages::index_page))
        .route("/user", get(handlers::pages::user_page))
        .route("/supporter", get(handlers::pages::supporter_page))
        .route("/reset-countdown", get(handlers::pages::reset_countdown))
        // New API endpoint for creating sessions without a form
        .route("/api/session/create", post(handlers::chat::create_session))
        .route("/ws/user/:session_id", get(handlers::ws::user_ws))
        .route(
            "/ws/supporter/:supporter_id",
            get(handlers::ws::supporter_ws),
        )
        .route("/messages/:session_id", get(handlers::chat::get_messages))
        .route(
            "/supporter/messages/:session_id",
            get(handlers::chat::get_supporter_messages),
        )
        .route("/send/:session_id", post(handlers::chat::send_message))
        .route(
            "/supporter/send/:session_id",
            post(handlers::chat::supporter_send_message),
        )
        .route("/sessions", get(handlers::chat::get_sessions))
        // Removed: .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:3111").await.unwrap();
    tracing::info!("Chat service running — http://0.0.0.0:3111");
    axum::Server::from_tcp(listener.into_std().unwrap())
        .unwrap()
        .serve(app.into_make_service_with_connect_info::<SocketAddr>())
        .await
        .unwrap();
}