feat: new method create_slashable_stake_quorum (#340)

### What Changed? Add new method `create_slashable_stake_quorum` to `avsregistry/writer`. Related issue #310 ### Reviewer Checklist - [ ] New features are tested and documented - [ ] PR updates the changelog with a description of changes - [ ] PR has one of the `changelog-X` labels (if applies) - [ ] Code deprecates any old functionality before removing it
Layr-Labs · Feb 12, 2025 · 80f5c62 · 80f5c62
1 parent f8cd9e0
commit 80f5c62
Show file tree

Hide file tree

Showing 2 changed files with 137 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -122,7 +122,33 @@ Those changes in added, changed or breaking changes, should include usage exampl
         .await
         .unwrap();
   ```
-
+
+* Added new method `create_slashable_stake_quorum` in `avsregistry/writer` in [#340](https://github.com/Layr-Labs/eigensdk-rs/pull/340).
+
+  ```rust
+      let operator_set_param = OperatorSetParam {
+          maxOperatorCount: 10,
+          kickBIPsOfOperatorStake: 50,
+          kickBIPsOfTotalStake: 50,
+      };
+      let minimum_stake = U96::from(100);
+      let strategy_param = StrategyParams {
+          strategy: get_erc20_mock_strategy(http_endpoint.clone()).await,
+          multiplier: U96::from(1),
+      };
+      let look_ahead_period = 10;
+
+      let tx_hash = avs_writer
+          .create_slashable_stake_quorum(
+              operator_set_param,
+              minimum_stake,
+              vec![strategy_param],
+              look_ahead_period,
+          )
+          .await
+          .unwrap();
+  ```
+
 * Added new method `set_ejector` in `avsregistry/writer` in [#330](https://github.com/Layr-Labs/eigensdk-rs/pull/330).
 
   ```rust
@@ -131,6 +157,7 @@ Those changes in added, changed or breaking changes, should include usage exampl
   ````
 
 * Added new method `set_ejection_cooldown` in `avsregistry/writer` in [#337](https://github.com/Layr-Labs/eigensdk-rs/pull/337).
+
   ```rust
       let new_cooldown = U256::from(120);
       let tx_hash = avs_writer
@@ -606,4 +633,4 @@ Those changes in added, changed or breaking changes, should include usage exampl
 ## Previous versions
 
 This changelog was introduced in-between v0.1.2 and v0.1.3.
-For changes from previous releases, you can check on our GitHub repo's releases page: [github.com/Layr-Labs/eigensdk-rs/releases](https://github.com/Layr-Labs/eigensdk-rs/releases)
+For changes from previous releases, you can check on our GitHub repo's releases page: [github.com/Layr-Labs/eigensdk-rs/releases](https://github.com/Layr-Labs/eigensdk-rs/releases)
diff --git a/crates/chainio/clients/avsregistry/src/writer.rs b/crates/chainio/clients/avsregistry/src/writer.rs
@@ -1,4 +1,5 @@
 use crate::error::AvsRegistryError;
+use alloy::primitives::aliases::U96;
 use alloy::primitives::{Address, Bytes, FixedBytes, TxHash, U256};
 use alloy::signers::local::PrivateKeySigner;
 use alloy::signers::Signer;
@@ -8,6 +9,7 @@ use eigen_crypto_bls::{
 };
 use eigen_logging::logger::SharedLogger;
 use eigen_utils::slashing::middleware::registrycoordinator::ISlashingRegistryCoordinatorTypes::OperatorSetParam;
+use eigen_utils::slashing::middleware::registrycoordinator::IStakeRegistryTypes::StrategyParams;
 use eigen_utils::slashing::middleware::registrycoordinator::{
     IBLSApkRegistryTypes::PubkeyRegistrationParams, ISignatureUtils::SignatureWithSaltAndExpiry,
     RegistryCoordinator,
@@ -459,6 +461,43 @@ impl AvsRegistryChainWriter {
         Ok(*tx.tx_hash())
     }
 
+    /// Create a new quorum that tracks slashable stake for operators
+    ///
+    /// # Arguments
+    ///
+    /// * `operator_set_param` - Configures the quorum's max operator count and churn parameters
+    /// * `minimum_stake` - Sets the minimum stake required for an operator to register or remain registered
+    /// * `strategy_params` - A list of strategies and multipliers used by the StakeRegistry to calculate an operator's stake weight for the quorum
+    /// * `look_ahead_period` - The number of blocks to look ahead when calculating slashable stake
+    ///
+    /// # Returns
+    ///
+    /// * `TxHash` - The transaction hash of the create slashable stake quorum transactions
+    pub async fn create_slashable_stake_quorum(
+        &self,
+        operator_set_param: OperatorSetParam,
+        minimum_stake: U96,
+        strategy_params: Vec<StrategyParams>,
+        look_ahead_period: u32,
+    ) -> Result<TxHash, AvsRegistryError> {
+        info!("creating slashable stake quorum with the AVS's registry coordinator");
+        let provider = get_signer(&self.signer.clone(), &self.provider);
+
+        let contract_registry_coordinator =
+            RegistryCoordinator::new(self.registry_coordinator_addr, provider);
+
+        contract_registry_coordinator.createSlashableStakeQuorum(
+            operator_set_param,
+            minimum_stake,
+            strategy_params,
+            look_ahead_period,
+            ).send()
+            .await
+            .map_err(AvsRegistryError::AlloyContractError)
+            .inspect(|tx| info!(tx_hash = ?tx,"successfully created slashable stake quorum with the AVS's registry coordinator" ))
+            .map(|tx| *tx.tx_hash())
+    }
+
     /// Set the ejection cooldow that an operator must wait in seconds after ejection before registering for any quorum
     ///
     /// # Arguments
@@ -518,19 +557,25 @@ mod tests {
     use crate::test_utils::{
         build_avs_registry_chain_writer, test_deregister_operator, test_register_operator,
     };
+    use alloy::primitives::aliases::U96;
     use alloy::primitives::U256;
     use alloy::primitives::{Address, Bytes};
+    use alloy::sol_types::SolCall;
     use eigen_common::{get_provider, get_signer};
     use eigen_testing_utils::anvil::{start_anvil_container, start_m2_anvil_container};
+    use eigen_testing_utils::anvil_constants::get_allocation_manager_address;
+    use eigen_testing_utils::anvil_constants::get_erc20_mock_strategy;
     use eigen_testing_utils::anvil_constants::get_service_manager_address;
     use eigen_testing_utils::anvil_constants::{
         FIFTH_ADDRESS, FIFTH_PRIVATE_KEY, FIRST_ADDRESS, FIRST_PRIVATE_KEY, OPERATOR_BLS_KEY,
         SECOND_ADDRESS,
     };
     use eigen_testing_utils::transaction::wait_transaction;
-    use eigen_utils::rewardsv2::middleware::servicemanagerbase::ServiceManagerBase;
+    use eigen_utils::slashing::core::allocationmanager::AllocationManager;
     use eigen_utils::slashing::middleware::registrycoordinator::ISlashingRegistryCoordinatorTypes::OperatorSetParam;
+    use eigen_utils::slashing::middleware::registrycoordinator::IStakeRegistryTypes::StrategyParams;
     use eigen_utils::slashing::middleware::registrycoordinator::RegistryCoordinator;
+    use eigen_utils::slashing::middleware::servicemanagerbase::ServiceManagerBase;
     use eigen_utils::slashing::middleware::stakeregistry::StakeRegistry;
     use futures_util::StreamExt;
 
@@ -788,6 +833,67 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_create_slashable_stake_quorum() {
+        let (_container, http_endpoint, _ws_endpoint) = start_anvil_container().await;
+        let private_key = FIRST_PRIVATE_KEY.to_string();
+        let avs_writer =
+            build_avs_registry_chain_writer(http_endpoint.clone(), private_key.clone()).await;
+
+        let service_manager_address = get_service_manager_address(http_endpoint.to_string()).await;
+
+        let service_manager = ServiceManagerBase::new(
+            service_manager_address,
+            get_signer(&avs_writer.signer.clone(), &avs_writer.provider),
+        );
+
+        let allocation_manager_addr = get_allocation_manager_address(http_endpoint.clone()).await;
+
+        service_manager
+            .setAppointee(
+                avs_writer.registry_coordinator_addr,
+                allocation_manager_addr,
+                alloy::primitives::FixedBytes(AllocationManager::createOperatorSetsCall::SELECTOR),
+            )
+            .send()
+            .await
+            .unwrap()
+            .get_receipt()
+            .await
+            .unwrap();
+
+        let operator_set_param = OperatorSetParam {
+            maxOperatorCount: 10,
+            kickBIPsOfOperatorStake: 50,
+            kickBIPsOfTotalStake: 50,
+        };
+        let minimum_stake = U96::from(100);
+
+        let strategy_param = StrategyParams {
+            strategy: get_erc20_mock_strategy(http_endpoint.clone()).await,
+            multiplier: U96::from(1),
+        };
+
+        let look_ahead_period = 10;
+
+        let tx_hash = avs_writer
+            .create_slashable_stake_quorum(
+                operator_set_param.clone(),
+                minimum_stake,
+                vec![strategy_param],
+                look_ahead_period,
+            )
+            .await
+            .unwrap();
+
+        let tx_status = wait_transaction(&http_endpoint, tx_hash)
+            .await
+            .unwrap()
+            .status();
+
+        assert!(tx_status);
+    }
+
     #[tokio::test]
     async fn test_set_ejector_cooldown() {
         let (_container, http_endpoint, _ws_endpoint) = start_m2_anvil_container().await;
@@ -863,7 +969,7 @@ mod tests {
     }
 
     #[tokio::test]
-    async fn test_set_ejecttor() {
+    async fn test_set_ejector() {
         let (_container, http_endpoint, _ws_endpoint) = start_m2_anvil_container().await;
         let private_key = FIRST_PRIVATE_KEY.to_string();
         let new_ejector_address = SECOND_ADDRESS;