rust-http-server 0.0.1

A simple and fast HTTP server written in rust. 
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

use std::usize;
use std::env;

use clap::{App, ArgMatches};

use config::{ConfigBuilder, Config};
use server::Server;

pub fn run_cli<'a, 'b>() -> App<'a, 'b> {
    let name = env!("CARGO_PKG_NAME");
    let version = env!("CARGO_PKG_VERSION");
    let author = env!("CARGO_PKG_AUTHORS");
    App::new(name)
        .version(version)
        .author(author)
        .args_from_usage(
            "[DIRECTORY]            'Serves the contents of the directory'
            -h, --host=[HOST]       'Sets the host address'
            -p, --port=[PORT]       'Sets the port'
            -a, --app=[APP]         '<module>:<callable> The server application'
            --cert=[CERT]           'Path to pkcs12 certificate'
            --pwd=[PWD]             'Password for the pkcs12'
            -t, --threads=[THREADS] 'Sets the number of threads to use'
            -l, --log=[LOG_LEVEL]   'Sets the log level (debug, error, info, trace, warn)'
            -v...                   'Sets verbosity'")

}

pub fn config_from_cli(args: &ArgMatches) -> Config {
    let mut config = ConfigBuilder::from_json_file("config.json").unwrap_or_default();

    if let Some(t) = args.value_of("threads") {
        let threads = usize::from_str_radix(t, 10)
            .expect("Please enter an integer for thread size");
        config.set_threads(threads);
    }

    if let Some(host) = args.value_of("host") {
        config.set_host(host);
    }

    if let Some(port) = args.value_of("port") {
        config.set_port(port);
    }

    if let Some(app) = args.value_of("app") {
        config.set_app(app);
    }

    if let Some(cert) = args.value_of("cert") {
        config.set_https_cert(cert);
    }

    if let Some(pwd) = args.value_of("pwd") {
        config.set_cert_password(pwd);
    }

    config.build()
}

pub fn cli_logging(args: &ArgMatches) {
    if let Some(log) = args.value_of("log") {
        env::set_var("RUST_HTTP_SERVER_LOG", log)
    }
}

pub fn cli_verbosity(app: &ArgMatches) {
    match app.occurrences_of("v") {
        0 => set_verbosity(false),
        _ => set_verbosity(true),
    }
}

pub fn cli_serve_directory(app: &ArgMatches, server: &mut Server) {
    if let Some(dir) = app.value_of("DIRECTORY") {
        server.serve_directory(dir);
    }
}

fn set_verbosity(value: bool) {
    unsafe {
        ::VERBOSE = value;
    }
}