@subwallet/extension-base 1.0.1 → 1.0.2-1b

package/cjs/koni/api/staking/bonding/relayChain.js

@@ -16,24 +16,105 @@ exports.getRelayPoolsInfo = getRelayPoolsInfo;
 exports.getRelayUnbondingExtrinsic = getRelayUnbondingExtrinsic;
 exports.getRelayValidatorsInfo = getRelayValidatorsInfo;
 exports.getRelayWithdrawalExtrinsic = getRelayWithdrawalExtrinsic;
+exports.validatePoolBondingCondition = validatePoolBondingCondition;
+exports.validateRelayBondingCondition = validateRelayBondingCondition;
+exports.validateRelayUnbondingCondition = validateRelayUnbondingCondition;
+var _TransactionError = require("@subwallet/extension-base/background/errors/TransactionError");
 var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
 var _utils = require("@subwallet/extension-base/koni/api/staking/bonding/utils");
 var _constants = require("@subwallet/extension-base/services/chain-service/constants");
+var _utils2 = require("@subwallet/extension-base/services/chain-service/utils");
+var _utils3 = require("@subwallet/extension-base/utils");
 var _util = require("@polkadot/util");
 var _utilCrypto = require("@polkadot/util-crypto");
 // Copyright 2019-2022 @subwallet/extension-koni authors & contributors
 // SPDX-License-Identifier: Apache-2.0
 
-async function getRelayChainStakingMetadata(chain, substrateApi) {
-  var _chainApi$api$query$a, _chainApi$api$query, _chainApi$api$query$n;
+function validateRelayUnbondingCondition(amount, chainStakingMetadata, nominatorMetadata) {
+  const errors = [];
+  const bnActiveStake = new _util.BN(nominatorMetadata.activeStake);
+  const bnRemainingStake = bnActiveStake.sub(new _util.BN(amount));
+  const minStake = new _util.BN(chainStakingMetadata.minPoolBonding || '0');
+  if (!(bnRemainingStake.isZero() || bnRemainingStake.gte(minStake))) {
+    errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.INVALID_ACTIVE_STAKE));
+  }
+  if (nominatorMetadata.unstakings.length > chainStakingMetadata.maxWithdrawalRequestPerValidator) {
+    errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.EXCEED_MAX_UNSTAKING));
+  }
+  return errors;
+}
+function validatePoolBondingCondition(chainInfo, amount, selectedPool, address, chainStakingMetadata, nominatorMetadata) {
+  // cannot stake when unstake all
+  // amount >= min stake
+  const errors = [];
+  let bnTotalStake = new _util.BN(amount);
+  const bnMinStake = new _util.BN(chainStakingMetadata.minPoolBonding || '0');
+  if (nominatorMetadata) {
+    const bnCurrentActiveStake = new _util.BN(nominatorMetadata.activeStake);
+    bnTotalStake = bnTotalStake.add(bnCurrentActiveStake);
+    if (!bnCurrentActiveStake.gt(_util.BN_ZERO)) {
+      errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.EXIST_UNSTAKING_REQUEST));
+    }
+  }
+  if (!bnTotalStake.gte(bnMinStake)) {
+    errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.NOT_ENOUGH_MIN_STAKE));
+  }
+  return errors;
+}
+function validateRelayBondingCondition(chainInfo, amount, selectedValidators, address, chainStakingMetadata, nominatorMetadata) {
+  const errors = [];
+  let bnTotalStake = new _util.BN(amount);
+  const bnMinStake = new _util.BN(chainStakingMetadata.minStake);
+  if (!nominatorMetadata) {
+    if (!bnTotalStake.gte(bnMinStake)) {
+      errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.NOT_ENOUGH_MIN_STAKE));
+    }
+    if (selectedValidators.length > chainStakingMetadata.maxValidatorPerNominator) {
+      errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.EXCEED_MAX_NOMINATIONS));
+    }
+    return errors;
+  }
+  const bnCurrentActiveStake = new _util.BN(nominatorMetadata.activeStake);
+  bnTotalStake = bnTotalStake.add(bnCurrentActiveStake);
+  if (!bnTotalStake.gte(bnMinStake)) {
+    errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.NOT_ENOUGH_MIN_STAKE));
+  }
+  if (selectedValidators.length > chainStakingMetadata.maxValidatorPerNominator) {
+    errors.push(new _TransactionError.TransactionError(_KoniTypes.StakingTxErrorType.EXCEED_MAX_NOMINATIONS));
+  }
+  return errors;
+}
+async function getRelayChainStakingMetadata(chainInfo, substrateApi) {
+  var _chainApi$api$query$a, _chainApi$api$query, _chainApi$api$query$s, _chainApi$api$query2, _chainApi$api$query2$, _chainApi$api$query3, _chainApi$api$query3$;
+  const chain = chainInfo.slug;
+  const {
+    decimals
+  } = (0, _utils2._getChainNativeTokenBasicInfo)(chainInfo);
   const chainApi = await substrateApi.isReady;
   const _era = await chainApi.api.query.staking.currentEra();
   const currentEra = _era.toString();
   const maxNominations = chainApi.api.consts.staking.maxNominations.toString();
   const maxUnlockingChunks = chainApi.api.consts.staking.maxUnlockingChunks.toString();
   const unlockingEras = chainApi.api.consts.staking.bondingDuration.toString();
-  const [_totalEraStake, _totalIssuance, _auctionCounter, _minimumActiveStake, _minPoolJoin] = await Promise.all([chainApi.api.query.staking.erasTotalStake(parseInt(currentEra)), chainApi.api.query.balances.totalIssuance(), (_chainApi$api$query$a = chainApi.api.query.auctions) === null || _chainApi$api$query$a === void 0 ? void 0 : _chainApi$api$query$a.auctionCounter(), chainApi.api.query.staking.minimumActiveStake(), (_chainApi$api$query = chainApi.api.query) === null || _chainApi$api$query === void 0 ? void 0 : (_chainApi$api$query$n = _chainApi$api$query.nominationPools) === null || _chainApi$api$query$n === void 0 ? void 0 : _chainApi$api$query$n.minJoinBond()]);
-  const minStake = _minimumActiveStake.toString();
+  const [_totalEraStake, _totalIssuance, _auctionCounter, _minimumActiveStake, _minNominatorBond, _minPoolJoin, _eraStakers] = await Promise.all([chainApi.api.query.staking.erasTotalStake(parseInt(currentEra)), chainApi.api.query.balances.totalIssuance(), (_chainApi$api$query$a = chainApi.api.query.auctions) === null || _chainApi$api$query$a === void 0 ? void 0 : _chainApi$api$query$a.auctionCounter(), ((_chainApi$api$query = chainApi.api.query) === null || _chainApi$api$query === void 0 ? void 0 : (_chainApi$api$query$s = _chainApi$api$query.staking) === null || _chainApi$api$query$s === void 0 ? void 0 : _chainApi$api$query$s.minimumActiveStake) && ((_chainApi$api$query2 = chainApi.api.query) === null || _chainApi$api$query2 === void 0 ? void 0 : (_chainApi$api$query2$ = _chainApi$api$query2.staking) === null || _chainApi$api$query2$ === void 0 ? void 0 : _chainApi$api$query2$.minimumActiveStake()), chainApi.api.query.staking.minNominatorBond(), (_chainApi$api$query3 = chainApi.api.query) === null || _chainApi$api$query3 === void 0 ? void 0 : (_chainApi$api$query3$ = _chainApi$api$query3.nominationPools) === null || _chainApi$api$query3$ === void 0 ? void 0 : _chainApi$api$query3$.minJoinBond(), chainApi.api.query.staking.erasStakers.entries(parseInt(currentEra))]);
+  const eraStakers = _eraStakers;
+  const nominatorList = [];
+  for (const item of eraStakers) {
+    // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+    const rawValidatorStat = item[1].toHuman();
+    const eraNominators = rawValidatorStat.others;
+    for (const nominator of eraNominators) {
+      // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
+      if (!nominatorList.includes(nominator.who)) {
+        nominatorList.push(nominator.who);
+      }
+    }
+  }
+  const minActiveStake = (_minimumActiveStake === null || _minimumActiveStake === void 0 ? void 0 : _minimumActiveStake.toString()) || '0';
+  const minNominatorBond = _minNominatorBond.toString();
+  const bnMinActiveStake = new _util.BN(minActiveStake);
+  const bnMinNominatorBond = new _util.BN(minNominatorBond);
+  const minStake = bnMinActiveStake.gt(bnMinNominatorBond) ? bnMinActiveStake : bnMinNominatorBond;
   const minPoolJoin = (_minPoolJoin === null || _minPoolJoin === void 0 ? void 0 : _minPoolJoin.toString()) || undefined;
   const rawTotalEraStake = _totalEraStake.toString();
   const rawTotalIssuance = _totalIssuance.toString();
@@ -51,23 +132,34 @@ async function getRelayChainStakingMetadata(chain, substrateApi) {
     expectedReturn,
     // in %, annually
     inflation,
-    minStake,
+    minStake: minStake.toString(),
+    minPoolBonding: (10 ** decimals).toString(),
+    // default is 1
     maxValidatorPerNominator: parseInt(maxNominations),
     maxWithdrawalRequestPerValidator: parseInt(maxUnlockingChunks),
     allowCancelUnstaking: true,
     unstakingPeriod: unlockingPeriod,
-    minJoinNominationPool: minPoolJoin
+    minJoinNominationPool: minPoolJoin,
+    nominatorCount: nominatorList.length
   };
 }
 async function getRelayChainNominatorMetadata(chainInfo, address, substrateApi) {
+  var _chainApi$api$query4, _chainApi$api$query4$, _chainApi$api$query5, _chainApi$api$query5$, _chainApi$api$query6, _chainApi$api$query6$, _chainApi$api$query7, _chainApi$api$query7$, _chainApi$api$query8, _chainApi$api$query8$, _chainApi$api$query9, _chainApi$api$query9$, _chainApi$api$query10, _chainApi$api$query11;
   if ((0, _utilCrypto.isEthereumAddress)(address)) {
     return;
   }
   const chain = chainInfo.slug;
   const chainApi = await substrateApi.isReady;
-  const [_ledger, _nominations, _currentEra, _bonded] = await Promise.all([chainApi.api.query.staking.ledger(address), chainApi.api.query.staking.nominators(address), chainApi.api.query.staking.currentEra(), chainApi.api.query.staking.bonded(address)]);
-  const ledger = _ledger.toJSON();
-  const nominations = _nominations.toJSON();
+  const [_ledger, _nominations, _currentEra, _bonded, _minimumActiveStake, _minNominatorBond] = await Promise.all([(_chainApi$api$query4 = chainApi.api.query) === null || _chainApi$api$query4 === void 0 ? void 0 : (_chainApi$api$query4$ = _chainApi$api$query4.staking) === null || _chainApi$api$query4$ === void 0 ? void 0 : _chainApi$api$query4$.ledger(address), (_chainApi$api$query5 = chainApi.api.query) === null || _chainApi$api$query5 === void 0 ? void 0 : (_chainApi$api$query5$ = _chainApi$api$query5.staking) === null || _chainApi$api$query5$ === void 0 ? void 0 : _chainApi$api$query5$.nominators(address), (_chainApi$api$query6 = chainApi.api.query) === null || _chainApi$api$query6 === void 0 ? void 0 : (_chainApi$api$query6$ = _chainApi$api$query6.staking) === null || _chainApi$api$query6$ === void 0 ? void 0 : _chainApi$api$query6$.currentEra(), (_chainApi$api$query7 = chainApi.api.query) === null || _chainApi$api$query7 === void 0 ? void 0 : (_chainApi$api$query7$ = _chainApi$api$query7.staking) === null || _chainApi$api$query7$ === void 0 ? void 0 : _chainApi$api$query7$.bonded(address), ((_chainApi$api$query8 = chainApi.api.query) === null || _chainApi$api$query8 === void 0 ? void 0 : (_chainApi$api$query8$ = _chainApi$api$query8.staking) === null || _chainApi$api$query8$ === void 0 ? void 0 : _chainApi$api$query8$.minimumActiveStake) && ((_chainApi$api$query9 = chainApi.api.query) === null || _chainApi$api$query9 === void 0 ? void 0 : (_chainApi$api$query9$ = _chainApi$api$query9.staking) === null || _chainApi$api$query9$ === void 0 ? void 0 : _chainApi$api$query9$.minimumActiveStake()), (_chainApi$api$query10 = chainApi.api.query) === null || _chainApi$api$query10 === void 0 ? void 0 : (_chainApi$api$query11 = _chainApi$api$query10.staking) === null || _chainApi$api$query11 === void 0 ? void 0 : _chainApi$api$query11.minNominatorBond()]);
+  const minActiveStake = (_minimumActiveStake === null || _minimumActiveStake === void 0 ? void 0 : _minimumActiveStake.toString()) || '0';
+  const minNominatorBond = _minNominatorBond.toString();
+  const bnMinActiveStake = new _util.BN(minActiveStake);
+  const bnMinNominatorBond = new _util.BN(minNominatorBond);
+  const minStake = bnMinActiveStake.gt(bnMinNominatorBond) ? bnMinActiveStake : bnMinNominatorBond;
+  const _maxNominatorRewardedPerValidator = chainApi.api.consts.staking.maxNominatorRewardedPerValidator.toString();
+  const maxNominatorRewardedPerValidator = parseInt(_maxNominatorRewardedPerValidator);
+  const ledger = _ledger.toPrimitive();
+  const nominations = _nominations.toPrimitive();
   const currentEra = _currentEra.toString();
   const bonded = _bonded.toHuman();
   if (!ledger) {
@@ -79,17 +171,47 @@ async function getRelayChainNominatorMetadata(chainInfo, address, substrateApi)
   if (nominations) {
     const validatorList = nominations.targets;
     await Promise.all(validatorList.map(async validatorAddress => {
-      const identityInfo = (await chainApi.api.query.identity.identityOf(validatorAddress)).toHuman();
+      let nominationStatus = _KoniTypes.StakingStatus.NOT_EARNING;
+      const [_identityInfo, _eraStaker] = await Promise.all([chainApi.api.query.identity.identityOf(validatorAddress), chainApi.api.query.staking.erasStakers(currentEra, validatorAddress)]);
+      const eraStaker = _eraStaker.toPrimitive();
+      const identityInfo = _identityInfo.toHuman();
       const identity = (0, _utils.parseIdentity)(identityInfo);
+      const topNominators = eraStaker.others.map(nominator => {
+        return nominator.who;
+      });
+      if (!topNominators.includes((0, _utils3.reformatAddress)(address, (0, _utils2._getChainSubstrateAddressPrefix)(chainInfo)))) {
+        // if nominator has target but not in nominator list
+        nominationStatus = _KoniTypes.StakingStatus.WAITING;
+      } else if (topNominators.slice(0, maxNominatorRewardedPerValidator).includes((0, _utils3.reformatAddress)(address, (0, _utils2._getChainSubstrateAddressPrefix)(chainInfo)))) {
+        // if address in top nominators
+        nominationStatus = _KoniTypes.StakingStatus.EARNING_REWARD;
+      }
       nominationList.push({
         chain,
         validatorAddress,
+        status: nominationStatus,
         validatorIdentity: identity,
         activeStake: '0' // relaychain allocates stake accordingly
       });
     }));
   }
 
+  let stakingStatus = _KoniTypes.StakingStatus.NOT_EARNING;
+  const bnActiveStake = new _util.BN(activeStake);
+  let waitingNominationCount = 0;
+  if (bnActiveStake.gte(minStake)) {
+    for (const nomination of nominationList) {
+      if (nomination.status === _KoniTypes.StakingStatus.EARNING_REWARD) {
+        // only need 1 earning nomination to count
+        stakingStatus = _KoniTypes.StakingStatus.EARNING_REWARD;
+      } else if (nomination.status === _KoniTypes.StakingStatus.WAITING) {
+        waitingNominationCount += 1;
+      }
+    }
+    if (waitingNominationCount === nominationList.length) {
+      stakingStatus = _KoniTypes.StakingStatus.WAITING;
+    }
+  }
   ledger.unlocking.forEach(unlockingChunk => {
     const isClaimable = unlockingChunk.era - parseInt(currentEra) <= 0;
     const remainingEra = unlockingChunk.era - (parseInt(currentEra) + 1);
@@ -104,6 +226,7 @@ async function getRelayChainNominatorMetadata(chainInfo, address, substrateApi)
   return {
     chain,
     type: _KoniTypes.StakingType.NOMINATED,
+    status: stakingStatus,
     address: address,
     activeStake,
     nominations: nominationList,
@@ -114,17 +237,39 @@ async function getRelayChainNominatorMetadata(chainInfo, address, substrateApi)
 async function getRelayChainPoolMemberMetadata(chainInfo, address, substrateApi) {
   const chainApi = await substrateApi.isReady;
   const [_poolMemberInfo, _currentEra] = await Promise.all([chainApi.api.query.nominationPools.poolMembers(address), chainApi.api.query.staking.currentEra()]);
+  const _maxNominatorRewardedPerValidator = chainApi.api.consts.staking.maxNominatorRewardedPerValidator.toString();
+  const maxNominatorRewardedPerValidator = parseInt(_maxNominatorRewardedPerValidator);
+  const poolsPalletId = chainApi.api.consts.nominationPools.palletId.toString();
   const poolMemberInfo = _poolMemberInfo.toPrimitive();
   const currentEra = _currentEra.toString();
   if (!poolMemberInfo) {
     return;
   }
+  let stakingStatus = _KoniTypes.StakingStatus.NOT_EARNING;
   const _poolMetadata = await chainApi.api.query.nominationPools.metadata(poolMemberInfo.poolId);
   const poolMetadata = _poolMetadata.toPrimitive();
   const poolName = (0, _utils.transformPoolName)(poolMetadata.isUtf8 ? poolMetadata.toUtf8() : poolMetadata.toString());
+  const poolStashAccount = (0, _utils.parsePoolStashAddress)(chainApi.api, 0, poolMemberInfo.poolId, poolsPalletId);
+  const _nominations = await chainApi.api.query.staking.nominators(poolStashAccount);
+  const nominations = _nominations.toJSON();
+  if (nominations) {
+    const validatorList = nominations.targets;
+    await Promise.all(validatorList.map(async validatorAddress => {
+      const _eraStaker = await chainApi.api.query.staking.erasStakers(currentEra, validatorAddress);
+      const eraStaker = _eraStaker.toPrimitive();
+      const topNominators = eraStaker.others.map(nominator => {
+        return nominator.who;
+      }).slice(0, maxNominatorRewardedPerValidator);
+      if (topNominators.includes((0, _utils3.reformatAddress)(poolStashAccount, (0, _utils2._getChainSubstrateAddressPrefix)(chainInfo)))) {
+        // if address in top nominators
+        stakingStatus = _KoniTypes.StakingStatus.EARNING_REWARD;
+      }
+    }));
+  }
   const joinedPoolInfo = {
     activeStake: poolMemberInfo.points.toString(),
     chain: chainInfo.slug,
+    status: stakingStatus,
     validatorIdentity: poolName,
     validatorAddress: poolMemberInfo.poolId.toString(),
     // use poolId
@@ -143,10 +288,15 @@ async function getRelayChainPoolMemberMetadata(chainInfo, address, substrateApi)
       waitingTime: waitingTime > 0 ? waitingTime : 0
     });
   });
+  const bnActiveStake = new _util.BN(poolMemberInfo.points.toString());
+  if (!bnActiveStake.gt(_util.BN_ZERO)) {
+    stakingStatus = _KoniTypes.StakingStatus.NOT_EARNING;
+  }
   return {
     chain: chainInfo.slug,
     type: _KoniTypes.StakingType.POOLED,
     address,
+    status: stakingStatus,
     activeStake: poolMemberInfo.points.toString(),
     nominations: [joinedPoolInfo],
     // can only join 1 pool at a time
@@ -202,9 +352,9 @@ async function getRelayValidatorsInfo(chain, substrateApi, decimals, chainStakin
   }
   const extraInfoMap = {};
   await Promise.all(allValidators.map(async address => {
-    var _chainApi$api$query2, _chainApi$api$query2$, _identityInfo$judgeme;
+    var _chainApi$api$query12, _chainApi$api$query13, _identityInfo$judgeme;
     // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
-    const [_commissionInfo, _identityInfo] = await Promise.all([chainApi.api.query.staking.validators(address), (_chainApi$api$query2 = chainApi.api.query) === null || _chainApi$api$query2 === void 0 ? void 0 : (_chainApi$api$query2$ = _chainApi$api$query2.identity) === null || _chainApi$api$query2$ === void 0 ? void 0 : _chainApi$api$query2$.identityOf(address)]);
+    const [_commissionInfo, _identityInfo] = await Promise.all([chainApi.api.query.staking.validators(address), (_chainApi$api$query12 = chainApi.api.query) === null || _chainApi$api$query12 === void 0 ? void 0 : (_chainApi$api$query13 = _chainApi$api$query12.identity) === null || _chainApi$api$query13 === void 0 ? void 0 : _chainApi$api$query13.identityOf(address)]);
     const commissionInfo = _commissionInfo.toHuman();
     const identityInfo = _identityInfo ? _identityInfo.toHuman() : null;
     let identity;
@@ -255,23 +405,31 @@ async function getRelayPoolsInfo(chain, substrateApi) {
   }));
   return nominationPools;
 }
-async function getRelayBondingExtrinsic(substrateApi, amount, targetValidators, nominatorMetadata, chainInfo) {
-  let bondDest = arguments.length > 5 && arguments[5] !== undefined ? arguments[5] : 'Staked';
+async function getRelayBondingExtrinsic(substrateApi, amount, targetValidators, chainInfo, address, nominatorMetadata) {
+  let bondDest = arguments.length > 6 && arguments[6] !== undefined ? arguments[6] : 'Staked';
   const chainApi = await substrateApi.isReady;
   const binaryAmount = new _util.BN(amount);
   let bondTx;
   let nominateTx;
+  const validatorParamList = targetValidators.map(validator => {
+    return validator.address;
+  });
+  if (!nominatorMetadata) {
+    bondTx = chainApi.api.tx.staking.bond(address, binaryAmount, bondDest);
+    nominateTx = chainApi.api.tx.staking.nominate(validatorParamList);
+    return chainApi.api.tx.utility.batchAll([bondTx, nominateTx]);
+  }
   if (!nominatorMetadata.isBondedBefore) {
     // first time
     bondTx = chainApi.api.tx.staking.bond(nominatorMetadata.address, binaryAmount, bondDest);
-    nominateTx = chainApi.api.tx.staking.nominate(targetValidators);
+    nominateTx = chainApi.api.tx.staking.nominate(validatorParamList);
     return chainApi.api.tx.utility.batchAll([bondTx, nominateTx]);
   } else {
     if (binaryAmount.gt(_util.BN_ZERO)) {
       bondTx = chainApi.api.tx.staking.bondExtra(binaryAmount);
     }
     if (nominatorMetadata.isBondedBefore && targetValidators.length > 0) {
-      nominateTx = chainApi.api.tx.staking.nominate(targetValidators);
+      nominateTx = chainApi.api.tx.staking.nominate(validatorParamList);
     }
   }
   if (bondTx && !nominateTx) {

package/cjs/koni/api/staking/bonding/utils.js

@@ -14,12 +14,15 @@ exports.getBondedValidators = getBondedValidators;
 exports.getCommission = getCommission;
 exports.getInflationParams = getInflationParams;
 exports.getParaCurrentInflation = getParaCurrentInflation;
-exports.getStakingAvailableActions = getStakingAvailableActions;
+exports.getStakingAvailableActionsByChain = getStakingAvailableActionsByChain;
+exports.getStakingAvailableActionsByNominator = getStakingAvailableActionsByNominator;
+exports.getStakingStatusByNominations = getStakingStatusByNominations;
 exports.getWithdrawalInfo = getWithdrawalInfo;
 exports.isActionFromValidator = isActionFromValidator;
 exports.isShowNominationByValidator = isShowNominationByValidator;
 exports.isUnstakeAll = isUnstakeAll;
 exports.parseIdentity = parseIdentity;
+exports.parsePoolStashAddress = parsePoolStashAddress;
 exports.transformPoolName = transformPoolName;
 var _KoniTypes = require("@subwallet/extension-base/background/KoniTypes");
 var _constants = require("@subwallet/extension-base/services/chain-service/constants");
@@ -34,6 +37,15 @@ exports.PalletParachainStakingRequestType = PalletParachainStakingRequestType;
   PalletParachainStakingRequestType["DECREASE"] = "decrease";
   PalletParachainStakingRequestType["BOND_LESS"] = "bondLess";
 })(PalletParachainStakingRequestType || (exports.PalletParachainStakingRequestType = PalletParachainStakingRequestType = {}));
+function parsePoolStashAddress(api, index, poolId, poolsPalletId) {
+  const ModPrefix = (0, _util.stringToU8a)('modl');
+  const U32Opts = {
+    bitLength: 32,
+    isLe: true
+  };
+  const EmptyH256 = new Uint8Array(32);
+  return api.registry.createType('AccountId32', (0, _util.u8aConcat)(ModPrefix, poolsPalletId, new Uint8Array([index]), (0, _util.bnToU8a)(new _util.BN(poolId.toString()), U32Opts), EmptyH256)).toString();
+}
 function transformPoolName(input) {
   return input.replace(/[^\x20-\x7E]/g, '');
 }
@@ -139,11 +151,8 @@ function getBondedValidators(nominations) {
   const bondedValidators = [];
   let nominationCount = 0;
   for (const nomination of nominations) {
-    const bnActiveStake = new _util.BN(nomination.activeStake);
-    if (bnActiveStake.gt(_util.BN_ZERO)) {
-      nominationCount += 1;
-      bondedValidators.push((0, _utils.reformatAddress)(nomination.validatorAddress, 0));
-    }
+    nominationCount += 1;
+    bondedValidators.push((0, _utils.reformatAddress)(nomination.validatorAddress, 0));
   }
   return {
     nominationCount,
@@ -153,7 +162,9 @@ function getBondedValidators(nominations) {
 function isUnstakeAll(selectedValidator, nominations, unstakeAmount) {
   let isUnstakeAll = false;
   for (const nomination of nominations) {
-    if (nomination.validatorAddress === selectedValidator) {
+    const parsedValidatorAddress = (0, _utils.reformatAddress)(nomination.validatorAddress, 0);
+    const parsedSelectedValidator = (0, _utils.reformatAddress)(selectedValidator, 0);
+    if (parsedValidatorAddress === parsedSelectedValidator) {
       if (unstakeAmount === nomination.activeStake) {
         isUnstakeAll = true;
       }
@@ -171,13 +182,28 @@ exports.StakingAction = StakingAction;
   StakingAction["CLAIM_REWARD"] = "CLAIM_REWARD";
   StakingAction["CANCEL_UNSTAKE"] = "CANCEL_UNSTAKE";
 })(StakingAction || (exports.StakingAction = StakingAction = {}));
-function getStakingAvailableActions(nominatorMetadata) {
+function getStakingAvailableActionsByChain(chain, type) {
+  if (type === _KoniTypes.StakingType.POOLED) {
+    return [StakingAction.STAKE, StakingAction.UNSTAKE, StakingAction.WITHDRAW, StakingAction.CLAIM_REWARD];
+  }
+  if (_constants._STAKING_CHAIN_GROUP.para.includes(chain)) {
+    return [StakingAction.STAKE, StakingAction.UNSTAKE, StakingAction.WITHDRAW, StakingAction.CANCEL_UNSTAKE];
+  } else if (_constants._STAKING_CHAIN_GROUP.astar.includes(chain)) {
+    return [StakingAction.STAKE, StakingAction.UNSTAKE, StakingAction.WITHDRAW, StakingAction.CLAIM_REWARD];
+  } else if (_constants._STAKING_CHAIN_GROUP.amplitude.includes(chain)) {
+    return [StakingAction.STAKE, StakingAction.UNSTAKE, StakingAction.WITHDRAW];
+  }
+  return [StakingAction.STAKE, StakingAction.UNSTAKE, StakingAction.WITHDRAW, StakingAction.CANCEL_UNSTAKE];
+}
+function getStakingAvailableActionsByNominator(nominatorMetadata, unclaimedReward) {
   const result = [StakingAction.STAKE];
   const bnActiveStake = new _util.BN(nominatorMetadata.activeStake);
   if (nominatorMetadata.activeStake && bnActiveStake.gt(_util.BN_ZERO)) {
     result.push(StakingAction.UNSTAKE);
-    const isChainAllowClaimReward = _constants._STAKING_CHAIN_GROUP.amplitude.includes(nominatorMetadata.chain) || _constants._STAKING_CHAIN_GROUP.astar.includes(nominatorMetadata.chain);
-    if (nominatorMetadata.type === _KoniTypes.StakingType.POOLED || isChainAllowClaimReward) {
+    const isAstarNetwork = _constants._STAKING_CHAIN_GROUP.astar.includes(nominatorMetadata.chain);
+    const isAmplitudeNetwork = _constants._STAKING_CHAIN_GROUP.amplitude.includes(nominatorMetadata.chain);
+    const bnUnclaimedReward = new _util.BN(unclaimedReward || '0');
+    if ((nominatorMetadata.type === _KoniTypes.StakingType.POOLED || isAmplitudeNetwork) && bnUnclaimedReward.gt(_util.BN_ZERO) || isAstarNetwork) {
       result.push(StakingAction.CLAIM_REWARD);
     }
   }
@@ -219,4 +245,23 @@ function getWithdrawalInfo(nominatorMetadata) {
     }
   }
   return result;
+}
+function getStakingStatusByNominations(bnTotalActiveStake, nominationList) {
+  let stakingStatus = _KoniTypes.StakingStatus.EARNING_REWARD;
+  if (bnTotalActiveStake.isZero()) {
+    stakingStatus = _KoniTypes.StakingStatus.NOT_EARNING;
+  } else {
+    let invalidDelegationCount = 0;
+    for (const nomination of nominationList) {
+      if (nomination.status === _KoniTypes.StakingStatus.NOT_EARNING) {
+        invalidDelegationCount += 1;
+      }
+    }
+    if (invalidDelegationCount > 0 && invalidDelegationCount < nominationList.length) {
+      stakingStatus = _KoniTypes.StakingStatus.PARTIALLY_EARNING;
+    } else if (invalidDelegationCount === nominationList.length) {
+      stakingStatus = _KoniTypes.StakingStatus.NOT_EARNING;
+    }
+  }
+  return stakingStatus;
 }

package/cjs/koni/api/tokens/wasm/index.js

@@ -6,6 +6,7 @@ Object.defineProperty(exports, "__esModule", {
 exports.getPSP22ContractPromise = getPSP22ContractPromise;
 exports.getPSP34ContractPromise = getPSP34ContractPromise;
 exports.getPSP34TransferExtrinsic = getPSP34TransferExtrinsic;
+var _utils = require("@subwallet/extension-base/koni/api/tokens/wasm/utils");
 var _helper = require("@subwallet/extension-base/services/chain-service/helper");
 var _apiContract = require("@polkadot/api-contract");
 // Copyright 2019-2022 @subwallet/extension-base
@@ -22,10 +23,10 @@ async function getPSP34TransferExtrinsic(networkKey, substrateApi, senderAddress
   const onChainOption = params.onChainOption;
   try {
     const contractPromise = getPSP34ContractPromise(substrateApi.api, contractAddress);
-    const transferQuery = await contractPromise.query['psp34::transfer'](senderAddress, {
-      gasLimit: -1
-    }, recipientAddress, onChainOption, {});
-    const gasLimit = transferQuery.gasRequired.toString();
+    // @ts-ignore
+    const gasLimit = await (0, _utils.getWasmContractGasLimit)(substrateApi.api, senderAddress, 'psp34::transfer', contractPromise, {}, [recipientAddress, onChainOption, {}]);
+
+    // @ts-ignore
     return contractPromise.tx['psp34::transfer']({
       gasLimit
     }, recipientAddress, onChainOption, {});

package/cjs/koni/api/tokens/wasm/utils.js

@@ -0,0 +1,63 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.getDefaultWeightV2 = getDefaultWeightV2;
+exports.getWasmContractGasLimit = getWasmContractGasLimit;
+var _util = require("@polkadot/util");
+// Copyright 2019-2022 @subwallet/extension-base
+// SPDX-License-Identifier: Apache-2.0
+
+const MAX_CALL_WEIGHT = '5000000000000';
+const DEFAULT_REF_TIME = '1000000000000';
+const toContractAbiMessage = (contractPromise, message) => {
+  const value = contractPromise.abi.messages.find(m => m.method === message);
+  if (!value) {
+    const messages = contractPromise === null || contractPromise === void 0 ? void 0 : contractPromise.abi.messages.map(m => m.method).join(', ');
+    const error = `"${message}" not found in metadata.spec.messages: [${messages}]`;
+    console.error(error);
+    return {
+      ok: false,
+      error
+    };
+  }
+  return {
+    ok: true,
+    value
+  };
+};
+async function getWasmContractGasLimit(api, callerAddress, message, contract) {
+  let options = arguments.length > 4 && arguments[4] !== undefined ? arguments[4] : {};
+  let args = arguments.length > 5 && arguments[5] !== undefined ? arguments[5] : [];
+  try {
+    var _abiMessage$value;
+    const abiMessage = toContractAbiMessage(contract, message);
+    if (!abiMessage.ok) {
+      return getDefaultWeightV2(api, true);
+    }
+
+    // @ts-ignore
+    const {
+      gasLimit,
+      storageDepositLimit,
+      value
+    } = options;
+
+    // @ts-ignore
+    const {
+      gasRequired
+    } = await api.call.contractsApi.call(callerAddress, contract.address, value !== null && value !== void 0 ? value : new _util.BN(0), gasLimit !== null && gasLimit !== void 0 ? gasLimit : null, storageDepositLimit !== null && storageDepositLimit !== void 0 ? storageDepositLimit : null, abiMessage === null || abiMessage === void 0 ? void 0 : (_abiMessage$value = abiMessage.value) === null || _abiMessage$value === void 0 ? void 0 : _abiMessage$value.toU8a(args));
+    return gasRequired;
+  } catch {
+    return getDefaultWeightV2(api, true);
+  }
+}
+function getDefaultWeightV2(apiPromise, isFallback) {
+  const proofSize = isFallback ? 3407872 : MAX_CALL_WEIGHT; // TODO: handle error better
+  const refTime = isFallback ? 32490000000 : DEFAULT_REF_TIME;
+  return apiPromise.registry.createType('WeightV2', {
+    refTime,
+    proofSize
+  });
+}

package/cjs/koni/api/xcm/index.js

@@ -35,115 +35,4 @@ const createXcmExtrinsic = async _ref => {
   console.log('XCM extrinsic: ', extrinsic.toHex());
   return extrinsic;
 };
-
-// TODO: add + refine logic for more chains
-// export function updateXcmResponseTxResult (
-//   networkKey: string,
-//   tokenInfo: _ChainAsset,
-//   response: TransactionResponse,
-//   records: EventRecord[]
-// ) {
-//   if (!response.txResult) {
-//     response.txResult = { change: '0' };
-//   }
-//
-//   let isFeeUseMainTokenSymbol = true;
-//
-//   for (let index = 0; index < records.length; index++) {
-//     const record = records[index];
-//
-//     if (_XCM_CHAIN_GROUP.acala.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'currencies' &&
-//         record.event.method.toLowerCase() === 'transferred') {
-//         if (index === 0) {
-//           // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//           response.txResult.fee = record.event.data[3]?.toString() || '0';
-//           response.txResult.feeSymbol = tokenInfo.symbol;
-//
-//           isFeeUseMainTokenSymbol = false;
-//         } else {
-//           response.txResult.change = record.event.data[3]?.toString() || '0';
-//           response.txResult.changeSymbol = tokenInfo.symbol;
-//         }
-//       }
-//
-//       if (record.event.section === 'tokens' &&
-//         record.event.method.toLowerCase() === 'withdrawn') {
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.kintsugi.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'tokens' &&
-//         record.event.method.toLowerCase() === 'transfer') {
-//         response.txResult.change = record.event.data[3]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.genshiro.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'eqBalances' &&
-//         record.event.method.toLowerCase() === 'transfer') {
-//         response.txResult.change = record.event.data[3]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.bifrost.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'tokens' &&
-//         record.event.method.toLowerCase() === 'withdrawn') {
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       } else if (record.event.section === 'balances' &&
-//         record.event.method.toLowerCase() === 'transfer') {
-//         // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.astar.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'assets' &&
-//         record.event.method.toLowerCase() === 'burned') {
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.moonbeam.includes(networkKey) && !_isNativeToken(tokenInfo)) {
-//       if (record.event.section === 'assets' &&
-//         record.event.method.toLowerCase() === 'burned') {
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//         response.txResult.changeSymbol = tokenInfo.symbol;
-//       }
-//     } else if (_XCM_CHAIN_GROUP.statemine.includes(networkKey)) {
-//       if (!_isNativeToken(tokenInfo)) {
-//         if (record.event.section === 'assets' &&
-//           record.event.method.toLowerCase() === 'transferred') {
-//           response.txResult.change = record.event.data[3]?.toString() || '0';
-//           response.txResult.changeSymbol = tokenInfo.symbol;
-//         }
-//       } else {
-//         if (record.event.section === 'balances' &&
-//           record.event.method.toLowerCase() === 'withdraw') {
-//           response.txResult.change = record.event.data[1]?.toString() || '0';
-//           response.txResult.changeSymbol = tokenInfo.symbol;
-//         }
-//       }
-//     } else {
-//       if (record.event.section === 'balances' &&
-//         record.event.method.toLowerCase() === 'transfer') {
-//         // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//       } else if (record.event.section === 'xTokens' &&
-//         record.event.method.toLowerCase() === 'transferred') {
-//         // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//         response.txResult.change = record.event.data[2]?.toString() || '0';
-//       }
-//     }
-//
-//     if (isFeeUseMainTokenSymbol && record.event.section === 'balances' && record.event.method.toLowerCase() === 'withdraw') {
-//       if (!response.txResult.fee) {
-//         // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//         response.txResult.fee = record.event.data[1]?.toString() || '0';
-//       }
-//     } else if (isFeeUseMainTokenSymbol && record.event.section === 'tokens' && record.event.method.toLowerCase() === 'withdrawn') {
-//       if (!response.txResult.fee) {
-//         // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-//         response.txResult.fee = record.event.data[2]?.toString() || '0';
-//       }
-//     }
-//   }
-// }
 exports.createXcmExtrinsic = createXcmExtrinsic;

package/cjs/koni/api/xcm/polkadotXcm.js

@@ -42,13 +42,13 @@ function getAssetLocation(tokenInfo, sendingValue) {
   };
 }
 function getExtrinsicByPolkadotXcmPallet(tokenInfo, originChainInfo, destinationChainInfo, recipientAddress, value, api) {
-  const weightParam = _utils.NETWORK_USE_UNLIMITED_WEIGHT.includes(originChainInfo.slug) ? _utils.POLKADOT_UNLIMITED_WEIGHT : _utils.FOUR_INSTRUCTIONS_WEIGHT;
+  const weightParam = (0, _utils.getDestWeight)();
   const beneficiary = (0, _utils.getBeneficiary)(originChainInfo, destinationChainInfo, recipientAddress);
   const destination = getDestinationChainLocation(destinationChainInfo);
   const assetLocation = getAssetLocation(tokenInfo, value);
   let method = 'limitedReserveTransferAssets';
   if (['astar', 'shiden'].includes(originChainInfo.slug)) {
-    method = 'reserveWithdrawAssets';
+    method = 'limitedReserveWithdrawAssets';
   } else if ((0, _utils2._isSubstrateRelayChain)(destinationChainInfo)) {
     method = 'limitedTeleportAssets';
   }