@subwallet/extension-base 1.1.23-0 → 1.1.24-1

package/koni/api/dotsama/balance.js

@@ -1,451 +0,0 @@
-// 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 { getDefaultWeightV2 } from '@subwallet/extension-base/koni/api/tokens/wasm/utils';
-import { state } from '@subwallet/extension-base/koni/background/handlers';
-import { _BALANCE_CHAIN_GROUP, _MANTA_ZK_CHAIN_GROUP, _PURE_EVM_CHAINS, _ZK_ASSET_PREFIX } from '@subwallet/extension-base/services/chain-service/constants';
-import { _checkSmartContractSupportByChain, _getChainNativeTokenSlug, _getContractAddressOfToken, _getTokenOnChainAssetId, _getTokenOnChainInfo, _isChainEvmCompatible, _isPureEvmChain, _isSubstrateRelayChain } from '@subwallet/extension-base/services/chain-service/utils';
-import { categoryAddresses, sumBN } from '@subwallet/extension-base/utils';
-import BigN from 'bignumber.js';
-import { BN, BN_ZERO } from '@polkadot/util';
-// 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);
-    } else if (_BALANCE_CHAIN_GROUP.centrifuge.includes(chain)) {
-      unsubLocalToken = await subscribeOrmlTokensPallet(addresses, chain, networkAPI.api, callBack);
-    }
-    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) {
-      var _networkAPI$query$nom;
-      const poolMemberDatas = await ((_networkAPI$query$nom = networkAPI.query.nominationPools.poolMembers) === null || _networkAPI$query$nom === void 0 ? void 0 : _networkAPI$query$nom.multi(addresses));
-      if (poolMemberDatas) {
-        for (const _poolMemberData of poolMemberDatas) {
-          const poolMemberData = _poolMemberData.toPrimitive();
-          if (poolMemberData) {
-            const pooledBalance = new BN(poolMemberData.points.toString());
-            pooledStakingBalance = pooledStakingBalance.add(pooledBalance);
-            Object.entries(poolMemberData.unbondingEras).forEach(([, amount]) => {
-              pooledStakingBalance = pooledStakingBalance.add(new BN(amount));
-            });
-          }
-        }
-      }
-    }
-    balances.forEach(balance => {
-      var _balance$data, _balance$data$free, _balance$data2, _balance$data2$reserv, _balance$data3, _balance$data3$miscFr, _balance$data4, _balance$data4$frozen, _balance$data5, _balance$data5$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));
-      // @ts-ignore
-      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
-      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()) || (balance === null || balance === void 0 ? void 0 : (_balance$data4 = balance.data) === null || _balance$data4 === void 0 ? void 0 : (_balance$data4$frozen = _balance$data4.frozen) === null || _balance$data4$frozen === void 0 ? void 0 : _balance$data4$frozen.toBn()) || new BN(0)); // TODO: update frozen
-      feeFrozen = feeFrozen.add(((_balance$data5 = balance.data) === null || _balance$data5 === void 0 ? void 0 : (_balance$data5$feeFro = _balance$data5.feeFrozen) === null || _balance$data5$feeFro === void 0 ? void 0 : _balance$data5$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.gte(BN_ZERO) ? free.toString() : '0',
-      locked: locked.toString(),
-      state: APIItemState.READY,
-      substrateInfo: {
-        miscFrozen: miscFrozen.toString(),
-        reserved: reserved.toString(),
-        feeFrozen: feeFrozen.toString()
-      }
-    });
-  });
-  return () => {
-    unsub();
-  };
-}
-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(tokenInfo.slug, 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 => {
-          var _balanceOf$output;
-          const _balanceOf = await contract.query['psp22::balanceOf'](address, {
-            gasLimit: getDefaultWeightV2(api)
-          }, address);
-          const balanceObj = _balanceOf === null || _balanceOf === void 0 ? void 0 : (_balanceOf$output = _balanceOf.output) === null || _balanceOf$output === void 0 ? void 0 : _balanceOf$output.toPrimitive();
-          return _balanceOf.output ? balanceObj.ok || balanceObj.Ok : '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.warn(tokenInfo.slug, err); // TODO: error createType
-      }
-    });
-  };
-
-  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(new BigN(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);
-    });
-  };
-}
-
-// eslint-disable-next-line @typescript-eslint/require-await
-async function subscribeOrmlTokensPallet(addresses, chain, api, callBack) {
-  const tokenTypes = [_AssetType.LOCAL];
-  const tokenMap = state.getAssetByChainAndAsset(chain, tokenTypes);
-  const unsubList = Object.values(tokenMap).map(async tokenInfo => {
-    try {
-      const onChainInfo = _getTokenOnChainInfo(tokenInfo);
-
-      // Get Token Balance
-      // @ts-ignore
-      const unsub = await api.query.ormlTokens.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()
-          }
-        });
-      });
-      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);
-      const assetId = _getTokenOnChainAssetId(tokenInfo);
-
-      // Get Token Balance
-      // @ts-ignore
-      return await api.query.tokens.accounts.multi(addresses.map(address => [address, onChainInfo || assetId]), 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]);
-  Object.values(tokenMap).forEach(token => {
-    if (_MANTA_ZK_CHAIN_GROUP.includes(token.originChain) && token.symbol.startsWith(_ZK_ASSET_PREFIX)) {
-      delete tokenMap[token.slug];
-    }
-  });
-  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.toHuman();
-          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).replaceAll(',', '') || '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();
-  };
-}

package/utils/address.d.ts

@@ -1,5 +0,0 @@
-import { AddressJson } from '@subwallet/extension-base/background/types';
-import { SubjectInfo } from '@subwallet/ui-keyring/observable/types';
-export declare const simpleAddress: (address: string) => string;
-export declare function quickFormatAddressToCompare(address?: string): string | undefined;
-export declare const convertSubjectInfoToAddresses: (subjectInfo: SubjectInfo) => AddressJson[];