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

package/cjs/services/balance-service/helpers/subscribe/balance.js

@@ -0,0 +1,111 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.subscribeBalance = subscribeBalance;
+var _types = require("@subwallet/chain-list/types");
+var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
+var _handlers = require("@subwallet/extension-base/koni/background/handlers");
+var _utils = require("@subwallet/extension-base/services/chain-service/utils");
+var _utils2 = require("@subwallet/extension-base/utils");
+var _account = require("@subwallet/extension-base/utils/account");
+var _evm = require("./evm");
+var _substrate = require("./substrate");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+const filterAddress = (addresses, chainInfo) => {
+  const isEvmChain = (0, _utils._isChainEvmCompatible)(chainInfo);
+  const [substrateAddresses, evmAddresses] = (0, _utils2.categoryAddresses)(addresses);
+  if (isEvmChain) {
+    return [evmAddresses, []];
+  } else {
+    const fetchList = [];
+    const unfetchList = [];
+    substrateAddresses.forEach(address => {
+      const account = (0, _account.getAccountJsonByAddress)(address);
+      if (account) {
+        if (account.isHardware) {
+          const availGen = account.availableGenesisHashes || [];
+          const gen = (0, _utils._getSubstrateGenesisHash)(chainInfo);
+          if (availGen.includes(gen)) {
+            fetchList.push(address);
+          } else {
+            unfetchList.push(address);
+          }
+        } else {
+          fetchList.push(address);
+        }
+      } else {
+        fetchList.push(address);
+      }
+    });
+    return [fetchList, [...unfetchList, ...evmAddresses]];
+  }
+};
+
+// main subscription
+function subscribeBalance(addresses, chainInfoMap, substrateApiMap, evmApiMap, callback) {
+  // Looping over each chain
+  const unsubList = Object.entries(chainInfoMap).map(async _ref => {
+    let [chainSlug, chainInfo] = _ref;
+    const [useAddresses, notSupportAddresses] = filterAddress(addresses, chainInfo);
+    if (notSupportAddresses.length) {
+      const tokens = _handlers.state.chainService.getAssetByChainAndType(chainSlug, [_types._AssetType.NATIVE, _types._AssetType.ERC20, _types._AssetType.PSP22, _types._AssetType.LOCAL]);
+      const assetSetting = await _handlers.state.chainService.getAssetSettings();
+      const filtered = Object.values(tokens).filter(_ref2 => {
+        var _assetSetting$slug;
+        let {
+          slug
+        } = _ref2;
+        return (_assetSetting$slug = assetSetting[slug]) === null || _assetSetting$slug === void 0 ? void 0 : _assetSetting$slug.visible;
+      });
+      const now = new Date().getTime();
+      notSupportAddresses.forEach(address => {
+        const items = filtered.map(token => {
+          return {
+            address,
+            tokenSlug: token.slug,
+            free: '0',
+            locked: '0',
+            state: _KoniTypes.APIItemState.NOT_SUPPORT,
+            timestamp: now
+          };
+        });
+        callback(items);
+      });
+    }
+    if ((0, _utils._isPureEvmChain)(chainInfo)) {
+      const nativeTokenInfo = _handlers.state.getNativeTokenInfo(chainSlug);
+      return (0, _evm.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
+    //     } as BalanceItem;
+    //   }).forEach(callback);
+    //
+    //   return undefined;
+    // }
+
+    const networkAPI = await substrateApiMap[chainSlug].isReady;
+    return (0, _substrate.subscribeSubstrateBalance)(useAddresses, chainInfo, chainSlug, networkAPI, evmApiMap, callback);
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}

package/cjs/services/balance-service/helpers/subscribe/evm.js

@@ -0,0 +1,95 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.subscribeERC20Interval = subscribeERC20Interval;
+exports.subscribeEVMBalance = subscribeEVMBalance;
+var _types = require("@subwallet/chain-list/types");
+var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
+var _constants = require("@subwallet/extension-base/constants");
+var _balance = require("@subwallet/extension-base/koni/api/tokens/evm/balance");
+var _web = require("@subwallet/extension-base/koni/api/tokens/evm/web3");
+var _handlers = require("@subwallet/extension-base/koni/background/handlers");
+var _utils = require("@subwallet/extension-base/services/chain-service/utils");
+var _util = require("@polkadot/util");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+function subscribeERC20Interval(addresses, chain, evmApiMap, callBack) {
+  let tokenList = {};
+  const erc20ContractMap = {};
+  const getTokenBalances = () => {
+    Object.values(tokenList).map(async tokenInfo => {
+      try {
+        const contract = erc20ContractMap[tokenInfo.slug];
+        const balances = await Promise.all(addresses.map(async address => {
+          try {
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-return,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+            return await contract.methods.balanceOf(address).call();
+          } catch (e) {
+            console.error(`Error on get balance of account ${address} for token ${tokenInfo.slug}`, e);
+            return '0';
+          }
+        }));
+        const items = balances.map((balance, index) => {
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            free: new _util.BN(balance || 0).toString(),
+            locked: '0',
+            state: _KoniTypes.APIItemState.READY
+          };
+        });
+        callBack(items);
+      } catch (err) {
+        console.log(tokenInfo.slug, err);
+      }
+    });
+  };
+  tokenList = _handlers.state.getAssetByChainAndAsset(chain, [_types._AssetType.ERC20]);
+  Object.entries(tokenList).forEach(_ref => {
+    let [slug, tokenInfo] = _ref;
+    erc20ContractMap[slug] = (0, _web.getERC20Contract)(chain, (0, _utils._getContractAddressOfToken)(tokenInfo), evmApiMap);
+  });
+  getTokenBalances();
+  const interval = setInterval(getTokenBalances, _constants.SUB_TOKEN_REFRESH_BALANCE_INTERVAL);
+  return () => {
+    clearInterval(interval);
+  };
+}
+function subscribeEVMBalance(chain, addresses, evmApiMap, callback, tokenInfo) {
+  function getBalance() {
+    (0, _balance.getEVMBalance)(chain, addresses, evmApiMap).then(balances => {
+      return balances.map((balance, index) => {
+        return {
+          address: addresses[index],
+          tokenSlug: tokenInfo.slug,
+          state: _KoniTypes.APIItemState.READY,
+          free: new _util.BN(balance || '0').toString(),
+          locked: '0'
+        };
+      });
+    }).catch(e => {
+      console.error(`Error on get native balance with token ${tokenInfo.slug}`, e);
+      return addresses.map(address => {
+        return {
+          address: address,
+          tokenSlug: tokenInfo.slug,
+          state: _KoniTypes.APIItemState.READY,
+          free: '0',
+          locked: '0'
+        };
+      });
+    }).then(items => {
+      callback(items);
+    }).catch(console.error);
+  }
+  getBalance();
+  const interval = setInterval(getBalance, _constants.ASTAR_REFRESH_BALANCE_INTERVAL);
+  const unsub2 = subscribeERC20Interval(addresses, chain, evmApiMap, callback);
+  return () => {
+    clearInterval(interval);
+    unsub2 && unsub2();
+  };
+}

package/cjs/services/balance-service/helpers/subscribe/substrate/equilibrium.js

@@ -0,0 +1,113 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.subscribeEqBalanceAccountPallet = subscribeEqBalanceAccountPallet;
+exports.subscribeEquilibriumTokenBalance = subscribeEquilibriumTokenBalance;
+var _types = require("@subwallet/chain-list/types");
+var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
+var _handlers = require("@subwallet/extension-base/koni/background/handlers");
+var _utils = require("@subwallet/extension-base/services/chain-service/utils");
+var _bignumber = _interopRequireDefault(require("bignumber.js"));
+var _util = require("@polkadot/util");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+async function subscribeEquilibriumTokenBalance(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_types._AssetType.NATIVE, _types._AssetType.LOCAL] : [_types._AssetType.LOCAL];
+  const tokenMap = _handlers.state.getAssetByChainAndAsset(chain, tokenTypes);
+  try {
+    const unsub = await api.query.system.account.multi(addresses, balances => {
+      // Equilibrium customizes the SystemAccount pallet
+      Object.values(tokenMap).forEach(tokenInfo => {
+        const assetId = (0, _utils._getTokenOnChainAssetId)(tokenInfo);
+        const items = balances.map((balance, index) => {
+          // 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 _util.BN(new _bignumber.default(freeTokenBalance[1].positive).toString()) : _util.BN_ZERO;
+          return {
+            address: addresses[index],
+            free: bnFreeTokenBalance.toString(),
+            locked: '0',
+            // Equilibrium doesn't show locked balance
+            state: _KoniTypes.APIItemState.READY,
+            tokenSlug: tokenInfo.slug
+          };
+        });
+        callBack(items);
+      });
+    });
+    return () => {
+      unsub();
+    };
+  } catch (e) {
+    Object.values(tokenMap).forEach(tokenInfo => {
+      const items = addresses.map(address => {
+        return {
+          address: address,
+          free: '0',
+          locked: '0',
+          // Equilibrium doesn't show locked balance
+          state: _KoniTypes.APIItemState.READY,
+          tokenSlug: tokenInfo.slug
+        };
+      });
+      callBack(items);
+    });
+    return () => {
+      // Empty
+    };
+  }
+}
+
+// eslint-disable-next-line @typescript-eslint/require-await
+async function subscribeEqBalanceAccountPallet(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_types._AssetType.NATIVE, _types._AssetType.LOCAL] : [_types._AssetType.LOCAL];
+  const tokenMap = _handlers.state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsubList = Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const assetId = (0, _utils._getTokenOnChainAssetId)(tokenInfo);
+      const unsub = await api.query.eqBalances.account.multi(addresses.map(address => [address, [assetId]]), balances => {
+        const items = balances.map((balance, index) => {
+          return {
+            address: addresses[index],
+            free: balance.asPositive.toString(),
+            locked: '0',
+            // Equilibrium doesn't show locked balance
+            state: _KoniTypes.APIItemState.READY,
+            tokenSlug: tokenInfo.slug
+          };
+        });
+        callBack(items);
+      });
+      return unsub;
+    } catch (err) {
+      console.warn(err);
+      const items = addresses.map(address => {
+        return {
+          address: address,
+          free: '0',
+          locked: '0',
+          // Equilibrium doesn't show locked balance
+          state: _KoniTypes.APIItemState.READY,
+          tokenSlug: tokenInfo.slug
+        };
+      });
+      callBack(items);
+      return undefined;
+    }
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}

package/cjs/services/balance-service/helpers/subscribe/substrate/index.js

@@ -0,0 +1,324 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.subscribeSubstrateBalance = subscribeSubstrateBalance;
+var _types = require("@subwallet/chain-list/types");
+var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
+var _constants = require("@subwallet/extension-base/constants");
+var _wasm = require("@subwallet/extension-base/koni/api/tokens/wasm");
+var _utils = require("@subwallet/extension-base/koni/api/tokens/wasm/utils");
+var _handlers = require("@subwallet/extension-base/koni/background/handlers");
+var _evm = require("@subwallet/extension-base/services/balance-service/helpers/subscribe/evm");
+var _constants2 = require("@subwallet/extension-base/services/chain-service/constants");
+var _utils2 = require("@subwallet/extension-base/services/chain-service/utils");
+var _util = require("@polkadot/util");
+var _equilibrium = require("./equilibrium");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+async function subscribeSubstrateBalance(addresses, chainInfo, chain, networkAPI, evmApiMap, callBack) {
+  let unsubNativeToken;
+  if (!_constants2._BALANCE_CHAIN_GROUP.kintsugi.includes(chain) && !_constants2._BALANCE_CHAIN_GROUP.genshiro.includes(chain) && !_constants2._BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+    unsubNativeToken = await subscribeWithSystemAccountPallet(addresses, chainInfo, networkAPI.api, callBack);
+  }
+  let unsubLocalToken;
+  let unsubEvmContractToken;
+  let unsubWasmContractToken;
+  try {
+    if (_constants2._BALANCE_CHAIN_GROUP.bifrost.includes(chain)) {
+      unsubLocalToken = await subscribeTokensAccountsPallet(addresses, chain, networkAPI.api, callBack);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.kintsugi.includes(chain)) {
+      unsubLocalToken = await subscribeTokensAccountsPallet(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.statemine.includes(chain)) {
+      unsubLocalToken = await subscribeAssetsAccountPallet(addresses, chain, networkAPI.api, callBack);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.genshiro.includes(chain)) {
+      unsubLocalToken = await (0, _equilibrium.subscribeEqBalanceAccountPallet)(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+      unsubLocalToken = await (0, _equilibrium.subscribeEquilibriumTokenBalance)(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.centrifuge.includes(chain)) {
+      unsubLocalToken = await subscribeOrmlTokensPallet(addresses, chain, networkAPI.api, callBack);
+    }
+    if ((0, _utils2._isChainEvmCompatible)(chainInfo)) {
+      unsubEvmContractToken = (0, _evm.subscribeERC20Interval)(addresses, chain, evmApiMap, callBack);
+    }
+    if ((0, _utils2._checkSmartContractSupportByChain)(chainInfo, _types._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 = (0, _utils2._getChainNativeTokenSlug)(chainInfo);
+
+  // TODO: Need handle case error
+  const unsub = await networkAPI.query.system.account.multi(addresses, async balances => {
+    const pooledStakingBalances = [];
+    if ((0, _utils2._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) {
+            let pooled = new _util.BN(poolMemberData.points.toString());
+            Object.entries(poolMemberData.unbondingEras).forEach(_ref => {
+              let [, amount] = _ref;
+              pooled = pooled.add(new _util.BN(amount));
+            });
+            pooledStakingBalances.push(pooled);
+          } else {
+            pooledStakingBalances.push(_util.BN_ZERO);
+          }
+        }
+      }
+    }
+    const items = balances.map((balance, index) => {
+      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;
+      let total = ((_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 _util.BN(0);
+      const reserved = ((_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 _util.BN(0);
+      // @ts-ignore
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      const miscFrozen = ((_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 _util.BN(0);
+      const feeFrozen = ((_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 _util.BN(0);
+      let locked = reserved.add(miscFrozen);
+      total = total.add(reserved);
+      const pooledStakingBalance = pooledStakingBalances[index] || _util.BN_ZERO;
+      if (pooledStakingBalance.gt(_util.BN_ZERO)) {
+        total = total.add(pooledStakingBalance);
+        locked = locked.add(pooledStakingBalance);
+      }
+      const free = total.sub(locked);
+      return {
+        address: addresses[index],
+        tokenSlug: chainNativeTokenSlug,
+        free: free.gte(_util.BN_ZERO) ? free.toString() : '0',
+        locked: locked.toString(),
+        state: _KoniTypes.APIItemState.READY,
+        substrateInfo: {
+          miscFrozen: miscFrozen.toString(),
+          reserved: reserved.toString(),
+          feeFrozen: feeFrozen.toString()
+        }
+      };
+    });
+    callBack(items);
+  });
+  return () => {
+    unsub();
+  };
+}
+function subscribePSP22Balance(addresses, chain, api, callBack) {
+  let tokenList = {};
+  const psp22ContractMap = {};
+  const getTokenBalances = () => {
+    Object.values(tokenList).map(async tokenInfo => {
+      try {
+        const contract = psp22ContractMap[tokenInfo.slug];
+        const balances = await Promise.all(addresses.map(async address => {
+          try {
+            var _balanceOf$output, _ref2;
+            const _balanceOf = await contract.query['psp22::balanceOf'](address, {
+              gasLimit: (0, _utils.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 {
+              address: address,
+              tokenSlug: tokenInfo.slug,
+              free: _balanceOf.output ? (_ref2 = balanceObj.ok) !== null && _ref2 !== void 0 ? _ref2 : balanceObj.Ok : '0',
+              locked: '0',
+              state: _KoniTypes.APIItemState.READY
+            };
+          } catch (err) {
+            console.error(`Error on get balance of account ${address} for token ${tokenInfo.slug}`, err);
+            return {
+              address: address,
+              tokenSlug: tokenInfo.slug,
+              free: '0',
+              locked: '0',
+              state: _KoniTypes.APIItemState.READY
+            };
+          }
+        }));
+        callBack(balances);
+      } catch (err) {
+        console.warn(tokenInfo.slug, err); // TODO: error createType
+      }
+    });
+  };
+
+  tokenList = _handlers.state.getAssetByChainAndAsset(chain, [_types._AssetType.PSP22]);
+  Object.entries(tokenList).forEach(_ref3 => {
+    let [slug, tokenInfo] = _ref3;
+    psp22ContractMap[slug] = (0, _wasm.getPSP22ContractPromise)(api, (0, _utils2._getContractAddressOfToken)(tokenInfo));
+  });
+  getTokenBalances();
+  const interval = setInterval(getTokenBalances, _constants.SUB_TOKEN_REFRESH_BALANCE_INTERVAL);
+  return () => {
+    clearInterval(interval);
+  };
+}
+async function subscribeTokensAccountsPallet(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_types._AssetType.NATIVE, _types._AssetType.LOCAL] : [_types._AssetType.LOCAL];
+  const tokenMap = _handlers.state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsubList = await Promise.all(Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const onChainInfo = (0, _utils2._getTokenOnChainInfo)(tokenInfo);
+      const assetId = (0, _utils2._getTokenOnChainAssetId)(tokenInfo);
+
+      // Get Token Balance
+      // @ts-ignore
+      return await api.query.tokens.accounts.multi(addresses.map(address => [address, onChainInfo || assetId]), balances => {
+        const items = balances.map((balance, index) => {
+          const tokenBalance = {
+            reserved: balance.reserved || new _util.BN(0),
+            frozen: balance.frozen || new _util.BN(0),
+            free: balance.free || new _util.BN(0) // free is actually total balance
+          };
+
+          const freeBalance = tokenBalance.free.sub(tokenBalance.frozen);
+          const lockedBalance = tokenBalance.frozen.add(tokenBalance.reserved);
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            state: _KoniTypes.APIItemState.READY,
+            free: freeBalance.toString(),
+            locked: lockedBalance.toString(),
+            substrateInfo: {
+              reserved: tokenBalance.reserved.toString(),
+              miscFrozen: tokenBalance.frozen.toString()
+            }
+          };
+        });
+        callBack(items);
+      });
+    } catch (err) {
+      console.warn(err);
+    }
+    return undefined;
+  }));
+  return () => {
+    unsubList.forEach(unsub => {
+      unsub && unsub();
+    });
+  };
+}
+async function subscribeAssetsAccountPallet(addresses, chain, api, callBack) {
+  const tokenMap = _handlers.state.getAssetByChainAndAsset(chain, [_types._AssetType.LOCAL]);
+  Object.values(tokenMap).forEach(token => {
+    if (_constants2._MANTA_ZK_CHAIN_GROUP.includes(token.originChain) && token.symbol.startsWith(_constants2._ZK_ASSET_PREFIX)) {
+      delete tokenMap[token.slug];
+    }
+  });
+  const unsubList = await Promise.all(Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const assetIndex = (0, _utils2._getTokenOnChainAssetId)(tokenInfo);
+
+      // Get Token Balance
+      return await api.query.assets.account.multi(addresses.map(address => [assetIndex, address]), balances => {
+        const items = balances.map((balance, index) => {
+          // @ts-ignore
+          // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-assignment
+          const bdata = balance === null || balance === void 0 ? void 0 : balance.toHuman();
+          let frozen = _util.BN_ZERO;
+          let total = _util.BN_ZERO;
+          if (bdata) {
+            // @ts-ignore
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-argument
+            const addressBalance = new _util.BN(String(bdata === null || bdata === void 0 ? void 0 : bdata.balance).replaceAll(',', '') || '0');
+
+            // @ts-ignore
+            if (bdata !== null && bdata !== void 0 && bdata.isFrozen) {
+              frozen = addressBalance;
+            } else {
+              total = addressBalance;
+            }
+          }
+          const free = total.sub(frozen);
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            free: free.toString(),
+            locked: frozen.toString(),
+            state: _KoniTypes.APIItemState.READY,
+            substrateInfo: {
+              miscFrozen: frozen.toString(),
+              reserved: '0'
+            }
+          };
+        });
+        callBack(items);
+      });
+    } catch (err) {
+      console.warn(err);
+    }
+    return undefined;
+  }));
+  return () => {
+    unsubList.forEach(unsub => {
+      unsub && unsub();
+    });
+  };
+}
+
+// eslint-disable-next-line @typescript-eslint/require-await
+async function subscribeOrmlTokensPallet(addresses, chain, api, callBack) {
+  const tokenTypes = [_types._AssetType.LOCAL];
+  const tokenMap = _handlers.state.getAssetByChainAndAsset(chain, tokenTypes);
+  const unsubList = Object.values(tokenMap).map(async tokenInfo => {
+    try {
+      const onChainInfo = (0, _utils2._getTokenOnChainInfo)(tokenInfo);
+
+      // Get Token Balance
+      // @ts-ignore
+      const unsub = await api.query.ormlTokens.accounts.multi(addresses.map(address => [address, onChainInfo]), balances => {
+        const items = balances.map((balance, index) => {
+          const tokenBalance = {
+            reserved: balance.reserved || new _util.BN(0),
+            frozen: balance.frozen || new _util.BN(0),
+            free: balance.free || new _util.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);
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            state: _KoniTypes.APIItemState.READY,
+            free: freeBalance.toString(),
+            locked: lockedBalance.toString(),
+            substrateInfo: {
+              reserved: tokenBalance.reserved.toString(),
+              miscFrozen: tokenBalance.frozen.toString()
+            }
+          };
+        });
+        callBack(items);
+      });
+      return unsub;
+    } catch (err) {
+      console.warn(err);
+      return undefined;
+    }
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}