airlab-web 0.1.2

airlab backend
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use crate::web::mw_auth::CtxW;
use crate::web::Result;
use crate::web_config;
use airlab_lib::model::validation::{Validation, ValidationBmc};
use airlab_lib::model::validation_file::{
    ValidationFile, ValidationFileBmc, ValidationFileForCreate, ValidationFileForUpdate,
};
use airlab_lib::model::ModelManager;
use axum::body::Body;
use axum::extract::{Json as eJson, Path, State};
use axum::routing::{get, patch, post};
use axum::{Json, Router};
use serde_json::{json, Value};
use tokio::{fs::File, io::BufReader};
use tokio_util::io::ReaderStream;
use tracing::debug;

pub fn routes(mm: ModelManager) -> Router {
    Router::new()
        .route(
            "/api/v1/validation_files",
            post(api_post_validation_file_handler),
        )
        .route(
            "/api/v1/validation_files/:validation_file_id",
            patch(api_patch_validation_file_handler),
        )
        .route(
            "/api/v1/validation_files/:validation_file_id",
            get(api_validation_file_handler),
        )
        .route(
            "/api/v1/validation_files",
            get(api_validation_files_handler),
        )
        .route(
            "/api/v1/validationFiles/:file_id/serve",
            get(api_serve_validation_handler),
        )
        .with_state(mm)
}

async fn api_post_validation_file_handler(
    State(mm): State<ModelManager>,
    ctx: CtxW,
    eJson(payload): eJson<ValidationFileForCreate>,
) -> Result<Json<Value>> {
    debug!("HANDLER - api_post_validation_file_handler: {:?}", payload);
    let ctx = ctx.0;
    let v_file_id = ValidationFileBmc::create(&ctx, &mm, payload).await?;

    let validation_file: ValidationFile = ValidationFileBmc::get(&ctx, &mm, v_file_id).await?;
    Ok(Json(json!(validation_file)))
}

async fn api_patch_validation_file_handler(
    State(mm): State<ModelManager>,
    ctx: CtxW,
    Path(v_file_id): Path<i32>,
    eJson(payload): eJson<ValidationFileForUpdate>,
) -> Result<Json<Value>> {
    debug!(
        "HANDLER - api_validation_file: {}; {:?}",
        v_file_id, payload
    );
    let ctx = ctx.0;

    ValidationFileBmc::update(&ctx, &mm, v_file_id, payload).await?;

    let validation_file: ValidationFile = ValidationFileBmc::get(&ctx, &mm, v_file_id).await?;
    Ok(Json(json!(validation_file)))
}

async fn api_validation_file_handler(
    State(mm): State<ModelManager>,
    ctx: CtxW,
    Path(v_file_id): Path<i32>,
) -> Result<Json<Value>> {
    debug!("HANDLER - api_validation_file: {}", v_file_id);
    let ctx = ctx.0;

    let validation_file: ValidationFile = ValidationFileBmc::get(&ctx, &mm, v_file_id).await?;
    Ok(Json(json!(validation_file)))
}

async fn api_serve_validation_handler(
    State(mm): State<ModelManager>,
    ctx: CtxW,
    Path(file_id): Path<i32>,
) -> Result<Body> {
    debug!("HANDLER - api_serve_validation_handler: {}", file_id);

    let ctx = ctx.0;
    let validation_file: ValidationFile = ValidationFileBmc::get(&ctx, &mm, file_id).await?;
    let validation: Validation =
        ValidationBmc::get(&ctx, &mm, validation_file.validation_id).await?;

    let file_path = format!(
        "{}/groups/{}/uploads/validation/{}/{}.{}",
        &web_config().DATA_PATH,
        validation.group_id,
        validation.id,
        validation_file.hash,
        validation_file.extension
    );
    let file = File::open(file_path).await?;

    let reader = BufReader::new(file);

    let stream = ReaderStream::new(reader);

    let body = Body::from_stream(stream);
    Ok(body)
}

async fn api_validation_files_handler(
    State(mm): State<ModelManager>,
    ctx: CtxW,
) -> Result<Json<Value>> {
    debug!("HANDLER - api_validation_files_handler");
    let ctx = ctx.0;

    let validation_files: Vec<ValidationFile> =
        ValidationFileBmc::list(&ctx, &mm, None, None).await?;
    Ok(Json(json!(validation_files)))
}