rung 0.2.0

Useful command-line tools for Angular projects written in Rust.
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

use actix_files::{Files, NamedFile};
use actix_web::http::StatusCode;
use actix_web::{guard, middleware, rt, web, App, HttpResponse, HttpServer, Result};
use log::info;
use std::path::Path;

struct AppState {
    root_dir: String,
}

/// 404 handler
async fn p404(data: web::Data<AppState>) -> Result<NamedFile> {
    // let file_path = Path::new(&data.root_dir).join("404.html");
    let file_path = Path::new(&data.root_dir).join("index.html");
    Ok(NamedFile::open(file_path)?.set_status_code(StatusCode::NOT_FOUND))
}

#[derive(Clone)]
pub struct ServerOptions {
    pub host: String,
    pub port: String,
    pub root_dir: String,
    pub open: bool,
}

impl ServerOptions {
    pub fn get_addr(&self) -> String {
        format!("{}:{}", self.host, self.port)
    }
    pub fn get_url(&self) -> String {
        format!("http://{}:{}", self.host, self.port)
    }
}

pub fn run_server(options: ServerOptions) -> std::io::Result<()> {
    let mut sys = rt::System::new("server");

    let addr = options.get_addr();
    let url = options.get_url();
    let open = options.open;

    let srv = HttpServer::new(move || {
        let root_dir = &options.root_dir;
        info!("Serving {}", root_dir);

        App::new()
            .data(AppState {
                root_dir: String::from(root_dir),
            })
            .wrap(middleware::Compress::default())
            // enable logger - always register actix-web Logger middleware last
            .wrap(middleware::Logger::default())
            .service(Files::new("/", &root_dir).index_file("index.html"))
            .default_service(
                // 404 for GET request
                web::resource("")
                    .route(web::get().to(p404))
                    // all requests that are not `GET`
                    .route(
                        web::route()
                            .guard(guard::Not(guard::Get()))
                            .to(HttpResponse::MethodNotAllowed),
                    ),
            )
    })
    .workers(1)
    .bind(addr)?
    .run();

    if open {
        webbrowser::open(url.as_str())?;
    }

    sys.block_on(srv)
}