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

Payback max

Browse Source
This commit is contained in:
Mubaris NK 2020-11-17 10:50:39 +05:30
parent b4ee90707d
commit 5b93c32856
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
contracts/connectors/aave_v2.sol
View File

@ -107,6 +107,16 @@ contract AaveHelpers is DSMath, Stores {
token.withdraw(amount); token.withdraw(amount);
} }
} }
function getPaybackBalance(AaveDataProviderInterface aaveData, address token, uint rateMode) internal view returns (uint) {
uint stableDebt;
uint variableDebt;
(, stableDebt, variableDebt, , , , , , ) = aaveData.getUserReserveData(token, address(this));
if (rateMode == 1) {
return stableDebt;
}
return variableDebt;
}
} }
contract BasicResolver is AaveHelpers { contract BasicResolver is AaveHelpers {
@ -196,17 +206,20 @@ contract BasicResolver is AaveHelpers {
uint _amt = getUint(getId, amt); uint _amt = getUint(getId, amt);
AaveInterface aave = AaveInterface(getAaveProvider().getLendingPool()); AaveInterface aave = AaveInterface(getAaveProvider().getLendingPool());
AaveDataProviderInterface aaveData = getAaveDataProvider();
bool isEth = token == getEthAddr(); bool isEth = token == getEthAddr();
address _token = isEth ? getWethAddr() : token; address _token = isEth ? getWethAddr() : token;
TokenInterface tokenContract = TokenInterface(_token); TokenInterface tokenContract = TokenInterface(_token);
uint debtPayback = getPaybackBalance(aaveData, _token, rateMode);
if (isEth) { if (isEth) {
_amt = _amt == uint(-1) ? address(this).balance : _amt; _amt = _amt == uint(-1) ? debtPayback : _amt;
convertEthToWeth(isEth, tokenContract, _amt); convertEthToWeth(isEth, tokenContract, _amt);
} else { } else {
_amt = _amt == uint(-1) ? tokenContract.balanceOf(address(this)) : _amt; _amt = _amt == uint(-1) ? debtPayback : _amt;
} }
tokenContract.approve(address(aave), _amt); tokenContract.approve(address(aave), _amt);