add startCheckpoint on module manager

PufferFinance · Jul 26, 2024 · e7d9dd8 · e7d9dd8
1 parent 6f1f071
commit e7d9dd8
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/mainnet-contracts/src/PufferModule.sol b/mainnet-contracts/src/PufferModule.sol
@@ -189,6 +189,15 @@ contract PufferModule is IPufferModule, Initializable, AccessManagedUpgradeable
         });
     }
 
+    /**
+     * @inheritdoc IPufferModule
+     * @dev Restricted to PufferModuleManager
+     */
+    function startCheckpoint() external virtual onlyPufferModuleManager {
+        ModuleStorage storage $ = _getPufferModuleStorage();
+        $.eigenPod.startCheckpoint({ revertIfNoBalance: true });
+    }
+
     /**
      * @dev Restricted to PufferProtocol
      */

diff --git a/mainnet-contracts/src/PufferModuleManager.sol b/mainnet-contracts/src/PufferModuleManager.sol
@@ -342,6 +342,17 @@ contract PufferModuleManager is IPufferModuleManager, AccessManagedUpgradeable,
         emit CustomCallSucceeded(address(restakingOperator), target, customCalldata, response);
     }
 
+    /**
+     * @inheritdoc IPufferModuleManager
+     * @dev Restricted to the DAO
+     */
+    function callStartCheckpoint(address[] calldata moduleAddresses) external virtual restricted {
+        for (uint256 i = 0; i < moduleAddresses.length; i++) {
+            // reverts if supplied with a duplicate module address
+            IPufferModule(moduleAddresses[i]).startCheckpoint();
+        }
+    }
+
     /**
      * @inheritdoc IPufferModuleManager
      * @dev Restricted to the DAO

diff --git a/mainnet-contracts/src/interface/IPufferModule.sol b/mainnet-contracts/src/interface/IPufferModule.sol
@@ -113,6 +113,11 @@ interface IPufferModule {
         bool[] calldata receiveAsTokens
     ) external;
 
+    /**
+     * @notice Starts the checkpoint for the module
+     */
+    function startCheckpoint() external;
+
     /**
      * @notice Function callable only by PufferProtocol
      * @param to is the destination address

diff --git a/mainnet-contracts/src/interface/IPufferModuleManager.sol b/mainnet-contracts/src/interface/IPufferModuleManager.sol
@@ -219,6 +219,11 @@ interface IPufferModuleManager {
      */
     function callSetProofSubmitter(bytes32 moduleName, address proofSubmitter) external;
 
+    /**
+     * @notice Starts the checkpointing on puffer modules
+     */
+    function callStartCheckpoint(address[] calldata moduleAddresses) external;
+
     /**
      * @notice Calls the modifyOperatorDetails function on the restaking operator
      * @param restakingOperator is the address of the restaking operator