crab-http 0.2.1

A lightweight HTTP server for serving static files
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

use colored::Colorize;
use std::net::TcpListener;
use std::thread;
use std::time::Duration;
mod argsparser;
mod h_res;
pub mod htmlgenerator;
mod req;
// todo, threading
fn main() {
    let (port, directory, allow_write, timeout) = argsparser::getargs();
    run(port, directory.as_deref(), allow_write, timeout);
}

fn run(port: u16, directory: Option<&str>, allow_write: bool, timeout: u64) {
    println!("🚀 Starting server on port: {}", port);

    let listener = TcpListener::bind(format!("0.0.0.0:{}", port)).unwrap();

    for stream in listener.incoming() {
        match stream {
            Ok(mut stream) => {
                let msg = "Accepted new connection";
                println!("{}", msg.green());

                let timeout = Duration::from_secs(timeout);
                stream.set_read_timeout(Some(timeout)).unwrap();
                stream.set_write_timeout(Some(timeout)).unwrap();

                let dir = directory.map(|s| s.to_string());
                thread::spawn(move || loop {
                    let should_close = req::handle_req(&mut stream, &dir, allow_write);

                    if should_close {
                        break;
                    }
                });
            }
            Err(e) => {
                eprintln!("error: {}", e);
            }
        }
    }
}