init test env

2020-10-26 19:19:10 +08:00
parent dd1cb18cc4
commit a61bdad959
26 changed files with 147 additions and 55 deletions
--- a/contracts/DODOVendorMachine/DVMFactory.sol
+++ b/contracts/DODOVendorMachine/DVMFactory.sol
@@ -0,0 +1,70 @@
+/*
+
+    Copyright 2020 DODO ZOO.
+    SPDX-License-Identifier: Apache-2.0
+
+*/
+
+pragma solidity 0.6.9;
+pragma experimental ABIEncoderV2;
+
+import {Ownable} from "../lib/Ownable.sol";
+import {ICloneFactory} from "../lib/CloneFactory.sol";
+import {DVMVault} from "./impl/DVMVault.sol";
+import {DVMController} from "./impl/DVMController.sol";
+
+contract DVMFactory is Ownable {
+    address public _CLONE_FACTORY_;
+    address public _VAULT_TEMPLATE_;
+    address public _CONTROLLER_TEMPLATE_;
+
+    // base -> quote -> DVM address list
+    mapping(address => mapping(address => address[])) _REGISTRY_;
+
+    constructor(
+        address cloneFactory,
+        address vaultTemplate,
+        address controllerTemplate
+    ) public {
+        _CLONE_FACTORY_ = cloneFactory;
+        _VAULT_TEMPLATE_ = vaultTemplate;
+        _CONTROLLER_TEMPLATE_ = controllerTemplate;
+    }
+
+    function createDODOVendorMachine(
+        address maintainer,
+        address baseToken,
+        address quoteToken,
+        address lpFeeRateModel,
+        address mtFeeRateModel,
+        uint256 i,
+        uint256 k,
+        uint256 gasPriceLimit
+    ) external returns (address newVendorMachine) {
+        newVendorMachine = ICloneFactory(_CLONE_FACTORY_).clone(_CONTROLLER_TEMPLATE_);
+        address vault = ICloneFactory(_CLONE_FACTORY_).clone(_VAULT_TEMPLATE_);
+        DVMVault(vault).init(newVendorMachine, baseToken, quoteToken); // vault owner is controller
+
+        DVMController(newVendorMachine).init(
+            msg.sender,
+            maintainer,
+            vault,
+            lpFeeRateModel,
+            mtFeeRateModel,
+            i,
+            k,
+            gasPriceLimit
+        );
+
+        _REGISTRY_[baseToken][quoteToken].push(newVendorMachine);
+        return newVendorMachine;
+    }
+
+    function getVendorMachine(address baseToken, address quoteToken)
+        external
+        view
+        returns (address[] memory machines)
+    {
+        return _REGISTRY_[baseToken][quoteToken];
+    }
+}