@velocity-exchange/sdk 0.4.0 → 0.6.0

package/scripts/find-flagged-users.ts

@@ -1,213 +0,0 @@
-import { Connection, Keypair, PublicKey } from '@solana/web3.js';
-import dotenv from 'dotenv';
-import {
-	VelocityClient,
-	VelocityClientConfig,
-	Wallet,
-	UserMap,
-	VELOCITY_PROGRAM_ID,
-	getMarketsAndOraclesForSubscription,
-	BulkAccountLoader,
-	BN,
-	PerpPosition,
-} from '../src';
-import { TransactionSignature } from '@solana/web3.js';
-import fs from 'fs';
-import os from 'os';
-import path from 'path';
-
-async function main() {
-	dotenv.config({ path: '../' });
-	// Simple CLI parsing
-	interface CliOptions {
-		mode: 'list' | 'one' | 'all';
-		targetUser?: string;
-	}
-
-	function parseCliArgs(): CliOptions {
-		const args = process.argv.slice(2);
-		let mode: CliOptions['mode'] = 'list';
-		let targetUser: string | undefined = undefined;
-		for (let i = 0; i < args.length; i++) {
-			const arg = args[i];
-			if (arg === '--mode' && i + 1 < args.length) {
-				const next = args[i + 1] as CliOptions['mode'];
-				if (next === 'list' || next === 'one' || next === 'all') {
-					mode = next;
-				}
-				i++;
-			} else if ((arg === '--user' || arg === '--target') && i + 1 < args.length) {
-				targetUser = args[i + 1];
-				i++;
-			}
-		}
-		return { mode, targetUser };
-	}
-
-	const { mode, targetUser } = parseCliArgs();
-
-	const RPC_ENDPOINT =
-		process.env.RPC_ENDPOINT ?? 'https://api.mainnet-beta.solana.com';
-
-	const connection = new Connection(RPC_ENDPOINT);
-	const keypairPath =
-		process.env.SOLANA_KEYPAIR ??
-		path.join(os.homedir(), '.config', 'solana', 'id.json');
-	const secret = JSON.parse(fs.readFileSync(keypairPath, 'utf-8')) as number[];
-	const wallet = new Wallet(Keypair.fromSecretKey(Uint8Array.from(secret)));
-
-	const { perpMarketIndexes, spotMarketIndexes, oracleInfos } =
-		getMarketsAndOraclesForSubscription('mainnet-beta');
-
-		const accountLoader = new BulkAccountLoader(connection, 'confirmed', 60_000);
-
-	const clientConfig: VelocityClientConfig = {
-		connection,
-		wallet,
-		programID: new PublicKey(VELOCITY_PROGRAM_ID),
-		accountSubscription: {
-			type: 'polling',
-			accountLoader,
-		},
-		perpMarketIndexes,
-		spotMarketIndexes,
-		oracleInfos,
-		env: 'mainnet-beta',
-	};
-
-	const client = new VelocityClient(clientConfig);
-	await client.subscribe();
-
-	const userMap = new UserMap({
-		velocityClient: client,
-		subscriptionConfig: {
-			type: 'polling',
-			frequency: 60_000,
-			commitment: 'confirmed',
-		},
-		includeIdle: false,
-		syncConfig: { type: 'paginated' },
-		throwOnFailedSync: false,
-	});
-	await userMap.subscribe();
-
-
-	const flaggedUsers: Array<{
-		userPubkey: string;
-		authority: string;
-		flags: Array<{ marketIndex: number; flag: number; isolatedPositionScaledBalance: BN }>;
-	}> = [];
-
-	console.log(`User map size: ${Array.from(userMap.entries()).length}`);
-
-	for (const [userPubkey, user] of userMap.entries()) {
-		const userAccount = user.getUserAccount();
-		const flaggedPositions = userAccount.perpPositions
-			.filter((p) => p.positionFlag >= 1 || p.isolatedPositionScaledBalance.toString() !== '0')
-			.map((p) => ({ marketIndex: p.marketIndex, flag: p.positionFlag, isolatedPositionScaledBalance: p.isolatedPositionScaledBalance, fullPosition: p }));
-
-		if (flaggedPositions.length > 0) {
-			if(mode === 'one' && userPubkey === targetUser) {
-				console.log(`flagged positions on user ${userPubkey}`);
-				console.log(flaggedPositions.map((p) => `mkt=${p.marketIndex}, flag=${p.flag}, isolatedPositionScaledBalance=${p.isolatedPositionScaledBalance.toString()}, fullPosition=${fullLogPerpPosition(p.fullPosition)}`).join('\n\n; '));
-			}
-			flaggedUsers.push({
-				userPubkey,
-				authority: userAccount.authority.toBase58(),
-				flags: flaggedPositions,
-			});
-		}
-	}
-
-	// Mode 1: list flagged users (default)
-	if (mode === 'list') {
-		console.log(`Flagged users (positionFlag >= 1 || isolatedPositionScaledBalance > 0): ${flaggedUsers.length}`);
-		for (const u of flaggedUsers) {
-			const flagsStr = u.flags
-				.map((f) => `mkt=${f.marketIndex}, flag=${f.flag}, isolatedPositionScaledBalance=${f.isolatedPositionScaledBalance.toString()}`)
-				.join('; ');
-			console.log(
-				`- authority=${u.authority} userAccount=${u.userPubkey} -> [${flagsStr}]`
-			);
-		}
-	}
-
-	// Helper to invoke updateUserIdle
-	async function updateUserIdleFor(userAccountPubkeyStr: string): Promise<TransactionSignature | undefined> {
-		const userObj = userMap.get(userAccountPubkeyStr);
-		if (!userObj) {
-			console.warn(`User ${userAccountPubkeyStr} not found in userMap`);
-			return undefined;
-		}
-		try {
-			const sig = await client.updateUserIdle(
-				new PublicKey(userAccountPubkeyStr),
-				userObj.getUserAccount()
-			);
-			console.log(`updateUserIdle sent for userAccount=${userAccountPubkeyStr} -> tx=${sig}`);
-			return sig;
-		} catch (e) {
-			console.error(`Failed updateUserIdle for userAccount=${userAccountPubkeyStr}`, e);
-			return undefined;
-		}
-	}
-
-	// Mode 2: updateUserIdle on a single flagged user
-	if (mode === 'one') {
-		if (flaggedUsers.length === 0) {
-			console.log('No flagged users to update.');
-		} else {
-			const chosen =
-				(targetUser && flaggedUsers.find((u) => u.userPubkey === targetUser)) ||
-				flaggedUsers[0];
-			console.log(
-				`Updating single flagged userAccount=${chosen.userPubkey} authority=${chosen.authority}`
-			);
-			await updateUserIdleFor(chosen.userPubkey);
-		}
-	}
-
-	// Mode 3: updateUserIdle on all flagged users
-	if (mode === 'all') {
-		if (flaggedUsers.length === 0) {
-			console.log('No flagged users to update.');
-		} else {
-			console.log(`Updating all ${flaggedUsers.length} flagged users...`);
-			for (const u of flaggedUsers) {
-				await updateUserIdleFor(u.userPubkey);
-			}
-			console.log('Finished updating all flagged users.');
-		}
-	}
-
-	await userMap.unsubscribe();
-	await client.unsubscribe();
-}
-
-main().catch((e) => {
-	console.error(e);
-	process.exit(1);
-});
-
-
-function fullLogPerpPosition(position: PerpPosition) {
-
-	return `
-	[PERP POSITION]
-	baseAssetAmount=${position.baseAssetAmount.toString()}
-	quoteAssetAmount=${position.quoteAssetAmount.toString()}
-	quoteBreakEvenAmount=${position.quoteBreakEvenAmount.toString()}
-	quoteEntryAmount=${position.quoteEntryAmount.toString()}
-	openBids=${position.openBids.toString()}
-	openAsks=${position.openAsks.toString()}
-	settledPnl=${position.settledPnl.toString()}
-	remainderBaseAssetAmount=${position.remainderBaseAssetAmount}
-	maxMarginRatio=${position.maxMarginRatio}
-	marketIndex=${position.marketIndex}
-	openOrders=${position.openOrders}
-	positionFlag=${position.positionFlag}
-	isolatedPositionScaledBalance=${position.isolatedPositionScaledBalance.toString()}
-	`;
-
-}
-

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

@@ -1,372 +0,0 @@
-import { VelocityClient } from '../src/velocityClient';
-import { grpcVelocityClientAccountSubscriberV2 } from '../src/accounts/grpcVelocityClientAccountSubscriberV2';
-import { grpcVelocityClientAccountSubscriber } from '../src/accounts/grpcVelocityClientAccountSubscriber';
-import { Connection, Keypair, PublicKey } from '@solana/web3.js';
-import { VelocityClientConfig } from '../src/velocityClientConfig';
-import {
-	VELOCITY_PROGRAM_ID,
-	PerpMarketAccount,
-	SpotMarketAccount,
-	Wallet,
-	OracleInfo,
-} 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 assert from 'assert';
-
-const GRPC_ENDPOINT = process.env.GRPC_ENDPOINT;
-const TOKEN = process.env.TOKEN;
-const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
-
-async function initializeGrpcVelocityClientV2VersusV1() {
-	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);
-
-	const allPerpMarketProgramAccounts =
-		(await program.account.perpMarket.all()) as ProgramAccount<PerpMarketAccount>[];
-	const perpMarketProgramAccounts = allPerpMarketProgramAccounts.filter((val) =>
-		[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15].includes(
-			val.account.marketIndex
-		)
-	);
-	const perpMarketIndexes = perpMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-
-	const allSpotMarketProgramAccounts =
-		(await program.account.spotMarket.all()) as ProgramAccount<SpotMarketAccount>[];
-	const spotMarketProgramAccounts = allSpotMarketProgramAccounts.filter((val) =>
-		[0, 1, 2, 3, 4, 5].includes(val.account.marketIndex)
-	);
-	const spotMarketIndexes = spotMarketProgramAccounts.map(
-		(val) => val.account.marketIndex
-	);
-
-	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,
-			});
-		}
-	}
-
-	const baseAccountSubscription = {
-		type: 'grpc' as const,
-		grpcConfigs: {
-			endpoint: GRPC_ENDPOINT,
-			token: TOKEN,
-			commitmentLevel: CommitmentLevel.PROCESSED,
-			channelOptions: {
-				grpcKeepAliveTimeout: 1_000,
-				grpcTcpKeepalive: 10_000,
-			},
-		},
-	};
-
-	const configV2: VelocityClientConfig = {
-		connection,
-		wallet,
-		programID: new PublicKey(VELOCITY_PROGRAM_ID),
-		accountSubscription: {
-			...baseAccountSubscription,
-			velocityClientAccountSubscriber: grpcVelocityClientAccountSubscriberV2,
-		},
-		perpMarketIndexes,
-		spotMarketIndexes,
-		oracleInfos,
-	};
-
-	const configV1: VelocityClientConfig = {
-		connection,
-		wallet,
-		programID: new PublicKey(VELOCITY_PROGRAM_ID),
-		accountSubscription: {
-			...baseAccountSubscription,
-			velocityClientAccountSubscriber: grpcVelocityClientAccountSubscriber,
-		},
-		perpMarketIndexes,
-		spotMarketIndexes,
-		oracleInfos,
-	};
-
-	const clientV2 = new VelocityClient(configV2);
-	const clientV1 = new VelocityClient(configV1);
-
-	await Promise.all([clientV1.subscribe(), clientV2.subscribe()]);
-	const compare = () => {
-		try {
-			// 1. Test getStateAccountAndSlot
-			const state1 = clientV1.accountSubscriber.getStateAccountAndSlot();
-			const state2 = clientV2.accountSubscriber.getStateAccountAndSlot();
-			assert.deepStrictEqual(
-				state1.data,
-				state2.data,
-				'State accounts should match'
-			);
-			if (
-				state1.slot !== undefined &&
-				state2.slot !== undefined &&
-				state2.slot < state1.slot
-			) {
-				console.error(
-					`State account slot regression: v2 slot ${state2.slot} < v1 slot ${state1.slot}`
-				);
-			}
-
-			// 2. Test getMarketAccountsAndSlots (all perp markets) - sorted comparison
-			const allPerpMarkets1 = clientV1.accountSubscriber
-				.getMarketAccountsAndSlots()
-				.sort((a, b) => a.data.marketIndex - b.data.marketIndex);
-			const allPerpMarkets2 = clientV2.accountSubscriber
-				.getMarketAccountsAndSlots()
-				.sort((a, b) => a.data.marketIndex - b.data.marketIndex);
-			assert.strictEqual(
-				allPerpMarkets1.length,
-				allPerpMarkets2.length,
-				'Number of perp markets should match'
-			);
-
-			// Compare each perp market in the sorted arrays
-			for (let i = 0; i < allPerpMarkets1.length; i++) {
-				const market1 = allPerpMarkets1[i];
-				const market2 = allPerpMarkets2[i];
-				assert.strictEqual(
-					market1.data.marketIndex,
-					market2.data.marketIndex,
-					`Perp market at position ${i} should have same marketIndex`
-				);
-				// assert.deepStrictEqual(
-				// 	market1.data,
-				// 	market2.data,
-				// 	`Perp market ${market1.data.marketIndex} (from getMarketAccountsAndSlots) should match`
-				// );
-			}
-
-			// 3. Test getMarketAccountAndSlot for each perp market
-			for (const idx of perpMarketIndexes) {
-				const market1 = clientV1.accountSubscriber.getMarketAccountAndSlot(idx);
-				const market2 = clientV2.accountSubscriber.getMarketAccountAndSlot(idx);
-				// assert.deepStrictEqual(
-				// 	market1?.data,
-				// 	market2?.data,
-				// 	`Perp market ${idx} data should match`
-				// );
-				// assert.strictEqual(
-				// 	market1?.slot,
-				// 	market2?.slot,
-				// 	`Perp market ${idx} slot should match`
-				// );
-				if (
-					market1?.slot !== undefined &&
-					market2?.slot !== undefined &&
-					market2.slot < market1.slot
-				) {
-					console.error(
-						`Perp market ${idx} slot regression: v2 slot ${market2.slot} < v1 slot ${market1.slot}`
-					);
-				} else if (
-					market1?.slot !== undefined &&
-					market2?.slot !== undefined &&
-					market2.slot > market1.slot
-				) {
-					console.info(
-						`Perp market ${idx} slot is FASTER! v2: ${market2.slot}, v1: ${market1.slot}`
-					);
-				}
-			}
-
-			// 4. Test getSpotMarketAccountsAndSlots (all spot markets) - sorted comparison
-			const allSpotMarkets1 = clientV1.accountSubscriber
-				.getSpotMarketAccountsAndSlots()
-				.sort((a, b) => a.data.marketIndex - b.data.marketIndex);
-			const allSpotMarkets2 = clientV2.accountSubscriber
-				.getSpotMarketAccountsAndSlots()
-				.sort((a, b) => a.data.marketIndex - b.data.marketIndex);
-			assert.strictEqual(
-				allSpotMarkets1.length,
-				allSpotMarkets2.length,
-				'Number of spot markets should match'
-			);
-
-			// Compare each spot market in the sorted arrays
-			for (let i = 0; i < allSpotMarkets1.length; i++) {
-				const market1 = allSpotMarkets1[i];
-				const market2 = allSpotMarkets2[i];
-				assert.strictEqual(
-					market1.data.marketIndex,
-					market2.data.marketIndex,
-					`Spot market at position ${i} should have same marketIndex`
-				);
-				// assert.deepStrictEqual(
-				// 	market1.data,
-				// 	market2.data,
-				// 	`Spot market ${market1.data.marketIndex} (from getSpotMarketAccountsAndSlots) should match`
-				// );
-			}
-
-			// 5. Test getSpotMarketAccountAndSlot for each spot market
-			for (const idx of spotMarketIndexes) {
-				const market1 =
-					clientV1.accountSubscriber.getSpotMarketAccountAndSlot(idx);
-				const market2 =
-					clientV2.accountSubscriber.getSpotMarketAccountAndSlot(idx);
-				// assert.deepStrictEqual(
-				// 	market1?.data,
-				// 	market2?.data,
-				// 	`Spot market ${idx} data should match`
-				// );
-				// assert.strictEqual(
-				// 	market1?.slot,
-				// 	market2?.slot,
-				// 	`Spot market ${idx} slot should match`
-				// );
-				if (
-					market1?.slot !== undefined &&
-					market2?.slot !== undefined &&
-					market2.slot < market1.slot
-				) {
-					console.error(
-						`Spot market ${idx} slot regression: v2 slot ${market2.slot} < v1 slot ${market1.slot}`
-					);
-				} else if (
-					market1?.slot !== undefined &&
-					market2?.slot !== undefined &&
-					market2.slot > market1.slot
-				) {
-					console.info(
-						`Spot market ${idx} slot is FASTER! v2: ${market2.slot}, v1: ${market1.slot}`
-					);
-				}
-			}
-
-			// 6. Test getOraclePriceDataAndSlotForPerpMarket
-			for (const idx of perpMarketIndexes) {
-				const oracle1 =
-					clientV1.accountSubscriber.getOraclePriceDataAndSlotForPerpMarket(
-						idx
-					);
-				const oracle2 =
-					clientV2.accountSubscriber.getOraclePriceDataAndSlotForPerpMarket(
-						idx
-					);
-				// assert.deepStrictEqual(
-				// 	oracle1?.data,
-				// 	oracle2?.data,
-				// 	`Perp market ${idx} oracle data should match`
-				// );
-				// Note: slots might differ slightly due to timing, so we can optionally skip this check or be lenient
-				// assert.strictEqual(oracle1?.slot, oracle2?.slot, `Perp market ${idx} oracle slot should match`);
-				if (
-					oracle1?.slot !== undefined &&
-					oracle2?.slot !== undefined &&
-					oracle2.slot < oracle1.slot
-				) {
-					console.error(
-						`Perp market ${idx} oracle slot regression: v2 slot ${oracle2.slot} < v1 slot ${oracle1.slot}`
-					);
-				} else if (
-					oracle1?.slot !== undefined &&
-					oracle2?.slot !== undefined &&
-					oracle2.slot > oracle1.slot
-				) {
-					console.info(
-						`Perp market ${idx} oracle slot is FASTER! v2: ${oracle2.slot}, v1: ${oracle1.slot}`
-					);
-				}
-			}
-
-			// 7. Test getOraclePriceDataAndSlotForSpotMarket
-			for (const idx of spotMarketIndexes) {
-				const oracle1 =
-					clientV1.accountSubscriber.getOraclePriceDataAndSlotForSpotMarket(
-						idx
-					);
-				const oracle2 =
-					clientV2.accountSubscriber.getOraclePriceDataAndSlotForSpotMarket(
-						idx
-					);
-				// assert.deepStrictEqual(
-				// 	oracle1?.data,
-				// 	oracle2?.data,
-				// 	`Spot market ${idx} oracle data should match`
-				// );
-				// Note: slots might differ slightly due to timing
-				// assert.strictEqual(oracle1?.slot, oracle2?.slot, `Spot market ${idx} oracle slot should match`);
-				if (
-					oracle1?.slot !== undefined &&
-					oracle2?.slot !== undefined &&
-					oracle2.slot < oracle1.slot
-				) {
-					console.error(
-						`Spot market ${idx} oracle slot regression: v2 slot ${oracle2.slot} < v1 slot ${oracle1.slot}`
-					);
-				} else if (
-					oracle1?.slot !== undefined &&
-					oracle2?.slot !== undefined &&
-					oracle2.slot > oracle1.slot
-				) {
-					console.info(
-						`Spot market ${idx} oracle slot is FASTER! v2: ${oracle2.slot}, v1: ${oracle1.slot}`
-					);
-				}
-			}
-
-			console.log('✓ All comparisons passed');
-		} catch (error) {
-			console.error('✗ Comparison failed:', error);
-		}
-	};
-
-	compare();
-	const interval = setInterval(compare, 1000);
-
-	const cleanup = async () => {
-		clearInterval(interval);
-		await Promise.all([clientV1.unsubscribe(), clientV2.unsubscribe()]);
-		process.exit(0);
-	};
-
-	process.on('SIGINT', cleanup);
-	process.on('SIGTERM', cleanup);
-}
-
-initializeGrpcVelocityClientV2VersusV1().catch(console.error);