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

Added missing things from eth pool

Browse Source
This commit is contained in:
Thrilok Kumar 2020-09-07 02:41:49 +05:30
parent d0da53a445
commit 4efb6a2fc7
1 changed files with 49 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
contracts/pools/erc20.sol
View File

@ -16,6 +16,7 @@ interface IndexInterface {
interface RegistryInterface { interface RegistryInterface {
function chief(address) external view returns (bool); function chief(address) external view returns (bool);
function poolLogic(address) external returns (address); function poolLogic(address) external returns (address);
function flusherLogic(address) external returns (address);
function fee(address) external view returns (uint); function fee(address) external view returns (uint);
function poolCap(address) external view returns (uint); function poolCap(address) external view returns (uint);
function checkSettleLogics(address, address[] calldata) external view returns (bool); function checkSettleLogics(address, address[] calldata) external view returns (bool);
@ -25,15 +26,17 @@ interface RateInterface {
function getTotalToken() external returns (uint totalUnderlyingTkn); function getTotalToken() external returns (uint totalUnderlyingTkn);
} }
contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable { interface FlusherLogicInterface {
function isFlusher(address) external returns (bool);
}
contract PoolToken is ReentrancyGuard, ERC20Pausable, DSMath {
using SafeERC20 for IERC20; using SafeERC20 for IERC20;
event LogDeploy(address indexed dsa, address token, uint amount);
event LogExchangeRate(uint exchangeRate, uint tokenBalance, uint insuranceAmt); event LogExchangeRate(uint exchangeRate, uint tokenBalance, uint insuranceAmt);
event LogSettle(uint settleBlock); event LogSettle(uint settleBlock);
event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt); event LogDeposit(address indexed user, uint depositAmt, uint poolMintAmt);
event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt); event LogWithdraw(address indexed user, uint withdrawAmt, uint poolBurnAmt);
event LogAddFee(uint amount);
event LogWithdrawFee(uint amount); event LogWithdrawFee(uint amount);
event LogPausePool(bool); event LogPausePool(bool);
@ -45,25 +48,30 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
uint public feeAmt; // fee collected on profits uint public feeAmt; // fee collected on profits
constructor( constructor(
address _registry, address _registry,
string memory _name, string memory _name,
string memory _symbol, string memory _symbol,
address _baseToken address _baseToken
) public ERC20(_name, _symbol) { ) public ERC20(_name, _symbol) {
baseToken = IERC20(_baseToken); baseToken = IERC20(_baseToken);
registry = RegistryInterface(_registry); registry = RegistryInterface(_registry);
exchangeRate = 10 ** uint(36 - ERC20(_baseToken).decimals()); exchangeRate = 10 ** uint(36 - ERC20(_baseToken).decimals());
} }
modifier isChief() { modifier isChief() {
require(registry.chief(msg.sender) || msg.sender == instaIndex.master(), "not-chief"); require(registry.chief(msg.sender) || msg.sender == instaIndex.master(), "not-chief");
_; _;
}
modifier isFlusher() {
require(FlusherLogicInterface(registry.flusherLogic(address(this))).isFlusher(msg.sender), "not-flusher");
_;
} }
/** /**
* @dev get pool token rate * @dev get pool token rate
* @param tokenAmt total token amount * @param tokenAmt total token amount
*/ */
function getCurrentRate(uint tokenAmt) internal view returns (uint) { function getCurrentRate(uint tokenAmt) internal view returns (uint) {
return wdiv(totalSupply(), tokenAmt); return wdiv(totalSupply(), tokenAmt);
} }
@ -73,26 +81,27 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
*/ */
function setExchangeRate() public { function setExchangeRate() public {
require(msg.sender == address(this), "not-pool-address"); require(msg.sender == address(this), "not-pool-address");
uint _previousRate = exchangeRate; uint _prevRate = exchangeRate;
uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken(); uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken();
_totalToken = sub(_totalToken, feeAmt); _totalToken = sub(_totalToken, feeAmt);
uint _currentRate = getCurrentRate(_totalToken); uint _newRate = getCurrentRate(_totalToken);
uint _tokenBal; uint _tokenBal;
require(_currentRate != 0, "current-rate-is-zero"); require(_newRate != 0, "current-rate-is-zero");
if (_currentRate > _previousRate) { // loss => deduct partially/fully from insurance amount if (_newRate > _prevRate) {
_currentRate = _previousRate; _newRate = _prevRate;
} else { // profit => add to insurance amount } else {
_tokenBal = wdiv(totalSupply(), _prevRate);
uint _newFee = wmul(sub(_totalToken, _tokenBal), registry.fee(address(this))); uint _newFee = wmul(sub(_totalToken, _tokenBal), registry.fee(address(this)));
feeAmt = add(feeAmt, _newFee); feeAmt = add(feeAmt, _newFee);
_tokenBal = sub(_totalToken, _newFee); _tokenBal = sub(_totalToken, _newFee);
_currentRate = getCurrentRate(_tokenBal); _newRate = getCurrentRate(_tokenBal);
} }
exchangeRate = _currentRate; exchangeRate = _newRate;
emit LogExchangeRate(exchangeRate, _tokenBal, feeAmt); emit LogExchangeRate(exchangeRate, _tokenBal, feeAmt);
} }
/** /**
* @dev Delegate the calls to Connector And this function is ran by cast(). * @dev delegate the calls to connector and this function is ran by settle()
* @param _target Target to of Connector. * @param _target Target to of Connector.
* @param _data CallData of function in Connector. * @param _data CallData of function in Connector.
*/ */
@ -128,48 +137,48 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
/** /**
* @dev Deposit token. * @dev Deposit token.
* @param tknAmt token amount * @param tknAmt token amount
* @return _mintAmt amount of wrap token minted * @return mintAmt amount of wrap token minted
*/ */
function deposit(uint tknAmt) external whenNotPaused payable returns (uint _mintAmt) { function deposit(uint tknAmt) public whenNotPaused payable isFlusher returns (uint mintAmt) {
require(msg.value == 0, "non-eth-pool"); require(msg.value == 0, "non-eth-pool");
uint _tokenBal = wdiv(totalSupply(), exchangeRate); uint _tokenBal = wdiv(totalSupply(), exchangeRate);
uint _newTknBal = add(_tokenBal, tknAmt); uint _newTknBal = add(_tokenBal, tknAmt);
require(_newTknBal < registry.poolCap(address(this)), "unmatched-amount"); require(_newTknBal < registry.poolCap(address(this)), "unmatched-amount");
baseToken.safeTransferFrom(msg.sender, address(this), tknAmt); baseToken.safeTransferFrom(msg.sender, address(this), tknAmt);
_mintAmt = wmul(tknAmt, exchangeRate); mintAmt = wmul(tknAmt, exchangeRate);
_mint(msg.sender, _mintAmt); _mint(msg.sender, mintAmt);
emit LogDeposit(msg.sender, tknAmt, _mintAmt); emit LogDeposit(msg.sender, tknAmt, mintAmt);
} }
/** /**
* @dev Withdraw tokens. * @dev Withdraw tokens.
* @param tknAmt token amount * @param tknAmt token amount
* @param to withdraw tokens to address * @param target withdraw tokens to address
* @return _tknAmt amount of token withdrawn * @return wdAmt amount of token withdrawn
*/ */
function withdraw(uint tknAmt, address to) external nonReentrant whenNotPaused returns (uint _tknAmt) { function withdraw(uint tknAmt, address target) external nonReentrant whenNotPaused returns (uint wdAmt) {
require(to != address(0), "to-address-not-vaild"); require(target != address(0), "invalid-target-address");
uint _userBal = wdiv(balanceOf(msg.sender), exchangeRate); uint _userBal = wdiv(balanceOf(msg.sender), exchangeRate);
uint _burnAmt; uint _burnAmt;
if (tknAmt >= _userBal) { if (tknAmt >= _userBal) {
_burnAmt = balanceOf(msg.sender); _burnAmt = balanceOf(msg.sender);
_tknAmt = _userBal; wdAmt = _userBal;
} else { } else {
_burnAmt = wmul(tknAmt, exchangeRate); _burnAmt = wmul(tknAmt, exchangeRate);
_tknAmt = tknAmt; wdAmt = tknAmt;
} }
require(_tknAmt <= baseToken.balanceOf(address(this)), "not-enough-liquidity-available"); require(wdAmt <= baseToken.balanceOf(address(this)), "not-enough-liquidity-available");
_burn(msg.sender, _burnAmt); _burn(msg.sender, _burnAmt);
baseToken.safeTransfer(to, _tknAmt); baseToken.safeTransfer(target, wdAmt);
emit LogWithdraw(msg.sender, _tknAmt, _burnAmt); emit LogWithdraw(msg.sender, wdAmt, _burnAmt);
} }
/** /**
* @dev Withdraw Insurance from the pool. * @dev withdraw fee from the pool
* @notice only master can call this function. * @notice only master can call this function
* @param wdAmt insurance token amount to remove * @param wdAmt fee amount to withdraw
*/ */
function withdrawFee(uint wdAmt) external { function withdrawFee(uint wdAmt) external {
require(msg.sender == instaIndex.master(), "not-master"); require(msg.sender == instaIndex.master(), "not-master");
@ -189,5 +198,4 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable {
} }
receive() external payable {} receive() external payable {}
} }