desire 0.0.1

A minimal Rust web application framework
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

use crate::Endpoint;
use crate::Request;
use crate::Result;
use bytes::Bytes;
use http_body_util::Full;
use std::path::Path;
use std::path::PathBuf;
use tracing::info;
pub struct ServeFile {
  path: PathBuf,
}
impl ServeFile {
  pub fn new(path: PathBuf) -> Self {
    ServeFile { path }
  }
}
#[async_trait::async_trait]
impl Endpoint for ServeFile {
  async fn call(&self, _req: Request) -> Result {
    let body = tokio::fs::read(&self.path).await?;
    let response = hyper::Response::new(Full::new(Bytes::from(body)));
    Ok(response.into())
  }
}

pub struct ServeDir {
  dir: PathBuf,
}

impl ServeDir {
  pub fn new(dir: PathBuf) -> Self {
    ServeDir { dir }
  }
}
#[async_trait::async_trait]
impl Endpoint for ServeDir {
  async fn call(&self, req: Request) -> Result {
    let file = req.param::<String>("file")?;
    let dir = self.dir.clone();
    let ext = Path::new(&file).extension();
    let file = PathBuf::from(format!("{}/{}", dir.to_string_lossy(), file));
    let body = tokio::fs::read(file).await?;
    info!("ext {:?}", ext);
    let mut response = hyper::Response::new(Full::new(Bytes::from(body)));
    if let Some(ext) = ext {
      let ext = ext.to_string_lossy();
      if ext.contains(&"js") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::TEXT_JAVASCRIPT.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"css") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::TEXT_CSS.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"png") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_PNG.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"png") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_PNG.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"svg") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_SVG.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"jpeg") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_JPEG.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"gif") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_GIF.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"bmp") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::IMAGE_BMP.as_ref().parse().unwrap(),
        );
      } else if ext.contains(&"mp4") {
        response.headers_mut().insert(
          hyper::header::CONTENT_TYPE,
          mime::MP4.as_ref().parse().unwrap(),
        );
      }
    }
    Ok(response.into())
  }
}