http-server-cc/src/main.rs
2024-05-10 20:50:36 +03:00

57 lines
1.6 KiB
Rust

use std::{ fmt::Display, io::{Read, Write}, net::TcpListener };
use anyhow::Result;
use thiserror::Error;
#[derive(Clone, Debug, Error)]
enum E {
#[error("Invalid request data found during parsing")]
InvalidRequest,
}
fn main() -> Result<()> {
let listener = TcpListener::bind("127.0.0.1:4221").unwrap();
for stream in listener.incoming() {
match stream {
Ok(mut stream) => {
let mut buf = String::new();
let _read = stream.read_to_string(&mut buf);
let mut data = buf.split("\r\n");
let _start @ (_method, path, _ver) = {
let start_line = data.next().ok_or(E::InvalidRequest)?; // should be 500;
let mut parts = start_line.split_whitespace();
let method = parts.next().ok_or(E::InvalidRequest)?;
let path = parts.next().ok_or(E::InvalidRequest)?;
let ver = parts.next().ok_or(E::InvalidRequest)?;
(method, path, ver)
};
let code = match path {
"/" => "200 OK",
_ => "404 Not Found",
};
println!("accepted new connection");
let resp = create_response_simple(code);
let _ = stream.write(&resp);
let _ = stream.flush();
}
Err(e) => {
println!("error: {}", e);
}
}
}
Ok(())
}
fn create_response_simple <T> (code: T) -> Vec<u8> where T: Display {
format!("HTTP/1.1 {code}\r\n\r\n").into()
}