chore: rewards per token getter

curvefi · Aug 30, 2024 · 3d0ee3c · 3d0ee3c
1 parent c9a08fa
commit 3d0ee3c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/contracts/implementations/ChildGauge.vy b/contracts/implementations/ChildGauge.vy
@@ -831,6 +831,18 @@ def rate_per_lp_token(_reward_id: uint256, _with_precision: bool=False) -> uint2
     return rate / precision
 
 
+@external
+@view
+def get_rewards_of(_token: ERC20) -> DynArray[RewardData, MAX_REWARDS]:
+    rewards: DynArray[RewardData, MAX_REWARDS] = []
+    for i in range(MAX_REWARDS):
+        if i >= len(self.reward_data):
+            break
+        if self.reward_data[i].token == _token:
+            rewards.append(self.reward_data[i])
+    return rewards
+
+
 @view
 @external
 def decimals() -> uint256: