use axum::{Router, middleware, routing::get};

use crate::{
    auth::require_bearer_auth,
    external_generation::{
        get_external_generation_job_status, get_external_generation_queue_overview,
    },
    state::AppState,
};

pub fn router(state: AppState) -> Router<AppState> {
    Router::new()
        .route(
            "/api/runtime/external-generation/queue-overview",
            get(get_external_generation_queue_overview).route_layer(
                middleware::from_fn_with_state(state.clone(), require_bearer_auth),
            ),
        )
        .route(
            "/api/runtime/external-generation/jobs/{job_id}",
            get(get_external_generation_job_status).route_layer(middleware::from_fn_with_state(
                state.clone(),
                require_bearer_auth,
            )),
        )
}