lariv-rs 0.1.0

Compile-time plugin web application framework built on Axum, SeaORM, Maud, and HTMX
Documentation 
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
127
128
129
130
131
132
133
134
135
136

//! HTTP handlers for import page and XLSX upload.
use axum::{
    extract::Multipart,
    http::StatusCode,
    response::{IntoResponse, Response},
};
use maud::Markup;

use crate::{
    components::{SharedChromeFolder, SlotCtx},
    export::ExportCapability,
    html_form::HtmlForm,
    http::Cap,
    plugins::{
        import::{forms::ImportForm, state::ImportState, templates::ImportPage, upsert, xlsx},
        users::middleware::RequireStaff,
        users::state::AuthContext,
    },
    web::{Htmx, html_built_page_or_app_layout},
};

const MAX_UPLOAD_BYTES: usize = crate::http::REQUEST_BODY_LIMIT_BYTES;

fn import_page(
    model_count: i64,
    error: String,
    result: Option<upsert::ImportReport>,
) -> ImportPage {
    ImportPage {
        error,
        result,
        model_count,
    }
}

fn render_page(
    page: &ImportPage,
    htmx: &Htmx,
    chrome: &SharedChromeFolder,
    ctx: &AuthContext,
) -> Markup {
    html_built_page_or_app_layout(page, htmx, chrome, &SlotCtx::from_auth(ctx))
}

/// HTTP handler: `page`.
pub async fn page(
    Cap(export): Cap<ExportCapability>,
    Cap(chrome): Cap<SharedChromeFolder>,
    RequireStaff(ctx): RequireStaff,
    htmx: Htmx,
) -> Response {
    let page = import_page(export.catalog().entries.len() as i64, String::new(), None);
    render_page(&page, &htmx, &chrome, &ctx).into_response()
}

/// HTTP handler: `import_post`.
pub async fn import_post(
    Cap(export): Cap<ExportCapability>,
    Cap(state): Cap<ImportState>,
    Cap(chrome): Cap<SharedChromeFolder>,
    RequireStaff(ctx): RequireStaff,
    htmx: Htmx,
    multipart: Multipart,
) -> Response {
    let catalog = export.catalog();
    let model_count = catalog.entries.len() as i64;

    let parsed_form = match ImportForm::from_multipart(multipart).await {
        Ok(form) => form,
        Err(err) => {
            let page = import_page(model_count, err.to_string(), None);
            return (
                StatusCode::BAD_REQUEST,
                render_page(&page, &htmx, &chrome, &ctx),
            )
                .into_response();
        }
    };

    let bytes = match parsed_form.file.into_bytes().await {
        Ok(bytes) => bytes,
        Err(err) => {
            let page = import_page(model_count, err.to_string(), None);
            return (
                StatusCode::BAD_REQUEST,
                render_page(&page, &htmx, &chrome, &ctx),
            )
                .into_response();
        }
    };
    if bytes.len() > MAX_UPLOAD_BYTES {
        let page = import_page(model_count, "xlsx file too large (max 50 MiB)".into(), None);
        return (
            StatusCode::BAD_REQUEST,
            render_page(&page, &htmx, &chrome, &ctx),
        )
            .into_response();
    }
    if bytes.is_empty() {
        let page = import_page(model_count, "empty file".into(), None);
        return (
            StatusCode::BAD_REQUEST,
            render_page(&page, &htmx, &chrome, &ctx),
        )
            .into_response();
    }

    let workbook = match xlsx::parse_workbook(&bytes, &catalog) {
        Ok(workbook) => workbook,
        Err(err) => {
            tracing::warn!(error = %err, "import parse failed");
            let page = import_page(model_count, err, None);
            return (
                StatusCode::BAD_REQUEST,
                render_page(&page, &htmx, &chrome, &ctx),
            )
                .into_response();
        }
    };

    match upsert::import_workbook(&state.db, &catalog, &workbook).await {
        Ok(report) => {
            let page = import_page(model_count, String::new(), Some(report));
            render_page(&page, &htmx, &chrome, &ctx).into_response()
        }
        Err(err) => {
            tracing::error!(error = %err, "import workbook failed");
            let page = import_page(model_count, err, None);
            (
                StatusCode::INTERNAL_SERVER_ERROR,
                render_page(&page, &htmx, &chrome, &ctx),
            )
                .into_response()
        }
    }
}