update solv oracle adapter

DODOEX · Jun 27, 2024 · 58c463e · 58c463e
1 parent 4f92eaf
commit 58c463e
Showing 1 changed file with 33 additions and 3 deletions.
diff --git a/contracts/GasSavingPool/adapter/SolvOracleAdapter.sol b/contracts/GasSavingPool/adapter/SolvOracleAdapter.sol
@@ -1,16 +1,21 @@
 /*
-
     Copyright 2020 DODO ZOO.
     SPDX-License-Identifier: Apache-2.0
-
 */
 
 pragma solidity 0.8.16;
 
-import {IOracle} from "../intf/IOracle.sol";
+interface IOracle {
+    function prices(address base) external view returns (uint256);
+}
 
 interface ISftWrappedToken {
     function getValueByShares(uint256 shares) external view returns (uint256 value);
+    function underlyingAsset() external view returns (address);
+}
+
+interface IERC20 {
+    function decimals() external view returns(uint8);
 }
 
 contract SolvOracleAdapter is IOracle {
@@ -19,5 +24,30 @@ contract SolvOracleAdapter is IOracle {
         uint256 shares = 1e18;
         uint256 value = ISftWrappedToken(base).getValueByShares(shares);
         price = value / 1;
+        (uint256 decimalCorrect, bool multiplyOrNot) = getDecimalCorrect(base);
+        if(decimalCorrect > 0) {
+            price = multiplyOrNot ? price * (10 ** decimalCorrect) : price / (10 ** decimalCorrect);
+        }
+    }
+
+    function getDecimalCorrect(address base) public view returns (uint256 decimalCorrect, bool multiplyOrNot) {
+        address underlyingAsset = ISftWrappedToken(base).underlyingAsset();
+        uint256 decimals = IERC20(underlyingAsset).decimals();
+
+        if(18 > decimals) {
+            decimalCorrect = 18 - decimals;
+            multiplyOrNot = true;
+        } else if(18 == decimals) {
+            decimalCorrect = 0;
+            multiplyOrNot = true;
+        } else if(18 < decimals) {
+            decimalCorrect = decimals - 18;
+            multiplyOrNot = false;
+        }
+    }
+
+    function getUnderlyingAsset(address base) public view returns(address asset, uint256 dec) {
+        asset = ISftWrappedToken(base).underlyingAsset();
+        dec = IERC20(asset).decimals();
     }
 }