src/routes/static.rs


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

use hitide_icons::ICONS_MAP;
use std::sync::Arc;

const FILE_MAIN_CSS: &[u8] = include_bytes!("../../res/main.css");

pub fn route_static() -> crate::RouteNode<()> {
    crate::RouteNode::new()
        .with_child_str(crate::RouteNode::new().with_handler_async("GET", handler_static_get))
}

async fn handler_static_get(
    params: (String,),
    _ctx: Arc<crate::RouteContext>,
    _req: hyper::Request<hyper::Body>,
) -> Result<hyper::Response<hyper::Body>, crate::Error> {
    if params.0 == "main.css" {
        let mut resp = hyper::Response::new(FILE_MAIN_CSS.into());
        resp.headers_mut().insert(
            hyper::header::CONTENT_TYPE,
            hyper::header::HeaderValue::from_static("text/css"),
        );

        Ok(resp)
    } else if let Some(icon) = ICONS_MAP.get(params.0.as_str()) {
        let mut resp = hyper::Response::new(icon.content.into());
        resp.headers_mut().insert(
            hyper::header::CONTENT_TYPE,
            hyper::header::HeaderValue::from_static("image/svg+xml"),
        );
        resp.headers_mut().insert(
            hyper::header::CACHE_CONTROL,
            hyper::header::HeaderValue::from_static("max-age=31536000, immutable"),
        );

        Ok(resp)
    } else {
        Err(crate::Error::RoutingError(trout::RoutingFailure::NotFound))
    }
}