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

use url::Url;

#[derive(Debug, Clone, PartialEq)]
pub(crate) struct HttpExtUrl(pub Url);

#[derive(Debug, Clone, PartialEq)]
pub(crate) struct HttpExtSerdeJsonValue(pub serde_json::Value);

#[derive(Debug, Clone, PartialEq)]
pub(crate) struct HttpExtVecU8(pub Vec<u8>);

/// Extension trait for http::response::Builder objects
///
/// Allows the user to add a `Url` to the http::Response
pub trait HttpResponseBuilderExt {
    /// A builder method for the `http::response::Builder` type that allows the user to add a `Url`
    /// to the `http::Response`
    fn url(self, url: Url) -> Self;
    fn json(self, value: &serde_json::Value) -> Self;
}

impl HttpResponseBuilderExt for http::response::Builder {
    fn url(self, url: Url) -> Self {
        self.extension(HttpExtUrl(url))
    }

    fn json(self, value: &serde_json::Value) -> Self {
        if let Ok(value) = serde_json::to_vec(value) {
            return self.extension(HttpExtVecU8(value));
        }

        self
    }
}

pub trait HttpResponseExt {
    fn url(&self) -> Option<Url>;
    fn json(&self) -> Option<serde_json::Value>;
}

impl<T> HttpResponseExt for http::Response<T> {
    fn url(&self) -> Option<Url> {
        self.extensions()
            .get::<HttpExtUrl>()
            .map(|url| url.clone().0)
    }

    fn json(&self) -> Option<serde_json::Value> {
        self.extensions()
            .get::<HttpExtVecU8>()
            .and_then(|value| serde_json::from_slice(value.0.as_slice()).ok())
    }
}