server

use std::net::TcpListener;
use std::io::{Read, Write};
use std::thread;

fn main() {
    let args: Vec<String> = std::env::args().collect();
    let port = &args[1];
    let server_name = &args[2];

    let addr = format!("127.0.0.1:{}", port);
    let listener = TcpListener::bind(&addr).unwrap();
    println!("{} listening on {}", server_name, addr);

    for stream in listener.incoming() {
        let stream = stream.unwrap();
        let server_name = server_name.clone();

        thread::spawn(move || {
            handle_connection(stream, &server_name);
        });
    }
}

fn handle_connection(mut stream: std::net::TcpStream, server_name: &str) {
    let mut buffer = [0; 1024];
    stream.read(&mut buffer).unwrap();

    let response = format!(
        "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\nHello from {}",
        server_name
    );

    stream.write(response.as_bytes()).unwrap();
    stream.flush().unwrap();
}