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

getSave and getLeverage

Browse Source
This commit is contained in:
Samyak Jain 2019-07-14 17:29:31 +05:30
parent c6725a9ae6
commit 1e1b094c6d
1 changed files with 49 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
contracts/ProxyLogics/InstaCompSave.sol
View File

@ -211,6 +211,48 @@ contract CompoundHelper is Helpers {
ratio = wdiv(totalBorrow, totalSupply); ratio = wdiv(totalBorrow, totalSupply);
} }
function getSave(
address user,
uint ethToFree,
address[] memory cTokenAddr,
uint[] memory ctokenFactor
) public returns (uint finalColInEth, uint finalDebtInEth, uint daiDebt, bool isOk)
{
(uint totalSupply, uint totalBorrow,,,uint maxWithdraw,) = getCompStats(user, cTokenAddr, ctokenFactor);
uint ethToSwap = ethToFree < maxWithdraw ? ethToFree : maxWithdraw;
(, uint expectedDAI) = SplitSwapInterface(getAddressSplitSwap()).getBest(getAddressETH(), getAddressDAI(), ethToSwap);
uint daiBorrowed = daiBorrowed(user);
uint daiInEth = CompOracleInterface(getCompOracleAddress()).getUnderlyingPrice(getCDAIAddress());
if (daiBorrowed < expectedDAI) {
finalColInEth = sub(totalSupply, ethToSwap);
finalDebtInEth = sub(totalBorrow, wmul(daiBorrowed, daiInEth));
daiDebt = 0;
isOk = false;
} else {
finalColInEth = sub(totalSupply, ethToSwap);
finalDebtInEth = sub(totalBorrow, wmul(expectedDAI, daiInEth));
daiDebt = sub(daiBorrowed, expectedDAI);
isOk = true;
}
}
function getLeverage(
address user,
uint daiToBorrow,
address[] memory cTokenAddr,
uint[] memory ctokenFactor
) public returns (uint finalColInEth, uint finalDebtInEth, uint ethCol)
{
(uint totalSupply, uint totalBorrow,, uint borrowRemain,,) = getCompStats(user, cTokenAddr, ctokenFactor);
uint daiToSwap = getDaiRemainBorrow(borrowRemain);
daiToSwap = daiToSwap < daiToBorrow ? daiToSwap : daiToBorrow;
(, uint expectedETH) = SplitSwapInterface(getAddressSplitSwap()).getBest(getAddressDAI(), getAddressETH(), daiToSwap);
uint daiInEth = CompOracleInterface(getCompOracleAddress()).getUnderlyingPrice(getCDAIAddress());
finalColInEth = add(totalSupply, expectedETH);
finalDebtInEth = add(totalBorrow, wmul(daiToSwap, daiInEth));
ethCol = add(getEthSupply(user), expectedETH);
}
function getEthSupply(address user) internal returns (uint ethSupply) { function getEthSupply(address user) internal returns (uint ethSupply) {
CTokenInterface cTokenContract = CTokenInterface(getCETHAddress()); CTokenInterface cTokenContract = CTokenInterface(getCETHAddress());
uint cTokenBal = sub(cTokenContract.balanceOf(user), 1); uint cTokenBal = sub(cTokenContract.balanceOf(user), 1);
@ -218,6 +260,11 @@ contract CompoundHelper is Helpers {
ethSupply = wmul(cTokenBal, cTokenExchangeRate); ethSupply = wmul(cTokenBal, cTokenExchangeRate);
} }
function daiBorrowed(address user) internal returns (uint daiAmt) {
CTokenInterface cTokenContract = CTokenInterface(getCDAIAddress());
daiAmt = cTokenContract.borrowBalanceCurrent(user);
}
function getDaiRemainBorrow(uint daiInEth) internal view returns (uint daiAmt) { function getDaiRemainBorrow(uint daiInEth) internal view returns (uint daiAmt) {
uint tokenPriceInEth = CompOracleInterface(getCompOracleAddress()).getUnderlyingPrice(getCDAIAddress()); uint tokenPriceInEth = CompOracleInterface(getCompOracleAddress()).getUnderlyingPrice(getCDAIAddress());
daiAmt = sub(wdiv(daiInEth, tokenPriceInEth), 10); daiAmt = sub(wdiv(daiInEth, tokenPriceInEth), 10);
@ -304,13 +351,13 @@ contract CompoundSave is CompoundResolver {
function save( function save(
uint ethToFree, uint ethToFree,
address[] memory cTokenAddr, address[] memory ctokenAddr,
uint[] memory ctokenFactor, uint[] memory ctokenFactor,
uint splitAmt, uint splitAmt,
uint slippageAmt uint slippageAmt
) public ) public
{ {
(,,,,uint maxWithdraw,) = getCompStats(address(this), cTokenAddr, ctokenFactor); (,,,,uint maxWithdraw,) = getCompStats(address(this), ctokenAddr, ctokenFactor);
uint ethToSwap = ethToFree < maxWithdraw ? ethToFree : maxWithdraw; uint ethToSwap = ethToFree < maxWithdraw ? ethToFree : maxWithdraw;
redeemEth(ethToSwap); redeemEth(ethToSwap);
uint destAmt = SplitSwapInterface(getAddressSplitSwap()).ethToDaiSwap.value(ethToSwap)(splitAmt, slippageAmt); uint destAmt = SplitSwapInterface(getAddressSplitSwap()).ethToDaiSwap.value(ethToSwap)(splitAmt, slippageAmt);