@drift-labs/sdk 2.67.0-beta.0 → 2.67.0-beta.2

package/src/oracles/switchboardClient.ts

@@ -0,0 +1,74 @@
+import { Connection, PublicKey } from '@solana/web3.js';
+import { PRICE_PRECISION, TEN } from '../constants/numericConstants';
+import { OracleClient, OraclePriceData } from './types';
+import switchboardV2Idl from '../idl/switchboard.json';
+import { BorshAccountsCoder, BN, Idl } from '@coral-xyz/anchor';
+
+type SwitchboardDecimal = {
+	scale: number;
+	mantissa: BN;
+};
+
+type AggregatorAccountData = {
+	latestConfirmedRound: {
+		result: SwitchboardDecimal;
+		stdDeviation: SwitchboardDecimal;
+		numSuccess: number;
+		roundOpenSlot: BN;
+	};
+	minOracleResults: number;
+};
+
+export class SwitchboardClient implements OracleClient {
+	connection: Connection;
+	coder: BorshAccountsCoder;
+
+	public constructor(connection: Connection) {
+		this.connection = connection;
+		this.coder = new BorshAccountsCoder(switchboardV2Idl as Idl);
+	}
+
+	public async getOraclePriceData(
+		pricePublicKey: PublicKey
+	): Promise<OraclePriceData> {
+		const accountInfo = await this.connection.getAccountInfo(pricePublicKey);
+		return this.getOraclePriceDataFromBuffer(accountInfo.data);
+	}
+
+	public getOraclePriceDataFromBuffer(buffer: Buffer): OraclePriceData {
+		const aggregatorAccountData = this.coder.decodeUnchecked(
+			'AggregatorAccountData',
+			buffer
+		) as AggregatorAccountData;
+
+		const price = convertSwitchboardDecimal(
+			aggregatorAccountData.latestConfirmedRound.result
+		);
+
+		const confidence = convertSwitchboardDecimal(
+			aggregatorAccountData.latestConfirmedRound.stdDeviation
+		);
+
+		const hasSufficientNumberOfDataPoints =
+			aggregatorAccountData.latestConfirmedRound.numSuccess >=
+			aggregatorAccountData.minOracleResults;
+
+		const slot: BN = aggregatorAccountData.latestConfirmedRound.roundOpenSlot;
+		return {
+			price,
+			slot,
+			confidence,
+			hasSufficientNumberOfDataPoints,
+		};
+	}
+}
+
+function convertSwitchboardDecimal(switchboardDecimal: {
+	scale: number;
+	mantissa: BN;
+}): BN {
+	const switchboardPrecision = TEN.pow(new BN(switchboardDecimal.scale));
+	return switchboardDecimal.mantissa
+		.mul(PRICE_PRECISION)
+		.div(switchboardPrecision);
+}

package/src/types.ts

@@ -93,7 +93,7 @@ export class OracleSource {
 	static readonly PYTH = { pyth: {} };
 	static readonly PYTH_1K = { pyth1K: {} };
 	static readonly PYTH_1M = { pyth1M: {} };
-	// static readonly SWITCHBOARD = { switchboard: {} };
+	static readonly SWITCHBOARD = { switchboard: {} };
 	static readonly QUOTE_ASSET = { quoteAsset: {} };
 	static readonly PYTH_STABLE_COIN = { pythStableCoin: {} };
 }

package/src/user.ts

@@ -1307,9 +1307,9 @@ export class User {
 			const quoteSpotMarket = this.driftClient.getSpotMarketAccount(
 				market.quoteSpotMarketIndex
 			);
-			const quoteOraclePriceData = this.driftClient.getOraclePriceDataAndSlot(
-				quoteSpotMarket.oracle
-			).data;
+			const quoteOraclePriceData = this.driftClient.getOracleDataForSpotMarket(
+				QUOTE_SPOT_MARKET_INDEX
+			);
 
 			let quotePrice;
 			if (strict) {
@@ -2205,7 +2205,7 @@ export class User {
 		if (signedTokenAmount.gt(ZERO)) {
 			const assetWeight = calculateAssetWeight(
 				signedTokenAmount,
-				this.driftClient.getOraclePriceDataAndSlot(market.oracle).data.price,
+				this.driftClient.getOracleDataForSpotMarket(market.marketIndex).price,
 				market,
 				marginCategory
 			);
@@ -2388,9 +2388,8 @@ export class User {
 		currentSpotMarketNetValue?: BN
 	): BN {
 		const market = this.driftClient.getSpotMarketAccount(targetMarketIndex);
-		const oraclePrice = this.driftClient.getOraclePriceDataAndSlot(
-			market.oracle
-		).data.price;
+		const oraclePrice =
+			this.driftClient.getOracleDataForSpotMarket(targetMarketIndex).price;
 
 		currentQuoteAssetValue = this.getSpotMarketAssetValue(
 			QUOTE_SPOT_MARKET_INDEX
@@ -3401,9 +3400,9 @@ export class User {
 			const perpMarket = this.driftClient.getPerpMarketAccount(
 				perpPosition.marketIndex
 			);
-			const oraclePriceData = this.driftClient.getOraclePriceDataAndSlot(
-				perpMarket.amm.oracle
-			).data;
+			const oraclePriceData = this.driftClient.getOracleDataForPerpMarket(
+				perpMarket.marketIndex
+			);
 			const oraclePrice = oraclePriceData.price;
 			const worstCaseBaseAmount =
 				calculateWorstCaseBaseAssetAmount(settledLpPosition);
@@ -3420,9 +3419,9 @@ export class User {
 			const quoteSpotMarket = this.driftClient.getSpotMarketAccount(
 				perpMarket.quoteSpotMarketIndex
 			);
-			const quoteOraclePriceData = this.driftClient.getOraclePriceDataAndSlot(
-				quoteSpotMarket.oracle
-			).data;
+			const quoteOraclePriceData = this.driftClient.getOracleDataForSpotMarket(
+				QUOTE_SPOT_MARKET_INDEX
+			);
 
 			const baseAssetValue = worstCaseBaseAmount
 				.abs()
@@ -3645,20 +3644,10 @@ export class User {
 	}
 
 	private getOracleDataForPerpMarket(marketIndex: number): OraclePriceData {
-		const oracleKey =
-			this.driftClient.getPerpMarketAccount(marketIndex).amm.oracle;
-		const oracleData =
-			this.driftClient.getOraclePriceDataAndSlot(oracleKey).data;
-
-		return oracleData;
+		return this.driftClient.getOracleDataForPerpMarket(marketIndex);
 	}
 
 	private getOracleDataForSpotMarket(marketIndex: number): OraclePriceData {
-		const oracleKey = this.driftClient.getSpotMarketAccount(marketIndex).oracle;
-
-		const oracleData =
-			this.driftClient.getOraclePriceDataAndSlot(oracleKey).data;
-
-		return oracleData;
+		return this.driftClient.getOracleDataForSpotMarket(marketIndex);
 	}
 }