@subwallet/extension-base 1.3.66-0 → 1.3.68-0

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

@@ -1193,6 +1193,9 @@ class KoniExtension {
     });
     return this.getNft();
   }
+  async handleGetNftFullList(request) {
+    return this.#koniState.nftDetectionService.getFullNftInstancesByCollection(request);
+  }
   getStakingReward() {
     return new Promise((resolve, reject) => {
       this.#koniState.getStakingReward(rs => {
@@ -1694,7 +1697,7 @@ class KoniExtension {
           error.length && inputTransaction.errors.push(...error);
         }
         if (isSubstrateXcm) {
-          const isDryRunSuccess = await (0, _xcm.dryRunXcmExtrinsicV2)(params);
+          const isDryRunSuccess = await (0, _xcm.dryRunXcmExtrinsicV2)(params, false);
           if (!isDryRunSuccess) {
             inputTransaction.errors.push(new _TransactionError.TransactionError(_types4.BasicTxErrorType.UNABLE_TO_SEND, 'Unable to perform transaction. Select another token or destination chain and try again'));
           }
@@ -2088,20 +2091,40 @@ class KoniExtension {
     var _data$metadata;
     const evmApi = this.#koniState.getEvmApi(data.originChain);
     const contractAddress = (_data$metadata = data.metadata) === null || _data$metadata === void 0 ? void 0 : _data$metadata.contractAddress;
-
-    // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
     const tokenContract = new evmApi.api.eth.Contract(_utils2._ERC721_ABI, contractAddress);
     try {
-      // eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
-      await tokenContract.methods.tokenOfOwnerByIndex('0xB7fdD27a8Df011816205a6e3cAA097DC4D8C2C5d', 1).call();
-      return true;
-    } catch (err) {
-      const error = err;
-      if (error.message.includes('index out of bounds')) {
-        return true;
-      } else {
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment,@typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      const supports1155 = await tokenContract.methods.supportsInterface('0xd9b67a26').call().catch(() => false);
+      if (supports1155) {
         return false;
       }
+
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-assignment
+      const supports721 = await tokenContract.methods.supportsInterface('0x80ac58cd').call().catch(() => false);
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access,@typescript-eslint/no-unsafe-assignment
+      const supportsMetadata = await tokenContract.methods.supportsInterface('0x5b5e139f').call().catch(() => false);
+      if (supports721 || supportsMetadata) {
+        return true;
+      }
+
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
+      if (tokenContract.methods.ownerOf) {
+        try {
+          // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+          await tokenContract.methods.ownerOf(1).call();
+          return true;
+        } catch (err) {
+          var _err$message, _err$message2;
+          // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+          if ((_err$message = err.message) !== null && _err$message !== void 0 && _err$message.includes('nonexistent token') || (_err$message2 = err.message) !== null && _err$message2 !== void 0 && _err$message2.includes('invalid token ID')) {
+            return true;
+          }
+        }
+      }
+      return false;
+    } catch (err) {
+      const error = err;
+      return error.message.includes('index out of bounds');
     }
   }
   async upsertCustomToken(data) {
@@ -2158,13 +2181,29 @@ class KoniExtension {
     }
     return await this.#koniState.balanceService.getTransferableBalance(address, networkKey, token, extrinsicType);
   }
-  async getAddressTotalBalance(_ref39) {
+  async getAddressAvailableBalanceByType(_ref39) {
     let {
       address,
+      balanceType,
       extrinsicType,
       networkKey,
       token
     } = _ref39;
+    if (token && _constants2._MANTA_ZK_CHAIN_GROUP.includes(networkKey)) {
+      const tokenInfo = this.#koniState.chainService.getAssetBySlug(token);
+      if (tokenInfo.symbol.startsWith(_constants2._ZK_ASSET_PREFIX)) {
+        return await this.#koniState.getMantaPayZkBalance(address, tokenInfo);
+      }
+    }
+    return await this.#koniState.balanceService.getBalanceByType(address, networkKey, token, balanceType, extrinsicType);
+  }
+  async getAddressTotalBalance(_ref40) {
+    let {
+      address,
+      extrinsicType,
+      networkKey,
+      token
+    } = _ref40;
     return await this.#koniState.balanceService.getTotalBalance(address, networkKey, token, extrinsicType);
   }
   async subscribeMaxTransferable(request, id, port) {
@@ -2221,17 +2260,17 @@ class KoniExtension {
       freeBalance: freeBalanceSubject,
       fee: feeSubject
     }).subscribe({
-      next: _ref40 => {
+      next: _ref41 => {
         let {
           fee,
           freeBalance
-        } = _ref40;
+        } = _ref41;
         (0, _utils8.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 => {
+        return await this.#koniState.balanceService.subscribeBalance(address, chain, token, _types4.BalanceType.TRANSFERABLE, extrinsicType, data => {
           freeBalanceSubject.next(data); // Must be called after subscription
         });
       } catch (e) {
@@ -2259,7 +2298,7 @@ class KoniExtension {
     });
     return (0, _utils8.calculateMaxTransferable)(id, _request, freeBalance, fee);
   }
-  async subscribeTransferableWhenConfirmation(_ref41, id, port) {
+  async subscribeTransferableWhenConfirmation(_ref42, id, port) {
     let {
       address,
       chain,
@@ -2268,7 +2307,7 @@ class KoniExtension {
       to,
       token,
       value
-    } = _ref41;
+    } = _ref42;
     const cb = (0, _subscriptions.createSubscription)(id, port);
     const freeBalanceSubject = new _rxjs.Subject();
     const feeSubject = new _rxjs.Subject();
@@ -2363,15 +2402,15 @@ class KoniExtension {
       freeBalance: freeBalanceSubject,
       fee: feeSubject
     }).subscribe({
-      next: _ref42 => {
+      next: _ref43 => {
         let {
           fee,
           freeBalance
-        } = _ref42;
+        } = _ref43;
         convertData(freeBalance, fee, _feeOptions, feeCustom).then(cb).catch(console.error);
       }
     });
-    const [unsubBalance, freeBalance] = await this.#koniState.balanceService.subscribeBalance(address, chain, token, 'transferable', _KoniTypes.ExtrinsicType.TRANSFER_BALANCE, data => {
+    const [unsubBalance, freeBalance] = await this.#koniState.balanceService.subscribeBalance(address, chain, token, _types4.BalanceType.TRANSFERABLE, _KoniTypes.ExtrinsicType.TRANSFER_BALANCE, data => {
       freeBalanceSubject.next(data); // Must be called after subscription
     });
 
@@ -2390,13 +2429,13 @@ class KoniExtension {
     });
     return convertData(freeBalance, fee, _feeOptions, feeCustom);
   }
-  async subscribeAddressTransferableBalance(_ref43, id, port) {
+  async subscribeAddressTransferableBalance(_ref44, id, port) {
     let {
       address,
       extrinsicType,
       networkKey,
       token
-    } = _ref43;
+    } = _ref44;
     const cb = (0, _subscriptions.createSubscription)(id, port);
     const convertData = data => {
       return {
@@ -2415,6 +2454,32 @@ class KoniExtension {
     });
     return convertData(currentFreeBalance);
   }
+  async subscribeAddressAvailableBalanceByType(_ref45, id, port) {
+    let {
+      address,
+      balanceType,
+      extrinsicType,
+      networkKey,
+      token
+    } = _ref45;
+    const cb = (0, _subscriptions.createSubscription)(id, port);
+    const convertData = data => {
+      return {
+        ...data,
+        id
+      };
+    };
+    const _cb = data => {
+      // eslint-disable-next-line node/no-callback-literal
+      cb(convertData(data));
+    };
+    const [unsub, currentFreeBalance] = await this.#koniState.balanceService.subscribeBalanceByType(address, networkKey, token, balanceType, extrinsicType, _cb);
+    this.createUnsubscriptionHandle(id, unsub);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return convertData(currentFreeBalance);
+  }
   async substrateNftSubmitTransaction(inputData) {
     const {
       params,
@@ -2447,11 +2512,11 @@ class KoniExtension {
       isSendingSelf
     };
   }
-  async enableChains(_ref44) {
+  async enableChains(_ref46) {
     let {
       chainSlugs,
       enableTokens
-    } = _ref44;
+    } = _ref46;
     try {
       await Promise.all(chainSlugs.map(chainSlug => this.enableChain({
         chainSlug,
@@ -2595,30 +2660,30 @@ class KoniExtension {
 
   // Parse transaction
 
-  parseSubstrateTransaction(_ref45) {
+  parseSubstrateTransaction(_ref47) {
     let {
       data,
       networkKey
-    } = _ref45;
+    } = _ref47;
     const apiProps = this.#koniState.getSubstrateApi(networkKey);
     const apiPromise = apiProps.api;
     return (0, _parseTransaction.parseSubstrateTransaction)(data, apiPromise);
   }
-  async parseEVMRLP(_ref46) {
+  async parseEVMRLP(_ref48) {
     let {
       data
-    } = _ref46;
+    } = _ref48;
     return await (0, _parseTransaction2.parseEvmRlp)(data, this.#koniState.getChainInfoMap(), this.#koniState.getEvmApiMap());
   }
 
   // Sign
 
-  qrSignSubstrate(_ref47) {
+  qrSignSubstrate(_ref49) {
     let {
       address,
       data,
       networkKey
-    } = _ref47;
+    } = _ref49;
     const pair = _uiKeyring.keyring.getPair(address);
     (0, _util.assert)(pair, (0, _i18next.t)('bg.koni.handler.Extension.unableToFindAccount'));
     if (pair.isLocked) {
@@ -2635,13 +2700,13 @@ class KoniExtension {
       signature: signed
     };
   }
-  async qrSignEVM(_ref48) {
+  async qrSignEVM(_ref50) {
     let {
       address,
       chainId,
       message,
       type
-    } = _ref48;
+    } = _ref50;
     let signed;
     const network = this.getNetworkJsonByChainId(chainId);
     if (!network) {
@@ -2725,11 +2790,11 @@ class KoniExtension {
     });
     return this.#koniState.getNominatorMetadata();
   }
-  async getBondingOptions(_ref49) {
+  async getBondingOptions(_ref51) {
     let {
       chain,
       type
-    } = _ref49;
+    } = _ref51;
     const apiProps = this.#koniState.getSubstrateApi(chain);
     const chainInfo = this.#koniState.getChainInfo(chain);
     const chainStakingMetadata = await this.#koniState.getStakingMetadataByChain(chain, type);
@@ -2919,12 +2984,12 @@ class KoniExtension {
   }
 
   // EVM Transaction
-  async parseContractInput(_ref50) {
+  async parseContractInput(_ref52) {
     let {
       chainId,
       contract,
       data
-    } = _ref50;
+    } = _ref52;
     const network = this.getNetworkJsonByChainId(chainId);
     return await (0, _parseTransaction2.parseContractInput)(data, contract, network);
   }
@@ -3024,10 +3089,10 @@ class KoniExtension {
 
   // Unlock wallet
 
-  keyringUnlock(_ref51) {
+  keyringUnlock(_ref53) {
     let {
       password
-    } = _ref51;
+    } = _ref53;
     try {
       _uiKeyring.keyring.unlockKeyring(password);
       // this.#koniState.initMantaPay(password)
@@ -3058,11 +3123,11 @@ class KoniExtension {
 
   // Export mnemonic
 
-  keyringExportMnemonic(_ref52) {
+  keyringExportMnemonic(_ref54) {
     let {
       address,
       password
-    } = _ref52;
+    } = _ref54;
     const pair = _uiKeyring.keyring.getPair(address);
     const result = pair.exportMnemonic(password);
     return {
@@ -3072,10 +3137,10 @@ class KoniExtension {
 
   // Reset wallet
 
-  async resetWallet(_ref53) {
+  async resetWallet(_ref55) {
     let {
       resetAll
-    } = _ref53;
+    } = _ref55;
     try {
       await this.#koniState.resetWallet(resetAll);
       return {
@@ -3091,10 +3156,10 @@ class KoniExtension {
   }
 
   // Signing substrate request
-  async signingApprovePasswordV2(_ref54) {
+  async signingApprovePasswordV2(_ref56) {
     let {
       id
-    } = _ref54;
+    } = _ref56;
     const queued = this.#koniState.getSignRequest(id);
     (0, _util.assert)(queued, (0, _i18next.t)('bg.koni.handler.Extension.unableToProceed'));
     const {
@@ -3253,10 +3318,10 @@ class KoniExtension {
   getSupportedSmartContractTypes() {
     return this.#koniState.getSupportedSmartContractTypes();
   }
-  getTransaction(_ref55) {
+  getTransaction(_ref57) {
     let {
       id
-    } = _ref55;
+    } = _ref57;
     // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
     const {
       transaction,
@@ -3267,8 +3332,8 @@ class KoniExtension {
   async subscribeTransactions(id, port) {
     const cb = (0, _subscriptions.createSubscription)(id, port);
     function convertRs(rs, processMap) {
-      return Object.fromEntries(Object.entries(rs).map(_ref56 => {
-        let [key, value] = _ref56;
+      return Object.fromEntries(Object.entries(rs).map(_ref58 => {
+        let [key, value] = _ref58;
         // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
         const {
           additionalValidator,
@@ -3294,11 +3359,11 @@ class KoniExtension {
     const subscription = (0, _rxjs.combineLatest)({
       transactions: transactionsObservable,
       processMap: processTransactionObservable
-    }).subscribe(_ref57 => {
+    }).subscribe(_ref59 => {
       let {
         processMap,
         transactions
-      } = _ref57;
+      } = _ref59;
       cb(convertRs(transactions, processMap));
     });
     port.onDisconnect.addListener(() => {
@@ -3319,10 +3384,10 @@ class KoniExtension {
     });
     return notificationSubject.value;
   }
-  async reloadCron(_ref58) {
+  async reloadCron(_ref60) {
     let {
       data
-    } = _ref58;
+    } = _ref60;
     if (data === 'nft') {
       return await this.#koniState.reloadNft();
     } else if (data === 'staking') {
@@ -3365,10 +3430,10 @@ class KoniExtension {
   }
 
   // Phishing detect
-  async passPhishingPage(_ref59) {
+  async passPhishingPage(_ref61) {
     let {
       url
-    } = _ref59;
+    } = _ref61;
     return await this.#koniState.approvePassPhishingPage(url);
   }
 
@@ -3389,10 +3454,10 @@ class KoniExtension {
   /// Wallet connect
 
   // Connect
-  async connectWalletConnect(_ref60) {
+  async connectWalletConnect(_ref62) {
     let {
       uri
-    } = _ref60;
+    } = _ref62;
     await this.#koniState.walletConnectService.connect(uri);
     return true;
   }
@@ -3405,11 +3470,11 @@ class KoniExtension {
     });
     return this.#koniState.requestService.allConnectWCRequests;
   }
-  async approveWalletConnectSession(_ref61) {
+  async approveWalletConnectSession(_ref63) {
     let {
       accounts: selectedAccounts,
       id
-    } = _ref61;
+    } = _ref63;
     const request = this.#koniState.requestService.getConnectWCRequest(id);
     if ((0, _helpers2.isProposalExpired)(request.request.params)) {
       throw new Error('The proposal has been expired');
@@ -3421,8 +3486,8 @@ class KoniExtension {
     const availableNamespaces = {};
     const namespaces = {};
     const chainInfoMap = this.#koniState.getChainInfoMap();
-    Object.entries(requiredNamespaces).forEach(_ref62 => {
-      let [key, namespace] = _ref62;
+    Object.entries(requiredNamespaces).forEach(_ref64 => {
+      let [key, namespace] = _ref64;
       if ((0, _helpers2.isSupportWalletConnectNamespace)(key)) {
         if (namespace.chains) {
           const unSupportChains = namespace.chains.filter(chain => !(0, _helpers2.isSupportWalletConnectChain)(chain, chainInfoMap));
@@ -3435,8 +3500,8 @@ class KoniExtension {
         throw new Error((0, _utils10.getSdkError)('UNSUPPORTED_NAMESPACE_KEY').message + ' ' + key);
       }
     });
-    Object.entries(optionalNamespaces).forEach(_ref63 => {
-      let [key, namespace] = _ref63;
+    Object.entries(optionalNamespaces).forEach(_ref65 => {
+      let [key, namespace] = _ref65;
       if ((0, _helpers2.isSupportWalletConnectNamespace)(key)) {
         if (namespace.chains) {
           const supportChains = namespace.chains.filter(chain => (0, _helpers2.isSupportWalletConnectChain)(chain, chainInfoMap)) || [];
@@ -3460,8 +3525,8 @@ class KoniExtension {
         }
       }
     });
-    Object.entries(availableNamespaces).forEach(_ref64 => {
-      let [key, namespace] = _ref64;
+    Object.entries(availableNamespaces).forEach(_ref66 => {
+      let [key, namespace] = _ref66;
       if (namespace.chains) {
         const accounts = selectedAccounts.filter(address => {
           const [_namespace] = address.split(':');
@@ -3485,10 +3550,10 @@ class KoniExtension {
     request.resolve();
     return true;
   }
-  async rejectWalletConnectSession(_ref65) {
+  async rejectWalletConnectSession(_ref67) {
     let {
       id
-    } = _ref65;
+    } = _ref67;
     const request = this.#koniState.requestService.getConnectWCRequest(id);
     const wcId = request.request.id;
     if ((0, _helpers2.isProposalExpired)(request.request.params)) {
@@ -3510,10 +3575,10 @@ class KoniExtension {
     });
     return this.#koniState.walletConnectService.sessions;
   }
-  async disconnectWalletConnectSession(_ref66) {
+  async disconnectWalletConnectSession(_ref68) {
     let {
       topic
-    } = _ref66;
+    } = _ref68;
     await this.#koniState.walletConnectService.disconnect(topic);
     return true;
   }
@@ -3526,18 +3591,18 @@ class KoniExtension {
     });
     return this.#koniState.requestService.allNotSupportWCRequests;
   }
-  approveWalletConnectNotSupport(_ref67) {
+  approveWalletConnectNotSupport(_ref69) {
     let {
       id
-    } = _ref67;
+    } = _ref69;
     const request = this.#koniState.requestService.getNotSupportWCRequest(id);
     request.resolve();
     return true;
   }
-  rejectWalletConnectNotSupport(_ref68) {
+  rejectWalletConnectNotSupport(_ref70) {
     let {
       id
-    } = _ref68;
+    } = _ref70;
     const request = this.#koniState.requestService.getNotSupportWCRequest(id);
     request.reject(new Error('USER_REJECTED'));
     return true;
@@ -3545,11 +3610,11 @@ class KoniExtension {
 
   /// Manta
 
-  async enableMantaPay(_ref69) {
+  async enableMantaPay(_ref71) {
     let {
       address,
       password
-    } = _ref69;
+    } = _ref71;
     // always takes the current account
     function timeout() {
       return new Promise(resolve => setTimeout(resolve, 1500));
@@ -3639,11 +3704,11 @@ class KoniExtension {
   async disableMantaPay(address) {
     return this.#koniState.disableMantaPay(address);
   }
-  async isTonBounceableAddress(_ref70) {
+  async isTonBounceableAddress(_ref72) {
     let {
       address,
       chain
-    } = _ref70;
+    } = _ref72;
     try {
       const tonApi = this.#koniState.getTonApi(chain);
       const state = await tonApi.getAccountState(address);
@@ -3689,10 +3754,10 @@ class KoniExtension {
 
   /* Metadata */
 
-  async findRawMetadata(_ref71) {
+  async findRawMetadata(_ref73) {
     let {
       genesisHash
-    } = _ref71;
+    } = _ref73;
     const {
       metadata,
       specVersion,
@@ -3706,20 +3771,20 @@ class KoniExtension {
       userExtensions
     };
   }
-  async calculateMetadataHash(_ref72) {
+  async calculateMetadataHash(_ref74) {
     let {
       chain
-    } = _ref72;
+    } = _ref74;
     const hash = await this.#koniState.calculateMetadataHash(chain);
     return {
       metadataHash: hash || ''
     };
   }
-  async shortenMetadata(_ref73) {
+  async shortenMetadata(_ref75) {
     let {
       chain,
       txBlob
-    } = _ref73;
+    } = _ref75;
     const shorten = await this.#koniState.shortenMetadata(chain, txBlob);
     return {
       txMetadata: shorten || ''
@@ -3890,6 +3955,10 @@ class KoniExtension {
         });
       }
     }
+    let balanceTypeForPool = _types4.BalanceType.TRANSFERABLE;
+    if (!!poolHandler && [_types4.YieldPoolType.NATIVE_STAKING, _types4.YieldPoolType.NOMINATION_POOL].includes(poolHandler.type)) {
+      balanceTypeForPool = _types4.BalanceType.TOTAL_MINUS_RESERVED;
+    }
     return await this.#koniState.transactionService.handleTransaction({
       address,
       chain: txChain,
@@ -3901,6 +3970,7 @@ class KoniExtension {
       chainType,
       resolveOnDone: !isLastStep,
       transferNativeAmount,
+      balanceType: balanceTypeForPool,
       skipFeeValidation: isMintingStep && isPoolSupportAlternativeFee,
       errorOnTimeOut,
       ...this.createPassConfirmationParams(isPassConfirmation),
@@ -4091,18 +4161,18 @@ class KoniExtension {
 
   /* Campaign */
 
-  unlockDotCheckCanMint(_ref74) {
+  unlockDotCheckCanMint(_ref76) {
     let {
       address,
       network,
       slug
-    } = _ref74;
+    } = _ref76;
     return this.#koniState.mintCampaignService.unlockDotCampaign.canMint(address, slug, network);
   }
-  unlockDotSubscribeMintedData(id, port, _ref75) {
+  unlockDotSubscribeMintedData(id, port, _ref77) {
     let {
       transactionId
-    } = _ref75;
+    } = _ref77;
     const cb = (0, _subscriptions.createSubscription)(id, port);
     const subscription = this.#koniState.mintCampaignService.unlockDotCampaign.subscribeMintedNft(transactionId, cb);
     this.createUnsubscriptionHandle(id, subscription.unsubscribe);
@@ -4134,10 +4204,10 @@ class KoniExtension {
     });
     return filterBanner(await this.#koniState.campaignService.getProcessingCampaign());
   }
-  async completeCampaignBanner(_ref76) {
+  async completeCampaignBanner(_ref78) {
     let {
       slug
-    } = _ref76;
+    } = _ref78;
     const campaign = await this.#koniState.dbService.getCampaign(slug);
     if (campaign) {
       await this.#koniState.dbService.upsertCampaign({
@@ -4672,8 +4742,8 @@ class KoniExtension {
                         resolve();
                       }
                     };
-                    this.#koniState.balanceService.subscribeTransferableBalance(address, waitXcmData.chain, waitXcmData.token, waitXcmData.nextTxType, onRs).then(_ref77 => {
-                      let [_unsub, rs] = _ref77;
+                    this.#koniState.balanceService.subscribeTransferableBalance(address, waitXcmData.chain, waitXcmData.token, waitXcmData.nextTxType, onRs).then(_ref79 => {
+                      let [_unsub, rs] = _ref79;
                       unsub = _unsub;
                       onRs(rs);
                     }).catch(console.error);
@@ -4903,6 +4973,8 @@ class KoniExtension {
         return await this.getNftCollection();
       case 'pri(nftCollection.getSubscription)':
         return await this.subscribeNftCollection(id, port);
+      case 'pri(nft.getFullList)':
+        return await this.handleGetNftFullList(request);
       case 'pri(staking.getStaking)':
         return this.getStaking();
       case 'pri(staking.getSubscription)':
@@ -5111,6 +5183,10 @@ class KoniExtension {
         return this.getAddressTransferableBalance(request);
       case 'pri(freeBalance.subscribe)':
         return this.subscribeAddressTransferableBalance(request, id, port);
+      case 'pri(availableBalance.getBalanceByType)':
+        return this.getAddressAvailableBalanceByType(request);
+      case 'pri(availableBalance.subscribeBalanceByType)':
+        return this.subscribeAddressAvailableBalanceByType(request, id, port);
       case 'pri(subscription.cancel)':
         return this.cancelSubscription(request);
       case 'pri(chainService.recoverSubstrateApi)':