@drift-labs/sdk 2.22.0-beta.3 → 2.23.0-beta.0

package/src/driftClient.ts

@@ -37,6 +37,7 @@ import {
 	PostOnlyParams,
 	SpotBalanceType,
 	PerpMarketExtendedInfo,
+	UserStatsAccount,
 } from './types';
 import * as anchor from '@project-serum/anchor';
 import driftIDL from './idl/drift.json';
@@ -776,6 +777,24 @@ export class DriftClient {
 		);
 	}
 
+	public async getReferredUserStatsAccountsByReferrer(
+		referrer: PublicKey
+	): Promise<UserStatsAccount[]> {
+		const programAccounts = await this.program.account.userStats.all([
+			{
+				memcmp: {
+					offset: 40,
+					/** data to match, as base-58 encoded string and limited to less than 129 bytes */
+					bytes: bs58.encode(referrer.toBuffer()),
+				},
+			},
+		]);
+
+		return programAccounts.map(
+			(programAccount) => programAccount.account as UserStatsAccount
+		);
+	}
+
 	public async getReferrerNameAccountsForAuthority(
 		authority: PublicKey
 	): Promise<ReferrerNameAccount[]> {
@@ -934,17 +953,32 @@ export class DriftClient {
 				// if cache has more recent slot than user positions account slot, add market to remaining accounts
 				// otherwise remove from slot
 				if (slot > lastUserSlot) {
-					const marketAccount = this.getPerpMarketAccount(marketIndex);
+					const perpMarketAccount = this.getPerpMarketAccount(marketIndex);
 					perpMarketAccountMap.set(marketIndex, {
-						pubkey: marketAccount.pubkey,
+						pubkey: perpMarketAccount.pubkey,
+						isSigner: false,
+						isWritable: false,
+					});
+					oracleAccountMap.set(perpMarketAccount.amm.oracle.toString(), {
+						pubkey: perpMarketAccount.amm.oracle,
 						isSigner: false,
 						isWritable: false,
 					});
-					oracleAccountMap.set(marketAccount.amm.oracle.toString(), {
-						pubkey: marketAccount.amm.oracle,
+					const spotMarketAccount = this.getSpotMarketAccount(
+						perpMarketAccount.quoteSpotMarketIndex
+					);
+					spotMarketAccountMap.set(perpMarketAccount.quoteSpotMarketIndex, {
+						pubkey: spotMarketAccount.pubkey,
 						isSigner: false,
 						isWritable: false,
 					});
+					if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
+						oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
+							pubkey: spotMarketAccount.oracle,
+							isSigner: false,
+							isWritable: false,
+						});
+					}
 				} else {
 					this.perpMarketLastSlotCache.delete(marketIndex);
 				}
@@ -957,15 +991,15 @@ export class DriftClient {
 				// if cache has more recent slot than user positions account slot, add market to remaining accounts
 				// otherwise remove from slot
 				if (slot > lastUserSlot) {
-					const marketAccount = this.getSpotMarketAccount(marketIndex);
+					const spotMarketAccount = this.getSpotMarketAccount(marketIndex);
 					spotMarketAccountMap.set(marketIndex, {
-						pubkey: marketAccount.pubkey,
+						pubkey: spotMarketAccount.pubkey,
 						isSigner: false,
 						isWritable: false,
 					});
-					if (!marketAccount.oracle.equals(PublicKey.default)) {
-						oracleAccountMap.set(marketAccount.oracle.toString(), {
-							pubkey: marketAccount.oracle,
+					if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
+						oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
+							pubkey: spotMarketAccount.oracle,
 							isSigner: false,
 							isWritable: false,
 						});
@@ -977,50 +1011,80 @@ export class DriftClient {
 		}
 
 		if (params.readablePerpMarketIndex !== undefined) {
-			const marketAccount = this.getPerpMarketAccount(
+			const perpMarketAccount = this.getPerpMarketAccount(
 				params.readablePerpMarketIndex
 			);
 			perpMarketAccountMap.set(params.readablePerpMarketIndex, {
-				pubkey: marketAccount.pubkey,
+				pubkey: perpMarketAccount.pubkey,
+				isSigner: false,
+				isWritable: false,
+			});
+			oracleAccountMap.set(perpMarketAccount.amm.oracle.toString(), {
+				pubkey: perpMarketAccount.amm.oracle,
 				isSigner: false,
 				isWritable: false,
 			});
-			oracleAccountMap.set(marketAccount.amm.oracle.toString(), {
-				pubkey: marketAccount.amm.oracle,
+			const spotMarketAccount = this.getSpotMarketAccount(
+				perpMarketAccount.quoteSpotMarketIndex
+			);
+			spotMarketAccountMap.set(perpMarketAccount.quoteSpotMarketIndex, {
+				pubkey: spotMarketAccount.pubkey,
 				isSigner: false,
 				isWritable: false,
 			});
+			if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
+				oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
+					pubkey: spotMarketAccount.oracle,
+					isSigner: false,
+					isWritable: false,
+				});
+			}
+		}
+
+		if (params.readableSpotMarketIndexes !== undefined) {
+			for (const readableSpotMarketIndex of params.readableSpotMarketIndexes) {
+				const spotMarketAccount = this.getSpotMarketAccount(
+					readableSpotMarketIndex
+				);
+				spotMarketAccountMap.set(readableSpotMarketIndex, {
+					pubkey: spotMarketAccount.pubkey,
+					isSigner: false,
+					isWritable: false,
+				});
+				if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
+					oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
+						pubkey: spotMarketAccount.oracle,
+						isSigner: false,
+						isWritable: false,
+					});
+				}
+			}
 		}
 
 		if (params.writablePerpMarketIndexes !== undefined) {
 			for (const writablePerpMarketIndex of params.writablePerpMarketIndexes) {
-				const marketAccount = this.getPerpMarketAccount(
+				const perpMarketAccount = this.getPerpMarketAccount(
 					writablePerpMarketIndex
 				);
 				perpMarketAccountMap.set(writablePerpMarketIndex, {
-					pubkey: marketAccount.pubkey,
+					pubkey: perpMarketAccount.pubkey,
 					isSigner: false,
 					isWritable: true,
 				});
-				oracleAccountMap.set(marketAccount.amm.oracle.toString(), {
-					pubkey: marketAccount.amm.oracle,
+				oracleAccountMap.set(perpMarketAccount.amm.oracle.toString(), {
+					pubkey: perpMarketAccount.amm.oracle,
 					isSigner: false,
 					isWritable: false,
 				});
-			}
-		}
-
-		if (params.readableSpotMarketIndexes !== undefined) {
-			for (const readableSpotMarketIndex of params.readableSpotMarketIndexes) {
 				const spotMarketAccount = this.getSpotMarketAccount(
-					readableSpotMarketIndex
+					perpMarketAccount.quoteSpotMarketIndex
 				);
-				spotMarketAccountMap.set(readableSpotMarketIndex, {
+				spotMarketAccountMap.set(perpMarketAccount.quoteSpotMarketIndex, {
 					pubkey: spotMarketAccount.pubkey,
 					isSigner: false,
 					isWritable: false,
 				});
-				if (spotMarketAccount.marketIndex !== 0) {
+				if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
 					oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
 						pubkey: spotMarketAccount.oracle,
 						isSigner: false,
@@ -1090,27 +1154,52 @@ export class DriftClient {
 						!spotPosition.openAsks.eq(ZERO) ||
 						!spotPosition.openBids.eq(ZERO)
 					) {
+						const quoteSpotMarket = this.getQuoteSpotMarketAccount();
 						spotMarketAccountMap.set(QUOTE_SPOT_MARKET_INDEX, {
-							pubkey: this.getQuoteSpotMarketAccount().pubkey,
+							pubkey: quoteSpotMarket.pubkey,
 							isSigner: false,
 							isWritable: false,
 						});
+						if (!quoteSpotMarket.oracle.equals(PublicKey.default)) {
+							oracleAccountMap.set(quoteSpotMarket.oracle.toString(), {
+								pubkey: quoteSpotMarket.oracle,
+								isSigner: false,
+								isWritable: false,
+							});
+						}
 					}
 				}
 			}
 			for (const position of userAccount.perpPositions) {
 				if (!positionIsAvailable(position)) {
-					const market = this.getPerpMarketAccount(position.marketIndex);
+					const perpMarketAccount = this.getPerpMarketAccount(
+						position.marketIndex
+					);
 					perpMarketAccountMap.set(position.marketIndex, {
-						pubkey: market.pubkey,
+						pubkey: perpMarketAccount.pubkey,
 						isWritable: false,
 						isSigner: false,
 					});
-					oracleAccountMap.set(market.amm.oracle.toString(), {
-						pubkey: market.amm.oracle,
+					oracleAccountMap.set(perpMarketAccount.amm.oracle.toString(), {
+						pubkey: perpMarketAccount.amm.oracle,
 						isWritable: false,
 						isSigner: false,
 					});
+					const spotMarketAccount = this.getSpotMarketAccount(
+						perpMarketAccount.quoteSpotMarketIndex
+					);
+					spotMarketAccountMap.set(perpMarketAccount.quoteSpotMarketIndex, {
+						pubkey: spotMarketAccount.pubkey,
+						isSigner: false,
+						isWritable: false,
+					});
+					if (!spotMarketAccount.oracle.equals(PublicKey.default)) {
+						oracleAccountMap.set(spotMarketAccount.oracle.toString(), {
+							pubkey: spotMarketAccount.oracle,
+							isSigner: false,
+							isWritable: false,
+						});
+					}
 				}
 			}
 		}

package/src/factory/oracleClient.ts

@@ -22,6 +22,10 @@ export function getOracleClient(
 		return new PythClient(connection, new BN(1000000));
 	}
 
+	if (isVariant(oracleSource, 'pythStableCoin')) {
+		return new PythClient(connection, undefined, true);
+	}
+
 	// if (isVariant(oracleSource, 'switchboard')) {
 	// 	return new SwitchboardClient(connection);
 	// }

package/src/idl/drift.json

@@ -1,5 +1,5 @@
 {
-  "version": "2.22.0-beta.3",
+  "version": "2.23.0-beta.0",
   "name": "drift",
   "instructions": [
     {
@@ -4081,12 +4081,20 @@
               "defined": "ContractTier"
             }
           },
+          {
+            "name": "padding1",
+            "type": "bool"
+          },
+          {
+            "name": "quoteSpotMarketIndex",
+            "type": "u16"
+          },
           {
             "name": "padding",
             "type": {
               "array": [
                 "u8",
-                51
+                48
               ]
             }
           }
@@ -5119,7 +5127,7 @@
         "fields": [
           {
             "name": "revenueWithdrawSinceLastSettle",
-            "type": "i64"
+            "type": "u64"
           },
           {
             "name": "maxRevenueWithdrawPerPeriod",
@@ -6383,6 +6391,9 @@
           },
           {
             "name": "Pyth1M"
+          },
+          {
+            "name": "PythStableCoin"
           }
         ]
       }

package/src/oracles/pythClient.ts

@@ -2,15 +2,26 @@ import { parsePriceData } from '@pythnetwork/client';
 import { Connection, PublicKey } from '@solana/web3.js';
 import { OracleClient, OraclePriceData } from './types';
 import { BN } from '@project-serum/anchor';
-import { ONE, PRICE_PRECISION, TEN } from '../constants/numericConstants';
+import {
+	ONE,
+	PRICE_PRECISION,
+	QUOTE_PRECISION,
+	TEN,
+} from '../constants/numericConstants';
 
 export class PythClient implements OracleClient {
 	private connection: Connection;
 	private multiple: BN;
+	private stableCoin: boolean;
 
-	public constructor(connection: Connection, multiple = ONE) {
+	public constructor(
+		connection: Connection,
+		multiple = ONE,
+		stableCoin = false
+	) {
 		this.connection = connection;
 		this.multiple = multiple;
+		this.stableCoin = stableCoin;
 	}
 
 	public async getOraclePriceData(
@@ -22,18 +33,24 @@ export class PythClient implements OracleClient {
 
 	public getOraclePriceDataFromBuffer(buffer: Buffer): OraclePriceData {
 		const priceData = parsePriceData(buffer);
+		const confidence = convertPythPrice(
+			priceData.confidence,
+			priceData.exponent,
+			this.multiple
+		);
+		let price = convertPythPrice(
+			priceData.aggregate.price,
+			priceData.exponent,
+			this.multiple
+		);
+		if (this.stableCoin) {
+			price = getStableCoinPrice(price, confidence);
+		}
+
 		return {
-			price: convertPythPrice(
-				priceData.aggregate.price,
-				priceData.exponent,
-				this.multiple
-			),
+			price,
 			slot: new BN(priceData.lastSlot.toString()),
-			confidence: convertPythPrice(
-				priceData.confidence,
-				priceData.exponent,
-				this.multiple
-			),
+			confidence,
 			twap: convertPythPrice(
 				priceData.twap.value,
 				priceData.exponent,
@@ -60,3 +77,12 @@ export function convertPythPrice(
 		.mul(PRICE_PRECISION)
 		.div(pythPrecision);
 }
+
+const fiveBPS = new BN(500);
+function getStableCoinPrice(price: BN, confidence: BN): BN {
+	if (price.sub(QUOTE_PRECISION).abs().lt(BN.min(confidence, fiveBPS))) {
+		return QUOTE_PRECISION;
+	} else {
+		return price;
+	}
+}

package/src/token/index.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseTokenAccount = void 0;
+const spl_token_1 = require("@solana/spl-token");
+const web3_js_1 = require("@solana/web3.js");
+function parseTokenAccount(data) {
+    const accountInfo = spl_token_1.AccountLayout.decode(data);
+    accountInfo.mint = new web3_js_1.PublicKey(accountInfo.mint);
+    accountInfo.owner = new web3_js_1.PublicKey(accountInfo.owner);
+    accountInfo.amount = spl_token_1.u64.fromBuffer(accountInfo.amount);
+    if (accountInfo.delegateOption === 0) {
+        accountInfo.delegate = null;
+        // eslint-disable-next-line new-cap
+        accountInfo.delegatedAmount = new spl_token_1.u64(0);
+    }
+    else {
+        accountInfo.delegate = new web3_js_1.PublicKey(accountInfo.delegate);
+        accountInfo.delegatedAmount = spl_token_1.u64.fromBuffer(accountInfo.delegatedAmount);
+    }
+    accountInfo.isInitialized = accountInfo.state !== 0;
+    accountInfo.isFrozen = accountInfo.state === 2;
+    if (accountInfo.isNativeOption === 1) {
+        accountInfo.rentExemptReserve = spl_token_1.u64.fromBuffer(accountInfo.isNative);
+        accountInfo.isNative = true;
+    }
+    else {
+        accountInfo.rentExemptReserve = null;
+        accountInfo.isNative = false;
+    }
+    if (accountInfo.closeAuthorityOption === 0) {
+        accountInfo.closeAuthority = null;
+    }
+    else {
+        accountInfo.closeAuthority = new web3_js_1.PublicKey(accountInfo.closeAuthority);
+    }
+    return accountInfo;
+}
+exports.parseTokenAccount = parseTokenAccount;

package/src/types.ts

@@ -80,6 +80,7 @@ export class OracleSource {
 	static readonly PYTH_1M = { pyth1M: {} };
 	// static readonly SWITCHBOARD = { switchboard: {} };
 	static readonly QUOTE_ASSET = { quoteAsset: {} };
+	static readonly PYTH_STABLE_COIN = { pythStableCoin: {} };
 }
 
 export class OrderType {
@@ -560,6 +561,7 @@ export type PerpMarketAccount = {
 		quoteSettledInsurance: BN;
 		quoteMaxInsurance: BN;
 	};
+	quoteSpotMarketIndex: number;
 };
 
 export type HistoricalOracleData = {

package/src/user.ts

@@ -424,7 +424,8 @@ export class User {
 		return this.getTotalPerpPositionValue(
 			marginCategory,
 			liquidationBuffer,
-			true
+			true,
+			strict
 		).add(
 			this.getSpotMarketLiabilityValue(
 				undefined,
@@ -467,9 +468,9 @@ export class User {
 	public getUnrealizedPNL(
 		withFunding?: boolean,
 		marketIndex?: number,
-		withWeightMarginCategory?: MarginCategory
+		withWeightMarginCategory?: MarginCategory,
+		strict = false
 	): BN {
-		const quoteSpotMarket = this.driftClient.getQuoteSpotMarketAccount();
 		return this.getActivePerpPositions()
 			.filter((pos) => (marketIndex ? pos.marketIndex === marketIndex : true))
 			.reduce((unrealizedPnl, perpPosition) => {
@@ -480,6 +481,13 @@ export class User {
 					market.marketIndex
 				);
 
+				const quoteSpotMarket = this.driftClient.getSpotMarketAccount(
+					market.quoteSpotMarketIndex
+				);
+				const quoteOraclePriceData = this.getOracleDataForSpotMarket(
+					market.quoteSpotMarketIndex
+				);
+
 				if (perpPosition.lpShares.gt(ZERO)) {
 					perpPosition = this.getSettledLPPosition(perpPosition.marketIndex)[0];
 				}
@@ -491,6 +499,25 @@ export class User {
 					oraclePriceData
 				);
 
+				let quotePrice;
+				if (strict && positionUnrealizedPnl.gt(ZERO)) {
+					quotePrice = BN.min(
+						quoteOraclePriceData.price,
+						quoteSpotMarket.historicalOracleData.lastOraclePriceTwap5Min
+					);
+				} else if (strict && positionUnrealizedPnl.lt(ZERO)) {
+					quotePrice = BN.max(
+						quoteOraclePriceData.price,
+						quoteSpotMarket.historicalOracleData.lastOraclePriceTwap5Min
+					);
+				} else {
+					quotePrice = quoteOraclePriceData.price;
+				}
+
+				positionUnrealizedPnl = positionUnrealizedPnl
+					.mul(quotePrice)
+					.div(new BN(PRICE_PRECISION));
+
 				if (withWeightMarginCategory !== undefined) {
 					if (positionUnrealizedPnl.gt(ZERO)) {
 						positionUnrealizedPnl = positionUnrealizedPnl
@@ -558,55 +585,60 @@ export class User {
 			const spotMarketAccount: SpotMarketAccount =
 				this.driftClient.getSpotMarketAccount(spotPosition.marketIndex);
 
+			const oraclePriceData = this.getOracleDataForSpotMarket(
+				spotPosition.marketIndex
+			);
+
 			if (
 				spotPosition.marketIndex === QUOTE_SPOT_MARKET_INDEX &&
 				countForQuote
 			) {
-				if (isVariant(spotPosition.balanceType, 'borrow')) {
-					const tokenAmount = getTokenAmount(
+				const tokenAmount = getSignedTokenAmount(
+					getTokenAmount(
 						spotPosition.scaledBalance,
 						spotMarketAccount,
 						spotPosition.balanceType
-					);
-
-					let weight = SPOT_MARKET_WEIGHT_PRECISION;
-					if (marginCategory === 'Initial') {
-						weight = BN.max(
-							weight,
-							new BN(this.getUserAccount().maxMarginRatio)
-						);
-					}
+					),
+					spotPosition.balanceType
+				);
 
-					const weightedTokenValue = tokenAmount
-						.mul(weight)
-						.div(SPOT_MARKET_WEIGHT_PRECISION);
+				if (isVariant(spotPosition.balanceType, 'borrow')) {
+					const weightedTokenValue = this.getSpotLiabilityValue(
+						tokenAmount,
+						oraclePriceData,
+						spotMarketAccount,
+						marginCategory,
+						liquidationBuffer,
+						strict,
+						now
+					).abs();
 
 					netQuoteValue = netQuoteValue.sub(weightedTokenValue);
-
-					continue;
 				} else {
-					const tokenAmount = getTokenAmount(
-						spotPosition.scaledBalance,
+					const weightedTokenValue = this.getSpotAssetValue(
+						tokenAmount,
+						oraclePriceData,
 						spotMarketAccount,
-						spotPosition.balanceType
+						marginCategory,
+						strict,
+						now
 					);
 
-					netQuoteValue = netQuoteValue.add(tokenAmount);
-
-					continue;
+					netQuoteValue = netQuoteValue.add(weightedTokenValue);
 				}
-			}
 
-			const oraclePriceData = this.getOracleDataForSpotMarket(
-				spotPosition.marketIndex
-			);
+				continue;
+			}
 
 			if (!includeOpenOrders && countForBase) {
 				if (isVariant(spotPosition.balanceType, 'borrow')) {
-					const tokenAmount = getTokenAmount(
-						spotPosition.scaledBalance,
-						spotMarketAccount,
-						spotPosition.balanceType
+					const tokenAmount = getSignedTokenAmount(
+						getTokenAmount(
+							spotPosition.scaledBalance,
+							spotMarketAccount,
+							spotPosition.balanceType
+						),
+						SpotBalanceType.BORROW
 					);
 					const liabilityValue = this.getSpotLiabilityValue(
 						tokenAmount,
@@ -616,7 +648,7 @@ export class User {
 						liquidationBuffer,
 						strict,
 						now
-					);
+					).abs();
 					totalLiabilityValue = totalLiabilityValue.add(liabilityValue);
 
 					continue;
@@ -662,14 +694,14 @@ export class User {
 
 			if (worstCaseTokenAmount.lt(ZERO) && countForBase) {
 				const baseLiabilityValue = this.getSpotLiabilityValue(
-					worstCaseTokenAmount.abs(),
+					worstCaseTokenAmount,
 					oraclePriceData,
 					spotMarketAccount,
 					marginCategory,
 					liquidationBuffer,
 					strict,
 					now
-				);
+				).abs();
 
 				totalLiabilityValue = totalLiabilityValue.add(baseLiabilityValue);
 			}
@@ -738,7 +770,7 @@ export class User {
 	): BN {
 		let liabilityValue = null;
 
-		if (strict && spotMarketAccount.marketIndex != QUOTE_SPOT_MARKET_INDEX) {
+		if (strict) {
 			const estOracleTwap = calculateLiveOracleTwap(
 				spotMarketAccount.historicalOracleData,
 				oraclePriceData,
@@ -809,7 +841,7 @@ export class User {
 		now?: BN
 	): BN {
 		let assetValue = null;
-		if (strict && spotMarketAccount.marketIndex != QUOTE_SPOT_MARKET_INDEX) {
+		if (strict) {
 			const estOracleTwap = calculateLiveOracleTwap(
 				spotMarketAccount.historicalOracleData,
 				oraclePriceData,
@@ -886,10 +918,16 @@ export class User {
 	 * calculates TotalCollateral: collateral + unrealized pnl
 	 * @returns : Precision QUOTE_PRECISION
 	 */
-	public getTotalCollateral(marginCategory: MarginCategory = 'Initial'): BN {
-		return this.getSpotMarketAssetValue(undefined, marginCategory, true).add(
-			this.getUnrealizedPNL(true, undefined, marginCategory)
-		);
+	public getTotalCollateral(
+		marginCategory: MarginCategory = 'Initial',
+		strict = false
+	): BN {
+		return this.getSpotMarketAssetValue(
+			undefined,
+			marginCategory,
+			true,
+			strict
+		).add(this.getUnrealizedPNL(true, undefined, marginCategory, strict));
 	}
 
 	/**
@@ -941,7 +979,8 @@ export class User {
 	getTotalPerpPositionValue(
 		marginCategory?: MarginCategory,
 		liquidationBuffer?: BN,
-		includeOpenOrders?: boolean
+		includeOpenOrders?: boolean,
+		strict = false
 	): BN {
 		return this.getActivePerpPositions().reduce(
 			(totalPerpValue, perpPosition) => {
@@ -1011,7 +1050,27 @@ export class User {
 						marginRatio = ZERO;
 					}
 
+					const quoteSpotMarket = this.driftClient.getSpotMarketAccount(
+						market.quoteSpotMarketIndex
+					);
+					const quoteOraclePriceData =
+						this.driftClient.getOraclePriceDataAndSlot(
+							quoteSpotMarket.oracle
+						).data;
+
+					let quotePrice;
+					if (strict) {
+						quotePrice = BN.max(
+							quoteOraclePriceData.price,
+							quoteSpotMarket.historicalOracleData.lastOraclePriceTwap5Min
+						);
+					} else {
+						quotePrice = quoteOraclePriceData.price;
+					}
+
 					baseAssetValue = baseAssetValue
+						.mul(quotePrice)
+						.div(PRICE_PRECISION)
 						.mul(marginRatio)
 						.div(MARGIN_PRECISION);