Added WETHGateway

This commit is contained in:
David Racero 2020-10-28 15:41:31 +01:00
parent 1ef22f036b
commit 8ad022979b
4 changed files with 156 additions and 0 deletions

View File

@ -0,0 +1,16 @@
// SPDX-License-Identifier: agpl-3.0
pragma solidity ^0.6.8;
interface IWETH {
function deposit() external payable;
function withdraw(uint256) external;
function approve(address guy, uint256 wad) external returns (bool);
function transferFrom(
address src,
address dst,
uint256 wad
) external returns (bool);
}

View File

@ -0,0 +1,14 @@
// SPDX-License-Identifier: agpl-3.0
pragma solidity ^0.6.8;
interface IWETHGateway {
function depositETH(address onBehalfOf, uint16 referralCode) external payable;
function withdrawETH(uint256 amount) external;
function repayETH(
uint256 amount,
uint256 rateMode,
address onBehalfOf
) external payable;
}

View File

@ -26,4 +26,15 @@ library Helpers {
DebtTokenBase(reserve.variableDebtTokenAddress).balanceOf(user)
);
}
function getUserCurrentDebtViaMemory(address user, ReserveLogic.ReserveData memory reserve)
internal
view
returns (uint256, uint256)
{
return (
DebtTokenBase(reserve.stableDebtTokenAddress).balanceOf(user),
DebtTokenBase(reserve.variableDebtTokenAddress).balanceOf(user)
);
}
}

View File

@ -0,0 +1,115 @@
// SPDX-License-Identifier: agpl-3.0
pragma solidity ^0.6.8;
import {IWETH} from '../interfaces/IWETH.sol';
import {IWETHGateway} from '../interfaces/IWETHGateway.sol';
import {ILendingPool} from '../interfaces/ILendingPool.sol';
import {ILendingPoolAddressesProvider} from '../interfaces/ILendingPoolAddressesProvider.sol';
import {IAToken} from '../tokenization/interfaces/IAToken.sol';
import {ReserveLogic} from '../libraries/logic/ReserveLogic.sol';
import {ReserveConfiguration} from '../libraries/configuration/ReserveConfiguration.sol';
import {UserConfiguration} from '../libraries/configuration/UserConfiguration.sol';
import {Helpers} from '../libraries/helpers/Helpers.sol';
contract WETHGateway is IWETHGateway {
using ReserveConfiguration for ReserveConfiguration.Map;
using UserConfiguration for UserConfiguration.Map;
IWETH public immutable WETH;
ILendingPoolAddressesProvider public immutable ADDRESSES_PROVIDER;
/**
* @dev Sets the WETH address and the LendingPoolAddressesProvider address. Infinite approves lending pool.
* @param _WETH Address of the Wrapped Ether contract
* @param _ADDRESSES_PROVIDER Address of the LendingPoolAddressesProvider contract
**/
constructor(address _WETH, address _ADDRESSES_PROVIDER) public {
WETH = IWETH(_WETH);
ADDRESSES_PROVIDER = ILendingPoolAddressesProvider(_ADDRESSES_PROVIDER);
IWETH(_WETH).approve(
address(ILendingPool(ILendingPoolAddressesProvider(_ADDRESSES_PROVIDER).getLendingPool())),
uint256(-1)
);
}
/**
* @dev deposits WETH into the reserve, using native ETH. A corresponding amount of the overlying asset (aTokens)
* is minted.
* @param onBehalfOf address of the user who will receive the aTokens representing the deposit
* @param referralCode integrators are assigned a referral code and can potentially receive rewards.
**/
function depositETH(address onBehalfOf, uint16 referralCode) external override payable {
ILendingPool pool = ILendingPool(ADDRESSES_PROVIDER.getLendingPool());
require(address(pool) != address(0));
WETH.deposit{value: msg.value}();
pool.deposit(address(WETH), msg.value, onBehalfOf, referralCode);
}
/**
* @dev withdraws the WETH _reserves of msg.sender.
* @param amount address of the user who will receive the aTokens representing the deposit
*/
function withdrawETH(uint256 amount) external override {
ILendingPool pool = ILendingPool(ADDRESSES_PROVIDER.getLendingPool());
require(address(pool) != address(0));
ReserveLogic.ReserveData memory reserve = pool.getReserveData(address(WETH));
uint256 userBalance = IAToken(reserve.aTokenAddress).balanceOf(msg.sender);
uint256 amountToWithdraw = amount;
// if amount is equal to uint(-1), the user wants to redeem everything
if (amount == type(uint256).max) {
amountToWithdraw = userBalance;
}
WETH.transferFrom(msg.sender, address(this), amountToWithdraw);
pool.withdraw(address(WETH), amount);
WETH.withdraw(amountToWithdraw);
safeTransferETH(msg.sender, amountToWithdraw);
}
/**
* @dev repays a borrow on the specific reserve, for the specified amount (or for the whole amount, if uint256(-1) is specified).
* @param amount address of the user who will receive the aTokens representing the deposit
*/
function repayETH(
uint256 amount,
uint256 rateMode,
address onBehalfOf
) external override payable {
ILendingPool pool = ILendingPool(ADDRESSES_PROVIDER.getLendingPool());
require(address(pool) != address(0));
(uint256 stableDebt, uint256 variableDebt) = Helpers.getUserCurrentDebtViaMemory(
onBehalfOf,
pool.getReserveData(address(WETH))
);
uint256 paybackAmount = ReserveLogic.InterestRateMode(rateMode) ==
ReserveLogic.InterestRateMode.STABLE
? stableDebt
: variableDebt;
if (amount < paybackAmount) {
paybackAmount = amount;
}
require(msg.value >= paybackAmount, 'msg.value is less than repayment amount');
WETH.deposit{value: paybackAmount}();
pool.repay(address(WETH), msg.value, rateMode, onBehalfOf);
// refund dust eth
if (msg.value > paybackAmount) safeTransferETH(msg.sender, msg.value - paybackAmount);
}
function safeTransferETH(address to, uint256 value) internal {
(bool success, ) = to.call{value: value}(new bytes(0));
require(success, 'ETH_TRANSFER_FAILED');
}
/**
* @dev Only WETH contract is allowed to transfer ETH here. Prevent other addresses to send Ether to this contract.
*/
receive() external payable {
assert(msg.sender == address(WETH));
}
}