diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 33 |
1 files changed, 32 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs index 0ea01c5..2f61490 100644 --- a/src/main.rs +++ b/src/main.rs @@ -60,7 +60,9 @@ impl From<&[&str]> for Headers { fn from(value: &[&str]) -> Self { let mut header_map = HashMap::new(); for header in value.iter().filter(|val| !val.is_empty()) { - let (key, val) = header.split_once(':').expect("Should be splitable by :"); + let (key, val) = header + .split_once(": ") + .expect("Should be splitable by :<space>"); header_map.insert(key.to_string(), val.to_string()); } Headers(header_map) @@ -221,6 +223,35 @@ fn handle_client(mut stream: TcpStream) { Err(e) => eprintln!("Failed to send response: {}", e), } } + ("user-agent", _) => { + let mut headers = HashMap::new(); + let user_agent: String = + request.headers.unwrap().0.get("User-Agent").unwrap().into(); + headers + .insert("Content-Type".to_string(), "text/plain".to_string()); + headers.insert( + "Content-Length".to_string(), + user_agent.len().to_string(), + ); + let body = user_agent; + let response: String = Response::new( + "1.1".to_string(), + StatusCode::Ok, + Some(Headers(headers)), + Some(body), + ) + .into(); + dbg!(&response); + let response = response.as_bytes(); + + match stream.write(response) { + Ok(_) => { + println!("Response sent successfully"); + println!("Hello user-agent"); + } + Err(e) => eprintln!("Failed to send response: {}", e), + } + } _ => match stream.write(not_found) { Ok(_) => println!("Response sent successfully"), Err(e) => eprintln!("Failed to send response: {}", e), |