@defisaver/positions-sdk 0.0.20 → 0.0.22

package/src/aaveV2/index.ts

@@ -1,221 +1,227 @@
-import Web3 from 'web3';
-import Dec from 'decimal.js';
-import { assetAmountInEth, getAssetInfo } from '@defisaver/tokens';
-import * as net from 'net';
-import {
-  Blockish, EthAddress, NetworkNumber, PositionBalances,
-} from '../types/common';
-import { calculateNetApy, getStETHApr } from '../staking';
-import { ethToWeth, wethToEth } from '../services/utils';
-import { AaveLoanInfoV2Contract } from '../contracts';
-import { calculateBorrowingAssetLimit } from '../moneymarket';
-import {
-  AaveMarketInfo, AaveV2AssetData, AaveV2AssetsData, AaveV2PositionData, AaveV2UsedAsset, AaveV2UsedAssets,
-} from '../types';
-import { EMPTY_AAVE_DATA } from '../aaveV3';
-import { AAVE_V2 } from '../markets/aave';
-import { aaveAnyGetAggregatedPositionData } from '../helpers/aaveHelpers';
-import { getEthPrice } from '../services/priceService';
-
-export const getAaveV2MarketsData = async (web3: Web3, network: NetworkNumber, selectedMarket: AaveMarketInfo, mainnetWeb3: Web3) => {
-  const ethPrice = await getEthPrice(mainnetWeb3);
-  const _addresses = selectedMarket.assets.map(a => getAssetInfo(ethToWeth(a)).address);
-  const loanInfoContract = AaveLoanInfoV2Contract(web3, network);
-  const marketAddress = selectedMarket.providerAddress;
-  const loanInfo = await loanInfoContract.methods.getFullTokensInfo(marketAddress, _addresses).call();
-  const markets = loanInfo
-    .map((market, i) => ({
-      symbol: selectedMarket.assets[i],
-      underlyingTokenAddress: market.underlyingTokenAddress,
-      supplyRate: new Dec(market.supplyRate.toString()).div(1e25).toString(),
-      borrowRate: new Dec(market.borrowRateVariable.toString()).div(1e25).toString(),
-      borrowRateStable: new Dec(market.borrowRateStable.toString()).div(1e25).toString(),
-      collateralFactor: new Dec(market.collateralFactor.toString()).div(10000).toString(),
-      liquidationRatio: new Dec(market.liquidationRatio.toString()).div(10000).toString(),
-      marketLiquidity: assetAmountInEth(new Dec(market.totalSupply.toString())
-        .sub(market.totalBorrow.toString())
-        .toString(), selectedMarket.assets[i]),
-      utilization: new Dec(market.totalBorrow.toString())
-        .div(new Dec(market.totalSupply.toString()))
-        .times(100)
-        .toString(),
-      usageAsCollateralEnabled: market.usageAsCollateralEnabled,
-      supplyCap: '0',
-      borrowCap: '0', // v2 doesnt have borrow cap but adding it for compatability with v3
-      totalSupply: assetAmountInEth(market.totalSupply.toString(), selectedMarket.assets[i]),
-      isInactive: !market.isActive,
-      isFrozen: market.isFrozen,
-      canBeBorrowed: market.isActive && market.borrowingEnabled && !market.isFrozen,
-      canBeSupplied: market.isActive && !market.isFrozen,
-      canBeWithdrawn: market.isActive,
-      canBePayBacked: market.isActive,
-      disabledStableBorrowing: !market.stableBorrowRateEnabled,
-      totalBorrow: assetAmountInEth(market.totalBorrow.toString(), selectedMarket.assets[i]),
-      totalBorrowVar: assetAmountInEth(market.totalBorrowVar.toString(), selectedMarket.assets[i]),
-      priceInEth: new Dec(market.price.toString()).div(1e18).toString(),
-      incentiveSupplyToken: 'AAVE',
-      incentiveBorrowToken: 'AAVE',
-      incentiveSupplyApy: '0',
-      price: new Dec(market.price.toString()).div(1e18).mul(ethPrice).toString(),
-    }));
-
-  const stEthMarket = markets.find(({ symbol }) => symbol === 'stETH');
-  if (stEthMarket) {
-    stEthMarket.incentiveSupplyApy = await getStETHApr(mainnetWeb3);
-    stEthMarket.incentiveSupplyToken = 'stETH';
-  }
-
-  const payload: AaveV2AssetsData = {};
-  // Sort by market size
-  markets
-    .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((assetData: AaveV2AssetData, i) => {
-      payload[assetData.symbol] = { ...assetData, sortIndex: i };
-    });
-
-  return { assetsData: payload };
-};
-
-export const getAaveV2AccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress): Promise<PositionBalances> => {
-  let balances: PositionBalances = {
-    collateral: {},
-    debt: {},
-  };
-
-  if (!address) {
-    return balances;
-  }
-
-  const market = AAVE_V2;
-
-  const loanInfoContract = AaveLoanInfoV2Contract(web3, network, block);
-
-  const marketAddress = market.providerAddress;
-  const _addresses = market.assets.map(a => getAssetInfo(ethToWeth(a), network).address);
-  const loanInfo = await loanInfoContract.methods.getTokenBalances(marketAddress, address, _addresses).call({}, block);
-
-  loanInfo.forEach((_tokenInfo: any, i: number) => {
-    const asset = wethToEth(market.assets[i]);
-    const tokenInfo = { ..._tokenInfo };
-
-    // known bug: stETH leaves 1 wei on every transfer
-    if (asset === 'stETH' && tokenInfo.balance.toString() === '1') {
-      tokenInfo.balance = '0';
-    }
-
-    balances = {
-      collateral: {
-        ...balances.collateral,
-        [addressMapping ? getAssetInfo(asset, network).address.toLowerCase() : asset]: tokenInfo.balance.toString(),
-      },
-      debt: {
-        ...balances.debt,
-        [addressMapping ? getAssetInfo(asset, network).address.toLowerCase() : asset]: new Dec(tokenInfo.borrowsStable.toString()).add(tokenInfo.borrowsVariable.toString()).toString(),
-      },
-    };
-  });
-
-  return balances;
-};
-
-export const getAaveV2AccountData = async (web3: Web3, network: NetworkNumber, address: string, assetsData: AaveV2AssetsData, market: AaveMarketInfo): Promise<AaveV2PositionData> => {
-  if (!address) throw new Error('Address is required');
-
-  let payload: AaveV2PositionData = {
-    ...EMPTY_AAVE_DATA,
-    lastUpdated: Date.now(),
-  };
-
-  const loanInfoContract = AaveLoanInfoV2Contract(web3, network);
-  const marketAddress = market.providerAddress;
-  const _addresses = market.assets.map(a => getAssetInfo(ethToWeth(a)).address);
-  const loanInfo = await loanInfoContract.methods.getTokenBalances(marketAddress, address, _addresses).call();
-  const usedAssets = {} as AaveV2UsedAssets;
-  loanInfo.forEach((_tokenInfo, i) => {
-    const asset = market.assets[i];
-    const tokenInfo = { ..._tokenInfo };
-
-    // known bug: stETH leaves 1 wei on every transfer
-    if (asset === 'stETH' && tokenInfo.balance.toString() === '1') tokenInfo.balance = '0';
-
-    const isSupplied = tokenInfo.balance.toString() !== '0';
-    const isBorrowed = tokenInfo.borrowsStable.toString() !== '0' || tokenInfo.borrowsVariable.toString() !== '0';
-    if (!isSupplied && !isBorrowed) return;
-
-    const supplied = assetAmountInEth(tokenInfo.balance.toString(), asset);
-    const borrowedStable = assetAmountInEth(tokenInfo.borrowsStable.toString(), asset);
-    const borrowedVariable = assetAmountInEth(tokenInfo.borrowsVariable.toString(), asset);
-    const enabledAsCollateral = assetsData[asset].usageAsCollateralEnabled ? tokenInfo.enabledAsCollateral : false;
-    let interestMode;
-    if (borrowedVariable === '0' && borrowedStable !== '0') {
-      interestMode = '1';
-    } else if (borrowedVariable !== '0' && borrowedStable === '0') {
-      interestMode = '2';
-    } else {
-      interestMode = 'both';
-    }
-    if (!usedAssets[asset]) usedAssets[asset] = {} as AaveV2UsedAsset;
-    usedAssets[asset] = {
-      ...usedAssets[asset],
-      symbol: asset,
-      supplied,
-      suppliedUsd: new Dec(supplied).mul(assetsData[asset].price).toString(),
-      isSupplied,
-      collateral: enabledAsCollateral,
-      stableBorrowRate: new Dec(tokenInfo.stableBorrowRate).div(1e25).toString(),
-      borrowedStable,
-      borrowedVariable,
-      borrowedUsdStable: new Dec(borrowedStable).mul(assetsData[asset].price).toString(),
-      borrowedUsdVariable: new Dec(borrowedVariable).mul(assetsData[asset].price).toString(),
-      borrowed: new Dec(borrowedStable).add(borrowedVariable).toString(),
-      borrowedUsd: new Dec(new Dec(borrowedVariable).add(borrowedStable)).mul(assetsData[asset].price).toString(),
-      isBorrowed,
-      interestMode,
-    };
-  });
-  payload = {
-    ...payload,
-    usedAssets,
-    ...aaveAnyGetAggregatedPositionData({
-      usedAssets, assetsData, eModeCategory: 0, selectedMarket: market,
-    }),
-  };
-
-  payload.ratio = payload.borrowedUsd && payload.borrowedUsd !== '0'
-    ? new Dec(payload.borrowLimitUsd).div(payload.borrowedUsd).mul(100).toString()
-    : '0';
-  payload.minRatio = '100';
-  payload.collRatio = payload.borrowedUsd && payload.borrowedUsd !== '0'
-    ? new Dec(payload.suppliedCollateralUsd).div(payload.borrowedUsd).mul(100).toString()
-    : '0';
-
-  // Calculate borrow limits per asset
-  Object.values(payload.usedAssets).forEach((item) => {
-    if (item.isBorrowed) {
-      // eslint-disable-next-line no-param-reassign
-      item.stableLimit = calculateBorrowingAssetLimit(item.borrowedUsdStable, payload.borrowLimitUsd);
-      // eslint-disable-next-line no-param-reassign
-      item.variableLimit = calculateBorrowingAssetLimit(item.borrowedUsdVariable, payload.borrowLimitUsd);
-      // eslint-disable-next-line no-param-reassign
-      item.limit = calculateBorrowingAssetLimit(item.borrowedUsd, payload.borrowLimitUsd);
-    }
-  });
-
-  const { netApy, incentiveUsd, totalInterestUsd } = calculateNetApy(usedAssets, assetsData);
-  payload.netApy = netApy;
-  payload.incentiveUsd = incentiveUsd;
-  payload.totalInterestUsd = totalInterestUsd;
-
-  return payload;
-};
-
-export const getAaveV2FullPositionData = async (web3: Web3, network: NetworkNumber, address: string, market: AaveMarketInfo, mainnetWeb3: Web3): Promise<AaveV2PositionData> => {
-  const marketData = await getAaveV2MarketsData(web3, network, market, mainnetWeb3);
-  const positionData = await getAaveV2AccountData(web3, network, address, marketData.assetsData, market);
-  return positionData;
+import Web3 from 'web3';
+import Dec from 'decimal.js';
+import { assetAmountInEth, getAssetInfo } from '@defisaver/tokens';
+import {
+  Blockish, EthAddress, NetworkNumber, PositionBalances,
+} from '../types/common';
+import { calculateNetApy, getStETHApr } from '../staking';
+import { ethToWeth, wethToEth, wethToEthByAddress } from '../services/utils';
+import { AaveLoanInfoV2Contract, createContractWrapper } from '../contracts';
+import { calculateBorrowingAssetLimit } from '../moneymarket';
+import {
+  AaveMarketInfo, AaveV2AssetData, AaveV2AssetsData, AaveV2PositionData, AaveV2UsedAsset, AaveV2UsedAssets,
+} from '../types';
+import { EMPTY_AAVE_DATA } from '../aaveV3';
+import { AAVE_V2 } from '../markets/aave';
+import { aaveAnyGetAggregatedPositionData } from '../helpers/aaveHelpers';
+import { getEthPrice } from '../services/priceService';
+
+export const getAaveV2MarketsData = async (web3: Web3, network: NetworkNumber, selectedMarket: AaveMarketInfo, mainnetWeb3: Web3) => {
+  const ethPrice = await getEthPrice(mainnetWeb3);
+  const _addresses = selectedMarket.assets.map(a => getAssetInfo(ethToWeth(a)).address);
+  const loanInfoContract = AaveLoanInfoV2Contract(web3, network);
+  const marketAddress = selectedMarket.providerAddress;
+  const loanInfo = await loanInfoContract.methods.getFullTokensInfo(marketAddress, _addresses).call();
+  const markets = loanInfo
+    .map((market, i) => ({
+      symbol: selectedMarket.assets[i],
+      underlyingTokenAddress: market.underlyingTokenAddress,
+      supplyRate: new Dec(market.supplyRate.toString()).div(1e25).toString(),
+      borrowRate: new Dec(market.borrowRateVariable.toString()).div(1e25).toString(),
+      borrowRateStable: new Dec(market.borrowRateStable.toString()).div(1e25).toString(),
+      collateralFactor: new Dec(market.collateralFactor.toString()).div(10000).toString(),
+      liquidationRatio: new Dec(market.liquidationRatio.toString()).div(10000).toString(),
+      marketLiquidity: assetAmountInEth(new Dec(market.totalSupply.toString())
+        .sub(market.totalBorrow.toString())
+        .toString(), selectedMarket.assets[i]),
+      utilization: new Dec(market.totalBorrow.toString())
+        .div(new Dec(market.totalSupply.toString()))
+        .times(100)
+        .toString(),
+      usageAsCollateralEnabled: market.usageAsCollateralEnabled,
+      supplyCap: '0',
+      borrowCap: '0', // v2 doesnt have borrow cap but adding it for compatability with v3
+      totalSupply: assetAmountInEth(market.totalSupply.toString(), selectedMarket.assets[i]),
+      isInactive: !market.isActive,
+      isFrozen: market.isFrozen,
+      canBeBorrowed: market.isActive && market.borrowingEnabled && !market.isFrozen,
+      canBeSupplied: market.isActive && !market.isFrozen,
+      canBeWithdrawn: market.isActive,
+      canBePayBacked: market.isActive,
+      disabledStableBorrowing: !market.stableBorrowRateEnabled,
+      totalBorrow: assetAmountInEth(market.totalBorrow.toString(), selectedMarket.assets[i]),
+      totalBorrowVar: assetAmountInEth(market.totalBorrowVar.toString(), selectedMarket.assets[i]),
+      priceInEth: new Dec(market.price.toString()).div(1e18).toString(),
+      incentiveSupplyToken: 'AAVE',
+      incentiveBorrowToken: 'AAVE',
+      incentiveSupplyApy: '0',
+      price: new Dec(market.price.toString()).div(1e18).mul(ethPrice).toString(),
+    }));
+
+  const stEthMarket = markets.find(({ symbol }) => symbol === 'stETH');
+  if (stEthMarket) {
+    stEthMarket.incentiveSupplyApy = await getStETHApr(mainnetWeb3);
+    stEthMarket.incentiveSupplyToken = 'stETH';
+  }
+
+  const payload: AaveV2AssetsData = {};
+  // Sort by market size
+  markets
+    .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((assetData: AaveV2AssetData, i) => {
+      payload[assetData.symbol] = { ...assetData, sortIndex: i };
+    });
+
+  return { assetsData: payload };
+};
+
+export const getAaveV2AccountBalances = async (web3: Web3, network: NetworkNumber, block: Blockish, addressMapping: boolean, address: EthAddress): Promise<PositionBalances> => {
+  let balances: PositionBalances = {
+    collateral: {},
+    debt: {},
+  };
+
+  if (!address) {
+    return balances;
+  }
+
+  const market = AAVE_V2;
+
+  const loanInfoContract = AaveLoanInfoV2Contract(web3, network, block);
+
+  const marketAddress = market.providerAddress;
+  // @ts-ignore
+  const protocolDataProviderContract = createContractWrapper(web3, network, market.protocolData, market.protocolDataAddress);
+
+  const reserveTokens = await protocolDataProviderContract.methods.getAllReservesTokens().call({}, block);
+  const symbols = reserveTokens.map(({ symbol }: { symbol: string }) => symbol);
+  const _addresses = reserveTokens.map(({ tokenAddress }: { tokenAddress: EthAddress }) => tokenAddress);
+
+  const loanInfo = await loanInfoContract.methods.getTokenBalances(marketAddress, address, _addresses).call({}, block);
+
+  loanInfo.forEach((_tokenInfo: any, i: number) => {
+    const asset = wethToEth(symbols[i]);
+    const assetAddr = wethToEthByAddress(_addresses[i], network).toLowerCase();
+    const tokenInfo = { ..._tokenInfo };
+
+    // known bug: stETH leaves 1 wei on every transfer
+    if (asset === 'stETH' && tokenInfo.balance.toString() === '1') {
+      tokenInfo.balance = '0';
+    }
+
+    balances = {
+      collateral: {
+        ...balances.collateral,
+        [addressMapping ? assetAddr : asset]: tokenInfo.balance.toString(),
+      },
+      debt: {
+        ...balances.debt,
+        [addressMapping ? assetAddr : asset]: new Dec(tokenInfo.borrowsStable.toString()).add(tokenInfo.borrowsVariable.toString()).toString(),
+      },
+    };
+  });
+
+  return balances;
+};
+
+export const getAaveV2AccountData = async (web3: Web3, network: NetworkNumber, address: string, assetsData: AaveV2AssetsData, market: AaveMarketInfo): Promise<AaveV2PositionData> => {
+  if (!address) throw new Error('Address is required');
+
+  let payload: AaveV2PositionData = {
+    ...EMPTY_AAVE_DATA,
+    lastUpdated: Date.now(),
+  };
+
+  const loanInfoContract = AaveLoanInfoV2Contract(web3, network);
+  const marketAddress = market.providerAddress;
+  const _addresses = market.assets.map(a => getAssetInfo(ethToWeth(a)).address);
+  const loanInfo = await loanInfoContract.methods.getTokenBalances(marketAddress, address, _addresses).call();
+  const usedAssets = {} as AaveV2UsedAssets;
+  loanInfo.forEach((_tokenInfo, i) => {
+    const asset = market.assets[i];
+    const tokenInfo = { ..._tokenInfo };
+
+    // known bug: stETH leaves 1 wei on every transfer
+    if (asset === 'stETH' && tokenInfo.balance.toString() === '1') tokenInfo.balance = '0';
+
+    const isSupplied = tokenInfo.balance.toString() !== '0';
+    const isBorrowed = tokenInfo.borrowsStable.toString() !== '0' || tokenInfo.borrowsVariable.toString() !== '0';
+    if (!isSupplied && !isBorrowed) return;
+
+    const supplied = assetAmountInEth(tokenInfo.balance.toString(), asset);
+    const borrowedStable = assetAmountInEth(tokenInfo.borrowsStable.toString(), asset);
+    const borrowedVariable = assetAmountInEth(tokenInfo.borrowsVariable.toString(), asset);
+    const enabledAsCollateral = assetsData[asset].usageAsCollateralEnabled ? tokenInfo.enabledAsCollateral : false;
+    let interestMode;
+    if (borrowedVariable === '0' && borrowedStable !== '0') {
+      interestMode = '1';
+    } else if (borrowedVariable !== '0' && borrowedStable === '0') {
+      interestMode = '2';
+    } else {
+      interestMode = 'both';
+    }
+    if (!usedAssets[asset]) usedAssets[asset] = {} as AaveV2UsedAsset;
+    usedAssets[asset] = {
+      ...usedAssets[asset],
+      symbol: asset,
+      supplied,
+      suppliedUsd: new Dec(supplied).mul(assetsData[asset].price).toString(),
+      isSupplied,
+      collateral: enabledAsCollateral,
+      stableBorrowRate: new Dec(tokenInfo.stableBorrowRate).div(1e25).toString(),
+      borrowedStable,
+      borrowedVariable,
+      borrowedUsdStable: new Dec(borrowedStable).mul(assetsData[asset].price).toString(),
+      borrowedUsdVariable: new Dec(borrowedVariable).mul(assetsData[asset].price).toString(),
+      borrowed: new Dec(borrowedStable).add(borrowedVariable).toString(),
+      borrowedUsd: new Dec(new Dec(borrowedVariable).add(borrowedStable)).mul(assetsData[asset].price).toString(),
+      isBorrowed,
+      interestMode,
+    };
+  });
+  payload = {
+    ...payload,
+    usedAssets,
+    ...aaveAnyGetAggregatedPositionData({
+      usedAssets, assetsData, eModeCategory: 0, selectedMarket: market,
+    }),
+  };
+
+  payload.ratio = payload.borrowedUsd && payload.borrowedUsd !== '0'
+    ? new Dec(payload.borrowLimitUsd).div(payload.borrowedUsd).mul(100).toString()
+    : '0';
+  payload.minRatio = '100';
+  payload.collRatio = payload.borrowedUsd && payload.borrowedUsd !== '0'
+    ? new Dec(payload.suppliedCollateralUsd).div(payload.borrowedUsd).mul(100).toString()
+    : '0';
+
+  // Calculate borrow limits per asset
+  Object.values(payload.usedAssets).forEach((item) => {
+    if (item.isBorrowed) {
+      // eslint-disable-next-line no-param-reassign
+      item.stableLimit = calculateBorrowingAssetLimit(item.borrowedUsdStable, payload.borrowLimitUsd);
+      // eslint-disable-next-line no-param-reassign
+      item.variableLimit = calculateBorrowingAssetLimit(item.borrowedUsdVariable, payload.borrowLimitUsd);
+      // eslint-disable-next-line no-param-reassign
+      item.limit = calculateBorrowingAssetLimit(item.borrowedUsd, payload.borrowLimitUsd);
+    }
+  });
+
+  const { netApy, incentiveUsd, totalInterestUsd } = calculateNetApy(usedAssets, assetsData);
+  payload.netApy = netApy;
+  payload.incentiveUsd = incentiveUsd;
+  payload.totalInterestUsd = totalInterestUsd;
+
+  return payload;
+};
+
+export const getAaveV2FullPositionData = async (web3: Web3, network: NetworkNumber, address: string, market: AaveMarketInfo, mainnetWeb3: Web3): Promise<AaveV2PositionData> => {
+  const marketData = await getAaveV2MarketsData(web3, network, market, mainnetWeb3);
+  const positionData = await getAaveV2AccountData(web3, network, address, marketData.assetsData, market);
+  return positionData;
 };