diff --git a/modules/sdk-core/src/bitgo/staking/iStakingWallet.ts b/modules/sdk-core/src/bitgo/staking/iStakingWallet.ts index 8cff5cdf9f..aac4e0ec4f 100644 --- a/modules/sdk-core/src/bitgo/staking/iStakingWallet.ts +++ b/modules/sdk-core/src/bitgo/staking/iStakingWallet.ts @@ -201,6 +201,10 @@ export interface SwitchValidatorOptions { validator: string; } +export interface TaoSwitchValidatorOptions extends SwitchValidatorOptions { + netUID: string; +} + export interface ClaimRewardsOptions { amount: string; clientId?: string; @@ -294,7 +298,7 @@ export interface IStakingWallet { readonly coin: string; stake(options: StakeOptions | TronStakeOptions | TaoStakeOptions): Promise; unstake(options: UnstakeOptions | EthUnstakeOptions): Promise; - switchValidator(options: SwitchValidatorOptions): Promise; + switchValidator(options: SwitchValidatorOptions | TaoSwitchValidatorOptions): Promise; claimRewards(options: ClaimRewardsOptions): Promise; getStakingRequest(stakingRequestId: string): Promise; getTransactionsReadyToSign(stakingRequestId: string): Promise; diff --git a/modules/sdk-core/src/bitgo/staking/stakingWallet.ts b/modules/sdk-core/src/bitgo/staking/stakingWallet.ts index 4e38f9c135..c1254ba0d3 100644 --- a/modules/sdk-core/src/bitgo/staking/stakingWallet.ts +++ b/modules/sdk-core/src/bitgo/staking/stakingWallet.ts @@ -22,6 +22,7 @@ import { StakingTxRequestPrebuildTransactionResult, TronStakeOptions, TaoStakeOptions, + TaoSwitchValidatorOptions, } from './iStakingWallet'; import { BitGoBase } from '../bitgoBase'; import { IWallet, PrebuildTransactionResult } from '../wallet'; @@ -78,7 +79,7 @@ export class StakingWallet implements IStakingWallet { * @param options - switch validator options * @return StakingRequest */ - async switchValidator(options: SwitchValidatorOptions): Promise { + async switchValidator(options: SwitchValidatorOptions | TaoSwitchValidatorOptions): Promise { return await this.createStakingRequest(options, 'SWITCH_VALIDATOR'); } @@ -317,7 +318,8 @@ export class StakingWallet implements IStakingWallet { | SwitchValidatorOptions | ClaimRewardsOptions | TronStakeOptions - | TaoStakeOptions, + | TaoStakeOptions + | TaoSwitchValidatorOptions, type: string ): Promise { return await this.bitgo