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

minor update

Browse Source
This commit is contained in:
Shriya Tyagi 2022-05-02 19:22:20 +04:00
parent 8bc347abb5
commit fb9abb2042
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
contracts/mainnet/connectors/aave/v2/main.sol
View File

@ -151,7 +151,7 @@ abstract contract AaveResolver is Events, Helpers {
TokenInterface tokenContract = TokenInterface(_token); TokenInterface tokenContract = TokenInterface(_token);
if (_amt == uint(-1)) { if (_amt == uint(-1)) {
uint _amtDSA = tokenContract.balanceOf(address(this)); uint _amtDSA = isEth ? address(this).balance : tokenContract.balanceOf(address(this));
uint _amtDebt = getPaybackBalance(_token, rateMode); uint _amtDebt = getPaybackBalance(_token, rateMode);
_amt = _amtDSA <= _amtDebt ? _amtDSA : _amtDebt; _amt = _amtDSA <= _amtDebt ? _amtDSA : _amtDebt;
} }
@ -200,7 +200,7 @@ abstract contract AaveResolver is Events, Helpers {
TokenInterface tokenContract = TokenInterface(_token); TokenInterface tokenContract = TokenInterface(_token);
if (_amt == uint(-1)) { if (_amt == uint(-1)) {
uint _amtDSA = tokenContract.balanceOf(onBehalfOf); uint _amtDSA = isEth ? onBehalfOf.balance : tokenContract.balanceOf(onBehalfOf);
uint _amtDebt = getOnBehalfOfPaybackBalance(_token, rateMode, onBehalfOf); uint _amtDebt = getOnBehalfOfPaybackBalance(_token, rateMode, onBehalfOf);
_amt = _amtDSA <= _amtDebt ? _amtDSA : _amtDebt; _amt = _amtDSA <= _amtDebt ? _amtDSA : _amtDebt;
} }