Add support for a rewards delegate

vesperfi · Jul 13, 2023 · d767b37 · d767b37
1 parent ec46174
commit d767b37
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/vesper-pools/contracts/interfaces/vesper/IPoolRewards.sol b/vesper-pools/contracts/interfaces/vesper/IPoolRewards.sol
@@ -9,6 +9,8 @@ interface IPoolRewards {
     event RewardPaid(address indexed user, address indexed rewardToken, uint256 reward);
     /// Emitted after adding new rewards token into rewardTokens array
     event RewardTokenAdded(address indexed rewardToken, address[] existingRewardTokens);
+    /// Emitted after setting the reward delegate for a user
+    event RewardsDelegateSet(address indexed user, address indexed delegate);
 
     function claimReward(address) external;
 

diff --git a/vesper-pools/contracts/pool/PoolRewards.sol b/vesper-pools/contracts/pool/PoolRewards.sol
@@ -39,6 +39,9 @@ abstract contract PoolRewardsStorage is IPoolRewards {
 
     /// RewardToken => User => Rewards earned till last reward update
     mapping(address => mapping(address => uint256)) public rewards;
+
+    // User => Address rewards are delegated to
+    mapping(address => address) public rewardsDelegate;
 }
 
 /// @title Distribute rewards based on vesper pool balance and supply
@@ -178,7 +181,11 @@ contract PoolRewards is Initializable, ReentrancyGuard, PoolRewardsStorage {
         // Mark reward as claimed
         rewards[rewardToken_][account_] = 0;
         // Transfer reward
-        IERC20(rewardToken_).safeTransfer(account_, reward_);
+        if (rewardsDelegate[account_] != address(0)) {
+            IERC20(rewardToken_).safeTransfer(rewardsDelegate[account_], reward_);
+        } else {
+            IERC20(rewardToken_).safeTransfer(account_, reward_);
+        }
     }
 
     // There are scenarios when extending contract will override external methods and
@@ -287,4 +294,11 @@ contract PoolRewards is Initializable, ReentrancyGuard, PoolRewardsStorage {
     ) external virtual override onlyAuthorized {
         _notifyRewardAmount(rewardToken_, rewardAmount_, rewardDuration_, IERC20(pool).totalSupply());
     }
+
+    function setRewardsDelegate(address user_, address delegate_) external onlyAuthorized {
+        require(user_ != delegate_, "rewards-delegate-is-user");
+        require(rewardsDelegate[user_] != delegate_, "rewards-delegate-already-set");
+        emit RewardsDelegateSet(user_, delegate_);
+        rewardsDelegate[user_] = delegate_;
+    }
 }