use std::env;
use std::net::SocketAddr;

use hyper::server::conn::http1;
use hyper::service::service_fn;
use hyper_util::rt::TokioIo;
use tokio::net::TcpListener;
use tokio_rusqlite::Connection;

mod db;
mod jobs;
mod model;
mod routes;
mod templates;
mod util;

#[tokio::main]
async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
    env_logger::init();

    let host = get_env("HOST");
    let port = get_env("PORT");
    let db_path = get_env("DB");
    let authorized_key = get_env("KEY");
    let files_dir = get_env("FILES_DIR");

    let db_conn = Connection::open(db_path)
        .await
        .expect("Error while openning DB conection");

    tokio::spawn(jobs::start(db_conn.clone(), files_dir.clone()));

    let addr: SocketAddr = format!("{host}:{port}")
        .parse()
        .unwrap_or_else(|_| panic!("Invalid address: {host}:{port}"));

    let listener = TcpListener::bind(addr).await?;
    log::info!("Listening on http://{}", addr);

    loop {
        let (stream, _) = listener.accept().await?;
        let io = TokioIo::new(stream);

        let db_conn = db_conn.clone();
        let authorized_key = authorized_key.clone();
        let files_dir = files_dir.clone();

        tokio::task::spawn(async move {
            if let Err(err) = http1::Builder::new()
                .serve_connection(
                    io,
                    service_fn(move |req| {
                        routes::routes(
                            req,
                            db_conn.clone(),
                            authorized_key.clone(),
                            files_dir.clone(),
                        )
                    }),
                )
                .await
            {
                log::error!("Failed to serve connection: {:?}", err);
            }
        });
    }
}

fn get_env(key: &str) -> String {
    env::var(key).unwrap_or_else(|_| panic!("Missing environment variable {key}"))
}