1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use crate::*;
use bridge_types::GenericNetworkId;
use sp_runtime::traits::Zero;

pub struct HashiBridgeLockedAssets;

impl Get<Vec<(AssetId, Balance)>> for HashiBridgeLockedAssets {
    fn get() -> Vec<(AssetId, Balance)> {
        let Ok(assets) = EthBridge::get_registered_assets(Some(GetEthNetworkId::get())) else {
            frame_support::log::warn!("Failed to get registered assets, skipping migration");
            return vec![];
        };
        let Some(bridge_account) = eth_bridge::BridgeAccount::<Runtime>::get(GetEthNetworkId::get()) else {
            frame_support::log::warn!("Failed to get Hashi bridge account, skipping migration");
            return vec![];
        };
        let mut result = vec![];
        for (kind, (asset_id, _precision), _) in assets {
            let reserved = if kind.is_owned() {
                Assets::total_balance(&asset_id, &bridge_account)
            } else {
                Assets::total_issuance(&asset_id)
            };
            let reserved = reserved.unwrap_or_default();
            if !reserved.is_zero() {
                result.push((asset_id, reserved));
            }
        }
        result
    }
}

parameter_types! {
    pub const HashiBridgeNetworkId: GenericNetworkId = GenericNetworkId::EVMLegacy(GetEthNetworkId::get());
}

pub type Migrations = (
    bridge_proxy::migrations::init::InitLockedAssets<
        Runtime,
        HashiBridgeLockedAssets,
        HashiBridgeNetworkId,
    >,
);