Almost done with aave connector

This commit is contained in:
Thrilok Kumar 2020-05-21 00:17:14 +05:30
parent 7da98bfb6d
commit 3c63d3369b

View File

@ -0,0 +1,200 @@
pragma solidity ^0.6.0;
// import files from common directory
import { TokenInterface , MemoryInterface, EventInterface} from "../common/interfaces.sol";
import { Stores } from "../common/stores.sol";
import { DSMath } from "../common/math.sol";
interface AaveInterface {
function deposit(address _reserve, uint256 _amount, uint16 _referralCode) external payable;
function redeemUnderlying(
address _reserve,
address payable _user,
uint256 _amount,
uint256 _aTokenBalanceAfterRedeem
) external;
function setUserUseReserveAsCollateral(address _reserve, bool _useAsCollateral) external;
function getUserReserveData(address _reserve, address _user) external view returns (
uint256 currentATokenBalance,
uint256 currentBorrowBalance,
uint256 principalBorrowBalance,
uint256 borrowRateMode,
uint256 borrowRate,
uint256 liquidityRate,
uint256 originationFee,
uint256 variableBorrowIndex,
uint256 lastUpdateTimestamp,
bool usageAsCollateralEnabled
);
function borrow(address _reserve, uint256 _amount, uint256 _interestRateMode, uint16 _referralCode) external;
function repay(address _reserve, uint256 _amount, address payable _onBehalfOf) external payable;
}
interface AaveProviderInterface {
function getLendingPool() external view returns (address);
function getLendingPoolCore() external view returns (address);
}
interface AaveCoreInterface {
function getReserveATokenAddress(address _reserve) external view returns (address);
}
interface ATokenInterface {
function redeem(uint256 _amount) external;
function balanceOf(address _user) external view returns(uint256);
function principalBalanceOf(address _user) external view returns(uint256);
}
contract AaveHelpers is DSMath, Stores {
/**
* @dev get Aave Address
*/
function getAaveProviderAddress() internal pure returns (address) {
// return 0x24a42fD28C976A61Df5D00D0599C34c4f90748c8; //mainnet
// return 0x506B0B2CF20FAA8f38a4E2B524EE43e1f4458Cc5; //kovan
return 0x1c8756FD2B28e9426CDBDcC7E3c4d64fa9A54728; //ropsten
}
function getWithdrawBalance(address token) internal view returns (uint bal) {
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveInterface aave = AaveInterface(lendingProviderPool.getLendingPool());
(bal, , , , , , , , , ) = aave.getUserReserveData(token, msg.sender);
}
function getPaybackBalance(address token) internal view returns (uint bal) {
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveInterface aave = AaveInterface(lendingProviderPool.getLendingPool());
(, bal, , , , , , , , ) = aave.getUserReserveData(token, msg.sender);
}
}
contract BasicResolver is AaveHelpers {
event LogDeposit(address indexed token, uint256 tokenAmt, uint256 getId, uint256 setId);
event LogWithdraw(address indexed token, uint256 tokenAmt, uint256 getId, uint256 setId);
event LogBorrow(address indexed token, uint256 tokenAmt, uint256 getId, uint256 setId);
event LogPayback(address indexed token, uint256 tokenAmt, uint256 getId, uint256 setId);
/**
* @dev Deposit ETH/ERC20_Token.
* @param token token address to deposit.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
* @param amt token amount to deposit.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function deposit(address token, uint amt, uint getId, uint setId) external payable{
uint _amt = getUint(getId, amt);
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveInterface aave = AaveInterface(lendingProviderPool.getLendingPool());
uint ethAmt;
if (token == getEthAddr()) {
require(_amt == msg.value, "not-enought-eth");
ethAmt = _amt;
} else {
TokenInterface tokenContract = TokenInterface(token);
require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token");
tokenContract.approve(lendingProviderPool.getLendingPoolCore(), _amt);
}
aave.deposit.value(ethAmt)(token, amt, 0); // TODO - need to set referralCode;
aave.setUserUseReserveAsCollateral(token, true);
setUint(setId, _amt);
emit LogDeposit(token, _amt, getId, setId);
bytes32 _eventCode = keccak256("LogDeposit(address,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, getId, setId);
(uint _type, uint _id) = connectorID();
EventInterface(getEventAddr()).emitEvent(_type, _id, _eventCode, _eventParam);
}
/**
* @dev Withdraw ETH/ERC20_Token.
* @param token token address to withdraw.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
* @param amt token amount to withdraw.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function withdraw(address token, uint amt, uint getId, uint setId) external payable{
uint _amt = getUint(getId, amt);
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveCoreInterface aaveCore = AaveCoreInterface(lendingProviderPool.getLendingPoolCore());
ATokenInterface atoken = ATokenInterface(aaveCore.getReserveATokenAddress(token));
uint totalBal = atoken.balanceOf(address(this));
_amt = _amt >= totalBal ? totalBal : _amt;
uint initialBal = token == getEthAddr() ? address(this).balance : TokenInterface(token).balanceOf(address(this));
atoken.redeem(_amt);
uint finialBal = token == getEthAddr() ? address(this).balance : TokenInterface(token).balanceOf(address(this));
uint withdrawnAmt = sub(finialBal, initialBal);
require(withdrawnAmt >= _amt, "withdraw-error");
setUint(setId, _amt);
emit LogWithdraw(token, _amt, getId, setId);
bytes32 _eventCode = keccak256("LogWithdraw(address,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, getId, setId);
(uint _type, uint _id) = connectorID();
EventInterface(getEventAddr()).emitEvent(_type, _id, _eventCode, _eventParam);
}
/**
* @dev Borrow ETH/ERC20_Token.
* @param token token address to borrow.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
* @param amt token amount to borrow.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function borrow(address token, uint amt, uint getId, uint setId) external payable {
uint _amt = getUint(getId, amt);
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveInterface aave = AaveInterface(lendingProviderPool.getLendingPool());
aave.borrow(token, amt, 2, 0); // TODO - need to set referralCode;
setUint(setId, _amt);
emit LogBorrow(token, _amt, getId, setId);
bytes32 _eventCode = keccak256("LogBorrow(address,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, getId, setId);
(uint _type, uint _id) = connectorID();
EventInterface(getEventAddr()).emitEvent(_type, _id, _eventCode, _eventParam);
}
/**
* @dev Payback borrowed ETH/ERC20_Token.
* @param token token address to payback.(For ETH: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
* @param amt token amount to payback.
* @param getId Get token amount at this ID from `InstaMemory` Contract.
* @param setId Set token amount at this ID in `InstaMemory` Contract.
*/
function payback(address token, uint amt, uint getId, uint setId) external payable {
uint _amt = getUint(getId, amt);
AaveProviderInterface lendingProviderPool = AaveProviderInterface(getAaveProviderAddress());
AaveInterface aave = AaveInterface(lendingProviderPool.getLendingPool());
_amt = _amt == uint(-1) ? getPaybackBalance(token) : _amt;
uint ethAmt;
if (token == getEthAddr()) {
require(address(this).balance >= _amt, "not-enough-eth");
ethAmt = _amt;
} else {
TokenInterface tokenContract = TokenInterface(token);
require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token");
tokenContract.approve(lendingProviderPool.getLendingPoolCore(), _amt);
}
aave.repay.value(ethAmt)(token, amt, payable(address(this)));
setUint(setId, _amt);
emit LogPayback(token, _amt, getId, setId);
bytes32 _eventCode = keccak256("LogPayback(address,uint256,uint256,uint256)");
bytes memory _eventParam = abi.encode(token, _amt, getId, setId);
(uint _type, uint _id) = connectorID();
EventInterface(getEventAddr()).emitEvent(_type, _id, _eventCode, _eventParam);
}
}
contract ConnectAave is BasicResolver {
string public name = "Aave-v1";
}