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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

#![feature(async_closure)]
#![feature(or_patterns)]
#![feature(exclusive_range_pattern)]

mod body;
mod request;
mod response;
mod router;
mod server;

use async_std::{fs, io::ReadExt, task};
use router::{Middleware, Path, Route, Router};
use std::{future::Future, pin::Pin, str, sync::Arc};

pub use body::ChunkedBody;
pub use request::{Encoding, Method, Request};
pub use response::{Response, Status};
pub use router::Ctx;

pub(crate) const BUF_LEN: usize = 2;
pub(crate) const KEEP_ALIVE_TIMEOUT: u64 = 10;

pub(crate) type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
pub(crate) type Sender<T> = futures::channel::mpsc::Sender<T>;
pub(crate) type Receiver<T> = futures::channel::mpsc::Receiver<T>;
pub(crate) type BoxFuture<'a, Response> = Pin<Box<dyn Future<Output = Response> + Send + 'static>>;

pub struct App<Routes> {
    router: Router<Routes>,
}

impl App<()> {
    pub fn new() -> App<()> {
        Self::with_routes(())
    }
    pub fn run(self, port: u32) -> Result<()> {
        let router = Arc::new(self.router);
        task::block_on(server::accept_loop("127.0.0.1", port, router))
    }
    pub fn get(&mut self, path: &'static str, handler: impl Route) -> &Self {
        self.add_route(Path::new(Method::Get, path.to_owned()), handler);
        self
    }
    pub fn post(&mut self, path: &'static str, handler: impl Route) -> &Self {
        self.add_route(Path::new(Method::Post, path.to_owned()), handler);
        self
    }
    pub fn put(&mut self, path: &'static str, handler: impl Route) -> &Self {
        self.add_route(Path::new(Method::Put, path.to_owned()), handler);
        self
    }
    pub fn delete(&mut self, path: &'static str, handler: impl Route) -> &Self {
        self.add_route(Path::new(Method::Delete, path.to_owned()), handler);
        self
    }
    pub fn middleware(&mut self, path: &'static str, middleware: impl Middleware) -> &Self {
        self.add_middleware(Path::all(path.to_owned()), middleware);
        self
    }
    pub fn get_middleware(&mut self, path: &'static str, middleware: impl Middleware) -> &Self {
        self.add_middleware(Path::new(Method::Get, path.to_owned()), middleware);
        self
    }
    pub fn post_middleware(&mut self, path: &'static str, middleware: impl Middleware) -> &Self {
        self.add_middleware(Path::new(Method::Post, path.to_owned()), middleware);
        self
    }
    pub fn put_middleware(&mut self, path: &'static str, middleware: impl Middleware) -> &Self {
        self.add_middleware(Path::new(Method::Put, path.to_owned()), middleware);
        self
    }
    pub fn delete_middleware(&mut self, path: &'static str, middleware: impl Middleware) -> &Self {
        self.add_middleware(Path::new(Method::Delete, path.to_owned()), middleware);
        self
    }
    pub fn files(&mut self, path: &'static str, src: &'static str) -> &Self {
        self.add_middleware(
            Path::new(Method::Get, path.to_owned()),
            Files::new(path, src),
        );
        self
    }
    fn add_route(&mut self, route: Path, handler: impl Route) {
        self.router.routes.insert(route, Box::new(handler));
    }
    fn add_middleware(&mut self, route: Path, middleware: impl Middleware) {
        self.router.middleware.push((route, Arc::new(middleware)));
    }
}

impl<Routes: Send + Sync + Copy + 'static> App<Routes> {
    pub fn with_routes(routes: Routes) -> App<Routes> {
        App {
            router: Router::new(routes),
        }
    }
}

async fn not_found(_req: Request) -> Response {
    Response::with_status(Status::NotFound)
}

pub struct Files {
    pub path: &'static str,
    pub src: &'static str,
}

impl Files {
    pub fn new(path: &'static str, src: &'static str) -> Files {
        Files { path, src }
    }
}

impl Middleware for Files {
    fn handle(&self, ctx: Ctx) -> BoxFuture<Response> {
        let filepath = ctx.req.path.replace(&self.path, "");
        let src = self.src.clone();
        Box::pin(async move {
            let path = format!("{}{}", src, filepath);
            match fs::canonicalize(path).await {
                Ok(path) => match async_std::fs::File::open(path).await {
                    Ok(mut file) => {
                        let mut res = Response::with_status(Status::Ok);
                        let mut buf = vec![];
                        let res = match file.read_to_end(&mut buf).await {
                            Ok(_) => {
                                let mime = mime_guess::from_path(filepath)
                                    .first()
                                    .unwrap_or(mime::APPLICATION_OCTET_STREAM);

                                res.set_bytes(buf, mime);
                                res
                            }
                            Err(_) => Response::with_status(Status::NotFound),
                        };
                        res
                    }
                    Err(_) => Response::with_status(Status::NotFound),
                },
                Err(_) => Response::with_status(Status::NotFound),
            }
        })
    }
}