musdk 0.1.0

mu Protocol app development SDK
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

use std::{
    borrow::Cow,
    collections::HashMap,
    ops::{Deref, DerefMut},
};

use musdk_common::{Request, Status};

use crate::{content_type, IntoResponse};

pub trait FromRequest<'a>: Sized {
    type Error: IntoResponse<'static>;

    fn from_request(req: &'a Request) -> Result<Self, Self::Error>;
}

impl<'a> FromRequest<'a> for &'a Request<'a> {
    type Error = ();

    fn from_request(req: &'a Request) -> Result<Self, Self::Error> {
        Ok(req)
    }
}

impl<'a> FromRequest<'a> for &'a [u8] {
    type Error = ();

    fn from_request(req: &'a Request) -> Result<Self, ()> {
        Ok(&req.body)
    }
}

impl<'a> FromRequest<'a> for Vec<u8> {
    type Error = ();

    fn from_request(req: &'a Request) -> Result<Self, ()> {
        Ok(req.body.to_vec())
    }
}

impl<'a> FromRequest<'a> for &'a str {
    //TODO: concrete error type
    type Error = (String, Status);

    fn from_request(req: &'a Request) -> Result<Self, Self::Error> {
        let content_type = req.content_type();
        let charset = content_type
            .as_ref()
            .and_then(|s| content_type::parse(s).1)
            .unwrap_or("us-ascii");

        match charset.to_lowercase().as_str() {
            "utf-8" | "us-ascii" => {
                core::str::from_utf8(&req.body).map_err(|e| (e.to_string(), Status::BadRequest))
            }
            ch => Err((format!("unsupported charset: {ch}"), Status::BadRequest)),
        }
    }
}

impl<'a> FromRequest<'a> for String {
    type Error = <&'a str as FromRequest<'a>>::Error;

    fn from_request(req: &'a Request) -> Result<Self, Self::Error> {
        <&'a str as FromRequest<'a>>::from_request(req).map(ToString::to_string)
    }
}

//TODO: Deserialize into the concrete struct, like `PathParam<Request>`
pub struct PathParams<'a>(HashMap<Cow<'a, str>, Cow<'a, str>>);
pub struct QueryParams<'a>(HashMap<Cow<'a, str>, Cow<'a, str>>);

impl<'a> FromRequest<'a> for PathParams<'a> {
    type Error = ();

    fn from_request(req: &'a Request) -> Result<Self, Self::Error> {
        let map = req
            .path_params
            .iter()
            .map(|(k, v)| (k.clone(), v.clone()))
            .collect();

        Ok(Self(map))
    }
}

impl<'a> FromRequest<'a> for QueryParams<'a> {
    type Error = ();

    fn from_request(req: &'a Request) -> Result<Self, Self::Error> {
        let map = req
            .query_params
            .iter()
            .map(|(k, v)| (k.clone(), v.clone()))
            .collect();

        Ok(Self(map))
    }
}

impl<'a> Deref for PathParams<'a> {
    type Target = HashMap<Cow<'a, str>, Cow<'a, str>>;

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

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

impl<'a> Deref for QueryParams<'a> {
    type Target = HashMap<Cow<'a, str>, Cow<'a, str>>;

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

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