@subwallet/extension-base 0.8.4-0 → 1.0.2-1

package/koni/api/dotsama/balance.js

@@ -0,0 +1,567 @@
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+import { _AssetType } from '@subwallet/chain-list/types';
+import { APIItemState } from '@subwallet/extension-base/background/KoniTypes';
+import { ASTAR_REFRESH_BALANCE_INTERVAL, SUB_TOKEN_REFRESH_BALANCE_INTERVAL } from '@subwallet/extension-base/constants';
+import { getEVMBalance } from '@subwallet/extension-base/koni/api/tokens/evm/balance';
+import { getERC20Contract } from '@subwallet/extension-base/koni/api/tokens/evm/web3';
+import { getPSP22ContractPromise } from '@subwallet/extension-base/koni/api/tokens/wasm';
+import { state } from '@subwallet/extension-base/koni/background/handlers';
+import { _BALANCE_CHAIN_GROUP, _BALANCE_TOKEN_GROUP, _PURE_EVM_CHAINS } from '@subwallet/extension-base/services/chain-service/constants';
+import { _checkSmartContractSupportByChain, _getChainNativeTokenSlug, _getContractAddressOfToken, _getTokenOnChainAssetId, _getTokenOnChainInfo, _isChainEvmCompatible, _isNativeToken, _isPureEvmChain, _isSmartContractToken, _isSubstrateRelayChain } from '@subwallet/extension-base/services/chain-service/utils';
+import { categoryAddresses, sumBN } from '@subwallet/extension-base/utils';
+import { BN, BN_ZERO } from '@polkadot/util';
+import { isEthereumAddress } from '@polkadot/util-crypto';
+// main subscription
+export function subscribeBalance(addresses, chainInfoMap, substrateApiMap, evmApiMap, callback) {
+  const [substrateAddresses, evmAddresses] = categoryAddresses(addresses);
+
+  // Looping over each chain
+  const unsubList = Object.entries(chainInfoMap).map(async ([chainSlug, chainInfo]) => {
+    const useAddresses = _isChainEvmCompatible(chainInfo) ? evmAddresses : substrateAddresses;
+    if (_isPureEvmChain(chainInfo)) {
+      const nativeTokenInfo = state.getNativeTokenInfo(chainSlug);
+      return subscribeEVMBalance(chainSlug, useAddresses, evmApiMap, callback, nativeTokenInfo);
+    }
+    if (!useAddresses || useAddresses.length === 0 || _PURE_EVM_CHAINS.indexOf(chainSlug) > -1) {
+      const fungibleTokensByChain = state.chainService.getFungibleTokensByChain(chainSlug, true);
+      const now = new Date().getTime();
+      Object.values(fungibleTokensByChain).map(token => {
+        return {
+          tokenSlug: token.slug,
+          free: '0',
+          locked: '0',
+          state: APIItemState.READY,
+          timestamp: now
+        };
+      }).forEach(callback);
+      return undefined;
+    }
+    const networkAPI = await substrateApiMap[chainSlug].isReady;
+    return subscribeSubstrateBalance(useAddresses, chainInfo, chainSlug, networkAPI, evmApiMap, callback);
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}
+export async function subscribeSubstrateBalance(addresses, chainInfo, chain, networkAPI, evmApiMap, callBack) {
+  let unsubNativeToken;
+  if (!_BALANCE_CHAIN_GROUP.kintsugi.includes(chain) && !_BALANCE_CHAIN_GROUP.genshiro.includes(chain) && !_BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+    unsubNativeToken = await subscribeWithSystemAccountPallet(addresses, chainInfo, networkAPI.api, callBack);
+  }
+  let unsubLocalToken;
+  let unsubEvmContractToken;
+  let unsubWasmContractToken;
+  try {
+    if (_BALANCE_CHAIN_GROUP.bifrost.includes(chain)) {
+      unsubLocalToken = await subscribeTokensAccountsPallet(addresses, chain, networkAPI.api, callBack);
+    } else if (_BALANCE_CHAIN_GROUP.kintsugi.includes(chain)) {
+      unsubLocalToken = await subscribeTokensAccountsPallet(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_BALANCE_CHAIN_GROUP.statemine.includes(chain)) {
+      unsubLocalToken = await subscribeAssetsAccountPallet(addresses, chain, networkAPI.api, callBack);
+    } else if (_BALANCE_CHAIN_GROUP.genshiro.includes(chain)) {
+      unsubLocalToken = await subscribeEqBalanceAccountPallet(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+      unsubLocalToken = await subscribeEquilibriumTokenBalance(addresses, chain, networkAPI.api, callBack, true);
+    }
+    if (_isChainEvmCompatible(chainInfo)) {
+      unsubEvmContractToken = subscribeERC20Interval(addresses, chain, evmApiMap, callBack);
+    }
+    if (_checkSmartContractSupportByChain(chainInfo, _AssetType.PSP22)) {
+      // Get sub-token for substrate-based chains
+      unsubWasmContractToken = subscribePSP22Balance(addresses, chain, networkAPI.api, callBack);
+    }
+  } catch (err) {
+    console.warn(err);
+  }
+  return () => {
+    unsubNativeToken && unsubNativeToken();
+    unsubLocalToken && unsubLocalToken();
+    unsubEvmContractToken && unsubEvmContractToken();
+    unsubWasmContractToken && unsubWasmContractToken();
+  };
+}
+
+// handler according to different logic
+async function subscribeWithSystemAccountPallet(addresses, chainInfo, networkAPI, callBack) {
+  const chainNativeTokenSlug = _getChainNativeTokenSlug(chainInfo);
+  const unsub = await networkAPI.query.system.account.multi(addresses, async balances => {
+    let [total, reserved, miscFrozen, feeFrozen] = [new BN(0), new BN(0), new BN(0), new BN(0)];
+    let pooledStakingBalance = BN_ZERO;
+    if (_isSubstrateRelayChain(chainInfo) && networkAPI.query.nominationPools) {
+      const poolMemberDatas = await networkAPI.query.nominationPools.poolMembers.multi(addresses);
+      for (const _poolMemberData of poolMemberDatas) {
+        const poolMemberData = _poolMemberData.toPrimitive();
+        if (poolMemberData) {
+          const pooledBalance = new BN(poolMemberData.points.toString());
+          pooledStakingBalance = pooledStakingBalance.add(pooledBalance);
+        }
+      }
+    }
+    balances.forEach(balance => {
+      var _balance$data, _balance$data$free, _balance$data2, _balance$data2$reserv, _balance$data3, _balance$data3$miscFr, _balance$data4, _balance$data4$feeFro;
+      total = total.add(((_balance$data = balance.data) === null || _balance$data === void 0 ? void 0 : (_balance$data$free = _balance$data.free) === null || _balance$data$free === void 0 ? void 0 : _balance$data$free.toBn()) || new BN(0)); // reserved is seperated
+      reserved = reserved.add(((_balance$data2 = balance.data) === null || _balance$data2 === void 0 ? void 0 : (_balance$data2$reserv = _balance$data2.reserved) === null || _balance$data2$reserv === void 0 ? void 0 : _balance$data2$reserv.toBn()) || new BN(0));
+      miscFrozen = miscFrozen.add(((_balance$data3 = balance.data) === null || _balance$data3 === void 0 ? void 0 : (_balance$data3$miscFr = _balance$data3.miscFrozen) === null || _balance$data3$miscFr === void 0 ? void 0 : _balance$data3$miscFr.toBn()) || new BN(0));
+      feeFrozen = feeFrozen.add(((_balance$data4 = balance.data) === null || _balance$data4 === void 0 ? void 0 : (_balance$data4$feeFro = _balance$data4.feeFrozen) === null || _balance$data4$feeFro === void 0 ? void 0 : _balance$data4$feeFro.toBn()) || new BN(0));
+    });
+    let locked = reserved.add(miscFrozen);
+    total = total.add(reserved); // total = free + reserved
+
+    if (pooledStakingBalance.gt(BN_ZERO)) {
+      total = total.add(pooledStakingBalance);
+      locked = locked.add(pooledStakingBalance);
+    }
+    const free = total.sub(locked);
+    callBack({
+      tokenSlug: chainNativeTokenSlug,
+      free: free.toString(),
+      locked: locked.toString(),
+      state: APIItemState.READY,
+      substrateInfo: {
+        miscFrozen: miscFrozen.toString(),
+        reserved: reserved.toString(),
+        feeFrozen: feeFrozen.toString()
+      }
+    });
+  });
+  return () => {
+    unsub();
+  };
+}
+
+// deprecated
+// function subscribeDarwiniaBalance (addresses: string[], chainInfo: _ChainInfo, chain: string, networkAPI: ApiPromise, callBack: (rs: BalanceItem) => void) {
+//   const tokenMap = state.getAssetByChainAndAsset(chain, [_AssetType.LOCAL]);
+//   const nativeTokenSlug = _getChainNativeTokenSlug(chainInfo);
+//
+//   let totalBalance: BN = new BN(0);
+//   let freeBalance: BN = new BN(0);
+//   let miscFrozen: BN = new BN(0);
+//   let reservedKtonBalance: BN = new BN(0);
+//   let totalKtonBalance: BN = new BN(0);
+//
+//   const unsubProms = addresses.map((address) => {
+//     return networkAPI.derive.balances?.all(address, async (balance: DeriveBalancesAll) => {
+//       // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+//       freeBalance = freeBalance.add(balance.availableBalance?.toBn() || new BN(0));
+//       miscFrozen = miscFrozen.add(balance.lockedBalance?.toBn() || new BN(0));
+//       totalBalance = totalBalance.add(balance.freeBalance?.toBn() || new BN(0));
+//
+//       const _systemBalance = await networkAPI.query.system.account(address);
+//       const systemBalance = _systemBalance.toHuman() as unknown as AccountInfo;
+//
+//       // @ts-ignore
+//       const rawTotalKton = (systemBalance.data?.freeKton as string).replaceAll(',', '');
+//       // @ts-ignore
+//       const rawReservedKton = (systemBalance.data?.reservedKton as string).replaceAll(',', '');
+//
+//       // @ts-ignore
+//       // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+//       totalKtonBalance = totalKtonBalance.add(new BN(rawTotalKton) || new BN(0));
+//       // @ts-ignore
+//       // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+//       reservedKtonBalance = reservedKtonBalance.add(new BN(rawReservedKton) || new BN(0));
+//
+//       if (Object.keys(tokenMap).length > 1) {
+//         for (const tokenInfo of Object.values(tokenMap)) {
+//           if (_BALANCE_TOKEN_GROUP.crab.includes(tokenInfo.symbol)) {
+//             const freeKton = totalKtonBalance.sub(reservedKtonBalance);
+//
+//             callBack({
+//               free: freeKton.toString(),
+//               locked: reservedKtonBalance.toString(),
+//               state: APIItemState.READY,
+//               tokenSlug: tokenInfo.slug
+//             });
+//             break;
+//           }
+//         }
+//       }
+//
+//       const free = totalBalance.sub(miscFrozen);
+//
+//       callBack({
+//         tokenSlug: nativeTokenSlug,
+//         free: free.toString(),
+//         locked: miscFrozen.toString(),
+//         state: APIItemState.READY,
+//         substrateInfo: {
+//           miscFrozen: miscFrozen.toString()
+//         }
+//       });
+//     });
+//   });
+//
+//   return () => {
+//     Promise.all(unsubProms).then((unsubs) => {
+//       unsubs.forEach((unsub) => {
+//         unsub && unsub();
+//       });
+//     }).catch(console.error);
+//   };
+// }
+
+function subscribeERC20Interval(addresses, chain, evmApiMap, callBack) {
+  let tokenList = {};
+  const erc20ContractMap = {};
+  const getTokenBalances = () => {
+    Object.values(tokenList).map(async tokenInfo => {
+      let free = new BN(0);
+      try {
+        const contract = erc20ContractMap[tokenInfo.slug];
+        const balanceList = await Promise.all(addresses.map(address => {
+          // eslint-disable-next-line @typescript-eslint/no-unsafe-return,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+          return contract.methods.balanceOf(address).call();
+        }));
+        free = sumBN(balanceList.map(balance => new BN(balance || 0)));
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: '0',
+          state: APIItemState.READY
+        });
+      } catch (err) {
+        console.log('There is a problem fetching ' + tokenInfo.slug + ' token balance', err);
+      }
+    });
+  };
+  tokenList = state.getAssetByChainAndAsset(chain, [_AssetType.ERC20]);
+  Object.entries(tokenList).forEach(([slug, tokenInfo]) => {
+    erc20ContractMap[slug] = getERC20Contract(chain, _getContractAddressOfToken(tokenInfo), evmApiMap);
+  });
+  getTokenBalances();
+  const interval = setInterval(getTokenBalances, SUB_TOKEN_REFRESH_BALANCE_INTERVAL);
+  return () => {
+    clearInterval(interval);
+  };
+}
+function subscribePSP22Balance(addresses, chain, api, callBack) {
+  let tokenList = {};
+  const psp22ContractMap = {};
+  const getTokenBalances = () => {
+    Object.values(tokenList).map(async tokenInfo => {
+      let free = new BN(0);
+      try {
+        const contract = psp22ContractMap[tokenInfo.slug];
+        const balances = await Promise.all(addresses.map(async address => {
+          const _balanceOf = await contract.query['psp22::balanceOf'](address, {
+            gasLimit: -1
+          }, address);
+          return _balanceOf.output ? _balanceOf.output.toString() : '0';
+        }));
+        free = sumBN(balances.map(bal => new BN(bal || 0)));
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: '0',
+          state: APIItemState.READY
+        });
+      } catch (err) {
+        console.log('There is a problem fetching ' + tokenInfo.slug + ' PSP-22 token balance', err);
+      }
+    });
+  };
+  tokenList = state.getAssetByChainAndAsset(chain, [_AssetType.PSP22]);
+  Object.entries(tokenList).forEach(([slug, tokenInfo]) => {
+    psp22ContractMap[slug] = getPSP22ContractPromise(api, _getContractAddressOfToken(tokenInfo));
+  });
+  getTokenBalances();
+  const interval = setInterval(getTokenBalances, SUB_TOKEN_REFRESH_BALANCE_INTERVAL);
+  return () => {
+    clearInterval(interval);
+  };
+}
+async function subscribeEquilibriumTokenBalance(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_AssetType.NATIVE, _AssetType.LOCAL] : [_AssetType.LOCAL];
+  const tokenMap = state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsub = await api.query.system.account.multi(addresses, balances => {
+    // Equilibrium customizes the SystemAccount pallet
+    Object.values(tokenMap).forEach(tokenInfo => {
+      const assetId = _getTokenOnChainAssetId(tokenInfo);
+      let tokenFreeBalance = BN_ZERO;
+      for (const balance of balances) {
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+        const balancesData = JSON.parse(balance.data.toString());
+        const balanceList = balancesData.v0.balance;
+
+        // @ts-ignore
+        const freeTokenBalance = balanceList.find(data => data[0] === parseInt(assetId));
+        const bnFreeTokenBalance = freeTokenBalance ? new BN(freeTokenBalance[1].positive.toString()) : BN_ZERO;
+        tokenFreeBalance = tokenFreeBalance.add(bnFreeTokenBalance);
+      }
+      const tokenBalance = {
+        free: tokenFreeBalance.toString(),
+        locked: '0',
+        // Equilibrium doesn't show locked balance
+        state: APIItemState.READY,
+        tokenSlug: tokenInfo.slug
+      };
+      callBack(tokenBalance);
+    });
+  });
+  return () => {
+    unsub();
+  };
+}
+
+// eslint-disable-next-line @typescript-eslint/require-await
+async function subscribeEqBalanceAccountPallet(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_AssetType.NATIVE, _AssetType.LOCAL] : [_AssetType.LOCAL];
+  const tokenMap = state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsubList = Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const assetId = _getTokenOnChainAssetId(tokenInfo);
+      const unsub = await api.query.eqBalances.account.multi(addresses.map(address => [address, [assetId]]), balances => {
+        const tokenBalance = {
+          free: sumBN(balances.map(b => b.asPositive)).toString(),
+          locked: '0',
+          // Equilibrium doesn't show locked balance
+          state: APIItemState.READY,
+          tokenSlug: tokenInfo.slug
+        };
+        callBack(tokenBalance);
+      });
+      return unsub;
+    } catch (err) {
+      console.warn(err);
+      return undefined;
+    }
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}
+async function subscribeTokensAccountsPallet(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_AssetType.NATIVE, _AssetType.LOCAL] : [_AssetType.LOCAL];
+  const tokenMap = state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsubList = await Promise.all(Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const onChainInfo = _getTokenOnChainInfo(tokenInfo);
+
+      // Get Token Balance
+      // @ts-ignore
+      return await api.query.tokens.accounts.multi(addresses.map(address => [address, onChainInfo]), balances => {
+        const tokenBalance = {
+          reserved: sumBN(balances.map(b => b.reserved || new BN(0))),
+          frozen: sumBN(balances.map(b => b.frozen || new BN(0))),
+          free: sumBN(balances.map(b => b.free || new BN(0))) // free is actually total balance
+        };
+
+        // free balance = total balance - frozen misc
+        // locked balance = reserved + frozen misc
+        const freeBalance = tokenBalance.free.sub(tokenBalance.frozen);
+        const lockedBalance = tokenBalance.frozen.add(tokenBalance.reserved);
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          state: APIItemState.READY,
+          free: freeBalance.toString(),
+          locked: lockedBalance.toString(),
+          substrateInfo: {
+            reserved: tokenBalance.reserved.toString(),
+            miscFrozen: tokenBalance.frozen.toString()
+          }
+        });
+      });
+    } catch (err) {
+      console.warn(err);
+    }
+    return undefined;
+  }));
+  return () => {
+    unsubList.forEach(unsub => {
+      unsub && unsub();
+    });
+  };
+}
+async function subscribeAssetsAccountPallet(addresses, chain, api, callBack) {
+  const tokenMap = state.getAssetByChainAndAsset(chain, [_AssetType.LOCAL]);
+  const unsubList = await Promise.all(Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const assetIndex = _getTokenOnChainAssetId(tokenInfo);
+
+      // Get Token Balance
+      return await api.query.assets.account.multi(addresses.map(address => [assetIndex, address]), balances => {
+        let total = new BN(0);
+        let frozen = new BN(0);
+        balances.forEach(b => {
+          // @ts-ignore
+          // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-assignment
+          const bdata = b === null || b === void 0 ? void 0 : b.toJSON();
+          if (bdata) {
+            // @ts-ignore
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-argument
+            const addressBalance = new BN(String(bdata === null || bdata === void 0 ? void 0 : bdata.balance) || '0');
+
+            // @ts-ignore
+            if (bdata !== null && bdata !== void 0 && bdata.isFrozen) {
+              frozen = frozen.add(addressBalance);
+            } else {
+              total = total.add(addressBalance);
+            }
+          }
+        });
+        const free = total.sub(frozen);
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: frozen.toString(),
+          state: APIItemState.READY,
+          substrateInfo: {
+            miscFrozen: frozen.toString(),
+            reserved: '0'
+          }
+        });
+      });
+    } catch (err) {
+      console.warn(err);
+    }
+    return undefined;
+  }));
+  return () => {
+    unsubList.forEach(unsub => {
+      unsub && unsub();
+    });
+  };
+}
+export function subscribeEVMBalance(chain, addresses, evmApiMap, callback, tokenInfo) {
+  const balanceItem = {
+    tokenSlug: tokenInfo.slug,
+    state: APIItemState.PENDING,
+    free: '0',
+    locked: '0'
+  };
+  function getBalance() {
+    getEVMBalance(chain, addresses, evmApiMap).then(balances => {
+      balanceItem.free = sumBN(balances.map(b => new BN(b || '0'))).toString();
+      balanceItem.state = APIItemState.READY;
+      callback(balanceItem);
+    }).catch(console.warn);
+  }
+  getBalance();
+  const interval = setInterval(getBalance, ASTAR_REFRESH_BALANCE_INTERVAL);
+  const unsub2 = subscribeERC20Interval(addresses, chain, evmApiMap, callback);
+  return () => {
+    clearInterval(interval);
+    unsub2 && unsub2();
+  };
+}
+export async function getFreeBalance(chain, address, substrateApiMap, evmApiMap, tokenSlug) {
+  const substrateApi = await substrateApiMap[chain].isReady;
+  const api = substrateApi.api;
+  const web3Api = evmApiMap[chain];
+  const tokenInfo = tokenSlug ? state.getAssetBySlug(tokenSlug) : state.getNativeTokenInfo(chain);
+  const chainInfo = state.getChainInfo(chain);
+
+  // Only EVM Address use with EVM network
+  if (Boolean(web3Api || _isChainEvmCompatible(chainInfo)) !== isEthereumAddress(address)) {
+    if (!isEthereumAddress(address)) {
+      return '0';
+    }
+  }
+
+  // web3Api support mean isEthereum Network support
+  if (web3Api) {
+    if (_isNativeToken(tokenInfo)) {
+      var _web3Api$api;
+      return (await ((_web3Api$api = web3Api.api) === null || _web3Api$api === void 0 ? void 0 : _web3Api$api.eth.getBalance(address))) || '0';
+    } else {
+      if (_getContractAddressOfToken(tokenInfo).length > 0) {
+        return '0';
+      }
+      const contract = getERC20Contract(chain, _getContractAddressOfToken(tokenInfo), evmApiMap);
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      const free = await contract.methods.balanceOf(address).call();
+
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-return
+      return (free === null || free === void 0 ? void 0 : free.toString()) || '0';
+    }
+  } else {
+    var _balance$availableBal, _balance$availableBal2;
+    if (tokenSlug) {
+      if (_isSmartContractToken(tokenInfo)) {
+        if (_getContractAddressOfToken(tokenInfo).length > 0) {
+          return '0';
+        }
+        const contractPromise = getPSP22ContractPromise(api, _getContractAddressOfToken(tokenInfo));
+        const balanceOf = await contractPromise.query['psp22::balanceOf'](address, {
+          gasLimit: -1
+        }, address);
+        return balanceOf.output ? balanceOf.output.toString() : '0';
+      } else if (_BALANCE_CHAIN_GROUP.genshiro.includes(chain)) {
+        var _balance$asPositive;
+        const onChainInfo = _getTokenOnChainInfo(tokenInfo);
+        const balance = await api.query.eqBalances.account(address, onChainInfo);
+
+        // @ts-ignore
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-return
+        return ((_balance$asPositive = balance.asPositive) === null || _balance$asPositive === void 0 ? void 0 : _balance$asPositive.toString()) || '0';
+      } else if (_BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+        const balance = await api.query.system.account(address);
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+        const balancesData = JSON.parse(balance.data.toString());
+        const balanceList = balancesData.v0.balance;
+        let freeTokenBalance;
+        const assetId = _getTokenOnChainAssetId(tokenInfo);
+        if (!_isNativeToken(tokenInfo)) {
+          // @ts-ignore
+          freeTokenBalance = balanceList.find(data => data[0] === assetId);
+        } else {
+          freeTokenBalance = balanceList[0];
+        }
+        return freeTokenBalance ? freeTokenBalance[1].positive.toString() : '0';
+      } else if (_BALANCE_TOKEN_GROUP.crab.includes(tokenInfo.symbol)) {
+        var _balance$data5, _balance$data5$freeKt;
+        // @ts-ignore
+        const balance = await api.query.system.account(address);
+        return ((_balance$data5 = balance.data) === null || _balance$data5 === void 0 ? void 0 : (_balance$data5$freeKt = _balance$data5.freeKton) === null || _balance$data5$freeKt === void 0 ? void 0 : _balance$data5$freeKt.toString()) || '0';
+      } else if (!_isNativeToken(tokenInfo) && _BALANCE_CHAIN_GROUP.statemine.includes(chain)) {
+        var _balanceInfo$balance;
+        const assetId = _getTokenOnChainAssetId(tokenInfo);
+        const balanceInfo = (await api.query.assets.account(assetId, address)).toHuman();
+
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-return
+        return (balanceInfo === null || balanceInfo === void 0 ? void 0 : (_balanceInfo$balance = balanceInfo.balance) === null || _balanceInfo$balance === void 0 ? void 0 : _balanceInfo$balance.replaceAll(',', '')) || '0';
+      } else if (!_isNativeToken(tokenInfo) || _BALANCE_CHAIN_GROUP.kintsugi.includes(chain)) {
+        var _balance$free;
+        const onChainInfo = _getTokenOnChainInfo(tokenInfo);
+        // @ts-ignore
+        const balance = await api.query.tokens.accounts(address, onChainInfo);
+        return ((_balance$free = balance.free) === null || _balance$free === void 0 ? void 0 : _balance$free.toString()) || '0';
+      }
+    }
+    if (_BALANCE_CHAIN_GROUP.kusama.includes(chain)) {
+      var _balance$data6, _balance$data7;
+      // @ts-ignore
+      const _balance = await api.query.system.account(address);
+
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-assignment
+      const balance = _balance.toHuman();
+
+      // @ts-ignore
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      const freeBalance = new BN((_balance$data6 = balance.data) === null || _balance$data6 === void 0 ? void 0 : _balance$data6.free.replaceAll(',', ''));
+
+      // @ts-ignore
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      const miscFrozen = new BN((_balance$data7 = balance.data) === null || _balance$data7 === void 0 ? void 0 : _balance$data7.miscFrozen.replaceAll(',', ''));
+      const transferable = freeBalance.sub(miscFrozen);
+
+      // @ts-ignore
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-argument,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-return
+      return transferable.toString() || '0';
+    }
+    const balance = await api.derive.balances.all(address);
+    return ((_balance$availableBal = balance.availableBalance) === null || _balance$availableBal === void 0 ? void 0 : (_balance$availableBal2 = _balance$availableBal.toBn()) === null || _balance$availableBal2 === void 0 ? void 0 : _balance$availableBal2.toString()) || '0';
+  }
+}

package/koni/api/dotsama/crowdloan.d.ts

@@ -0,0 +1,7 @@
+import { _ChainInfo } from '@subwallet/chain-list/types';
+import { CrowdloanItem, CrowdloanParaState } from '@subwallet/extension-base/background/KoniTypes';
+import { _SubstrateApi } from '@subwallet/extension-base/services/chain-service/types';
+import { ApiPromise } from '@polkadot/api';
+export declare const subscribeAcalaContributeInterval: (polkadotAddresses: string[], paraState: CrowdloanParaState, callback: (rs: CrowdloanItem) => void) => () => void;
+export declare function getCrowdloanFundsStatus(api: ApiPromise): Promise<Record<string, CrowdloanParaState>>;
+export declare function subscribeCrowdloan(addresses: string[], substrateApiMap: Record<string, _SubstrateApi>, callback: (networkKey: string, rs: CrowdloanItem) => void, chainInfoMap: Record<string, _ChainInfo>): Promise<() => void>;