@subwallet/extension-base 1.3.75-2 → 1.3.77-0

package/services/chain-service/utils/patch.d.ts

@@ -1,5 +1,5 @@
 import { _ChainAsset, _ChainInfo, _MultiChainAsset } from '@subwallet/chain-list/types';
-export declare const ChainListVersion = "0.2.123";
+export declare const ChainListVersion = "0.2.126";
 export interface PatchInfo {
     patchVersion: string;
     appliedVersion: string;

package/services/chain-service/utils/patch.js

@@ -5,7 +5,7 @@ const PRODUCTION_BRANCHES = ['master', 'webapp', 'webapp-dev'];
 const branchName = process.env.BRANCH_NAME || 'subwallet-dev';
 const fetchDomain = process.env.PATCH_CHAIN_LIST_URL || (PRODUCTION_BRANCHES.indexOf(branchName) > -1 ? 'https://chain-list-assets.subwallet.app' : 'https://dev.sw-chain-list-assets.pages.dev');
 const fetchFile = PRODUCTION_BRANCHES.indexOf(branchName) > -1 ? 'list.json' : 'preview.json';
-export const ChainListVersion = '0.2.123'; // update this when build chain-list
+export const ChainListVersion = '0.2.126'; // update this when build chain-list
 
 // todo: move this interface to chainlist
 

package/services/earning-service/constants/chains.d.ts

@@ -17,6 +17,7 @@ export declare const _STAKING_CHAIN_GROUP: {
     mythos: string[];
     tanssi: string[];
 };
+export declare const SUNSETTED_YIELD_POOL_SLUGS: string[];
 export declare const RELAY_HANDLER_DIRECT_STAKING_CHAINS: string[];
 export declare const TON_CHAINS: string[];
 export declare const MaxEraRewardPointsEras = 14;

package/services/earning-service/constants/chains.js

@@ -1,6 +1,7 @@
 // Copyright 2019-2022 @subwallet/extension-base
 // SPDX-License-Identifier: Apache-2.0
 
+import { STELLA_SWAP_LIQUID_STAKING_SLUG } from "../handlers/liquid-staking/stella-swap.js";
 export const _STAKING_CHAIN_GROUP = {
   relay: ['polkadot', 'kusama', 'aleph', 'polkadex', 'ternoa', 'alephTest', 'polkadexTest', 'westend', 'kate', 'edgeware', 'creditcoin', 'vara_network', 'goldberg_testnet', 'availTuringTest', 'avail_mainnet', 'vara_testnet', 'dentnet', 'cere', 'zkverify', 'zkverify_testnet', 'paseoTest'],
   assetHub: ['statemine', 'statemint', 'westend_assethub', 'paseo_assethub'],
@@ -22,6 +23,7 @@ export const _STAKING_CHAIN_GROUP = {
   mythos: ['mythos', 'muse_testnet'],
   tanssi: ['tanssi', 'dancelight']
 };
+export const SUNSETTED_YIELD_POOL_SLUGS = [STELLA_SWAP_LIQUID_STAKING_SLUG, 'MANTA___native_staking___manta_network'];
 export const RELAY_HANDLER_DIRECT_STAKING_CHAINS = [..._STAKING_CHAIN_GROUP.relay, ..._STAKING_CHAIN_GROUP.assetHub];
 export const TON_CHAINS = ['ton', 'ton_testnet'];
 export const MaxEraRewardPointsEras = 14;

package/services/earning-service/handlers/liquid-staking/stella-swap.d.ts

@@ -6,6 +6,7 @@ import { BaseYieldStepDetail, HandleYieldStepData, LiquidYieldPoolInfo, OptimalY
 import { Contract } from 'web3-eth-contract';
 import BaseLiquidStakingPoolHandler from './base';
 export declare const getStellaswapLiquidStakingContract: (networkKey: string, assetAddress: string, evmApi: _EvmApi, options?: {}) => Contract;
+export declare const STELLA_SWAP_LIQUID_STAKING_SLUG = "xcDOT___liquid_staking___stellaswap";
 export default class StellaSwapLiquidStakingPoolHandler extends BaseLiquidStakingPoolHandler {
     slug: string;
     protected readonly name: string;

package/services/earning-service/handlers/liquid-staking/stella-swap.js

@@ -19,6 +19,7 @@ export const getStellaswapLiquidStakingContract = (networkKey, assetAddress, evm
 const APR_STATS_URL = 'https://stdot-apr.stellaswap.com/';
 const SUBWALLET_REFERRAL = '0x7e6815f45E624768548d085231f2d453f16FD7DD';
 const TOP_HOLDER = '0x4300e09284e3bB4d9044DdAB31EfAF5f3301DABa';
+export const STELLA_SWAP_LIQUID_STAKING_SLUG = 'xcDOT___liquid_staking___stellaswap';
 export default class StellaSwapLiquidStakingPoolHandler extends BaseLiquidStakingPoolHandler {
   inputAsset = 'moonbeam-LOCAL-xcDOT';
   altInputAsset = '';
@@ -28,8 +29,8 @@ export default class StellaSwapLiquidStakingPoolHandler extends BaseLiquidStakin
   transactionChainType = ChainType.EVM;
   rateDecimals = 10; // Derivative asset decimals
   availableMethod = {
-    join: true,
-    defaultUnstake: true,
+    join: false,
+    defaultUnstake: false,
     fastUnstake: false,
     cancelUnstake: false,
     withdraw: true,
@@ -38,7 +39,7 @@ export default class StellaSwapLiquidStakingPoolHandler extends BaseLiquidStakin
   };
   constructor(state, chain) {
     super(state, chain);
-    this.slug = 'xcDOT___liquid_staking___stellaswap';
+    this.slug = STELLA_SWAP_LIQUID_STAKING_SLUG;
     this.name = 'StellaSwap Liquid Staking';
     this._logo = 'stellaswap';
     this.shortName = 'StellaSwap';
@@ -161,7 +162,8 @@ export default class StellaSwapLiquidStakingPoolHandler extends BaseLiquidStakin
             unstakeBalance: unlockBalance.toString(),
             isBondedBefore: totalBalance.gt(BN_ZERO),
             derivativeToken: derivativeTokenSlug,
-            status: activeBalance.gt(BN_ZERO) ? EarningStatus.EARNING_REWARD : EarningStatus.NOT_EARNING,
+            // StellaSwap announced all stDOT positions are auto-unstaking, so rewards are no longer earned.
+            status: EarningStatus.NOT_EARNING,
             nominations: [],
             unstakings
           };

package/services/earning-service/handlers/native-staking/dtao.d.ts

@@ -1,6 +1,6 @@
 import { _ChainInfo } from '@subwallet/chain-list/types';
 import KoniState from '@subwallet/extension-base/koni/background/handlers/State';
-import { BaseYieldPositionInfo, RequestEarningImpact, YieldPoolInfo, YieldPoolType, YieldPositionInfo } from '@subwallet/extension-base/types';
+import { BaseYieldPositionInfo, RequestEarningImpact, TransactionData, YieldPoolInfo, YieldPoolType, YieldPositionInfo } from '@subwallet/extension-base/types';
 import TaoNativeStakingPoolHandler, { TaoStakingStakeOption } from './tao';
 export interface SubnetData {
     netuid: number;
@@ -11,15 +11,6 @@ export interface SubnetData {
     taoIn: number;
     taoInEmission: number;
 }
-export interface RawDelegateState {
-    data: Array<{
-        hotkey_name: string;
-        hotkey: {
-            ss58: string;
-        };
-        stake: string;
-    }>;
-}
 export interface EarningImpactResult {
     slippage: number;
     rate: number;
@@ -40,4 +31,5 @@ export default class SubnetTaoStakingPoolHandler extends TaoNativeStakingPoolHan
     subscribePoolInfo(callback: (data: YieldPoolInfo) => void): Promise<VoidFunction>;
     parseNominatorMetadata(chainInfo: _ChainInfo, delegatorState: TaoStakingStakeOption[], netuid?: number): Promise<Omit<YieldPositionInfo, keyof BaseYieldPositionInfo>>;
     subscribePoolPosition(useAddresses: string[], rsCallback: (rs: YieldPositionInfo) => void): Promise<VoidFunction>;
+    handleChangeRootClaimType(): Promise<TransactionData>;
 }

package/services/earning-service/handlers/native-staking/dtao.js

@@ -1,57 +1,15 @@
 // Copyright 2019-2022 @subwallet/extension-base
 // SPDX-License-Identifier: Apache-2.0
 
+import { TransactionError } from '@subwallet/extension-base/background/errors/TransactionError';
 import { ExtrinsicType } from '@subwallet/extension-base/background/KoniTypes';
 import { BITTENSOR_REFRESH_STAKE_APY, BITTENSOR_REFRESH_STAKE_INFO } from '@subwallet/extension-base/constants';
-import { EarningStatus, YieldPoolType } from '@subwallet/extension-base/types';
+import { BasicTxErrorType, EarningStatus, YieldPoolType } from '@subwallet/extension-base/types';
 import { reformatAddress } from '@subwallet/extension-base/utils';
 import BigN from 'bignumber.js';
-import { BN, BN_ZERO } from '@polkadot/util';
+import { BN_ZERO } from '@polkadot/util';
 import TaoNativeStakingPoolHandler, { DEFAULT_DTAO_MINBOND } from "./tao.js";
-
-// interface ApiResponse {
-//   data: SubnetData[];
-// }
-
-// interface PoolData {
-//   netuid: number;
-//   name: string;
-//   symbol: string;
-// }
-
-// interface PoolApiResponse {
-//   data: PoolData[];
-// }
-
-// const SUBNET_API_URL = 'https://dash.taostats.io/api/subnet';
-// const POOL_API_URL = 'https://dash.taostats.io/api/dtao/pool';
-
-// export async function fetchSubnetData () {
-//   try {
-//     const [subnetResponse, poolResponse] = await Promise.all([
-//       fetch(SUBNET_API_URL).then((res) => res.json()) as Promise<ApiResponse>,
-//       fetch(POOL_API_URL).then((res) => res.json()) as Promise<PoolApiResponse>
-//     ]);
-
-//     const poolMap = new Map(poolResponse.data.map((pool) => [pool.netuid, pool]));
-
-//     const filteredSubnets = subnetResponse.data.filter((subnet) => subnet.netuid !== 0);
-
-//     const mergedData = filteredSubnets.map((subnet) => ({
-//       ...subnet,
-//       name: poolMap.get(subnet.netuid)?.name || 'Unknown',
-//       symbol: poolMap.get(subnet.netuid)?.symbol || 'Unknown'
-//     }));
-
-//     return mergedData;
-//   } catch (err) {
-//     console.error('Error:', err);
-
-//     return [];
-//   }
-// }
-
-const getAlphaToTaoMapping = async substrateApi => {
+const getAlphaToTaoRateMap = async substrateApi => {
   const allSubnets = (await substrateApi.api.call.subnetInfoRuntimeApi.getAllDynamicInfo()).toJSON();
   if (!allSubnets || allSubnets.length === 0) {
     return {};
@@ -244,87 +202,91 @@ export default class SubnetTaoStakingPoolHandler extends TaoNativeStakingPoolHan
     const chainInfo = this.chainInfo;
     const _delegateInfo = await this.bittensorCache.get();
     const getPoolPosition = async () => {
-      const rawDelegateStateInfos = await Promise.all(useAddresses.map(async address => (await substrateApi.api.call.stakeInfoRuntimeApi.getStakeInfoForColdkey(address)).toJSON()));
-      const price = await getAlphaToTaoMapping(this.substrateApi);
-      if (rawDelegateStateInfos && rawDelegateStateInfos.length > 0) {
-        rawDelegateStateInfos.forEach((rawDelegateStateInfo, i) => {
-          const owner = reformatAddress(useAddresses[i], 42);
-          const delegateStateInfo = rawDelegateStateInfo;
-          const subnetPositions = {};
-          for (const delegate of delegateStateInfo) {
-            const hotkey = delegate.hotkey;
-            const netuid = delegate.netuid;
-            const stake = new BigN(delegate.stake);
-            const aplhaToTaoPrice = new BigN(price[netuid]);
-            if (!subnetPositions[netuid]) {
-              subnetPositions[netuid] = {
-                delegatorState: [],
-                totalBalance: BN_ZERO,
-                originalTotalStake: BN_ZERO
-              };
-            }
-            let identity = '';
-            if (_delegateInfo) {
-              const delegateInfo = _delegateInfo.data.find(info => info.hotkey.ss58 === hotkey);
-              identity = delegateInfo ? delegateInfo.name : '';
-            }
-            subnetPositions[netuid].delegatorState.push({
-              owner: hotkey,
-              amount: stake.toString(),
-              rate: aplhaToTaoPrice,
-              identity: identity
-            });
-            subnetPositions[netuid].totalBalance = subnetPositions[netuid].totalBalance.add(new BN(stake.toString()));
-            subnetPositions[netuid].originalTotalStake = subnetPositions[netuid].originalTotalStake.add(new BN(stake.toString()));
+      const rawDelegateStateInfos = await substrateApi.api.call.stakeInfoRuntimeApi.getStakeInfoForColdkeys(useAddresses);
+      const delegateStateInfos = rawDelegateStateInfos.toPrimitive();
+      const alphaToTaoRateMap = await getAlphaToTaoRateMap(this.substrateApi);
+      if (!delegateStateInfos || delegateStateInfos.length === 0) {
+        return;
+      }
+      delegateStateInfos.forEach(([coldkey, stakeInfos]) => {
+        const owner = reformatAddress(coldkey, 42);
+        const subnetPositions = {};
+        for (const delegate of stakeInfos) {
+          const hotkey = delegate.hotkey;
+          const netuid = delegate.netuid;
+          if (netuid === 0) {
+            continue;
+          }
+          const stake = new BigN(delegate.stake);
+          const alphaToTaoPrice = new BigN(alphaToTaoRateMap[netuid]);
+          if (!subnetPositions[netuid]) {
+            subnetPositions[netuid] = {
+              delegatorState: [],
+              totalBalance: BigN(0),
+              originalTotalStake: BigN(0)
+            };
           }
-          Object.values(this.subnetData).forEach(subnet => {
-            const netuid = subnet.netuid;
-            const subnetSlug = `${this.slug}__subnet_${netuid.toString().padStart(2, '0')}`;
-            const subnetName = `${subnet.name || 'Unknown'} ${netuid}`;
-            const subnetSymbol = subnet.symbol || 'dTAO';
-            const {
-              delegatorState = [],
-              originalTotalStake = BN_ZERO
-            } = subnetPositions[netuid] || {};
-            if (delegatorState.length > 0) {
-              this.parseNominatorMetadata(chainInfo, delegatorState, netuid).then(nominatorMetadata => {
-                rsCallback({
-                  ...defaultInfo,
-                  ...nominatorMetadata,
-                  address: owner,
-                  type: this.type,
-                  slug: subnetSlug,
-                  subnetData: {
-                    subnetSymbol,
-                    subnetShortName: subnetName,
-                    originalTotalStake: originalTotalStake.toString()
-                  }
-                });
-              }).catch(console.error);
-            } else {
+          let identity = '';
+          if (_delegateInfo) {
+            const delegateInfo = _delegateInfo.data.find(info => info.hotkey.ss58 === hotkey);
+            identity = delegateInfo ? delegateInfo.name : '';
+          }
+          subnetPositions[netuid].delegatorState.push({
+            owner: hotkey,
+            amount: stake.toString(),
+            rate: alphaToTaoPrice,
+            identity
+          });
+          subnetPositions[netuid].totalBalance = subnetPositions[netuid].totalBalance.plus(new BigN(stake.toString()));
+          subnetPositions[netuid].originalTotalStake = subnetPositions[netuid].originalTotalStake.plus(new BigN(stake.toString()));
+        }
+        Object.values(this.subnetData).forEach(subnet => {
+          const netuid = subnet.netuid;
+          const subnetSlug = `${this.slug}__subnet_${netuid.toString().padStart(2, '0')}`;
+          const subnetName = `${subnet.name || 'Unknown'} ${netuid}`;
+          const subnetSymbol = subnet.symbol || 'dTAO';
+          const {
+            delegatorState = [],
+            originalTotalStake = BN_ZERO
+          } = subnetPositions[netuid] || {};
+          if (delegatorState.length > 0) {
+            this.parseNominatorMetadata(chainInfo, delegatorState, netuid).then(nominatorMetadata => {
               rsCallback({
                 ...defaultInfo,
-                type: this.type,
+                ...nominatorMetadata,
                 address: owner,
-                balanceToken: this.nativeToken.slug,
-                totalStake: '0',
-                activeStake: '0',
-                unstakeBalance: '0',
-                status: EarningStatus.NOT_STAKING,
-                isBondedBefore: false,
-                nominations: [],
-                unstakings: [],
+                type: this.type,
                 slug: subnetSlug,
                 subnetData: {
                   subnetSymbol,
                   subnetShortName: subnetName,
-                  originalTotalStake: '0'
+                  originalTotalStake: originalTotalStake.toString()
                 }
               });
-            }
-          });
+            }).catch(console.error);
+          } else {
+            rsCallback({
+              ...defaultInfo,
+              type: this.type,
+              address: owner,
+              balanceToken: this.nativeToken.slug,
+              totalStake: '0',
+              activeStake: '0',
+              unstakeBalance: '0',
+              status: EarningStatus.NOT_STAKING,
+              isBondedBefore: false,
+              nominations: [],
+              unstakings: [],
+              slug: subnetSlug,
+              subnetData: {
+                subnetSymbol,
+                subnetShortName: subnetName,
+                originalTotalStake: '0'
+              }
+            });
+          }
         });
-      }
+      });
     };
     const getStakingPositionInterval = async () => {
       if (cancel) {
@@ -343,4 +305,12 @@ export default class SubnetTaoStakingPoolHandler extends TaoNativeStakingPoolHan
   }
 
   /* Subscribe pool position */
+
+  /* Unimplemented function  */
+
+  handleChangeRootClaimType() {
+    return Promise.reject(new TransactionError(BasicTxErrorType.UNSUPPORTED));
+  }
+
+  /* Unimplemented function  */
 }

package/services/earning-service/handlers/native-staking/tanssi.js

@@ -291,7 +291,8 @@ export default class TanssiNativeStakingPoolHandler extends BaseParaNativeStakin
           metadata: {
             pendingStake: bnJoiningStake.toString(),
             compoundingStake: bnCompoundingStake.toString(),
-            manualStake: bnManualStake.toString()
+            manualStake: bnManualStake.toString(),
+            isShowActiveStakeDetails: bnJoiningStake.gt(0) || bnCompoundingStake.gt(0) || !bnManualStake.gt(0)
           }
         });
       };

package/services/earning-service/handlers/native-staking/tao.d.ts

@@ -4,7 +4,7 @@ import { ExtrinsicType } from '@subwallet/extension-base/background/KoniTypes';
 import KoniState from '@subwallet/extension-base/koni/background/handlers/State';
 import { _SubstrateApi } from '@subwallet/extension-base/services/chain-service/types';
 import BaseParaStakingPoolHandler from '@subwallet/extension-base/services/earning-service/handlers/native-staking/base-para';
-import { BaseYieldPositionInfo, OptimalYieldPath, StakeCancelWithdrawalParams, SubmitBittensorChangeValidatorStaking, SubmitJoinNativeStaking, TransactionData, UnstakingInfo, ValidatorInfo, YieldPoolInfo, YieldPoolMethodInfo, YieldPositionInfo, YieldTokenBaseInfo } from '@subwallet/extension-base/types';
+import { BaseYieldPositionInfo, BittensorRootClaimType, OptimalYieldPath, StakeCancelWithdrawalParams, SubmitBittensorChangeValidatorStaking, SubmitJoinNativeStaking, TransactionData, UnstakingInfo, ValidatorInfo, YieldPoolInfo, YieldPoolMethodInfo, YieldPositionInfo, YieldTokenBaseInfo } from '@subwallet/extension-base/types';
 import BigN from 'bignumber.js';
 declare type Nominators = [Array<[number, number]>];
 export interface TestnetBittensorDelegateInfo {
@@ -88,8 +88,6 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     protected bittensorCache: BittensorCache;
     protected getMinBond(netuid?: number): Promise<BigN>;
     constructor(state: KoniState, chain: string);
-    handleYieldWithdraw(address: string, unstakingInfo: UnstakingInfo): Promise<TransactionData>;
-    handleYieldCancelUnstake(params: StakeCancelWithdrawalParams): Promise<TransactionData>;
     get maintainBalance(): string;
     subscribePoolInfo(callback: (data: YieldPoolInfo) => void): Promise<VoidFunction>;
     protected parseNominatorMetadataBase(chainInfo: _ChainInfo, delegatorState: TaoStakingStakeOption[], minBond: string, applyRate?: boolean): Omit<YieldPositionInfo, keyof BaseYieldPositionInfo>;
@@ -104,5 +102,8 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     handleYieldUnstake(amount: string, address: string, selectedTarget?: string, netuid?: number, slippage?: number): Promise<[ExtrinsicType, TransactionData]>;
     validateYieldLeave(amount: string, address: string, fastLeave: boolean, selectedTarget?: string, slug?: string, poolInfo?: YieldPoolInfo): Promise<TransactionError[]>;
     handleChangeEarningValidator(data: SubmitBittensorChangeValidatorStaking): Promise<TransactionData>;
+    handleChangeRootClaimType(type: BittensorRootClaimType): Promise<TransactionData>;
+    handleYieldWithdraw(address: string, unstakingInfo: UnstakingInfo): Promise<TransactionData>;
+    handleYieldCancelUnstake(params: StakeCancelWithdrawalParams): Promise<TransactionData>;
 }
 export {};

package/services/earning-service/handlers/native-staking/tao.js

@@ -113,36 +113,6 @@ export class BittensorCache {
     }
   }
 }
-
-// export async function fetchTaoDelegateState (address: string): Promise<RawDelegateState> {
-//   const apiKey = bittensorApiKey();
-
-//   return new Promise(function (resolve) {
-//     fetch(`https://api.taostats.io/api/stake_balance/latest/v1?coldkey=${address}`, {
-//       method: 'GET',
-//       headers: {
-//         'Content-Type': 'application/json',
-//         Authorization: `${apiKey}`
-//       }
-//     }).then((resp) => {
-//       resolve(resp.json());
-//     }).catch(console.error);
-//   });
-// }
-
-/* Fetch data */
-
-// const testnetDelegate = {
-//   '5G6wdAdS7hpBuH1tjuZDhpzrGw9Wf71WEVakDCxHDm1cxEQ2': {
-//     name: '0x436c6f776e4e616d65f09fa4a1',
-//     url: 'https://example.com  ',
-//     image: 'https://example.com/image.png',
-//     discord: '0xe28094446973636f7264',
-//     description: 'This is an example identity.',
-//     additional: ''
-//   }
-// };
-
 export const getAlphaToTaoRate = async (substrateApi, netuid) => {
   const subnetInfo = (await substrateApi.api.call.subnetInfoRuntimeApi.getDynamicInfo(netuid)).toJSON();
   if (!subnetInfo) {
@@ -196,20 +166,9 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     super(state, chain);
     this.bittensorCache = BittensorCache.getInstance();
   }
-
-  /* Unimplemented function  */
-  handleYieldWithdraw(address, unstakingInfo) {
-    return Promise.reject(new TransactionError(BasicTxErrorType.UNSUPPORTED));
-  }
-  handleYieldCancelUnstake(params) {
-    return Promise.reject(new TransactionError(BasicTxErrorType.UNSUPPORTED));
-  }
-  /* Unimplemented function  */
-
   get maintainBalance() {
     const ed = new BigN(this.nativeToken.minAmount || '0');
-    const calculateMaintainBalance = new BigN(15).multipliedBy(ed).dividedBy(10);
-    const maintainBalance = calculateMaintainBalance;
+    const maintainBalance = new BigN(15).multipliedBy(ed).dividedBy(10);
     return maintainBalance.toString();
   }
 
@@ -333,66 +292,74 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     const chainInfo = this.chainInfo;
     const _delegateInfo = await this.bittensorCache.get();
     const getPoolPosition = async () => {
-      const rawDelegateStateInfos = await Promise.all(useAddresses.map(async address => (await substrateApi.api.call.stakeInfoRuntimeApi.getStakeInfoForColdkey(address)).toJSON()));
-      if (rawDelegateStateInfos && rawDelegateStateInfos.length > 0) {
-        rawDelegateStateInfos.forEach((rawDelegateStateInfo, i) => {
-          const owner = reformatAddress(useAddresses[i], 42);
-          const delegatorState = [];
-          let bnTotalBalance = BN_ZERO;
-          const delegateStateInfo = rawDelegateStateInfo;
-          const totalDelegate = {};
-          for (const delegate of delegateStateInfo) {
-            const hotkey = delegate.hotkey;
-            const netuid = delegate.netuid;
-            const stake = new BigN(delegate.stake);
-            if (netuid === 0) {
-              const taoStake = stake.toFixed(0);
-              if (totalDelegate[hotkey]) {
-                totalDelegate[hotkey] = new BigN(totalDelegate[hotkey]).plus(taoStake).toFixed();
-              } else {
-                totalDelegate[hotkey] = taoStake;
-              }
+      const rawDelegateStateInfo = await substrateApi.api.call.stakeInfoRuntimeApi.getStakeInfoForColdkeys(useAddresses);
+      const delegateStateInfo = rawDelegateStateInfo.toPrimitive();
+      const rootClaimType = await substrateApi.api.query.subtensorModule.rootClaimType.multi(useAddresses);
+      if (!delegateStateInfo || delegateStateInfo.length === 0) {
+        return;
+      }
+      delegateStateInfo.forEach(([coldkey, stakeInfos], i) => {
+        var _rootClaimType$i;
+        const owner = reformatAddress(coldkey, 42);
+        const delegatorState = [];
+        let bnTotalBalance = BN_ZERO;
+        const totalDelegate = {};
+        for (const delegate of stakeInfos) {
+          const hotkey = delegate.hotkey;
+          const netuid = delegate.netuid;
+          const stake = new BigN(delegate.stake);
+          if (netuid === 0) {
+            const taoStake = stake.toFixed(0);
+            if (totalDelegate[hotkey]) {
+              totalDelegate[hotkey] = new BigN(totalDelegate[hotkey]).plus(taoStake).toFixed();
+            } else {
+              totalDelegate[hotkey] = taoStake;
             }
           }
-          for (const hotkey in totalDelegate) {
-            bnTotalBalance = bnTotalBalance.add(new BN(totalDelegate[hotkey]));
-            let identity = '';
-            if (_delegateInfo) {
-              const delegateInfo = _delegateInfo.data.find(info => info.hotkey.ss58 === hotkey);
-              identity = delegateInfo ? delegateInfo.name : '';
-            }
-            delegatorState.push({
-              owner: hotkey,
-              amount: totalDelegate[hotkey],
-              identity: identity
-            });
+        }
+        for (const hotkey in totalDelegate) {
+          bnTotalBalance = bnTotalBalance.add(new BN(totalDelegate[hotkey]));
+          let identity = '';
+          if (_delegateInfo) {
+            const delegateInfo = _delegateInfo.data.find(info => info.hotkey.ss58 === hotkey);
+            identity = delegateInfo ? delegateInfo.name : '';
           }
-          if (delegateStateInfo && delegateStateInfo.length > 0) {
-            this.parseNominatorMetadata(chainInfo, delegatorState).then(nominatorMetadata => {
-              rsCallback({
-                ...defaultInfo,
-                ...nominatorMetadata,
-                address: owner,
-                type: this.type
-              });
-            }).catch(console.error);
-          } else {
+          delegatorState.push({
+            owner: hotkey,
+            amount: totalDelegate[hotkey],
+            identity: identity
+          });
+        }
+        const rawType = (_rootClaimType$i = rootClaimType[i]) === null || _rootClaimType$i === void 0 ? void 0 : _rootClaimType$i.toString();
+        const rootClaimForAddress = rawType === 'Swap' || rawType === 'Keep' ? rawType : 'Others';
+        if (delegateStateInfo && delegateStateInfo.length > 0) {
+          this.parseNominatorMetadata(chainInfo, delegatorState).then(nominatorMetadata => {
             rsCallback({
               ...defaultInfo,
-              type: this.type,
+              ...nominatorMetadata,
               address: owner,
-              balanceToken: this.nativeToken.slug,
-              totalStake: '0',
-              activeStake: '0',
-              unstakeBalance: '0',
-              status: EarningStatus.NOT_STAKING,
-              isBondedBefore: false,
-              nominations: [],
-              unstakings: []
+              metadata: {
+                bittensorRootClaimType: rootClaimForAddress
+              },
+              type: this.type
             });
-          }
-        });
-      }
+          }).catch(console.error);
+        } else {
+          rsCallback({
+            ...defaultInfo,
+            type: this.type,
+            address: owner,
+            balanceToken: this.nativeToken.slug,
+            totalStake: '0',
+            activeStake: '0',
+            unstakeBalance: '0',
+            status: EarningStatus.NOT_STAKING,
+            isBondedBefore: false,
+            nominations: [],
+            unstakings: []
+          });
+        }
+      });
     };
     const getStakingPositionInterval = async () => {
       if (cancel) {
@@ -440,8 +407,7 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     }));
   }
   async getMainnetPoolTargets(netuid) {
-    const _topValidator = await this.bittensorCache.get();
-    const topValidator = _topValidator;
+    const topValidator = await this.bittensorCache.get();
     const bnMinBond = await this.getMinBond();
     const validatorList = topValidator.data;
     const aprResponse = await this.bittensorCache.fetchApr(netuid);
@@ -449,7 +415,7 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
     aprResponse.data.forEach(item => {
       aprMap[item.hotkey.ss58] = item.thirty_day_apy;
     });
-    const results = await Promise.all(validatorList.map(validator => {
+    return await Promise.all(validatorList.map(validator => {
       const address = validator.hotkey.ss58;
       // With bittensor we use total weight, root weight and alpha staked insted of total stake, own stake and other stake
       const bnTotalWeightStake = new BigN(validator.global_weighted_stake);
@@ -477,7 +443,6 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
         identity: name
       };
     }));
-    return results;
   }
   async getPoolTargets(netuid) {
     if (this.chain === 'bittensor') {
@@ -625,7 +590,25 @@ export default class TaoNativeStakingPoolHandler extends BaseParaStakingPoolHand
         }
       })));
     }
-    const extrinsic = chainApi.api.tx.subtensorModule.moveStake(originValidator, destValidator, netuid, netuid, amount);
-    return extrinsic;
+    return chainApi.api.tx.subtensorModule.moveStake(originValidator, destValidator, netuid, netuid, amount);
   }
+
+  /* Others function  */
+  async handleChangeRootClaimType(type) {
+    if (type !== 'Keep' && type !== 'Swap') {
+      return Promise.reject(new TransactionError(BasicTxErrorType.INVALID_PARAMS));
+    }
+    const chainApi = await this.substrateApi.isReady;
+    return chainApi.api.tx.subtensorModule.setRootClaimType(type);
+  }
+  /* Others function  */
+
+  /* Unimplemented function  */
+  handleYieldWithdraw(address, unstakingInfo) {
+    return Promise.reject(new TransactionError(BasicTxErrorType.UNSUPPORTED));
+  }
+  handleYieldCancelUnstake(params) {
+    return Promise.reject(new TransactionError(BasicTxErrorType.UNSUPPORTED));
+  }
+  /* Unimplemented function  */
 }