first commit

contracts/impl/Settlement.sol

@@ -0,0 +1,143 @@
+/*
+
+    Copyright 2020 DODO ZOO.
+    SPDX-License-Identifier: Apache-2.0
+
+*/
+
+pragma solidity 0.6.9;
+pragma experimental ABIEncoderV2;
+
+import {SafeMath} from "../lib/SafeMath.sol";
+import {SafeERC20} from "../lib/SafeERC20.sol";
+import {DecimalMath} from "../lib/DecimalMath.sol";
+import {Types} from "../lib/Types.sol";
+import {IERC20} from "../intf/IERC20.sol";
+import {Storage} from "./Storage.sol";
+
+/**
+ * @title Settlement
+ * @author DODO Breeder
+ *
+ * @notice Functions for assets settlement
+ */
+contract Settlement is Storage {
+    using SafeMath for uint256;
+    using SafeERC20 for IERC20;
+
+    // ============ Events ============
+
+    event DonateBaseToken(uint256 amount);
+
+    event DonateQuoteToken(uint256 amount);
+
+    event Claim(address indexed user, uint256 baseTokenAmount, uint256 quoteTokenAmount);
+
+    // ============ Assets IN/OUT Functions ============
+
+    function _baseTokenTransferIn(address from, uint256 amount) internal {
+        IERC20(_BASE_TOKEN_).safeTransferFrom(from, address(this), amount);
+        _BASE_BALANCE_ = _BASE_BALANCE_.add(amount);
+    }
+
+    function _quoteTokenTransferIn(address from, uint256 amount) internal {
+        IERC20(_QUOTE_TOKEN_).safeTransferFrom(from, address(this), amount);
+        _QUOTE_BALANCE_ = _QUOTE_BALANCE_.add(amount);
+    }
+
+    function _baseTokenTransferOut(address to, uint256 amount) internal {
+        IERC20(_BASE_TOKEN_).safeTransfer(to, amount);
+        _BASE_BALANCE_ = _BASE_BALANCE_.sub(amount);
+    }
+
+    function _quoteTokenTransferOut(address to, uint256 amount) internal {
+        IERC20(_QUOTE_TOKEN_).safeTransfer(to, amount);
+        _QUOTE_BALANCE_ = _QUOTE_BALANCE_.sub(amount);
+    }
+
+    // ============ Donate to Liquidity Pool Functions ============
+
+    function _donateBaseToken(uint256 amount) internal {
+        _TARGET_BASE_TOKEN_AMOUNT_ = _TARGET_BASE_TOKEN_AMOUNT_.add(amount);
+        emit DonateBaseToken(amount);
+    }
+
+    function _donateQuoteToken(uint256 amount) internal {
+        _TARGET_QUOTE_TOKEN_AMOUNT_ = _TARGET_QUOTE_TOKEN_AMOUNT_.add(amount);
+        emit DonateQuoteToken(amount);
+    }
+
+    function donateBaseToken(uint256 amount) external {
+        _baseTokenTransferIn(msg.sender, amount);
+        _donateBaseToken(amount);
+    }
+
+    function donateQuoteToken(uint256 amount) external {
+        _quoteTokenTransferIn(msg.sender, amount);
+        _donateQuoteToken(amount);
+    }
+
+    // ============ Final Settlement Functions ============
+
+    // last step to shut down dodo
+    function finalSettlement() external onlyOwner notClosed {
+        _CLOSED_ = true;
+        _DEPOSIT_QUOTE_ALLOWED_ = false;
+        _DEPOSIT_BASE_ALLOWED_ = false;
+        _TRADE_ALLOWED_ = false;
+        uint256 totalBaseCapital = getTotalBaseCapital();
+        uint256 totalQuoteCapital = getTotalQuoteCapital();
+
+        if (_QUOTE_BALANCE_ > _TARGET_QUOTE_TOKEN_AMOUNT_) {
+            uint256 spareQuote = _QUOTE_BALANCE_.sub(_TARGET_QUOTE_TOKEN_AMOUNT_);
+            _BASE_CAPITAL_RECEIVE_QUOTE_ = DecimalMath.divFloor(spareQuote, totalBaseCapital);
+        } else {
+            _TARGET_QUOTE_TOKEN_AMOUNT_ = _QUOTE_BALANCE_;
+        }
+
+        if (_BASE_BALANCE_ > _TARGET_BASE_TOKEN_AMOUNT_) {
+            uint256 spareBase = _BASE_BALANCE_.sub(_TARGET_BASE_TOKEN_AMOUNT_);
+            _QUOTE_CAPITAL_RECEIVE_BASE_ = DecimalMath.divFloor(spareBase, totalQuoteCapital);
+        } else {
+            _TARGET_BASE_TOKEN_AMOUNT_ = _BASE_BALANCE_;
+        }
+
+        _R_STATUS_ = Types.RStatus.ONE;
+    }
+
+    // claim remaining assets after final settlement
+    function claim() external preventReentrant {
+        require(_CLOSED_, "DODO_IS_NOT_CLOSED");
+        require(!_CLAIMED_[msg.sender], "ALREADY_CLAIMED");
+        _CLAIMED_[msg.sender] = true;
+        uint256 quoteAmount = DecimalMath.mul(
+            getBaseCapitalBalanceOf(msg.sender),
+            _BASE_CAPITAL_RECEIVE_QUOTE_
+        );
+        uint256 baseAmount = DecimalMath.mul(
+            getQuoteCapitalBalanceOf(msg.sender),
+            _QUOTE_CAPITAL_RECEIVE_BASE_
+        );
+        _baseTokenTransferOut(msg.sender, baseAmount);
+        _quoteTokenTransferOut(msg.sender, quoteAmount);
+        emit Claim(msg.sender, baseAmount, quoteAmount);
+        return;
+    }
+
+    // in case someone transfer to contract directly
+    function retrieve(address token, uint256 amount) external onlyOwner {
+        if (token == _BASE_TOKEN_) {
+            require(
+                IERC20(_BASE_TOKEN_).balanceOf(address(this)) >= _BASE_BALANCE_.add(amount),
+                "DODO_BASE_BALANCE_NOT_ENOUGH"
+            );
+        }
+        if (token == _QUOTE_TOKEN_) {
+            require(
+                IERC20(_QUOTE_TOKEN_).balanceOf(address(this)) >= _QUOTE_BALANCE_.add(amount),
+                "DODO_QUOTE_BALANCE_NOT_ENOUGH"
+            );
+        }
+        IERC20(token).safeTransfer(msg.sender, amount);
+    }
+}