feat(sdk-coin-sol): implement staking activate for jito

TICKET: SC-2314

Author: haritkapadia

examples/ts/sol/stake-jito.ts

@@ -0,0 +1,101 @@
+/**
+ * Stakes JitoSOL tokens on Solana devnet.
+ *
+ * Copyright 2025, BitGo, Inc.  All Rights Reserved.
+ */
+import { BitGoAPI } from '@bitgo/sdk-api'
+import { TransactionBuilderFactory, Tsol } from '@bitgo/sdk-coin-sol'
+import { coins } from '@bitgo/statics'
+import { Connection, PublicKey, clusterApiUrl, Transaction, Keypair, LAMPORTS_PER_SOL } from "@solana/web3.js"
+import { getStakePoolAccount, updateStakePool } from '@solana/spl-stake-pool'
+import * as bs58 from 'bs58';
+
+require('dotenv').config({ path: '../../.env' })
+
+const AMOUNT_LAMPORTS = 1000
+const JITO_STAKE_POOL_ADDRESS = 'Jito4APyf642JPZPx3hGc6WWJ8zPKtRbRs4P815Awbb'
+const NETWORK = 'devnet'
+
+const bitgo = new BitGoAPI({
+  accessToken: process.env.TESTNET_ACCESS_TOKEN,
+  env: 'test',
+})
+const coin = coins.get("tsol")
+bitgo.register(coin.name, Tsol.createInstance)
+
+async function main() {
+  const account = getAccount()
+  const connection = new Connection(clusterApiUrl(NETWORK), 'confirmed')
+  const recentBlockhash = await connection.getLatestBlockhash()
+  const stakePoolAccount = await getStakePoolAccount(connection, new PublicKey(JITO_STAKE_POOL_ADDRESS))
+
+
+  // Account should have sufficient balance
+  const accountBalance = await connection.getBalance(account.publicKey)
+  if (accountBalance < 0.1 * LAMPORTS_PER_SOL) {
+    console.info(`Your account balance is ${accountBalance / LAMPORTS_PER_SOL} SOL, requesting airdrop`)
+    const sig = await connection.requestAirdrop(account.publicKey, 2 * LAMPORTS_PER_SOL)
+    await connection.confirmTransaction(sig)
+    console.info(`Airdrop successful: ${sig}`)
+  }
+
+  // Stake pool should be up to date
+  const epochInfo = await connection.getEpochInfo()
+  if (stakePoolAccount.account.data.lastUpdateEpoch.ltn(epochInfo.epoch)) {
+    console.info('Stake pool is out of date.')
+    const usp = await updateStakePool(connection, stakePoolAccount)
+    const tx = new Transaction()
+    tx.add(...usp.updateListInstructions, ...usp.finalInstructions)
+    const signer = Keypair.fromSecretKey(account.secretKeyArray)
+    const sig = await connection.sendTransaction(tx, [signer])
+    await connection.confirmTransaction(sig)
+    console.info(`Stake pool updated: ${sig}`)
+  }
+
+  // Use BitGoAPI to build depositSol instruction
+  const txBuilder = new TransactionBuilderFactory(coin).getStakingActivateBuilder()
+  txBuilder
+    .amount(`${AMOUNT_LAMPORTS}`)
+    .sender(account.publicKey.toBase58())
+    .stakingAddress(JITO_STAKE_POOL_ADDRESS)
+    .validator(JITO_STAKE_POOL_ADDRESS)
+    .isJito(true)
+    .nonce(recentBlockhash.blockhash)
+  txBuilder.sign({ key: account.secretKey })
+  const tx = await txBuilder.build()
+  const serializedTx = tx.toBroadcastFormat()
+  console.info(`Transaction JSON:\n${JSON.stringify(tx.toJson(), undefined, 2)}`)
+
+  // Send transaction
+  try {
+    const sig = await connection.sendRawTransaction(Buffer.from(serializedTx, 'base64'))
+    await connection.confirmTransaction(sig)
+    console.log(`${AMOUNT_LAMPORTS / LAMPORTS_PER_SOL} SOL deposited`, sig)
+  } catch (e) {
+    console.log('Error sending transaction')
+    console.error(e)
+    if (e.transactionMessage === 'Transaction simulation failed: Error processing Instruction 0: Provided owner is not allowed') {
+      console.error('If you successfully staked JitoSOL once, you cannot stake again.')
+    }
+  }
+}
+
+const getAccount = () => {
+  const publicKey = process.env.ACCOUNT_PUBLIC_KEY
+  const secretKey = process.env.ACCOUNT_SECRET_KEY
+  if (publicKey === undefined || secretKey === undefined) {
+    const { publicKey, secretKey } = Keypair.generate()
+    console.log('Here is a new account to save into your .env file.')
+    console.log(`ACCOUNT_PUBLIC_KEY=${publicKey.toBase58()}`)
+    console.log(`ACCOUNT_SECRET_KEY=${bs58.encode(secretKey)}`)
+    throw new Error("Missing account information")
+  }
+
+  return {
+    publicKey: new PublicKey(publicKey),
+    secretKey,
+    secretKeyArray: new Uint8Array(bs58.decode(secretKey)),
+  }
+}
+
+main().catch((e) => console.error(e))

modules/sdk-coin-sol/package.json

@@ -43,6 +43,7 @@
     "@bitgo/sdk-core": "^36.0.0",
     "@bitgo/sdk-lib-mpc": "^10.6.0",
     "@bitgo/statics": "^57.0.0",
+    "@solana/spl-stake-pool": "1.1.8",
     "@solana/spl-token": "0.3.1",
     "@solana/web3.js": "1.92.1",
     "bignumber.js": "^9.0.0",

modules/sdk-coin-sol/src/lib/constants.ts

@@ -9,6 +9,13 @@ export const STAKE_ACCOUNT_RENT_EXEMPT_AMOUNT = 2282880;
 
 export const UNAVAILABLE_TEXT = 'UNAVAILABLE';
 
+export const JITO_STAKE_POOL_ADDRESS = 'Jito4APyf642JPZPx3hGc6WWJ8zPKtRbRs4P815Awbb';
+export const JITOSOL_MINT_ADDRESS = 'J1toso1uCk3RLmjorhTtrVwY9HJ7X8V9yYac6Y7kGCPn';
+export const JITO_STAKE_POOL_RESERVE_ACCOUNT = 'BgKUXdS29YcHCFrPm5M8oLHiTzZaMDjsebggjoaQ6KFL';
+export const JITO_STAKE_POOL_RESERVE_ACCOUNT_TESTNET = 'rrWBQqRqBXYZw3CmPCCcjFxQ2Ds4JFJd7oRQJ997dhz';
+export const JITO_MANAGER_FEE_ACCOUNT = 'feeeFLLsam6xZJFc6UQFrHqkvVt4jfmVvi2BRLkUZ4i';
+export const JITO_MANAGER_FEE_ACCOUNT_TESTNET = 'DH7tmjoQ5zjqcgfYJU22JqmXhP5EY1tkbYpgVWUS2oNo';
+
 // Sdk instructions, mainly to check decoded types.
 export enum ValidInstructionTypesEnum {
   AdvanceNonceAccount = 'AdvanceNonceAccount',
@@ -30,6 +37,7 @@ export enum ValidInstructionTypesEnum {
   SetPriorityFee = 'SetPriorityFee',
   MintTo = 'MintTo',
   Burn = 'Burn',
+  DepositSol = 'DepositSol',
 }
 
 // Internal instructions types
@@ -72,6 +80,7 @@ export const VALID_SYSTEM_INSTRUCTION_TYPES: ValidInstructionTypes[] = [
   ValidInstructionTypesEnum.SetPriorityFee,
   ValidInstructionTypesEnum.MintTo,
   ValidInstructionTypesEnum.Burn,
+  ValidInstructionTypesEnum.DepositSol,
 ];
 
 /** Const to check the order of the Wallet Init instructions when decode */
@@ -96,6 +105,12 @@ export const marinadeStakingActivateInstructionsIndexes = {
   Memo: 2,
 } as const;
 
+/** Const to check the order of the Jito Staking Activate instructions when decode */
+export const jitoStakingActivateInstructionsIndexes = {
+  AtaInit: 0,
+  DepositSol: 1,
+} as const;
+
 /** Const to check the order of the Staking Authorize instructions when decode */
 export const stakingAuthorizeInstructionsIndexes = {
   Authorize: 0,

modules/sdk-coin-sol/src/lib/iface.ts

@@ -8,6 +8,7 @@ import {
   TransactionSignature,
 } from '@solana/web3.js';
 import { InstructionBuilderTypes } from './constants';
+import { StakePoolInstructionType } from '@solana/spl-stake-pool';
 
 // TODO(STLX-9890): Add the interfaces for validityWindow and SequenceId
 export interface SolanaKeys {
@@ -121,6 +122,7 @@ export interface StakingActivate {
     amount: string;
     validator: string;
     isMarinade?: boolean;
+    isJito?: boolean;
   };
 }
 
@@ -184,6 +186,7 @@ export interface AtaClose {
 export type ValidInstructionTypes =
   | SystemInstructionType
   | StakeInstructionType
+  | StakePoolInstructionType
   | 'Memo'
   | 'InitializeAssociatedTokenAccount'
   | 'CloseAssociatedTokenAccount'

modules/sdk-coin-sol/src/lib/instructionParamsFactory.ts

@@ -27,7 +27,12 @@ import {
 import { NotSupported, TransactionType } from '@bitgo/sdk-core';
 import { coins, SolCoin } from '@bitgo/statics';
 import assert from 'assert';
-import { InstructionBuilderTypes, ValidInstructionTypesEnum, walletInitInstructionIndexes } from './constants';
+import {
+  InstructionBuilderTypes,
+  JITO_STAKE_POOL_ADDRESS,
+  ValidInstructionTypesEnum,
+  walletInitInstructionIndexes,
+} from './constants';
 import {
   AtaClose,
   AtaInit,
@@ -47,6 +52,8 @@ import {
   SetPriorityFee,
 } from './iface';
 import { getInstructionType } from './utils';
+import { DepositSolParams } from '@solana/spl-stake-pool';
+import { decodeDepositSol } from './jitoStakePoolOperations';
 
 /**
  * Construct instructions params from Solana instructions
@@ -303,6 +310,14 @@ function parseSendInstructions(
   return instructionData;
 }
 
+function stakingInstructionsIsMarinade(si: StakingInstructions): boolean {
+  return !!(si.delegate === undefined && si.depositSol === undefined);
+}
+
+function stakingInstructionsIsJito(si: StakingInstructions): boolean {
+  return !!(si.delegate === undefined && si.depositSol?.stakePool.toString() === JITO_STAKE_POOL_ADDRESS);
+}
+
 /**
  * Parses Solana instructions to create staking tx and delegate tx instructions params
  * Only supports Nonce, StakingActivate and Memo Solana instructions
@@ -312,8 +327,8 @@ function parseSendInstructions(
  */
 function parseStakingActivateInstructions(
   instructions: TransactionInstruction[]
-): Array<Nonce | StakingActivate | Memo> {
-  const instructionData: Array<Nonce | StakingActivate | Memo> = [];
+): Array<Nonce | StakingActivate | Memo | AtaInit> {
+  const instructionData: Array<Nonce | StakingActivate | Memo | AtaInit> = [];
   const stakingInstructions = {} as StakingInstructions;
   for (const instruction of instructions) {
     const type = getInstructionType(instruction);
@@ -346,21 +361,48 @@ function parseStakingActivateInstructions(
       case ValidInstructionTypesEnum.StakingDelegate:
         stakingInstructions.delegate = StakeInstruction.decodeDelegate(instruction);
         break;
+
+      case ValidInstructionTypesEnum.DepositSol:
+        stakingInstructions.depositSol = decodeDepositSol(instruction);
+        break;
+
+      case ValidInstructionTypesEnum.InitializeAssociatedTokenAccount:
+        instructionData.push({
+          type: InstructionBuilderTypes.CreateAssociatedTokenAccount,
+          params: {
+            mintAddress: instruction.keys[ataInitInstructionKeysIndexes.MintAddress].pubkey.toString(),
+            ataAddress: instruction.keys[ataInitInstructionKeysIndexes.ATAAddress].pubkey.toString(),
+            ownerAddress: instruction.keys[ataInitInstructionKeysIndexes.OwnerAddress].pubkey.toString(),
+            payerAddress: instruction.keys[ataInitInstructionKeysIndexes.PayerAddress].pubkey.toString(),
+            tokenName: findTokenName(instruction.keys[ataInitInstructionKeysIndexes.MintAddress].pubkey.toString()),
+          },
+        });
+        break;
     }
   }
 
   validateStakingInstructions(stakingInstructions);
+
   const stakingActivate: StakingActivate = {
     type: InstructionBuilderTypes.StakingActivate,
     params: {
-      fromAddress: stakingInstructions.create?.fromPubkey.toString() || '',
-      stakingAddress: stakingInstructions.initialize?.stakePubkey.toString() || '',
-      amount: stakingInstructions.create?.lamports.toString() || '',
+      fromAddress:
+        stakingInstructions.create?.fromPubkey.toString() ||
+        stakingInstructions.depositSol?.fundingAccount.toString() ||
+        '',
+      stakingAddress:
+        stakingInstructions.initialize?.stakePubkey.toString() ||
+        stakingInstructions.depositSol?.stakePool.toString() ||
+        '',
+      amount:
+        stakingInstructions.create?.lamports.toString() || stakingInstructions.depositSol?.lamports.toString() || '',
       validator:
         stakingInstructions.delegate?.votePubkey.toString() ||
         stakingInstructions.initialize?.authorized.staker.toString() ||
+        stakingInstructions.depositSol?.stakePool.toString() ||
         '',
-      isMarinade: stakingInstructions.delegate === undefined,
+      isMarinade: stakingInstructionsIsMarinade(stakingInstructions),
+      isJito: stakingInstructionsIsJito(stakingInstructions),
     },
   };
   instructionData.push(stakingActivate);
@@ -413,22 +455,23 @@ interface StakingInstructions {
   initialize?: InitializeStakeParams;
   delegate?: DelegateStakeParams;
   authorize?: AuthorizeStakeParams[];
+  depositSol?: DepositSolParams;
 }
 
 function validateStakingInstructions(stakingInstructions: StakingInstructions) {
-  if (!stakingInstructions.create) {
-    throw new NotSupported('Invalid staking activate transaction, missing create stake account instruction');
-  }
-
-  if (!stakingInstructions.initialize && stakingInstructions.delegate) {
-    return;
-  } else if (!stakingInstructions.delegate && stakingInstructions.initialize) {
-    return;
-  } else if (!stakingInstructions.delegate && !stakingInstructions.initialize) {
-    // If both are missing something is wrong
-    throw new NotSupported(
-      'Invalid staking activate transaction, missing initialize stake account/delegate instruction'
-    );
+  if (stakingInstructionsIsJito(stakingInstructions)) {
+    if (!stakingInstructions.depositSol) {
+      throw new NotSupported('Invalid staking activate transaction, missing deposit sol instruction');
+    }
+  } else {
+    if (!stakingInstructions.create) {
+      throw new NotSupported('Invalid staking activate transaction, missing create stake account instruction');
+    }
+    if (!stakingInstructions.delegate && !stakingInstructions.initialize) {
+      throw new NotSupported(
+        'Invalid staking activate transaction, missing initialize stake account/delegate instruction'
+      );
+    }
   }
 }
 
@@ -776,6 +819,9 @@ function parseAtaInitInstructions(
         };
         instructionData.push(ataInit);
         break;
+      case ValidInstructionTypesEnum.DepositSol:
+        // AtaInit is a part of spl-stake-pool's depositSol process
+        break;
       default:
         throw new NotSupported(
           'Invalid transaction, instruction type not supported: ' + getInstructionType(instruction)