@velocity-exchange/sdk 0.0.5 → 0.2.0

package/src/accounts/webSocketVelocityClientAccountSubscriberV2.ts

@@ -7,6 +7,7 @@ import {
 	NotSubscribedError,
 	ResubOpts,
 } from './types';
+import { assertDataAndSlot } from './utils';
 import {
 	isVariant,
 	PerpMarketAccount,
@@ -66,12 +67,38 @@ export class WebSocketVelocityClientAccountSubscriberV2
 
 	eventEmitter: StrictEventEmitter<EventEmitter, VelocityClientAccountEvents>;
 	stateAccountSubscriber?: WebSocketAccountSubscriberV2<StateAccount>;
-	perpMarketAllAccountsSubscriber: WebSocketProgramAccountsSubscriberV2<PerpMarketAccount>;
+	private _perpMarketAllAccountsSubscriber?: WebSocketProgramAccountsSubscriberV2<PerpMarketAccount>;
+	get perpMarketAllAccountsSubscriber(): WebSocketProgramAccountsSubscriberV2<PerpMarketAccount> {
+		if (!this._perpMarketAllAccountsSubscriber) {
+			throw new Error(
+				'perpMarketAllAccountsSubscriber accessed before subscribe()'
+			);
+		}
+		return this._perpMarketAllAccountsSubscriber;
+	}
+	set perpMarketAllAccountsSubscriber(
+		subscriber: WebSocketProgramAccountsSubscriberV2<PerpMarketAccount>
+	) {
+		this._perpMarketAllAccountsSubscriber = subscriber;
+	}
 	perpMarketAccountLatestData = new Map<
 		number,
 		DataAndSlot<PerpMarketAccount>
 	>();
-	spotMarketAllAccountsSubscriber: WebSocketProgramAccountsSubscriberV2<SpotMarketAccount>;
+	private _spotMarketAllAccountsSubscriber?: WebSocketProgramAccountsSubscriberV2<SpotMarketAccount>;
+	get spotMarketAllAccountsSubscriber(): WebSocketProgramAccountsSubscriberV2<SpotMarketAccount> {
+		if (!this._spotMarketAllAccountsSubscriber) {
+			throw new Error(
+				'spotMarketAllAccountsSubscriber accessed before subscribe()'
+			);
+		}
+		return this._spotMarketAllAccountsSubscriber;
+	}
+	set spotMarketAllAccountsSubscriber(
+		subscriber: WebSocketProgramAccountsSubscriberV2<SpotMarketAccount>
+	) {
+		this._spotMarketAllAccountsSubscriber = subscriber;
+	}
 	spotMarketAccountLatestData = new Map<
 		number,
 		DataAndSlot<SpotMarketAccount>
@@ -83,13 +110,13 @@ export class WebSocketVelocityClientAccountSubscriberV2
 	oracleSubscribers = new Map<string, AccountSubscriber<OraclePriceData>>();
 	delistedMarketSetting: DelistedMarketSetting;
 
-	initialPerpMarketAccountData: Map<number, PerpMarketAccount>;
-	initialSpotMarketAccountData: Map<number, SpotMarketAccount>;
-	initialOraclePriceData: Map<string, OraclePriceData>;
+	initialPerpMarketAccountData: Map<number, PerpMarketAccount> = new Map();
+	initialSpotMarketAccountData: Map<number, SpotMarketAccount> = new Map();
+	initialOraclePriceData: Map<string, OraclePriceData> = new Map();
 
 	protected isSubscribing = false;
-	protected subscriptionPromise: Promise<boolean>;
-	protected subscriptionPromiseResolver: (val: boolean) => void;
+	private subscriptionPromiseResolver: (val: boolean) => void = () => {};
+	protected subscriptionPromise: Promise<boolean> = Promise.resolve(false);
 
 	private rpc: Rpc<any>;
 	private rpcSubscriptions: RpcSubscriptions<SolanaRpcSubscriptionsApi> &
@@ -349,9 +376,7 @@ export class WebSocketVelocityClientAccountSubscriberV2
 			this.isSubscribed = true;
 			this.isSubscribing = false;
 			// Before calling subscriptionPromiseResolver, check if it's defined
-			if (this.subscriptionPromiseResolver) {
-				this.subscriptionPromiseResolver(true);
-			}
+			this.subscriptionPromiseResolver(true);
 
 			return true;
 		} catch (error) {
@@ -415,27 +440,34 @@ export class WebSocketVelocityClientAccountSubscriberV2
 			)
 		).flat();
 		this.initialOraclePriceData = new Map(
-			this.oracleInfos.reduce((result, oracleInfo, i) => {
-				if (!oracleAccountInfos[i]) {
-					return result;
-				}
+			this.oracleInfos.reduce(
+				(result, oracleInfo, i) => {
+					const oracleAccountInfo = oracleAccountInfos[i];
+					if (!oracleAccountInfo) {
+						return result;
+					}
 
-				const oracleClient = this.oracleClientCache.get(
-					oracleInfo.source,
-					connection,
-					this.program
-				);
+					const oracleClient = this.oracleClientCache.get(
+						oracleInfo.source,
+						connection,
+						this.program
+					);
+					if (!oracleClient) {
+						return result;
+					}
 
-				const oraclePriceData = oracleClient.getOraclePriceDataFromBuffer(
-					oracleAccountInfos[i].data
-				);
+					const oraclePriceData = oracleClient.getOraclePriceDataFromBuffer(
+						oracleAccountInfo.data
+					);
 
-				result.push([
-					getOracleId(oracleInfo.publicKey, oracleInfo.source),
-					oraclePriceData,
-				]);
-				return result;
-			}, [])
+					result.push([
+						getOracleId(oracleInfo.publicKey, oracleInfo.source),
+						oraclePriceData,
+					]);
+					return result;
+				},
+				[] as [string, OraclePriceData][]
+			)
 		);
 		const oracleSetupEndTime = performance.now();
 		const oracleSetupDuration = oracleSetupEndTime - oracleSetupStartTime;
@@ -503,6 +535,9 @@ export class WebSocketVelocityClientAccountSubscriberV2
 				this.program.provider.connection,
 				this.program
 			);
+			if (!client) {
+				return false;
+			}
 			const accountSubscriber =
 				new WebSocketAccountSubscriberV2<OraclePriceData>(
 					'oracle',
@@ -600,13 +635,13 @@ export class WebSocketVelocityClientAccountSubscriberV2
 			}
 			const perpMarketAccount = perpMarket.data;
 			const perpMarketIndex = perpMarketAccount.marketIndex;
-			const oracle = perpMarketAccount.amm.oracle;
-			const oracleId = getOracleId(oracle, perpMarket.data.amm.oracleSource);
+			const oracle = perpMarketAccount.oracle;
+			const oracleId = getOracleId(oracle, perpMarket.data.oracleSource);
 			if (!this.oracleSubscribers.has(oracleId)) {
 				addOraclePromises.push(
 					this.addOracle({
 						publicKey: oracle,
-						source: perpMarket.data.amm.oracleSource,
+						source: perpMarket.data.oracleSource,
 					})
 				);
 			}
@@ -653,8 +688,9 @@ export class WebSocketVelocityClientAccountSubscriberV2
 
 		for (const oracle of oracles) {
 			const oracleId = getOracleId(oracle.publicKey, oracle.source);
-			if (this.oracleSubscribers.has(oracleId)) {
-				await this.oracleSubscribers.get(oracleId).unsubscribe();
+			const subscriber = this.oracleSubscribers.get(oracleId);
+			if (subscriber) {
+				await subscriber.unsubscribe();
 				if (this.delistedMarketSetting === DelistedMarketSetting.Discard) {
 					this.oracleSubscribers.delete(oracleId);
 				}
@@ -672,7 +708,10 @@ export class WebSocketVelocityClientAccountSubscriberV2
 
 	public getStateAccountAndSlot(): DataAndSlot<StateAccount> {
 		this.assertIsSubscribed();
-		return this.stateAccountSubscriber.dataAndSlot;
+		return assertDataAndSlot(
+			this.stateAccountSubscriber?.dataAndSlot,
+			'State account data not available'
+		);
 	}
 
 	public getMarketAccountAndSlot(
@@ -720,8 +759,8 @@ export class WebSocketVelocityClientAccountSubscriberV2
 			return undefined;
 		}
 
-		if (!perpMarketAccount.data.amm.oracle.equals(oracle)) {
-			// If the oracle has changed, we need to update the oracle map in background
+		if (!oracle || !perpMarketAccount.data.oracle.equals(oracle)) {
+			// If the oracle has changed (or not yet cached), update the oracle map in background
 			this.setPerpOracleMap();
 		}
 
@@ -738,18 +777,11 @@ export class WebSocketVelocityClientAccountSubscriberV2
 			return undefined;
 		}
 
-		if (!spotMarketAccount.data.oracle.equals(oracle)) {
-			// If the oracle has changed, we need to update the oracle map in background
+		if (!oracle || !spotMarketAccount.data.oracle.equals(oracle)) {
+			// If the oracle has changed (or not yet cached), update the oracle map in background
 			this.setSpotOracleMap();
 		}
 
 		return this.getOraclePriceDataAndSlot(oracleId);
 	}
 }
-
-/** @deprecated Use `WebSocketVelocityClientAccountSubscriberV2` instead. `WebSocketDriftClientAccountSubscriberV2` will be removed in a future major. */
-export const WebSocketDriftClientAccountSubscriberV2 =
-	WebSocketVelocityClientAccountSubscriberV2;
-/** @deprecated Use `WebSocketVelocityClientAccountSubscriberV2` instead. `WebSocketDriftClientAccountSubscriberV2` will be removed in a future major. */
-export type WebSocketDriftClientAccountSubscriberV2 =
-	WebSocketVelocityClientAccountSubscriberV2;

package/src/accounts/websocketProgramUserAccountSubscriber.ts

@@ -83,12 +83,7 @@ export class WebSocketProgramUserAccountSubscriber
 		this.isSubscribed = false;
 	}
 
-	getUserAccountAndSlot(): DataAndSlot<UserAccount> {
-		if (!this.userAccountAndSlot) {
-			throw new NotSubscribedError(
-				'Must subscribe before getting user account data'
-			);
-		}
+	getUserAccountAndSlot(): DataAndSlot<UserAccount> | undefined {
 		return this.userAccountAndSlot;
 	}
 }

package/src/addresses/marketAddresses.ts

@@ -7,8 +7,9 @@ export async function getMarketAddress(
 	marketIndex: number
 ): Promise<PublicKey> {
 	const cacheKey = `${programId.toString()}-${marketIndex.toString()}`;
-	if (CACHE.has(cacheKey)) {
-		return CACHE.get(cacheKey);
+	const cached = CACHE.get(cacheKey);
+	if (cached !== undefined) {
+		return cached;
 	}
 
 	const publicKey = await getPerpMarketPublicKey(programId, marketIndex);

package/src/addresses/pda.ts

@@ -1,12 +1,11 @@
 /**
- * PDA derivation helpers for all Drift protocol accounts.
+ * PDA derivation helpers for all Velocity protocol accounts.
  * Covers: User, UserStats, PerpMarket, SpotMarket, InsuranceFundStake,
  * HighLeverageModeConfig, State, and vault PDAs.
  * All functions are pure (no RPC calls).
  */
 import { PublicKey } from '@solana/web3.js';
 import * as anchor from '../isomorphic/anchor';
-import { BN } from '../isomorphic/anchor';
 import {
 	getAssociatedTokenAddress,
 	TOKEN_2022_PROGRAM_ID,
@@ -18,24 +17,17 @@ export async function getVelocityStateAccountPublicKeyAndNonce(
 	programId: PublicKey
 ): Promise<[PublicKey, number]> {
 	return PublicKey.findProgramAddress(
-		[Buffer.from(anchor.utils.bytes.utf8.encode('drift_state'))],
+		[Buffer.from(anchor.utils.bytes.utf8.encode('velocity_state'))],
 		programId
 	);
 }
 
-/** @deprecated Use `getVelocityStateAccountPublicKeyAndNonce` instead. `getDriftStateAccountPublicKeyAndNonce` will be removed in a future major. */
-export const getDriftStateAccountPublicKeyAndNonce =
-	getVelocityStateAccountPublicKeyAndNonce;
-
 export async function getVelocityStateAccountPublicKey(
 	programId: PublicKey
 ): Promise<PublicKey> {
 	return (await getVelocityStateAccountPublicKeyAndNonce(programId))[0];
 }
 
-/** @deprecated Use `getVelocityStateAccountPublicKey` instead. `getDriftStateAccountPublicKey` will be removed in a future major. */
-export const getDriftStateAccountPublicKey = getVelocityStateAccountPublicKey;
-
 export async function getUserAccountPublicKeyAndNonce(
 	programId: PublicKey,
 	authority: PublicKey,
@@ -218,14 +210,11 @@ export function getInsuranceFundStakeAccountPublicKey(
 
 export function getVelocitySignerPublicKey(programId: PublicKey): PublicKey {
 	return PublicKey.findProgramAddressSync(
-		[Buffer.from(anchor.utils.bytes.utf8.encode('drift_signer'))],
+		[Buffer.from(anchor.utils.bytes.utf8.encode('velocity_signer'))],
 		programId
 	)[0];
 }
 
-/** @deprecated Use `getVelocitySignerPublicKey` instead. `getDriftSignerPublicKey` will be removed in a future major. */
-export const getDriftSignerPublicKey = getVelocitySignerPublicKey;
-
 export function getReferrerNamePublicKeySync(
 	programId: PublicKey,
 	nameBuffer: number[]
@@ -239,15 +228,6 @@ export function getReferrerNamePublicKeySync(
 	)[0];
 }
 
-export function getProtocolIfSharesTransferConfigPublicKey(
-	programId: PublicKey
-): PublicKey {
-	return PublicKey.findProgramAddressSync(
-		[Buffer.from(anchor.utils.bytes.utf8.encode('if_shares_transfer_config'))],
-		programId
-	)[0];
-}
-
 export function getPrelaunchOraclePublicKey(
 	programId: PublicKey,
 	marketIndex: number
@@ -287,21 +267,6 @@ export function getTokenProgramForSpotMarket(
 	return TOKEN_PROGRAM_ID;
 }
 
-export function getIfRebalanceConfigPublicKey(
-	programId: PublicKey,
-	inMarketIndex: number,
-	outMarketIndex: number
-): PublicKey {
-	return PublicKey.findProgramAddressSync(
-		[
-			Buffer.from(anchor.utils.bytes.utf8.encode('if_rebalance_config')),
-			new BN(inMarketIndex).toArrayLike(Buffer, 'le', 2),
-			new BN(outMarketIndex).toArrayLike(Buffer, 'le', 2),
-		],
-		programId
-	)[0];
-}
-
 export function getRevenueShareAccountPublicKey(
 	programId: PublicKey,
 	authority: PublicKey