use std::{
    io::{BufRead, BufReader, Write},
    net::{TcpListener, TcpStream},
    process::exit,
};

fn main() {
    println!("Starting the server");
    let target_address = "127.0.0.1:7878";
    let listener = match TcpListener::bind(target_address) {
        Ok(listener) => listener,
        Err(err) => {
            eprintln!("Could not start server: {err}");
            exit(1);
        }
    };
    for stream in listener.incoming() {
        let stream = stream.unwrap();

        handle_connection(stream);
    }
}

fn handle_connection(mut stream: TcpStream) {
    let buf_rd = BufReader::new(&mut stream);
    let request: Vec<_> = buf_rd
        .lines()
        .map(|result| result.unwrap())
        .take_while(|line| !line.is_empty())
        .collect();
    println!("Request: {:#?}", request);
    match request[0].as_str() {
        "ping" => {
            writeln!(stream, "> pong!").unwrap();
        }
        _ => {
            writeln!(stream, "> command not recognized.").unwrap();
        }
    }
}