From f5d345ec66e551a3ed61bb7a1579794f1449820a Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Tue, 26 Nov 2019 05:12:46 +0800
Subject: [PATCH] added cToken import

---
 .../ProxyLogics/import/InstaTokenImport.sol   | 64 ++++++++++++++++++-
 1 file changed, 63 insertions(+), 1 deletion(-)

diff --git a/contracts/ProxyLogics/import/InstaTokenImport.sol b/contracts/ProxyLogics/import/InstaTokenImport.sol
index f06cceb..44acbf6 100644
--- a/contracts/ProxyLogics/import/InstaTokenImport.sol
+++ b/contracts/ProxyLogics/import/InstaTokenImport.sol
@@ -8,6 +8,11 @@ interface ERC20Interface {
     function transferFrom(address, address, uint) external returns (bool);
 }
 
+interface ComptrollerInterface {
+    function enterMarkets(address[] calldata cTokens) external returns (uint[] memory);
+    function getAssetsIn(address account) external view returns (address[] memory);
+}
+
 
 contract DSMath {
     function add(uint x, uint y) internal pure returns (uint z) {
@@ -26,8 +31,41 @@ contract DSMath {
 }
 
 
-contract ImportResolver is DSMath {
+contract Helper is DSMath {
+
+    /**
+     * @dev get Compound Comptroller Address
+     */
+    function getComptrollerAddress() public pure returns (address troller) {
+        troller = 0x3d9819210A31b4961b30EF54bE2aeD79B9c9Cd3B;
+    }
+
+    function enterMarket(address[] memory cErc20) internal {
+        ComptrollerInterface troller = ComptrollerInterface(getComptrollerAddress());
+        address[] memory markets = troller.getAssetsIn(address(this));
+        address[] memory toEnter = new address[](cErc20.length);
+        uint count = 0;
+        for (uint j = 0; j < cErc20.length; j++) {
+            bool isEntered = false;
+            for (uint i = 0; i < markets.length; i++) {
+                if (markets[i] == cErc20[j]) {
+                    isEntered = true;
+                    break;
+                }
+            }
+            if (!isEntered) {
+                toEnter[count] = cErc20[j];
+                count += 1;
+            }
+        }
+        troller.enterMarkets(toEnter);
+    }
+}
+
+
+contract ImportResolver is Helper {
     event LogTokensImport(address owner, uint percentage, address[] tokenAddr, uint[] tokenBalArr);
+    event LogCTokensImport(address owner, uint percentage, address[] tokenAddr, uint[] tokenBalArr);
 
     function importTokens(uint toConvert, address[] memory tokenAddrArr) public {
         uint[] memory tokenBalArr = new uint[](tokenAddrArr.length);
@@ -51,6 +89,30 @@ contract ImportResolver is DSMath {
             tokenBalArr
         );
     }
+
+    function importCTokens(uint toConvert, address[] memory ctokenAddrArr) public {
+        uint[] memory tokenBalArr = new uint[](ctokenAddrArr.length);
+
+         // transfer tokens to InstaDApp smart wallet from user wallet
+        enterMarket(ctokenAddrArr);
+        for (uint i = 0; i < ctokenAddrArr.length; i++) {
+            address erc20 = ctokenAddrArr[i];
+            ERC20Interface tknContract = ERC20Interface(erc20);
+            uint tokenBal = tknContract.balanceOf(msg.sender);
+            tokenBal = toConvert < 10**18 ? wmul(tokenBal, toConvert) : tokenBal;
+            if (tokenBal > 0) {
+                require(tknContract.transferFrom(msg.sender, address(this), tokenBal), "Allowance?");
+            }
+            tokenBalArr[i] = tokenBal;
+        }
+
+        emit LogCTokensImport(
+            msg.sender,
+            toConvert,
+            ctokenAddrArr,
+            tokenBalArr
+        );
+    }
 }