remove challengers

hyperlane-xyz · Apr 25, 2024 · f737489 · f737489
1 parent 0488bc5
commit f737489
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/solidity/contracts/avs/HyperlaneServiceManager.sol b/solidity/contracts/avs/HyperlaneServiceManager.sol
@@ -46,6 +46,23 @@ contract HyperlaneServiceManager is ECDSAServiceManagerBase {
     function deregisterOperatorFromAVS(
         address operator
     ) public virtual override onlyStakeRegistry {
+        address[] memory challengers = enrolledChallengers[operator].keys();
+        for (uint256 i = 0; i < challengers.length; i++) {
+            IRemoteChallenger challenger = IRemoteChallenger(challengers[i]);
+            Enrollment memory enrollment = enrolledChallengers[operator].get(
+                challengers[i]
+            );
+            require(
+                enrollment.status != EnrollmentStatus.ENROLLED,
+                string(
+                    abi.encodePacked(
+                        "HyperlaneServiceManager: Operator still enrolled in challenger",
+                        challengers[i]
+                    )
+                )
+            );
+            enrolledChallengers[operator].remove(challengers[i]);
+        }
         elAvsDirectory.deregisterOperatorFromAVS(operator);
     }
 

diff --git a/solidity/contracts/libs/EnumerableMapEnrollment.sol b/solidity/contracts/libs/EnumerableMapEnrollment.sol
@@ -9,8 +9,7 @@ import {TypeCasts} from "./TypeCasts.sol";
 
 enum EnrollmentStatus {
     ENROLLED,
-    PENDING_UNENROLLMENT,
-    UNENROLLED
+    PENDING_UNENROLLMENT
 }
 
 struct Enrollment {