seva 0.1.3

Simple directory http server inspired by Python's 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
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

#![deny(unused, missing_docs)]
/*!
A simple directory http server.

Seva serves files from a directory, directly mapping
the directory structure to HTTP requests.

Inspired by Python's http.server
*/
use std::{env, path::PathBuf};

use clap::Parser;
use tracing::{info, Level};
use tracing_subscriber::FmtSubscriber;

mod errors;
mod fs;
mod http;
mod mime;
mod server;
mod utils;

/// A simple directory http server.
///
/// Seva serves files from a directory, directly mapping
/// the directory structure to HTTP requests.
///
/// Inspired by Python's http.server
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    /// Directory to server files from.
    #[arg(short, long, default_value = ".")]
    directory: String,

    /// Http port to listen on.
    #[arg(short, long, default_value = "8001")]
    port: u16,

    /// Interface to bind to.
    #[arg(long, default_value = "127.0.0.1")]
    host: String,

    /// Log Level
    #[arg(long, default_value = "INFO")]
    log_level: Level,

    /// Timeout duration in seconds
    #[arg(long, default_value = "10")]
    timeout: u64,
}

fn expand_tilde(path: &str) -> String {
    let home = match home_dir() {
        None => return path.to_string(),
        Some(home) => home.to_string_lossy().into_owned(),
    };
    path.replace('~', &home)
}

fn home_dir() -> Option<PathBuf> {
    #![allow(deprecated)]
    std::env::home_dir()
}

fn main() -> errors::Result<()> {
    let args = Args::parse();
    let subscriber = FmtSubscriber::builder()
        .with_max_level(args.log_level)
        .compact()
        .with_file(false)
        .with_line_number(false)
        .finish();
    tracing::subscriber::set_global_default(subscriber)
        .expect("setting default subscriber failed");

    let dir = match args.directory.as_str() {
        "." => env::current_dir()?,
        _ => PathBuf::from(expand_tilde(&args.directory)),
    };

    info!("Starting seva in: {}", dir.display());
    info!(
        "Serving HTTP on {bind} port {port} (http://{bind}:{port}/) ...",
        bind = args.host,
        port = args.port,
    );

    let mut server =
        server::HttpServer::new(&args.host, args.port, dir, false, args.timeout)?;
    server.run()?;
    Ok(())
}