construct bonus wrapper with gauge

tranchess · Oct 18, 2024 · 5dc7cc0 · 5dc7cc0
1 parent ce1fc1d
commit 5dc7cc0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/contracts/swap/SwapBonusWrapper.sol b/contracts/swap/SwapBonusWrapper.sol
@@ -6,6 +6,7 @@ import "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
 import "@openzeppelin/contracts/access/Ownable.sol";
 import "@openzeppelin/contracts/math/SafeMath.sol";
 
+import "./LiquidityGaugeV3.sol";
 import "./SwapBonus.sol";
 
 contract SwapBonusWrapper is Ownable {
@@ -14,13 +15,17 @@ contract SwapBonusWrapper is Ownable {
 
     address public immutable swapBonus;
     address public immutable bonusToken;
+    LiquidityGaugeV3 public immutable liquidityGauge;
 
-    constructor(address swapBonus_) public {
+    constructor(LiquidityGaugeV3 liquidityGauge_) public {
+        liquidityGauge = liquidityGauge_;
+        address swapBonus_ = liquidityGauge_.swapBonus();
         swapBonus = swapBonus_;
         bonusToken = SwapBonus(swapBonus_).bonusToken();
     }
 
     function updateBonus(uint256 amount, uint256 interval) external onlyOwner {
+        liquidityGauge.syncWithVotingEscrow(address(0));
         uint256 realAmount = amount.div(interval).mul(interval);
         IERC20(bonusToken).safeTransferFrom(msg.sender, address(this), realAmount);
         IERC20(bonusToken).approve(swapBonus, realAmount);