use std::collections::HashMap;

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

use crate::{AppState, Error, Status};

pub fn router() -> axum::Router<AppState> {
    axum::Router::new()
        .route("/", get(services))
        .route("/:name", get(service))
}

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

impl ServiceQuery {
    pub fn filter(&self, value: &(String, Status)) -> bool {
        !self.name.as_ref().is_some_and(|n| *n != value.0)
            && !self.status.as_ref().is_some_and(|s| *s != value.1)
    }
}

pub async fn services(
    Query(query): Query<ServiceQuery>,
    State(state): State<AppState>,
) -> Json<HashMap<String, Status>> {
    Json(
        state
            .statuses()
            .into_iter()
            .filter(|item| query.filter(item))
            .collect(),
    )
}

pub async fn service(
    Path(name): Path<String>,
    State(state): State<AppState>,
) -> Result<Status, Error> {
    state
        .status(&name)
        .ok_or_else(|| Error::ServiceNotFound(name))
}