CLI: Add validator to pool

utils/steward-cli/src/commands/actions/add_validator_to_pool.rs

@@ -0,0 +1,117 @@
+use std::sync::Arc;
+
+use anchor_lang::{InstructionData, ToAccountMetas};
+use anyhow::Result;
+use clap::Parser;
+use solana_client::nonblocking::rpc_client::RpcClient;
+use solana_program::instruction::Instruction;
+use solana_sdk::sysvar::{rent, stake_history};
+#[allow(deprecated)]
+use solana_sdk::{
+    pubkey::Pubkey, signature::read_keypair_file, signer::Signer, stake, system_program, sysvar,
+    transaction::Transaction,
+};
+use spl_stake_pool::find_stake_program_address;
+use stakenet_sdk::utils::{
+    accounts::get_all_steward_accounts,
+    transactions::{configure_instruction, print_base58_tx},
+};
+
+use crate::commands::command_args::PermissionedParameters;
+
+#[derive(Parser)]
+#[command(about = "Add validator to pool")]
+pub struct AddValidatorToPool {
+    #[command(flatten)]
+    pub permissioned_parameters: PermissionedParameters,
+
+    /// Vote account
+    #[arg(long, env)]
+    pub vote_account: Pubkey,
+}
+
+pub async fn command_add_validator_to_pool(
+    args: AddValidatorToPool,
+    client: &Arc<RpcClient>,
+    program_id: Pubkey,
+) -> Result<()> {
+    // Creates config account
+    let authority = Arc::new(
+        read_keypair_file(args.permissioned_parameters.authority_keypair_path)
+            .expect("Failed reading keypair file ( Payer )"),
+    );
+
+    let steward_config = args.permissioned_parameters.steward_config;
+
+    let steward_accounts = get_all_steward_accounts(client, &program_id, &steward_config).await?;
+
+    let vote_account = args.vote_account;
+
+    let (stake_address, _) = find_stake_program_address(
+        &spl_stake_pool::id(),
+        &vote_account,
+        &steward_accounts.stake_pool_address,
+        None,
+    );
+
+    let ix = Instruction {
+        program_id,
+        accounts: jito_steward::accounts::AddValidatorToPool {
+            config: steward_config,
+            state_account: steward_accounts.state_address,
+            stake_pool_program: spl_stake_pool::id(),
+            stake_pool: steward_accounts.stake_pool_address,
+            reserve_stake: steward_accounts.stake_pool_account.reserve_stake,
+            withdraw_authority: steward_accounts.stake_pool_withdraw_authority,
+            validator_list: steward_accounts.validator_list_address,
+            stake_account: stake_address,
+            vote_account,
+            admin: authority.pubkey(),
+            rent: rent::id(),
+            clock: sysvar::clock::id(),
+            stake_history: stake_history::id(),
+            stake_config: stake::config::ID,
+            system_program: system_program::id(),
+            stake_program: stake::program::id(),
+        }
+        .to_account_metas(None),
+        data: jito_steward::instruction::AddValidatorToPool {
+            validator_seed: None,
+        }
+        .data(),
+    };
+
+    let blockhash = client.get_latest_blockhash().await?;
+
+    let configured_ix = configure_instruction(
+        &[ix],
+        args.permissioned_parameters
+            .transaction_parameters
+            .priority_fee,
+        args.permissioned_parameters
+            .transaction_parameters
+            .compute_limit,
+        args.permissioned_parameters
+            .transaction_parameters
+            .heap_size,
+    );
+
+    let transaction = Transaction::new_signed_with_payer(
+        &configured_ix,
+        Some(&authority.pubkey()),
+        &[&authority],
+        blockhash,
+    );
+
+    if args.permissioned_parameters.transaction_parameters.print_tx {
+        print_base58_tx(&configured_ix)
+    } else {
+        let signature = client
+            .send_and_confirm_transaction_with_spinner(&transaction)
+            .await?;
+
+        println!("Signature: {}", signature);
+    }
+
+    Ok(())
+}

utils/steward-cli/src/commands/actions/mod.rs

@@ -1,5 +1,6 @@
 pub mod add_to_blacklist;
 pub mod add_to_directed_stake_whitelist;
+pub mod add_validator_to_pool;
 pub mod auto_add_validator_from_pool;
 pub mod auto_remove_validator_from_pool;
 pub mod close_directed_stake_meta;

utils/steward-cli/src/commands/command_args.rs

@@ -6,6 +6,7 @@ use std::path::PathBuf;
 use crate::commands::{
     actions::{
         add_to_directed_stake_whitelist::AddToDirectedStakeWhitelist,
+        add_validator_to_pool::AddValidatorToPool,
         close_directed_stake_meta::CloseDirectedStakeMeta,
         close_directed_stake_ticket::CloseDirectedStakeTicket,
         close_directed_stake_whitelist::CloseDirectedStakeWhitelist,
@@ -335,6 +336,7 @@ pub enum Commands {
     RemoveBadValidators(RemoveBadValidators),
     ManuallyCopyVoteAccount(ManuallyCopyVoteAccount),
     ManuallyCopyAllVoteAccounts(ManuallyCopyAllVoteAccounts),
+    AddValidatorToPool(AddValidatorToPool),
     ManuallyRemoveValidator(ManuallyRemoveValidator),
     AutoRemoveValidatorFromPool(AutoRemoveValidatorFromPool),
     AutoAddValidatorFromPool(AutoAddValidatorFromPool),

utils/steward-cli/src/main.rs

@@ -47,6 +47,7 @@ use crate::{
     commands::{
         actions::{
             add_to_directed_stake_whitelist::command_add_to_directed_stake_whitelist,
+            add_validator_to_pool::command_add_validator_to_pool,
             close_directed_stake_meta::command_close_directed_stake_meta,
             close_directed_stake_ticket::command_close_directed_stake_ticket,
             close_directed_stake_whitelist::command_close_directed_stake_whitelist,
@@ -173,6 +174,9 @@ async fn main() -> Result<()> {
         Commands::ResetValidatorLamportBalances(args) => {
             command_reset_validator_lamport_balances(args, &client, steward_program_id).await
         }
+        Commands::AddValidatorToPool(args) => {
+            command_add_validator_to_pool(args, &client, steward_program_id).await
+        }
         Commands::ManuallyRemoveValidator(args) => {
             command_manually_remove_validator(args, &client, steward_program_id).await
         }