flashapi 0.2.3

A lightweight 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
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 crate::{
    Request,
    server::{
        handler::{AsyncHandler, Handler},
        http::{HttpMethod, HttpStatus},
        request::get_request_info,
        response::Response,
        url::match_route,
    },
};
use std::{collections::HashMap, sync::Arc};
use tokio::net::{TcpListener, TcpStream};

type Routes<S> = HashMap<(HttpMethod, String), Arc<dyn Handler<S>>>;

pub struct HttpServer<S = ()> {
    routes: Routes<S>,
    state: Option<Arc<S>>,
}

impl<S> HttpServer<S>
where
    S: Send + Sync + 'static,
{
    pub fn new() -> Self {
        Self {
            routes: HashMap::new(),
            state: None,
        }
    }

    pub fn with_state(mut self, state: S) -> Self {
        self.state = Some(Arc::new(state));
        self
    }

    // ------------------------------------------------------------
    // Routes start
    // ------------------------------------------------------------
    pub fn delete<F, Fut>(&mut self, path: String, handler: F)
    where
        F: Fn(Request, Response, Arc<S>) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = ()> + Send + Sync + 'static,
    {
        let h = AsyncHandler::new(handler);
        self.routes.insert((HttpMethod::Delete, path), Arc::new(h));
    }

    pub fn get<F, Fut>(&mut self, path: String, handler: F)
    where
        F: Fn(Request, Response, Arc<S>) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = ()> + Send + 'static,
    {
        let h = AsyncHandler::new(handler);

        self.routes.insert((HttpMethod::Get, path), Arc::new(h));
    }

    pub fn patch<F, Fut>(&mut self, path: String, handler: F)
    where
        F: Fn(Request, Response, Arc<S>) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = ()> + Send + Sync + 'static,
    {
        let h = AsyncHandler::new(handler);

        self.routes.insert((HttpMethod::Patch, path), Arc::new(h));
    }

    pub fn post<F, Fut>(&mut self, path: String, handler: F)
    where
        F: Fn(Request, Response, Arc<S>) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = ()> + Send + Sync + 'static,
    {
        let h = AsyncHandler::new(handler);

        self.routes.insert((HttpMethod::Post, path), Arc::new(h));
    }
    // ------------------------------------------------------------
    // Routes end
    // ------------------------------------------------------------

    async fn handle_connection(routes: Routes<S>, mut stream: TcpStream, state: Option<Arc<S>>) {
        let mut request_info = get_request_info(&mut stream).await;

        let mut handler: Option<&Arc<dyn Handler<S>>> = None;

        for ((method, path), h) in &routes {
            if method == &request_info.method {
                // ------------------------------------------------------------
                // Handling dynamic route check and adding params if found any
                // ------------------------------------------------------------
                if let Some(params) = match_route(&path, &request_info.path.to_string()) {
                    request_info.params = Some(params);
                    handler = Some(h);
                }
            }
        }

        let mut response: Response = Response::new(stream);

        if let Some(handler) = handler {
            handler.call(request_info, response, state).await;
        } else {
            response
                .send(HttpStatus::NotFound, "Route not found.", "text/plain")
                .await;
        }
    }

    pub async fn listen(&mut self, port: u16) -> std::io::Result<()> {
        let listener = TcpListener::bind("0.0.0.0:".to_string() + &port.to_string()).await?;

        let routes = self.routes.clone();
        let state = self.state.clone();

        loop {
            let (stream, _) = listener.accept().await?;

            let routes = routes.clone();
            let state = state.clone();

            tokio::spawn(async {
                HttpServer::handle_connection(routes, stream, state).await;
            });
        }
    }
}