From d5eaf43bcb7aa076e6ac149b9aa5b982bf4cf964 Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Fri, 11 Sep 2020 02:34:56 +0530
Subject: [PATCH] Almost done with ethLogic exchangeRate

---
 contracts/logics/exchangeRate/ethLogic.sol | 96 ++++++++++++++++++----
 1 file changed, 81 insertions(+), 15 deletions(-)

diff --git a/contracts/logics/exchangeRate/ethLogic.sol b/contracts/logics/exchangeRate/ethLogic.sol
index 0b4e273..a68df45 100644
--- a/contracts/logics/exchangeRate/ethLogic.sol
+++ b/contracts/logics/exchangeRate/ethLogic.sol
@@ -3,36 +3,102 @@ pragma solidity ^0.6.8;
 
 import { DSMath } from "../../libs/safeMath.sol";
 
-contract EthRateLogic {
-    address poolToken;
+// TODO - have to check y pool virtual price
+interface CTokenInterface {
+    function borrowBalanceCurrent(address account) external returns (uint256);
+    function exchangeRateCurrent() external returns (uint256);
 
-    function getCompoundNetAssetsInEth(address _dsa) private returns (uint _netBal) {
-        // Logics
-        // Take price from Compound's Oracle?
+    function balanceOf(address owner) external view returns (uint256);
+}
+
+interface CompTroller {
+    function getAllMarkets() external view returns (address[] memory);
+}
+
+interface OracleComp {
+    function getUnderlyingPrice(address) external view returns (uint);
+}
+
+interface InstaMapping {
+    function cTokenMapping(address) external view returns (address);
+}
+
+interface CurveMapping {
+    function curvePoolMapping(address poolAddr) external view returns (address);
+}
+
+interface CurveRegistry {
+    function pool_list(uint) external view returns (address poolAddress, address poolToken, address ctokenPrice);
+    function pool_count() external view returns (uint);
+}
+
+interface ICurve {
+    function get_virtual_price() external view returns (uint256 out);
+}
+
+interface TokenInterface {
+    function balanceOf(address owner) external view returns (uint256);
+}
+
+
+contract EthRateLogic is DSMath {
+    address public immutable poolToken;
+    address public constant compTrollerAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant cethAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant compOracleAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant ctokenMapping = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant curveRegistryAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+
+    function getCompoundNetAssetsInEth(address _dsa) private returns (uint256 _netBal) {
+        uint totalSupplyInETH;
+        uint totalBorrowInETH;
+        address[] memory allMarkets = CompTroller(compTrollerAddr).getAllMarkets();
+        OracleComp oracleContract = OracleComp(compOracleAddr);
+        uint ethPrice = oracleContract.getUnderlyingPrice(cethAddr);
+        for (uint i = 0; i < allMarkets.length; i++) {
+            CTokenInterface ctoken = CTokenInterface(allMarkets[i]);
+            uint tokenPriceInETH = wdiv(oracleContract.getUnderlyingPrice(address(ctoken)), ethPrice);
+            uint supply = wmul(ctoken.balanceOf(_dsa), ctoken.exchangeRateCurrent());
+            uint supplyInETH = wmul(supply, tokenPriceInETH);
+
+            uint borrow = ctoken.borrowBalanceCurrent(_dsa);
+            uint borrowInETH = wmul(borrow, tokenPriceInETH);
+
+            totalSupplyInETH += add(totalSupplyInETH, supplyInETH);
+            totalBorrowInETH = add(totalBorrowInETH, borrowInETH);
+        }
+        _netBal = sub(totalSupplyInETH, totalBorrowInETH);
     }
 
-    function getCurveNetAssetsInEth(address _dsa) private returns (uint _netBal) {
-        // Logics
-        // Take price from ChainLink's Oracle?
+    function getCurveNetAssetsInEth(address _dsa) private returns (uint256 _netBal) {
+        CurveRegistry curveRegistry = CurveRegistry(curveRegistryAddr);
+        uint poolLen = curveRegistry.pool_count();
+        OracleComp oracleContract = OracleComp(compOracleAddr);
+        uint ethPrice = oracleContract.getUnderlyingPrice(cethAddr);
+        for (uint i = 0; i < poolLen; i++) {
+            (address curvePoolAddr, address curveTokenAddr, address ctokenPriceAddr) = curveRegistry.pool_list(i);
+            uint tokenPriceInETH = wdiv(oracleContract.getUnderlyingPrice(address(ctokenPriceAddr)), ethPrice);
+            uint virtualPrice = ICurve(curvePoolAddr).get_virtual_price();
+            uint curveTokenBal = TokenInterface(curveTokenAddr).balanceOf(_dsa);
+            uint amtInUSD = wmul(curveTokenBal, virtualPrice);
+            uint amtInETH = wmul(amtInUSD, tokenPriceInETH);
+            _netBal = add(_netBal, amtInETH);
+        }
     }
 
-    function getNetDsaAssets(address _dsa) private returns (uint _netBal) {
+    function getNetDsaAssets(address _dsa) private returns (uint256 _netBal) {
         _netBal = _dsa.balance;
         _netBal += getCompoundNetAssetsInEth(_dsa);
         _netBal += getCurveNetAssetsInEth(_dsa);
     }
     
-    function getTotalToken() public returns (uint) {
+    function getTotalToken() public returns (uint256) {
         address _dsa = 0x0000000000000000000000000000000000000000;
-        uint bal = poolToken.balance;
+        uint256 bal = poolToken.balance;
         bal += getNetDsaAssets(_dsa);
         return bal;
     }
 
-    function reduceETH(uint amt) public {
-        payable(address(0)).transfer(amt);
-    }
-
     constructor (address ethPool) public {
         poolToken = address(ethPool);
     }