about summary refs log tree commit diff stats
path: root/crates/rocie-server/src/api/get/unit.rs
blob: caafaa34596c0d51dfb7a4e7b46f895dca246f3d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
use actix_web::{HttpResponse, Responder, Result, get, web};

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

/// 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))
}

/// Return all registered units for a specific unit property
#[utoipa::path(
    responses(
        (
            status = OK,
            description = "All units founds",
            body = Vec<Unit>
        ),
        (
            status = INTERNAL_SERVER_ERROR,
            description = "Server encountered error",
            body = String
        )
    ),
    params(
        (
            "id" = UnitPropertyId,
            description = "Unit property id"
        ),
    )
)]
#[get("/units-by-property/{id}")]
pub(crate) async fn units_by_property_id(
    app: web::Data<App>,
    id: web::Path<UnitPropertyIdStub>,
) -> Result<impl Responder> {
    let id = id.into_inner();
    let all = Unit::get_all(&app)
        .await?
        .into_iter()
        .filter(|unit| unit.unit_property == id.into())
        .collect::<Vec<_>>();

    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()),
    }
}