Code refactoring

This commit is contained in:
Thrilok Kumar 2020-11-17 21:38:09 +05:30
parent 5b93c32856
commit ad141f0e4a

View File

@ -109,13 +109,8 @@ contract AaveHelpers is DSMath, Stores {
} }
function getPaybackBalance(AaveDataProviderInterface aaveData, address token, uint rateMode) internal view returns (uint) { function getPaybackBalance(AaveDataProviderInterface aaveData, address token, uint rateMode) internal view returns (uint) {
uint stableDebt; (, uint stableDebt, uint variableDebt, , , , , , ) = aaveData.getUserReserveData(token, address(this));
uint variableDebt; return rateMode == 1 ? stableDebt : variableDebt;
(, stableDebt, variableDebt, , , , , , ) = aaveData.getUserReserveData(token, address(this));
if (rateMode == 1) {
return stableDebt;
}
return variableDebt;
} }
} }
@ -213,14 +208,9 @@ contract BasicResolver is AaveHelpers {
TokenInterface tokenContract = TokenInterface(_token); TokenInterface tokenContract = TokenInterface(_token);
uint debtPayback = getPaybackBalance(aaveData, _token, rateMode); _amt = _amt == uint(-1) ? getPaybackBalance(aaveData, _token, rateMode) : _amt;
if (isEth) { if (isEth) convertEthToWeth(isEth, tokenContract, _amt);
_amt = _amt == uint(-1) ? debtPayback : _amt;
convertEthToWeth(isEth, tokenContract, _amt);
} else {
_amt = _amt == uint(-1) ? debtPayback : _amt;
}
tokenContract.approve(address(aave), _amt); tokenContract.approve(address(aave), _amt);