cardano-serialization-lib 15.0.3

use super::indexes::{AssetIndex, PolicyIndex, UtxoIndex};
use crate::{Coin, JsError};
use std::collections::{HashMap, HashSet};

#[derive(Clone)]
pub(super) struct UtxosStat {
    pub(super) total_policies: usize,
    pub(super) assets_in_policy: HashMap<PolicyIndex, usize>,
    pub(super) coins_in_assets: HashMap<AssetIndex, Coin>,
    pub(super) ada_coins: Coin,
}

impl UtxosStat {
    pub(super) fn new(
        total_ada: &Coin,
        policy_to_asset: &HashMap<PolicyIndex, HashSet<AssetIndex>>,
        amounts: &Vec<HashMap<UtxoIndex, Coin>>,
    ) -> Result<Self, JsError> {
        let mut utxos_stat = UtxosStat {
            total_policies: 0,
            assets_in_policy: HashMap::new(),
            coins_in_assets: HashMap::new(),
            ada_coins: Coin::zero(),
        };
        for (policy_index, assets) in policy_to_asset {
            utxos_stat
                .assets_in_policy
                .insert(policy_index.clone(), assets.len());
        }

        for i in 0..amounts.len() {
            for (_, amount) in &amounts[i] {
                let asset_index = AssetIndex(i);
                if let Some(coins) = utxos_stat.coins_in_assets.get(&asset_index) {
                    let new_total = coins.checked_add(amount)?;
                    utxos_stat.coins_in_assets.insert(asset_index, new_total);
                } else {
                    utxos_stat
                        .coins_in_assets
                        .insert(asset_index, amount.clone());
                }
            }
        }

        utxos_stat.total_policies = policy_to_asset.len();
        utxos_stat.ada_coins = total_ada.clone();

        Ok(utxos_stat)
    }
}