use actix_web::{get, web, HttpResponse, Responder, Result};

use crate::{app::App, storage::sql::unit::{Unit, UnitId, UnitIdStub}};

/// Return all registered units
#[utoipa::path(
    responses(
        (
            status = OK,
            description = "All units founds",
            body = Vec<Unit>
        ),
        (
            status = INTERNAL_SERVER_ERROR,
            description = "Server encountered error",
            body = String
        )
    ),
)]
#[get("/units/")]
pub(crate) async fn units(app: web::Data<App>) -> Result<impl Responder> {
    let all = Unit::get_all(&app).await?;

    Ok(HttpResponse::Ok().json(all))
}

/// Get Unit by id
#[utoipa::path(
    responses(
        (
            status = OK,
            description = "Unit found from database",
            body = Unit
        ),
        (
            status = NOT_FOUND,
            description = "Unit not found in database"
        ),
        (
            status = INTERNAL_SERVER_ERROR,
            description = "Server encountered error",
            body = String
        )
    ),
    params(
        (
            "id" = UnitId,
            description = "Unit id"
        ),
    )
)]
#[get("/unit/{id}")]
pub(crate) async fn unit_by_id(
    app: web::Data<App>,
    id: web::Path<UnitIdStub>,
) -> Result<impl Responder> {
    let id = id.into_inner();

    match Unit::from_id(&app, id.into()).await? {
        Some(unit) => Ok(HttpResponse::Ok().json(unit)),
        None => Ok(HttpResponse::NotFound().finish()),
    }
}