@gearbox-protocol/sdk 3.0.0-next.5 → 3.0.0-next.50

package/lib/core/creditAccount.js

@@ -2,10 +2,11 @@
 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;
     creditManager;
@@ -14,6 +15,7 @@ class CreditAccountData {
     since;
     expirationDate;
     version;
+    cmName;
     enabledTokenMask;
     healthFactor;
     isDeleting;
@@ -27,7 +29,6 @@ class CreditAccountData {
     totalValue;
     totalValueUSD;
     twvUSD;
-    cumulativeIndexNow;
     cumulativeIndexLastUpdate;
     cumulativeQuotaInterest;
     activeBots;
@@ -39,6 +40,7 @@ class CreditAccountData {
     quotedTokens = {};
     schedultedWithdrawals;
     constructor(payload) {
+        this.isSuccessful = payload.isSuccessful;
         this.addr = payload.addr.toLowerCase();
         this.borrower = payload.borrower.toLowerCase();
         this.creditManager = payload.creditManager.toLowerCase();
@@ -47,16 +49,17 @@ class CreditAccountData {
         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.healthFactor = Number((0, sdk_gov_1.toBigInt)(payload.healthFactor));
+        this.cmName = payload.cmName;
+        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);
-        this.accruedFees = (0, sdk_gov_1.toBigInt)(payload.accruedFees);
+        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);
+        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) *
@@ -65,7 +68,6 @@ class CreditAccountData {
         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());
@@ -79,7 +81,8 @@ class CreditAccountData {
                 isEnabled: b.isEnabled,
                 isQuoted: b.isQuoted,
                 quota: (0, sdk_gov_1.toBigInt)(b.quota),
-                quotaRate: b.quotaRate,
+                quotaRate: Number((0, sdk_gov_1.toBigInt)(b.quotaRate) * sdk_gov_1.PERCENTAGE_DECIMALS),
+                quotaCumulativeIndexLU: (0, sdk_gov_1.toBigInt)(b.quotaCumulativeIndexLU),
             };
             if (!b.isForbidden) {
                 this.balances[token] = balance.balance;
@@ -103,9 +106,6 @@ class CreditAccountData {
     setDeleteInProgress(d) {
         this.isDeleting = d;
     }
-    balancesSorted(prices, tokens) {
-        return CreditAccountData.sortBalances(this.balances, prices, tokens).map(([token, balance]) => ({ token, balance }));
-    }
     static sortBalances(balances, prices, tokens) {
         return Object.entries(balances).sort(([addr1, amount1], [addr2, amount2]) => {
             const addr1Lc = addr1.toLowerCase();
@@ -147,41 +147,42 @@ 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 { rate: quotaAPY = 0, isActive = false } = quotaRates?.[tokenAddressLC] || {};
+            const { balance: quotaBalance = 0n } = quotas[tokenAddressLC] || {};
+            const quotaAmount = isActive ? quotaBalance : 0n;
+            const quotaMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice || 0n, quotaAmount, underlyingTokenDecimals);
+            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 /
-            sdk_gov_1.PERCENTAGE_FACTOR;
-        return Number((0, formatter_1.toSignificant)(apyInPercent, sdk_gov_1.WAD_DECIMALS_POW));
+        const apyInPercent = (assetAPYAmountInUnderlying - debtAPY) / yourAssets;
+        return apyInPercent;
     }
     hash() {
         return CreditAccountData.hash(this.creditManager, this.borrower);
@@ -189,22 +190,85 @@ class CreditAccountData {
     static hash(creditManager, borrower) {
         return `${creditManager.toLowerCase()}:${borrower.toLowerCase()}`;
     }
-    static calcHealthFactor({ assets, prices, liquidationThresholds, underlyingToken, borrowed, }) {
+    static calcHealthFactor({ assets, quotas, quotasInfo, 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 { isActive = false } = quotasInfo?.[tokenAddress] || {};
+            const quota = quotas[tokenAddress];
+            const quotaBalance = isActive ? quota?.balance || 0n : 0n;
+            const quotaMoney = price_1.PriceUtils.calcTotalPrice(underlyingPrice, quotaBalance, 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, isActive } = cmQuota;
+            const { quota: initialQuota = 0n } = initialQuotas[token] || {};
+            if (!isActive) {
+                acc.desiredQuota[token] = {
+                    balance: initialQuota,
+                    token,
+                };
+                return acc;
+            }
+            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, isActive = false } = quotaRates?.[token] || {};
+            const quotaBalance = isActive ? balance : 0n;
+            const rateBalance = quotaBalance * BigInt(rate);
+            return acc + rateBalance;
+        }, 0n);
+        const quotaBorrowRate = Number(totalRateBalance / borrowAmount);
+        return quotaBorrowRate;
+    }
 }
 exports.CreditAccountData = CreditAccountData;