@velocity-exchange/sdk 0.3.0 → 0.6.0

package/src/userConfig.ts

@@ -1,38 +0,0 @@
-import { VelocityClient } from './velocityClient';
-import { Commitment, PublicKey } from '@solana/web3.js';
-import { BulkAccountLoader } from './accounts/bulkAccountLoader';
-import { GrpcConfigs, UserAccountSubscriber } from './accounts/types';
-import { WebSocketProgramAccountSubscriber } from './accounts/webSocketProgramAccountSubscriber';
-import { UserAccount } from './types';
-import { grpcMultiUserAccountSubscriber } from './accounts/grpcMultiUserAccountSubscriber';
-
-type UserConfigBase = {
-	accountSubscription?: UserSubscriptionConfig;
-	userAccountPublicKey: PublicKey;
-};
-
-export type UserConfig = UserConfigBase & { velocityClient: VelocityClient };
-
-export type UserSubscriptionConfig =
-	| {
-			type: 'grpc';
-			resubTimeoutMs?: number;
-			logResubMessages?: boolean;
-			grpcConfigs: GrpcConfigs;
-			grpcMultiUserAccountSubscriber?: grpcMultiUserAccountSubscriber;
-	  }
-	| {
-			type: 'websocket';
-			resubTimeoutMs?: number;
-			logResubMessages?: boolean;
-			commitment?: Commitment;
-			programUserAccountSubscriber?: WebSocketProgramAccountSubscriber<UserAccount>;
-	  }
-	| {
-			type: 'polling';
-			accountLoader: BulkAccountLoader;
-	  }
-	| {
-			type: 'custom';
-			userAccountSubscriber: UserAccountSubscriber;
-	  };

package/src/userMap/PollingSubscription.ts

@@ -1,47 +0,0 @@
-import { UserMap } from './userMap';
-
-export class PollingSubscription {
-	private userMap: UserMap;
-	private frequency: number;
-	private skipInitialLoad: boolean;
-
-	intervalId?: ReturnType<typeof setTimeout>;
-
-	constructor({
-		userMap,
-		frequency,
-		skipInitialLoad = false,
-	}: {
-		userMap: UserMap;
-		frequency: number;
-		skipInitialLoad?: boolean;
-		includeIdle?: boolean;
-	}) {
-		this.userMap = userMap;
-		this.frequency = frequency;
-		this.skipInitialLoad = skipInitialLoad;
-	}
-
-	public async subscribe(): Promise<void> {
-		if (this.intervalId || this.frequency <= 0) {
-			return;
-		}
-
-		const executeSync = async () => {
-			await this.userMap.sync();
-			this.intervalId = setTimeout(executeSync, this.frequency);
-		};
-
-		if (!this.skipInitialLoad) {
-			await this.userMap.sync();
-		}
-		executeSync();
-	}
-
-	public async unsubscribe(): Promise<void> {
-		if (this.intervalId) {
-			clearInterval(this.intervalId);
-			this.intervalId = undefined;
-		}
-	}
-}

package/src/userMap/WebsocketSubscription.ts

@@ -1,84 +0,0 @@
-import { UserMap } from './userMap';
-import { getNonIdleUserFilter, getUserFilter } from '../memcmp';
-import { WebSocketProgramAccountSubscriber } from '../accounts/webSocketProgramAccountSubscriber';
-import { UserAccount } from '../types';
-import { Commitment, Context, MemcmpFilter, PublicKey } from '@solana/web3.js';
-import { ResubOpts } from '../accounts/types';
-
-export class WebsocketSubscription {
-	private userMap: UserMap;
-	private commitment: Commitment;
-	private skipInitialLoad: boolean;
-	private resubOpts?: ResubOpts;
-	private includeIdle?: boolean;
-	private additionalFilters?: MemcmpFilter[];
-	private decodeFn: (name: string, data: Buffer) => UserAccount;
-
-	private subscriber?: WebSocketProgramAccountSubscriber<UserAccount>;
-
-	constructor({
-		userMap,
-		commitment,
-		skipInitialLoad = false,
-		resubOpts,
-		includeIdle = false,
-		decodeFn,
-		additionalFilters = undefined,
-	}: {
-		userMap: UserMap;
-		commitment: Commitment;
-		skipInitialLoad?: boolean;
-		resubOpts?: ResubOpts;
-		includeIdle?: boolean;
-		decodeFn: (name: string, data: Buffer) => UserAccount;
-		additionalFilters?: MemcmpFilter[];
-	}) {
-		this.userMap = userMap;
-		this.commitment = commitment;
-		this.skipInitialLoad = skipInitialLoad;
-		this.resubOpts = resubOpts;
-		this.includeIdle = includeIdle || false;
-		this.decodeFn = decodeFn;
-		this.additionalFilters = additionalFilters;
-	}
-
-	public async subscribe(): Promise<void> {
-		if (!this.subscriber) {
-			const filters = [getUserFilter()];
-			if (!this.includeIdle) {
-				filters.push(getNonIdleUserFilter());
-			}
-			if (this.additionalFilters) {
-				filters.push(...this.additionalFilters);
-			}
-			this.subscriber = new WebSocketProgramAccountSubscriber<UserAccount>(
-				'UserMap',
-				'user',
-				this.userMap.velocityClient.program,
-				this.decodeFn,
-				{
-					filters,
-					commitment: this.commitment,
-				},
-				this.resubOpts
-			);
-		}
-
-		await this.subscriber.subscribe(
-			(accountId: PublicKey, account: UserAccount, context: Context) => {
-				const userKey = accountId.toBase58();
-				this.userMap.updateUserAccount(userKey, account, context.slot);
-			}
-		);
-
-		if (!this.skipInitialLoad) {
-			await this.userMap.sync();
-		}
-	}
-
-	public async unsubscribe(): Promise<void> {
-		if (!this.subscriber) return;
-		await this.subscriber.unsubscribe();
-		this.subscriber = undefined;
-	}
-}

package/src/userMap/grpcSubscription.ts

@@ -1,85 +0,0 @@
-import { UserMap } from './userMap';
-import { getNonIdleUserFilter, getUserFilter } from '../memcmp';
-import { WebSocketProgramAccountSubscriber } from '../accounts/webSocketProgramAccountSubscriber';
-import { UserAccount } from '../types';
-import { Context, MemcmpFilter, PublicKey } from '@solana/web3.js';
-import { GrpcConfigs, ResubOpts } from '../accounts/types';
-import { grpcProgramAccountSubscriber } from '../accounts/grpcProgramAccountSubscriber';
-
-export class grpcSubscription {
-	private grpcConfigs: GrpcConfigs;
-	private userMap: UserMap;
-	private skipInitialLoad: boolean;
-	private resubOpts?: ResubOpts;
-	private includeIdle?: boolean;
-	private additionalFilters?: MemcmpFilter[];
-	private decodeFn: (name: string, data: Buffer) => UserAccount;
-
-	private subscriber?: WebSocketProgramAccountSubscriber<UserAccount>;
-
-	constructor({
-		grpcConfigs,
-		userMap,
-		skipInitialLoad = false,
-		resubOpts,
-		includeIdle = false,
-		decodeFn,
-		additionalFilters = undefined,
-	}: {
-		grpcConfigs: GrpcConfigs;
-		userMap: UserMap;
-		skipInitialLoad?: boolean;
-		resubOpts?: ResubOpts;
-		includeIdle?: boolean;
-		decodeFn: (name: string, data: Buffer) => UserAccount;
-		additionalFilters?: MemcmpFilter[];
-	}) {
-		this.userMap = userMap;
-		this.skipInitialLoad = skipInitialLoad;
-		this.resubOpts = resubOpts;
-		this.includeIdle = includeIdle || false;
-		this.decodeFn = decodeFn;
-		this.grpcConfigs = grpcConfigs;
-		this.additionalFilters = additionalFilters;
-	}
-
-	public async subscribe(): Promise<void> {
-		if (!this.subscriber) {
-			const filters = [getUserFilter()];
-			if (!this.includeIdle) {
-				filters.push(getNonIdleUserFilter());
-			}
-			if (this.additionalFilters) {
-				filters.push(...this.additionalFilters);
-			}
-			this.subscriber = await grpcProgramAccountSubscriber.create<UserAccount>(
-				this.grpcConfigs,
-				'UserMap',
-				'user',
-				this.userMap.velocityClient.program,
-				this.decodeFn,
-				{
-					filters,
-				},
-				this.resubOpts
-			);
-		}
-
-		await this.subscriber.subscribe(
-			(accountId: PublicKey, account: UserAccount, context: Context) => {
-				const userKey = accountId.toBase58();
-				this.userMap.updateUserAccount(userKey, account, context.slot);
-			}
-		);
-
-		if (!this.skipInitialLoad) {
-			await this.userMap.sync();
-		}
-	}
-
-	public async unsubscribe(): Promise<void> {
-		if (!this.subscriber) return;
-		await this.subscriber.unsubscribe();
-		this.subscriber = undefined;
-	}
-}

package/src/userMap/referrerMap.ts

@@ -1,271 +0,0 @@
-import {
-	MemcmpFilter,
-	PublicKey,
-	RpcResponseAndContext,
-} from '@solana/web3.js';
-import { VelocityClient } from '../velocityClient';
-import { ReferrerInfo } from '../types';
-import {
-	getUserAccountPublicKeySync,
-	getUserStatsAccountPublicKey,
-} from '../addresses/pda';
-import {
-	getUserStatsFilter,
-	getUserStatsIsReferredFilter,
-	getUserStatsIsReferredOrReferrerFilter,
-} from '../memcmp';
-import bs58 from 'bs58';
-
-const DEFAULT_PUBLIC_KEY = PublicKey.default.toBase58();
-
-export class ReferrerMap {
-	/**
-	 * map from authority pubkey to referrer pubkey.
-	 */
-	private authorityReferrerMap = new Map<string, string>();
-	/**
-	 * map from referrer pubkey to ReferrerInfo.
-	 * Will be undefined if the referrer is not in the map yet.
-	 */
-	private referrerReferrerInfoMap = new Map<string, ReferrerInfo>();
-	private velocityClient: VelocityClient;
-	private parallelSync: boolean;
-
-	private fetchPromise?: Promise<void>;
-	private fetchPromiseResolver: () => void = () => {};
-
-	/**
-	 * Creates a new UserStatsMap instance.
-	 *
-	 * @param {VelocityClient} velocityClient - The VelocityClient instance.
-	 */
-	constructor(velocityClient: VelocityClient, parallelSync?: boolean) {
-		this.velocityClient = velocityClient;
-		this.parallelSync = parallelSync !== undefined ? parallelSync : true;
-	}
-
-	/**
-	 * Subscribe to all UserStats accounts.
-	 */
-	public async subscribe() {
-		if (this.size() > 0) {
-			return;
-		}
-
-		await this.velocityClient.subscribe();
-		await this.sync();
-	}
-
-	public has(authorityPublicKey: string): boolean {
-		return this.authorityReferrerMap.has(authorityPublicKey);
-	}
-
-	public get(authorityPublicKey: string): ReferrerInfo | undefined {
-		return this.getReferrer(authorityPublicKey);
-	}
-
-	public async addReferrer(authority: string, referrer?: string) {
-		if (referrer) {
-			this.authorityReferrerMap.set(authority, referrer);
-		} else if (referrer === undefined) {
-			const userStatsAccountPublicKey = getUserStatsAccountPublicKey(
-				this.velocityClient.program.programId,
-				new PublicKey(authority)
-			);
-			const accountInfo = await this.velocityClient.connection.getAccountInfo(
-				userStatsAccountPublicKey,
-				'processed'
-			);
-			if (!accountInfo) {
-				throw new Error(
-					`ReferrerMap: UserStats account not found for authority ${authority}`
-				);
-			}
-			const buffer = accountInfo.data;
-
-			const referrer = bs58.encode(buffer.subarray(40, 72));
-
-			this.addReferrer(authority, referrer);
-		}
-	}
-
-	/**
-	 * Enforce that a UserStats will exist for the given authorityPublicKey,
-	 * reading one from the blockchain if necessary.
-	 * @param authorityPublicKey
-	 * @returns
-	 */
-	public async mustGet(
-		authorityPublicKey: string
-	): Promise<ReferrerInfo | undefined> {
-		if (!this.has(authorityPublicKey)) {
-			await this.addReferrer(authorityPublicKey);
-		}
-		return this.getReferrer(authorityPublicKey);
-	}
-
-	public getReferrer(authorityPublicKey: string): ReferrerInfo | undefined {
-		const referrer = this.authorityReferrerMap.get(authorityPublicKey);
-		if (!referrer) {
-			// return undefined if the referrer is not in the map
-			return undefined;
-		}
-
-		if (referrer === DEFAULT_PUBLIC_KEY) {
-			return undefined;
-		}
-
-		if (this.referrerReferrerInfoMap.has(referrer)) {
-			return this.referrerReferrerInfoMap.get(referrer);
-		}
-
-		const referrerKey = new PublicKey(referrer);
-		const referrerInfo = {
-			referrer: getUserAccountPublicKeySync(
-				this.velocityClient.program.programId,
-				referrerKey,
-				0
-			),
-			referrerStats: getUserStatsAccountPublicKey(
-				this.velocityClient.program.programId,
-				referrerKey
-			),
-		};
-
-		this.referrerReferrerInfoMap.set(referrer, referrerInfo);
-		return referrerInfo;
-	}
-
-	public size(): number {
-		return this.authorityReferrerMap.size;
-	}
-
-	public numberOfReferred(): number {
-		return Array.from(this.authorityReferrerMap.values()).filter(
-			(referrer) => referrer !== DEFAULT_PUBLIC_KEY
-		).length;
-	}
-
-	public async sync(): Promise<void> {
-		if (this.fetchPromise) {
-			return this.fetchPromise;
-		}
-
-		this.fetchPromise = new Promise((resolver) => {
-			this.fetchPromiseResolver = resolver;
-		});
-
-		try {
-			if (this.parallelSync) {
-				await Promise.all([
-					this.syncAll(),
-					this.syncReferrer(getUserStatsIsReferredFilter()),
-					this.syncReferrer(getUserStatsIsReferredOrReferrerFilter()),
-				]);
-			} else {
-				await this.syncAll();
-				await this.syncReferrer(getUserStatsIsReferredFilter());
-				await this.syncReferrer(getUserStatsIsReferredOrReferrerFilter());
-			}
-		} finally {
-			this.fetchPromiseResolver();
-			this.fetchPromise = undefined;
-		}
-	}
-
-	public async syncAll(): Promise<void> {
-		const rpcRequestArgs = [
-			this.velocityClient.program.programId.toBase58(),
-			{
-				commitment: this.velocityClient.opts?.commitment,
-				filters: [getUserStatsFilter()],
-				encoding: 'base64',
-				dataSlice: {
-					offset: 0,
-					length: 0,
-				},
-				withContext: true,
-			},
-		];
-
-		const rpcJSONResponse: any =
-			// @ts-ignore
-			await this.velocityClient.connection._rpcRequest(
-				'getProgramAccounts',
-				rpcRequestArgs
-			);
-
-		const rpcResponseAndContext: RpcResponseAndContext<
-			Array<{
-				pubkey: string;
-				account: {
-					data: [string, string];
-				};
-			}>
-		> = rpcJSONResponse.result;
-
-		for (const account of rpcResponseAndContext.value) {
-			// only add if it isn't already in the map
-			// so that if syncReferrer already set it, we dont overwrite
-			if (!this.has(account.pubkey)) {
-				this.addReferrer(account.pubkey, DEFAULT_PUBLIC_KEY);
-			}
-		}
-	}
-
-	async syncReferrer(referrerFilter: MemcmpFilter): Promise<void> {
-		const rpcRequestArgs = [
-			this.velocityClient.program.programId.toBase58(),
-			{
-				commitment: this.velocityClient.opts?.commitment,
-				filters: [getUserStatsFilter(), referrerFilter],
-				encoding: 'base64',
-				dataSlice: {
-					offset: 0,
-					length: 72,
-				},
-				withContext: true,
-			},
-		];
-
-		const rpcJSONResponse: any =
-			// @ts-ignore
-			await this.velocityClient.connection._rpcRequest(
-				'getProgramAccounts',
-				rpcRequestArgs
-			);
-
-		const rpcResponseAndContext: RpcResponseAndContext<
-			Array<{
-				pubkey: string;
-				account: {
-					data: [string, string];
-				};
-			}>
-		> = rpcJSONResponse.result;
-
-		const batchSize = 1000;
-		for (let i = 0; i < rpcResponseAndContext.value.length; i += batchSize) {
-			const batch = rpcResponseAndContext.value.slice(i, i + batchSize);
-			await Promise.all(
-				batch.map(async (programAccount) => {
-					// @ts-ignore
-					const buffer = Buffer.from(
-						programAccount.account.data[0],
-						programAccount.account.data[1] as BufferEncoding
-					);
-					const authority = bs58.encode(buffer.subarray(8, 40));
-					const referrer = bs58.encode(buffer.subarray(40, 72));
-
-					this.addReferrer(authority, referrer);
-				})
-			);
-			await new Promise((resolve) => setTimeout(resolve, 0));
-		}
-	}
-
-	public async unsubscribe() {
-		this.authorityReferrerMap.clear();
-		this.referrerReferrerInfoMap.clear();
-	}
-}