Skip to content

Commit dbe4428

Browse files
chore: missing onlyOwner and events
1 parent 22efc13 commit dbe4428

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

contracts/StargateV2Strategy/StargateV2Strategy.sol

+6-4
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
8181
error TransferFailed();
8282
error DepositPaused();
8383
error WithdrawPaused();
84-
error NotEnough();
8584
error PauserNotAuthorized();
8685
error EmptyAddress();
8786
error SwapFailed();
@@ -154,7 +153,7 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
154153
depositPaused = true;
155154
withdrawPaused = true;
156155

157-
uint256 amount = farm.balanceOf(address(lpToken), address(this)); // TODO: check if lpToken <> token ratio is 1:1
156+
uint256 amount = farm.balanceOf(address(lpToken), address(this));
158157

159158
// withdraw from farm
160159
farm.withdraw(address(lpToken), amount);
@@ -250,7 +249,7 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
250249
/**
251250
* @notice invests currently available STG for compounding interest
252251
*/
253-
function invest(bytes calldata arbData, bytes calldata stgData) external {
252+
function invest(bytes calldata arbData, bytes calldata stgData) external onlyOwner {
254253
IERC20 _stg = IERC20(STG);
255254
IERC20 _arb = IERC20(ARB);
256255

@@ -331,7 +330,7 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
331330
// *********************************** //
332331
function _currentBalance() internal view override returns (uint256 amount) {
333332
/// @dev: wrap fees are not taken into account here because it's 0
334-
amount = farm.balanceOf(address(lpToken), address(this)); // TODO: check if lpToken <> token ratio is 1:1
333+
amount = farm.balanceOf(address(lpToken), address(this));
335334
amount += IERC20(contractAddress).balanceOf(address(this));
336335
amount += pendingRewards();
337336
}
@@ -385,6 +384,7 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
385384

386385
// send `contractAddress`
387386
IERC20(contractAddress).safeTransfer(to, amount);
387+
emit AmountWithdrawn(to, amount);
388388
}
389389

390390
// ********************************* //
@@ -409,6 +409,8 @@ contract StargateV2Strategy is BaseERC20Strategy, Ownable, ReentrancyGuard {
409409
lpToken.safeApprove(address(farm), type(uint256).max);
410410
farm.deposit(address(lpToken), lpAmount);
411411
lpToken.safeApprove(address(farm), 0);
412+
413+
event AmountDeposited(uint256 lpAmount);
412414
}
413415

414416
}

0 commit comments

Comments
 (0)