feat: update EL middleware interface

chainbound · Jan 24, 2025 · 0205433 · 0205433
1 parent bf598c4
commit 0205433
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/bolt-cli/src/commands/operators/eigenlayer.rs b/bolt-cli/src/commands/operators/eigenlayer.rs
@@ -147,11 +147,7 @@ impl EigenLayerSubcommand {
                         BoltEigenLayerMiddlewareMainnet::new(bolt_avs_address, provider.clone());
 
                     match el_middleware
-                        .registerThroughAVSDirectory(
-                            operator_rpc.to_string(),
-                            extra_data,
-                            signature,
-                        )
+                        .registerOperatorToAVS(operator_rpc.to_string(), extra_data, signature)
                         .send()
                         .await
                     {
@@ -223,7 +219,7 @@ impl EigenLayerSubcommand {
                     let el_middleware =
                         BoltEigenLayerMiddlewareMainnet::new(bolt_avs_address, provider);
 
-                    match el_middleware.deregisterThroughAVSDirectory().send().await {
+                    match el_middleware.deregisterOperatorFromAVS().send().await {
                         Ok(pending) => {
                             info!(
                                 hash = ?pending.tx_hash(),
@@ -384,7 +380,7 @@ impl EigenLayerSubcommand {
                         Err(e) => {
                             let other = try_parse_contract_error::<OperatorsRegistryV1Errors>(e)?;
                             bail!("Unexpected error with selector {:?}", other.selector())
-                        },
+                        }
                     }
 
                     match registry.isActiveOperator(address).call().await {
@@ -398,7 +394,7 @@ impl EigenLayerSubcommand {
                         Err(e) => {
                             let other = try_parse_contract_error::<OperatorsRegistryV1Errors>(e)?;
                             bail!("Unexpected error with selector {:?}", other.selector())
-                        },
+                        }
                     }
 
                     match el_middleware.getOperatorCollaterals(address).call().await {

diff --git a/bolt-cli/src/contracts/bolt.rs b/bolt-cli/src/contracts/bolt.rs
@@ -120,13 +120,15 @@ sol! {
 
         function getOperatorStake(address operator, address collateral) public view returns (uint256);
 
-        function registerThroughAVSDirectory(
+        function registerOperatorToAVS(
             string memory rpcEndpoint,
             string memory extraData,
             SignatureWithSaltAndExpiry calldata operatorSignature
         ) public;
 
-        function deregisterThroughAVSDirectory() public;
+        function deregisterOperatorFromAVS() public;
+
+        function updateOperatorsRegistryAddress(address newOperatorsRegistry) public;
 
         error InvalidRpc();
         error InvalidSigner();