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

[Compound] Max repay and withdraw

Browse Source
This commit is contained in:
Mubaris NK 2021-01-04 22:24:08 +05:30
parent ff26b8e135
commit 37996015ce
No known key found for this signature in database
GPG Key ID: 1F132294E8700320
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
contracts/connectors/refinance.sol
View File

@ -382,6 +382,10 @@ contract CompoundHelpers is Helpers {
if (amts[i] > 0) {
address cToken = InstaMapping(getMappingAddr()).cTokenMapping(tokens[i]);
CTokenInterface cTokenContract = CTokenInterface(cToken);
uint _amt = amts[i];
if (_amt == uint(-1)) {
_amt = cTokenContract.balanceOf(address(this));
}
require(cTokenContract.redeemUnderlying(amts[i]) == 0, "withdraw-failed");
}
}
@ -395,12 +399,18 @@ contract CompoundHelpers is Helpers {
for (uint i = 0; i < length; i++) {
if (amts[i] > 0) {
address cToken = InstaMapping(getMappingAddr()).cTokenMapping(tokens[i]);
CTokenInterface cTokenContract = CTokenInterface(cToken);
uint _amt = amts[i];
if (_amt == uint(-1)) {
_amt = cTokenContract.borrowBalanceCurrent(address(this));
}
if (tokens[i] != getEthAddr()) {
TokenInterface tokenContract = TokenInterface(tokens[i]);
tokenContract.approve(cToken, amts[i]);
require(cTokenContract.repayBorrow(amts[i]) == 0, "repay-failed.");
tokenContract.approve(cToken, _amt);
require(cTokenContract.repayBorrow(_amt) == 0, "repay-failed.");
} else {
CETHInterface(cToken).repayBorrow.value(amts[i])();
CETHInterface(cToken).repayBorrow.value(_amt)();
}
}
}