furumi-ng/furumi-agent/src/web/mod.rs

pub mod api;

use std::sync::Arc;

use axum::{Router, routing::{get, post, put}};
use sqlx::PgPool;

use crate::config::Args;

#[derive(Clone)]
pub struct AppState {
    pub pool: PgPool,
    pub config: Arc<Args>,
    pub system_prompt: Arc<String>,
    pub merge_prompt: Arc<String>,
}

pub fn build_router(state: Arc<AppState>) -> Router {
    let api = Router::new()
        .route("/stats", get(api::stats))
        .route("/queue", get(api::list_queue))
        .route("/queue/:id", get(api::get_queue_item).delete(api::delete_queue_item))
        .route("/queue/:id/approve", post(api::approve_queue_item))
        .route("/queue/:id/reject", post(api::reject_queue_item))
        .route("/queue/:id/retry", post(api::retry_queue_item))
        .route("/queue/:id/update", put(api::update_queue_item))
        .route("/queue/batch/approve", post(api::batch_approve))
        .route("/queue/batch/reject", post(api::batch_reject))
        .route("/queue/batch/retry", post(api::batch_retry))
        .route("/queue/batch/delete", post(api::batch_delete))
        .route("/artists/search", get(api::search_artists))
        .route("/artists", get(api::list_artists))
        .route("/artists/:id", put(api::update_artist))
        .route("/artists/:id/albums", get(api::list_albums))
        .route("/albums/:id", put(api::update_album))
        .route("/merges", get(api::list_merges).post(api::create_merge))
        .route("/merges/:id", get(api::get_merge).put(api::update_merge))
        .route("/merges/:id/approve", post(api::approve_merge))
        .route("/merges/:id/reject", post(api::reject_merge))
        .route("/merges/:id/retry", post(api::retry_merge))
        .route("/library/tracks", get(api::library_tracks))
        .route("/library/albums", get(api::library_albums))
        .route("/library/artists", get(api::library_artists));

    Router::new()
        .route("/", get(admin_html))
        .nest("/api", api)
        .with_state(state)
}

async fn admin_html() -> axum::response::Html<&'static str> {
    axum::response::Html(include_str!("admin.html"))
}