@defisaver/positions-sdk 0.0.186 → 0.0.188-dev-markets

package/cjs/compoundV3/index.js

@@ -25,8 +25,18 @@ const compoundHelpers_1 = require("../helpers/compoundHelpers");
 const compound_1 = require("../markets/compound");
 const priceService_1 = require("../services/priceService");
 const getSupportedAssetsAddressesForMarket = (selectedMarket, network) => selectedMarket.collAssets.map(asset => (0, tokens_1.getAssetInfo)((0, utils_1.ethToWeth)(asset), network)).map(addr => addr.address.toLowerCase());
+const getBaseAssetPriceFunction = (asset) => {
+    switch (asset) {
+        case 'wstETH':
+            return priceService_1.getWstETHPrice;
+        case 'ETH':
+            return priceService_1.getEthPrice;
+        default:
+            return priceService_1.getUSDCPrice;
+    }
+};
 const getCompoundV3MarketsData = (web3, network, selectedMarket, defaultWeb3) => __awaiter(void 0, void 0, void 0, function* () {
-    const baseAssetPrice = selectedMarket.baseAsset === 'ETH' ? yield (0, priceService_1.getEthPrice)(defaultWeb3) : yield (0, priceService_1.getUSDCPrice)(defaultWeb3);
+    const baseAssetPrice = yield getBaseAssetPriceFunction(selectedMarket.baseAsset)(defaultWeb3);
     const compPrice = yield (0, priceService_1.getCompPrice)(defaultWeb3);
     const contract = (0, contracts_1.CompV3ViewContract)(web3, network);
     const CompV3ViewAddress = contract.options.address;
@@ -178,14 +188,9 @@ const getCompoundV3AccountData = (web3, network, address, proxyAddress, extracte
     if (loanData.borrowAmount.toString() !== '0') {
         usedAssets[baseAssetSymbol].isBorrowed = true;
         usedAssets[baseAssetSymbol].borrowed = (0, tokens_1.assetAmountInEth)(loanData.borrowAmount, baseAssetInfo.symbol);
-        if (selectedMarket.value === (0, compound_1.COMPOUND_V3_ETH)(network).value) {
-            usedAssets[baseAssetSymbol].borrowedUsd = new decimal_js_1.default((0, tokens_1.assetAmountInEth)(loanData.borrowValue, baseAssetInfo.symbol))
-                .mul(assetsData[baseAssetSymbol].price)
-                .toString();
-        }
-        else {
-            usedAssets[baseAssetSymbol].borrowedUsd = (0, tokens_1.assetAmountInEth)(loanData.borrowValue, baseAssetInfo.symbol);
-        }
+        usedAssets[baseAssetSymbol].borrowedUsd = new decimal_js_1.default((0, tokens_1.assetAmountInEth)(loanData.borrowValue, baseAssetInfo.symbol))
+            .mul(assetsData[baseAssetSymbol].price)
+            .toString();
     }
     const supportedAssetsAddresses = getSupportedAssetsAddressesForMarket(selectedMarket, network);
     loanData.collAddr.forEach((coll, i) => {