From 80164b1683d5f620b88fa04f68bb63f454078424 Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Sun, 27 Sep 2020 22:35:21 +0530
Subject: [PATCH] done with dai pool exchange rate

---
 contracts/logics/exchangeRate/daiLogic.sol | 101 +++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 contracts/logics/exchangeRate/daiLogic.sol

diff --git a/contracts/logics/exchangeRate/daiLogic.sol b/contracts/logics/exchangeRate/daiLogic.sol
new file mode 100644
index 0000000..a8b6a48
--- /dev/null
+++ b/contracts/logics/exchangeRate/daiLogic.sol
@@ -0,0 +1,101 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.6.8;
+
+import { DSMath } from "../../libs/safeMath.sol";
+
+interface CTokenInterface {
+    function borrowBalanceCurrent(address account) external returns (uint256);
+    function exchangeRateCurrent() external returns (uint256);
+
+    function balanceOf(address owner) external view returns (uint256);
+
+    function underlying() external view returns (address);
+}
+
+interface CompTroller {
+    function getAllMarkets() external view returns (address[] memory);
+}
+
+interface ICurve {
+    function get_virtual_price() external view returns (uint256 out);
+}
+
+interface TokenInterface {
+    function balanceOf(address owner) external view returns (uint256);
+}
+
+interface PriceFeedInterface {
+    function getPrices(address[] memory tokens) external view returns (uint256[] memory pricesInETH);
+    function getPrice(address token) external view returns (uint256 priceInETH);
+    function getEthPrice() external view returns (uint256 ethPriceUSD);
+}
+
+contract DaiRateLogic is DSMath {
+    address public immutable poolToken;
+    address public immutable dsa;
+
+    address public constant ethAddr = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
+    address public constant daiAddr = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
+
+    address public constant PriceFeedAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+
+    address public constant compTrollerAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant cethAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    
+    address public constant curve3poolAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+    address public constant curve3poolTokenAddr = address(0xe81F70Cc7C0D46e12d70efc60607F16bbD617E88);
+
+    function getCompoundNetAssetsInEth(address _dsa) private returns (uint256 _netBal) {
+        uint totalSupplyInETH;
+        uint totalBorrowInETH;
+        address[] memory allMarkets = CompTroller(compTrollerAddr).getAllMarkets();
+        PriceFeedInterface priceFeedContract = PriceFeedInterface(PriceFeedAddr);
+
+        for (uint i = 0; i < allMarkets.length; i++) {
+            CTokenInterface ctoken = CTokenInterface(allMarkets[i]);
+            uint tokenPriceInETH = address(ctoken) == cethAddr ? 10 ** 18 : priceFeedContract.getPrice(ctoken.underlying());
+            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 view returns (uint256 _netBal) {
+        PriceFeedInterface priceFeedContract = PriceFeedInterface(PriceFeedAddr);
+        uint ethPriceUSD = priceFeedContract.getEthPrice();
+        uint virtualPrice = ICurve(curve3poolAddr).get_virtual_price();
+        uint curveTokenBal = TokenInterface(curve3poolTokenAddr).balanceOf(_dsa);
+        uint amtInUSD = wmul(curveTokenBal, virtualPrice);
+        uint amtInETH = wdiv(amtInUSD, ethPriceUSD);
+        _netBal = add(_netBal, amtInETH);
+    }
+
+    function getNetDsaAssets(address _dsa) private returns (uint256 _netBal) {
+        _netBal = _dsa.balance;
+        _netBal += getCompoundNetAssetsInEth(_dsa);
+        _netBal += getCurveNetAssetsInEth(_dsa);
+    }
+    
+    function getTotalToken() public returns (uint256) {
+        address _dsa = 0x0000000000000000000000000000000000000000;
+        PriceFeedInterface priceFeedContract = PriceFeedInterface(PriceFeedAddr);
+        uint daiPriceInETH = priceFeedContract.getPrice(daiAddr);
+        
+        uint256 balInEth = poolToken.balance;
+        balInEth += getNetDsaAssets(_dsa);
+        uint balInDai = wdiv(balInEth, daiPriceInETH);
+        return balInDai;
+    }
+
+    constructor (address daiPool, address _dsa) public {
+        poolToken = address(daiPool);
+        dsa = _dsa;
+
+    }
+}
\ No newline at end of file