stake-pool: add redelegate js bindings (#3960)

* - add ts/js binding for redelegate functionality

* - add redelegate instructions

* - refactor

* - force rebuild

* - refactor

* - force rebuild

* - force rebuild

Author: AlexanderRay

stake-pool/js/README.md

@@ -51,4 +51,4 @@ console.log(solanaStakePool);
 ```javascript
 // `solanaStakePool` is provided in the global namespace by the script bundle.
 console.log(solanaStakePool);
-```
+```

stake-pool/js/src/constants.ts

@@ -7,6 +7,9 @@ export const STAKE_POOL_PROGRAM_ID = new PublicKey('SPoo1Ku8WFXoNDMHPsrGSTSG1Y47
 // Maximum number of validators to update during UpdateValidatorListBalance.
 export const MAX_VALIDATORS_TO_UPDATE = 5;
 
+// Seed for ephemeral stake account
+export const EPHEMERAL_STAKE_SEED_PREFIX = Buffer.from('ephemeral');
+
 // Seed used to derive transient stake accounts.
 export const TRANSIENT_STAKE_SEED_PREFIX = Buffer.from('transient');
 

stake-pool/js/src/index.ts

@@ -24,6 +24,7 @@ import {
   prepareWithdrawAccounts,
   lamportsToSol,
   solToLamports,
+  findEphemeralStakeProgramAddress,
 } from './utils';
 import { StakePoolInstruction } from './instructions';
 import {
@@ -36,6 +37,7 @@ import {
 } from './layouts';
 import { MAX_VALIDATORS_TO_UPDATE, MINIMUM_ACTIVE_STAKE, STAKE_POOL_PROGRAM_ID } from './constants';
 import { create } from 'superstruct';
+import BN from 'bn.js';
 
 export type { StakePool, AccountType, ValidatorList, ValidatorStakeInfo } from './layouts';
 export { STAKE_POOL_PROGRAM_ID } from './constants';
@@ -66,6 +68,17 @@ export interface StakePoolAccounts {
   validatorList: ValidatorListAccount | undefined;
 }
 
+interface RedelegateProps {
+  connection: Connection;
+  stakePoolAddress: PublicKey;
+  sourceVoteAccount: PublicKey;
+  destinationVoteAccount: PublicKey;
+  sourceTransientStakeSeed: number | BN;
+  destinationTransientStakeSeed: number | BN;
+  ephemeralStakeSeed: number | BN;
+  lamports: number | BN;
+}
+
 /**
  * Retrieves and deserializes a StakePool account using a web3js connection and the stake pool address.
  * @param connection: An active web3js connection.
@@ -401,7 +414,7 @@ export async function withdrawStake(
       val.voteAccountAddress.equals(voteAccount),
     );
     if (voteAccountAddress && voteAccountAddress !== voteAccount) {
-      throw new Error(`Provided withdrawal vote account ${voteAccountAddress} does not match delegation on stake receiver account ${voteAccount}, 
+      throw new Error(`Provided withdrawal vote account ${voteAccountAddress} does not match delegation on stake receiver account ${voteAccount},
       remove this flag or provide a different stake account delegated to ${voteAccountAddress}`);
     }
     if (isValidVoter) {
@@ -668,6 +681,7 @@ export async function increaseValidatorStake(
   stakePoolAddress: PublicKey,
   validatorVote: PublicKey,
   lamports: number,
+  ephemeralStakeSeed?: number,
 ) {
   const stakePool = await getStakePoolAccount(connection, stakePoolAddress);
 
@@ -705,8 +719,14 @@ export async function increaseValidatorStake(
   );
 
   const instructions: TransactionInstruction[] = [];
-  instructions.push(
-    StakePoolInstruction.increaseValidatorStake({
+
+  if (ephemeralStakeSeed != undefined) {
+    const ephemeralStake = await findEphemeralStakeProgramAddress(
+      STAKE_POOL_PROGRAM_ID,
+      stakePoolAddress,
+      new BN(ephemeralStakeSeed),
+    );
+    StakePoolInstruction.increaseAdditionalValidatorStake({
       stakePool: stakePoolAddress,
       staker: stakePool.account.data.staker,
       validatorList: stakePool.account.data.validatorList,
@@ -717,8 +737,25 @@ export async function increaseValidatorStake(
       validatorStake,
       validatorVote,
       lamports,
-    }),
-  );
+      ephemeralStake,
+      ephemeralStakeSeed,
+    });
+  } else {
+    instructions.push(
+      StakePoolInstruction.increaseValidatorStake({
+        stakePool: stakePoolAddress,
+        staker: stakePool.account.data.staker,
+        validatorList: stakePool.account.data.validatorList,
+        reserveStake: stakePool.account.data.reserveStake,
+        transientStakeSeed: transientStakeSeed.toNumber(),
+        withdrawAuthority,
+        transientStake,
+        validatorStake,
+        validatorVote,
+        lamports,
+      }),
+    );
+  }
 
   return {
     instructions,
@@ -733,6 +770,7 @@ export async function decreaseValidatorStake(
   stakePoolAddress: PublicKey,
   validatorVote: PublicKey,
   lamports: number,
+  ephemeralStakeSeed?: number,
 ) {
   const stakePool = await getStakePoolAccount(connection, stakePoolAddress);
   const validatorList = await getValidatorListAccount(
@@ -769,18 +807,41 @@ export async function decreaseValidatorStake(
   );
 
   const instructions: TransactionInstruction[] = [];
-  instructions.push(
-    StakePoolInstruction.decreaseValidatorStake({
-      stakePool: stakePoolAddress,
-      staker: stakePool.account.data.staker,
-      validatorList: stakePool.account.data.validatorList,
-      transientStakeSeed: transientStakeSeed.toNumber(),
-      withdrawAuthority,
-      validatorStake,
-      transientStake,
-      lamports,
-    }),
-  );
+
+  if (ephemeralStakeSeed != undefined) {
+    const ephemeralStake = await findEphemeralStakeProgramAddress(
+      STAKE_POOL_PROGRAM_ID,
+      stakePoolAddress,
+      new BN(ephemeralStakeSeed),
+    );
+    instructions.push(
+      StakePoolInstruction.decreaseAdditionalValidatorStake({
+        stakePool: stakePoolAddress,
+        staker: stakePool.account.data.staker,
+        validatorList: stakePool.account.data.validatorList,
+        transientStakeSeed: transientStakeSeed.toNumber(),
+        withdrawAuthority,
+        validatorStake,
+        transientStake,
+        lamports,
+        ephemeralStake,
+        ephemeralStakeSeed,
+      }),
+    );
+  } else {
+    instructions.push(
+      StakePoolInstruction.decreaseValidatorStake({
+        stakePool: stakePoolAddress,
+        staker: stakePool.account.data.staker,
+        validatorList: stakePool.account.data.validatorList,
+        transientStakeSeed: transientStakeSeed.toNumber(),
+        withdrawAuthority,
+        validatorStake,
+        transientStake,
+        lamports,
+      }),
+    );
+  }
 
   return {
     instructions,
@@ -993,3 +1054,82 @@ export async function stakePoolInfo(connection: Connection, stakePoolAddress: Pu
     }, // CliStakePoolDetails
   };
 }
+
+/**
+ * Creates instructions required to redelegate stake.
+ */
+export async function redelegate(props: RedelegateProps) {
+  const {
+    connection,
+    stakePoolAddress,
+    sourceVoteAccount,
+    sourceTransientStakeSeed,
+    destinationVoteAccount,
+    destinationTransientStakeSeed,
+    ephemeralStakeSeed,
+    lamports,
+  } = props;
+  const stakePool = await getStakePoolAccount(connection, stakePoolAddress);
+
+  const stakePoolWithdrawAuthority = await findWithdrawAuthorityProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    stakePoolAddress,
+  );
+
+  const sourceValidatorStake = await findStakeProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    sourceVoteAccount,
+    stakePoolAddress,
+  );
+
+  const sourceTransientStake = await findTransientStakeProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    sourceVoteAccount,
+    stakePoolAddress,
+    new BN(sourceTransientStakeSeed),
+  );
+
+  const destinationValidatorStake = await findStakeProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    destinationVoteAccount,
+    stakePoolAddress,
+  );
+
+  const destinationTransientStake = await findTransientStakeProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    destinationVoteAccount,
+    stakePoolAddress,
+    new BN(destinationTransientStakeSeed),
+  );
+
+  const ephemeralStake = await findEphemeralStakeProgramAddress(
+    STAKE_POOL_PROGRAM_ID,
+    stakePoolAddress,
+    new BN(ephemeralStakeSeed),
+  );
+
+  const instructions: TransactionInstruction[] = [];
+
+  instructions.push(
+    StakePoolInstruction.redelegate({
+      stakePool: stakePool.pubkey,
+      staker: stakePool.account.data.staker,
+      validatorList: stakePool.account.data.validatorList,
+      stakePoolWithdrawAuthority,
+      ephemeralStake,
+      ephemeralStakeSeed,
+      sourceValidatorStake,
+      sourceTransientStake,
+      sourceTransientStakeSeed,
+      destinationValidatorStake,
+      destinationTransientStake,
+      destinationTransientStakeSeed,
+      validator: destinationVoteAccount,
+      lamports,
+    }),
+  );
+
+  return {
+    instructions,
+  };
+}