mirror of
https://github.com/Instadapp/smart-contract.git
synced 2024-07-29 22:08:07 +00:00
InstaLiquidity done
This commit is contained in:
Samyak Jain
parent
f0dfd5dc5d
commit
49ef605c23
195
contracts/ProxyLogics/InstaLiquidity.sol
Normal file
195
contracts/ProxyLogics/InstaLiquidity.sol
Normal file
|
|
@ -0,0 +1,195 @@
|
|||
pragma solidity ^0.5.7;
|
||||
|
||||
interface ERC20Interface {
|
||||
function allowance(address, address) external view returns (uint);
|
||||
function balanceOf(address) external view returns (uint);
|
||||
function approve(address, uint) external;
|
||||
function transfer(address, uint) external returns (bool);
|
||||
function transferFrom(address, address, uint) external returns (bool);
|
||||
function deposit() external payable;
|
||||
function withdraw(uint) external;
|
||||
}
|
||||
|
||||
interface CTokenInterface {
|
||||
function mint(uint mintAmount) external returns (uint); // For ERC20
|
||||
function redeem(uint redeemTokens) external returns (uint);
|
||||
function redeemUnderlying(uint redeemAmount) external returns (uint);
|
||||
function borrow(uint borrowAmount) external returns (uint);
|
||||
function exchangeRateCurrent() external returns (uint);
|
||||
function transfer(address, uint) external returns (bool);
|
||||
function transferFrom(address, address, uint) external returns (bool);
|
||||
function balanceOf(address) external view returns (uint);
|
||||
function repayBorrow(uint repayAmount) external returns (uint); // For ERC20
|
||||
function borrowBalanceCurrent(address account) external returns (uint);
|
||||
function underlying() external view returns (address);
|
||||
}
|
||||
|
||||
interface CETHInterface {
|
||||
function exchangeRateCurrent() external returns (uint);
|
||||
function mint() external payable; // For ETH
|
||||
function repayBorrow() external payable; // For ETH
|
||||
function transfer(address, uint) external returns (bool);
|
||||
function borrowBalanceCurrent(address account) external returns (uint);
|
||||
function balanceOf(address) external view returns (uint);
|
||||
}
|
||||
|
||||
interface ComptrollerInterface {
|
||||
function enterMarkets(address[] calldata cTokens) external returns (uint[] memory);
|
||||
function exitMarket(address cTokenAddress) external returns (uint);
|
||||
function getAssetsIn(address account) external view returns (address[] memory);
|
||||
function getAccountLiquidity(address account) external view returns (uint, uint, uint);
|
||||
}
|
||||
|
||||
interface LiquidityInterface {
|
||||
function deposits(address userWallet, address ctkn) external view returns(uint ctknAmt);
|
||||
function depositCTkn(address ctknAddr, uint amt) external;
|
||||
function withdrawCTkn(address ctknAddr, uint amt) external returns(uint ctknAmt);
|
||||
}
|
||||
|
||||
|
||||
contract DSMath {
|
||||
|
||||
function add(uint x, uint y) internal pure returns (uint z) {
|
||||
require((z = x + y) >= x, "math-not-safe");
|
||||
}
|
||||
|
||||
function mul(uint x, uint y) internal pure returns (uint z) {
|
||||
require(y == 0 || (z = x * y) / y == x, "math-not-safe");
|
||||
}
|
||||
|
||||
uint constant WAD = 10 ** 18;
|
||||
|
||||
function wmul(uint x, uint y) internal pure returns (uint z) {
|
||||
z = add(mul(x, y), WAD / 2) / WAD;
|
||||
}
|
||||
|
||||
function wdiv(uint x, uint y) internal pure returns (uint z) {
|
||||
z = add(mul(x, WAD), y / 2) / y;
|
||||
}
|
||||
|
||||
function sub(uint a, uint b) internal pure returns (uint c) {
|
||||
require(b <= a, "SafeMath: subtraction overflow");
|
||||
c = a - b;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract Helper is DSMath {
|
||||
|
||||
/**
|
||||
* @dev get ethereum address for trade
|
||||
*/
|
||||
function getEthAddr() public pure returns (address eth) {
|
||||
eth = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
|
||||
}
|
||||
|
||||
function getCEthAddr() public pure returns(address ceth) {
|
||||
ceth = 0x4Ddc2D193948926D02f9B1fE9e1daa0718270ED5;
|
||||
}
|
||||
|
||||
function getComptrollerAddr() public pure returns (address troller) {
|
||||
troller = 0x3d9819210A31b4961b30EF54bE2aeD79B9c9Cd3B;
|
||||
}
|
||||
|
||||
function getLiquidityAddr() public pure returns (address liquidity) {
|
||||
// liquidity = ;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev setting allowance to compound for the "user proxy" if required
|
||||
*/
|
||||
function setApproval(address erc20, uint srcAmt, address to) internal {
|
||||
ERC20Interface erc20Contract = ERC20Interface(erc20);
|
||||
uint tokenAllowance = erc20Contract.allowance(address(this), to);
|
||||
if (srcAmt > tokenAllowance) {
|
||||
erc20Contract.approve(to, srcAmt);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
contract ProvideLiquidity is Helper {
|
||||
|
||||
event LogDepositToken(address tknAddr, uint amt);
|
||||
event LogWithdrawToken(address tknAddr, uint amt);
|
||||
event LogDepositCToken(address ctknAddr, uint amt);
|
||||
event LogWithdrawCToken(address ctknAddr, uint amt);
|
||||
|
||||
/**
|
||||
* @dev Deposit Token to liquidity.
|
||||
*/
|
||||
function depositToken(address ctknAddr, uint amt) public payable {
|
||||
if (ctknAddr != getCEthAddr()) {
|
||||
CTokenInterface cTokenContract = CTokenInterface(ctknAddr);
|
||||
address tknAddr = cTokenContract.underlying();
|
||||
require(ERC20Interface(tknAddr).transferFrom(msg.sender, address(this), amt), "Not enough tkn to deposit");
|
||||
setApproval(tknAddr, amt, ctknAddr);
|
||||
assert(cTokenContract.mint(amt) == 0);
|
||||
uint exchangeRate = cTokenContract.exchangeRateCurrent();
|
||||
uint cTknAmt = wdiv(amt, exchangeRate);
|
||||
uint cTknBal = cTokenContract.balanceOf(address(this));
|
||||
cTknAmt = cTknAmt <= cTknBal ? cTknAmt : cTknBal;
|
||||
setApproval(ctknAddr, cTknAmt, getLiquidityAddr());
|
||||
LiquidityInterface(getLiquidityAddr()).depositCTkn(ctknAddr, cTknAmt);
|
||||
emit LogDepositToken(tknAddr, amt);
|
||||
} else {
|
||||
CETHInterface cEthContract = CETHInterface(ctknAddr);
|
||||
cEthContract.mint.value(msg.value)();
|
||||
uint exchangeRate = cEthContract.exchangeRateCurrent();
|
||||
uint cEthAmt = wdiv(msg.value, exchangeRate);
|
||||
uint cEthBal = cEthContract.balanceOf(address(this));
|
||||
cEthAmt = cEthAmt <= cEthBal ? cEthAmt : cEthBal;
|
||||
setApproval(ctknAddr, cEthAmt, getLiquidityAddr());
|
||||
LiquidityInterface(getLiquidityAddr()).depositCTkn(ctknAddr, cEthAmt);
|
||||
emit LogDepositToken(getEthAddr(), amt);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Withdraw Token from liquidity.
|
||||
*/
|
||||
function withdrawToken(address ctknAddr, uint amt) public {
|
||||
CTokenInterface cTokenContract = CTokenInterface(ctknAddr);
|
||||
uint exchangeRate = cTokenContract.exchangeRateCurrent();
|
||||
uint withdrawAmt = wdiv(amt, exchangeRate); // withdraw CToken Amount
|
||||
withdrawAmt = LiquidityInterface(getLiquidityAddr()).withdrawCTkn(ctknAddr, withdrawAmt);
|
||||
if (ctknAddr != getCEthAddr()) {
|
||||
require(cTokenContract.redeem(withdrawAmt) == 0, "something went wrong");
|
||||
uint tknAmt = wmul(withdrawAmt, exchangeRate);
|
||||
address tknAddr = cTokenContract.underlying();
|
||||
uint tknBal = ERC20Interface(tknAddr).balanceOf(address(this));
|
||||
tknAmt = tknAmt <= tknBal ? tknAmt : tknBal;
|
||||
require(ERC20Interface(tknAddr).transfer(msg.sender, tknAmt), "not enough tkn to Transfer");
|
||||
emit LogWithdrawToken(tknAddr, tknAmt);
|
||||
} else {
|
||||
require(cTokenContract.redeem(withdrawAmt) == 0, "something went wrong");
|
||||
uint ethAmt = wmul(withdrawAmt, exchangeRate);
|
||||
uint ethBal = address(this).balance;
|
||||
ethAmt = ethAmt <= ethBal ? ethAmt : ethBal;
|
||||
msg.sender.transfer(ethAmt);
|
||||
emit LogWithdrawToken(getEthAddr(), ethAmt);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Deposit CToken in liquidity
|
||||
*/
|
||||
function depositCTkn(address ctknAddr, uint amt) public {
|
||||
require(CTokenInterface(ctknAddr).transferFrom(msg.sender, address(this), amt), "Nothing to deposit");
|
||||
setApproval(ctknAddr, amt, getLiquidityAddr());
|
||||
LiquidityInterface(getLiquidityAddr()).depositCTkn(ctknAddr, amt);
|
||||
emit LogDepositCToken(ctknAddr, amt);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Withdraw CToken from liquidity
|
||||
*/
|
||||
function withdrawCTkn(address ctknAddr, uint amt) public {
|
||||
uint withdrawAmt = LiquidityInterface(getLiquidityAddr()).withdrawCTkn(ctknAddr, amt);
|
||||
assert(CTokenInterface(ctknAddr).transfer(msg.sender, withdrawAmt));
|
||||
emit LogWithdrawCToken(ctknAddr, withdrawAmt);
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user