getOperatorSetLeaves

Layr-Labs · Sep 16, 2024 · 82cf8d1 · 82cf8d1
1 parent b855db0
commit 82cf8d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/contracts/core/StakeRootCompendium.sol b/src/contracts/core/StakeRootCompendium.sol
@@ -466,7 +466,7 @@ contract StakeRootCompendium is StakeRootCompendiumStorage {
         uint256 operatorSetIndex,
         uint256 startOperatorIndex,
         uint256 numOperators
-    ) external view returns (OperatorLeaf[] memory) {
+    ) external view returns (OperatorSet memory, address[] memory, OperatorLeaf[] memory) {
         require(
             operatorSetIndex < operatorSets.length,
             "StakeRootCompendium.getOperatorSetLeaves: operator set index out of bounds"
@@ -486,7 +486,7 @@ contract StakeRootCompendium is StakeRootCompendiumStorage {
                 extraData: operatorExtraDatas[operatorSet.avs][operatorSet.operatorSetId][operators[i]]
             });
         }
-        return operatorLeaves;
+        return (operatorSet, operators, operatorLeaves);
     }
 
     /// @inheritdoc IStakeRootCompendium

diff --git a/src/contracts/interfaces/IStakeRootCompendium.sol b/src/contracts/interfaces/IStakeRootCompendium.sol
@@ -99,13 +99,13 @@ interface IStakeRootCompendium {
      * @param operatorSetIndex the index of the operatorSet within the SRC's operatorSets list to calculate the operator set leaves for
      * @param startOperatorIndex the index of the first operator to get the leaves for
      * @param numOperators the number of operators to get the leaves for
-     * @return the operatorSet leaves
+     * @return the operatorSet, the operators, and the operator leaves
      */
     function getOperatorSetLeaves(
         uint256 operatorSetIndex,
         uint256 startOperatorIndex,
         uint256 numOperators
-    ) external view returns (OperatorLeaf[] memory);
+    ) external view returns (OperatorSet memory, address[] memory, OperatorLeaf[] memory);
 
     /**
      * @notice deposits funds for an operator set