@velocity-exchange/sdk 0.4.0 → 0.6.0

package/scripts/grpc-multiuser-client-test-comparison.ts

@@ -1,158 +0,0 @@
-import { grpcUserAccountSubscriber } from '../src/accounts/grpcUserAccountSubscriber';
-import { grpcMultiUserAccountSubscriber } from '../src/accounts/grpcMultiUserAccountSubscriber';
-import { Connection, Keypair, PublicKey } from '@solana/web3.js';
-import { VELOCITY_PROGRAM_ID } from '../src';
-import { CommitmentLevel } from '@triton-one/yellowstone-grpc';
-import { AnchorProvider, Idl, Program } from '@coral-xyz/anchor';
-import velocityIDL from '../src/idl/velocity.json';
-import assert from 'assert';
-import { Wallet } from '../src';
-
-const GRPC_ENDPOINT = process.env.GRPC_ENDPOINT;
-const TOKEN = process.env.TOKEN;
-const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
-
-const USER_ACCOUNT_PUBKEYS = [
-	// Add user account public keys here, e.g.:
-	// new PublicKey('...')
-];
-
-async function testGrpcUserAccountSubscriberV1VsV2() {
-	console.log('🚀 Initializing User Account Subscriber V1 vs V2 Test...');
-
-	if (USER_ACCOUNT_PUBKEYS.length === 0) {
-		console.error(
-			'❌ No user account public keys provided. Please add some to USER_ACCOUNT_PUBKEYS array.'
-		);
-		process.exit(1);
-	}
-
-	const connection = new Connection(RPC_ENDPOINT);
-	const wallet = new Wallet(new Keypair());
-
-	const programId = new PublicKey(VELOCITY_PROGRAM_ID);
-	const provider = new AnchorProvider(
-		connection,
-		// @ts-ignore
-		wallet,
-		{
-			commitment: 'processed',
-		}
-	);
-
-	const program = new Program(velocityIDL as Idl, programId, provider);
-
-	const grpcConfigs = {
-		endpoint: GRPC_ENDPOINT,
-		token: TOKEN,
-		commitmentLevel: CommitmentLevel.PROCESSED,
-		channelOptions: {
-			grpcKeepAliveTimeout: 1_000,
-			grpcTcpKeepalive: 10_000,
-		},
-	};
-
-	console.log(`📊 Testing ${USER_ACCOUNT_PUBKEYS.length} user accounts...`);
-
-	// V1: Create individual subscribers for each user account
-	const v1Subscribers = USER_ACCOUNT_PUBKEYS.map(
-		(pubkey) =>
-			new grpcUserAccountSubscriber(grpcConfigs, program, pubkey, {
-				logResubMessages: true,
-			})
-	);
-
-	// V2: Create a single multi-subscriber and get per-user interfaces
-	const v2MultiSubscriber = new grpcMultiUserAccountSubscriber(
-		program,
-		grpcConfigs,
-		{ logResubMessages: true }
-	);
-	const v2Subscribers = USER_ACCOUNT_PUBKEYS.map((pubkey) =>
-		v2MultiSubscriber.forUser(pubkey)
-	);
-
-	// Subscribe all V1 subscribers
-	console.log('🔗 Subscribing V1 subscribers...');
-	await Promise.all(v1Subscribers.map((sub) => sub.subscribe()));
-	console.log('✅ V1 subscribers ready');
-
-	// Subscribe all V2 subscribers
-	console.log('🔗 Subscribing V2 subscribers...');
-	await v2MultiSubscriber.subscribe();
-	console.log('✅ V2 subscribers ready');
-
-	const compare = () => {
-		try {
-			let passedTests = 0;
-			let totalTests = 0;
-
-			// Test each user account
-			for (let i = 0; i < USER_ACCOUNT_PUBKEYS.length; i++) {
-				const pubkey = USER_ACCOUNT_PUBKEYS[i];
-				const v1Sub = v1Subscribers[i];
-				const v2Sub = v2Subscribers[i];
-
-				totalTests++;
-
-				// 1. Test isSubscribed
-				assert.strictEqual(
-					v1Sub.isSubscribed,
-					v2Sub.isSubscribed,
-					`User ${pubkey.toBase58()}: isSubscribed should match`
-				);
-
-				// 2. Test getUserAccountAndSlot
-				const v1Data = v1Sub.getUserAccountAndSlot();
-				const v2Data = v2Sub.getUserAccountAndSlot();
-
-				// Compare the user account data
-				assert.deepStrictEqual(
-					v1Data.data,
-					v2Data.data,
-					`User ${pubkey.toBase58()}: account data should match`
-				);
-
-				// Slots might differ slightly due to timing, but let's check if they're close
-				const slotDiff = Math.abs(v1Data.slot - v2Data.slot);
-				if (slotDiff > 10) {
-					console.warn(
-						`⚠️  User ${pubkey.toBase58()}: slot difference is ${slotDiff} (v1: ${
-							v1Data.slot
-						}, v2: ${v2Data.slot})`
-					);
-				}
-
-				passedTests++;
-			}
-
-			console.log(
-				`✅ All comparisons passed (${passedTests}/${totalTests} user accounts)`
-			);
-		} catch (error) {
-			console.error('❌ Comparison failed:', error);
-		}
-	};
-
-	// Run initial comparison
-	compare();
-
-	// Run comparison every second to verify live updates
-	const interval = setInterval(compare, 1000);
-
-	const cleanup = async () => {
-		clearInterval(interval);
-		console.log('🧹 Cleaning up...');
-		await Promise.all([
-			...v1Subscribers.map((sub) => sub.unsubscribe()),
-			...v2Subscribers.map((sub) => sub.unsubscribe()),
-		]);
-		console.log('✅ Cleanup complete');
-		process.exit(0);
-	};
-
-	process.on('SIGINT', cleanup);
-	process.on('SIGTERM', cleanup);
-}
-
-testGrpcUserAccountSubscriberV1VsV2().catch(console.error);

package/scripts/postbuild.js

@@ -1,95 +0,0 @@
-// scripts/postbuild.js
-const fs = require('fs');
-const path = require('path');
-const yargs = require('yargs/yargs');
-const { hideBin } = require('yargs/helpers');
-
-const forceEnv = yargs(hideBin(process.argv))
-	.option('force-env', {
-		type: 'string',
-		description: 'Specify environment to force (node or browser)',
-		choices: ['node', 'browser']
-	})
-	.argv?.forceEnv;
-
-const isomorphicPackages = ['grpc', 'anchor', 'anchor29'];
-const environments = ['node', 'browser'];
-
-environments.forEach((environment) => {
-	console.log(`Running ${environment} environment postbuild script`);
-	console.log(``);
-
-	isomorphicPackages.forEach((package) => {
-
-		// We want to overwrite the base isomorphic files (the "target" files) with the concrete implementation code and definition files (the "source" files).
-
-		const isomorphicFolderPath = path.join(
-			__dirname,
-			'..',
-			'lib',
-			environment,
-			'isomorphic'
-		);
-
-		const targetEnv = forceEnv ? forceEnv : environment;
-
-		const filesToSwap = [
-			{
-				source: `${package}.${targetEnv}.js`,
-				target: `${package}.js`,
-			},
-			{
-				source: `${package}.${targetEnv}.d.ts`,
-				target: `${package}.d.ts`,
-			},
-		];
-
-		for (const file of filesToSwap) {
-			const sourcePath = path.join(
-				isomorphicFolderPath,
-				file.source
-			);
-
-			const targetPath = path.join(
-				isomorphicFolderPath,
-				file.target
-			);
-
-			try {
-				const sourceContent = fs.readFileSync(sourcePath, 'utf8');
-				fs.writeFileSync(targetPath, sourceContent);
-			} catch (error) {
-				console.error(
-					`Error processing isomophic package : ${package} :: ${error.message}`
-				);
-			}
-		}
-
-		// Delete other environment files for safety
-		environments.forEach((otherEnvironment) => {
-			if (otherEnvironment === targetEnv) {
-				return;
-			}
-
-			const otherTargetFiles = [
-				`${package}.${otherEnvironment}.js`,
-				`${package}.${otherEnvironment}.d.ts`,
-			];
-
-			for (const otherTargetFile of otherTargetFiles) {
-				const otherTargetPath = path.join(
-					__dirname,
-					'..',
-					'lib',
-					environment,
-					'isomorphic',
-					otherTargetFile
-				);
-
-				if (fs.existsSync(otherTargetPath)) {
-					fs.unlinkSync(otherTargetPath);
-				}
-			}
-		});
-	});
-});

package/scripts/single-grpc-client-test.ts

@@ -1,284 +0,0 @@
-import { VelocityClient } from '../src/velocityClient';
-import { grpcVelocityClientAccountSubscriberV2 } from '../src/accounts/grpcVelocityClientAccountSubscriberV2';
-import { Connection, Keypair, PublicKey } from '@solana/web3.js';
-import { VelocityClientConfig } from '../src/velocityClientConfig';
-import {
-	VELOCITY_PROGRAM_ID,
-	PerpMarketAccount,
-	SpotMarketAccount,
-	Wallet,
-	OracleInfo,
-	decodeName,
-} from '../src';
-import { CommitmentLevel } from '@triton-one/yellowstone-grpc';
-import dotenv from 'dotenv';
-import {
-	AnchorProvider,
-	Idl,
-	Program,
-	ProgramAccount,
-} from '@coral-xyz/anchor';
-import velocityIDL from '../src/idl/velocity.json';
-import { grpcMultiUserAccountSubscriber } from '../src/accounts/grpcMultiUserAccountSubscriber';
-
-const GRPC_ENDPOINT = process.env.GRPC_ENDPOINT;
-const TOKEN = process.env.TOKEN;
-const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
-
-async function initializeSingleGrpcClient() {
-	console.log('🚀 Initializing single gRPC Velocity Client...');
-
-	const connection = new Connection(RPC_ENDPOINT);
-	const wallet = new Wallet(new Keypair());
-	dotenv.config({ path: '../' });
-
-	const programId = new PublicKey(VELOCITY_PROGRAM_ID);
-	const provider = new AnchorProvider(
-		connection,
-		// @ts-ignore
-		wallet,
-		{
-			commitment: 'processed',
-		}
-	);
-
-	const program = new Program(velocityIDL as Idl, programId, provider);
-
-	// Get perp market accounts
-	const allPerpMarketProgramAccounts =
-		(await program.account.perpMarket.all()) as ProgramAccount<PerpMarketAccount>[];
-	const perpMarketProgramAccounts = allPerpMarketProgramAccounts.filter((val) =>
-		[46].includes(val.account.marketIndex)
-	);
-	const perpMarketIndexes = perpMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-
-	// Get spot market accounts
-	const allSpotMarketProgramAccounts =
-		(await program.account.spotMarket.all()) as ProgramAccount<SpotMarketAccount>[];
-	const spotMarketProgramAccounts = allSpotMarketProgramAccounts.filter((val) =>
-		[0].includes(val.account.marketIndex)
-	);
-	const spotMarketIndexes = spotMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-
-	// Get oracle infos
-	const seen = new Set<string>();
-	const oracleInfos: OracleInfo[] = [];
-	for (const acct of perpMarketProgramAccounts) {
-		const key = `${acct.account.amm.oracle.toBase58()}-${
-			Object.keys(acct.account.amm.oracleSource)[0]
-		}`;
-		if (!seen.has(key)) {
-			seen.add(key);
-			oracleInfos.push({
-				publicKey: acct.account.amm.oracle,
-				source: acct.account.amm.oracleSource,
-			});
-		}
-	}
-	for (const acct of spotMarketProgramAccounts) {
-		const key = `${acct.account.oracle.toBase58()}-${
-			Object.keys(acct.account.oracleSource)[0]
-		}`;
-		if (!seen.has(key)) {
-			seen.add(key);
-			oracleInfos.push({
-				publicKey: acct.account.oracle,
-				source: acct.account.oracleSource,
-			});
-		}
-	}
-
-	console.log(
-		`📊 Markets: ${perpMarketIndexes.length} perp, ${spotMarketIndexes.length} spot`
-	);
-	console.log(`🔮 Oracles: ${oracleInfos.length}`);
-
-	const grpcConfigs = {
-		endpoint: GRPC_ENDPOINT,
-		token: TOKEN,
-		commitmentLevel: CommitmentLevel.PROCESSED,
-		channelOptions: {
-			grpcKeepAliveTimeout: 1_000,
-			grpcTcpKeepalive: 10_000,
-		},
-	};
-
-	const multiUserSubsciber = new grpcMultiUserAccountSubscriber(
-		program,
-		grpcConfigs
-	);
-
-	const baseAccountSubscription = {
-		type: 'grpc' as const,
-		grpcConfigs,
-		grpcMultiUserAccountSubscriber: multiUserSubsciber,
-	};
-
-	const config: VelocityClientConfig = {
-		connection,
-		wallet,
-		programID: new PublicKey(VELOCITY_PROGRAM_ID),
-		accountSubscription: {
-			...baseAccountSubscription,
-			velocityClientAccountSubscriber: grpcVelocityClientAccountSubscriberV2,
-		},
-		perpMarketIndexes,
-		spotMarketIndexes,
-		oracleInfos,
-	};
-
-	const client = new VelocityClient(config);
-
-	// Set up event listeners
-	const eventCounts = {
-		stateAccountUpdate: 0,
-		perpMarketAccountUpdate: 0,
-		spotMarketAccountUpdate: 0,
-		oraclePriceUpdate: 0,
-		update: 0,
-	};
-
-	console.log('🎧 Setting up event listeners...');
-
-	client.eventEmitter.on('stateAccountUpdate', (_data) => {
-		eventCounts.stateAccountUpdate++;
-	});
-
-	client.eventEmitter.on('perpMarketAccountUpdate', (_data) => {
-		eventCounts.perpMarketAccountUpdate++;
-	});
-
-	client.eventEmitter.on('spotMarketAccountUpdate', (_data) => {
-		eventCounts.spotMarketAccountUpdate++;
-	});
-
-	client.eventEmitter.on('oraclePriceUpdate', (_publicKey, _source, _data) => {
-		eventCounts.oraclePriceUpdate++;
-	});
-
-	client.accountSubscriber.eventEmitter.on('update', () => {
-		eventCounts.update++;
-	});
-
-	// Subscribe
-	console.log('🔗 Subscribing to accounts...');
-	await client.subscribe();
-
-	console.log('✅ Client subscribed successfully!');
-	console.log(
-		'🚀 Starting high-load testing (50 reads/sec per perp market)...'
-	);
-
-	// High-frequency load testing - 50 reads per second per perp market
-	const loadTestInterval = setInterval(async () => {
-		try {
-			// Test getPerpMarketAccount for each perp market (50 times per second per market)
-			for (const marketIndex of perpMarketIndexes) {
-				const perpMarketAccount = client.getPerpMarketAccount(marketIndex);
-				if (!perpMarketAccount) {
-					console.log(`Perp market ${marketIndex} not found`);
-					continue;
-				}
-				console.log(
-					'perpMarketAccount name: ',
-					decodeName(perpMarketAccount.name)
-				);
-				console.log(
-					'perpMarketAccount data: ',
-					JSON.stringify({
-						marketIndex: perpMarketAccount.marketIndex,
-						name: decodeName(perpMarketAccount.name),
-						baseAssetReserve: perpMarketAccount.amm.baseAssetReserve.toString(),
-						quoteAssetReserve:
-							perpMarketAccount.amm.quoteAssetReserve.toString(),
-					})
-				);
-			}
-
-			// Test getMMOracleDataForPerpMarket for each perp market (50 times per second per market)
-			for (const marketIndex of perpMarketIndexes) {
-				try {
-					const oracleData = client.getMMOracleDataForPerpMarket(marketIndex);
-					console.log('oracleData price: ', oracleData.price.toString());
-					console.log(
-						'oracleData: ',
-						JSON.stringify({
-							price: oracleData.price.toString(),
-							confidence: oracleData.confidence?.toString(),
-							slot: oracleData.slot?.toString(),
-						})
-					);
-				} catch (error) {
-					// Ignore errors for load testing
-				}
-			}
-
-			for (const marketIndex of perpMarketIndexes) {
-				try {
-					const { data, slot } =
-						client.accountSubscriber.getMarketAccountAndSlot(marketIndex);
-					if (!data) {
-						console.log(
-							`Perp market getMarketAccountAndSlot ${marketIndex} not found`
-						);
-						continue;
-					}
-					console.log(
-						'marketAccountAndSlot: ',
-						JSON.stringify({
-							marketIndex: data.marketIndex,
-							name: decodeName(data.name),
-							slot: slot?.toString(),
-							baseAssetReserve: data.amm.baseAssetReserve.toString(),
-							quoteAssetReserve: data.amm.quoteAssetReserve.toString(),
-						})
-					);
-				} catch (error) {
-					console.error(
-						`Error getting market account and slot for market ${marketIndex}:`,
-						error
-					);
-				}
-			}
-		} catch (error) {
-			console.error('Load test error:', error);
-		}
-	}, 20); // 50 times per second = 1000ms / 50 = 20ms interval
-
-	// Log periodic stats
-	const statsInterval = setInterval(() => {
-		console.log('\n📈 Event Counts:', eventCounts);
-		console.log(`⏱️  Client subscribed: ${client.isSubscribed}`);
-		console.log(
-			`🔗 Account subscriber subscribed: ${client.accountSubscriber.isSubscribed}`
-		);
-		console.log(
-			`🔥 Load: ${perpMarketIndexes.length * 50 * 2} reads/sec (${
-				perpMarketIndexes.length
-			} markets × 50 getPerpMarketAccount + 50 getMMOracleDataForPerpMarket)`
-		);
-	}, 5000);
-
-	// Handle shutdown signals - just exit without cleanup since they never unsubscribe
-	process.on('SIGINT', () => {
-		console.log('\n🛑 Shutting down...');
-		clearInterval(loadTestInterval);
-		clearInterval(statsInterval);
-		process.exit(0);
-	});
-
-	process.on('SIGTERM', () => {
-		console.log('\n🛑 Shutting down...');
-		clearInterval(loadTestInterval);
-		clearInterval(statsInterval);
-		process.exit(0);
-	});
-
-	return client;
-}
-
-initializeSingleGrpcClient().catch(console.error);

package/scripts/withdraw-isolated-positions.ts

@@ -1,174 +0,0 @@
-import { Connection, Keypair, PublicKey } from '@solana/web3.js';
-import dotenv from 'dotenv';
-import {
-	AnchorProvider,
-	Idl,
-	Program,
-	ProgramAccount,
-} from '@coral-xyz/anchor';
-import velocityIDL from '../src/idl/velocity.json';
-import {
-	VELOCITY_PROGRAM_ID,
-	PerpMarketAccount,
-	SpotMarketAccount,
-	OracleInfo,
-	Wallet,
-	ZERO,
-} from '../src';
-import { VelocityClient } from '../src/velocityClient';
-import { VelocityClientConfig } from '../src/velocityClientConfig';
-
-function isStatusOpen(status: any) {
-	return !!status && 'open' in status;
-}
-
-function isPerpMarketType(marketType: any) {
-	return !!marketType && 'perp' in marketType;
-}
-
-async function main() {
-	dotenv.config({ path: '../' });
-
-	const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
-	if (!RPC_ENDPOINT) throw new Error('RPC_ENDPOINT env var required');
-
-	// Load wallet
-	// For safety this creates a new ephemeral wallet unless PRIVATE_KEY is provided (base58 array)
-	let keypair: Keypair;
-	const pk = process.env.PRIVATE_KEY;
-	if (pk) {
-		const secret = Uint8Array.from(JSON.parse(pk));
-		keypair = Keypair.fromSecretKey(secret);
-	} else {
-		keypair = new Keypair();
-		console.warn(
-			'Using ephemeral keypair. Provide PRIVATE_KEY for real withdrawals.'
-		);
-	}
-	const wallet = new Wallet(keypair);
-
-	// Connection and program for market discovery
-	const connection = new Connection(RPC_ENDPOINT);
-	const provider = new AnchorProvider(connection, wallet as any, {
-		commitment: 'processed',
-	});
-	const programId = new PublicKey(VELOCITY_PROGRAM_ID);
-	const program = new Program(velocityIDL as Idl, programId, provider);
-
-	// Discover markets and oracles (like the example test script)
-	const allPerpMarketProgramAccounts =
-		(await program.account.perpMarket.all()) as ProgramAccount<PerpMarketAccount>[];
-	const perpMarketIndexes = allPerpMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-	const allSpotMarketProgramAccounts =
-		(await program.account.spotMarket.all()) as ProgramAccount<SpotMarketAccount>[];
-	const spotMarketIndexes = allSpotMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-
-	const seen = new Set<string>();
-	const oracleInfos: OracleInfo[] = [];
-	for (const acct of allPerpMarketProgramAccounts) {
-		const key = `${acct.account.amm.oracle.toBase58()}-${
-			Object.keys(acct.account.amm.oracleSource)[0]
-		}`;
-		if (!seen.has(key)) {
-			seen.add(key);
-			oracleInfos.push({
-				publicKey: acct.account.amm.oracle,
-				source: acct.account.amm.oracleSource,
-			});
-		}
-	}
-	for (const acct of allSpotMarketProgramAccounts) {
-		const key = `${acct.account.oracle.toBase58()}-${
-			Object.keys(acct.account.oracleSource)[0]
-		}`;
-		if (!seen.has(key)) {
-			seen.add(key);
-			oracleInfos.push({
-				publicKey: acct.account.oracle,
-				source: acct.account.oracleSource,
-			});
-		}
-	}
-
-	// Build VelocityClient with websocket subscription (lightweight)
-	const clientConfig: VelocityClientConfig = {
-		connection,
-		wallet,
-		programID: programId,
-		accountSubscription: {
-			type: 'websocket',
-			commitment: 'processed',
-		},
-		perpMarketIndexes,
-		spotMarketIndexes,
-		oracleInfos,
-		env: 'devnet',
-	};
-	const client = new VelocityClient(clientConfig);
-	await client.subscribe();
-
-	// Ensure user exists and is subscribed
-	const user = client.getUser();
-	await user.subscribe();
-
-	const userAccount = user.getUserAccount();
-	const openOrders = user.getOpenOrders();
-
-	const marketsWithOpenOrders = new Set<number>();
-	for (const o of openOrders ?? []) {
-		if (isStatusOpen(o.status) && isPerpMarketType(o.marketType)) {
-			marketsWithOpenOrders.add(o.marketIndex);
-		}
-	}
-
-	const withdrawTargets = userAccount.perpPositions.filter((pos) => {
-		const isZeroBase = pos.baseAssetAmount.eq(ZERO);
-		const hasIso = pos.isolatedPositionScaledBalance.gt(ZERO);
-		const hasOpenOrders = marketsWithOpenOrders.has(pos.marketIndex);
-		return isZeroBase && hasIso && !hasOpenOrders;
-	});
-
-	console.log(
-		`Found ${withdrawTargets.length} isolated perp positions to withdraw`
-	);
-
-	for (const pos of withdrawTargets) {
-		try {
-			const amount = client.getIsolatedPerpPositionTokenAmount(pos.marketIndex);
-			if (amount.lte(ZERO)) continue;
-
-			const perpMarketAccount = client.getPerpMarketAccount(pos.marketIndex);
-			const quoteAta = await client.getAssociatedTokenAccount(
-				perpMarketAccount.quoteSpotMarketIndex
-			);
-
-			const ixs = await client.getWithdrawFromIsolatedPerpPositionIxsBundle(
-				amount,
-				pos.marketIndex,
-				0,
-				quoteAta,
-				true
-			);
-
-			const tx = await client.buildTransaction(ixs);
-			const { txSig } = await client.sendTransaction(tx);
-			console.log(
-				`Withdrew isolated deposit for perp market ${pos.marketIndex}: ${txSig}`
-			);
-		} catch (e) {
-			console.error(`Failed to withdraw for market ${pos.marketIndex}:`, e);
-		}
-	}
-
-	await user.unsubscribe();
-	await client.unsubscribe();
-}
-
-main().catch((e) => {
-	console.error(e);
-	process.exit(1);
-});