fix dvm test context

2020-11-20 18:58:35 +08:00
parent 7777750ff2
commit 047d7f1d94
82 changed files with 107 additions and 851293 deletions
--- a/contracts/SmartRoute/DVMProxy.sol
+++ b/contracts/SmartRoute/DVMProxy.sol
@@ -0,0 +1,99 @@
+/*
+
+    Copyright 2020 DODO ZOO.
+    SPDX-License-Identifier: Apache-2.0
+
+*/
+
+pragma solidity 0.6.9;
+
+import {IDVM} from "../DODOVendingMachine/intf/IDVM.sol";
+import {IERC20} from "../intf/IERC20.sol";
+import {SafeERC20} from "../lib/SafeERC20.sol";
+import {SafeMath} from "../lib/SafeMath.sol";
+import {DecimalMath} from "../lib/DecimalMath.sol";
+
+contract DVMProxy {
+    using SafeMath for uint256;
+    using SafeERC20 for IERC20;
+
+    function sellBaseOnDVM(
+        address DVMAddress,
+        address to,
+        uint256 baseAmount,
+        uint256 minReceive
+    ) public returns (uint256 receiveAmount) {
+        IERC20(IDVM(DVMAddress)._BASE_TOKEN_()).safeTransferFrom(
+            msg.sender,
+            DVMAddress,
+            baseAmount
+        );
+        receiveAmount = IDVM(DVMAddress).sellBase(to);
+        require(receiveAmount >= minReceive, "RECEIVE_NOT_ENOUGH");
+        return receiveAmount;
+    }
+
+    function sellQuoteOnDVM(
+        address DVMAddress,
+        address to,
+        uint256 quoteAmount,
+        uint256 minReceive
+    ) public returns (uint256 receiveAmount) {
+        IERC20(IDVM(DVMAddress)._QUOTE_TOKEN_()).safeTransferFrom(
+            msg.sender,
+            DVMAddress,
+            quoteAmount
+        );
+        receiveAmount = IDVM(DVMAddress).sellQuote(to);
+        require(receiveAmount >= minReceive, "RECEIVE_NOT_ENOUGU");
+        return receiveAmount;
+    }
+
+    function depositToDVM(
+        address DVMAddress,
+        address to,
+        uint256 baseAmount,
+        uint256 quoteAmount
+    ) public returns (uint256 shares) {
+        uint256 adjustedBaseAmount;
+        uint256 adjustedQuoteAmount;
+        (uint256 baseReserve, uint256 quoteReserve) = IDVM(DVMAddress).getVaultReserve();
+
+        if (quoteReserve == 0 && baseReserve == 0) {
+            adjustedBaseAmount = baseAmount;
+            adjustedQuoteAmount = quoteAmount;
+        }
+
+        if (quoteReserve == 0 && baseReserve > 0) {
+            adjustedBaseAmount = baseAmount;
+            adjustedQuoteAmount = 0;
+        }
+
+        if (quoteReserve > 0 && baseReserve > 0) {
+            uint256 baseIncreaseRatio = DecimalMath.divFloor(baseAmount, baseReserve);
+            uint256 quoteIncreaseRatio = DecimalMath.divFloor(quoteAmount, quoteReserve);
+            if (baseIncreaseRatio <= quoteIncreaseRatio) {
+                adjustedBaseAmount = baseAmount;
+                adjustedQuoteAmount = DecimalMath.mulFloor(quoteReserve, baseIncreaseRatio);
+            } else {
+                adjustedQuoteAmount = quoteAmount;
+                adjustedBaseAmount = DecimalMath.mulFloor(baseReserve, quoteIncreaseRatio);
+            }
+        }
+
+        IERC20(IDVM(DVMAddress)._BASE_TOKEN_()).safeTransferFrom(
+            msg.sender,
+            DVMAddress,
+            adjustedBaseAmount
+        );
+        IERC20(IDVM(DVMAddress)._QUOTE_TOKEN_()).safeTransferFrom(
+            msg.sender,
+            DVMAddress,
+            adjustedQuoteAmount
+        );
+
+        shares = IDVM(DVMAddress).buyShares(to);
+
+        return shares;
+    }
+}