mod chat_handlers;
mod message_response_stream;
mod references;

use axum::{
    extract::FromRef,
    routing::{get, post},
    Router,
};
use chat_handlers::{
    new_chat_user_message, new_user_message, show_chat_base, show_existing_chat,
    show_initialized_chat,
};
use message_response_stream::get_response_stream;
use references::show_reference_tooltip;

use crate::html_state::HtmlState;

pub fn router<S>() -> Router<S>
where
    S: Clone + Send + Sync + 'static,
    HtmlState: FromRef<S>,
{
    Router::new()
        .route("/chat", get(show_chat_base).post(new_chat_user_message))
        .route("/chat/:id", get(show_existing_chat).post(new_user_message))
        .route("/initialized-chat", post(show_initialized_chat))
        .route("/chat/response-stream", get(get_response_stream))
        .route("/chat/reference/:id", get(show_reference_tooltip))
}