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

/// To let a struct which implements the `SimpleAuthorization` trait become an authorizer.
#[macro_export]
macro_rules! authorizer {
    ($name:ty) => {
        #[$crate::rocket::async_trait]
        impl<'r> $crate::rocket::request::FromRequest<'r> for $name {
            type Error = ();

            async fn from_request(
                request: &'r $crate::rocket::request::Request<'_>,
            ) -> $crate::rocket::request::Outcome<Self, Self::Error> {
                let key: Option<&str> = request.headers().get("authorization").next();

                match <$name as $crate::SimpleAuthorization>::authorizing(request, key).await {
                    Some(ins) => $crate::rocket::outcome::Outcome::Success(ins),
                    None => $crate::rocket::outcome::Outcome::Forward(()),
                }
            }
        }
    };
    (ref $name:ty) => {
        $crate::authorizer!($name);

        #[$crate::rocket::async_trait]
        impl<'r> $crate::rocket::request::FromRequest<'r> for &'r $name {
            type Error = ();

            async fn from_request(
                request: &'r $crate::rocket::request::Request<'_>,
            ) -> $crate::rocket::request::Outcome<Self, Self::Error> {
                async fn f(request: &$crate::rocket::request::Request<'_>) -> Option<$name> {
                    let key: Option<&str> = request.headers().get("authorization").next();

                    match <$name as $crate::SimpleAuthorization>::authorizing(request, key).await {
                        Some(ins) => Some(ins),
                        None => None,
                    }
                }

                let cache = request.local_cache_async(f(request)).await;

                match cache.as_ref() {
                    Some(cache) => $crate::rocket::outcome::Outcome::Success(cache),
                    None => $crate::rocket::outcome::Outcome::Forward(()),
                }
            }
        }
    };
}