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

package/lib/types.d.ts

@@ -149,6 +149,9 @@ export declare class OracleSource {
     static readonly PYTH_STABLE_COIN: {
         pythStableCoin: {};
     };
+    static readonly Prelaunch: {
+        prelaunch: {};
+    };
 }
 export declare class OrderType {
     static readonly LIMIT: {
@@ -1084,6 +1087,14 @@ export type OracleGuardRails = {
         tooVolatileRatio: BN;
     };
 };
+export type PrelaunchOracle = {
+    price: BN;
+    maxPrice: BN;
+    confidence: BN;
+    ammLastUpdateSlot: BN;
+    lastUpdateSlot: BN;
+    perpMarketIndex: number;
+};
 export type MarginCategory = 'Initial' | 'Maintenance';
 export type InsuranceFundStake = {
     costBasis: BN;

package/lib/types.js

@@ -98,6 +98,7 @@ OracleSource.PYTH_1M = { pyth1M: {} };
 // static readonly SWITCHBOARD = { switchboard: {} };
 OracleSource.QUOTE_ASSET = { quoteAsset: {} };
 OracleSource.PYTH_STABLE_COIN = { pythStableCoin: {} };
+OracleSource.Prelaunch = { prelaunch: {} };
 class OrderType {
 }
 exports.OrderType = OrderType;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.71.0-beta.0",
+	"version": "2.71.0-beta.2",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/accounts/pollingDriftClientAccountSubscriber.ts

@@ -257,7 +257,8 @@ export class PollingDriftClientAccountSubscriber
 	async addOracleToAccountLoader(oracleToPoll: OraclesToPoll): Promise<void> {
 		const oracleClient = this.oracleClientCache.get(
 			oracleToPoll.source,
-			this.program.provider.connection
+			this.program.provider.connection,
+			this.program
 		);
 
 		oracleToPoll.callbackId = await this.accountLoader.addAccount(
@@ -316,7 +317,8 @@ export class PollingDriftClientAccountSubscriber
 			if (buffer) {
 				const oracleClient = this.oracleClientCache.get(
 					oracleToPoll.source,
-					this.program.provider.connection
+					this.program.provider.connection,
+					this.program
 				);
 				const oraclePriceData =
 					oracleClient.getOraclePriceDataFromBuffer(buffer);

package/src/accounts/webSocketDriftClientAccountSubscriber.ts

@@ -204,7 +204,8 @@ export class WebSocketDriftClientAccountSubscriber
 	async subscribeToOracle(oracleInfo: OracleInfo): Promise<boolean> {
 		const client = this.oracleClientCache.get(
 			oracleInfo.source,
-			this.program.provider.connection
+			this.program.provider.connection,
+			this.program
 		);
 		const accountSubscriber = new WebSocketAccountSubscriber<OraclePriceData>(
 			'oracle',

package/src/addresses/pda.ts

@@ -223,3 +223,16 @@ export function getProtocolIfSharesTransferConfigPublicKey(
 		programId
 	)[0];
 }
+
+export function getPrelaunchOraclePublicKey(
+	programId: PublicKey,
+	marketIndex: number
+): PublicKey {
+	return PublicKey.findProgramAddressSync(
+		[
+			Buffer.from(anchor.utils.bytes.utf8.encode('prelaunch_oracle')),
+			new anchor.BN(marketIndex).toArrayLike(Buffer, 'le', 2),
+		],
+		programId
+	)[0];
+}

package/src/adminClient.ts

@@ -26,6 +26,7 @@ import {
 	getSerumFulfillmentConfigPublicKey,
 	getPhoenixFulfillmentConfigPublicKey,
 	getProtocolIfSharesTransferConfigPublicKey,
+	getPrelaunchOraclePublicKey,
 } from './addresses/pda';
 import { squareRootBN } from './math/utils';
 import { TOKEN_PROGRAM_ID } from '@solana/spl-token';
@@ -1794,4 +1795,50 @@ export class AdminClient extends DriftClient {
 			}
 		);
 	}
+
+	public async initializePrelaunchOracle(
+		perpMarketIndex: number,
+		price?: BN,
+		maxPrice?: BN
+	): Promise<TransactionSignature> {
+		const params = {
+			perpMarketIndex,
+			price: price || null,
+			maxPrice: maxPrice || null,
+		};
+		return await this.program.rpc.initializePrelaunchOracle(params, {
+			accounts: {
+				admin: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				prelaunchOracle: await getPrelaunchOraclePublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+				rent: SYSVAR_RENT_PUBKEY,
+				systemProgram: anchor.web3.SystemProgram.programId,
+			},
+		});
+	}
+
+	public async updatePrelaunchOracleParams(
+		perpMarketIndex: number,
+		price?: BN,
+		maxPrice?: BN
+	): Promise<TransactionSignature> {
+		const params = {
+			perpMarketIndex,
+			price: price || null,
+			maxPrice: maxPrice || null,
+		};
+		return await this.program.rpc.updatePrelaunchOracleParams(params, {
+			accounts: {
+				admin: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				prelaunchOracle: await getPrelaunchOraclePublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+			},
+		});
+	}
 }

package/src/constants/perpMarkets.ts

@@ -244,16 +244,16 @@ export const DevnetPerpMarkets: PerpMarketConfig[] = [
 		launchTs: 1704209558000,
 		oracleSource: OracleSource.PYTH,
 	},
-	// {
-	// 	fullName: 'WIF',
-	// 	category: ['Meme', 'Dog'],
-	// 	symbol: 'WIF-PERP',
-	// 	baseAssetSymbol: 'WIF',
-	// 	marketIndex: 23,
-	// 	oracle: new PublicKey('5i1sz2QQjCQt9PnhuPvqbiYUAYCgjdRnza1JbiH2qRCo'),
-	// 	launchTs: 1706219971000,
-	// 	oracleSource: OracleSource.PYTH,
-	// },
+	{
+		fullName: 'Wormhole',
+		category: ['Bridge'],
+		symbol: 'W-PERP',
+		baseAssetSymbol: 'W',
+		marketIndex: 23,
+		oracle: new PublicKey('BM2UWqREbt7ktsPCA438dqAVqhU7UZFVg11CQyPXFr49'),
+		launchTs: 1709852537000,
+		oracleSource: OracleSource.Prelaunch,
+	},
 ];
 
 export const MainnetPerpMarkets: PerpMarketConfig[] = [

package/src/driftClient.ts

@@ -1566,10 +1566,12 @@ export class DriftClient {
 			isSigner: false,
 			isWritable: writable,
 		});
+		const oracleWritable =
+			writable && isVariant(perpMarketAccount.amm.oracleSource, 'prelaunch');
 		oracleAccountMap.set(perpMarketAccount.amm.oracle.toString(), {
 			pubkey: perpMarketAccount.amm.oracle,
 			isSigner: false,
-			isWritable: false,
+			isWritable: oracleWritable,
 		});
 		this.addSpotMarketToRemainingAccountMaps(
 			perpMarketAccount.quoteSpotMarketIndex,
@@ -5787,6 +5789,39 @@ export class DriftClient {
 		});
 	}
 
+	public async updatePrelaunchOracle(
+		perpMarketIndex: number,
+		txParams?: TxParams
+	): Promise<TransactionSignature> {
+		const { txSig } = await this.sendTransaction(
+			await this.buildTransaction(
+				await this.getUpdatePrelaunchOracleIx(perpMarketIndex),
+				txParams
+			),
+			[],
+			this.opts
+		);
+		return txSig;
+	}
+
+	public async getUpdatePrelaunchOracleIx(
+		perpMarketIndex: number
+	): Promise<TransactionInstruction> {
+		const perpMarket = this.getPerpMarketAccount(perpMarketIndex);
+
+		if (!isVariant(perpMarket.amm.oracleSource, 'prelaunch')) {
+			throw new Error(`Wrong oracle source ${perpMarket.amm.oracleSource}`);
+		}
+
+		return await this.program.instruction.updatePrelaunchOracle({
+			accounts: {
+				state: await this.getStatePublicKey(),
+				perpMarket: perpMarket.pubkey,
+				oracle: perpMarket.amm.oracle,
+			},
+		});
+	}
+
 	public async updatePerpBidAskTwap(
 		perpMarketIndex: number,
 		makers: [PublicKey, PublicKey][],

package/src/factory/oracleClient.ts

@@ -4,11 +4,13 @@ import { OracleClient } from '../oracles/types';
 import { PythClient } from '../oracles/pythClient';
 // import { SwitchboardClient } from '../oracles/switchboardClient';
 import { QuoteAssetOracleClient } from '../oracles/quoteAssetOracleClient';
-import { BN } from '@coral-xyz/anchor';
+import { BN, Program } from '@coral-xyz/anchor';
+import { PrelaunchOracleClient } from '../oracles/prelaunchOracleClient';
 
 export function getOracleClient(
 	oracleSource: OracleSource,
-	connection: Connection
+	connection: Connection,
+	program: Program
 ): OracleClient {
 	if (isVariant(oracleSource, 'pyth')) {
 		return new PythClient(connection);
@@ -30,6 +32,10 @@ export function getOracleClient(
 	// 	return new SwitchboardClient(connection);
 	// }
 
+	if (isVariant(oracleSource, 'prelaunch')) {
+		return new PrelaunchOracleClient(connection, program);
+	}
+
 	if (isVariant(oracleSource, 'quoteAsset')) {
 		return new QuoteAssetOracleClient();
 	}