@evaafi/sdk 0.5.0 → 0.5.1

package/dist/api/math.d.ts

@@ -1,6 +1,6 @@
 import { AssetConfig, AssetData, AssetInterest, ExtendedAssetData } from '../types/Master';
 import { Dictionary } from '@ton/core';
-import { LiquidationData, UserBalance } from '../types/User';
+import { LiquidationData, PredictHealthFactorArgs, UserBalance } from '../types/User';
 export declare function mulFactor(decimal: bigint, a: bigint, b: bigint): bigint;
 export declare function mulDiv(x: bigint, y: bigint, z: bigint): bigint;
 export declare function bigIntMax(...args: bigint[]): bigint;
@@ -18,3 +18,4 @@ export declare function calculateAssetInterest(assetConfig: AssetConfig, assetDa
 export declare function getAvailableToBorrow(assetsConfig: Dictionary<bigint, AssetConfig>, assetsData: Dictionary<bigint, ExtendedAssetData>, principals: Dictionary<bigint, bigint>, prices: Dictionary<bigint, bigint>): bigint;
 export declare function presentValue(sRate: bigint, bRate: bigint, principalValue: bigint): UserBalance;
 export declare function calculateLiquidationData(assetsConfig: Dictionary<bigint, AssetConfig>, assetsData: Dictionary<bigint, ExtendedAssetData>, principals: Dictionary<bigint, bigint>, prices: Dictionary<bigint, bigint>): LiquidationData;
+export declare function predictHealthFactor(args: PredictHealthFactorArgs): number;

package/dist/api/math.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateLiquidationData = exports.presentValue = exports.getAvailableToBorrow = exports.calculateAssetInterest = exports.calculateAssetData = exports.calculateCurrentRates = exports.calculatePresentValue = exports.bigIntMin = exports.bigIntMax = exports.mulDiv = exports.mulFactor = void 0;
+exports.predictHealthFactor = exports.calculateLiquidationData = exports.presentValue = exports.getAvailableToBorrow = exports.calculateAssetInterest = exports.calculateAssetData = exports.calculateCurrentRates = exports.calculatePresentValue = exports.bigIntMin = exports.bigIntMax = exports.mulDiv = exports.mulFactor = void 0;
 const constants_1 = require("../constants");
 const User_1 = require("../types/User");
+const sha256BigInt_1 = require("../utils/sha256BigInt");
 function mulFactor(decimal, a, b) {
     return (a * b) / decimal;
 }
@@ -210,3 +211,34 @@ function calculateLiquidationData(assetsConfig, assetsData, principals, prices)
     };
 }
 exports.calculateLiquidationData = calculateLiquidationData;
+function predictHealthFactor(args) {
+    const liquidationData = calculateLiquidationData(args.assetsConfig, args.assetsData, args.balances, args.prices);
+    const tokenHash = (0, sha256BigInt_1.sha256Hash)(args.tokenSymbol);
+    const assetConfig = args.assetsConfig.get(tokenHash);
+    const assetPrice = Number(args.prices.get(tokenHash));
+    let totalLimit = Number(liquidationData.totalLimit);
+    let totalBorrow = Number(liquidationData.totalDebt);
+    const currentAmount = args.amount;
+    const decimals = Number(assetConfig.decimals);
+    const currentBalance = assetPrice * Number(currentAmount) / Math.pow(10, decimals);
+    const changeType = args.balanceChangeType;
+    if (currentAmount != null && currentAmount != 0n) {
+        if (changeType == User_1.BalanceChangeType.Borrow) {
+            totalBorrow += currentBalance * (1 + Number(assetConfig.originationFee) / Number(constants_1.MASTER_CONSTANTS.ASSET_ORIGINATION_FEE_SCALE));
+        }
+        else if (changeType == User_1.BalanceChangeType.Repay) {
+            totalBorrow -= currentBalance;
+        }
+        else if (changeType == User_1.BalanceChangeType.Withdraw) {
+            totalLimit -= currentBalance * Number(assetConfig.liquidationThreshold) / Number(constants_1.MASTER_CONSTANTS.ASSET_COEFFICIENT_SCALE);
+        }
+        else if (changeType == User_1.BalanceChangeType.Supply) {
+            totalLimit += currentBalance * Number(assetConfig.liquidationThreshold) / Number(constants_1.MASTER_CONSTANTS.ASSET_COEFFICIENT_SCALE);
+        }
+    }
+    if (Number(totalLimit) == 0) {
+        return 1;
+    }
+    return Math.min(Math.max(1 - totalBorrow / totalLimit, 0), 1); // let's limit a result to zero below and one above
+}
+exports.predictHealthFactor = predictHealthFactor;

package/dist/constants.d.ts

@@ -36,6 +36,7 @@ export declare const MASTER_CONSTANTS: {
     ASSET_PRICE_SCALE: bigint;
     ASSET_RESERVE_FACTOR_SCALE: bigint;
     ASSET_LIQUIDATION_RESERVE_FACTOR_SCALE: bigint;
+    ASSET_ORIGINATION_FEE_SCALE: bigint;
 };
 export declare const LENDING_CODE: Cell;
 export declare const JETTON_WALLETS_CODE: {

package/dist/constants.js

@@ -42,6 +42,7 @@ exports.MASTER_CONSTANTS = {
     ASSET_PRICE_SCALE: BigInt(1e8),
     ASSET_RESERVE_FACTOR_SCALE: 10000n,
     ASSET_LIQUIDATION_RESERVE_FACTOR_SCALE: 10000n,
+    ASSET_ORIGINATION_FEE_SCALE: BigInt(1e9)
 };
 exports.LENDING_CODE = core_1.Cell.fromBoc(Buffer.from('b5ee9c72c1010e0100fd000d12182a555a6065717691969efd0114ff00f4a413f4bcf2c80b010202c8050202039f740403001ff2f8276a2687d2018fd201800f883b840051d38642c678b64e4400780e58fc10802faf07f80e59fa801e78b096664c02078067c07c100627a7978402014807060007a0ddb0c60201c709080013a0fd007a026900aa90400201200b0a0031b8e1002191960aa00b9e2ca007f4042796d225e8019203f6010201200d0c000bf7c147d2218400b9d10e86981fd201840b07f8138d809797976a2687d2029116382f970fd9178089910374daf81b619fd20182c7883b8701981684100627910eba56001797a6a6ba610fd8200e8768f76a9f6aa00cc2a32a8292878809bef2f1889f883bbcdeb86f01', 'hex'))[0];
 exports.JETTON_WALLETS_CODE = {

package/dist/types/User.d.ts

@@ -1,4 +1,5 @@
 import { Address, Cell, Dictionary } from '@ton/core';
+import { AssetConfig, ExtendedAssetData } from './Master';
 export declare enum BalanceType {
     supply = "supply",
     borrow = "borrow"
@@ -57,3 +58,18 @@ export type UserRewards = {
     trackingIndex: bigint;
     trackingAccured: bigint;
 };
+export declare enum BalanceChangeType {
+    Borrow = 0,
+    Repay = 1,
+    Supply = 2,
+    Withdraw = 3
+}
+export type PredictHealthFactorArgs = {
+    balanceChangeType: BalanceChangeType;
+    amount: bigint;
+    tokenSymbol: string;
+    balances: Dictionary<bigint, bigint>;
+    prices: Dictionary<bigint, bigint>;
+    assetsData: Dictionary<bigint, ExtendedAssetData>;
+    assetsConfig: Dictionary<bigint, AssetConfig>;
+};

package/dist/types/User.js

@@ -1,8 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BalanceType = void 0;
+exports.BalanceChangeType = exports.BalanceType = void 0;
 var BalanceType;
 (function (BalanceType) {
     BalanceType["supply"] = "supply";
     BalanceType["borrow"] = "borrow";
 })(BalanceType || (exports.BalanceType = BalanceType = {}));
+var BalanceChangeType;
+(function (BalanceChangeType) {
+    BalanceChangeType[BalanceChangeType["Borrow"] = 0] = "Borrow";
+    BalanceChangeType[BalanceChangeType["Repay"] = 1] = "Repay";
+    BalanceChangeType[BalanceChangeType["Supply"] = 2] = "Supply";
+    BalanceChangeType[BalanceChangeType["Withdraw"] = 3] = "Withdraw";
+})(BalanceChangeType || (exports.BalanceChangeType = BalanceChangeType = {}));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@evaafi/sdk",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "SDK for EVAA contracts",
   "main": "dist/index.js",
   "files": [

package/src/api/math.ts

@@ -1,7 +1,8 @@
 import { AssetConfig, AssetData, AssetInterest, ExtendedAssetData } from '../types/Master';
 import { MASTER_CONSTANTS } from '../constants';
 import { Dictionary } from '@ton/core';
-import { BalanceType, LiquidationData, UserBalance } from '../types/User';
+import { BalanceChangeType, BalanceType, LiquidationData, PredictHealthFactorArgs, UserBalance } from '../types/User';
+import { sha256Hash } from '../utils/sha256BigInt';
 
 export function mulFactor(decimal: bigint, a: bigint, b: bigint): bigint {
     return (a * b) / decimal;
@@ -248,3 +249,38 @@ export function calculateLiquidationData(
         liquidable: false,
     };
 }
+
+export function predictHealthFactor(args: PredictHealthFactorArgs): number {
+    const liquidationData = calculateLiquidationData(args.assetsConfig, args.assetsData, args.balances, args.prices);
+    const tokenHash = sha256Hash(args.tokenSymbol);
+    
+    const assetConfig = args.assetsConfig.get(tokenHash)!;
+    const assetPrice = Number(args.prices.get(tokenHash)!);
+   
+    let totalLimit = Number(liquidationData.totalLimit);
+    let totalBorrow = Number(liquidationData.totalDebt);
+
+    const currentAmount = args.amount;
+
+    const decimals = Number(assetConfig.decimals)
+
+    const currentBalance = assetPrice * Number(currentAmount) / Math.pow(10, decimals);
+    const changeType = args.balanceChangeType;
+
+    if (currentAmount != null && currentAmount != 0n) { 
+        if (changeType == BalanceChangeType.Borrow) {
+            totalBorrow += currentBalance * (1 + Number(assetConfig.originationFee) / Number(MASTER_CONSTANTS.ASSET_ORIGINATION_FEE_SCALE));
+        } else if (changeType == BalanceChangeType.Repay) {
+            totalBorrow -= currentBalance;
+        } else if (changeType == BalanceChangeType.Withdraw) {
+            totalLimit -= currentBalance * Number(assetConfig.liquidationThreshold) / Number(MASTER_CONSTANTS.ASSET_COEFFICIENT_SCALE);
+        } else if (changeType == BalanceChangeType.Supply) {
+            totalLimit += currentBalance * Number(assetConfig.liquidationThreshold) / Number(MASTER_CONSTANTS.ASSET_COEFFICIENT_SCALE);
+        }
+    }
+    if (Number(totalLimit) == 0) {
+        return 1;
+    }
+
+    return Math.min(Math.max(1 - totalBorrow / totalLimit, 0), 1);  // let's limit a result to zero below and one above
+}

package/src/constants.ts

@@ -45,6 +45,7 @@ export const MASTER_CONSTANTS = {
     ASSET_PRICE_SCALE: BigInt(1e8),
     ASSET_RESERVE_FACTOR_SCALE: 10000n,
     ASSET_LIQUIDATION_RESERVE_FACTOR_SCALE: 10000n,
+    ASSET_ORIGINATION_FEE_SCALE: BigInt(1e9)
 };
 
 export const LENDING_CODE = Cell.fromBoc(

package/src/types/User.ts

@@ -1,4 +1,5 @@
 import { Address, Cell, Dictionary } from '@ton/core';
+import { AssetConfig, ExtendedAssetData } from './Master';
 
 export enum BalanceType {
     supply = 'supply',
@@ -73,3 +74,20 @@ export type UserRewards = {
     trackingIndex: bigint;
     trackingAccured: bigint;
 };
+
+export enum BalanceChangeType {
+    Borrow = 0,
+    Repay = 1,
+    Supply = 2,
+    Withdraw = 3
+}
+
+export type PredictHealthFactorArgs = {
+    balanceChangeType: BalanceChangeType;
+    amount: bigint;  // always positive
+    tokenSymbol: string;
+    balances: Dictionary<bigint, bigint>;
+    prices: Dictionary<bigint, bigint>;
+    assetsData: Dictionary<bigint, ExtendedAssetData>;
+    assetsConfig: Dictionary<bigint, AssetConfig>;
+};