feat: chainIDs

Author: 0xrajath

src/contracts/interfaces/ICrossChainRegistry.sol

@@ -37,6 +37,9 @@ interface ICrossChainRegistryErrors {
 
     /// @notice Thrown when the operator set is not valid
     error InvalidOperatorSet();
+
+    /// @notice Thrown when the chainIDs array is empty
+    error EmptyChainIDsArray();
 }
 
 interface ICrossChainRegistryTypes {
@@ -140,21 +143,21 @@ interface ICrossChainRegistry is ICrossChainRegistryErrors, ICrossChainRegistryE
     function removeTransportDestinations(OperatorSet calldata operatorSet, uint32[] calldata chainIDs) external;
 
     /**
-     * @notice Adds a chainID to the whitelist of chainIDs that can be transported to
-     * @param chainID the chainID to add to the whitelist
+     * @notice Adds chainIDs to the whitelist of chainIDs that can be transported to
+     * @param chainIDs the chainIDs to add to the whitelist
      * @dev msg.sender must be the owner of the CrossChainRegistry
      */
-    function addChainIDToWhitelist(
-        uint32 chainID
+    function addChainIDsToWhitelist(
+        uint32[] calldata chainIDs
     ) external;
 
     /**
-     * @notice Removes a chainID from the whitelist of chainIDs that can be transported to
-     * @param chainID the chainID to remove from the whitelist
+     * @notice Removes chainIDs from the whitelist of chainIDs that can be transported to
+     * @param chainIDs the chainIDs to remove from the whitelist
      * @dev msg.sender must be the owner of the CrossChainRegistry
      */
-    function removeChainIDFromWhitelist(
-        uint32 chainID
+    function removeChainIDsFromWhitelist(
+        uint32[] calldata chainIDs
     ) external;
 
     /**

src/multichain/CrossChainRegistry.sol

@@ -43,6 +43,17 @@ contract CrossChainRegistry is
         _;
     }
 
+    /**
+     * @dev Validates that the chainIDs array is not empty
+     * @param chainIDs The array of chain IDs to validate
+     */
+    modifier nonEmptyChainIDsArray(
+        uint32[] calldata chainIDs
+    ) {
+        require(chainIDs.length > 0, EmptyChainIDsArray());
+        _;
+    }
+
     /**
      *
      *                         INITIALIZING FUNCTIONS
@@ -183,6 +194,7 @@ contract CrossChainRegistry is
         onlyWhenNotPaused(PAUSED_TRANSPORT_DESTINATIONS)
         checkCanCall(operatorSet.avs)
         isValidOperatorSet(operatorSet)
+        nonEmptyChainIDsArray(chainIDs)
     {
         bytes32 operatorSetKey = operatorSet.key();
 
@@ -205,6 +217,7 @@ contract CrossChainRegistry is
         onlyWhenNotPaused(PAUSED_TRANSPORT_DESTINATIONS)
         checkCanCall(operatorSet.avs)
         isValidOperatorSet(operatorSet)
+        nonEmptyChainIDsArray(chainIDs)
     {
         bytes32 operatorSetKey = operatorSet.key();
 
@@ -222,33 +235,41 @@ contract CrossChainRegistry is
     }
 
     /// @inheritdoc ICrossChainRegistry
-    function addChainIDToWhitelist(
-        uint32 chainID
-    ) external onlyOwner onlyWhenNotPaused(PAUSED_CHAIN_WHITELIST) {
-        // Validate chainID
-        require(chainID != 0, InvalidChainId());
-        // Check if already whitelisted
-        require(!_whitelistedChainIDs.contains(chainID), ChainIDAlreadyWhitelisted());
+    function addChainIDsToWhitelist(
+        uint32[] calldata chainIDs
+    ) external onlyOwner onlyWhenNotPaused(PAUSED_CHAIN_WHITELIST) nonEmptyChainIDsArray(chainIDs) {
+        for (uint256 i = 0; i < chainIDs.length; i++) {
+            uint32 chainID = chainIDs[i];
+
+            // Validate chainID
+            require(chainID != 0, InvalidChainId());
+            // Check if already whitelisted
+            require(!_whitelistedChainIDs.contains(chainID), ChainIDAlreadyWhitelisted());
 
-        // Add to whitelist
-        _whitelistedChainIDs.add(chainID);
+            // Add to whitelist
+            _whitelistedChainIDs.add(chainID);
 
-        emit ChainIDAddedToWhitelist(chainID);
+            emit ChainIDAddedToWhitelist(chainID);
+        }
     }
 
     /// @inheritdoc ICrossChainRegistry
-    function removeChainIDFromWhitelist(
-        uint32 chainID
-    ) external onlyOwner onlyWhenNotPaused(PAUSED_CHAIN_WHITELIST) {
-        // Validate chainID
-        require(chainID != 0, InvalidChainId());
-        // Check if whitelisted
-        require(_whitelistedChainIDs.contains(chainID), ChainIDNotWhitelisted());
+    function removeChainIDsFromWhitelist(
+        uint32[] calldata chainIDs
+    ) external onlyOwner onlyWhenNotPaused(PAUSED_CHAIN_WHITELIST) nonEmptyChainIDsArray(chainIDs) {
+        for (uint256 i = 0; i < chainIDs.length; i++) {
+            uint32 chainID = chainIDs[i];
 
-        // Remove from whitelist
-        _whitelistedChainIDs.remove(chainID);
+            // Validate chainID
+            require(chainID != 0, InvalidChainId());
+            // Check if whitelisted
+            require(_whitelistedChainIDs.contains(chainID), ChainIDNotWhitelisted());
 
-        emit ChainIDRemovedFromWhitelist(chainID);
+            // Remove from whitelist
+            _whitelistedChainIDs.remove(chainID);
+
+            emit ChainIDRemovedFromWhitelist(chainID);
+        }
     }
 
     /**