feat: add permission for harvesting protocol rewards

astrodevs-labs · Nov 22, 2023 · c97bdcd · c97bdcd
1 parent 764abce
commit c97bdcd
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/contracts/src/abstracts/AProtocolClaimer.sol b/contracts/src/abstracts/AProtocolClaimer.sol
@@ -1,9 +1,15 @@
 // SPDX-License-Identifier: GPL-3.0-or-later
 pragma solidity 0.8.20;
 
-abstract contract AProtocolClaimer {
-    event ProtocolRewardsHarvest();
+import { ReentrancyGuard } from "solmate/utils/ReentrancyGuard.sol";
+import { AOperator } from "./AOperator.sol";
 
+/**
+ * @title Incentivized Locker contract
+ * @author 0xtekgrinder
+ * @notice Locker contract capable of claiming protocol rewards
+ */
+abstract contract AProtocolClaimer is ReentrancyGuard, AOperator {
     /**
      * @dev Harvest rewards & send them to the Controller
      */
@@ -12,8 +18,7 @@ abstract contract AProtocolClaimer {
     /**
      * @notice Harvest rewards
      */
-    function harvest() external {
-        emit ProtocolRewardsHarvest();
+    function harvest() external nonReentrant onlyOperatorOrOwner {
         _harvest();
     }
 }