Cap Notifier

Author: MirthFutures

contracts/feeReceiver/FeeReceiver.sol

@@ -39,11 +39,12 @@ contract FeeReceiver is IFeeReceiver, UUPSUpgradeable, Access, FeeReceiverStorag
             if ($.protocolFeePercentage > 0) _claimProtocolFees();
             uint256 bal = $.capToken.balanceOf(address(this));
             $.capToken.safeTransfer(address($.stakedCapToken), bal);
-            if ($.stakedCapToken.lastNotify() + $.stakedCapToken.lockDuration() < block.timestamp) {
-                $.stakedCapToken.notify();
-            }
             emit FeesDistributed(bal);
         }
+
+        if ($.stakedCapToken.lastNotify() + $.stakedCapToken.lockDuration() < block.timestamp) {
+            $.stakedCapToken.notify();
+        }
     }
 
     /// @inheritdoc IFeeReceiver

contracts/gelato/CapNotify.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.28;
+
+import { ICapNotify } from "../interfaces/ICapNotify.sol";
+
+import { IFeeReceiver } from "../interfaces/IFeeReceiver.sol";
+import { IStakedCap } from "../interfaces/IStakedCap.sol";
+
+import { CapNotifyStorageUtils } from "../storage/CapNotifyStorageUtils.sol";
+import { Initializable } from "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
+import { UUPSUpgradeable } from "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
+
+/// @title Cap Notifier
+/// @author weso, Cap Labs
+/// @notice Calls distribute from the Fee Receiver contract
+contract CapNotify is ICapNotify, CapNotifyStorageUtils, Initializable {
+    /// @inheritdoc ICapNotify
+    function initialize(address _feeReceiver, address _stakedCap) external initializer {
+        CapNotifyStorage storage $ = getCapNotifyStorage();
+        $.feeReceiver = _feeReceiver;
+        $.stakedCap = _stakedCap;
+    }
+
+    /// @inheritdoc ICapNotify
+    function notify() external {
+        CapNotifyStorage storage $ = getCapNotifyStorage();
+        IFeeReceiver($.feeReceiver).distribute();
+    }
+
+    /// @inheritdoc ICapNotify
+    function checker() external view returns (bool canExec, bytes memory execPayload) {
+        CapNotifyStorage storage $ = getCapNotifyStorage();
+        uint256 lastNotify = IStakedCap($.stakedCap).lastNotify();
+        uint256 duration = IStakedCap($.stakedCap).lockDuration();
+        if (lastNotify + duration > block.timestamp) return (false, bytes("Not time to notify"));
+
+        return (true, abi.encodeCall(this.notify, ()));
+    }
+}

contracts/interfaces/ICapNotify.sol

@@ -0,0 +1,20 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.28;
+
+interface ICapNotify {
+    struct CapNotifyStorage {
+        address feeReceiver;
+        address stakedCap;
+    }
+    /// @notice Initialize the CapNotify contract
+    /// @param _feeReceiver Fee receiver address
+    /// @param _stakedCap Staked cap address
+
+    function initialize(address _feeReceiver, address _stakedCap) external;
+
+    /// @notice Notify the fee receiver
+    function notify() external;
+
+    /// @notice Gelato checker
+    function checker() external view returns (bool canExec, bytes memory execPayload);
+}

contracts/storage/CapNotifyStorageUtils.sol

@@ -0,0 +1,21 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.28;
+
+import { ICapNotify } from "../interfaces/ICapNotify.sol";
+
+/// @title Cap Notify Storage Utils
+/// @author weso, Cap Labs
+/// @notice Storage utilities for cap notify
+abstract contract CapNotifyStorageUtils {
+    /// @dev keccak256(abi.encode(uint256(keccak256("cap.storage.CapNotify")) - 1)) & ~bytes32(uint256(0xff))
+    bytes32 private constant CapNotifyStorageLocation =
+        0xae7da03710cf6ca7cf92719d8e00ba51842bad60a95a2e09406012de26726900;
+
+    /// @dev Get cap notify storage
+    /// @return $ Storage pointer
+    function getCapNotifyStorage() internal pure returns (ICapNotify.CapNotifyStorage storage $) {
+        assembly {
+            $.slot := CapNotifyStorageLocation
+        }
+    }
+}