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

use crate::{Error, Additional, Extractor, http::Request};
use serde::de::DeserializeOwned;
use std::sync::Arc;
use std::ops::{Deref, DerefMut};

/// Json extractor
///
/// Allows to use a structure that implements `DeserializeOwned` to extract information as json from the body of a request
///
/// ```rust, no_run
/// use cataclysm::http::{Response, Json};
/// use serde::{Deserialize};
///
/// #[derive(Deserialize, Debug)]
/// struct BodyParams {
///     name: String,
///     last_name: Option<String>
/// }
/// 
/// async fn check_body(json: Json<BodyParams>) -> Response {
///     log::info!("Http call containing {:?}", json.into_inner());
///     Response::ok()
/// }
/// ```
///
/// Deserialization error will result always in a bad request response
pub struct Json<J>(pub J);

impl<J> Json<J> {
    /// Retrieves the inner instance of the generic type
    pub fn into_inner(self) -> J {
        self.0
    }
}

impl<T: Sync, J: 'static + DeserializeOwned + Send + Sync> Extractor<T> for Json<J> {
    fn extract(req: &Request, _additional: Arc<Additional<T>>) -> Result<Self, Error> {
        let content_type_header = req.headers.get("content-type").or_else(|| req.headers.get("Content-Type"));
        if let Some(content_type_header) = content_type_header {
            if content_type_header == "application/json" {
                match String::from_utf8(req.content.clone()) {
                    Ok(body) => {
                        serde_json::from_str::<J>(&body)
                            .map(|j| Json(j))
                            .map_err(|e| Error::ExtractionBR(format!("json deserialization failure, {}", e)))
                    },
                    Err(e) => {
                        Err(Error::ExtractionBR(format!("body encoding error, {}", e)))
                    }
                }
            } else {
                Err(Error::ExtractionBR(format!("content-type header should be 'application/json' (found {}) for correct parsing", content_type_header)))
            }
        } else {
            Err(Error::ExtractionBR(format!("missing header content-type (or Content-Type) required for json parsing")))
        }
    }
}

impl<J> Deref for Json<J> {
    type Target = J;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl<J> DerefMut for Json<J> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}