use dioxus::prelude::*;

fn main() {
    #[cfg(not(feature = "server"))]
    dioxus::launch(App);
    #[cfg(feature = "server")]
    server::server_main();
}

#[cfg(feature = "server")]
mod server {
    use axum::extract::WebSocketUpgrade;
    use axum::response::IntoResponse;
    use dioxus::logger::tracing::{info, warn};
    use dioxus::prelude::*;
    use futures::StreamExt;

    #[tokio::main]
    pub async fn server_main() {
        dioxus::logger::initialize_default();
        let socket_addr = dioxus_cli_config::fullstack_address_or_localhost();
        let router = axum::Router::new()
            .serve_dioxus_application(ServeConfigBuilder::new(), super::App)
            .route("/events", axum::routing::any(websocket_handler))
            .into_make_service();

        let listener = tokio::net::TcpListener::bind(socket_addr).await.unwrap();
        axum::serve(listener, router).await.unwrap();
    }

    async fn websocket_handler(ws: WebSocketUpgrade) -> impl IntoResponse {
        ws.on_upgrade(|mut ws| async move {
            while let Some(msg) = ws.next().await {
                match msg {
                    Ok(msg) => info!("Got message: {msg:?}"),
                    Err(e) => warn!("Error receiving message: {e}"),
                }
            }
        })
    }
}

#[component]
fn App() -> Element {
    #[cfg(feature = "web")]
    {
        _ = use_future(|| async {
            use dioxus::logger::tracing::error;
            use futures::SinkExt;
            use gloo::net::websocket::futures::WebSocket;

            let mut ws = match WebSocket::open("/events") {
                Ok(ws) => ws,
                Err(e) => {
                    error!("Could not open websocket connection: {e}");
                    return;
                }
            };

            if let Err(e) = ws
                .send(gloo::net::websocket::Message::Text(
                    "this is a websocket message".into(),
                ))
                .await
            {
                error!("Could not send websocket message: {e}");
            }
        });
    }

    rsx! {
        "hi"
    }
}