feat(contracts): add vault / strategy deregistration

chainbound · Oct 18, 2024 · 096e76e · 096e76e
1 parent f4c4131
commit 096e76e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/bolt-contracts/src/contracts/BoltEigenLayerMiddlewareV1.sol b/bolt-contracts/src/contracts/BoltEigenLayerMiddlewareV1.sol
@@ -153,6 +153,18 @@ contract BoltEigenLayerMiddlewareV1 is IBoltMiddlewareV1, OwnableUpgradeable, UU
         strategies.enable(strategy);
     }
 
+    /// @notice Deregister a strategy from working in Bolt Protocol.
+    /// @param strategy The EigenLayer strategy address.
+    function deregisterStrategy(
+        address strategy
+    ) public onlyOwner {
+        if (!strategies.contains(strategy)) {
+            revert NotRegistered();
+        }
+
+        strategies.remove(strategy);
+    }
+
     // ========= EIGENLAYER MIDDLEWARE LOGIC =========
 
     /// @notice Allow an operator to signal opt-in to Bolt Protocol.

diff --git a/bolt-contracts/src/contracts/BoltSymbioticMiddlewareV1.sol b/bolt-contracts/src/contracts/BoltSymbioticMiddlewareV1.sol
@@ -171,6 +171,18 @@ contract BoltSymbioticMiddlewareV1 is IBoltMiddlewareV1, OwnableUpgradeable, UUP
         vaults.enable(vault);
     }
 
+    /// @notice Deregister a vault from working in Bolt Protocol.
+    /// @param vault The vault address to deregister.
+    function deregisterVault(
+        address vault
+    ) public onlyOwner {
+        if (!vaults.contains(vault)) {
+            revert NotRegistered();
+        }
+
+        vaults.remove(vault);
+    }
+
     // ========= SYMBIOTIC MIDDLEWARE LOGIC =========
 
     /// @notice Allow an operator to signal opt-in to Bolt Protocol.