httpz 0.0.6

Code once, support every Rust webserver!
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

use actix_web::{
    guard::{Guard, GuardContext},
    web::{self, Bytes},
    HttpRequest, HttpResponse as ActixHttpResponse, Resource,
};
use http::{header::HeaderName, Request, StatusCode};

use crate::{Endpoint, HttpEndpoint, HttpResponse, Server};

/// TODO
pub struct ActixMounter<TEndpoint: HttpEndpoint>(TEndpoint)
where
    TEndpoint: HttpEndpoint + Clone;

impl<TEndpoint> Clone for ActixMounter<TEndpoint>
where
    TEndpoint: HttpEndpoint + Clone,
{
    fn clone(&self) -> Self {
        Self(self.0.clone())
    }
}

impl<TEndpoint> ActixMounter<TEndpoint>
where
    TEndpoint: HttpEndpoint + Clone,
{
    /// TODO
    pub fn mount(&self) -> Resource {
        let mut endpoint = self.0.clone();
        let (url, routes) = endpoint.register();

        web::resource(url.as_ref())
            .guard(MethodGuard::<TEndpoint>(routes))
            .to(move |request: HttpRequest, body: Bytes| {
                let endpoint = endpoint.clone();
                async move {
                    let mut parts = Request::<()>::default().into_parts().0;
                    // TODO: Reducing the cloning here
                    parts.method = request.method().clone();
                    parts.uri = request.uri().clone();
                    parts.version = request.version();
                    for (k, v) in request.headers() {
                        parts.headers.insert(HeaderName::from(k), v.clone());
                    }
                    // req.extensions = request.extensions().get_mut() // TODO: Pass extensions through

                    match endpoint
                        .handler(crate::Request(parts, body.to_vec(), Server::ActixWeb))
                        .await
                        .into_response()
                    {
                        Ok(resp) => {
                            let (parts, body) = resp.into_parts();
                            let mut resp = ActixHttpResponse::new(parts.status);
                            for (k, v) in parts.headers {
                                if let Some(k) = k {
                                    resp.headers_mut().insert(k, v);
                                }
                            }
                            resp.set_body(body)
                        }
                        Err(err) => ActixHttpResponse::new(StatusCode::INTERNAL_SERVER_ERROR)
                            .set_body(err.to_string().as_bytes().to_vec()),
                    }
                }
            })
    }
}

impl<TEndpoint> Endpoint<TEndpoint>
where
    TEndpoint: HttpEndpoint + Clone,
{
    /// is called to create a builder for mounting this endpoint to your actix-web router.
    pub fn actix(self) -> ActixMounter<TEndpoint> {
        ActixMounter(self.endpoint)
    }
}

struct MethodGuard<TEndpoint: HttpEndpoint>(TEndpoint::Routes);

impl<TEndpoint: HttpEndpoint> Guard for MethodGuard<TEndpoint> {
    fn check(&self, ctx: &GuardContext<'_>) -> bool {
        for method in self.0.as_ref().iter() {
            if method == ctx.head().method {
                return true;
            }
        }

        false
    }
}