stage 3 attempt 1

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 <T> (code: T) -> Vec<u8> where T: Display {
+    format!("HTTP/1.1 {code}\r\n\r\n").into()
 }