http-server 1.0.0-pre.6fa7635

Simple and configurable command-line HTTP server
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

use std::net::IpAddr;
use std::process::exit;
use std::sync::Arc;

use anyhow::Result;
use clap::Parser;
use tokio::runtime::Builder;
use tracing::{error, info};

use crate::config::Config;
use crate::server::Server;

const THREAD_NAME: &str = "http-server";

#[derive(Debug, Parser)]
pub struct StartOpt {
    /// Host (IP) to bind the server
    #[clap(long, default_value = "0.0.0.0")]
    pub host: IpAddr,
    /// Port to bind the server
    #[clap(short = 'p', long, default_value = "7878")]
    pub port: u16,
    /// Enable CORS with a permissive policy
    #[clap(long, default_value = "false")]
    pub cors: bool,
}

impl From<&StartOpt> for Config {
    fn from(val: &StartOpt) -> Self {
        Config {
            host: val.host,
            port: val.port,
            cors: val.cors,
        }
    }
}

impl StartOpt {
    pub fn exec(&self) -> Result<()> {
        let rt = Builder::new_multi_thread()
            .enable_all()
            .thread_name(THREAD_NAME)
            .build()?;
        let rt = Arc::new(rt);
        let config: Config = self.into();
        let server = Server::new(config);

        rt.block_on(async {
            match server.run().await {
                Ok(_) => {
                    info!("Server exited successfuly");
                    Ok(())
                }
                Err(error) => {
                    error!(%error, "Server exited with error");
                    exit(1);
                }
            }
        })
    }
}