use core::panic;
use std::sync::Arc;

use tauri::{path::BaseDirectory, Manager};
use tokio::sync::Mutex;
use tower::{Service, ServiceExt};

#[cfg_attr(mobile, tauri::mobile_entry_point)]
pub fn run() {
    tauri::Builder::default()
        .setup(|app| {
            let resource_path_buf = app
                .path()
                .resolve("assets", BaseDirectory::Resource)
                .expect("Path should be resolvable");

            let router = Arc::new(Mutex::new(
                clego::get_router(resource_path_buf.as_path()).clone(),
            ));

            app.manage(router);

            Ok(())
        })
        .register_asynchronous_uri_scheme_protocol("axum", move |app, request, responder| {
            let router = Arc::clone(&app.state::<Arc<Mutex<axum::Router>>>());

            tauri::async_runtime::spawn(async move {
                let mut router = router.lock().await;

                let (parts, body) = request.into_parts();
                let body = axum::body::Body::from(body);

                let request = axum::extract::Request::from_parts(parts, body);

                let response = match router.as_service().ready().await {
                    Ok(ready_service) => ready_service.call(request).await,
                    Err(_error) => panic!("Failed to get ready service"),
                };

                let response = match response {
                    Ok(response) => response,
                    Err(_error) => panic!("Problem getting response from request."),
                };

                let (parts, body) = response.into_parts();
                let body = match axum::body::to_bytes(body, usize::MAX).await {
                    Ok(bytes) => bytes.to_vec(),
                    Err(_error) => panic!("Problem converting response body to bytes."),
                };

                let response = tauri::http::Response::from_parts(parts, body);

                responder.respond(response);
            });
        })
        .run(tauri::generate_context!())
        .expect("error while running tauri application");
}