use clap::Parser;
use emerwen_protocol::Message;
use tokio::net::TcpListener;
use tracing::{debug, error, info, level_filters::LevelFilter};
use tracing_subscriber::{prelude::*, EnvFilter};

#[derive(Parser)]
#[command(version)]
struct Args {
    /// Enable debug logging
    #[arg(long)]
    debug: bool,
}

#[tokio::main]
async fn main() {
    let args = Args::parse();

    tracing_subscriber::registry()
        .with(
            tracing_subscriber::fmt::layer()
                .with_file(args.debug)
                .with_line_number(args.debug)
                .with_level(true),
        )
        .with(
            EnvFilter::builder()
                .with_default_directive(if args.debug {
                    LevelFilter::DEBUG.into()
                } else {
                    LevelFilter::INFO.into()
                })
                .from_env_lossy(),
        )
        .init();

    info!("Starting emerwen worker...");

    debug!("Hello, debug!");

    let worker_server = WorkerServer::new("127.0.0.1:8000");
    let _ = worker_server.run().await;
}

struct WorkerServer {
    addr: String,
}

impl WorkerServer {
    pub fn new(addr: impl Into<String>) -> WorkerServer {
        WorkerServer { addr: addr.into() }
    }

    pub async fn run(&self) -> Result<(), Box<dyn std::error::Error>> {
        let listener = TcpListener::bind(&self.addr).await?;
        info!("Worker Server listening on {}", self.addr);

        loop {
            let (mut stream, addr) = listener.accept().await?;
            info!("Client '{}' connected", addr);

            tokio::task::spawn(async move {
                loop {
                    let message = match Message::decode(&mut stream).await {
                        Ok(message) => message,
                        Err(e) => {
                            error!("Error decoding message: {:?}", e);

                            continue;
                        }
                    };
                    debug!("{:?}", message);
                }
            });
        }
    }
}