runique 1.1.25

A Django-inspired web framework for Rust with ORM, templates, and comprehensive security middleware
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

use crate::config::RuniqueConfig;
use crate::utils::aliases::{StrMap, StrVecMap};
use crate::utils::parse_html::parse_multipart;
use axum::{
    body::Body,
    extract::{FromRequest, Multipart},
    http::{Request, StatusCode},
    response::{IntoResponse, Response},
};
use http_body_util::BodyExt;
use std::collections::HashMap;
use std::sync::Arc;

/// Aegis : extraction unique du body (multipart/urlencoded/json) et normalisation.
pub async fn aegis<S>(
    req: Request<Body>,
    state: &S,
    config: Arc<RuniqueConfig>,
    content_type: &str,
) -> Result<StrVecMap, Response>
where
    S: Send + Sync,
{
    let mut parsed: StrVecMap = HashMap::new();

    if content_type.starts_with("multipart/form-data") {
        let multipart = Multipart::from_request(req, state)
            .await
            .map_err(|_e| (StatusCode::BAD_REQUEST, "Multipart error").into_response())?;

        let upload_dir = std::path::Path::new(&config.static_files.media_root);
        parsed = parse_multipart(multipart, upload_dir).await?;
    } else {
        let bytes = req
            .into_body()
            .collect()
            .await
            .map_err(|_e| (StatusCode::BAD_REQUEST, "Body error").into_response())?
            .to_bytes();

        if content_type.starts_with("application/x-www-form-urlencoded") {
            parsed = serde_urlencoded::from_bytes::<StrMap>(&bytes)
                .unwrap_or_default()
                .into_iter()
                .map(|(k, v)| (k, vec![v]))
                .collect();
        } else if content_type.starts_with("application/json") {
            parsed = serde_json::from_slice::<StrMap>(&bytes)
                .unwrap_or_default()
                .into_iter()
                .map(|(k, v)| (k, vec![v]))
                .collect();
        }
    }

    Ok(parsed)
}