Skip to content

Commit

Permalink
always be linting
Browse files Browse the repository at this point in the history
  • Loading branch information
alcueca committed Apr 16, 2024
1 parent 8e2b141 commit 822dc8c
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 9 deletions.
14 changes: 8 additions & 6 deletions src/gnosissafe/GnosisSafeWrapper.sol
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ contract GnosisSafeWrapper is BaseWrapper, AccessControl {

IGnosisSafe public immutable safe;

mapping (address asset => LendingData data) public lending;
mapping(address asset => LendingData data) public lending;

constructor(IGnosisSafe _safe) {
safe = _safe;
Expand All @@ -40,20 +40,22 @@ contract GnosisSafeWrapper is BaseWrapper, AccessControl {
function flashFee(address asset, uint256 amount) public view returns (uint256) {
uint256 max = maxFlashLoan(asset);
if (max == 0) revert UnsupportedAsset(asset);
return amount >= max ? type(uint256).max : amount * lending[asset].fee / 10000;
return amount >= max ? type(uint256).max : amount * lending[asset].fee / 10_000;
}

function _flashLoan(address asset, uint256 amount, bytes memory params) internal override {

Data memory decodedParams = abi.decode(params, (Data));

if (lending[asset].enabled == false) revert UnsupportedAsset(asset);
uint256 fee = flashFee(asset, amount);
uint256 balanceAfter = IERC20(asset).balanceOf(address(safe)) + fee;

// Take assets from safe
bytes memory transferCall = abi.encodeWithSignature("transfer(address,uint256)", decodedParams.loanReceiver, amount);
if (!safe.execTransactionFromModule(asset, 0, transferCall, Enum.Operation.Call)) revert FailedTransfer(asset, amount);
bytes memory transferCall =
abi.encodeWithSignature("transfer(address,uint256)", decodedParams.loanReceiver, amount);
if (!safe.execTransactionFromModule(asset, 0, transferCall, Enum.Operation.Call)) {
revert FailedTransfer(asset, amount);
}

// Call callback
_bridgeToCallback(asset, amount, 0, params);
Expand Down
6 changes: 4 additions & 2 deletions src/gnosissafe/interfaces/IGnosisSafe.sol
Original file line number Diff line number Diff line change
Expand Up @@ -14,5 +14,7 @@ interface IGnosisSafe {
uint256 value,
bytes calldata data,
Enum.Operation operation
) external returns (bool success);
}
)
external
returns (bool success);
}
2 changes: 1 addition & 1 deletion src/gnosissafe/lib/Enum.sol
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,4 @@ contract Enum {
Call,
DelegateCall
}
}
}

0 comments on commit 822dc8c

Please sign in to comment.