@gearbox-protocol/sdk 3.0.0-next.2 → 3.0.0-next.21

package/lib/core/creditAccount.js

@@ -2,73 +2,110 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CreditAccountData = void 0;
 const sdk_gov_1 = require("@gearbox-protocol/sdk-gov");
-const apy_1 = require("../apy");
 const formatter_1 = require("../utils/formatter");
+const math_1 = require("../utils/math");
 const price_1 = require("../utils/price");
 class CreditAccountData {
+    isSuccessful;
     addr;
     borrower;
-    inUse;
     creditManager;
+    creditFacade;
     underlyingToken;
     since;
-    cumulativeIndexAtOpen;
+    expirationDate;
+    version;
+    cmDescription;
+    enabledTokenMask;
+    healthFactor;
+    isDeleting;
+    baseBorrowRate;
+    borrowRate;
     borrowedAmount;
+    accruedInterest;
+    accruedFees;
+    totalDebtUSD;
     borrowedAmountPlusInterestAndFees;
-    borrowedAmountPlusInterest;
     totalValue;
-    healthFactor;
-    borrowRate;
-    isDeleting;
-    enabledTokenMask;
-    version = 1;
-    collateralTokens = [];
-    allTokens = [];
-    forbiddenTokens = [];
+    totalValueUSD;
+    twvUSD;
+    cumulativeIndexNow;
+    cumulativeIndexLastUpdate;
+    cumulativeQuotaInterest;
+    activeBots;
+    maxApprovedBots;
     balances = {};
+    collateralTokens = [];
     allBalances = {};
-    /// V1 Artifacts
-    repayAmount;
-    liquidationAmount;
-    canBeClosed;
+    forbiddenTokens = {};
+    quotedTokens = {};
+    schedultedWithdrawals;
     constructor(payload) {
+        this.isSuccessful = payload.isSuccessful;
         this.addr = payload.addr.toLowerCase();
         this.borrower = payload.borrower.toLowerCase();
-        this.inUse = payload.inUse;
         this.creditManager = payload.creditManager.toLowerCase();
+        this.creditFacade = payload.creditFacade.toLowerCase();
         this.underlyingToken = payload.underlying.toLowerCase();
-        this.since = Number((0, formatter_1.toBigInt)(payload.since || 0));
-        this.cumulativeIndexAtOpen = (0, formatter_1.toBigInt)(payload.cumulativeIndexAtOpen || 0);
-        this.borrowedAmount = (0, formatter_1.toBigInt)(payload.borrowedAmount || 0);
-        this.borrowedAmountPlusInterest = (0, formatter_1.toBigInt)(payload.borrowedAmountPlusInterest || 0);
-        this.borrowedAmountPlusInterestAndFees = (0, formatter_1.toBigInt)(payload.borrowedAmountPlusInterestAndFees || 0);
-        this.totalValue = (0, formatter_1.toBigInt)(payload.totalValue || 0);
-        this.healthFactor = payload.healthFactor.toNumber();
-        this.borrowRate = Number(((0, formatter_1.toBigInt)(payload.borrowRate || 0) *
-            sdk_gov_1.PERCENTAGE_FACTOR *
-            sdk_gov_1.PERCENTAGE_DECIMALS) /
-            sdk_gov_1.RAY);
+        this.since = Number((0, sdk_gov_1.toBigInt)(payload.since));
+        this.expirationDate = Number((0, sdk_gov_1.toBigInt)(payload.expirationDate));
+        this.version = payload.cfVersion?.toNumber();
+        this.cmDescription = payload.cmDescription;
+        this.healthFactor = Number((0, sdk_gov_1.toBigInt)(payload.healthFactor || 0n));
+        this.enabledTokenMask = (0, sdk_gov_1.toBigInt)(payload.enabledTokensMask);
+        this.isDeleting = false;
+        this.borrowedAmount = (0, sdk_gov_1.toBigInt)(payload.debt);
+        this.accruedInterest = (0, sdk_gov_1.toBigInt)(payload.accruedInterest || 0n);
+        this.accruedFees = (0, sdk_gov_1.toBigInt)(payload.accruedFees || 0n);
+        this.borrowedAmountPlusInterestAndFees =
+            this.borrowedAmount + this.accruedInterest + this.accruedFees;
+        this.totalDebtUSD = (0, sdk_gov_1.toBigInt)(payload.totalDebtUSD);
+        this.totalValue = (0, sdk_gov_1.toBigInt)(payload.totalValue || 0n);
+        this.totalValueUSD = (0, sdk_gov_1.toBigInt)(payload.totalValueUSD);
+        this.twvUSD = (0, sdk_gov_1.toBigInt)(payload.twvUSD);
+        this.baseBorrowRate = (0, formatter_1.rayToNumber)((0, sdk_gov_1.toBigInt)(payload.baseBorrowRate) *
+            sdk_gov_1.PERCENTAGE_DECIMALS *
+            sdk_gov_1.PERCENTAGE_FACTOR);
+        this.borrowRate = (0, formatter_1.rayToNumber)((0, sdk_gov_1.toBigInt)(payload.aggregatedBorrowRate) *
+            sdk_gov_1.PERCENTAGE_DECIMALS *
+            sdk_gov_1.PERCENTAGE_FACTOR);
+        this.cumulativeIndexNow = (0, sdk_gov_1.toBigInt)(payload.cumulativeIndexNow);
+        this.cumulativeIndexLastUpdate = (0, sdk_gov_1.toBigInt)(payload.cumulativeIndexLastUpdate);
+        this.cumulativeQuotaInterest = (0, sdk_gov_1.toBigInt)(payload.cumulativeQuotaInterest);
+        this.activeBots = payload.activeBots.map(b => b.toLowerCase());
+        this.maxApprovedBots = (0, sdk_gov_1.toBigInt)(payload.maxApprovedBots);
         payload.balances.forEach(b => {
-            const tokenLC = b.token.toLowerCase();
-            if (b.isAllowed) {
-                this.balances[tokenLC] = (0, formatter_1.toBigInt)(b.balance || 0);
-                this.collateralTokens.push(tokenLC);
+            const token = b.token.toLowerCase();
+            const balance = {
+                token,
+                balance: (0, sdk_gov_1.toBigInt)(b.balance),
+                isForbidden: b.isForbidden,
+                isEnabled: b.isEnabled,
+                isQuoted: b.isQuoted,
+                quota: (0, sdk_gov_1.toBigInt)(b.quota),
+                quotaRate: b.quotaRate,
+            };
+            if (!b.isForbidden) {
+                this.balances[token] = balance.balance;
+                this.collateralTokens.push(token);
             }
-            else {
-                this.forbiddenTokens.push(tokenLC);
+            if (b.isForbidden) {
+                this.forbiddenTokens[token] = true;
             }
-            this.allBalances[tokenLC] = (0, formatter_1.toBigInt)(b.balance || 0);
-            this.allTokens.push(tokenLC);
+            if (b.isQuoted) {
+                this.quotedTokens[token] = true;
+            }
+            this.allBalances[token] = balance;
         });
-        this.enabledTokenMask = (0, formatter_1.toBigInt)(payload.enabledTokenMask || 0);
-        this.isDeleting = false;
-        this.version = payload.version;
-        this.repayAmount = (0, formatter_1.toBigInt)(payload.repayAmount || 0);
-        this.liquidationAmount = (0, formatter_1.toBigInt)(payload.liquidationAmount || 0);
-        this.canBeClosed = payload.canBeClosed;
+        this.schedultedWithdrawals = payload.schedultedWithdrawals.map(w => ({
+            tokenIndex: w.tokenIndex,
+            maturity: w.maturity,
+            token: w.token.toLowerCase(),
+            amount: (0, sdk_gov_1.toBigInt)(w.amount),
+        }));
     }
-    balancesSorted(prices, tokens) {
-        return CreditAccountData.sortBalances(this.balances, prices, tokens).map(([token, balance]) => ({ token, balance }));
+    setDeleteInProgress(d) {
+        this.isDeleting = d;
     }
     static sortBalances(balances, prices, tokens) {
         return Object.entries(balances).sort(([addr1, amount1], [addr2, amount2]) => {
@@ -96,21 +133,14 @@ class CreditAccountData {
     static amountAbcComparator(t1, t2) {
         return t1 > t2 ? -1 : 1;
     }
-    calcBorrowAmountPlusInterestRate(currentCumulativeIndex) {
-        return ((this.borrowedAmount * currentCumulativeIndex) /
-            this.cumulativeIndexAtOpen);
-    }
     isForbidden(token) {
-        return this.balances[token] === undefined;
+        return !!this.forbiddenTokens[token];
+    }
+    isQuoted(token) {
+        return !!this.quotedTokens[token];
     }
-    updateHealthFactor(creditManager, currentCumulativeIndex, priceOracle) {
-        const twvUSDValue = this.collateralTokens
-            .filter((_, num) => CreditAccountData.isTokenEnabled(num, this.enabledTokenMask))
-            .map(token => priceOracle.convertToUSD(this.balances[token], token) *
-            creditManager.liquidationThresholds[token])
-            .reduce((a, b) => a + b);
-        this.healthFactor = Number(priceOracle.convertFromUSD(twvUSDValue, this.underlyingToken) /
-            this.calcBorrowAmountPlusInterestRate(currentCumulativeIndex));
+    isTokenEnabled(index) {
+        return ((2n ** BigInt(index)) & this.enabledTokenMask) !== 0n;
     }
     static isTokenEnabled(index, enabledTokenMask) {
         return ((2n ** BigInt(index)) & enabledTokenMask) !== 0n;
@@ -118,36 +148,38 @@ class CreditAccountData {
     static calcMaxDebtIncrease(healthFactor, borrowAmountPlusInterest, underlyingLT, minHf = Number(sdk_gov_1.PERCENTAGE_FACTOR)) {
         const result = (borrowAmountPlusInterest * BigInt(healthFactor - minHf)) /
             BigInt(minHf - underlyingLT);
-        return result < 0 ? 0n : result;
+        return math_1.BigIntMath.max(0n, result);
     }
-    static calcOverallAPY({ caAssets, lpAPY, prices, totalValue, debt, borrowRate, underlyingToken, }) {
+    static calcOverallAPY({ caAssets, lpAPY, prices, quotas, quotaRates, totalValue, debt, baseBorrowRate, underlyingToken, }) {
         if (!lpAPY ||
             !totalValue ||
             totalValue <= 0n ||
             !debt ||
             totalValue <= debt)
             return undefined;
+        const underlyingTokenAddressLC = underlyingToken.toLowerCase();
+        const underlyingTokenSymbol = sdk_gov_1.tokenSymbolByAddress[underlyingTokenAddressLC] || "";
+        const underlyingTokenDecimals = sdk_gov_1.decimals[underlyingTokenSymbol] || 18;
+        const underlyingPrice = prices[underlyingTokenAddressLC];
         const assetAPYMoney = caAssets.reduce((acc, { token: tokenAddress, balance: amount }) => {
             const tokenAddressLC = tokenAddress.toLowerCase();
-            const symbol = sdk_gov_1.tokenSymbolByAddress[tokenAddressLC];
-            if (!(0, apy_1.isTokenWithAPY)(symbol))
-                return acc;
+            const symbol = sdk_gov_1.tokenSymbolByAddress[tokenAddressLC] || "";
             const apy = lpAPY[symbol] || 0;
             const price = prices[tokenAddressLC] || 0n;
             const tokenDecimals = sdk_gov_1.decimals[symbol];
             const money = price_1.PriceUtils.calcTotalPrice(price, amount, tokenDecimals);
             const apyMoney = money * BigInt(apy);
-            return acc + apyMoney;
+            const { balance: quotaAmount = 0n } = quotas[tokenAddressLC] || {};
+            const quotaMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice || 0n, quotaAmount, underlyingTokenDecimals);
+            const { rate: quotaAPY = 0 } = quotaRates[tokenAddressLC] || {};
+            const quotaAPYMoney = quotaMoney * BigInt(quotaAPY);
+            return acc + apyMoney - quotaAPYMoney;
         }, 0n);
-        const underlyingTokenAddressLC = underlyingToken.toLowerCase();
-        const underlyingTokenSymbol = sdk_gov_1.tokenSymbolByAddress[underlyingTokenAddressLC] || "";
-        const underlyingTokenDecimals = sdk_gov_1.decimals[underlyingTokenSymbol] || 18;
-        const underlyingPrice = prices[underlyingTokenAddressLC] || sdk_gov_1.PRICE_DECIMALS;
         const assetAPYAmountInUnderlying = price_1.PriceUtils.convertByPrice(assetAPYMoney, {
-            price: underlyingPrice,
+            price: underlyingPrice || sdk_gov_1.PRICE_DECIMALS,
             decimals: underlyingTokenDecimals,
         });
-        const debtAPY = debt * BigInt(borrowRate);
+        const debtAPY = debt * BigInt(baseBorrowRate);
         const yourAssets = totalValue - debt;
         const apyInPercent = ((assetAPYAmountInUnderlying - debtAPY) * sdk_gov_1.WAD) /
             yourAssets /
@@ -160,22 +192,75 @@ class CreditAccountData {
     static hash(creditManager, borrower) {
         return `${creditManager.toLowerCase()}:${borrower.toLowerCase()}`;
     }
-    static calcHealthFactor({ assets, prices, liquidationThresholds, underlyingToken, borrowed, }) {
+    static calcHealthFactor({ assets, quotas, liquidationThresholds, underlyingToken, borrowed, prices, }) {
+        const [, underlyingDecimals] = (0, sdk_gov_1.extractTokenData)(underlyingToken);
+        const underlyingPrice = prices[underlyingToken] || 0n;
         const assetLTMoney = assets.reduce((acc, { token: tokenAddress, balance: amount }) => {
-            const tokenSymbol = sdk_gov_1.tokenSymbolByAddress[tokenAddress.toLowerCase()];
-            const tokenDecimals = sdk_gov_1.decimals[tokenSymbol];
-            const lt = liquidationThresholds[tokenAddress.toLowerCase()] || 0n;
-            const price = prices[tokenAddress.toLowerCase()] || 0n;
-            const money = price_1.PriceUtils.calcTotalPrice(price, amount, tokenDecimals);
+            const [, tokenDecimals] = (0, sdk_gov_1.extractTokenData)(tokenAddress);
+            const lt = liquidationThresholds[tokenAddress] || 0n;
+            const price = prices[tokenAddress] || 0n;
+            const tokenMoney = price_1.PriceUtils.calcTotalPrice(price, amount, tokenDecimals);
+            const quota = quotas[tokenAddress];
+            const quotaMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice, quota?.balance || 0n, underlyingDecimals);
+            // if quota is undefined, then it is not a quoted token
+            const money = quota
+                ? math_1.BigIntMath.min(quotaMoney, tokenMoney)
+                : tokenMoney;
             const ltMoney = money * lt;
             return acc + ltMoney;
         }, 0n);
-        const underlyingSymbol = sdk_gov_1.tokenSymbolByAddress[underlyingToken.toLowerCase()];
-        const underlyingDecimals = sdk_gov_1.decimals[underlyingSymbol];
-        const underlyingPrice = prices[underlyingToken.toLowerCase()] || sdk_gov_1.PRICE_DECIMALS;
-        const borrowedMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice, borrowed, underlyingDecimals);
+        const borrowedMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice || sdk_gov_1.PRICE_DECIMALS, borrowed, underlyingDecimals);
         const hfInPercent = borrowedMoney > 0n ? assetLTMoney / borrowedMoney : 0n;
         return Number(hfInPercent);
     }
+    static calcQuotaUpdate({ quotas, initialQuotas, assetsAfterUpdate, allowedToSpend, allowedToObtain, }) {
+        const r = Object.values(quotas).reduce((acc, cmQuota) => {
+            const { token } = cmQuota;
+            const { quota: initialQuota = 0n } = initialQuotas[token] || {};
+            const after = assetsAfterUpdate[token];
+            const { amountInTarget = 0n } = after || {};
+            const desiredQuota = (amountInTarget * 101n) / 100n;
+            const quotaChange = desiredQuota - initialQuota;
+            const correctIncrease = after && allowedToObtain[token] && quotaChange > 0;
+            const correctDecrease = after && allowedToSpend[token] && quotaChange < 0;
+            if (correctIncrease || correctDecrease) {
+                acc.desiredQuota[token] = {
+                    balance: desiredQuota,
+                    token,
+                };
+            }
+            else {
+                acc.desiredQuota[token] = {
+                    balance: initialQuota,
+                    token,
+                };
+            }
+            if (correctIncrease) {
+                acc.quotaIncrease.push({
+                    balance: quotaChange,
+                    token,
+                });
+            }
+            if (correctDecrease) {
+                acc.quotaDecrease.push({
+                    balance: quotaChange,
+                    token,
+                });
+            }
+            return acc;
+        }, { desiredQuota: {}, quotaIncrease: [], quotaDecrease: [] });
+        return r;
+    }
+    static calcQuotaBorrowRate({ quotas, quotaRates, borrowAmount, }) {
+        if (borrowAmount <= 0)
+            return 0;
+        const totalRateBalance = Object.values(quotas).reduce((acc, { token, balance }) => {
+            const { rate = 0 } = quotaRates[token] || {};
+            const rateBalance = balance * BigInt(rate);
+            return acc + rateBalance;
+        }, 0n);
+        const quotaBorrowRate = Number(totalRateBalance / borrowAmount);
+        return quotaBorrowRate;
+    }
 }
 exports.CreditAccountData = CreditAccountData;