add getNumOperatorSets

Layr-Labs · Sep 8, 2024 · 45b65da · 45b65da
1 parent 2445306
commit 45b65da
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/contracts/core/StakeRootCompendium.sol b/src/contracts/core/StakeRootCompendium.sol
@@ -214,6 +214,11 @@ contract StakeRootCompendium is StakeRootCompendiumStorage {
 
     /// VIEW FUNCTIONS
 
+    /// @inheritdoc IStakeRootCompendium
+    function getNumOperatorSets() external view returns (uint256) {
+        return operatorSets.length;
+    }
+
     /// @inheritdoc IStakeRootCompendium
     function getStakes(IAVSDirectory.OperatorSet calldata operatorSet, address operator)
         external

diff --git a/src/contracts/interfaces/IStakeRootCompendium.sol b/src/contracts/interfaces/IStakeRootCompendium.sol
@@ -52,6 +52,9 @@ interface IStakeRootCompendium {
     function verifier() external view returns (address);
     function imageId() external view returns (bytes32);
 
+    /// @notice the number of operator sets in the StakeTree
+    function getNumOperatorSets() external view returns (uint256);
+
     /// @notice the interval at which proofs can be posted, to not overcharge the operatorSets
     function proofInterval() external view returns (uint32);