use std::net::SocketAddr;

use tower_http::trace::TraceLayer;
use warpui::{Entity, ModelContext, SingletonEntity};

// Spells "Warp" - should hopefully not conflict with other ports.
// Does not conflict with known ports on https://en.wikipedia.org/wiki/List_of_TCP_and_UDP_port_numbers
const PORT: u16 = 9277;

/// A singleton model for the small HTTP server that is run by the Warp client.
pub struct HttpServer {
    /// The tokio runtime that the HTTP server runs on.
    ///
    /// We use a private runtime only because we don't currently have a shared
    /// tokio runtime.
    ///
    /// TODO(vorporeal): Remove this when we have a shared tokio runtime.
    _runtime: Option<tokio::runtime::Runtime>,
}

impl HttpServer {
    pub fn new(
        routers: impl IntoIterator<Item = axum::Router>,
        _ctx: &mut ModelContext<Self>,
    ) -> Self {
        let runtime = Self::spawn_server(routers)
            .inspect_err(|err| {
                log::warn!("Failed to start local HTTP server: {err:#}");
            })
            .ok();

        Self { _runtime: runtime }
    }

    fn spawn_server(
        routers: impl IntoIterator<Item = axum::Router>,
    ) -> Result<tokio::runtime::Runtime, std::io::Error> {
        let runtime = tokio::runtime::Builder::new_multi_thread()
            .worker_threads(1)
            .enable_io()
            .build()?;

        let mut root = axum::Router::new();
        for router in routers {
            root = root.merge(router);
        }

        runtime.spawn(async move {
            let addr = SocketAddr::from(([127, 0, 0, 1], PORT));
            let listener = tokio::net::TcpListener::bind(addr).await?;

            axum::serve(listener, root.layer(TraceLayer::new_for_http())).await
        });

        Ok(runtime)
    }
}

impl Entity for HttpServer {
    type Event = ();
}

impl SingletonEntity for HttpServer {}