From d58cd90a1f09669b1a816e4d3dd1d4dbce31cd5f Mon Sep 17 00:00:00 2001
From: mingda <leimingda@pku.edu.cn>
Date: Sat, 25 Jul 2020 15:48:50 +0800
Subject: [PATCH] Add Admin control to dodo zoo

---
 contracts/DODOZoo.sol | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/contracts/DODOZoo.sol b/contracts/DODOZoo.sol
index a7347ba..75124ad 100644
--- a/contracts/DODOZoo.sol
+++ b/contracts/DODOZoo.sol
@@ -57,6 +57,24 @@ contract DODOZoo is Ownable {
         _DEFAULT_SUPERVISOR_ = _defaultSupervisor;
     }
 
+    // ============ Admin Function ============
+
+    function setDODOLogic(address _dodoLogic) external onlyOwner {
+        _DODO_LOGIC_ = _dodoLogic;
+    }
+
+    function setCloneFactory(address _cloneFactory) external onlyOwner {
+        _CLONE_FACTORY_ = _cloneFactory;
+    }
+
+    function setDefaultSupervisor(address _defaultSupervisor) external onlyOwner {
+        _DEFAULT_SUPERVISOR_ = _defaultSupervisor;
+    }
+
+    function removeDODO(address baseToken, address quoteToken) external onlyOwner {
+        _DODO_REGISTER_[baseToken][quoteToken] = address(0);
+    }
+
     // ============ Breed DODO Function ============
 
     function breedDODO(
@@ -90,10 +108,6 @@ contract DODOZoo is Ownable {
         return newBornDODO;
     }
 
-    function removeDODO(address baseToken, address quoteToken) external onlyOwner {
-        _DODO_REGISTER_[baseToken][quoteToken] = address(0);
-    }
-
     // ============ View Functions ============
 
     function isDODORegistered(address baseToken, address quoteToken) public view returns (bool) {