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

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

@@ -0,0 +1,87 @@
+// 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 { state } from '@subwallet/extension-base/koni/background/handlers';
+import { _getContractAddressOfToken } from '@subwallet/extension-base/services/chain-service/utils';
+import { BN } from '@polkadot/util';
+export 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 BN(balance || 0).toString(),
+            locked: '0',
+            state: APIItemState.READY
+          };
+        });
+        callBack(items);
+      } 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);
+  };
+}
+export function subscribeEVMBalance(chain, addresses, evmApiMap, callback, tokenInfo) {
+  function getBalance() {
+    getEVMBalance(chain, addresses, evmApiMap).then(balances => {
+      return balances.map((balance, index) => {
+        return {
+          address: addresses[index],
+          tokenSlug: tokenInfo.slug,
+          state: APIItemState.READY,
+          free: new 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: APIItemState.READY,
+          free: '0',
+          locked: '0'
+        };
+      });
+    }).then(items => {
+      callback(items);
+    }).catch(console.error);
+  }
+  getBalance();
+  const interval = setInterval(getBalance, ASTAR_REFRESH_BALANCE_INTERVAL);
+  const unsub2 = subscribeERC20Interval(addresses, chain, evmApiMap, callback);
+  return () => {
+    clearInterval(interval);
+    unsub2 && unsub2();
+  };
+}

package/services/balance-service/helpers/subscribe/substrate/equilibrium.d.ts

@@ -0,0 +1,4 @@
+import { BalanceItem } from '@subwallet/extension-base/types';
+import { ApiPromise } from '@polkadot/api';
+export declare function subscribeEquilibriumTokenBalance(addresses: string[], chain: string, api: ApiPromise, callBack: (rs: BalanceItem[]) => void, includeNativeToken?: boolean): Promise<() => void>;
+export declare function subscribeEqBalanceAccountPallet(addresses: string[], chain: string, api: ApiPromise, callBack: (rs: BalanceItem[]) => void, includeNativeToken?: boolean): Promise<() => void>;

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

@@ -0,0 +1,105 @@
+// 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 { state } from '@subwallet/extension-base/koni/background/handlers';
+import { _getTokenOnChainAssetId } from '@subwallet/extension-base/services/chain-service/utils';
+import BigN from 'bignumber.js';
+import { BN, BN_ZERO } from '@polkadot/util';
+export async function subscribeEquilibriumTokenBalance(addresses, chain, api, callBack, includeNativeToken) {
+  const tokenTypes = includeNativeToken ? [_AssetType.NATIVE, _AssetType.LOCAL] : [_AssetType.LOCAL];
+  const tokenMap = 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 = _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 BN(new BigN(freeTokenBalance[1].positive).toString()) : BN_ZERO;
+          return {
+            address: addresses[index],
+            free: bnFreeTokenBalance.toString(),
+            locked: '0',
+            // Equilibrium doesn't show locked balance
+            state: 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: APIItemState.READY,
+          tokenSlug: tokenInfo.slug
+        };
+      });
+      callBack(items);
+    });
+    return () => {
+      // Empty
+    };
+  }
+}
+
+// eslint-disable-next-line @typescript-eslint/require-await
+export 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 items = balances.map((balance, index) => {
+          return {
+            address: addresses[index],
+            free: balance.asPositive.toString(),
+            locked: '0',
+            // Equilibrium doesn't show locked balance
+            state: 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: APIItemState.READY,
+          tokenSlug: tokenInfo.slug
+        };
+      });
+      callBack(items);
+      return undefined;
+    }
+  });
+  return () => {
+    unsubList.forEach(subProm => {
+      subProm.then(unsub => {
+        unsub && unsub();
+      }).catch(console.error);
+    });
+  };
+}

package/services/balance-service/helpers/subscribe/substrate/index.d.ts

@@ -0,0 +1,4 @@
+import { _ChainInfo } from '@subwallet/chain-list/types';
+import { _EvmApi, _SubstrateApi } from '@subwallet/extension-base/services/chain-service/types';
+import { BalanceItem } from '@subwallet/extension-base/types';
+export declare function subscribeSubstrateBalance(addresses: string[], chainInfo: _ChainInfo, chain: string, networkAPI: _SubstrateApi, evmApiMap: Record<string, _EvmApi>, callBack: (rs: BalanceItem[]) => void): Promise<() => void>;

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

@@ -0,0 +1,316 @@
+// 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 { SUB_TOKEN_REFRESH_BALANCE_INTERVAL } from '@subwallet/extension-base/constants';
+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 { subscribeERC20Interval } from '@subwallet/extension-base/services/balance-service/helpers/subscribe/evm';
+import { _BALANCE_CHAIN_GROUP, _MANTA_ZK_CHAIN_GROUP, _ZK_ASSET_PREFIX } from '@subwallet/extension-base/services/chain-service/constants';
+import { _checkSmartContractSupportByChain, _getChainNativeTokenSlug, _getContractAddressOfToken, _getTokenOnChainAssetId, _getTokenOnChainInfo, _isChainEvmCompatible, _isSubstrateRelayChain } from '@subwallet/extension-base/services/chain-service/utils';
+import { BN, BN_ZERO } from '@polkadot/util';
+import { subscribeEqBalanceAccountPallet, subscribeEquilibriumTokenBalance } from "./equilibrium.js";
+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);
+
+  // TODO: Need handle case error
+  const unsub = await networkAPI.query.system.account.multi(addresses, async balances => {
+    const pooledStakingBalances = [];
+    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) {
+            let pooled = new BN(poolMemberData.points.toString());
+            Object.entries(poolMemberData.unbondingEras).forEach(([, amount]) => {
+              pooled = pooled.add(new BN(amount));
+            });
+            pooledStakingBalances.push(pooled);
+          } else {
+            pooledStakingBalances.push(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 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 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 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 BN(0);
+      let locked = reserved.add(miscFrozen);
+      total = total.add(reserved);
+      const pooledStakingBalance = pooledStakingBalances[index] || BN_ZERO;
+      if (pooledStakingBalance.gt(BN_ZERO)) {
+        total = total.add(pooledStakingBalance);
+        locked = locked.add(pooledStakingBalance);
+      }
+      const free = total.sub(locked);
+      return {
+        address: addresses[index],
+        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()
+        }
+      };
+    });
+    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, _ref;
+            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 {
+              address: address,
+              tokenSlug: tokenInfo.slug,
+              free: _balanceOf.output ? (_ref = balanceObj.ok) !== null && _ref !== void 0 ? _ref : balanceObj.Ok : '0',
+              locked: '0',
+              state: 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: APIItemState.READY
+            };
+          }
+        }));
+        callBack(balances);
+      } 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 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 items = balances.map((balance, index) => {
+          const tokenBalance = {
+            reserved: balance.reserved || new BN(0),
+            frozen: balance.frozen || new BN(0),
+            free: balance.free || new 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: 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 = 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 => {
+        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 = BN_ZERO;
+          let total = BN_ZERO;
+          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 = addressBalance;
+            } else {
+              total = addressBalance;
+            }
+          }
+          const free = total.sub(frozen);
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            free: free.toString(),
+            locked: frozen.toString(),
+            state: 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 = [_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 items = balances.map((balance, index) => {
+          const tokenBalance = {
+            reserved: balance.reserved || new BN(0),
+            frozen: balance.frozen || new BN(0),
+            free: balance.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);
+          return {
+            address: addresses[index],
+            tokenSlug: tokenInfo.slug,
+            state: 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);
+    });
+  };
+}

package/services/balance-service/index.d.ts

@@ -1,9 +1,28 @@
-import { AmountData, BalanceItem } from '@subwallet/extension-base/background/KoniTypes';
-import { ChainService } from '@subwallet/extension-base/services/chain-service';
+import { AmountData } from '@subwallet/extension-base/background/KoniTypes';
+import KoniState from '@subwallet/extension-base/koni/background/handlers/State';
+import { BalanceItem } from '@subwallet/extension-base/types';
+/**
+ * Balance service
+ * @class
+*/
 export declare class BalanceService {
-    private chainService;
-    constructor(chainService: ChainService);
+    private state;
+    /**
+     * @constructor
+     * @param {KoniState} state - The state of extension.
+     */
+    constructor(state: KoniState);
     subscribeTokenFreeBalance(address: string, chain: string, tokenSlug: string | undefined, callback?: (rs: AmountData) => void): Promise<[() => void, AmountData]>;
+    /**
+     * @public
+     * @async
+     * @function getTokenFreeBalance
+     * @desc Fetch free balance on chain
+     * @param {string} address - Address
+     * @param {string} chain - Slug of chain
+     * @param {string} [tokenSlug] - Slug of token
+     * @return {Promise<AmountData>} - Free token balance of address on chain
+    */
     getTokenFreeBalance(address: string, chain: string, tokenSlug?: string): Promise<AmountData>;
-    subscribeBalance(addresses: string[], chains: string[] | null, callback: (rs: BalanceItem) => void): () => void;
+    subscribeBalance(addresses: string[], chains: string[] | null, _callback: (rs: BalanceItem) => void): () => void;
 }

package/services/balance-service/index.js

@@ -3,16 +3,25 @@
 
 import { BalanceError } from '@subwallet/extension-base/background/errors/BalanceError';
 import { BalanceErrorType } from '@subwallet/extension-base/background/KoniTypes';
-import { subscribeEVMBalance, subscribeSubstrateBalance } from '@subwallet/extension-base/koni/api/dotsama/balance';
-import { state } from '@subwallet/extension-base/koni/background/handlers';
+import { groupBalance } from '@subwallet/extension-base/services/balance-service/helpers/group';
+import { subscribeEVMBalance } from '@subwallet/extension-base/services/balance-service/helpers/subscribe/evm';
+import { subscribeSubstrateBalance } from '@subwallet/extension-base/services/balance-service/helpers/subscribe/substrate';
 import { _PURE_EVM_CHAINS } from '@subwallet/extension-base/services/chain-service/constants';
 import { _getChainNativeTokenSlug, _isChainEvmCompatible, _isPureEvmChain } from '@subwallet/extension-base/services/chain-service/utils';
 import { categoryAddresses } from '@subwallet/extension-base/utils';
 import { t } from 'i18next';
-export class BalanceService {
-  constructor(chainService) {
-    this.chainService = chainService;
 
+/**
+ * Balance service
+ * @class
+*/
+export class BalanceService {
+  /**
+   * @constructor
+   * @param {KoniState} state - The state of extension.
+   */
+  constructor(state) {
+    this.state = state;
     // Todo: Load data from db to balanceSubject
     // Todo: Start  subscribe balance and data
     // Todo: Listen change and apply to balanceSubject
@@ -23,9 +32,10 @@ export class BalanceService {
     // Todo: Move everything of fetching balance to this service
   }
 
+  /* Subscribe token free balance on chain */
   async subscribeTokenFreeBalance(address, chain, tokenSlug, callback) {
-    const chainInfo = this.chainService.getChainInfoByKey(chain);
-    const chainState = this.chainService.getChainStateByKey(chain);
+    const chainInfo = this.state.chainService.getChainInfoByKey(chain);
+    const chainState = this.state.chainService.getChainStateByKey(chain);
     if (!chainInfo || !chainState || !chainState.active) {
       return Promise.reject(new BalanceError(BalanceErrorType.NETWORK_ERROR, t('{{chain}} is inactive. Please enable network', {
         replace: {
@@ -34,7 +44,7 @@ export class BalanceService {
       })));
     }
     const tSlug = tokenSlug || _getChainNativeTokenSlug(chainInfo);
-    const tokenInfo = this.chainService.getAssetBySlug(tSlug);
+    const tokenInfo = this.state.chainService.getAssetBySlug(tSlug);
     if (!tokenInfo) {
       return Promise.reject(new BalanceError(BalanceErrorType.TOKEN_ERROR, t('Transfer is currently not available for this token: {{tSlug}}', {
         replace: {
@@ -69,28 +79,44 @@ export class BalanceService {
       }, 9999);
     });
   }
+
+  /**
+   * @public
+   * @async
+   * @function getTokenFreeBalance
+   * @desc Fetch free balance on chain
+   * @param {string} address - Address
+   * @param {string} chain - Slug of chain
+   * @param {string} [tokenSlug] - Slug of token
+   * @return {Promise<AmountData>} - Free token balance of address on chain
+  */
   async getTokenFreeBalance(address, chain, tokenSlug) {
     const [, balance] = await this.subscribeTokenFreeBalance(address, chain, tokenSlug);
     return balance;
   }
-  subscribeBalance(addresses, chains, callback) {
+  subscribeBalance(addresses, chains, _callback) {
     const [substrateAddresses, evmAddresses] = categoryAddresses(addresses);
-    const chainInfoMap = this.chainService.getChainInfoMap();
-    const chainStateMap = this.chainService.getChainStateMap();
-    const substrateApiMap = this.chainService.getSubstrateApiMap();
-    const evmApiMap = this.chainService.getEvmApiMap();
+    const chainInfoMap = this.state.chainService.getChainInfoMap();
+    const chainStateMap = this.state.chainService.getChainStateMap();
+    const substrateApiMap = this.state.chainService.getSubstrateApiMap();
+    const evmApiMap = this.state.chainService.getEvmApiMap();
 
     // Get data from chain or all chains
     const chainList = chains || Object.keys(chainInfoMap);
     // Filter active chain only
     const useChainInfos = chainList.filter(c => chainStateMap[c] && chainStateMap[c].active).map(c => chainInfoMap[c]);
+    const callback = items => {
+      if (items.length) {
+        _callback(groupBalance(items, 'GROUPED', items[0].tokenSlug));
+      }
+    };
 
     // Looping over each chain
     const unsubList = useChainInfos.map(async chainInfo => {
       const chainSlug = chainInfo.slug;
       const useAddresses = _isChainEvmCompatible(chainInfo) ? evmAddresses : substrateAddresses;
       if (_isPureEvmChain(chainInfo)) {
-        const nativeTokenInfo = state.getNativeTokenInfo(chainSlug);
+        const nativeTokenInfo = this.state.getNativeTokenInfo(chainSlug);
         return subscribeEVMBalance(chainSlug, useAddresses, evmApiMap, callback, nativeTokenInfo);
       }
       if (!useAddresses || useAddresses.length === 0 || _PURE_EVM_CHAINS.indexOf(chainSlug) > -1) {