ddoc 0.18.0

doc site generator
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

//! nothing yet
use {
    crate::*,
    rouille::Response,
    std::path::PathBuf,
};

pub struct Server {
    base_url: String,
    addr: String,
    static_path: PathBuf,
}

impl Server {
    pub fn new(
        static_path: PathBuf,
        port: u16,
    ) -> DdResult<Self> {
        let addr = format!("localhost:{port}");
        let base_url = format!("http://{addr}/");
        Ok(Self {
            base_url,
            addr,
            static_path,
        })
    }
    pub fn base_url(&self) -> &str {
        &self.base_url
    }
    pub fn run(self) -> DdResult<()> {
        let static_path = self.static_path;
        let rouille_server = rouille::Server::new(self.addr, move |request| {
            // build the file path
            let mut path = static_path.clone();
            path.push(&request.url()[1..]); // Remove leading /

            if path.is_dir() {
                if request.url().ends_with('/') {
                    // If it's a directory with trailing /,
                    // the URL is correct but we must send index.html
                    path.push("index.html");
                    if path.exists() {
                        if let Ok(file) = std::fs::File::open(&path) {
                            return Response::from_file("text/html", file);
                        }
                    }
                } else {
                    // The URL is missing a trailing /
                    let new_url = format!("{}/", request.url());
                    return Response::redirect_301(new_url);
                }
            }

            // Try to serve the file
            rouille::match_assets(request, &static_path)
        })
        .map_err(|e| DdError::Server(e.to_string()))?;
        rouille_server.run();
        Ok(())
    }
}