axol 0.2.0

Axol Web Framework
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

use axol_http::request::Request;
use axol_http::response::Response;
use futures::Future;

use crate::Result;
use crate::{FromRequestParts, IntoResponse};

mod realip;
pub use realip::RealIp;

#[async_trait::async_trait]
pub trait RequestHook: Send + Sync + 'static {
    /// Called on an inbound request
    async fn handle_request(&self, request: &mut Request) -> Result<Option<Response>>;
}

#[async_trait::async_trait]
impl<F, Fut, Res> RequestHookExpansion<()> for F
where
    F: Fn() -> Fut + Send + Sync + 'static,
    Fut: Future<Output = Result<Option<Res>>> + Send,
    Res: IntoResponse,
{
    async fn handle_request(&self, _request: &mut Request) -> Result<Option<Response>> {
        Ok(self().await?.map(IntoResponse::into_response).transpose()?)
    }
}

#[async_trait::async_trait]
pub trait RequestHookExpansion<G>: Send + Sync + 'static {
    async fn handle_request(&self, request: &mut Request) -> Result<Option<Response>>;
}

#[async_trait::async_trait]
impl<G: 'static> RequestHook for Box<dyn RequestHookExpansion<G>> {
    async fn handle_request(&self, request: &mut Request) -> Result<Option<Response>> {
        (**self).handle_request(request).await
    }
}

macro_rules! impl_handler {
    ( $($ty:ident),* $(,)? ) => {
        #[allow(non_snake_case)]
        #[allow(unused_variables)]
        #[async_trait::async_trait]
        impl<F, Fut, Res, $($ty,)*> RequestHookExpansion<(($($ty,)*), Fut, Res)> for F
        where F: Fn($($ty,)*) -> Fut + Send + Sync + 'static,
            Fut: Future<Output = Result<Option<Res>>> + Send + 'static,
            Res: IntoResponse,
            $( for<'a> $ty: FromRequestParts<'a> + Send, )*
        {
            async fn handle_request(&self, request: &mut Request) -> Result<Option<Response>> {
                let parts = request.parts();
                $(
                    let $ty = $ty::from_request_parts(parts).await?;
                )*

                let res = self($($ty,)*).await?;

                res.map(IntoResponse::into_response).transpose()
            }
        }
    };
}

all_the_tuples_no_last_special_case!(impl_handler);