@subwallet/extension-base 1.1.39-0 → 1.1.39-beta.0

package/background/KoniTypes.d.ts

@@ -1645,6 +1645,7 @@ export interface KoniRequestSignatures {
     'pri(bonding.nominationPool.submitUnbonding)': [RequestStakePoolingUnbonding, SWTransactionResponse];
     'pri(chainService.subscribeChainInfoMap)': [null, Record<string, any>, Record<string, any>];
     'pri(chainService.subscribeChainStateMap)': [null, Record<string, any>, Record<string, any>];
+    'pri(chainService.subscribeChainStatusMap)': [null, Record<string, any>, Record<string, any>];
     'pri(chainService.subscribeAssetRegistry)': [null, Record<string, any>, Record<string, any>];
     'pri(chainService.subscribeMultiChainAssetMap)': [null, Record<string, _MultiChainAsset>, Record<string, _MultiChainAsset>];
     'pri(chainService.subscribeXcmRefMap)': [null, Record<string, _AssetRef>, Record<string, _AssetRef>];

package/cjs/koni/api/dotsama/balance.js

@@ -0,0 +1,464 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.subscribeBalance = subscribeBalance;
+exports.subscribeEVMBalance = subscribeEVMBalance;
+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 _balance = require("@subwallet/extension-base/koni/api/tokens/evm/balance");
+var _web = require("@subwallet/extension-base/koni/api/tokens/evm/web3");
+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 _constants2 = require("@subwallet/extension-base/services/chain-service/constants");
+var _utils2 = require("@subwallet/extension-base/services/chain-service/utils");
+var _utils3 = require("@subwallet/extension-base/utils");
+var _bignumber = _interopRequireDefault(require("bignumber.js"));
+var _util = require("@polkadot/util");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+// main subscription
+function subscribeBalance(addresses, chainInfoMap, substrateApiMap, evmApiMap, callback) {
+  const [substrateAddresses, evmAddresses] = (0, _utils3.categoryAddresses)(addresses);
+
+  // Looping over each chain
+  const unsubList = Object.entries(chainInfoMap).map(async _ref => {
+    let [chainSlug, chainInfo] = _ref;
+    const useAddresses = (0, _utils2._isChainEvmCompatible)(chainInfo) ? evmAddresses : substrateAddresses;
+    if ((0, _utils2._isPureEvmChain)(chainInfo)) {
+      const nativeTokenInfo = _handlers.state.getNativeTokenInfo(chainSlug);
+      return subscribeEVMBalance(chainSlug, useAddresses, evmApiMap, callback, nativeTokenInfo);
+    }
+    if (!useAddresses || useAddresses.length === 0 || _constants2._PURE_EVM_CHAINS.indexOf(chainSlug) > -1) {
+      const fungibleTokensByChain = _handlers.state.chainService.getFungibleTokensByChain(chainSlug, true);
+      const now = new Date().getTime();
+      Object.values(fungibleTokensByChain).map(token => {
+        return {
+          tokenSlug: token.slug,
+          free: '0',
+          locked: '0',
+          state: _KoniTypes.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);
+    });
+  };
+}
+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 subscribeEqBalanceAccountPallet(addresses, chain, networkAPI.api, callBack, true);
+    } else if (_constants2._BALANCE_CHAIN_GROUP.equilibrium_parachain.includes(chain)) {
+      unsubLocalToken = await 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 = 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);
+  const unsub = await networkAPI.query.system.account.multi(addresses, async balances => {
+    let [total, reserved, miscFrozen, feeFrozen] = [new _util.BN(0), new _util.BN(0), new _util.BN(0), new _util.BN(0)];
+    let pooledStakingBalance = _util.BN_ZERO;
+    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) {
+            const pooledBalance = new _util.BN(poolMemberData.points.toString());
+            pooledStakingBalance = pooledStakingBalance.add(pooledBalance);
+            Object.entries(poolMemberData.unbondingEras).forEach(_ref2 => {
+              let [, amount] = _ref2;
+              pooledStakingBalance = pooledStakingBalance.add(new _util.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 _util.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 _util.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 _util.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 _util.BN(0));
+    });
+    let locked = reserved.add(miscFrozen);
+    total = total.add(reserved); // total = free + reserved
+
+    if (pooledStakingBalance.gt(_util.BN_ZERO)) {
+      total = total.add(pooledStakingBalance);
+      locked = locked.add(pooledStakingBalance);
+    }
+    const free = total.sub(locked);
+    callBack({
+      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()
+      }
+    });
+  });
+  return () => {
+    unsub();
+  };
+}
+function subscribeERC20Interval(addresses, chain, evmApiMap, callBack) {
+  let tokenList = {};
+  const erc20ContractMap = {};
+  const getTokenBalances = () => {
+    Object.values(tokenList).map(async tokenInfo => {
+      let free = new _util.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 = (0, _utils3.sumBN)(balanceList.map(balance => new _util.BN(balance || 0)));
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: '0',
+          state: _KoniTypes.APIItemState.READY
+        });
+      } catch (err) {
+        console.log(tokenInfo.slug, err);
+      }
+    });
+  };
+  tokenList = _handlers.state.getAssetByChainAndAsset(chain, [_types._AssetType.ERC20]);
+  Object.entries(tokenList).forEach(_ref3 => {
+    let [slug, tokenInfo] = _ref3;
+    erc20ContractMap[slug] = (0, _web.getERC20Contract)(chain, (0, _utils2._getContractAddressOfToken)(tokenInfo), evmApiMap);
+  });
+  getTokenBalances();
+  const interval = setInterval(getTokenBalances, _constants.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 _util.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: (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 _balanceOf.output ? balanceObj.ok || balanceObj.Ok : '0';
+        }));
+        free = (0, _utils3.sumBN)(balances.map(bal => new _util.BN(bal || 0)));
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: '0',
+          state: _KoniTypes.APIItemState.READY
+        });
+      } catch (err) {
+        console.warn(tokenInfo.slug, err); // TODO: error createType
+      }
+    });
+  };
+
+  tokenList = _handlers.state.getAssetByChainAndAsset(chain, [_types._AssetType.PSP22]);
+  Object.entries(tokenList).forEach(_ref4 => {
+    let [slug, tokenInfo] = _ref4;
+    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 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);
+  const unsub = await api.query.system.account.multi(addresses, balances => {
+    // Equilibrium customizes the SystemAccount pallet
+    Object.values(tokenMap).forEach(tokenInfo => {
+      const assetId = (0, _utils2._getTokenOnChainAssetId)(tokenInfo);
+      let tokenFreeBalance = _util.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 _util.BN(new _bignumber.default(freeTokenBalance[1].positive).toString()) : _util.BN_ZERO;
+        tokenFreeBalance = tokenFreeBalance.add(bnFreeTokenBalance);
+      }
+      const tokenBalance = {
+        free: tokenFreeBalance.toString(),
+        locked: '0',
+        // Equilibrium doesn't show locked balance
+        state: _KoniTypes.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 ? [_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, _utils2._getTokenOnChainAssetId)(tokenInfo);
+      const unsub = await api.query.eqBalances.account.multi(addresses.map(address => [address, [assetId]]), balances => {
+        const tokenBalance = {
+          free: (0, _utils3.sumBN)(balances.map(b => b.asPositive)).toString(),
+          locked: '0',
+          // Equilibrium doesn't show locked balance
+          state: _KoniTypes.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 = [_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 tokenBalance = {
+          reserved: (0, _utils3.sumBN)(balances.map(b => b.reserved || new _util.BN(0))),
+          frozen: (0, _utils3.sumBN)(balances.map(b => b.frozen || new _util.BN(0))),
+          free: (0, _utils3.sumBN)(balances.map(b => b.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);
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          state: _KoniTypes.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 ? [_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 tokenBalance = {
+          reserved: (0, _utils3.sumBN)(balances.map(b => b.reserved || new _util.BN(0))),
+          frozen: (0, _utils3.sumBN)(balances.map(b => b.frozen || new _util.BN(0))),
+          free: (0, _utils3.sumBN)(balances.map(b => b.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);
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          state: _KoniTypes.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 = _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 => {
+        let total = new _util.BN(0);
+        let frozen = new _util.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 _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 = frozen.add(addressBalance);
+            } else {
+              total = total.add(addressBalance);
+            }
+          }
+        });
+        const free = total.sub(frozen);
+        callBack({
+          tokenSlug: tokenInfo.slug,
+          free: free.toString(),
+          locked: frozen.toString(),
+          state: _KoniTypes.APIItemState.READY,
+          substrateInfo: {
+            miscFrozen: frozen.toString(),
+            reserved: '0'
+          }
+        });
+      });
+    } catch (err) {
+      console.warn(err);
+    }
+    return undefined;
+  }));
+  return () => {
+    unsubList.forEach(unsub => {
+      unsub && unsub();
+    });
+  };
+}
+function subscribeEVMBalance(chain, addresses, evmApiMap, callback, tokenInfo) {
+  const balanceItem = {
+    tokenSlug: tokenInfo.slug,
+    state: _KoniTypes.APIItemState.PENDING,
+    free: '0',
+    locked: '0'
+  };
+  function getBalance() {
+    (0, _balance.getEVMBalance)(chain, addresses, evmApiMap).then(balances => {
+      balanceItem.free = (0, _utils3.sumBN)(balances.map(b => new _util.BN(b || '0'))).toString();
+      balanceItem.state = _KoniTypes.APIItemState.READY;
+      callback(balanceItem);
+    }).catch(console.warn);
+  }
+  getBalance();
+  const interval = setInterval(getBalance, _constants.ASTAR_REFRESH_BALANCE_INTERVAL);
+  const unsub2 = subscribeERC20Interval(addresses, chain, evmApiMap, callback);
+  return () => {
+    clearInterval(interval);
+    unsub2 && unsub2();
+  };
+}

package/cjs/koni/background/handlers/Extension.js

@@ -3255,6 +3255,19 @@ class KoniExtension {
     });
     return this.#koniState.getChainStateMap();
   }
+  subscribeChainStatusMap(id, port) {
+    const cb = (0, _subscriptions.createSubscription)(id, port);
+    const chainStateMapSubscription = this.#koniState.chainService.subscribeChainStatusMap().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, chainStateMapSubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.chainService.getChainStatusMap();
+  }
   async subscribeAssetRegistry(id, port) {
     const cb = (0, _subscriptions.createSubscription)(id, port);
     let ready = false;
@@ -3558,7 +3571,7 @@ class KoniExtension {
       });
       const {
         connectionStatus
-      } = this.#koniState.chainService.getChainStateByKey(_constants2._DEFAULT_MANTA_ZK_CHAIN);
+      } = this.#koniState.chainService.getChainStatusByKey(_constants2._DEFAULT_MANTA_ZK_CHAIN);
       if (connectionStatus !== _types2._ChainConnectionStatus.CONNECTED) {
         // TODO: do better
         await timeout();
@@ -4298,6 +4311,8 @@ class KoniExtension {
         return this.subscribeChainInfoMap(id, port);
       case 'pri(chainService.subscribeChainStateMap)':
         return this.subscribeChainStateMap(id, port);
+      case 'pri(chainService.subscribeChainStatusMap)':
+        return this.subscribeChainStatusMap(id, port);
       case 'pri(chainService.subscribeXcmRefMap)':
         return this.subscribeXcmRefMap(id, port);
       case 'pri(chainService.getSupportedContractTypes)':