@@ -309,11 +309,15 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
309
309
(address [] memory tokens , uint256 [] memory rewards ) = IStargateV2MultiRewarder (_rewarder).getRewards (address (lpToken), address (this ));
310
310
311
311
uint256 _index = _findIndex (tokens, STG);
312
- uint256 rewardAmount = rewards[_index];
313
- if (rewardAmount == 0 ) return 0 ;
312
+ uint256 stgRewardAmount = rewards[_index];
313
+ _index = _findIndex (tokens, ARB);
314
+ uint256 arbRewardAmount = rewards[_index];
315
+ if (stgRewardAmount == 0 && arbRewardAmount == 0 ) return 0 ;
314
316
315
317
(, uint256 stgPrice ) = stgInputTokenOracle.peek (stgInputTokenOracleData);
316
- amount = (rewardAmount * stgPrice) / 1e18 ;
318
+ (, uint256 arbPrice ) = arbInputTokenOracle.peek (arbInputTokenOracleData);
319
+ amount = (stgRewardAmount * stgPrice) / 1e18 ;
320
+ amount += (arbRewardAmount * arbPrice) / 1e18 ;
317
321
}
318
322
319
323
/**
@@ -410,7 +414,7 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
410
414
farm.deposit (address (lpToken), lpAmount);
411
415
lpToken.safeApprove (address (farm), 0 );
412
416
413
- event AmountDeposited (uint256 lpAmount );
417
+ emit AmountDeposited (lpAmount);
414
418
}
415
419
416
420
}
0 commit comments