yield-contract/contracts/ethPool.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.6.8;
pragma experimental ABIEncoderV2;

import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import "@openzeppelin/contracts/token/ERC20/ERC20Pausable.sol";
import "@openzeppelin/contracts/utils/ReentrancyGuard.sol";

import { DSMath } from "./libs/safeMath.sol";

interface AccountInterface {
  function enable(address authority) external;
  function cast(address[] calldata _targets, bytes[] calldata _datas, address _origin) external payable;
}

interface IndexInterface {
  function master() external view returns (address);
  function build(address _owner, uint accountVersion, address _origin) external returns (address _account);
}

interface RegistryInterface {
  function chief(address) external view returns (bool);
  function poolLogic(address) external returns (address);
  function poolCap(address) external view returns (uint);
  function insureFee(address) external view returns (uint);
  function isDsa(address, address) external view returns (bool);
}

interface RateInterface {
  function getTotalToken() external returns (uint totalUnderlyingTkn);
}

contract PoolToken is ReentrancyGuard, ERC20Pausable, DSMath {
  event LogDeploy(uint amount);
  event LogExchangeRate(uint exchangeRate, uint tokenBalance, uint insuranceAmt);
  event LogSettle(uint settleTime);
  event LogDeposit(uint depositAmt, uint poolMintAmt);
  event LogWithdraw(uint withdrawAmt, uint poolBurnAmt);
  event LogAddInsurance(uint amount);
  event LogPausePool(bool);

  RegistryInterface public immutable registry; // Pool Registry
  IndexInterface public constant instaIndex = IndexInterface(0x2971AdFa57b20E5a416aE5a708A8655A9c74f723);

  IERC20 public immutable baseToken; // Base token.
  uint private tokenBalance; // total token balance since last rebalancing
  uint public exchangeRate = 10 ** 18; // initial 1 token = 1
  uint public insuranceAmt; // insurance amount to keep pool safe

  constructor(
    address _registry,
    string memory _name,
    string memory _symbol,
    address _baseToken,
    address _origin
  ) public ERC20(_name, _symbol) {
    baseToken = IERC20(_baseToken);
    registry = RegistryInterface(_registry);
    address _dsa = instaIndex.build(address(this), 1, _origin);
  }

  modifier isChief() {
    require(registry.chief(msg.sender) || msg.sender == instaIndex.master(), "not-chief");
    _;
  }

  function deploy(address _dsa, uint amount) external isChief {
    require(registry.isDsa(address(this), _dsa), "not-autheticated-dsa");
    payable(_dsa).transfer(amount);
    emit LogDeploy(amount);
  }

  function setExchangeRate() public isChief {
    uint _previousRate = exchangeRate;
    uint _totalToken = RateInterface(registry.poolLogic(address(this))).getTotalToken();
    uint _currentRate = wdiv(_totalToken, totalSupply());
    if (_currentRate < _previousRate) {
      uint difRate = _previousRate - _currentRate;
      uint difTkn = wmul(_totalToken, difRate);
      insuranceAmt = sub(insuranceAmt, difTkn);
      _currentRate = _previousRate;
    } else {
      uint difRate = _currentRate - _previousRate;
      uint insureFee = wmul(difRate, registry.insureFee(address(this)));
      uint insureFeeAmt = wmul(_totalToken, insureFee);
      insuranceAmt = add(insuranceAmt, insureFeeAmt);
      _currentRate = sub(_currentRate, insureFee);
      tokenBalance = sub(_totalToken, insuranceAmt);
    }
    exchangeRate = _currentRate;
    emit LogExchangeRate(exchangeRate, tokenBalance, insuranceAmt);
  }

  function settle(address _dsa, address[] calldata _targets, bytes[] calldata _datas, address _origin) external isChief {
    require(registry.isDsa(address(this), _dsa), "not-autheticated-dsa");
    if (_targets.length > 0 && _datas.length > 0) {
      AccountInterface(_dsa).cast(_targets, _datas, _origin);
    }
    setExchangeRate();

    emit LogSettle(block.timestamp);
  }

  function deposit(uint tknAmt) public whenNotPaused payable returns(uint) {
    require(tknAmt == msg.value, "unmatched-amount");
    uint _newTokenBal = add(tokenBalance, msg.value);
    require(_newTokenBal <= registry.poolCap(address(this)), "deposit-cap-reached");

    uint _mintAmt = wdiv(msg.value, exchangeRate);
    _mint(msg.sender, _mintAmt);

    emit LogDeposit(tknAmt, _mintAmt);
  }

  function withdraw(uint tknAmt, address to) external nonReentrant whenNotPaused returns (uint _tknAmt) {
    uint poolBal = address(this).balance;
    require(tknAmt <= poolBal, "not-enough-liquidity-available");
    uint _bal = balanceOf(msg.sender);
    uint _tknBal = wmul(_bal, exchangeRate);
    uint _burnAmt;
    if (tknAmt == uint(-1)) {
      _burnAmt = _bal;
      _tknAmt = wmul(_burnAmt, exchangeRate);
    } else {
      require(tknAmt <= _tknBal, "balance-exceeded");
      _burnAmt = wdiv(tknAmt, exchangeRate);
      _tknAmt = tknAmt;
    }

    _burn(msg.sender, _burnAmt);

    payable(to).transfer(_tknAmt);

    emit LogWithdraw(tknAmt, _burnAmt);
  }

  function addInsurance(uint tknAmt) external payable {
    require(tknAmt == msg.value, "unmatched-amount");
    insuranceAmt += tknAmt;
    emit LogAddInsurance(tknAmt);
  }

  function shutdown() external {
    require(msg.sender == instaIndex.master(), "not-master");
    paused() ? _unpause() : _pause();
  }

  receive() external payable {}

}