Add CLI for setting deposit capacity (#155)

Author: buffalu

cli/src/bin/main.rs

@@ -31,12 +31,22 @@ pub fn get_cli_config(args: &Cli) -> Result<CliConfig, anyhow::Error> {
             .as_ref()
             .ok_or_else(|| anyhow!("unable to get config file path"))?;
         if let Ok(config) = Config::load(config_file) {
+            let keypair = if let Some(keypair_path) = &args.keypair {
+                read_keypair_file(keypair_path)
+            } else {
+                read_keypair_file(config.keypair_path)
+            }
+            .map_err(|e| anyhow!(e.to_string()))?;
+            let rpc = if let Some(rpc) = &args.rpc_url {
+                rpc.to_string()
+            } else {
+                config.json_rpc_url
+            };
+
             CliConfig {
-                rpc_url: config.json_rpc_url,
+                rpc_url: rpc,
                 commitment: CommitmentConfig::from_str(&config.commitment)?,
-                keypair: Some(
-                    read_keypair_file(config.keypair_path).map_err(|e| anyhow!(e.to_string()))?,
-                ),
+                keypair: Some(keypair),
             }
         } else {
             CliConfig {

cli/src/vault.rs

@@ -60,4 +60,11 @@ pub enum VaultActions {
     },
     /// List all vaults
     List,
+    /// Sets the deposit capacity in the vault
+    SetCapacity {
+        /// The vault pubkey
+        vault: String,
+        /// The new capacity
+        amount: u64,
+    },
 }

cli/src/vault_handler.rs

@@ -4,6 +4,7 @@ use anyhow::{anyhow, Result};
 use jito_bytemuck::{AccountDeserialize, Discriminator};
 use jito_vault_client::instructions::{
     CreateTokenMetadataBuilder, InitializeConfigBuilder, InitializeVaultBuilder,
+    SetDepositCapacityBuilder,
 };
 use jito_vault_core::{config::Config, vault::Vault};
 use jito_vault_sdk::inline_mpl_token_metadata;
@@ -93,6 +94,9 @@ impl VaultCliHandler {
                         uri,
                     },
             } => self.create_token_metadata(vault, name, symbol, uri).await,
+            VaultCommands::Vault {
+                action: VaultActions::SetCapacity { vault, amount },
+            } => self.set_capacity(vault, amount).await,
         }
     }
 
@@ -296,4 +300,42 @@ impl VaultCliHandler {
 
         Ok(())
     }
+
+    pub async fn set_capacity(&self, vault: String, amount: u64) -> Result<()> {
+        let keypair = self
+            .cli_config
+            .keypair
+            .as_ref()
+            .ok_or_else(|| anyhow!("Keypair not provided"))?;
+        let vault_pubkey = Pubkey::from_str(&vault)?;
+        let rpc_client = self.get_rpc_client();
+
+        let mut builder = SetDepositCapacityBuilder::new();
+        builder
+            .config(Config::find_program_address(&self.vault_program_id).0)
+            .vault(vault_pubkey)
+            .admin(keypair.pubkey())
+            .amount(amount);
+
+        let recent_blockhash = rpc_client.get_latest_blockhash().await?;
+        let tx = Transaction::new_signed_with_payer(
+            &[builder.instruction()],
+            Some(&keypair.pubkey()),
+            &[keypair],
+            recent_blockhash,
+        );
+
+        info!("Vault capacity instruction: {:?}", builder);
+        info!(
+            "Vault capacity transaction signature: {:?}",
+            tx.get_signature()
+        );
+        rpc_client
+            .send_and_confirm_transaction(&tx)
+            .await
+            .map_err(|e| anyhow!(e.to_string()))?;
+        info!("Transaction confirmed: {:?}", tx.get_signature());
+
+        Ok(())
+    }
 }

docs/_tools/00_cli.md

@@ -212,6 +212,7 @@ Vault commands
 * `create-token-metadata` — Creates token metadata for the vault's LRT token
 * `get` — Gets a vault
 * `list` — List all vaults
+* `set-capacity` — Sets the deposit capacity in the vault
 
 
 
@@ -266,6 +267,19 @@ List all vaults
 
 
 
+## `jito-restaking-cli vault vault set-capacity`
+
+Sets the deposit capacity in the vault
+
+**Usage:** `jito-restaking-cli vault vault set-capacity <VAULT> <AMOUNT>`
+
+###### **Arguments:**
+
+* `<VAULT>` — The vault pubkey
+* `<AMOUNT>` — The new capacity
+
+
+
 <hr/>
 
 <small><i>