@defisaver/positions-sdk 0.0.1

package/src/compoundV2/index.ts

@@ -0,0 +1,207 @@
+import { assetAmountInEth, getAssetInfo, getAssetInfoByAddress } from '@defisaver/tokens';
+import Dec from 'decimal.js';
+import Web3 from 'web3';
+import { BLOCKS_IN_A_YEAR } from '../constants';
+import { aprToApy } from '../moneymarket';
+import { compareAddresses, handleWbtcLegacy, wethToEth } from '../services/utils';
+import {
+  Blockish, EthAddress, NetworkNumber, PositionBalances,
+} from '../types/common';
+import { CompoundLoanInfoContract, ComptrollerContract } from '../contracts';
+import { compoundV2CollateralAssets } from '../markets/compound/marketsAssets';
+import {
+  CompoundV2AssetsData, CompoundV2MarketsData, CompoundV2PositionData, CompoundV2UsedAsset, CompoundV2UsedAssets,
+} from '../types';
+import { getCompoundV2AggregatedData } from '../helpers/compoundHelpers';
+
+const compAddress = '0xc00e94cb662c3520282e6f5717214004a7f26888';
+
+export const getCompoundV2MarketsData = async (web3: Web3, network: NetworkNumber): Promise<CompoundV2MarketsData> => {
+  const cAddresses = compoundV2CollateralAssets.map(a => a.address);
+
+  const loanInfoContract = CompoundLoanInfoContract(web3, network);
+  const loanInfo = await loanInfoContract.methods.getFullTokensInfo(cAddresses).call();
+
+  const compPrice = loanInfo.find(m => compareAddresses(m.underlyingTokenAddress, compAddress))!.price.toString();
+
+  const assetsData = loanInfo
+    .map((market, i) => {
+      let symbol = getAssetInfoByAddress(cAddresses[i]).underlyingAsset;
+      let isWbtcLegacy = false;
+      const totalSupply = new Dec(market.totalSupply.toString()).div(1e18).times(market.exchangeRate.toString());
+      const totalBorrow = market.totalBorrow.toString();
+      const borrowCap = market.borrowCap.toString();
+      const compSupplySpeeds = market.compSupplySpeeds.toString();
+      const compBorrowSpeeds = market.compBorrowSpeeds.toString();
+      const assetPrice = market.price.toString();
+
+      const pricePrecisionDiff = 18 - getAssetInfo(getAssetInfoByAddress(cAddresses[i]).underlyingAsset).decimals;
+
+      // compSupplySpeeds/compBorrowSpeeds is per block per market (borrow & supply are separate markets)
+      const incentiveSupplyApy = aprToApy((100 * BLOCKS_IN_A_YEAR * +compSupplySpeeds * +compPrice) / +assetPrice / +totalSupply).toString();
+      const incentiveBorrowApy = aprToApy((100 * BLOCKS_IN_A_YEAR * +compBorrowSpeeds * +compPrice) / +assetPrice / +totalBorrow).toString();
+
+      if (cAddresses[i].toLowerCase() === '0xc11b1268c1a384e55c48c2391d8d480264a3a7f4'.toLowerCase()) {
+        symbol = 'WBTC Legacy';
+        isWbtcLegacy = true;
+      }
+      return {
+        symbol,
+        underlyingTokenAddress: market.underlyingTokenAddress,
+        supplyRate: aprToApy(new Dec(BLOCKS_IN_A_YEAR).times(market.supplyRate.toString()).div(1e16).toString()).toString(),
+        borrowRate: aprToApy(new Dec(BLOCKS_IN_A_YEAR).times(market.borrowRate.toString()).div(1e16).toString()).toString(),
+        incentiveSupplyToken: 'COMP',
+        incentiveBorrowToken: 'COMP',
+        incentiveSupplyApy,
+        incentiveBorrowApy,
+        collateralFactor: new Dec(market.collateralFactor.toString()).div(1e18).toString(),
+        marketLiquidity: assetAmountInEth(market.marketLiquidity.toString(), handleWbtcLegacy(symbol)),
+        utilization: new Dec(market.totalBorrow.toString()).div(totalSupply).times(100).toString(),
+        totalSupply: assetAmountInEth(totalSupply, handleWbtcLegacy(symbol)),
+        totalBorrow: assetAmountInEth(totalBorrow, handleWbtcLegacy(symbol)),
+        exchangeRate: new Dec(market.exchangeRate.toString()).div(1e28).toString(),
+        borrowCap: assetAmountInEth(borrowCap, handleWbtcLegacy(symbol)),
+        canBeBorrowed: market.canBorrow,
+        canBeSupplied: market.canMint,
+        price: new Dec(market.price.toString()).div(10 ** (18 + pricePrecisionDiff)).toString(),
+      };
+    });
+
+  const payload = {} as CompoundV2AssetsData;
+  // Sort by market size
+  assetsData
+    .sort((a, b) => {
+      const aMarket = new Dec(a.price).times(a.totalSupply).toString();
+      const bMarket = new Dec(b.price).times(b.totalSupply).toString();
+
+      return new Dec(bMarket).minus(aMarket).toNumber();
+    })
+    .forEach((market, i) => { payload[market.symbol] = { ...market, sortIndex: i }; });
+
+  return { assetsData: payload };
+};
+
+export const EMPTY_COMPOUND_DATA = {
+  usedAssets: {},
+  suppliedUsd: '0',
+  borrowedUsd: '0',
+  borrowLimitUsd: '0',
+  leftToBorrowUsd: '0',
+  ratio: '0',
+  minRatio: '0',
+  netApy: '0',
+  incentiveUsd: '0',
+  totalInterestUsd: '0',
+  borrowStableSupplyUnstable: false,
+};
+
+export const getCollateralAssetsAddresses = async (web3: Web3, network: NetworkNumber, account: string) => {
+  const contract = ComptrollerContract(web3, network);
+
+  return contract.methods.getAssetsIn(account).call();
+};
+
+export const getCompoundV2AccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress): Promise<PositionBalances> => {
+  let balances: PositionBalances = {
+    collateral: {},
+    debt: {},
+  };
+
+  if (!address) {
+    return balances;
+  }
+
+  const loanInfoContract = CompoundLoanInfoContract(web3, network, block);
+  const loanInfo = await loanInfoContract.methods.getTokenBalances(address, compoundV2CollateralAssets.map(a => a.address)).call({}, block);
+
+  loanInfo.balances.forEach((weiAmount: any, i: number) => {
+    const asset = wethToEth(compoundV2CollateralAssets[i].symbol === 'cWBTC Legacy'
+      ? `${compoundV2CollateralAssets[i].underlyingAsset} Legacy`
+      : compoundV2CollateralAssets[i].underlyingAsset);
+
+    balances = {
+      collateral: {
+        ...balances.collateral,
+        [addressMapping ? getAssetInfo(asset, network).address.toLowerCase() : asset]: weiAmount.toString(),
+      },
+    };
+  });
+  loanInfo.borrows.forEach((weiAmount: any, i: number) => {
+    const asset = wethToEth(compoundV2CollateralAssets[i].symbol === 'cWBTC Legacy'
+      ? `${compoundV2CollateralAssets[i].underlyingAsset} Legacy`
+      : compoundV2CollateralAssets[i].underlyingAsset);
+
+    balances = {
+      ...balances,
+      debt: {
+        ...balances.debt,
+        [addressMapping ? getAssetInfo(asset, network).address.toLowerCase() : asset]: weiAmount.toString(),
+      },
+    };
+  });
+
+  return balances;
+};
+
+export const getCompoundV2AccountData = async (web3: Web3, network: NetworkNumber, address: string, assetsData: CompoundV2AssetsData): Promise<CompoundV2PositionData> => {
+  if (!address) throw new Error('No address provided');
+
+  let payload = { ...EMPTY_COMPOUND_DATA, lastUpdated: Date.now() };
+
+  const loanInfoContract = CompoundLoanInfoContract(web3, network);
+  const loanInfo = await loanInfoContract.methods.getTokenBalances(address, compoundV2CollateralAssets.map(a => a.address)).call();
+  const collateralAssetsAddresses = await getCollateralAssetsAddresses(web3, network, address);
+
+  const usedAssets = {} as CompoundV2UsedAssets;
+
+  loanInfo.balances.forEach((weiAmount: string, i: number) => {
+    const asset = compoundV2CollateralAssets[i].symbol === 'cWBTC Legacy'
+      ? `${compoundV2CollateralAssets[i].underlyingAsset} Legacy`
+      : compoundV2CollateralAssets[i].underlyingAsset;
+    const amount = assetAmountInEth(weiAmount.toString(), asset);
+    const collateral = !!collateralAssetsAddresses.find(a => compareAddresses(a, compoundV2CollateralAssets[i].address));
+    if (weiAmount.toString() === '0' && !collateral) return;
+    if (!usedAssets[asset]) usedAssets[asset] = {} as CompoundV2UsedAsset;
+    usedAssets[asset] = {
+      ...usedAssets[asset],
+      symbol: asset,
+      supplied: amount,
+      suppliedUsd: new Dec(amount).mul(assetsData[handleWbtcLegacy(asset)].price).toString(),
+      isSupplied: +amount > 0,
+      collateral,
+    };
+  });
+
+  loanInfo.borrows.forEach((weiAmount, i) => {
+    if (weiAmount.toString() === '0') return;
+    const asset = compoundV2CollateralAssets[i].symbol === 'cWBTC Legacy'
+      ? `${compoundV2CollateralAssets[i].underlyingAsset} Legacy`
+      : compoundV2CollateralAssets[i].underlyingAsset;
+    const amount = assetAmountInEth(weiAmount.toString(), asset);
+    if (!usedAssets[asset]) usedAssets[asset] = {} as CompoundV2UsedAsset;
+    usedAssets[asset] = {
+      ...usedAssets[asset],
+      symbol: asset,
+      borrowed: amount,
+      borrowedUsd: new Dec(amount).mul(assetsData[handleWbtcLegacy(asset)].price).toString(),
+      isBorrowed: true,
+      collateral: !!usedAssets[asset].collateral,
+    };
+  });
+
+  payload = {
+    ...payload,
+    usedAssets,
+    ...getCompoundV2AggregatedData({
+      usedAssets, assetsData,
+    }),
+  };
+
+  return payload;
+};
+
+export const getCompoundV2FullPositionData = async (web3: Web3, network: NetworkNumber, address: string): Promise<CompoundV2PositionData> => {
+  const marketData = await getCompoundV2MarketsData(web3, network);
+  const positionData = await getCompoundV2AccountData(web3, network, address, marketData.assetsData);
+  return positionData;
+};

package/src/compoundV3/index.ts

@@ -0,0 +1,259 @@
+import Web3 from 'web3';
+import Dec from 'decimal.js';
+import {
+  assetAmountInEth, assetAmountInWei, getAssetInfo, getAssetInfoByAddress,
+} from '@defisaver/tokens';
+import { CompV3ViewContract } from '../contracts';
+import { multicall } from '../multicall';
+import {
+  CompoundV3AssetData, CompoundMarketData, CompoundV3AssetsData, CompoundV3UsedAssets, CompoundV3MarketsData, CompoundV3PositionData, CompoundVersions,
+} from '../types/compound';
+import {
+  Blockish, EthAddress, NetworkNumber, PositionBalances,
+} from '../types/common';
+import {
+  getCbETHApr, getStETHApr, getStETHByWstETHMultiple, getWstETHByStETH,
+} from '../staking';
+import { wethToEth } from '../services/utils';
+import { ZERO_ADDRESS } from '../constants';
+import { calculateBorrowingAssetLimit } from '../moneymarket';
+import {
+  formatBaseData, formatMarketData, getCompoundV3AggregatedData, getIncentiveApys,
+} from '../helpers/compoundHelpers';
+import { COMPOUND_V3_ETH, COMPOUND_V3_USDBC, COMPOUND_V3_USDC } from '../markets/compound';
+
+export const getCompoundV3MarketsData = async (web3: Web3, network: NetworkNumber, selectedMarket: CompoundMarketData, compPrice: string, defaultWeb3: Web3): Promise<CompoundV3MarketsData> => {
+  const contract = CompV3ViewContract(web3, network);
+  const CompV3ViewAddress = contract.options.address;
+  const calls = [
+    {
+      target: CompV3ViewAddress,
+      abiItem: contract.options.jsonInterface.find((props) => props.name === 'getFullBaseTokenInfo'),
+      params: [selectedMarket.baseMarketAddress],
+    },
+    {
+      target: CompV3ViewAddress,
+      abiItem: contract.options.jsonInterface.find((props) => props.name === 'getFullCollInfos'),
+      params: [selectedMarket.baseMarketAddress],
+    },
+  ];
+  const data = await multicall(calls, web3, network);
+  const colls = data[1].colls.map((coll: any) => formatMarketData(coll, network)) as CompoundV3AssetData[];
+  if (selectedMarket.value === CompoundVersions.CompoundV3ETH) {
+    for (const coll of colls) {
+      if (coll.symbol === 'wstETH') {
+        // eslint-disable-next-line no-await-in-loop
+        const [[totalSupplyAlternative, supplyCapAlternative], priceAlternative] = await Promise.all([
+          getStETHByWstETHMultiple([
+            assetAmountInWei(coll.totalSupply, 'wstETH'),
+            assetAmountInWei(coll.supplyCap, 'wstETH'),
+          ], defaultWeb3),
+          getWstETHByStETH(assetAmountInWei(1, 'stETH'), defaultWeb3),
+        ]);
+        coll.totalSupplyAlternative = assetAmountInEth(totalSupplyAlternative, 'stETH');
+        coll.supplyCapAlternative = assetAmountInEth(supplyCapAlternative, 'stETH');
+        coll.priceAlternative = assetAmountInEth(priceAlternative, 'wstETH');
+        // const stEthMarket = markets.find(({ symbol }) => symbol === 'stETH');
+        // eslint-disable-next-line no-await-in-loop
+        coll.incentiveSupplyApy = await getStETHApr(defaultWeb3);
+        coll.incentiveSupplyToken = 'wstETH';
+      }
+      if (coll.symbol === 'cbETH') {
+        // eslint-disable-next-line no-await-in-loop
+        coll.incentiveSupplyApy = await getCbETHApr(defaultWeb3);
+        coll.incentiveSupplyToken = 'cbETH';
+      }
+    }
+  }
+  const base = formatBaseData(data[0].baseToken, network);
+
+  const payload: CompoundV3AssetsData = {};
+
+  const baseObj = { ...base, ...getIncentiveApys(base, compPrice) };
+  const allAssets = [baseObj, ...colls];
+
+  allAssets
+    .sort((a, b) => {
+      const aMarket = new Dec(a.price).times(a.totalSupply).toString();
+      const bMarket = new Dec(b.price).times(b.totalSupply).toString();
+
+      return new Dec(bMarket).minus(aMarket).toNumber();
+    })
+    .forEach((market, i) => {
+      payload[market.symbol] = { ...market, sortIndex: i };
+    });
+
+  return { assetsData: payload };
+};
+
+export const EMPTY_COMPOUND_V3_DATA = {
+  usedAssets: {},
+  suppliedUsd: '0',
+  borrowedUsd: '0',
+  borrowLimitUsd: '0',
+  leftToBorrowUsd: '0',
+  ratio: '0',
+  minRatio: '0',
+  netApy: '0',
+  incentiveUsd: '0',
+  totalInterestUsd: '0',
+  isSubscribedToAutomation: false,
+  automationResubscribeRequired: false,
+  isAllowed: false,
+  lastUpdated: Date.now(),
+};
+
+export const EMPTY_USED_ASSET = {
+  isSupplied: false,
+  isBorrowed: false,
+  supplied: '0',
+  suppliedUsd: '0',
+  borrowed: '0',
+  borrowedUsd: '0',
+  symbol: '',
+  collateral: true,
+  debt: '0',
+};
+
+export const getCompoundV3AccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress, marketAddress: EthAddress): Promise<PositionBalances> => {
+  let balances: PositionBalances = {
+    collateral: {},
+    debt: {},
+  };
+
+  if (!address) {
+    return balances;
+  }
+
+  const market = ({
+    [COMPOUND_V3_ETH(network).baseMarketAddress.toLowerCase()]: COMPOUND_V3_ETH(network),
+    [COMPOUND_V3_USDC(network).baseMarketAddress.toLowerCase()]: COMPOUND_V3_USDC(network),
+    [COMPOUND_V3_USDBC(network).baseMarketAddress.toLowerCase()]: COMPOUND_V3_USDBC(network),
+  })[marketAddress];
+
+  const loanInfoContract = CompV3ViewContract(web3, network, block);
+  const loanInfo = await loanInfoContract.methods.getLoanData(market.baseMarketAddress, address).call({}, block);
+  const baseAssetInfo = getAssetInfo(wethToEth(market.baseAsset), network);
+
+  balances = {
+    collateral: {
+      [addressMapping ? baseAssetInfo.address.toLowerCase() : baseAssetInfo.symbol]: loanInfo.depositAmount,
+    },
+    debt: {
+      [addressMapping ? baseAssetInfo.address.toLowerCase() : baseAssetInfo.symbol]: loanInfo.borrowAmount,
+    },
+  };
+
+  loanInfo.collAddr.forEach((coll: string, i: number): void => {
+    const symbol = wethToEth(getAssetInfoByAddress(coll, network).symbol);
+    balances = {
+      ...balances,
+      collateral: {
+        ...balances.collateral,
+        [addressMapping ? getAssetInfo(symbol, network).address.toLowerCase() : symbol]: loanInfo.collAmounts[i].toString(),
+      },
+    };
+  });
+
+  return balances;
+};
+
+export const getCompoundV3AccountData = async (
+  web3: Web3,
+  network: NetworkNumber,
+  address: string,
+  proxyAddress: string,
+  extractedState: ({
+    selectedMarket: CompoundMarketData,
+    assetsData: CompoundV3AssetsData,
+  })): Promise<CompoundV3PositionData> => {
+  if (!address) throw new Error('No address provided');
+  const {
+    selectedMarket, assetsData,
+  } = extractedState;
+
+  let payload = {
+    ...EMPTY_COMPOUND_V3_DATA,
+    lastUpdated: Date.now(),
+  };
+
+  const contract = CompV3ViewContract(web3, network);
+  const CompV3ViewAddress = contract.options.address;
+
+  const calls = [
+    {
+      target: CompV3ViewAddress,
+      abiItem: contract.options.jsonInterface.find((props) => props.name === 'getLoanData'),
+      params: [selectedMarket.baseMarketAddress, address],
+    },
+    {
+      target: CompV3ViewAddress,
+      abiItem: contract.options.jsonInterface.find((props) => props.name === 'isAllowed'),
+      params: [selectedMarket.baseMarketAddress, address, proxyAddress || ZERO_ADDRESS],
+    },
+  ];
+
+  const data: any[] = await multicall(calls, web3, network);
+
+  const loanData = data[0][0];
+
+  const usedAssets: CompoundV3UsedAssets = {};
+
+  const baseAssetInfo = getAssetInfo(selectedMarket.baseAsset);
+  const baseAssetSymbol = wethToEth(selectedMarket.baseAsset);
+  usedAssets[baseAssetSymbol] = { ...EMPTY_USED_ASSET, symbol: baseAssetSymbol, collateral: false };
+  if (loanData.depositAmount.toString() !== '0') {
+    usedAssets[baseAssetSymbol].isSupplied = true;
+    usedAssets[baseAssetSymbol].supplied = assetAmountInEth(loanData.depositAmount, baseAssetInfo.symbol);
+    usedAssets[baseAssetSymbol].suppliedUsd = assetAmountInEth(loanData.depositValue, baseAssetInfo.symbol);
+  }
+  if (loanData.borrowAmount.toString() !== '0') {
+    usedAssets[baseAssetSymbol].isBorrowed = true;
+    usedAssets[baseAssetSymbol].borrowed = assetAmountInEth(loanData.borrowAmount, baseAssetInfo.symbol);
+    usedAssets[baseAssetSymbol].borrowedUsd = assetAmountInEth(loanData.borrowValue, baseAssetInfo.symbol);
+  }
+  loanData.collAddr.forEach((coll: string, i: number): void => {
+    const assetInfo = getAssetInfoByAddress(coll, network);
+    const symbol = wethToEth(assetInfo.symbol);
+    const supplied = assetAmountInEth(loanData.collAmounts[i].toString(), symbol);
+    const isSupplied = supplied !== '0';
+    const price = assetsData[symbol].price;
+    const suppliedUsd = new Dec(supplied).mul(price).toString();
+    usedAssets[symbol] = {
+      ...usedAssets[symbol],
+      borrowed: '0',
+      borrowedUsd: '0',
+      isSupplied,
+      supplied,
+      suppliedUsd,
+      isBorrowed: false,
+      symbol,
+      collateral: true,
+    };
+  });
+
+  payload = {
+    ...payload,
+    usedAssets,
+    ...getCompoundV3AggregatedData({
+      usedAssets, assetsData, network, selectedMarket,
+    }),
+    isAllowed: data[1][0],
+  };
+
+  // Calculate borrow limits per asset
+  Object.values(payload.usedAssets).forEach((item: any) => {
+    if (item.isBorrowed) {
+      // eslint-disable-next-line no-param-reassign
+      item.limit = calculateBorrowingAssetLimit(item.borrowedUsd, payload.borrowLimitUsd);
+    }
+  });
+
+  return payload;
+};
+
+export const getCompoundV3FullPositionData = async (web3: Web3, network: NetworkNumber, address: string, proxyAddress: string, selectedMarket: CompoundMarketData, compPrice: string, mainnetWeb3: Web3): Promise<CompoundV3PositionData> => {
+  const marketData = await getCompoundV3MarketsData(web3, network, selectedMarket, compPrice, mainnetWeb3);
+  const positionData = await getCompoundV3AccountData(web3, network, address, proxyAddress, { selectedMarket, assetsData: marketData.assetsData });
+  return positionData;
+};