use axum::{routing::get, Router, extract::State};

use crate::state::AppState;

pub fn create_router(app_state: AppState) -> Router {
    Router::new()
        .route("/health", get(health_check))
        .route("/tasks", get(get_tasks))
        .with_state(app_state)
}

async fn health_check(State(_state): State<AppState>) -> &'static str {
    "OK"
}

async fn get_tasks(State(state): State<AppState>) -> axum::Json<Vec<common_contracts::observability::TaskProgress>> {
    let tasks = state
        .tasks
        .iter()
        .map(|kv| kv.value().clone())
        .collect::<Vec<_>>();
    axum::Json(tasks)
}