feat: add xtask to create serialized protocol config update instruction

Author: ananas-block

xtask/Cargo.toml

@@ -16,7 +16,6 @@ light-hasher = { workspace = true }
 light-indexed-merkle-tree = { workspace = true }
 light-compressed-account = { workspace = true }
 light-merkle-tree-metadata = { workspace = true }
-light-registry = { workspace = true }
 num-bigint = { workspace = true }
 rand = { workspace = true }
 quote = "1.0"
@@ -33,3 +32,4 @@ serde_json = "1.0.139"
 solana-client = { workspace = true }
 solana-transaction-status = { workspace = true }
 light-batched-merkle-tree = { workspace = true }
+light-registry = { workspace = true }

xtask/src/create_update_protocol_config_ix.rs

@@ -0,0 +1,72 @@
+use account_compression::processor::initialize_address_merkle_tree::AnchorDeserialize;
+use clap::Parser;
+use light_client::rpc::{RpcConnection, SolanaRpcConnection};
+use light_registry::{
+    protocol_config::state::ProtocolConfigPda, sdk::create_update_protocol_config_instruction,
+    utils::get_protocol_config_pda_address,
+};
+use solana_sdk::{bs58, message::Message, pubkey};
+
+/// Updateable Parameters:
+/// 1. slot_length
+/// 2. cpi_context_size
+/// 3. min_weight
+#[derive(Debug, Parser)]
+pub struct Options {
+    #[clap(long)]
+    slot_length: Option<u64>,
+    #[clap(long)]
+    cpi_context_size: Option<u64>,
+    #[clap(long)]
+    min_weight: Option<u64>,
+}
+
+/// Steps:
+/// 1. fetch protocol config account
+///     1.1. print protocol config account
+/// 2. create updated protocol config based on inputs
+///     2.1. print updated protocol config
+/// 3. create instruction
+///     - signer is the multisig
+/// 4. serialize instruction to bs58
+/// 5. print bs58
+pub async fn create_update_protocol_config_ix(options: Options) -> anyhow::Result<()> {
+    let rpc_url = String::from("https://api.mainnet-beta.solana.com");
+    let mut rpc = SolanaRpcConnection::new(rpc_url, None);
+    let (protocol_config_pda, _) = get_protocol_config_pda_address();
+    let account = rpc
+        .get_account(protocol_config_pda)
+        .await?
+        .expect("Protocol Config Account not found");
+    let mut deserialized_account = ProtocolConfigPda::deserialize(&mut &account.data[8..]).unwrap();
+    println!("current protocol config: {:?}", deserialized_account);
+    if let Some(slot_length) = options.slot_length {
+        deserialized_account.config.slot_length = slot_length;
+    }
+    if let Some(cpi_context_size) = options.cpi_context_size {
+        deserialized_account.config.cpi_context_size = cpi_context_size;
+    }
+    if let Some(min_weight) = options.min_weight {
+        deserialized_account.config.min_weight = min_weight;
+    }
+    println!("updated protocol config: {:?}", deserialized_account.config);
+    let authority = pubkey!("7PeqkcCXeqgsp5Mi15gjJh8qvSLk7n3dgNuyfPhJJgqY");
+    let instruction = create_update_protocol_config_instruction(
+        authority,
+        None,
+        Some(deserialized_account.config),
+    );
+    println!("instruction: {:?}", instruction);
+    println!(
+        "Serialized instruction data: {}",
+        bs58::encode(instruction.data.clone()).into_string()
+    );
+    let message = Message::new(&[instruction], Some(&authority));
+    let serialized_message = bs58::encode(message.serialize()).into_string();
+    println!(
+        "\n ----------- Use the serialized message in the squads tx builder. --------------- \n"
+    );
+    println!("serialized message: {}", serialized_message);
+
+    Ok(())
+}

xtask/src/main.rs

@@ -3,6 +3,7 @@ use clap::{Parser, ValueEnum};
 mod bench;
 mod create_batch_state_tree;
 mod create_state_tree;
+mod create_update_protocol_config_ix;
 mod create_vkeyrs_from_gnark_key;
 mod export_photon_test_data;
 mod fee;
@@ -52,6 +53,8 @@ enum Command {
     /// cargo xtask init-new-deployment --keypairs ../light-keypairs --network local --num-foresters 3
     /// Requires program ids to be changed manually in programs.
     InitNewDeployment(new_deployment::Options),
+    /// cargo xtask create-update-protocol-config --slot-length <u64>
+    CreateUpdateProtocolConfigIx(create_update_protocol_config_ix::Options),
 }
 
 #[tokio::main]
@@ -78,5 +81,8 @@ async fn main() -> Result<(), anyhow::Error> {
             create_batch_state_tree::create_batch_state_tree(opts).await
         }
         Command::InitNewDeployment(opts) => new_deployment::init_new_deployment(opts).await,
+        Command::CreateUpdateProtocolConfigIx(opts) => {
+            create_update_protocol_config_ix::create_update_protocol_config_ix(opts).await
+        }
     }
 }