diff --git a/src/main.rs b/src/main.rs index 454d338..909121e 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,18 +1,56 @@ -use std::{ io::Write, net::TcpListener }; +use std::{ fmt::Display, io::{Read, Write}, net::TcpListener }; +use anyhow::Result; +use thiserror::Error; -fn main() { +#[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 _200 = b"HTTP/1.1 200 OK\r\n\r\n"; - let _ = stream.write(_200); + let resp = create_response_simple(code); + + let _ = stream.write(&resp); + let _ = stream.flush(); + } Err(e) => { println!("error: {}", e); } } } + + Ok(()) +} + + +fn create_response_simple (code: T) -> Vec where T: Display { + format!("HTTP/1.1 {code}\r\n\r\n").into() }