use std::sync::Arc;

use axum::extract::rejection::PathRejection;
use axum::extract::Path;
use axum::extract::State;
use axum::response::Html;
use axum::response::Response;
use html_escaper::Escape;
use html_escaper::Trusted;
use neptune_privacy::api::export::Tip5;
use neptune_privacy::prelude::tasm_lib::prelude::Digest;
use tarpc::context;

use crate::html::component::header::HeaderHtml;
use crate::html::page::not_found::not_found_html_response;
use crate::http_util::rpc_method_err;
use crate::model::app_state::AppState;
use crate::model::transparent_utxo_tuple::TransparentUtxoTuple;

#[axum::debug_handler]
pub async fn utxo_page(
    index_maybe: Result<Path<u64>, PathRejection>,
    State(state_rw): State<Arc<AppState>>,
) -> Result<Html<String>, Response> {
    #[derive(boilerplate::Boilerplate)]
    #[boilerplate(filename = "web/html/page/utxo.html")]
    pub struct UtxoHtmlPage<'a> {
        header: HeaderHtml<'a>,
        index: u64,
        digest: Digest,
        transparent_utxo_info: Option<TransparentUtxoTuple>,
    }

    let state = &state_rw.load();
    let cache = state.transparent_utxos_cache.clone();

    let Path(index) =
        index_maybe.map_err(|e| not_found_html_response(state, Some(e.to_string())))?;

    let digest = match state
        .rpc_client
        .utxo_digest(context::current(), state.token(), index, cache)
        .await
        .map_err(|e| not_found_html_response(state, Some(e.to_string())))?
        .map_err(rpc_method_err)?
    {
        Some(digest) => digest,
        None => {
            return Err(not_found_html_response(
                state,
                Some("The requested UTXO does not exist".to_string()),
            ))
        }
    };

    let header = HeaderHtml { state };

    let transparent_utxo_info = state
        .transparent_utxos_cache
        .lock()
        .await
        .iter()
        .find(|tu| tu.aocl_leaf_index().is_some_and(|li| li == index))
        .cloned();

    let utxo_page = UtxoHtmlPage {
        index,
        header,
        digest,
        transparent_utxo_info,
    };
    Ok(Html(utxo_page.to_string()))
}