use std::collections::HashMap;

use axum::{
    extract::{Path, Query, State},
    Json, Router,
};
use axum_extra::routing::Resource;
use serde::{Deserialize, Serialize};

use crate::{service::ServiceHandles, Error, Status};

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct ServiceQuery {
    pub name: Option<String>,
    pub state: Option<Status>,
}

pub fn router() -> Router<ServiceHandles> {
    Resource::named("services").index(index).show(show).into()
}

pub async fn index(
    Query(query): Query<ServiceQuery>,
    State(services): State<ServiceHandles>,
) -> Json<HashMap<String, Status>> {
    let map = match query.name {
        Some(n) => services
            .iter()
            .filter(|(name, _)| n == **name)
            .map(|(name, srv)| (name.clone(), srv.status()))
            .collect(),
        None => services
            .iter()
            .map(|(name, srv)| (name.clone(), srv.status()))
            .collect(),
    };

    Json(map)
}

pub async fn show(
    Path(name): Path<String>,
    State(services): State<ServiceHandles>,
) -> Result<Status, Error> {
    services
        .get(&name)
        .map(|s| s.status())
        .ok_or_else(|| Error::ServiceNotFound(name))
}