Instadapp/dsa-connectors-old
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors-old.git synced 2024-07-29 22:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed payback issue

Browse Source
This commit is contained in:
Thrilok Kumar 2020-05-23 03:23:25 +05:30
parent 9ec4a5a0bb
commit 0fd86b7e6d
1 changed files with 37 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
contracts/connectors/aave.sol
View File

@ -71,8 +71,8 @@ contract AaveHelpers is DSMath, Stores {
(bal, , , , , , , , , ) = aave.getUserReserveData(token, address(this)); (bal, , , , , , , , , ) = aave.getUserReserveData(token, address(this));
} }
function getPaybackBalance(AaveInterface aave, address token) internal view returns (uint bal) { function getPaybackBalance(AaveInterface aave, address token) internal view returns (uint bal, uint fee) {
(, bal, , , , , , , , ) = aave.getUserReserveData(token, address(this)); (, bal, , , , , fee, , , ) = aave.getUserReserveData(token, address(this));
} }
} }
@ -177,10 +177,44 @@ contract BasicResolver is AaveHelpers {
AaveInterface aave = AaveInterface(getAaveAddress()); AaveInterface aave = AaveInterface(getAaveAddress());
if (_amt == uint(-1)) { if (_amt == uint(-1)) {
uint fee;
(_amt, fee) = getPaybackBalance(aave, token);
_amt = add(_amt, fee);
}
uint ethAmt;
if (token == getEthAddr()) { if (token == getEthAddr()) {
uint ethAmt = getPaybackBalance(aave, token) + 100000000; ethAmt = _amt;
require(address(this).balance >= _amt, "not-enough-eth");
} else {
TokenInterface tokenContract = TokenInterface(token);
require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token");
tokenContract.approve(getAaveCoreAddress(), _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);
}
function payback2(address token, uint amt, uint getId, uint setId) external payable {
uint _amt = getUint(getId, amt);
AaveInterface aave = AaveInterface(getAaveAddress());
if (_amt == uint(-1)) {
if (token == getEthAddr()) {
(uint ethAmt, uint fee) = getPaybackBalance(aave, token);
ethAmt += fee;
uint initalBal = address(this).balance;
require(address(this).balance >= ethAmt, "not-enough-eth"); require(address(this).balance >= ethAmt, "not-enough-eth");
aave.repay.value(ethAmt)(token, ethAmt, payable(address(this))); aave.repay.value(ethAmt)(token, ethAmt, payable(address(this)));
uint finalBal = address(this).balance;
_amt = sub(initalBal, finalBal);
} else { } else {
TokenInterface tokenContract = TokenInterface(token); TokenInterface tokenContract = TokenInterface(token);
require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token"); require(tokenContract.balanceOf(address(this)) >= _amt, "not-enough-token");