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

setAmt

Browse Source
This commit is contained in:
Richa-iitr 2022-09-02 19:43:42 +05:30
parent 826051ac50
commit 28ef1ae2bb
1 changed files with 18 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
contracts/mainnet/connectors/compound/v3/helpers.sol
View File

@ -181,51 +181,26 @@ abstract contract Helpers is DSMath, Basic {
bool isEth,
ACTION action
) internal returns (uint256) {
if (isEth) {
if (amt == uint256(-1)) {
uint256 allowance_ = TokenInterface(token).allowance(
src,
market
);
uint256 bal_;
if (action == ACTION.repay) {
bal_ = CometInterface(market).borrowBalanceOf(src);
} else if (action == ACTION.deposit) {
bal_ = src.balance;
} else if (action == ACTION.transfer) {
bal_ = (token == getBaseToken(market))
? TokenInterface(market).balanceOf(src)
: CometInterface(market)
.userCollateral(src, token)
.balance;
}
if (action == ACTION.transfer) amt = bal_;
else amt = bal_ < allowance_ ? bal_ : allowance_;
}
if (src == address(this))
convertEthToWeth(isEth, TokenInterface(token), amt);
} else {
if (amt == uint256(-1)) {
uint256 allowance_ = TokenInterface(token).allowance(
src,
market
);
uint256 bal_;
if (action == ACTION.repay) {
bal_ = CometInterface(market).borrowBalanceOf(src);
} else if (action == ACTION.deposit) {
bal_ = TokenInterface(token).balanceOf(src);
} else if (action == ACTION.transfer) {
bal_ = (token == getBaseToken(market))
? TokenInterface(market).balanceOf(src)
: CometInterface(market)
.userCollateral(src, token)
.balance;
}
if (action == ACTION.transfer) amt = bal_;
else amt = bal_ < allowance_ ? bal_ : allowance_;
if (amt == uint256(-1)) {
uint256 allowance_ = TokenInterface(token).allowance(src, market);
uint256 bal_;
if (action == ACTION.repay) {
bal_ = CometInterface(market).borrowBalanceOf(src);
} else if (action == ACTION.deposit) {
if (isEth) bal_ = src.balance;
else bal_ = TokenInterface(token).balanceOf(src);
} else if (action == ACTION.transfer) {
bal_ = (token == getBaseToken(market))
? TokenInterface(market).balanceOf(src)
: CometInterface(market).userCollateral(src, token).balance;
}
if (action == ACTION.transfer) amt = bal_;
else amt = bal_ < allowance_ ? bal_ : allowance_;
}
if (src == address(this))
convertEthToWeth(isEth, TokenInterface(token), amt);
return amt;
}
}