fix: add missing struct defintion

Layr-Labs · Sep 4, 2024 · e5b375b · e5b375b
1 parent 9f95d3b
commit e5b375b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/contracts/core/AVSDirectory.sol b/src/contracts/core/AVSDirectory.sol
@@ -1041,7 +1041,7 @@ contract AVSDirectory is
         uint16 numToComplete
     ) external view returns (uint64) {
         OperatorMagnitudeInfo storage info = operatorMagnitudeInfo[operator][strategy];
-        (, uint64 freeMagnitudeToAdd) =
+        (uint64 freeMagnitudeToAdd, ) =
             _getPendingFreeMagnitude(operator, strategy, numToComplete, info.nextPendingFreeMagnitudeIndex);
         return info.freeMagnitude + freeMagnitudeToAdd;
     }

diff --git a/src/contracts/interfaces/IAVSDirectory.sol b/src/contracts/interfaces/IAVSDirectory.sol
@@ -54,6 +54,16 @@ interface IAVSDirectory is ISignatureUtils {
         uint32 completableTimestamp;
     }
 
+    /**
+     * @notice Struct containing info regarding free allocatable magnitude.
+     * @param nextPendingFreeMagnitudeIndex The next available update index.
+     * @param freeMagnitude The total amount of free allocatable magnitude.
+     */
+    struct OperatorMagnitudeInfo {
+        uint192 nextPendingFreeMagnitudeIndex;
+        uint64 freeMagnitude;
+    }
+
     /// @notice Emitted when an operator set is created by an AVS.
     event OperatorSetCreated(OperatorSet operatorSet);