@drift-labs/sdk 2.96.0-beta.2 → 2.96.0-beta.20

package/src/accounts/grpcDriftClientAccountSubscriber.ts

@@ -1,196 +0,0 @@
-import { WebSocketDriftClientAccountSubscriber } from './webSocketDriftClientAccountSubscriber';
-import { OracleInfo, OraclePriceData } from '../oracles/types';
-import { Program } from '@coral-xyz/anchor';
-import { findAllMarketAndOracles } from '../config';
-import {
-	getDriftStateAccountPublicKey,
-	getPerpMarketPublicKey,
-	getSpotMarketPublicKey,
-} from '../addresses/pda';
-import { GrpcConfigs, ResubOpts } from './types';
-import { grpcAccountSubscriber } from './grpcAccountSubscriber';
-import { PerpMarketAccount, SpotMarketAccount, StateAccount } from '../types';
-import { Commitment } from '@solana/web3.js';
-
-export class gprcDriftClientAccountSubscriber extends WebSocketDriftClientAccountSubscriber {
-	private grpcConfigs: GrpcConfigs;
-
-	constructor(
-		grpcConfigs: GrpcConfigs,
-		program: Program,
-		perpMarketIndexes: number[],
-		spotMarketIndexes: number[],
-		oracleInfos: OracleInfo[],
-		shouldFindAllMarketsAndOracles: boolean,
-		resubOpts?: ResubOpts,
-		commitment?: Commitment
-	) {
-		super(
-			program,
-			perpMarketIndexes,
-			spotMarketIndexes,
-			oracleInfos,
-			shouldFindAllMarketsAndOracles,
-			resubOpts,
-			commitment
-		);
-		this.grpcConfigs = grpcConfigs;
-	}
-
-	public async subscribe(): Promise<boolean> {
-		if (this.isSubscribed) {
-			return true;
-		}
-
-		if (this.isSubscribing) {
-			return await this.subscriptionPromise;
-		}
-
-		this.isSubscribing = true;
-
-		this.subscriptionPromise = new Promise((res) => {
-			this.subscriptionPromiseResolver = res;
-		});
-
-		if (this.shouldFindAllMarketsAndOracles) {
-			const {
-				perpMarketIndexes,
-				perpMarketAccounts,
-				spotMarketIndexes,
-				spotMarketAccounts,
-				oracleInfos,
-			} = await findAllMarketAndOracles(this.program);
-			this.perpMarketIndexes = perpMarketIndexes;
-			this.spotMarketIndexes = spotMarketIndexes;
-			this.oracleInfos = oracleInfos;
-			// front run and set the initial data here to save extra gma call in set initial data
-			this.initialPerpMarketAccountData = new Map(
-				perpMarketAccounts.map((market) => [market.marketIndex, market])
-			);
-			this.initialSpotMarketAccountData = new Map(
-				spotMarketAccounts.map((market) => [market.marketIndex, market])
-			);
-		}
-
-		const statePublicKey = await getDriftStateAccountPublicKey(
-			this.program.programId
-		);
-
-		// create and activate main state account subscription
-		this.stateAccountSubscriber = new grpcAccountSubscriber(
-			this.grpcConfigs,
-			'state',
-			this.program,
-			statePublicKey,
-			undefined,
-			undefined
-		);
-		await this.stateAccountSubscriber.subscribe((data: StateAccount) => {
-			this.eventEmitter.emit('stateAccountUpdate', data);
-			this.eventEmitter.emit('update');
-		});
-
-		// set initial data to avoid spamming getAccountInfo calls in webSocketAccountSubscriber
-		await this.setInitialData();
-
-		await Promise.all([
-			// subscribe to market accounts
-			this.subscribeToPerpMarketAccounts(),
-			// subscribe to spot market accounts
-			this.subscribeToSpotMarketAccounts(),
-			// subscribe to oracles
-			this.subscribeToOracles(),
-		]);
-
-		this.eventEmitter.emit('update');
-
-		await Promise.all([this.setPerpOracleMap(), this.setSpotOracleMap()]);
-
-		this.subscriptionPromiseResolver(true);
-
-		this.isSubscribing = false;
-		this.isSubscribed = true;
-
-		// delete initial data
-		this.removeInitialData();
-
-		return true;
-	}
-
-	override async subscribeToSpotMarketAccount(
-		marketIndex: number
-	): Promise<boolean> {
-		const marketPublicKey = await getSpotMarketPublicKey(
-			this.program.programId,
-			marketIndex
-		);
-		const accountSubscriber = new grpcAccountSubscriber<SpotMarketAccount>(
-			this.grpcConfigs,
-			'spotMarket',
-			this.program,
-			marketPublicKey,
-			undefined,
-			this.resubOpts
-		);
-		accountSubscriber.setData(
-			this.initialSpotMarketAccountData.get(marketIndex)
-		);
-		await accountSubscriber.subscribe((data: SpotMarketAccount) => {
-			this.eventEmitter.emit('spotMarketAccountUpdate', data);
-			this.eventEmitter.emit('update');
-		});
-		this.spotMarketAccountSubscribers.set(marketIndex, accountSubscriber);
-		return true;
-	}
-
-	async subscribeToPerpMarketAccount(marketIndex: number): Promise<boolean> {
-		const perpMarketPublicKey = await getPerpMarketPublicKey(
-			this.program.programId,
-			marketIndex
-		);
-		const accountSubscriber = new grpcAccountSubscriber<PerpMarketAccount>(
-			this.grpcConfigs,
-			'perpMarket',
-			this.program,
-			perpMarketPublicKey,
-			undefined,
-			this.resubOpts
-		);
-		accountSubscriber.setData(
-			this.initialPerpMarketAccountData.get(marketIndex)
-		);
-		await accountSubscriber.subscribe((data: PerpMarketAccount) => {
-			this.eventEmitter.emit('perpMarketAccountUpdate', data);
-			this.eventEmitter.emit('update');
-		});
-		this.perpMarketAccountSubscribers.set(marketIndex, accountSubscriber);
-		return true;
-	}
-
-	async subscribeToOracle(oracleInfo: OracleInfo): Promise<boolean> {
-		const oracleString = oracleInfo.publicKey.toString();
-		const client = this.oracleClientCache.get(
-			oracleInfo.source,
-			this.program.provider.connection,
-			this.program
-		);
-		const accountSubscriber = new grpcAccountSubscriber<OraclePriceData>(
-			this.grpcConfigs,
-			'oracle',
-			this.program,
-			oracleInfo.publicKey,
-			(buffer: Buffer) => {
-				return client.getOraclePriceDataFromBuffer(buffer);
-			},
-			this.resubOpts
-		);
-		accountSubscriber.setData(this.initialOraclePriceData.get(oracleString));
-		await accountSubscriber.subscribe((data: OraclePriceData) => {
-			this.eventEmitter.emit('oraclePriceUpdate', oracleInfo.publicKey, data);
-			this.eventEmitter.emit('update');
-		});
-
-		this.oracleSubscribers.set(oracleString, accountSubscriber);
-		return true;
-	}
-}

package/src/accounts/grpcInsuranceFundStakeAccountSubscriber.ts

@@ -1,62 +0,0 @@
-import { GrpcConfigs } from './types';
-import { Program } from '@coral-xyz/anchor';
-import { Commitment, PublicKey } from '@solana/web3.js';
-import { InsuranceFundStake } from '../types';
-import { WebSocketInsuranceFundStakeAccountSubscriber } from './webSocketInsuranceFundStakeAccountSubscriber';
-import { grpcAccountSubscriber } from './grpcAccountSubscriber';
-
-export class grpcInsuranceFundStakeAccountSubscriber extends WebSocketInsuranceFundStakeAccountSubscriber {
-	private grpcConfigs: GrpcConfigs;
-
-	public constructor(
-		grpcConfigs: GrpcConfigs,
-		program: Program,
-		insuranceFundStakeAccountPublicKey: PublicKey,
-		resubTimeoutMs?: number,
-		commitment?: Commitment
-	) {
-		super(
-			program,
-			insuranceFundStakeAccountPublicKey,
-			resubTimeoutMs,
-			commitment
-		);
-		this.grpcConfigs = grpcConfigs;
-	}
-
-	async subscribe(
-		insuranceFundStakeAccount?: InsuranceFundStake
-	): Promise<boolean> {
-		if (this.isSubscribed) {
-			return true;
-		}
-
-		this.insuranceFundStakeDataAccountSubscriber = new grpcAccountSubscriber(
-			this.grpcConfigs,
-			'insuranceFundStake',
-			this.program,
-			this.insuranceFundStakeAccountPublicKey,
-			undefined,
-			{
-				resubTimeoutMs: this.resubTimeoutMs,
-			}
-		);
-
-		if (insuranceFundStakeAccount) {
-			this.insuranceFundStakeDataAccountSubscriber.setData(
-				insuranceFundStakeAccount
-			);
-		}
-
-		await this.insuranceFundStakeDataAccountSubscriber.subscribe(
-			(data: InsuranceFundStake) => {
-				this.eventEmitter.emit('insuranceFundStakeAccountUpdate', data);
-				this.eventEmitter.emit('update');
-			}
-		);
-
-		this.eventEmitter.emit('update');
-		this.isSubscribed = true;
-		return true;
-	}
-}

package/src/accounts/grpcProgramAccountSubscriber.ts

@@ -1,181 +0,0 @@
-import { ResubOpts, GrpcConfigs } from './types';
-import { Program } from '@coral-xyz/anchor';
-import { Commitment, Context, MemcmpFilter, PublicKey } from '@solana/web3.js';
-import * as Buffer from 'buffer';
-import { WebSocketProgramAccountSubscriber } from './webSocketProgramAccountSubscriber';
-import { ClientDuplexStream } from '@grpc/grpc-js';
-import Client, {
-	SubscribeRequest,
-	SubscribeUpdate,
-	CommitmentLevel,
-} from '@triton-one/yellowstone-grpc';
-
-export class grpcProgramAccountSubscriber<
-	T,
-> extends WebSocketProgramAccountSubscriber<T> {
-	client: Client;
-	stream: ClientDuplexStream<SubscribeRequest, SubscribeUpdate>;
-	commitmentLevel: CommitmentLevel;
-	listenerId?: number;
-
-	public constructor(
-		grpcConfigs: GrpcConfigs,
-		subscriptionName: string,
-		accountDiscriminator: string,
-		program: Program,
-		decodeBufferFn: (accountName: string, ix: Buffer) => T,
-		options: { filters: MemcmpFilter[]; commitment?: Commitment } = {
-			filters: [],
-		},
-		resubOpts?: ResubOpts
-	) {
-		super(
-			subscriptionName,
-			accountDiscriminator,
-			program,
-			decodeBufferFn,
-			options,
-			resubOpts
-		);
-		this.client = new Client(
-			grpcConfigs.endpoint,
-			grpcConfigs.token,
-			grpcConfigs.channelOptions ?? {}
-		);
-		this.commitmentLevel =
-			grpcConfigs.commitmentLevel ?? CommitmentLevel.CONFIRMED;
-	}
-
-	async subscribe(
-		onChange: (
-			accountId: PublicKey,
-			data: T,
-			context: Context,
-			buffer: Buffer
-		) => void
-	): Promise<void> {
-		if (this.listenerId != null || this.isUnsubscribing) {
-			return;
-		}
-
-		this.onChange = onChange;
-
-		// Subscribe with grpc
-		this.stream = await this.client.subscribe();
-		const request: SubscribeRequest = {
-			slots: {
-				slots: {},
-			},
-			accounts: {
-				drift: {
-					account: [],
-					owner: [this.program.programId.toBase58()],
-					// @ts-ignore
-					filters: this.options.filters,
-				},
-			},
-			transactions: {},
-			blocks: {},
-			blocksMeta: {},
-			accountsDataSlice: [],
-			commitment: this.commitmentLevel,
-			entry: {},
-			transactionsStatus: {},
-		};
-		this.stream.on('data', (chunk: SubscribeUpdate) => {
-			if (!chunk.account) {
-				return;
-			}
-			const slot = Number(chunk.account.slot);
-			const accountInfo = {
-				owner: new PublicKey(chunk.account.account.owner),
-				lamports: Number(chunk.account.account.lamports),
-				data: Buffer.Buffer.from(chunk.account.account.data),
-				executable: chunk.account.account.executable,
-				rentEpoch: Number(chunk.account.account.rentEpoch),
-			};
-
-			if (this.resubOpts?.resubTimeoutMs) {
-				this.receivingData = true;
-				clearTimeout(this.timeoutId);
-				this.handleRpcResponse(
-					{
-						slot,
-					},
-					{
-						accountId: new PublicKey(chunk.account.account.pubkey),
-						accountInfo,
-					}
-				);
-				this.setTimeout();
-			} else {
-				this.handleRpcResponse(
-					{
-						slot,
-					},
-					{
-						accountId: new PublicKey(chunk.account.account.pubkey),
-						accountInfo,
-					}
-				);
-			}
-		});
-
-		return new Promise<void>((resolve, reject) => {
-			this.stream.write(request, (err) => {
-				if (err === null || err === undefined) {
-					this.listenerId = 1;
-					if (this.resubOpts?.resubTimeoutMs) {
-						this.receivingData = true;
-						this.setTimeout();
-					}
-					resolve();
-				} else {
-					reject(err);
-				}
-			});
-		}).catch((reason) => {
-			console.error(reason);
-			throw reason;
-		});
-	}
-
-	public async unsubscribe(onResub = false): Promise<void> {
-		if (!onResub && this.resubOpts) {
-			this.resubOpts.resubTimeoutMs = undefined;
-		}
-		this.isUnsubscribing = true;
-		clearTimeout(this.timeoutId);
-		this.timeoutId = undefined;
-
-		if (this.listenerId != null) {
-			const promise = new Promise<void>((resolve, reject) => {
-				const request: SubscribeRequest = {
-					slots: {},
-					accounts: {},
-					transactions: {},
-					blocks: {},
-					blocksMeta: {},
-					accountsDataSlice: [],
-					entry: {},
-					transactionsStatus: {},
-				};
-				this.stream.write(request, (err) => {
-					if (err === null || err === undefined) {
-						this.listenerId = undefined;
-						this.isUnsubscribing = false;
-						resolve();
-					} else {
-						reject(err);
-					}
-				});
-			}).catch((reason) => {
-				console.error(reason);
-				throw reason;
-			});
-			return promise;
-		} else {
-			this.isUnsubscribing = false;
-		}
-	}
-}

package/src/accounts/grpcUserAccountSubscriber.ts

@@ -1,48 +0,0 @@
-import { ResubOpts, GrpcConfigs } from './types';
-import { Program } from '@coral-xyz/anchor';
-import { PublicKey } from '@solana/web3.js';
-import { UserAccount } from '../types';
-import { WebSocketUserAccountSubscriber } from './webSocketUserAccountSubscriber';
-import { grpcAccountSubscriber } from './grpcAccountSubscriber';
-
-export class grpcUserAccountSubscriber extends WebSocketUserAccountSubscriber {
-	private grpcConfigs: GrpcConfigs;
-
-	public constructor(
-		grpcConfigs: GrpcConfigs,
-		program: Program,
-		userAccountPublicKey: PublicKey,
-		resubOpts?: ResubOpts
-	) {
-		super(program, userAccountPublicKey, resubOpts);
-		this.grpcConfigs = grpcConfigs;
-	}
-
-	async subscribe(userAccount?: UserAccount): Promise<boolean> {
-		if (this.isSubscribed) {
-			return true;
-		}
-
-		this.userDataAccountSubscriber = new grpcAccountSubscriber(
-			this.grpcConfigs,
-			'user',
-			this.program,
-			this.userAccountPublicKey,
-			undefined,
-			this.resubOpts
-		);
-
-		if (userAccount) {
-			this.userDataAccountSubscriber.setData(userAccount);
-		}
-
-		await this.userDataAccountSubscriber.subscribe((data: UserAccount) => {
-			this.eventEmitter.emit('userAccountUpdate', data);
-			this.eventEmitter.emit('update');
-		});
-
-		this.eventEmitter.emit('update');
-		this.isSubscribed = true;
-		return true;
-	}
-}

package/src/accounts/grpcUserStatsAccountSubscriber.ts

@@ -1,51 +0,0 @@
-import { ResubOpts, GrpcConfigs } from './types';
-import { Program } from '@coral-xyz/anchor';
-import { Commitment, PublicKey } from '@solana/web3.js';
-import { UserStatsAccount } from '../types';
-import { WebSocketUserStatsAccountSubscriber } from './webSocketUserStatsAccountSubsriber';
-import { grpcAccountSubscriber } from './grpcAccountSubscriber';
-
-export class grpcUserStatsAccountSubscriber extends WebSocketUserStatsAccountSubscriber {
-	private grpcConfigs: GrpcConfigs;
-
-	public constructor(
-		grpcConfigs: GrpcConfigs,
-		program: Program,
-		userStatsAccountPublicKey: PublicKey,
-		resubOpts?: ResubOpts,
-		commitment?: Commitment
-	) {
-		super(program, userStatsAccountPublicKey, resubOpts, commitment);
-		this.grpcConfigs = grpcConfigs;
-	}
-
-	async subscribe(userStatsAccount?: UserStatsAccount): Promise<boolean> {
-		if (this.isSubscribed) {
-			return true;
-		}
-
-		this.userStatsAccountSubscriber = new grpcAccountSubscriber(
-			this.grpcConfigs,
-			'userStats',
-			this.program,
-			this.userStatsAccountPublicKey,
-			undefined,
-			this.resubOpts
-		);
-
-		if (userStatsAccount) {
-			this.userStatsAccountSubscriber.setData(userStatsAccount);
-		}
-
-		await this.userStatsAccountSubscriber.subscribe(
-			(data: UserStatsAccount) => {
-				this.eventEmitter.emit('userStatsAccountUpdate', data);
-				this.eventEmitter.emit('update');
-			}
-		);
-
-		this.eventEmitter.emit('update');
-		this.isSubscribed = true;
-		return true;
-	}
-}

package/src/orderSubscriber/grpcSubscription.ts

@@ -1,126 +0,0 @@
-import { Commitment, Context, PublicKey } from '@solana/web3.js';
-import { Buffer } from 'buffer';
-import { grpcProgramAccountSubscriber } from '../accounts/grpcProgramAccountSubscriber';
-import { OrderSubscriber } from './OrderSubscriber';
-import { GrpcConfigs, ResubOpts } from '../accounts/types';
-import { UserAccount } from '../types';
-import { getNonIdleUserFilter, getUserFilter } from '../memcmp';
-
-export class grpcSubscription {
-	private orderSubscriber: OrderSubscriber;
-	private commitment: Commitment;
-	private skipInitialLoad: boolean;
-	private resubOpts?: ResubOpts;
-	private resyncIntervalMs?: number;
-
-	private subscriber?: grpcProgramAccountSubscriber<UserAccount>;
-	private resyncTimeoutId?: NodeJS.Timeout;
-
-	private decoded?: boolean;
-
-	private grpcConfigs: GrpcConfigs;
-
-	constructor({
-		grpcConfigs,
-		orderSubscriber,
-		commitment,
-		skipInitialLoad = false,
-		resubOpts,
-		resyncIntervalMs,
-		decoded = true,
-	}: {
-		grpcConfigs: GrpcConfigs;
-		orderSubscriber: OrderSubscriber;
-		commitment: Commitment;
-		skipInitialLoad?: boolean;
-		resubOpts?: ResubOpts;
-		resyncIntervalMs?: number;
-		decoded?: boolean;
-	}) {
-		this.orderSubscriber = orderSubscriber;
-		this.commitment = commitment;
-		this.skipInitialLoad = skipInitialLoad;
-		this.resubOpts = resubOpts;
-		this.resyncIntervalMs = resyncIntervalMs;
-		this.decoded = decoded;
-		this.grpcConfigs = grpcConfigs;
-	}
-
-	public async subscribe(): Promise<void> {
-		if (this.subscriber) {
-			return;
-		}
-
-		this.subscriber = new grpcProgramAccountSubscriber<UserAccount>(
-			this.grpcConfigs,
-			'OrderSubscriber',
-			'User',
-			this.orderSubscriber.driftClient.program,
-			this.orderSubscriber.decodeFn,
-			{
-				filters: [getUserFilter(), getNonIdleUserFilter()],
-				commitment: this.commitment,
-			},
-			this.resubOpts
-		);
-
-		await this.subscriber.subscribe(
-			(
-				accountId: PublicKey,
-				account: UserAccount,
-				context: Context,
-				buffer: Buffer
-			) => {
-				const userKey = accountId.toBase58();
-				if (this.decoded ?? true) {
-					this.orderSubscriber.tryUpdateUserAccount(
-						userKey,
-						'decoded',
-						account,
-						context.slot
-					);
-				} else {
-					this.orderSubscriber.tryUpdateUserAccount(
-						userKey,
-						'buffer',
-						buffer,
-						context.slot
-					);
-				}
-			}
-		);
-
-		if (!this.skipInitialLoad) {
-			await this.orderSubscriber.fetch();
-		}
-
-		if (this.resyncIntervalMs) {
-			const recursiveResync = () => {
-				this.resyncTimeoutId = setTimeout(() => {
-					this.orderSubscriber
-						.fetch()
-						.catch((e) => {
-							console.error('Failed to resync in OrderSubscriber');
-							console.log(e);
-						})
-						.finally(() => {
-							// eslint-disable-next-line @typescript-eslint/no-unused-vars
-							if (!this.resyncTimeoutId) return;
-							recursiveResync();
-						});
-				}, this.resyncIntervalMs);
-			};
-			recursiveResync();
-		}
-	}
-
-	public async unsubscribe(): Promise<void> {
-		if (!this.subscriber) return;
-		await this.subscriber.unsubscribe();
-		this.subscriber = undefined;
-		if (this.resyncTimeoutId !== undefined) {
-			clearTimeout(this.resyncTimeoutId);
-			this.resyncTimeoutId = undefined;
-		}
-	}
-}