mod chat_handlers;
mod message_response_stream;
mod reference_validation;
mod references;

use axum::{Router, extract::FromRef, routing::get};
pub use chat_handlers::{
    delete_conversation, new_chat_user_message, new_user_message, patch_conversation_title,
    reload_sidebar, show_chat_base as show_base, show_conversation_editing_title,
    show_existing_chat as show_existing,
};
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_base).post(new_chat_user_message))
        .route(
            "/chat/{id}",
            get(show_existing)
                .post(new_user_message)
                .delete(delete_conversation),
        )
        .route(
            "/chat/{id}/title",
            get(show_conversation_editing_title).patch(patch_conversation_title),
        )
        .route("/chat/sidebar", get(reload_sidebar))
        .route("/chat/response-stream", get(get_response_stream))
        .route("/chat/reference/{id}", get(show_reference_tooltip))
}