@subwallet/extension-base 1.3.17-0 → 1.3.18-1

package/koni/background/handlers/Extension.js

@@ -3,13 +3,12 @@
 
 import { Common } from '@ethereumjs/common';
 import { LegacyTransaction } from '@ethereumjs/tx';
-import { COMMON_CHAIN_SLUGS } from '@subwallet/chain-list';
 import { _AssetType } from '@subwallet/chain-list/types';
 import { TransactionError } from '@subwallet/extension-base/background/errors/TransactionError';
 import { withErrorLog } from '@subwallet/extension-base/background/handlers/helpers';
 import { createSubscription } from '@subwallet/extension-base/background/handlers/subscriptions';
 import { CampaignDataType, ChainType, ExternalRequestPromiseStatus, ExtrinsicType, MantaPayEnableMessage, StakingType } from '@subwallet/extension-base/background/KoniTypes';
-import { ALL_ACCOUNT_KEY, LATEST_SESSION, XCM_FEE_RATIO } from '@subwallet/extension-base/constants';
+import { ALL_ACCOUNT_KEY, ASSET_HUB_CHAIN_SLUGS, LATEST_SESSION } from '@subwallet/extension-base/constants';
 import { additionalValidateTransferForRecipient, additionalValidateXcmTransfer, validateTransferRequest, validateXcmTransferRequest } from '@subwallet/extension-base/core/logic-validation/transfer';
 import { _isSnowBridgeXcm } from '@subwallet/extension-base/core/substrate/xcm-parser';
 import { ALLOWED_PATH } from '@subwallet/extension-base/defaults';
@@ -25,23 +24,26 @@ import { getPoolingBondingExtrinsic, getPoolingUnbondingExtrinsic, validatePoolB
 import { YIELD_EXTRINSIC_TYPES } from '@subwallet/extension-base/koni/api/yield/helper/utils';
 import { isBounceableAddress } from '@subwallet/extension-base/services/balance-service/helpers/subscribe/ton/utils';
 import { getERC20TransactionObject, getERC721Transaction, getEVMTransactionObject, getPSP34TransferExtrinsic } from '@subwallet/extension-base/services/balance-service/transfer/smart-contract';
-import { createTransferExtrinsic, getTransferMockTxFee } from '@subwallet/extension-base/services/balance-service/transfer/token';
+import { createTransferExtrinsic } from '@subwallet/extension-base/services/balance-service/transfer/token';
 import { createTonTransaction } from '@subwallet/extension-base/services/balance-service/transfer/ton-transfer';
-import { createAvailBridgeExtrinsicFromAvail, createAvailBridgeTxFromEth, createPolygonBridgeExtrinsic, createSnowBridgeExtrinsic, createXcmExtrinsic, getXcmMockTxFee } from '@subwallet/extension-base/services/balance-service/transfer/xcm';
+import { createAvailBridgeExtrinsicFromAvail, createAvailBridgeTxFromEth, createPolygonBridgeExtrinsic, createSnowBridgeExtrinsic, createXcmExtrinsic } from '@subwallet/extension-base/services/balance-service/transfer/xcm';
 import { getClaimTxOnAvail, getClaimTxOnEthereum, isAvailChainBridge } from '@subwallet/extension-base/services/balance-service/transfer/xcm/availBridge';
 import { _isPolygonChainBridge, getClaimPolygonBridge, isClaimedPolygonBridge } from '@subwallet/extension-base/services/balance-service/transfer/xcm/polygonBridge';
 import { _isPosChainBridge, getClaimPosBridge } from '@subwallet/extension-base/services/balance-service/transfer/xcm/posBridge';
 import { _DEFAULT_MANTA_ZK_CHAIN, _MANTA_ZK_CHAIN_GROUP, _ZK_ASSET_PREFIX, SUFFICIENT_CHAIN } from '@subwallet/extension-base/services/chain-service/constants';
 import { _ChainConnectionStatus } from '@subwallet/extension-base/services/chain-service/types';
-import { _getAssetDecimals, _getAssetSymbol, _getChainNativeTokenBasicInfo, _getContractAddressOfToken, _getEvmChainId, _getTokenMinAmount, _getTokenOnChainAssetId, _getXcmAssetMultilocation, _isAssetSmartContractNft, _isBridgedToken, _isChainEvmCompatible, _isChainSubstrateCompatible, _isChainTonCompatible, _isCustomAsset, _isLocalToken, _isMantaZkAsset, _isNativeToken, _isPureEvmChain, _isTokenEvmSmartContract, _isTokenTransferredByEvm, _isTokenTransferredByTon } from '@subwallet/extension-base/services/chain-service/utils';
+import { _getAssetDecimals, _getAssetSymbol, _getChainNativeTokenBasicInfo, _getContractAddressOfToken, _getEvmChainId, _getTokenOnChainAssetId, _getXcmAssetMultilocation, _isAssetSmartContractNft, _isBridgedToken, _isChainEvmCompatible, _isChainSubstrateCompatible, _isCustomAsset, _isLocalToken, _isMantaZkAsset, _isNativeToken, _isPureEvmChain, _isTokenEvmSmartContract, _isTokenTransferredByEvm, _isTokenTransferredByTon } from '@subwallet/extension-base/services/chain-service/utils';
+import { calculateToAmountByReservePool } from '@subwallet/extension-base/services/fee-service/utils';
 import { EXTENSION_REQUEST_URL } from '@subwallet/extension-base/services/request-service/constants';
 import { DEFAULT_AUTO_LOCK_TIME } from '@subwallet/extension-base/services/setting-service/constants';
+import { checkLiquidityForPool, estimateTokensForPool, getReserveForPool } from '@subwallet/extension-base/services/swap-service/handler/asset-hub/utils';
 import { isProposalExpired, isSupportWalletConnectChain, isSupportWalletConnectNamespace } from '@subwallet/extension-base/services/wallet-connect-service/helpers';
 import { SWStorage } from '@subwallet/extension-base/storage';
 import { AccountsStore } from '@subwallet/extension-base/stores';
 import { BasicTxErrorType, BasicTxWarningCode, StakingTxErrorType, YieldPoolType } from '@subwallet/extension-base/types';
-import { _analyzeAddress, BN_ZERO, combineAllAccountProxy, createTransactionFromRLP, isSameAddress, MODULE_SUPPORT, reformatAddress, signatureToHex, toBNString, transformAccounts, transformAddresses, uniqueStringArray } from '@subwallet/extension-base/utils';
+import { _analyzeAddress, calculateMaxTransferable, combineAllAccountProxy, createTransactionFromRLP, detectTransferTxType, isSameAddress, MODULE_SUPPORT, reformatAddress, signatureToHex, transformAccounts, transformAddresses, uniqueStringArray } from '@subwallet/extension-base/utils';
 import { parseContractInput, parseEvmRlp } from '@subwallet/extension-base/utils/eth/parseTransaction';
+import { getId } from '@subwallet/extension-base/utils/getId';
 import { getKeypairTypeByAddress, isAddress, isSubstrateAddress, isTonAddress } from '@subwallet/keyring';
 import { EthereumKeypairTypes, SubstrateKeypairTypes, TonKeypairTypes } from '@subwallet/keyring/types';
 import { keyring } from '@subwallet/ui-keyring';
@@ -50,7 +52,7 @@ import BigN from 'bignumber.js';
 import { t } from 'i18next';
 import { combineLatest, Subject } from 'rxjs';
 import { TypeRegistry } from '@polkadot/types';
-import { assert, hexStripPrefix, hexToU8a, isAscii, isHex, u8aToHex } from '@polkadot/util';
+import { assert, hexStripPrefix, hexToU8a, isAscii, isHex, noop, u8aToHex } from '@polkadot/util';
 import { decodeAddress, isEthereumAddress } from '@polkadot/util-crypto';
 import { getSuitableRegistry, setupApiRegistry, setupDappRegistry, setupDatabaseRegistry } from "../utils.js";
 export function isJsonPayload(value) {
@@ -1101,8 +1103,11 @@ export default class KoniExtension {
   }
   async makeTransfer(inputData) {
     const {
+      chain,
+      feeCustom,
+      feeOption,
       from,
-      networkKey,
+      nonNativeTokenPayFeeSlug,
       to,
       tokenSlug,
       transferAll,
@@ -1110,12 +1115,13 @@ export default class KoniExtension {
       value
     } = inputData;
     const transferTokenInfo = this.#koniState.chainService.getAssetBySlug(tokenSlug);
-    const [errors,,] = validateTransferRequest(transferTokenInfo, from, to, value, transferAll);
+    const errors = validateTransferRequest(transferTokenInfo, from, to, value, transferAll);
     const warnings = [];
-    const chainInfo = this.#koniState.getChainInfo(networkKey);
-    const nativeTokenInfo = this.#koniState.getNativeTokenInfo(networkKey);
+    const chainInfo = this.#koniState.getChainInfo(chain);
+    const nativeTokenInfo = this.#koniState.getNativeTokenInfo(chain);
     const nativeTokenSlug = nativeTokenInfo.slug;
     const isTransferNativeToken = nativeTokenSlug === tokenSlug;
+    const isTransferLocalTokenAndPayThatTokenAsFee = !isTransferNativeToken && nonNativeTokenPayFeeSlug === tokenSlug;
     const extrinsicType = isTransferNativeToken ? ExtrinsicType.TRANSFER_BALANCE : ExtrinsicType.TRANSFER_TOKEN;
     let chainType = ChainType.SUBSTRATE;
     const transferAmount = {
@@ -1126,7 +1132,7 @@ export default class KoniExtension {
     let transaction;
     const transferTokenAvailable = await this.getAddressTransferableBalance({
       address: from,
-      networkKey,
+      networkKey: chain,
       token: tokenSlug,
       extrinsicType
     });
@@ -1134,38 +1140,60 @@ export default class KoniExtension {
       if (isEthereumAddress(from) && isEthereumAddress(to) && _isTokenTransferredByEvm(transferTokenInfo)) {
         chainType = ChainType.EVM;
         const txVal = transferAll ? transferTokenAvailable.value : value || '0';
-        const evmApi = this.#koniState.getEvmApi(networkKey);
+        const evmApi = this.#koniState.getEvmApi(chain);
+        const feeInfo = await this.#koniState.feeService.subscribeChainFee(getId(), chain, 'evm');
 
         // todo: refactor: merge getERC20TransactionObject & getEVMTransactionObject
         // Estimate with EVM API
         if (_isTokenEvmSmartContract(transferTokenInfo) || _isLocalToken(transferTokenInfo)) {
-          [transaction, transferAmount.value] = await getERC20TransactionObject(_getContractAddressOfToken(transferTokenInfo), chainInfo, from, to, txVal, !!transferAll, evmApi);
+          [transaction, transferAmount.value] = await getERC20TransactionObject({
+            assetAddress: _getContractAddressOfToken(transferTokenInfo),
+            chain,
+            evmApi,
+            feeCustom,
+            feeInfo,
+            feeOption,
+            from,
+            to,
+            transferAll,
+            value: txVal
+          });
         } else {
-          [transaction, transferAmount.value] = await getEVMTransactionObject(chainInfo, from, to, txVal, !!transferAll, evmApi);
+          [transaction, transferAmount.value] = await getEVMTransactionObject({
+            chain,
+            evmApi,
+            feeCustom,
+            feeInfo,
+            feeOption,
+            from,
+            to,
+            transferAll,
+            value: txVal
+          });
         }
       } else if (_isMantaZkAsset(transferTokenInfo)) {
         transaction = undefined;
         transferAmount.value = '0';
       } else if (isTonAddress(from) && isTonAddress(to) && _isTokenTransferredByTon(transferTokenInfo)) {
         chainType = ChainType.TON;
-        const tonApi = this.#koniState.getTonApi(networkKey);
+        const tonApi = this.#koniState.getTonApi(chain);
         [transaction, transferAmount.value] = await createTonTransaction({
           tokenInfo: transferTokenInfo,
           from,
           to,
-          networkKey,
+          networkKey: chain,
           value: value || '0',
           transferAll: !!transferAll,
           // currently not used
           tonApi
         });
       } else {
-        const substrateApi = this.#koniState.getSubstrateApi(networkKey);
+        const substrateApi = this.#koniState.getSubstrateApi(chain);
         [transaction, transferAmount.value] = await createTransferExtrinsic({
           transferAll: !!transferAll,
           value: value || '0',
           from: from,
-          networkKey,
+          networkKey: chain,
           tokenInfo: transferTokenInfo,
           to: to,
           substrateApi
@@ -1186,16 +1214,27 @@ export default class KoniExtension {
         return undefined;
       }
 
+      // Check enough free local to pay fee local
+      if (nonNativeTokenPayFeeSlug) {
+        var _inputTransaction$est;
+        const nonNativeFee = BigInt(((_inputTransaction$est = inputTransaction.estimateFee) === null || _inputTransaction$est === void 0 ? void 0 : _inputTransaction$est.value) || '0'); // todo: estimateFee should be must-have, need to refactor interface
+        const nonNativeTokenPayFeeInfo = await this.#koniState.balanceService.getTokensHasBalance(reformatAddress(from), chain, nonNativeTokenPayFeeSlug);
+        const nonNativeTokenPayFeeBalance = BigInt(nonNativeTokenPayFeeInfo[nonNativeTokenPayFeeSlug].free);
+        if (nonNativeFee > nonNativeTokenPayFeeBalance) {
+          inputTransaction.errors.push(new TransactionError(BasicTxErrorType.NOT_ENOUGH_BALANCE));
+        }
+      }
+
       // Check ed for sender
       if (!isTransferNativeToken) {
         const [_senderSendingTokenTransferable, _receiverNativeTotal] = await Promise.all([this.getAddressTransferableBalance({
           address: from,
-          networkKey,
+          networkKey: chain,
           token: tokenSlug,
           extrinsicType
         }), this.getAddressTotalBalance({
           address: to,
-          networkKey,
+          networkKey: chain,
           token: nativeTokenSlug,
           extrinsicType: ExtrinsicType.TRANSFER_BALANCE
         })]);
@@ -1206,13 +1245,13 @@ export default class KoniExtension {
         value: _receiverSendingTokenKeepAliveBalance
       } = await this.getAddressTotalBalance({
         address: to,
-        networkKey,
+        networkKey: chain,
         token: tokenSlug,
         extrinsicType
       }); // todo: shouldn't be just transferable, locked also counts
       const receiverSendingTokenKeepAliveBalance = BigInt(_receiverSendingTokenKeepAliveBalance);
       const amount = BigInt(transferAmount.value);
-      const substrateApi = this.#koniState.getSubstrateApi(networkKey);
+      const substrateApi = this.#koniState.getSubstrateApi(chain);
       const isSendingTokenSufficient = await this.isSufficientToken(transferTokenInfo, substrateApi);
       const [warnings, errors] = additionalValidateTransferForRecipient(transferTokenInfo, nativeTokenInfo, extrinsicType, receiverSendingTokenKeepAliveBalance, amount, senderSendingTokenTransferable, receiverSystemAccountInfo, isSendingTokenSufficient);
       warnings.length && inputTransaction.warnings.push(...warnings);
@@ -1229,7 +1268,10 @@ export default class KoniExtension {
       errors,
       warnings,
       address: from,
-      chain: networkKey,
+      chain,
+      feeCustom,
+      feeOption,
+      nonNativeTokenPayFeeSlug,
       chainType,
       transferNativeAmount,
       transaction,
@@ -1237,6 +1279,7 @@ export default class KoniExtension {
       extrinsicType,
       ignoreWarnings,
       isTransferAll: isTransferNativeToken ? transferAll : false,
+      isTransferLocalTokenAndPayThatTokenAsFee,
       edAsWarning: isTransferNativeToken,
       additionalValidator: additionalValidator
     });
@@ -1244,7 +1287,10 @@ export default class KoniExtension {
   async makeCrossChainTransfer(inputData) {
     const {
       destinationNetworkKey,
+      feeCustom,
+      feeOption,
       from,
+      nonNativeTokenPayFeeSlug,
       originNetworkKey,
       to,
       tokenSlug,
@@ -1265,33 +1311,46 @@ export default class KoniExtension {
     const isSnowBridgeEvmTransfer = _isPureEvmChain(chainInfoMap[originNetworkKey]) && _isSnowBridgeXcm(chainInfoMap[originNetworkKey], chainInfoMap[destinationNetworkKey]) && !isAvailBridgeFromEvm;
     const isPolygonBridgeTransfer = _isPolygonChainBridge(originNetworkKey, destinationNetworkKey);
     const isPosBridgeTransfer = _isPosChainBridge(originNetworkKey, destinationNetworkKey);
+    const isTransferNative = this.#koniState.getNativeTokenInfo(originNetworkKey).slug === tokenSlug;
+    const isTransferLocalTokenAndPayThatTokenAsFee = !isTransferNative && tokenSlug === nonNativeTokenPayFeeSlug;
     let additionalValidator;
     let eventsHandler;
     if (fromKeyPair && destinationTokenInfo) {
       const evmApi = this.#koniState.getEvmApi(originNetworkKey);
       const substrateApi = this.#koniState.getSubstrateApi(originNetworkKey);
-      const params = {
-        destinationTokenInfo,
-        originTokenInfo,
-        sendingValue: value,
-        sender: from,
-        recipient: to,
-        chainInfoMap,
-        substrateApi,
-        evmApi
-      };
       let funcCreateExtrinsic;
+      let type;
       if (isPosBridgeTransfer || isPolygonBridgeTransfer) {
         funcCreateExtrinsic = createPolygonBridgeExtrinsic;
+        type = 'evm';
       } else if (isSnowBridgeEvmTransfer) {
         funcCreateExtrinsic = createSnowBridgeExtrinsic;
+        type = 'evm';
       } else if (isAvailBridgeFromEvm) {
         funcCreateExtrinsic = createAvailBridgeTxFromEth;
+        type = 'evm';
       } else if (isAvailBridgeFromAvail) {
         funcCreateExtrinsic = createAvailBridgeExtrinsicFromAvail;
+        type = 'substrate';
       } else {
         funcCreateExtrinsic = createXcmExtrinsic;
+        type = 'substrate';
       }
+      const feeInfo = await this.#koniState.feeService.subscribeChainFee(getId(), originNetworkKey, type);
+      const params = {
+        destinationTokenInfo,
+        originTokenInfo,
+        sendingValue: value,
+        sender: from,
+        recipient: to,
+        destinationChain: chainInfoMap[destinationTokenInfo.originChain],
+        originChain: chainInfoMap[originTokenInfo.originChain],
+        substrateApi,
+        evmApi,
+        feeCustom,
+        feeOption,
+        feeInfo
+      };
       extrinsic = await funcCreateExtrinsic(params);
       additionalValidator = async inputTransaction => {
         const {
@@ -1351,12 +1410,77 @@ export default class KoniExtension {
       chainType: !isSnowBridgeEvmTransfer && !isAvailBridgeFromEvm && !isPolygonBridgeTransfer && !isPosBridgeTransfer ? ChainType.SUBSTRATE : ChainType.EVM,
       transferNativeAmount: _isNativeToken(originTokenInfo) ? value : '0',
       ignoreWarnings,
+      nonNativeTokenPayFeeSlug,
       isTransferAll: transferAll,
+      isTransferLocalTokenAndPayThatTokenAsFee,
       errors,
       additionalValidator: additionalValidator,
       eventsHandler: eventsHandler
     });
   }
+  async getTokensCanPayFee(request) {
+    var _tokensHasBalanceInfo;
+    const {
+      address: _address,
+      chain,
+      feeAmount
+    } = request;
+    const chainService = this.#koniState.chainService;
+    const substrateApi = this.#koniState.getSubstrateApi(chain);
+    const address = reformatAddress(_address);
+
+    // ensure nativeTokenInfo and localTokenInfo have multi-location metadata beforehand to improve performance.
+    const tokensHasBalanceInfoMap = await this.#koniState.balanceService.getTokensHasBalance(address, chain);
+    const tokensHasBalanceSlug = Object.keys(tokensHasBalanceInfoMap);
+    const tokenInfos = tokensHasBalanceSlug.map(tokenSlug => chainService.getAssetBySlug(tokenSlug)).filter(token => token.originChain === chain && token.assetType !== _AssetType.NATIVE && token.metadata && token.metadata.multilocation);
+    const nativeTokenInfo = chainService.getNativeTokenInfo(chain);
+    const nativeTokenBalanceInfo = {
+      slug: nativeTokenInfo.slug,
+      free: ((_tokensHasBalanceInfo = tokensHasBalanceInfoMap[nativeTokenInfo.slug]) === null || _tokensHasBalanceInfo === void 0 ? void 0 : _tokensHasBalanceInfo.free) || '0',
+      rate: '1'
+    };
+    const tokensCanPayFee = [nativeTokenBalanceInfo];
+    if (!ASSET_HUB_CHAIN_SLUGS.includes(chain) || !nativeTokenInfo.metadata || !nativeTokenInfo.metadata.multilocation) {
+      return tokensCanPayFee;
+    }
+    await Promise.all(tokenInfos.map(async tokenInfo => {
+      const tokenSlug = tokenInfo.slug;
+      const reserve = await getReserveForPool(substrateApi.api, nativeTokenInfo, tokenInfo);
+      if (!reserve || !reserve[0] || !reserve[1] || reserve[0] === '0' || reserve[1] === '0') {
+        return;
+      }
+      const rate = new BigN(reserve[1]).div(reserve[0]).toFixed();
+      const tokenCanPayFee = {
+        slug: tokenSlug,
+        free: tokensHasBalanceInfoMap[tokenSlug].free,
+        rate
+      };
+      if (feeAmount === undefined) {
+        tokensCanPayFee.push(tokenCanPayFee);
+      } else {
+        const amount = estimateTokensForPool(feeAmount, reserve);
+        const liquidityError = checkLiquidityForPool(amount, reserve[0], reserve[1]);
+        if (!liquidityError) {
+          tokensCanPayFee.push(tokenCanPayFee);
+        }
+      }
+    }));
+    return tokensCanPayFee;
+  }
+  async getAmountForPair(request) {
+    const {
+      nativeTokenFeeAmount,
+      nativeTokenSlug,
+      toTokenSlug
+    } = request;
+    if (nativeTokenSlug === toTokenSlug) {
+      return nativeTokenFeeAmount;
+    }
+    const nativeTokenInfo = this.#koniState.chainService.getAssetBySlug(nativeTokenSlug);
+    const toTokenInfo = this.#koniState.chainService.getAssetBySlug(toTokenSlug);
+    const substrateApi = this.#koniState.chainService.getSubstrateApi(nativeTokenInfo.originChain);
+    return await calculateToAmountByReservePool(substrateApi.api, nativeTokenInfo, toTokenInfo, nativeTokenFeeAmount);
+  }
   async evmNftSubmitTransaction(inputData) {
     const {
       networkKey,
@@ -1377,7 +1501,8 @@ export default class KoniExtension {
         url: EXTENSION_REQUEST_URL
       });
     }
-    const transaction = await getERC721Transaction(this.#koniState.getEvmApi(networkKey), networkKey, contractAddress, senderAddress, recipientAddress, tokenId);
+    const feeInfo = await this.#koniState.feeService.subscribeChainFee(getId(), networkKey, 'evm');
+    const transaction = await getERC721Transaction(this.#koniState.getEvmApi(networkKey), networkKey, contractAddress, senderAddress, recipientAddress, tokenId, feeInfo);
 
     // this.addContact(recipientAddress);
 
@@ -1530,72 +1655,87 @@ export default class KoniExtension {
   }) {
     return await this.#koniState.balanceService.getTotalBalance(address, networkKey, token, extrinsicType);
   }
-  async getMaxTransferable({
-    address,
-    destChain,
-    isXcmTransfer,
-    networkKey,
-    token
-  }) {
-    const tokenInfo = token ? this.#koniState.chainService.getAssetBySlug(token) : this.#koniState.chainService.getNativeTokenInfo(networkKey);
-    if (!_isNativeToken(tokenInfo)) {
-      return await this.getAddressTransferableBalance({
-        extrinsicType: ExtrinsicType.TRANSFER_TOKEN,
-        address,
-        networkKey,
-        token
-      });
-    } else {
-      let maxTransferable;
-      if (isXcmTransfer) {
-        maxTransferable = await this.getXcmMaxTransferable(tokenInfo, destChain, address);
-      } else {
-        // regular transfer with native token
-        maxTransferable = await this.getNativeTokenMaxTransferable(tokenInfo, networkKey, address);
-      }
-      return {
-        value: maxTransferable.gt(BN_ZERO) ? maxTransferable.toFixed(0) || '0' : '0',
-        decimals: tokenInfo.decimals,
-        symbol: tokenInfo.symbol
-      };
-    }
-  }
-  async getXcmMaxTransferable(originTokenInfo, destChain, address) {
-    const substrateApi = this.#koniState.chainService.getSubstrateApi(originTokenInfo.originChain);
-    const chainInfoMap = this.#koniState.chainService.getChainInfoMap();
-    const destinationTokenInfo = this.#koniState.getXcmEqualAssetByChain(destChain, originTokenInfo.slug);
-
-    // todo: improve this case. Currently set 1 AVAIL for covering fee as default.
-    const isSpecialBridgeFromAvail = originTokenInfo.slug === 'avail_mainnet-NATIVE-AVAIL' && destChain === COMMON_CHAIN_SLUGS.ETHEREUM;
-    const specialBridgeFromAvailFee = new BigN(toBNString(1, _getAssetDecimals(originTokenInfo))).minus(new BigN(_getTokenMinAmount(originTokenInfo)));
-    if (destinationTokenInfo) {
-      const [bnMockExecutionFee, {
-        value
-      }] = await Promise.all([getXcmMockTxFee(substrateApi, chainInfoMap, originTokenInfo, destinationTokenInfo), this.getAddressTransferableBalance({
-        extrinsicType: ExtrinsicType.TRANSFER_XCM,
-        address,
-        networkKey: originTokenInfo.originChain,
-        token: originTokenInfo.slug
-      })]);
-      const bnMaxTransferable = new BigN(value);
-      const txFee = isSpecialBridgeFromAvail ? specialBridgeFromAvailFee : bnMockExecutionFee.multipliedBy(XCM_FEE_RATIO);
-      return bnMaxTransferable.minus(txFee);
-    }
-    return new BigN(0);
-  }
-  async getNativeTokenMaxTransferable(tokenInfo, networkKey, address) {
-    const chainInfo = this.#koniState.chainService.getChainInfoByKey(networkKey);
-    const api = _isChainEvmCompatible(chainInfo) && _isTokenTransferredByEvm(tokenInfo) ? this.#koniState.chainService.getEvmApi(networkKey) : _isChainTonCompatible(chainInfo) && _isTokenTransferredByTon(tokenInfo) ? this.#koniState.chainService.getTonApi(networkKey) : this.#koniState.chainService.getSubstrateApi(networkKey);
-    const [mockTxFee, {
-      value
-    }] = await Promise.all([getTransferMockTxFee(address, chainInfo, tokenInfo, api), this.getAddressTransferableBalance({
-      extrinsicType: ExtrinsicType.TRANSFER_BALANCE,
+  async subscribeMaxTransferable(request, id, port) {
+    const {
       address,
-      networkKey,
-      token: tokenInfo.slug
-    })]);
-    const bnMaxTransferable = new BigN(value);
-    return bnMaxTransferable.minus(mockTxFee);
+      chain,
+      destChain: _destChain,
+      feeCustom,
+      feeOption,
+      nonNativeTokenPayFeeSlug,
+      token
+    } = request;
+    const cb = createSubscription(id, port);
+    const transferTokenInfo = this.#koniState.chainService.getAssetBySlug(token);
+    const isTransferLocalTokenAndPayThatTokenAsFee = !_isNativeToken(transferTokenInfo) && !!nonNativeTokenPayFeeSlug && nonNativeTokenPayFeeSlug === token;
+    const isTransferNativeTokenAndPayLocalTokenAsFee = _isNativeToken(transferTokenInfo) && !!nonNativeTokenPayFeeSlug;
+    const srcToken = token ? this.#koniState.chainService.getAssetBySlug(token) : this.#koniState.chainService.getNativeTokenInfo(chain);
+    const destToken = _destChain !== chain ? this.#koniState.getXcmEqualAssetByChain(_destChain, srcToken.slug) : srcToken;
+    const srcChain = this.#koniState.chainService.getChainInfoByKey(chain);
+    const destChain = this.#koniState.chainService.getChainInfoByKey(_destChain);
+    const nativeToken = this.#koniState.chainService.getNativeTokenInfo(chain);
+    const extrinsicType = srcChain.slug !== destChain.slug ? ExtrinsicType.TRANSFER_XCM : ExtrinsicType.TRANSFER_BALANCE;
+    const freeBalanceSubject = new Subject();
+    const feeSubject = new Subject();
+    const feeChainType = detectTransferTxType(srcToken, srcChain, destChain);
+    if (!destToken) {
+      throw new Error('Destination token not found');
+    }
+    const _request = {
+      address: address,
+      destChain,
+      destToken,
+      evmApi: this.#koniState.chainService.getEvmApi(chain),
+      feeCustom,
+      feeOption,
+      srcChain,
+      srcToken,
+      substrateApi: this.#koniState.chainService.getSubstrateApi(chain),
+      tonApi: this.#koniState.chainService.getTonApi(chain),
+      isTransferLocalTokenAndPayThatTokenAsFee,
+      isTransferNativeTokenAndPayLocalTokenAsFee,
+      nativeToken
+    };
+    const subscription = combineLatest({
+      freeBalance: freeBalanceSubject,
+      fee: feeSubject
+    }).subscribe({
+      next: ({
+        fee,
+        freeBalance
+      }) => {
+        calculateMaxTransferable(id, _request, freeBalance, fee).then(cb).catch(console.error);
+      }
+    });
+    const [unsubBalance, freeBalance] = await (async () => {
+      try {
+        return await this.#koniState.balanceService.subscribeBalance(address, chain, token, 'transferable', extrinsicType, data => {
+          freeBalanceSubject.next(data); // Must be called after subscription
+        });
+      } catch (e) {
+        const fallBackValue = {
+          value: '0',
+          decimals: srcToken.decimals || 0,
+          symbol: srcToken.symbol
+        };
+        freeBalanceSubject.next(fallBackValue);
+        return [noop, fallBackValue];
+      }
+    })();
+    const fee = await this.#koniState.feeService.subscribeChainFee(id, chain, feeChainType, data => {
+      feeSubject.next(data); // Must be called after subscription
+    });
+
+    const unsub = () => {
+      subscription.unsubscribe();
+      unsubBalance();
+      this.#koniState.feeService.unsubscribeChainFee(id, chain, feeChainType);
+    };
+    this.createUnsubscriptionHandle(id, unsub);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return calculateMaxTransferable(id, _request, freeBalance, fee);
   }
   async subscribeAddressTransferableBalance({
     address,
@@ -3330,6 +3470,7 @@ export default class KoniExtension {
       // skipFeeValidation: chosenFeeToken && allowSkipValidation
     });
   }
+
   /* Swap service */
 
   /* Notification service */
@@ -3361,6 +3502,7 @@ export default class KoniExtension {
     }
     return result;
   }
+
   /* Notification service */
 
   async submitClaimAvailBridge(data) {
@@ -3378,7 +3520,8 @@ export default class KoniExtension {
       chainType = ChainType.SUBSTRATE;
     } else {
       const evmApi = this.#koniState.getEvmApi(chain);
-      transaction = await getClaimTxOnEthereum(chain, notification, evmApi);
+      const feeInfo = await this.#koniState.feeService.subscribeChainFee(getId(), chain, 'evm');
+      transaction = await getClaimTxOnEthereum(chain, notification, evmApi, feeInfo);
       chainType = ChainType.EVM;
     }
     return await this.#koniState.transactionService.handleTransaction({
@@ -3406,10 +3549,11 @@ export default class KoniExtension {
     let transaction = null;
     const evmApi = this.#koniState.getEvmApi(chain);
     const metadata = notification.metadata;
+    const feeInfo = await this.#koniState.feeService.subscribeChainFee(getId(), chain, 'evm');
     if (metadata.bridgeType === 'POS') {
-      transaction = await getClaimPosBridge(chain, notification, evmApi);
+      transaction = await getClaimPosBridge(chain, notification, evmApi, feeInfo);
     } else {
-      transaction = await getClaimPolygonBridge(chain, notification, evmApi);
+      transaction = await getClaimPolygonBridge(chain, notification, evmApi, feeInfo);
     }
     const chainType = ChainType.EVM;
     return await this.#koniState.transactionService.handleTransaction({
@@ -3761,8 +3905,8 @@ export default class KoniExtension {
         return this.subscribeAssetSetting(id, port);
       case 'pri(assetSetting.update)':
         return await this.updateAssetSetting(request);
-      case 'pri(transfer.getMaxTransferable)':
-        return this.getMaxTransferable(request);
+      case 'pri(transfer.subscribe)':
+        return this.subscribeMaxTransferable(request, id, port);
       case 'pri(freeBalance.get)':
         return this.getAddressTransferableBalance(request);
       case 'pri(freeBalance.subscribe)':
@@ -3787,6 +3931,10 @@ export default class KoniExtension {
         return await this.getOptimalTransferProcess(request);
       case 'pri(accounts.approveSpending)':
         return await this.approveSpending(request);
+      case 'pri(customFee.getTokensCanPayFee)':
+        return this.getTokensCanPayFee(request);
+      case 'pri(customFee.getAmountForPair)':
+        return this.getAmountForPair(request);
 
       /// Sign QR
       case 'pri(qr.transaction.parse.substrate)':
@@ -4032,10 +4180,8 @@ export default class KoniExtension {
       /* Avail Bridge */
 
       /* Polygon Bridge */
-
       case 'pri(polygonBridge.submitClaimPolygonBridge)':
         return this.submitClaimPolygonBridge(request);
-
       /* Polygon Bridge */
 
       /* Ledger */

package/koni/background/handlers/State.js

@@ -907,14 +907,7 @@ export default class KoniState {
         }
       })();
       promiseList.push(Promise.race([promise, timeoutPromise]).then(result => {
-        var _result$gasPrice, _result$maxFeePerGas, _result$maxPriorityFe, _result$baseGasFee;
-        return [slug, result ? {
-          ...result,
-          gasPrice: (_result$gasPrice = result.gasPrice) === null || _result$gasPrice === void 0 ? void 0 : _result$gasPrice.toString(),
-          maxFeePerGas: (_result$maxFeePerGas = result.maxFeePerGas) === null || _result$maxFeePerGas === void 0 ? void 0 : _result$maxFeePerGas.toString(),
-          maxPriorityFeePerGas: (_result$maxPriorityFe = result.maxPriorityFeePerGas) === null || _result$maxPriorityFe === void 0 ? void 0 : _result$maxPriorityFe.toString(),
-          baseGasFee: (_result$baseGasFee = result.baseGasFee) === null || _result$baseGasFee === void 0 ? void 0 : _result$baseGasFee.toString()
-        } : null];
+        return [slug, result || null];
       }));
     });
     return Object.fromEntries(await Promise.all(promiseList));