@drift-labs/sdk 2.92.0-beta.2 → 2.92.0-beta.3

package/VERSION

@@ -1 +1 @@
-2.92.0-beta.2
+2.92.0-beta.3

package/lib/adminClient.d.ts

@@ -21,6 +21,8 @@ export declare class AdminClient extends DriftClient {
     getInitializeOpenbookV2FulfillmentConfigIx(marketIndex: number, openbookMarket: PublicKey): Promise<TransactionInstruction>;
     initializePerpMarket(marketIndex: number, priceOracle: PublicKey, baseAssetReserve: BN, quoteAssetReserve: BN, periodicity: BN, pegMultiplier?: BN, oracleSource?: OracleSource, contractTier?: ContractTier, marginRatioInitial?: number, marginRatioMaintenance?: number, liquidatorFee?: number, ifLiquidatorFee?: number, imfFactor?: number, activeStatus?: boolean, baseSpread?: number, maxSpread?: number, maxOpenInterest?: BN, maxRevenueWithdrawPerPeriod?: BN, quoteMaxInsurance?: BN, orderStepSize?: BN, orderTickSize?: BN, minOrderSize?: BN, concentrationCoefScale?: BN, curveUpdateIntensity?: number, ammJitIntensity?: number, name?: string): Promise<TransactionSignature>;
     getInitializePerpMarketIx(marketIndex: number, priceOracle: PublicKey, baseAssetReserve: BN, quoteAssetReserve: BN, periodicity: BN, pegMultiplier?: BN, oracleSource?: OracleSource, contractTier?: ContractTier, marginRatioInitial?: number, marginRatioMaintenance?: number, liquidatorFee?: number, ifLiquidatorFee?: number, imfFactor?: number, activeStatus?: boolean, baseSpread?: number, maxSpread?: number, maxOpenInterest?: BN, maxRevenueWithdrawPerPeriod?: BN, quoteMaxInsurance?: BN, orderStepSize?: BN, orderTickSize?: BN, minOrderSize?: BN, concentrationCoefScale?: BN, curveUpdateIntensity?: number, ammJitIntensity?: number, name?: string): Promise<TransactionInstruction>;
+    initializePredictionMarket(perpMarketIndex: number): Promise<TransactionSignature>;
+    getInitializePredictionMarketIx(perpMarketIndex: number): Promise<TransactionInstruction>;
     deleteInitializedPerpMarket(marketIndex: number): Promise<TransactionSignature>;
     getDeleteInitializedPerpMarketIx(marketIndex: number): Promise<TransactionInstruction>;
     moveAmmPrice(perpMarketIndex: number, baseAssetReserve: BN, quoteAssetReserve: BN, sqrtK?: BN): Promise<TransactionSignature>;

package/lib/adminClient.js

@@ -236,6 +236,23 @@ class AdminClient extends driftClient_1.DriftClient {
             },
         });
     }
+    async initializePredictionMarket(perpMarketIndex) {
+        const updatePerpMarketConcentrationCoefIx = await this.getInitializePredictionMarketIx(perpMarketIndex);
+        const tx = await this.buildTransaction(updatePerpMarketConcentrationCoefIx);
+        const { txSig } = await this.sendTransaction(tx, [], this.opts);
+        return txSig;
+    }
+    async getInitializePredictionMarketIx(perpMarketIndex) {
+        return await this.program.instruction.initializePredictionMarket({
+            accounts: {
+                state: await this.getStatePublicKey(),
+                admin: this.isSubscribed
+                    ? this.getStateAccount().admin
+                    : this.wallet.publicKey,
+                perpMarket: await (0, pda_1.getPerpMarketPublicKey)(this.program.programId, perpMarketIndex),
+            },
+        });
+    }
     async deleteInitializedPerpMarket(marketIndex) {
         const deleteInitializeMarketIx = await this.getDeleteInitializedPerpMarketIx(marketIndex);
         const tx = await this.buildTransaction(deleteInitializeMarketIx);

package/lib/constants/numericConstants.d.ts

@@ -68,3 +68,4 @@ export declare const SLOT_TIME_ESTIMATE_MS = 400;
 export declare const DUST_POSITION_SIZE: BN;
 export declare const FUEL_WINDOW: BN;
 export declare const FUEL_START_TS: BN;
+export declare const MAX_PREDICTION_PRICE: BN;

package/lib/constants/numericConstants.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MARGIN_PRECISION = exports.AMM_TIMES_PEG_TO_QUOTE_PRECISION_RATIO = exports.PRICE_TO_QUOTE_PRECISION = exports.PRICE_DIV_PEG = exports.AMM_TO_QUOTE_PRECISION_RATIO = exports.BASE_PRECISION_EXP = exports.BASE_PRECISION = exports.AMM_RESERVE_PRECISION = exports.PEG_PRECISION = exports.FUNDING_RATE_BUFFER_PRECISION = exports.FUNDING_RATE_PRECISION = exports.PRICE_PRECISION = exports.QUOTE_PRECISION = exports.LIQUIDATION_FEE_PRECISION = exports.SPOT_MARKET_IMF_PRECISION = exports.SPOT_MARKET_IMF_PRECISION_EXP = exports.SPOT_MARKET_BALANCE_PRECISION = exports.SPOT_MARKET_BALANCE_PRECISION_EXP = exports.SPOT_MARKET_WEIGHT_PRECISION = exports.SPOT_MARKET_UTILIZATION_PRECISION = exports.SPOT_MARKET_UTILIZATION_PRECISION_EXP = exports.SPOT_MARKET_CUMULATIVE_INTEREST_PRECISION = exports.SPOT_MARKET_CUMULATIVE_INTEREST_PRECISION_EXP = exports.SPOT_MARKET_RATE_PRECISION = exports.SPOT_MARKET_RATE_PRECISION_EXP = exports.AMM_RESERVE_PRECISION_EXP = exports.PEG_PRECISION_EXP = exports.FUNDING_RATE_PRECISION_EXP = exports.PRICE_PRECISION_EXP = exports.FUNDING_RATE_BUFFER_PRECISION_EXP = exports.QUOTE_PRECISION_EXP = exports.CONCENTRATION_PRECISION = exports.PERCENTAGE_PRECISION = exports.PERCENTAGE_PRECISION_EXP = exports.MAX_LEVERAGE_ORDER_SIZE = exports.MAX_LEVERAGE = exports.TEN_MILLION = exports.BN_MAX = exports.TEN_THOUSAND = exports.TEN = exports.NINE = exports.EIGHT = exports.SEVEN = exports.SIX = exports.FIVE = exports.FOUR = exports.THREE = exports.TWO = exports.ONE = exports.ZERO = void 0;
-exports.FUEL_START_TS = exports.FUEL_WINDOW = exports.DUST_POSITION_SIZE = exports.SLOT_TIME_ESTIMATE_MS = exports.IDLE_TIME_SLOTS = exports.ACCOUNT_AGE_DELETION_CUTOFF_SECONDS = exports.DEFAULT_REVENUE_SINCE_LAST_FUNDING_SPREAD_RETREAT = exports.OPEN_ORDER_MARGIN_REQUIREMENT = exports.LAMPORTS_EXP = exports.LAMPORTS_PRECISION = exports.GOV_SPOT_MARKET_INDEX = exports.QUOTE_SPOT_MARKET_INDEX = exports.ONE_YEAR = exports.ONE_HOUR = exports.FIVE_MINUTE = exports.FUNDING_RATE_OFFSET_DENOMINATOR = exports.LIQUIDATION_PCT_PRECISION = exports.BID_ASK_SPREAD_PRECISION = void 0;
+exports.MAX_PREDICTION_PRICE = exports.FUEL_START_TS = exports.FUEL_WINDOW = exports.DUST_POSITION_SIZE = exports.SLOT_TIME_ESTIMATE_MS = exports.IDLE_TIME_SLOTS = exports.ACCOUNT_AGE_DELETION_CUTOFF_SECONDS = exports.DEFAULT_REVENUE_SINCE_LAST_FUNDING_SPREAD_RETREAT = exports.OPEN_ORDER_MARGIN_REQUIREMENT = exports.LAMPORTS_EXP = exports.LAMPORTS_PRECISION = exports.GOV_SPOT_MARKET_INDEX = exports.QUOTE_SPOT_MARKET_INDEX = exports.ONE_YEAR = exports.ONE_HOUR = exports.FIVE_MINUTE = exports.FUNDING_RATE_OFFSET_DENOMINATOR = exports.LIQUIDATION_PCT_PRECISION = exports.BID_ASK_SPREAD_PRECISION = void 0;
 const web3_js_1 = require("@solana/web3.js");
 const __1 = require("../");
 exports.ZERO = new __1.BN(0);
@@ -72,3 +72,4 @@ exports.SLOT_TIME_ESTIMATE_MS = 400;
 exports.DUST_POSITION_SIZE = exports.QUOTE_PRECISION.divn(100); // Dust position is any position smaller than 1c
 exports.FUEL_WINDOW = new __1.BN(60 * 60 * 24 * 28); // 28 days
 exports.FUEL_START_TS = new __1.BN(1723147200); // unix timestamp
+exports.MAX_PREDICTION_PRICE = exports.PRICE_PRECISION;

package/lib/dlob/orderBookLevels.js

@@ -104,7 +104,7 @@ function getVammL2Generator({ marketAccount, oraclePriceData, numOrders, now, to
         openAsks = __1.ZERO;
     }
     now = now !== null && now !== void 0 ? now : new __1.BN(Date.now() / 1000);
-    const [bidReserves, askReserves] = (0, __1.calculateSpreadReserves)(updatedAmm, oraclePriceData, now);
+    const [bidReserves, askReserves] = (0, __1.calculateSpreadReserves)(updatedAmm, oraclePriceData, now, (0, __1.isVariant)(marketAccount.contractType, 'prediction'));
     let numBids = 0;
     let topOfBookBidSize = __1.ZERO;
     let bidSize = openBids.div(new __1.BN(numBaseOrders));

package/lib/driftClient.js

@@ -3980,6 +3980,7 @@ class DriftClient {
         }
         const [pullIx, _responses, success] = await feedAccount.fetchUpdateIx({
             numSignatures,
+            // @ts-ignore :: TODO someone needs to look at this
             feedConfigs: this.sbProgramFeedConfigs.get(feed.toString()),
         });
         if (!success) {

package/lib/idl/drift.json

@@ -3403,6 +3403,27 @@
                 }
             ]
         },
+        {
+            "name": "initializePredictionMarket",
+            "accounts": [
+                {
+                    "name": "admin",
+                    "isMut": false,
+                    "isSigner": true
+                },
+                {
+                    "name": "state",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "perpMarket",
+                    "isMut": true,
+                    "isSigner": false
+                }
+            ],
+            "args": []
+        },
         {
             "name": "deleteInitializedPerpMarket",
             "accounts": [
@@ -12906,6 +12927,11 @@
             "code": 6283,
             "name": "LiquidationOrderFailedToFill",
             "msg": "Liquidation order failed to fill"
+        },
+        {
+            "code": 6284,
+            "name": "InvalidPredictionMarketOrder",
+            "msg": "Invalid prediction market order"
         }
     ],
     "metadata": {

package/lib/math/amm.d.ts

@@ -6,13 +6,13 @@ export declare function calculatePegFromTargetPrice(targetPrice: BN, baseAssetRe
 export declare function calculateOptimalPegAndBudget(amm: AMM, oraclePriceData: OraclePriceData): [BN, BN, BN, boolean];
 export declare function calculateNewAmm(amm: AMM, oraclePriceData: OraclePriceData): [BN, BN, BN, BN];
 export declare function calculateUpdatedAMM(amm: AMM, oraclePriceData: OraclePriceData): AMM;
-export declare function calculateUpdatedAMMSpreadReserves(amm: AMM, direction: PositionDirection, oraclePriceData: OraclePriceData): {
+export declare function calculateUpdatedAMMSpreadReserves(amm: AMM, direction: PositionDirection, oraclePriceData: OraclePriceData, isPrediction?: boolean): {
     baseAssetReserve: BN;
     quoteAssetReserve: BN;
     sqrtK: BN;
     newPeg: BN;
 };
-export declare function calculateBidAskPrice(amm: AMM, oraclePriceData: OraclePriceData, withUpdate?: boolean): [BN, BN];
+export declare function calculateBidAskPrice(amm: AMM, oraclePriceData: OraclePriceData, withUpdate?: boolean, isPrediction?: boolean): [BN, BN];
 /**
  * Calculates a price given an arbitrary base and quote amount (they must have the same precision)
  *
@@ -64,7 +64,8 @@ export declare function calculateSpreadBN(baseSpread: number, lastOracleReserveP
     shortSpread: number;
 };
 export declare function calculateSpread(amm: AMM, oraclePriceData: OraclePriceData, now?: BN, reservePrice?: BN): [number, number];
-export declare function calculateSpreadReserves(amm: AMM, oraclePriceData: OraclePriceData, now?: BN): {
+export declare function getQuoteAssetReservePredictionMarketBounds(amm: AMM, direction: PositionDirection): [BN, BN];
+export declare function calculateSpreadReserves(amm: AMM, oraclePriceData: OraclePriceData, now?: BN, isPrediction?: boolean): {
     baseAssetReserve: any;
     quoteAssetReserve: any;
 }[];
@@ -92,6 +93,6 @@ export declare function getSwapDirection(inputAssetType: AssetType, positionDire
  * @returns cost : Precision PRICE_PRECISION
  */
 export declare function calculateTerminalPrice(market: PerpMarketAccount): BN;
-export declare function calculateMaxBaseAssetAmountToTrade(amm: AMM, limit_price: BN, direction: PositionDirection, oraclePriceData?: OraclePriceData, now?: BN): [BN, PositionDirection];
+export declare function calculateMaxBaseAssetAmountToTrade(amm: AMM, limit_price: BN, direction: PositionDirection, oraclePriceData?: OraclePriceData, now?: BN, isPrediction?: boolean): [BN, PositionDirection];
 export declare function calculateQuoteAssetAmountSwapped(quoteAssetReserves: BN, pegMultiplier: BN, swapDirection: SwapDirection): BN;
 export declare function calculateMaxBaseAssetAmountFillable(amm: AMM, orderDirection: PositionDirection): BN;

package/lib/math/amm.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateMaxBaseAssetAmountFillable = exports.calculateQuoteAssetAmountSwapped = exports.calculateMaxBaseAssetAmountToTrade = exports.calculateTerminalPrice = exports.getSwapDirection = exports.calculateSwapOutput = exports.calculateSpreadReserves = exports.calculateSpread = exports.calculateSpreadBN = exports.calculateVolSpreadBN = exports.calculateMaxSpread = exports.calculateEffectiveLeverage = exports.calculateReferencePriceOffset = exports.calculateInventoryScale = exports.calculateInventoryLiquidityRatio = exports.calculateMarketOpenBidAsk = exports.calculateAmmReservesAfterSwap = exports.calculatePrice = exports.calculateBidAskPrice = exports.calculateUpdatedAMMSpreadReserves = exports.calculateUpdatedAMM = exports.calculateNewAmm = exports.calculateOptimalPegAndBudget = exports.calculatePegFromTargetPrice = void 0;
+exports.calculateMaxBaseAssetAmountFillable = exports.calculateQuoteAssetAmountSwapped = exports.calculateMaxBaseAssetAmountToTrade = exports.calculateTerminalPrice = exports.getSwapDirection = exports.calculateSwapOutput = exports.calculateSpreadReserves = exports.getQuoteAssetReservePredictionMarketBounds = exports.calculateSpread = exports.calculateSpreadBN = exports.calculateVolSpreadBN = exports.calculateMaxSpread = exports.calculateEffectiveLeverage = exports.calculateReferencePriceOffset = exports.calculateInventoryScale = exports.calculateInventoryLiquidityRatio = exports.calculateMarketOpenBidAsk = exports.calculateAmmReservesAfterSwap = exports.calculatePrice = exports.calculateBidAskPrice = exports.calculateUpdatedAMMSpreadReserves = exports.calculateUpdatedAMM = exports.calculateNewAmm = exports.calculateOptimalPegAndBudget = exports.calculatePegFromTargetPrice = void 0;
 const anchor_1 = require("@coral-xyz/anchor");
 const numericConstants_1 = require("../constants/numericConstants");
 const types_1 = require("../types");
@@ -103,9 +103,9 @@ function calculateUpdatedAMM(amm, oraclePriceData) {
     return newAmm;
 }
 exports.calculateUpdatedAMM = calculateUpdatedAMM;
-function calculateUpdatedAMMSpreadReserves(amm, direction, oraclePriceData) {
+function calculateUpdatedAMMSpreadReserves(amm, direction, oraclePriceData, isPrediction = false) {
     const newAmm = calculateUpdatedAMM(amm, oraclePriceData);
-    const [shortReserves, longReserves] = calculateSpreadReserves(newAmm, oraclePriceData);
+    const [shortReserves, longReserves] = calculateSpreadReserves(newAmm, oraclePriceData, undefined, isPrediction);
     const dirReserves = (0, types_1.isVariant)(direction, 'long')
         ? longReserves
         : shortReserves;
@@ -118,7 +118,7 @@ function calculateUpdatedAMMSpreadReserves(amm, direction, oraclePriceData) {
     return result;
 }
 exports.calculateUpdatedAMMSpreadReserves = calculateUpdatedAMMSpreadReserves;
-function calculateBidAskPrice(amm, oraclePriceData, withUpdate = true) {
+function calculateBidAskPrice(amm, oraclePriceData, withUpdate = true, isPrediction = false) {
     let newAmm;
     if (withUpdate) {
         newAmm = calculateUpdatedAMM(amm, oraclePriceData);
@@ -126,7 +126,7 @@ function calculateBidAskPrice(amm, oraclePriceData, withUpdate = true) {
     else {
         newAmm = amm;
     }
-    const [bidReserves, askReserves] = calculateSpreadReserves(newAmm, oraclePriceData);
+    const [bidReserves, askReserves] = calculateSpreadReserves(newAmm, oraclePriceData, undefined, isPrediction);
     const askPrice = calculatePrice(askReserves.baseAssetReserve, askReserves.quoteAssetReserve, newAmm.pegMultiplier);
     const bidPrice = calculatePrice(bidReserves.baseAssetReserve, bidReserves.quoteAssetReserve, newAmm.pegMultiplier);
     return [bidPrice, askPrice];
@@ -442,7 +442,30 @@ function calculateSpread(amm, oraclePriceData, now, reservePrice) {
     return [longSpread, shortSpread];
 }
 exports.calculateSpread = calculateSpread;
-function calculateSpreadReserves(amm, oraclePriceData, now) {
+function getQuoteAssetReservePredictionMarketBounds(amm, direction) {
+    let quoteAssetReserveLowerBound = numericConstants_1.ZERO;
+    const pegSqrt = (0, __1.squareRootBN)(amm.pegMultiplier.mul(numericConstants_1.PEG_PRECISION).addn(1)).addn(1);
+    let quoteAssetReserveUpperBound = amm.sqrtK
+        .mul(pegSqrt)
+        .div(amm.pegMultiplier);
+    if (direction === types_1.PositionDirection.LONG) {
+        quoteAssetReserveLowerBound = this.amm.sqrtK
+            .muln(22361)
+            .mul(pegSqrt)
+            .divn(100000)
+            .div(amm.pegMultiplier);
+    }
+    else {
+        quoteAssetReserveUpperBound = amm.sqrtK
+            .muln(97467)
+            .mul(pegSqrt)
+            .divn(100000)
+            .div(amm.pegMultiplier);
+    }
+    return [quoteAssetReserveLowerBound, quoteAssetReserveUpperBound];
+}
+exports.getQuoteAssetReservePredictionMarketBounds = getQuoteAssetReservePredictionMarketBounds;
+function calculateSpreadReserves(amm, oraclePriceData, now, isPrediction = false) {
     function calculateSpreadReserve(spread, direction, amm) {
         if (spread === 0) {
             return {
@@ -463,6 +486,10 @@ function calculateSpreadReserves(amm, oraclePriceData, now) {
         else {
             quoteAssetReserve = amm.quoteAssetReserve.sub(quoteAssetReserveDelta.abs());
         }
+        if (isPrediction) {
+            const [qarLower, qarUpper] = getQuoteAssetReservePredictionMarketBounds(amm, direction);
+            quoteAssetReserve = (0, __1.clampBN)(quoteAssetReserve, qarLower, qarUpper);
+        }
         const baseAssetReserve = amm.sqrtK.mul(amm.sqrtK).div(quoteAssetReserve);
         return {
             baseAssetReserve,
@@ -542,7 +569,7 @@ function calculateTerminalPrice(market) {
     return terminalPrice;
 }
 exports.calculateTerminalPrice = calculateTerminalPrice;
-function calculateMaxBaseAssetAmountToTrade(amm, limit_price, direction, oraclePriceData, now) {
+function calculateMaxBaseAssetAmountToTrade(amm, limit_price, direction, oraclePriceData, now, isPrediction = false) {
     const invariant = amm.sqrtK.mul(amm.sqrtK);
     const newBaseAssetReserveSquared = invariant
         .mul(numericConstants_1.PRICE_PRECISION)
@@ -550,7 +577,7 @@ function calculateMaxBaseAssetAmountToTrade(amm, limit_price, direction, oracleP
         .div(limit_price)
         .div(numericConstants_1.PEG_PRECISION);
     const newBaseAssetReserve = (0, __1.squareRootBN)(newBaseAssetReserveSquared);
-    const [shortSpreadReserves, longSpreadReserves] = calculateSpreadReserves(amm, oraclePriceData, now);
+    const [shortSpreadReserves, longSpreadReserves] = calculateSpreadReserves(amm, oraclePriceData, now, isPrediction);
     const baseAssetReserveBefore = (0, types_1.isVariant)(direction, 'long')
         ? longSpreadReserves.baseAssetReserve
         : shortSpreadReserves.baseAssetReserve;

package/lib/math/margin.d.ts

@@ -7,5 +7,18 @@ imfFactor: BN, liabilityWeight: BN, precision: BN): BN;
 export declare function calculateSizeDiscountAssetWeight(size: BN, // AMM_RESERVE_PRECISION
 imfFactor: BN, assetWeight: BN): BN;
 export declare function calculateOraclePriceForPerpMargin(perpPosition: PerpPosition, market: PerpMarketAccount, oraclePriceData: OraclePriceData): BN;
+/**
+ * This is _not_ the same as liability value as for prediction markets, the liability for the short in prediction market is (1 - oracle price) * base
+ * See {@link calculatePerpLiabilityValue} to get the liabiltiy value
+ * @param market
+ * @param perpPosition
+ * @param oraclePriceData
+ * @param includeOpenOrders
+ */
 export declare function calculateBaseAssetValueWithOracle(market: PerpMarketAccount, perpPosition: PerpPosition, oraclePriceData: OraclePriceData, includeOpenOrders?: boolean): BN;
-export declare function calculateWorstCaseBaseAssetAmount(perpPosition: PerpPosition): BN;
+export declare function calculateWorstCaseBaseAssetAmount(perpPosition: PerpPosition, perpMarket: PerpMarketAccount, oraclePrice: BN): BN;
+export declare function calculateWorstCasePerpLiabilityValue(perpPosition: PerpPosition, perpMarket: PerpMarketAccount, oraclePrice: BN): {
+    worstCaseBaseAssetAmount: BN;
+    worstCaseLiabilityValue: BN;
+};
+export declare function calculatePerpLiabilityValue(baseAssetAmount: BN, oraclePrice: BN, isPredictionMarket: boolean): BN;

package/lib/math/margin.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateWorstCaseBaseAssetAmount = exports.calculateBaseAssetValueWithOracle = exports.calculateOraclePriceForPerpMargin = exports.calculateSizeDiscountAssetWeight = exports.calculateSizePremiumLiabilityWeight = void 0;
+exports.calculatePerpLiabilityValue = exports.calculateWorstCasePerpLiabilityValue = exports.calculateWorstCaseBaseAssetAmount = exports.calculateBaseAssetValueWithOracle = exports.calculateOraclePriceForPerpMargin = exports.calculateSizeDiscountAssetWeight = exports.calculateSizePremiumLiabilityWeight = void 0;
 const utils_1 = require("./utils");
 const numericConstants_1 = require("../constants/numericConstants");
 const anchor_1 = require("@coral-xyz/anchor");
@@ -56,25 +56,63 @@ function calculateOraclePriceForPerpMargin(perpPosition, market, oraclePriceData
     return marginPrice;
 }
 exports.calculateOraclePriceForPerpMargin = calculateOraclePriceForPerpMargin;
+/**
+ * This is _not_ the same as liability value as for prediction markets, the liability for the short in prediction market is (1 - oracle price) * base
+ * See {@link calculatePerpLiabilityValue} to get the liabiltiy value
+ * @param market
+ * @param perpPosition
+ * @param oraclePriceData
+ * @param includeOpenOrders
+ */
 function calculateBaseAssetValueWithOracle(market, perpPosition, oraclePriceData, includeOpenOrders = false) {
     let price = oraclePriceData.price;
     if ((0, types_1.isVariant)(market.status, 'settlement')) {
         price = market.expiryPrice;
     }
     const baseAssetAmount = includeOpenOrders
-        ? calculateWorstCaseBaseAssetAmount(perpPosition)
+        ? calculateWorstCaseBaseAssetAmount(perpPosition, market, oraclePriceData.price)
         : perpPosition.baseAssetAmount;
     return baseAssetAmount.abs().mul(price).div(numericConstants_1.AMM_RESERVE_PRECISION);
 }
 exports.calculateBaseAssetValueWithOracle = calculateBaseAssetValueWithOracle;
-function calculateWorstCaseBaseAssetAmount(perpPosition) {
+function calculateWorstCaseBaseAssetAmount(perpPosition, perpMarket, oraclePrice) {
+    return calculateWorstCasePerpLiabilityValue(perpPosition, perpMarket, oraclePrice).worstCaseBaseAssetAmount;
+}
+exports.calculateWorstCaseBaseAssetAmount = calculateWorstCaseBaseAssetAmount;
+function calculateWorstCasePerpLiabilityValue(perpPosition, perpMarket, oraclePrice) {
     const allBids = perpPosition.baseAssetAmount.add(perpPosition.openBids);
     const allAsks = perpPosition.baseAssetAmount.add(perpPosition.openAsks);
-    if (allBids.abs().gt(allAsks.abs())) {
-        return allBids;
+    const isPredictionMarket = (0, types_1.isVariant)(perpMarket.contractType, 'prediction');
+    const allBidsLiabilityValue = calculatePerpLiabilityValue(allBids, oraclePrice, isPredictionMarket);
+    const allAsksLiabilityValue = calculatePerpLiabilityValue(allAsks, oraclePrice, isPredictionMarket);
+    if (allAsksLiabilityValue.gte(allBidsLiabilityValue)) {
+        return {
+            worstCaseBaseAssetAmount: allAsks,
+            worstCaseLiabilityValue: allAsksLiabilityValue,
+        };
     }
     else {
-        return allAsks;
+        return {
+            worstCaseBaseAssetAmount: allBids,
+            worstCaseLiabilityValue: allBidsLiabilityValue,
+        };
     }
 }
-exports.calculateWorstCaseBaseAssetAmount = calculateWorstCaseBaseAssetAmount;
+exports.calculateWorstCasePerpLiabilityValue = calculateWorstCasePerpLiabilityValue;
+function calculatePerpLiabilityValue(baseAssetAmount, oraclePrice, isPredictionMarket) {
+    if (isPredictionMarket) {
+        if (baseAssetAmount.gt(numericConstants_1.ZERO)) {
+            return baseAssetAmount.mul(oraclePrice).div(numericConstants_1.BASE_PRECISION);
+        }
+        else {
+            return baseAssetAmount
+                .abs()
+                .mul(numericConstants_1.MAX_PREDICTION_PRICE.sub(oraclePrice))
+                .div(numericConstants_1.BASE_PRECISION);
+        }
+    }
+    else {
+        return baseAssetAmount.abs().mul(oraclePrice).div(numericConstants_1.BASE_PRECISION);
+    }
+}
+exports.calculatePerpLiabilityValue = calculatePerpLiabilityValue;

package/lib/types.d.ts

@@ -78,6 +78,9 @@ export declare class ContractType {
     static readonly FUTURE: {
         future: {};
     };
+    static readonly PREDICTION: {
+        prediction: {};
+    };
 }
 export declare class ContractTier {
     static readonly A: {

package/lib/types.js

@@ -63,6 +63,7 @@ class ContractType {
 exports.ContractType = ContractType;
 ContractType.PERPETUAL = { perpetual: {} };
 ContractType.FUTURE = { future: {} };
+ContractType.PREDICTION = { prediction: {} };
 class ContractTier {
 }
 exports.ContractTier = ContractTier;

package/lib/user.d.ts

@@ -164,7 +164,7 @@ export declare class User {
      * @returns : number (value from [0, 100])
      */
     getHealth(): number;
-    calculateWeightedPerpPositionValue(perpPosition: PerpPosition, marginCategory?: MarginCategory, liquidationBuffer?: BN, includeOpenOrders?: boolean, strict?: boolean): BN;
+    calculateWeightedPerpPositionLiability(perpPosition: PerpPosition, marginCategory?: MarginCategory, liquidationBuffer?: BN, includeOpenOrders?: boolean, strict?: boolean): BN;
     /**
      * calculates position value of a single perp market in margin system
      * @returns : Precision QUOTE_PRECISION
@@ -174,12 +174,17 @@ export declare class User {
      * calculates sum of position value across all positions in margin system
      * @returns : Precision QUOTE_PRECISION
      */
-    getTotalPerpPositionValue(marginCategory?: MarginCategory, liquidationBuffer?: BN, includeOpenOrders?: boolean, strict?: boolean): BN;
+    getTotalPerpPositionLiability(marginCategory?: MarginCategory, liquidationBuffer?: BN, includeOpenOrders?: boolean, strict?: boolean): BN;
     /**
-     * calculates position value in margin system
+     * calculates position value based on oracle
      * @returns : Precision QUOTE_PRECISION
      */
     getPerpPositionValue(marketIndex: number, oraclePriceData: OraclePriceData, includeOpenOrders?: boolean): BN;
+    /**
+     * calculates position liabiltiy value in margin system
+     * @returns : Precision QUOTE_PRECISION
+     */
+    getPerpLiabilityValue(marketIndex: number, oraclePriceData: OraclePriceData, includeOpenOrders?: boolean): BN;
     getPositionSide(currentPosition: Pick<PerpPosition, 'baseAssetAmount'>): PositionDirection | undefined;
     /**
      * calculates average exit price (optionally for closing up to 100% of position)
@@ -264,7 +269,7 @@ export declare class User {
      */
     liquidationPrice(marketIndex: number, positionBaseSizeChange?: BN, estimatedEntryPrice?: BN, marginCategory?: MarginCategory, includeOpenOrders?: boolean): BN;
     calculateEntriesEffectOnFreeCollateral(market: PerpMarketAccount, oraclePrice: BN, perpPosition: PerpPosition, positionBaseSizeChange: BN, estimatedEntryPrice: BN, includeOpenOrders: boolean): BN;
-    calculateFreeCollateralDeltaForPerp(market: PerpMarketAccount, perpPosition: PerpPosition, positionBaseSizeChange: BN, marginCategory?: MarginCategory, includeOpenOrders?: boolean): BN | undefined;
+    calculateFreeCollateralDeltaForPerp(market: PerpMarketAccount, perpPosition: PerpPosition, positionBaseSizeChange: BN, oraclePrice: BN, marginCategory?: MarginCategory, includeOpenOrders?: boolean): BN | undefined;
     calculateFreeCollateralDeltaForSpot(market: SpotMarketAccount, signedTokenAmount: BN, marginCategory?: MarginCategory): BN;
     /**
      * Calculates the estimated liquidation price for a position after closing a quote amount of the position.
@@ -292,9 +297,13 @@ export declare class User {
      *
      * @param targetMarketIndex
      * @param tradeSide
-     * @returns tradeSizeAllowed : Precision QUOTE_PRECISION
+     * @param isLp
+     * @returns { tradeSize: BN, oppositeSideTradeSize: BN} : Precision QUOTE_PRECISION
      */
-    getMaxTradeSizeUSDCForPerp(targetMarketIndex: number, tradeSide: PositionDirection, isLp?: boolean): BN;
+    getMaxTradeSizeUSDCForPerp(targetMarketIndex: number, tradeSide: PositionDirection, isLp?: boolean): {
+        tradeSize: BN;
+        oppositeSideTradeSize: BN;
+    };
     /**
      * Get the maximum trade size for a given market, taking into account the user's current leverage, positions, collateral, etc.
      *