@subwallet/extension-base 1.1.31-1 → 1.1.31-beta.0

package/koni/background/handlers/Extension.js

@@ -15,12 +15,13 @@ import { resolveAzeroAddressToDomain, resolveAzeroDomainToAddress } from '@subwa
 import { parseSubstrateTransaction } from '@subwallet/extension-base/koni/api/dotsama/parseTransaction';
 import { checkReferenceCount, checkSupportTransfer, createTransferExtrinsic } from '@subwallet/extension-base/koni/api/dotsama/transfer';
 import { getNftTransferExtrinsic, isRecipientSelf } from '@subwallet/extension-base/koni/api/nft/transfer';
-import { getBondingExtrinsic, getCancelWithdrawalExtrinsic, getClaimRewardExtrinsic, getNominationPoolsInfo, getUnbondingExtrinsic, getValidatorsInfo, getWithdrawalExtrinsic, validateBondingCondition, validateUnbondingCondition } from '@subwallet/extension-base/koni/api/staking/bonding';
+import { getBondingExtrinsic, getCancelWithdrawalExtrinsic, getClaimRewardExtrinsic, getNominationPoolsInfo, getUnbondingExtrinsic, getValidatorsInfo, validateBondingCondition, validateUnbondingCondition } from '@subwallet/extension-base/koni/api/staking/bonding';
 import { getTuringCancelCompoundingExtrinsic, getTuringCompoundExtrinsic } from '@subwallet/extension-base/koni/api/staking/bonding/paraChain';
 import { getPoolingBondingExtrinsic, getPoolingUnbondingExtrinsic, validatePoolBondingCondition, validateRelayUnbondingCondition } from '@subwallet/extension-base/koni/api/staking/bonding/relayChain';
 import { getERC20TransactionObject, getERC721Transaction, getEVMTransactionObject } from '@subwallet/extension-base/koni/api/tokens/evm/transfer';
 import { getPSP34TransferExtrinsic } from '@subwallet/extension-base/koni/api/tokens/wasm';
 import { createXcmExtrinsic } from '@subwallet/extension-base/koni/api/xcm';
+import { YIELD_EXTRINSIC_TYPES } from '@subwallet/extension-base/koni/api/yield/helper/utils';
 import { _API_OPTIONS_CHAIN_GROUP, _DEFAULT_MANTA_ZK_CHAIN, _MANTA_ZK_CHAIN_GROUP, _ZK_ASSET_PREFIX } from '@subwallet/extension-base/services/chain-service/constants';
 import { _ChainConnectionStatus } from '@subwallet/extension-base/services/chain-service/types';
 import { _getChainNativeTokenBasicInfo, _getContractAddressOfToken, _getEvmChainId, _getSubstrateGenesisHash, _getTokenMinAmount, _isAssetSmartContractNft, _isChainEvmCompatible, _isCustomAsset, _isLocalToken, _isMantaZkAsset, _isNativeToken, _isTokenEvmSmartContract, _isTokenTransferredByEvm } from '@subwallet/extension-base/services/chain-service/utils';
@@ -28,6 +29,7 @@ import { EXTENSION_REQUEST_URL } from '@subwallet/extension-base/services/reques
 import { DEFAULT_AUTO_LOCK_TIME } from '@subwallet/extension-base/services/setting-service/constants';
 import { WALLET_CONNECT_EIP155_NAMESPACE } from '@subwallet/extension-base/services/wallet-connect-service/constants';
 import { isProposalExpired, isSupportWalletConnectChain, isSupportWalletConnectNamespace } from '@subwallet/extension-base/services/wallet-connect-service/helpers';
+import { YieldPoolType } from '@subwallet/extension-base/types';
 import { convertSubjectInfoToAddresses, isSameAddress, reformatAddress, uniqueStringArray } from '@subwallet/extension-base/utils';
 import { createTransactionFromRLP, recalculateGasPrice, signatureToHex } from '@subwallet/extension-base/utils/eth';
 import { parseContractInput, parseEvmRlp } from '@subwallet/extension-base/utils/eth/parseTransaction';
@@ -468,6 +470,7 @@ export default class KoniExtension {
     const cb = createSubscription(id, port);
     const keyringService = this.#koniState.keyringService;
     await this.#koniState.eventService.waitAccountReady;
+    await this.#koniState.eventService.waitInjectReady;
     const currentAccount = keyringService.currentAccount;
     const transformedAccounts = transformAccounts(keyringService.accounts);
     const responseData = {
@@ -1022,6 +1025,9 @@ export default class KoniExtension {
   getCrowdloan(reset) {
     return this.#koniState.getCrowdloan(reset);
   }
+  getCrowdloanContributions(request) {
+    return this.#koniState.getCrowdloanContributions(request);
+  }
   subscribeCrowdloan(id, port) {
     const cb = createSubscription(id, port);
     const crowdloanSubscription = this.#koniState.subscribeCrowdloan().subscribe({
@@ -1910,12 +1916,22 @@ export default class KoniExtension {
     token
   }, id, port) {
     const cb = createSubscription(id, port);
-    const [unsub, currentFreeBalance] = await this.#koniState.balanceService.subscribeTokenFreeBalance(address, networkKey, token, cb);
+    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.subscribeTokenFreeBalance(address, networkKey, token, _cb);
     this.createUnsubscriptionHandle(id, unsub);
     port.onDisconnect.addListener(() => {
       this.cancelSubscription(id);
     });
-    return currentFreeBalance;
+    return convertData(currentFreeBalance);
   }
   async transferCheckReferenceCount({
     address,
@@ -2530,36 +2546,18 @@ export default class KoniExtension {
       chainType: ChainType.SUBSTRATE
     });
   }
-  async submitStakeWithdrawal(inputData) {
-    const {
-      chain,
-      nominatorMetadata,
-      validatorAddress
-    } = inputData;
-    if (!nominatorMetadata) {
-      return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
-    }
-    const dotSamaApi = this.#koniState.getSubstrateApi(chain);
-    const extrinsic = await getWithdrawalExtrinsic(dotSamaApi, chain, nominatorMetadata, validatorAddress);
-    return await this.#koniState.transactionService.handleTransaction({
-      address: nominatorMetadata.address,
-      chain: chain,
-      transaction: extrinsic,
-      data: inputData,
-      extrinsicType: ExtrinsicType.STAKING_WITHDRAW,
-      chainType: ChainType.SUBSTRATE
-    });
-  }
   async submitStakeClaimReward(inputData) {
     const {
       address,
       bondReward,
-      chain,
-      stakingType
+      slug
     } = inputData;
-    if (!address) {
+    const poolHandler = this.#koniState.earningService.getPoolHandler(slug);
+    if (!address || !poolHandler) {
       return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
     }
+    const chain = poolHandler.chain;
+    const stakingType = poolHandler.type === YieldPoolType.NOMINATION_POOL ? StakingType.POOLED : StakingType.NOMINATED;
     const substrateApi = this.#koniState.getSubstrateApi(chain);
     const extrinsic = await getClaimRewardExtrinsic(substrateApi, chain, address, stakingType, bondReward);
     return await this.#koniState.transactionService.handleTransaction({
@@ -2572,15 +2570,18 @@ export default class KoniExtension {
     });
   }
   async submitCancelStakeWithdrawal(inputData) {
+    var _this$koniState$earni;
     const {
       address,
-      chain,
-      selectedUnstaking
+      selectedUnstaking,
+      slug
     } = inputData;
+    const chain = (_this$koniState$earni = this.#koniState.earningService.getPoolHandler(slug)) === null || _this$koniState$earni === void 0 ? void 0 : _this$koniState$earni.chain;
     if (!chain || !selectedUnstaking) {
       return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
     }
     const substrateApi = this.#koniState.getSubstrateApi(chain);
+    // @ts-ignore
     const extrinsic = await getCancelWithdrawalExtrinsic(substrateApi, chain, selectedUnstaking);
     return await this.#koniState.transactionService.handleTransaction({
       address,
@@ -3553,9 +3554,237 @@ export default class KoniExtension {
     this.#koniState.keyringService.removeInjectAccounts(request.addresses);
     return true;
   }
+  subscribeYieldPoolInfo(id, port) {
+    const cb = createSubscription(id, port);
+    const yieldPoolSubscription = this.#koniState.subscribeYieldPoolInfo().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, yieldPoolSubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.getYieldPoolInfo();
+  }
+  async earlyValidateJoin(request) {
+    return await this.#koniState.earningService.earlyValidateJoin(request);
+  }
+  async getOptimalYieldPath(request) {
+    return await this.#koniState.earningService.generateOptimalSteps(request);
+  }
+  async handleYieldStep(inputData) {
+    const {
+      data,
+      path
+    } = inputData;
+    const {
+      address
+    } = data;
+    if (!data) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INTERNAL_ERROR)]);
+    }
+    const isLastStep = inputData.currentStep + 1 === path.steps.length;
+    const yieldValidation = await this.#koniState.earningService.validateYieldJoin({
+      data,
+      path
+    }); // TODO: validate, set to fail upon submission
+
+    if (yieldValidation.length > 0) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors(yieldValidation);
+    }
+
+    // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+    const {
+      extrinsic,
+      extrinsicType,
+      transferNativeAmount,
+      txChain,
+      txData
+    } = await this.#koniState.earningService.handleYieldJoin(inputData);
+    const isPoolSupportAlternativeFee = this.#koniState.earningService.isPoolSupportAlternativeFee(inputData.data.slug);
+    const isMintingStep = YIELD_EXTRINSIC_TYPES.includes(extrinsicType);
+    const chainInfo = this.#koniState.getChainInfo(txChain);
+    return await this.#koniState.transactionService.handleTransaction({
+      address,
+      chain: txChain,
+      transaction: extrinsic,
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+      data: txData,
+      extrinsicType,
+      // change this depends on step
+      chainType: _isChainEvmCompatible(chainInfo) ? ChainType.EVM : ChainType.SUBSTRATE,
+      resolveOnDone: !isLastStep,
+      transferNativeAmount,
+      skipFeeValidation: isMintingStep && isPoolSupportAlternativeFee
+    });
+  }
+  async handleYieldLeave(params) {
+    const {
+      address,
+      slug
+    } = params;
+    const leaveValidation = await this.#koniState.earningService.validateYieldLeave(params);
+    if (leaveValidation.length > 0) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors(leaveValidation);
+    }
+    const [extrinsicType, extrinsic] = await this.#koniState.earningService.handleYieldLeave(params);
+    const handler = this.#koniState.earningService.getPoolHandler(slug);
+    return await this.#koniState.transactionService.handleTransaction({
+      address,
+      chain: (handler === null || handler === void 0 ? void 0 : handler.chain) || '',
+      transaction: extrinsic,
+      data: params,
+      // TODO
+      extrinsicType,
+      chainType: (handler === null || handler === void 0 ? void 0 : handler.transactionChainType) || ChainType.SUBSTRATE
+    });
+  }
+  async getYieldPoolTargets(request) {
+    const {
+      slug
+    } = request;
+    const targets = await this.#koniState.earningService.getPoolTargets(slug);
+    return {
+      slug,
+      targets
+    };
+  }
+  async subscribeYieldPosition(id, port) {
+    const cb = createSubscription(id, port);
+    const yieldPositionSubscription = this.#koniState.subscribeYieldPosition().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, yieldPositionSubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.getYieldPositionInfo();
+  }
+  subscribeYieldReward(id, port) {
+    const cb = createSubscription(id, port);
+    const stakingRewardSubscription = this.#koniState.earningService.subscribeEarningReward().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, stakingRewardSubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.earningService.getEarningRewards();
+  }
+  subscribeYieldRewardHistory(id, port) {
+    const cb = createSubscription(id, port);
+    const rewardHistorySubscription = this.#koniState.earningService.subscribeEarningRewardHistory().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, rewardHistorySubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.earningService.getEarningRewardHistory();
+  }
+  subscribeEarningMinAmountPercent(id, port) {
+    const cb = createSubscription(id, port);
+    const earningMinAmountPercentSubscription = this.#koniState.earningService.subscribeMinAmountPercent().subscribe({
+      next: rs => {
+        cb(rs);
+      }
+    });
+    this.createUnsubscriptionHandle(id, earningMinAmountPercentSubscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.earningService.getMinAmountPercent();
+  }
+  handleValidateYieldProcess(inputData) {
+    return this.#koniState.earningService.validateYieldJoin(inputData);
+  }
+  async yieldSubmitWithdrawal(params) {
+    const {
+      address,
+      slug
+    } = params;
+    const poolHandler = this.#koniState.earningService.getPoolHandler(slug);
+    if (!poolHandler) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
+    }
+    const extrinsic = await this.#koniState.earningService.handleYieldWithdraw(params);
+    return await this.#koniState.transactionService.handleTransaction({
+      address: address,
+      chain: poolHandler.chain,
+      transaction: extrinsic,
+      data: params,
+      extrinsicType: ExtrinsicType.STAKING_WITHDRAW,
+      chainType: (poolHandler === null || poolHandler === void 0 ? void 0 : poolHandler.transactionChainType) || ChainType.SUBSTRATE
+    });
+  }
+  async yieldSubmitCancelWithdrawal(params) {
+    const {
+      address,
+      selectedUnstaking,
+      slug
+    } = params;
+    const poolHandler = this.#koniState.earningService.getPoolHandler(slug);
+    if (!poolHandler || !selectedUnstaking) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
+    }
+    const chain = poolHandler.chain;
+    const extrinsic = await this.#koniState.earningService.handleYieldCancelUnstake(params);
+    return await this.#koniState.transactionService.handleTransaction({
+      address,
+      chain,
+      transaction: extrinsic,
+      data: params,
+      extrinsicType: ExtrinsicType.STAKING_CANCEL_UNSTAKE,
+      chainType: (poolHandler === null || poolHandler === void 0 ? void 0 : poolHandler.transactionChainType) || ChainType.SUBSTRATE
+    });
+  }
+  async yieldSubmitClaimReward(params) {
+    const {
+      address,
+      slug
+    } = params;
+    const poolHandler = this.#koniState.earningService.getPoolHandler(slug);
+    if (!address || !poolHandler) {
+      return this.#koniState.transactionService.generateBeforeHandleResponseErrors([new TransactionError(BasicTxErrorType.INVALID_PARAMS)]);
+    }
+    const extrinsic = await this.#koniState.earningService.handleYieldClaimReward(params);
+    return await this.#koniState.transactionService.handleTransaction({
+      address,
+      chain: poolHandler.chain,
+      transaction: extrinsic,
+      data: params,
+      extrinsicType: ExtrinsicType.STAKING_CLAIM_REWARD,
+      chainType: (poolHandler === null || poolHandler === void 0 ? void 0 : poolHandler.transactionChainType) || ChainType.SUBSTRATE
+    });
+  }
 
   /* Campaign */
 
+  unlockDotCheckCanMint({
+    address,
+    network,
+    slug
+  }) {
+    return this.#koniState.mintCampaignService.unlockDotCampaign.canMint(address, slug, network);
+  }
+  unlockDotSubscribeMintedData(id, port, {
+    transactionId
+  }) {
+    const cb = createSubscription(id, port);
+    const subscription = this.#koniState.mintCampaignService.unlockDotCampaign.subscribeMintedNft(transactionId, cb);
+    this.createUnsubscriptionHandle(id, subscription.unsubscribe);
+    port.onDisconnect.addListener(() => {
+      this.cancelSubscription(id);
+    });
+    return this.#koniState.mintCampaignService.unlockDotCampaign.getMintedNft(transactionId);
+  }
   async subscribeProcessingBanner(id, port) {
     const cb = createSubscription(id, port);
     const filterBanner = data => {
@@ -3758,6 +3987,8 @@ export default class KoniExtension {
         return await this.subscribeBalance(id, port);
       case 'pri(crowdloan.getCrowdloan)':
         return this.getCrowdloan();
+      case 'pri(crowdloan.getCrowdloanContributions)':
+        return this.getCrowdloanContributions(request);
       case 'pri(crowdloan.getSubscription)':
         return this.subscribeCrowdloan(id, port);
       case 'pri(derivation.createV2)':
@@ -3787,6 +4018,57 @@ export default class KoniExtension {
       case 'pri(transaction.history.subscribe)':
         return this.subscribeHistoryByChainAndAddress(request, id, port);
 
+      /* Earning */
+
+      /* Info */
+
+      case 'pri(yield.subscribePoolInfo)':
+        return this.subscribeYieldPoolInfo(id, port);
+      case 'pri(yield.getTargets)':
+        return this.getYieldPoolTargets(request);
+      case 'pri(yield.subscribeYieldPosition)':
+        return this.subscribeYieldPosition(id, port);
+      case 'pri(yield.subscribeYieldReward)':
+        return this.subscribeYieldReward(id, port);
+      case 'pri(yield.subscribeRewardHistory)':
+        return this.subscribeYieldRewardHistory(id, port);
+      case 'pri(yield.minAmountPercent)':
+        return this.subscribeEarningMinAmountPercent(id, port);
+
+      /* Info */
+
+      /* Actions */
+
+      /* Join */
+
+      case 'pri(yield.join.earlyValidate)':
+        return await this.earlyValidateJoin(request);
+      case 'pri(yield.join.getOptimalPath)':
+        return await this.getOptimalYieldPath(request);
+      case 'pri(yield.join.handleStep)':
+        return await this.handleYieldStep(request);
+      case 'pri(yield.join.validateProcess)':
+        return await this.handleValidateYieldProcess(request);
+
+      /* Join */
+
+      /* Others */
+
+      case 'pri(yield.leave.submit)':
+        return await this.handleYieldLeave(request);
+      case 'pri(yield.withdraw.submit)':
+        return await this.yieldSubmitWithdrawal(request);
+      case 'pri(yield.cancelWithdrawal.submit)':
+        return await this.yieldSubmitCancelWithdrawal(request);
+      case 'pri(yield.claimReward.submit)':
+        return await this.yieldSubmitClaimReward(request);
+
+      /* Others */
+
+      /* Actions */
+
+      /* Earning */
+
       /* Account management */
       // Add account
       case 'pri(accounts.create.suriV2)':
@@ -3905,6 +4187,8 @@ export default class KoniExtension {
         return this.transferGetExistentialDeposit(request);
       case 'pri(transfer.getMaxTransferable)':
         return this.transferGetMaxTransferable(request);
+      case 'pri(transfer.subscribeMaxTransferable)':
+        return this.transferGetMaxTransferable(request);
       case 'pri(freeBalance.get)':
         return this.getAddressFreeBalance(request);
       case 'pri(freeBalance.subscribe)':
@@ -3963,8 +4247,6 @@ export default class KoniExtension {
         return await this.submitBonding(request);
       case 'pri(unbonding.submitTransaction)':
         return await this.submitUnbonding(request);
-      case 'pri(unbonding.submitWithdrawal)':
-        return await this.submitStakeWithdrawal(request);
       case 'pri(staking.submitClaimReward)':
         return await this.submitStakeClaimReward(request);
       case 'pri(staking.submitCancelWithdrawal)':
@@ -4068,6 +4350,15 @@ export default class KoniExtension {
       case 'pri(mantaPay.subscribeSyncingState)':
         return this.subscribeMantaPaySyncState(id, port);
 
+      /* Campaign */
+
+      case 'pri(campaign.unlockDot.canMint)':
+        return this.unlockDotCheckCanMint(request);
+      case 'pri(campaign.unlockDot.subscribe)':
+        return this.unlockDotSubscribeMintedData(id, port, request);
+
+      /* Campaign */
+
       // Metadata
       case 'pri(metadata.find)':
         return this.findRawMetadata(request);

package/koni/background/handlers/State.d.ts

@@ -1,16 +1,18 @@
 /// <reference types="chrome" />
 import { _AssetRef, _AssetType, _ChainAsset, _ChainInfo, _MultiChainAsset } from '@subwallet/chain-list/types';
-import { AddTokenRequestExternal, AmountData, ApiMap, AuthRequestV2, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, MantaPayConfig, MantaPaySyncState, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, UiSettings } from '@subwallet/extension-base/background/KoniTypes';
+import { AddTokenRequestExternal, AmountData, ApiMap, AuthRequestV2, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, MantaPayConfig, MantaPaySyncState, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestCrowdloanContributions, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, UiSettings } from '@subwallet/extension-base/background/KoniTypes';
 import { AccountJson, RequestAuthorizeTab, RequestRpcSend, RequestRpcSubscribe, RequestRpcUnsubscribe, RequestSign, ResponseRpcListProviders, ResponseSigning } from '@subwallet/extension-base/background/types';
 import { BalanceService } from '@subwallet/extension-base/services/balance-service';
 import BuyService from '@subwallet/extension-base/services/buy-service';
 import CampaignService from '@subwallet/extension-base/services/campaign-service';
 import { ChainService } from '@subwallet/extension-base/services/chain-service';
 import { _ChainState, _NetworkUpsertParams, _ValidateCustomAssetRequest } from '@subwallet/extension-base/services/chain-service/types';
+import EarningService from '@subwallet/extension-base/services/earning-service/service';
 import { EventService } from '@subwallet/extension-base/services/event-service';
 import { HistoryService } from '@subwallet/extension-base/services/history-service';
 import { KeyringService } from '@subwallet/extension-base/services/keyring-service';
 import MigrationService from '@subwallet/extension-base/services/migration-service';
+import MintCampaignService from '@subwallet/extension-base/services/mint-campaign-service';
 import NotificationService from '@subwallet/extension-base/services/notification-service/NotificationService';
 import { PriceService } from '@subwallet/extension-base/services/price-service';
 import RequestService from '@subwallet/extension-base/services/request-service';
@@ -20,7 +22,7 @@ import DatabaseService from '@subwallet/extension-base/services/storage-service/
 import { SubscanService } from '@subwallet/extension-base/services/subscan-service';
 import TransactionService from '@subwallet/extension-base/services/transaction-service';
 import WalletConnectService from '@subwallet/extension-base/services/wallet-connect-service';
-import { BalanceItem, BalanceJson, BalanceMap } from '@subwallet/extension-base/types';
+import { BalanceItem, BalanceJson, BalanceMap, YieldPoolInfo, YieldPoolType, YieldPositionInfo } from '@subwallet/extension-base/types';
 import { MetadataDef, ProviderMeta } from '@subwallet/extension-inject/types';
 import SimpleKeyring from 'eth-simple-keyring';
 import { BehaviorSubject, Subject } from 'rxjs';
@@ -46,6 +48,8 @@ export default class KoniState {
     private stakingNominatorMetadataSubject;
     private stakingRewardSubject;
     private stakingRewardState;
+    private yieldPoolInfoSubject;
+    private yieldPositionSubject;
     private lazyMap;
     readonly notificationService: NotificationService;
     readonly eventService: EventService;
@@ -64,8 +68,10 @@ export default class KoniState {
     readonly migrationService: MigrationService;
     readonly subscanService: SubscanService;
     readonly walletConnectService: WalletConnectService;
+    readonly mintCampaignService: MintCampaignService;
     readonly campaignService: CampaignService;
     readonly buyService: BuyService;
+    readonly earningService: EarningService;
     private generalStatus;
     private waitSleeping;
     private waitStarting;
@@ -108,6 +114,11 @@ export default class KoniState {
     getStaking(): Promise<StakingJson>;
     getStakingOwnersByChains(chains: string[]): Promise<string[]>;
     getPooledStakingRecordsByAddress(addresses: string[]): Promise<StakingItem[]>;
+    getPooledPositionByAddress(addresses: string[]): Promise<YieldPositionInfo[]>;
+    subscribeYieldPoolInfo(): Subject<YieldPoolInfo[]>;
+    subscribeYieldPosition(): Subject<YieldPositionInfo[]>;
+    getYieldPoolInfo(): Promise<YieldPoolInfo[]>;
+    getYieldPositionInfo(): Promise<YieldPositionInfo[]>;
     subscribeMantaPayConfig(): Subject<MantaPayConfig[]>;
     subscribeStaking(): Subject<StakingJson>;
     subscribeChainStakingMetadata(): Subject<ChainStakingMetadata[]>;
@@ -251,5 +262,10 @@ export default class KoniState {
         metadata: string;
         specVersion: number;
     }>;
+    updateYieldPoolInfo(data: YieldPoolInfo): void;
+    resetYieldPoolInfo(chains: string[]): void;
+    updateYieldPosition(data: YieldPositionInfo): void;
+    getYieldPoolStakingInfo(chain: string, poolType: YieldPoolType): Promise<YieldPoolInfo | undefined>;
+    getCrowdloanContributions({ address, page, relayChain }: RequestCrowdloanContributions): Promise<import("../../../services/subscan-service/types").CrowdloanContributionsResponse>;
 }
 export {};

package/koni/background/handlers/State.js

@@ -14,10 +14,12 @@ import CampaignService from '@subwallet/extension-base/services/campaign-service
 import { ChainService } from '@subwallet/extension-base/services/chain-service';
 import { _DEFAULT_MANTA_ZK_CHAIN, _MANTA_ZK_CHAIN_GROUP, _PREDEFINED_SINGLE_MODES } from '@subwallet/extension-base/services/chain-service/constants';
 import { _getEvmChainId, _getSubstrateGenesisHash, _getTokenOnChainAssetId, _isAssetFungibleToken, _isChainEnabled, _isChainTestNet, _parseMetadataForSmartContractAsset } from '@subwallet/extension-base/services/chain-service/utils';
+import EarningService from '@subwallet/extension-base/services/earning-service/service';
 import { EventService } from '@subwallet/extension-base/services/event-service';
 import { HistoryService } from '@subwallet/extension-base/services/history-service';
 import { KeyringService } from '@subwallet/extension-base/services/keyring-service';
 import MigrationService from '@subwallet/extension-base/services/migration-service';
+import MintCampaignService from '@subwallet/extension-base/services/mint-campaign-service';
 import NotificationService from '@subwallet/extension-base/services/notification-service/NotificationService';
 import { PriceService } from '@subwallet/extension-base/services/price-service';
 import RequestService from '@subwallet/extension-base/services/request-service';
@@ -75,6 +77,10 @@ export default class KoniState {
     ready: false,
     data: {}
   };
+
+  // earning
+  yieldPoolInfoSubject = new Subject();
+  yieldPositionSubject = new Subject();
   lazyMap = {};
   // Handle the general status of the extension
   generalStatus = ServiceStatus.INITIALIZING;
@@ -93,11 +99,13 @@ export default class KoniState {
     this.priceService = new PriceService(this.dbService, this.eventService, this.chainService);
     this.balanceService = new BalanceService(this);
     this.historyService = new HistoryService(this.dbService, this.chainService, this.eventService, this.keyringService, this.subscanService);
-    this.transactionService = new TransactionService(this.chainService, this.eventService, this.requestService, this.balanceService, this.historyService, this.notificationService, this.dbService);
+    this.mintCampaignService = new MintCampaignService(this);
     this.walletConnectService = new WalletConnectService(this, this.requestService);
     this.migrationService = new MigrationService(this, this.eventService);
     this.campaignService = new CampaignService(this);
     this.buyService = new BuyService(this);
+    this.transactionService = new TransactionService(this);
+    this.earningService = new EarningService(this);
     this.subscription = new KoniSubscription(this, this.dbService);
     this.cron = new KoniCron(this, this.subscription, this.dbService);
     this.logger = createLogger('State');
@@ -240,6 +248,7 @@ export default class KoniState {
   }
   async startSubscription() {
     await this.eventService.waitKeyringReady;
+    await this.eventService.waitAssetReady;
     this.dbService.subscribeChainStakingMetadata([], data => {
       this.chainStakingMetadataSubject.next(data);
     });
@@ -250,8 +259,8 @@ export default class KoniState {
     this.keyringService.accountSubject.subscribe(accounts => {
       // TODO: improve this
       unsub && unsub.unsubscribe();
-      unsub = this.dbService.subscribeNominatorMetadata(Object.keys(accounts), data => {
-        this.stakingNominatorMetadataSubject.next(data);
+      unsub = this.dbService.subscribeYieldPosition(Object.keys(accounts), data => {
+        this.yieldPositionSubject.next(data);
       });
     });
   }
@@ -343,6 +352,22 @@ export default class KoniState {
   async getPooledStakingRecordsByAddress(addresses) {
     return this.dbService.getPooledStakings(addresses, this.activeChainSlugs);
   }
+  async getPooledPositionByAddress(addresses) {
+    return this.dbService.getYieldNominationPoolPosition(addresses, this.activeChainSlugs);
+  }
+  subscribeYieldPoolInfo() {
+    return this.yieldPoolInfoSubject;
+  }
+  subscribeYieldPosition() {
+    return this.yieldPositionSubject;
+  }
+  getYieldPoolInfo() {
+    return this.dbService.getYieldPools();
+  }
+  getYieldPositionInfo() {
+    const addresses = this.getDecodedAddresses(this.keyringService.currentAccount.address);
+    return this.dbService.getYieldPositionByAddress(addresses);
+  }
   subscribeMantaPayConfig() {
     return this.mantaPayConfigSubject;
   }
@@ -1741,4 +1766,26 @@ export default class KoniState {
       specVersion: parseInt((metadata === null || metadata === void 0 ? void 0 : metadata.specVersion) || '0')
     };
   }
+  updateYieldPoolInfo(data) {
+    this.dbService.updateYieldPoolStore(data).catch(e => this.logger.warn(e));
+  }
+  resetYieldPoolInfo(chains) {
+    this.dbService.subscribeYieldPoolInfo(chains, data => {
+      // TODO: no unsub
+      this.yieldPoolInfoSubject.next(data);
+    });
+  }
+  updateYieldPosition(data) {
+    this.dbService.updateYieldPosition(data).catch(e => this.logger.warn(e));
+  }
+  getYieldPoolStakingInfo(chain, poolType) {
+    return this.dbService.getYieldPoolStakingInfo(chain, poolType);
+  }
+  getCrowdloanContributions({
+    address,
+    page,
+    relayChain
+  }) {
+    return this.subscanService.getCrowdloanContributions(relayChain, address, page);
+  }
 }

package/koni/background/handlers/index.js

@@ -58,7 +58,8 @@ export default function handlers({
     assert(port, 'Port has been disconnected');
     port.postMessage({
       id,
-      response
+      response,
+      sender: 'BACKGROUND'
     });
   }).catch(error => {
     console.error(error);
@@ -70,7 +71,8 @@ export default function handlers({
         error: error.message,
         errorCode: error.code,
         errorData: error.data,
-        id
+        id,
+        sender: 'BACKGROUND'
       });
     }
   });