@defisaver/positions-sdk 1.0.3-dev → 1.0.3

package/src/helpers/sparkHelpers/index.ts

@@ -1,151 +1,151 @@
-import Dec from 'decimal.js';
-import Web3 from 'web3';
-import { assetAmountInWei, getAssetInfo, getAssetInfoByAddress } from '@defisaver/tokens';
-import {
-  aprToApy, calcLeverageLiqPrice, getAssetsTotal, isLeveragedPos,
-} from '../../moneymarket';
-import {
-  SparkAggregatedPositionData,
-  SparkAssetsData, SparkHelperCommon, SparkMarketData, SparkUsedAssets,
-} from '../../types';
-import { calculateNetApy } from '../../staking';
-import { ethToWeth, wethToEth } from '../../services/utils';
-import { SparkViewContract } from '../../contracts';
-import { NetworkNumber } from '../../types/common';
-import { borrowOperations } from '../../constants';
-
-export const sparkIsInIsolationMode = ({ usedAssets, assetsData }: { usedAssets: SparkUsedAssets, assetsData: SparkAssetsData }) => Object.values(usedAssets).some(({ symbol, collateral }) => collateral && assetsData[symbol].isIsolated);
-
-export const sparkGetCollSuppliedAssets = ({ usedAssets }: { usedAssets: SparkUsedAssets }) => Object.values(usedAssets).filter(({ isSupplied, collateral }) => isSupplied && collateral);
-
-export const sparkGetSuppliableAssets = ({
-  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest
-}: SparkHelperCommon) => {
-  const data = {
-    usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
-  };
-
-  const collAccountAssets = sparkGetCollSuppliedAssets(data);
-  const marketAssets = Object.values(assetsData);
-
-  if (sparkIsInIsolationMode(data)) {
-    const collAsset = collAccountAssets[0].symbol;
-    return marketAssets.filter(d => d.canBeSupplied).map(({ symbol }) => ({ symbol, canBeCollateral: symbol === collAsset }));
-  }
-
-  return marketAssets.filter(d => d.canBeSupplied).map(({ symbol, isIsolated }) => ({ symbol, canBeCollateral: !isIsolated }));
-};
-
-export const sparkGetSuppliableAsCollAssets = ({
-  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest
-}: SparkHelperCommon) => sparkGetSuppliableAssets({
-  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
-}).filter(({ canBeCollateral }) => canBeCollateral);
-
-export const sparkGetEmodeMutableProps = ({
-  eModeCategory,
-  assetsData,
-}: SparkHelperCommon,
-_asset: string) => {
-  const asset = wethToEth(_asset);
-
-  const assetData = assetsData[asset];
-  if (
-    eModeCategory === 0
-    || assetData.eModeCategory !== eModeCategory
-    || new Dec(assetData?.eModeCategoryData?.collateralFactor || 0).eq(0)
-  ) {
-    const { liquidationRatio, collateralFactor } = assetData;
-    return ({ liquidationRatio, collateralFactor });
-  }
-  const { liquidationRatio, collateralFactor } = assetData.eModeCategoryData;
-  return ({ liquidationRatio, collateralFactor });
-};
-
-export const sparkGetAggregatedPositionData = ({
-  usedAssets,
-  eModeCategory,
-  eModeCategories,
-  assetsData,
-  selectedMarket,
-  network,
-  ...rest
-}: SparkHelperCommon): SparkAggregatedPositionData => {
-  const data = {
-    usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
-  };
-  const payload = {} as SparkAggregatedPositionData;
-  payload.suppliedUsd = getAssetsTotal(usedAssets, ({ isSupplied }: { isSupplied: boolean }) => isSupplied, ({ suppliedUsd }: { suppliedUsd: string }) => suppliedUsd);
-  payload.suppliedCollateralUsd = getAssetsTotal(usedAssets, ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral, ({ suppliedUsd }: { suppliedUsd: string }) => suppliedUsd);
-  payload.borrowedUsd = getAssetsTotal(usedAssets, ({ isBorrowed }: { isBorrowed: boolean }) => isBorrowed, ({ borrowedUsd }: { borrowedUsd: string }) => borrowedUsd);
-  payload.borrowLimitUsd = getAssetsTotal(
-    usedAssets,
-    ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral,
-    ({ symbol, suppliedUsd }: { symbol: string, suppliedUsd: string }) => new Dec(suppliedUsd).mul(sparkGetEmodeMutableProps(data, symbol).collateralFactor),
-  );
-  payload.liquidationLimitUsd = getAssetsTotal(
-    usedAssets,
-    ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral,
-    ({ symbol, suppliedUsd }: { symbol: string, suppliedUsd: string }) => new Dec(suppliedUsd).mul(sparkGetEmodeMutableProps(data, symbol).liquidationRatio),
-  );
-  const leftToBorrowUsd = new Dec(payload.borrowLimitUsd).sub(payload.borrowedUsd);
-  payload.leftToBorrowUsd = leftToBorrowUsd.lte('0') ? '0' : leftToBorrowUsd.toString();
-  payload.ratio = +payload.suppliedUsd ? new Dec(payload.borrowLimitUsd).div(payload.borrowedUsd).mul(100).toString() : '0';
-  payload.collRatio = +payload.suppliedUsd ? new Dec(payload.suppliedCollateralUsd).div(payload.borrowedUsd).mul(100).toString() : '0';
-  const { netApy, incentiveUsd, totalInterestUsd } = calculateNetApy({ usedAssets, assetsData });
-  payload.netApy = netApy;
-  payload.incentiveUsd = incentiveUsd;
-  payload.totalInterestUsd = totalInterestUsd;
-  payload.liqRatio = new Dec(payload.borrowLimitUsd).div(payload.liquidationLimitUsd).toString();
-  payload.liqPercent = new Dec(payload.borrowLimitUsd).div(payload.liquidationLimitUsd).mul(100).toString();
-  const { leveragedType, leveragedAsset } = isLeveragedPos(usedAssets);
-  payload.leveragedType = leveragedType;
-  if (leveragedType !== '') {
-    payload.leveragedAsset = leveragedAsset;
-    let assetPrice = data.assetsData[leveragedAsset].price; // TODO sparkPrice or price??
-    if (leveragedType === 'lsd-leverage') {
-      // Treat ETH like a stablecoin in a long stETH position
-      payload.leveragedLsdAssetRatio = new Dec(assetsData[leveragedAsset].price).div(assetsData.ETH.price).toDP(18).toString();
-      assetPrice = new Dec(assetPrice).div(assetsData.ETH.price).toString();
-    }
-    payload.liquidationPrice = calcLeverageLiqPrice(leveragedType, assetPrice, payload.borrowedUsd, payload.liquidationLimitUsd);
-  }
-  return payload;
-};
-
-export const getApyAfterValuesEstimation = async (selectedMarket: SparkMarketData, actions: [{ action: string, amount: string, asset: string }], web3: Web3) => {
-  const sparkViewContract = SparkViewContract(web3, NetworkNumber.Eth);
-  const params = actions.map(({ action, asset, amount }: { action: string, amount: string, asset: string }) => {
-    const isDebtAsset = borrowOperations.includes(action);
-    const amountInWei = assetAmountInWei(amount, asset);
-    const assetInfo = getAssetInfo(ethToWeth(asset));
-    let liquidityAdded;
-    let liquidityTaken;
-    if (isDebtAsset) {
-      liquidityAdded = action === 'payback' ? amountInWei : '0';
-      liquidityTaken = action === 'borrow' ? amountInWei : '0';
-    } else {
-      liquidityAdded = action === 'collateral' ? amountInWei : '0';
-      liquidityTaken = action === 'withdraw' ? amountInWei : '0';
-    }
-    return {
-      reserveAddress: assetInfo.address,
-      liquidityAdded,
-      liquidityTaken,
-      isDebtAsset,
-    };
-  });
-  const data = await sparkViewContract.methods.getApyAfterValuesEstimation(
-    selectedMarket.providerAddress,
-    params,
-  ).call();
-  const rates: { [key: string]: { supplyRate: string, borrowRate: string } } = {};
-  data.forEach((d) => {
-    const asset = wethToEth(getAssetInfoByAddress(d.reserveAddress).symbol);
-    rates[asset] = {
-      supplyRate: aprToApy(new Dec(d.supplyRate.toString()).div(1e25).toString()),
-      borrowRate: aprToApy(new Dec(d.variableBorrowRate.toString()).div(1e25).toString()),
-    };
-  });
-  return rates;
+import Dec from 'decimal.js';
+import Web3 from 'web3';
+import { assetAmountInWei, getAssetInfo, getAssetInfoByAddress } from '@defisaver/tokens';
+import {
+  aprToApy, calcLeverageLiqPrice, getAssetsTotal, isLeveragedPos,
+} from '../../moneymarket';
+import {
+  SparkAggregatedPositionData,
+  SparkAssetsData, SparkHelperCommon, SparkMarketData, SparkUsedAssets,
+} from '../../types';
+import { calculateNetApy } from '../../staking';
+import { ethToWeth, wethToEth } from '../../services/utils';
+import { SparkViewContract } from '../../contracts';
+import { NetworkNumber } from '../../types/common';
+import { borrowOperations } from '../../constants';
+
+export const sparkIsInIsolationMode = ({ usedAssets, assetsData }: { usedAssets: SparkUsedAssets, assetsData: SparkAssetsData }) => Object.values(usedAssets).some(({ symbol, collateral }) => collateral && assetsData[symbol].isIsolated);
+
+export const sparkGetCollSuppliedAssets = ({ usedAssets }: { usedAssets: SparkUsedAssets }) => Object.values(usedAssets).filter(({ isSupplied, collateral }) => isSupplied && collateral);
+
+export const sparkGetSuppliableAssets = ({
+  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest
+}: SparkHelperCommon) => {
+  const data = {
+    usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
+  };
+
+  const collAccountAssets = sparkGetCollSuppliedAssets(data);
+  const marketAssets = Object.values(assetsData);
+
+  if (sparkIsInIsolationMode(data)) {
+    const collAsset = collAccountAssets[0].symbol;
+    return marketAssets.filter(d => d.canBeSupplied).map(({ symbol }) => ({ symbol, canBeCollateral: symbol === collAsset }));
+  }
+
+  return marketAssets.filter(d => d.canBeSupplied).map(({ symbol, isIsolated }) => ({ symbol, canBeCollateral: !isIsolated }));
+};
+
+export const sparkGetSuppliableAsCollAssets = ({
+  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest
+}: SparkHelperCommon) => sparkGetSuppliableAssets({
+  usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
+}).filter(({ canBeCollateral }) => canBeCollateral);
+
+export const sparkGetEmodeMutableProps = ({
+  eModeCategory,
+  assetsData,
+}: SparkHelperCommon,
+_asset: string) => {
+  const asset = wethToEth(_asset);
+
+  const assetData = assetsData[asset];
+  if (
+    eModeCategory === 0
+    || assetData.eModeCategory !== eModeCategory
+    || new Dec(assetData?.eModeCategoryData?.collateralFactor || 0).eq(0)
+  ) {
+    const { liquidationRatio, collateralFactor } = assetData;
+    return ({ liquidationRatio, collateralFactor });
+  }
+  const { liquidationRatio, collateralFactor } = assetData.eModeCategoryData;
+  return ({ liquidationRatio, collateralFactor });
+};
+
+export const sparkGetAggregatedPositionData = ({
+  usedAssets,
+  eModeCategory,
+  eModeCategories,
+  assetsData,
+  selectedMarket,
+  network,
+  ...rest
+}: SparkHelperCommon): SparkAggregatedPositionData => {
+  const data = {
+    usedAssets, eModeCategory, eModeCategories, assetsData, selectedMarket, network, ...rest,
+  };
+  const payload = {} as SparkAggregatedPositionData;
+  payload.suppliedUsd = getAssetsTotal(usedAssets, ({ isSupplied }: { isSupplied: boolean }) => isSupplied, ({ suppliedUsd }: { suppliedUsd: string }) => suppliedUsd);
+  payload.suppliedCollateralUsd = getAssetsTotal(usedAssets, ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral, ({ suppliedUsd }: { suppliedUsd: string }) => suppliedUsd);
+  payload.borrowedUsd = getAssetsTotal(usedAssets, ({ isBorrowed }: { isBorrowed: boolean }) => isBorrowed, ({ borrowedUsd }: { borrowedUsd: string }) => borrowedUsd);
+  payload.borrowLimitUsd = getAssetsTotal(
+    usedAssets,
+    ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral,
+    ({ symbol, suppliedUsd }: { symbol: string, suppliedUsd: string }) => new Dec(suppliedUsd).mul(sparkGetEmodeMutableProps(data, symbol).collateralFactor),
+  );
+  payload.liquidationLimitUsd = getAssetsTotal(
+    usedAssets,
+    ({ isSupplied, collateral }: { isSupplied: boolean, collateral: boolean }) => isSupplied && collateral,
+    ({ symbol, suppliedUsd }: { symbol: string, suppliedUsd: string }) => new Dec(suppliedUsd).mul(sparkGetEmodeMutableProps(data, symbol).liquidationRatio),
+  );
+  const leftToBorrowUsd = new Dec(payload.borrowLimitUsd).sub(payload.borrowedUsd);
+  payload.leftToBorrowUsd = leftToBorrowUsd.lte('0') ? '0' : leftToBorrowUsd.toString();
+  payload.ratio = +payload.suppliedUsd ? new Dec(payload.borrowLimitUsd).div(payload.borrowedUsd).mul(100).toString() : '0';
+  payload.collRatio = +payload.suppliedUsd ? new Dec(payload.suppliedCollateralUsd).div(payload.borrowedUsd).mul(100).toString() : '0';
+  const { netApy, incentiveUsd, totalInterestUsd } = calculateNetApy({ usedAssets, assetsData });
+  payload.netApy = netApy;
+  payload.incentiveUsd = incentiveUsd;
+  payload.totalInterestUsd = totalInterestUsd;
+  payload.liqRatio = new Dec(payload.borrowLimitUsd).div(payload.liquidationLimitUsd).toString();
+  payload.liqPercent = new Dec(payload.borrowLimitUsd).div(payload.liquidationLimitUsd).mul(100).toString();
+  const { leveragedType, leveragedAsset } = isLeveragedPos(usedAssets);
+  payload.leveragedType = leveragedType;
+  if (leveragedType !== '') {
+    payload.leveragedAsset = leveragedAsset;
+    let assetPrice = data.assetsData[leveragedAsset].price; // TODO sparkPrice or price??
+    if (leveragedType === 'lsd-leverage') {
+      // Treat ETH like a stablecoin in a long stETH position
+      payload.leveragedLsdAssetRatio = new Dec(assetsData[leveragedAsset].price).div(assetsData.ETH.price).toDP(18).toString();
+      assetPrice = new Dec(assetPrice).div(assetsData.ETH.price).toString();
+    }
+    payload.liquidationPrice = calcLeverageLiqPrice(leveragedType, assetPrice, payload.borrowedUsd, payload.liquidationLimitUsd);
+  }
+  return payload;
+};
+
+export const getApyAfterValuesEstimation = async (selectedMarket: SparkMarketData, actions: [{ action: string, amount: string, asset: string }], web3: Web3) => {
+  const sparkViewContract = SparkViewContract(web3, NetworkNumber.Eth);
+  const params = actions.map(({ action, asset, amount }: { action: string, amount: string, asset: string }) => {
+    const isDebtAsset = borrowOperations.includes(action);
+    const amountInWei = assetAmountInWei(amount, asset);
+    const assetInfo = getAssetInfo(ethToWeth(asset));
+    let liquidityAdded;
+    let liquidityTaken;
+    if (isDebtAsset) {
+      liquidityAdded = action === 'payback' ? amountInWei : '0';
+      liquidityTaken = action === 'borrow' ? amountInWei : '0';
+    } else {
+      liquidityAdded = action === 'collateral' ? amountInWei : '0';
+      liquidityTaken = action === 'withdraw' ? amountInWei : '0';
+    }
+    return {
+      reserveAddress: assetInfo.address,
+      liquidityAdded,
+      liquidityTaken,
+      isDebtAsset,
+    };
+  });
+  const data = await sparkViewContract.methods.getApyAfterValuesEstimation(
+    selectedMarket.providerAddress,
+    params,
+  ).call();
+  const rates: { [key: string]: { supplyRate: string, borrowRate: string } } = {};
+  data.forEach((d) => {
+    const asset = wethToEth(getAssetInfoByAddress(d.reserveAddress).symbol);
+    rates[asset] = {
+      supplyRate: aprToApy(new Dec(d.supplyRate.toString()).div(1e25).toString()),
+      borrowRate: aprToApy(new Dec(d.variableBorrowRate.toString()).div(1e25).toString()),
+    };
+  });
+  return rates;
 };

package/src/index.ts

@@ -1,52 +1,52 @@
-import './setup';
-
-import * as fluid from './fluid';
-import * as aaveV3 from './aaveV3';
-import * as morphoAaveV3 from './morphoAaveV3';
-import * as aaveV2 from './aaveV2';
-import * as morphoAaveV2 from './morphoAaveV2';
-import * as compoundV3 from './compoundV3';
-import * as compoundV2 from './compoundV2';
-import * as spark from './spark';
-import * as curveUsd from './curveUsd';
-import * as liquity from './liquity';
-import * as liquityV2 from './liquityV2';
-import * as maker from './maker';
-import * as staking from './staking';
-import * as multicall from './multicall';
-import * as moneymarket from './moneymarket';
-import * as assets from './assets';
-import * as markets from './markets';
-import * as helpers from './helpers';
-import * as chickenBonds from './chickenBonds';
-import * as exchange from './exchange';
-import * as morphoBlue from './morphoBlue';
-import * as llamaLend from './llamaLend';
-import * as eulerV2 from './eulerV2';
-
-export * from './types';
-
-export {
-  aaveV2,
-  aaveV3,
-  morphoAaveV2,
-  morphoAaveV3,
-  compoundV2,
-  compoundV3,
-  spark,
-  curveUsd,
-  liquity,
-  liquityV2,
-  maker,
-  chickenBonds,
-  exchange,
-  staking,
-  multicall,
-  moneymarket,
-  markets,
-  helpers,
-  morphoBlue,
-  llamaLend,
-  eulerV2,
-  fluid,
-};
+import './setup';
+
+import * as fluid from './fluid';
+import * as aaveV3 from './aaveV3';
+import * as morphoAaveV3 from './morphoAaveV3';
+import * as aaveV2 from './aaveV2';
+import * as morphoAaveV2 from './morphoAaveV2';
+import * as compoundV3 from './compoundV3';
+import * as compoundV2 from './compoundV2';
+import * as spark from './spark';
+import * as curveUsd from './curveUsd';
+import * as liquity from './liquity';
+import * as liquityV2 from './liquityV2';
+import * as maker from './maker';
+import * as staking from './staking';
+import * as multicall from './multicall';
+import * as moneymarket from './moneymarket';
+import * as assets from './assets';
+import * as markets from './markets';
+import * as helpers from './helpers';
+import * as chickenBonds from './chickenBonds';
+import * as exchange from './exchange';
+import * as morphoBlue from './morphoBlue';
+import * as llamaLend from './llamaLend';
+import * as eulerV2 from './eulerV2';
+
+export * from './types';
+
+export {
+  aaveV2,
+  aaveV3,
+  morphoAaveV2,
+  morphoAaveV3,
+  compoundV2,
+  compoundV3,
+  spark,
+  curveUsd,
+  liquity,
+  liquityV2,
+  maker,
+  chickenBonds,
+  exchange,
+  staking,
+  multicall,
+  moneymarket,
+  markets,
+  helpers,
+  morphoBlue,
+  llamaLend,
+  eulerV2,
+  fluid,
+};

package/src/liquity/index.ts

@@ -1,117 +1,117 @@
-import Web3 from 'web3';
-import Dec from 'decimal.js';
-import { assetAmountInEth, getAssetInfo } from '@defisaver/tokens';
-import {
-  Blockish, EthAddress, NetworkNumber, PositionBalances,
-} from '../types/common';
-import {
-  LiquityActivePoolContract, LiquityCollSurplusPoolContract, LiquityPriceFeedContract, LiquityTroveManagerContract, LiquityViewContract,
-} from '../contracts';
-import { multicall } from '../multicall';
-import { LIQUITY_TROVE_STATUS_ENUM, LiquityTroveInfo } from '../types';
-import { ZERO_ADDRESS } from '../constants';
-
-export const LIQUITY_NORMAL_MODE_RATIO = 110; // MCR
-export const LIQUITY_RECOVERY_MODE_RATIO = 150; // CCR
-
-export const getLiquityAccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress): Promise<PositionBalances> => {
-  let balances: PositionBalances = {
-    collateral: {},
-    debt: {},
-  };
-
-  if (!address) {
-    return balances;
-  }
-
-  const viewContract = LiquityViewContract(web3, network, block);
-  const troveInfo = await viewContract.methods.getTroveInfo(address).call({}, block);
-
-  balances = {
-    collateral: {
-      [addressMapping ? getAssetInfo('ETH', network).address.toLowerCase() : 'ETH']: troveInfo[1],
-    },
-    debt: {
-      [addressMapping ? getAssetInfo('LUSD', network).address.toLowerCase() : 'LUSD']: troveInfo[2],
-    },
-  };
-
-  return balances;
-};
-
-const _getDebtInFront = async (viewContract: any, address: string, accumulatedSum = '0', iterations = 2000) => viewContract.methods.getDebtInFront(address, accumulatedSum, iterations).call();
-
-export const getDebtInFront = async (viewContract: any, address: string, accumulatedSum = '0', iterations = 2000): Promise<string> => {
-  const { debt, next } = await _getDebtInFront(viewContract, address, accumulatedSum, iterations);
-  if (next === ZERO_ADDRESS) return assetAmountInEth(debt, 'LUSD');
-  return getDebtInFront(viewContract, next, debt, iterations);
-};
-
-export const getLiquityTroveInfo = async (web3: Web3, network: NetworkNumber, address: string): Promise<LiquityTroveInfo> => {
-  const viewContract = LiquityViewContract(web3, network);
-  const collSurplusPoolContract = LiquityCollSurplusPoolContract(web3, network);
-  const troveManagerContract = LiquityTroveManagerContract(web3, network);
-  const priceFeedContract = LiquityPriceFeedContract(web3, network);
-  const activePoolContract = LiquityActivePoolContract(web3, network);
-
-  const multicallData = [
-    {
-      target: viewContract.options.address,
-      abiItem: viewContract.options.jsonInterface.find(({ name }) => name === 'getTroveInfo'),
-      params: [address],
-    },
-    {
-      target: collSurplusPoolContract.options.address,
-      abiItem: collSurplusPoolContract.options.jsonInterface.find(({ name }) => name === 'getCollateral'),
-      params: [address],
-    },
-    {
-      target: troveManagerContract.options.address,
-      abiItem: troveManagerContract.options.jsonInterface.find(({ name }) => name === 'getBorrowingRateWithDecay'),
-      params: [],
-    },
-    {
-      target: priceFeedContract.options.address,
-      abiItem: priceFeedContract.options.jsonInterface.find(({ name }) => name === 'fetchPrice'),
-      params: [],
-    },
-    {
-      target: activePoolContract.options.address,
-      abiItem: activePoolContract.options.jsonInterface.find(({ name }) => name === 'getETH'),
-      params: [],
-    },
-    {
-      target: activePoolContract.options.address,
-      abiItem: activePoolContract.options.jsonInterface.find(({ name }) => name === 'getLUSDDebt'),
-      params: [],
-    },
-  ];
-
-  const [multiRes, debtInFront] = await Promise.all([
-    multicall(multicallData, web3, network),
-    getDebtInFront(viewContract, address),
-  ]);
-
-  const recoveryMode = multiRes[0][6];
-  const totalETH = multiRes[4][0];
-  const totalLUSD = multiRes[5][0];
-
-  const payload = {
-    troveStatus: LIQUITY_TROVE_STATUS_ENUM[+multiRes[0][0].toString()],
-    collateral: assetAmountInEth(multiRes[0][1]),
-    debtInAsset: assetAmountInEth(multiRes[0][2]),
-    TCRatio: assetAmountInEth(multiRes[0][4]),
-    recoveryMode,
-    claimableCollateral: assetAmountInEth(multiRes[1][0]),
-    borrowingRateWithDecay: assetAmountInEth(multiRes[2][0]),
-    assetPrice: assetAmountInEth(multiRes[3][0]),
-    totalETH,
-    totalLUSD,
-    debtInFront,
-    minCollateralRatio: recoveryMode ? LIQUITY_RECOVERY_MODE_RATIO : LIQUITY_NORMAL_MODE_RATIO,
-    priceForRecovery: new Dec(recoveryMode ? LIQUITY_RECOVERY_MODE_RATIO : LIQUITY_NORMAL_MODE_RATIO).mul(totalLUSD).div(totalETH).div(100)
-      .toString(),
-  };
-
-  return payload;
+import Web3 from 'web3';
+import Dec from 'decimal.js';
+import { assetAmountInEth, getAssetInfo } from '@defisaver/tokens';
+import {
+  Blockish, EthAddress, NetworkNumber, PositionBalances,
+} from '../types/common';
+import {
+  LiquityActivePoolContract, LiquityCollSurplusPoolContract, LiquityPriceFeedContract, LiquityTroveManagerContract, LiquityViewContract,
+} from '../contracts';
+import { multicall } from '../multicall';
+import { LIQUITY_TROVE_STATUS_ENUM, LiquityTroveInfo } from '../types';
+import { ZERO_ADDRESS } from '../constants';
+
+export const LIQUITY_NORMAL_MODE_RATIO = 110; // MCR
+export const LIQUITY_RECOVERY_MODE_RATIO = 150; // CCR
+
+export const getLiquityAccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress): Promise<PositionBalances> => {
+  let balances: PositionBalances = {
+    collateral: {},
+    debt: {},
+  };
+
+  if (!address) {
+    return balances;
+  }
+
+  const viewContract = LiquityViewContract(web3, network, block);
+  const troveInfo = await viewContract.methods.getTroveInfo(address).call({}, block);
+
+  balances = {
+    collateral: {
+      [addressMapping ? getAssetInfo('ETH', network).address.toLowerCase() : 'ETH']: troveInfo[1],
+    },
+    debt: {
+      [addressMapping ? getAssetInfo('LUSD', network).address.toLowerCase() : 'LUSD']: troveInfo[2],
+    },
+  };
+
+  return balances;
+};
+
+const _getDebtInFront = async (viewContract: any, address: string, accumulatedSum = '0', iterations = 2000) => viewContract.methods.getDebtInFront(address, accumulatedSum, iterations).call();
+
+export const getDebtInFront = async (viewContract: any, address: string, accumulatedSum = '0', iterations = 2000): Promise<string> => {
+  const { debt, next } = await _getDebtInFront(viewContract, address, accumulatedSum, iterations);
+  if (next === ZERO_ADDRESS) return assetAmountInEth(debt, 'LUSD');
+  return getDebtInFront(viewContract, next, debt, iterations);
+};
+
+export const getLiquityTroveInfo = async (web3: Web3, network: NetworkNumber, address: string): Promise<LiquityTroveInfo> => {
+  const viewContract = LiquityViewContract(web3, network);
+  const collSurplusPoolContract = LiquityCollSurplusPoolContract(web3, network);
+  const troveManagerContract = LiquityTroveManagerContract(web3, network);
+  const priceFeedContract = LiquityPriceFeedContract(web3, network);
+  const activePoolContract = LiquityActivePoolContract(web3, network);
+
+  const multicallData = [
+    {
+      target: viewContract.options.address,
+      abiItem: viewContract.options.jsonInterface.find(({ name }) => name === 'getTroveInfo'),
+      params: [address],
+    },
+    {
+      target: collSurplusPoolContract.options.address,
+      abiItem: collSurplusPoolContract.options.jsonInterface.find(({ name }) => name === 'getCollateral'),
+      params: [address],
+    },
+    {
+      target: troveManagerContract.options.address,
+      abiItem: troveManagerContract.options.jsonInterface.find(({ name }) => name === 'getBorrowingRateWithDecay'),
+      params: [],
+    },
+    {
+      target: priceFeedContract.options.address,
+      abiItem: priceFeedContract.options.jsonInterface.find(({ name }) => name === 'fetchPrice'),
+      params: [],
+    },
+    {
+      target: activePoolContract.options.address,
+      abiItem: activePoolContract.options.jsonInterface.find(({ name }) => name === 'getETH'),
+      params: [],
+    },
+    {
+      target: activePoolContract.options.address,
+      abiItem: activePoolContract.options.jsonInterface.find(({ name }) => name === 'getLUSDDebt'),
+      params: [],
+    },
+  ];
+
+  const [multiRes, debtInFront] = await Promise.all([
+    multicall(multicallData, web3, network),
+    getDebtInFront(viewContract, address),
+  ]);
+
+  const recoveryMode = multiRes[0][6];
+  const totalETH = multiRes[4][0];
+  const totalLUSD = multiRes[5][0];
+
+  const payload = {
+    troveStatus: LIQUITY_TROVE_STATUS_ENUM[+multiRes[0][0].toString()],
+    collateral: assetAmountInEth(multiRes[0][1]),
+    debtInAsset: assetAmountInEth(multiRes[0][2]),
+    TCRatio: assetAmountInEth(multiRes[0][4]),
+    recoveryMode,
+    claimableCollateral: assetAmountInEth(multiRes[1][0]),
+    borrowingRateWithDecay: assetAmountInEth(multiRes[2][0]),
+    assetPrice: assetAmountInEth(multiRes[3][0]),
+    totalETH,
+    totalLUSD,
+    debtInFront,
+    minCollateralRatio: recoveryMode ? LIQUITY_RECOVERY_MODE_RATIO : LIQUITY_NORMAL_MODE_RATIO,
+    priceForRecovery: new Dec(recoveryMode ? LIQUITY_RECOVERY_MODE_RATIO : LIQUITY_NORMAL_MODE_RATIO).mul(totalLUSD).div(totalETH).div(100)
+      .toString(),
+  };
+
+  return payload;
 };