use sd_notify;

#[cfg(feature = "ssr")]
#[tokio::main]
async fn main() {
    use axum::Router;
    use leptos::logging::log;
    use leptos::prelude::*;
    use leptos_axum::{LeptosRoutes, generate_route_list};
    use tlaternet_webserver::app::*;
    use tlaternet_webserver::{AppState, Config};

    let config = Config::parse();

    let (addr, leptos_options) = {
        let conf = get_configuration(None).unwrap();
        let addr = conf.leptos_options.site_addr;
        let leptos_options = conf.leptos_options;
        (addr, leptos_options)
    };

    // Generate the list of routes in your Leptos App
    let routes = generate_route_list(App);
    let http_client = reqwest::Client::new();

    let state = AppState {
        config,
        http_client,
        leptos_options,
    };

    let app = Router::new()
        .leptos_routes(&state, routes, {
            let leptos_options = state.leptos_options.clone();
            move || shell(leptos_options.clone())
        })
        .fallback::<_, (_, _, axum::extract::State<AppState>, _)>(
            leptos_axum::file_and_error_handler(shell),
        )
        .with_state(state);

    // run our app with hyper
    // `axum::Server` is a re-export of `hyper::Server`
    log!("listening on http://{}", &addr);
    let _ = sd_notify::notify(true, &[sd_notify::NotifyState::Ready]);
    let listener = tokio::net::TcpListener::bind(&addr).await.unwrap();
    axum::serve(listener, app.into_make_service())
        .await
        .unwrap();
}

#[cfg(not(feature = "ssr"))]
pub fn main() {
    // no client-side main function
    // unless we want this to work with e.g., Trunk for pure client-side testing
    // see lib.rs for hydration function instead
}
