@drift-labs/sdk-browser 2.155.0-beta.3 → 2.155.0-beta.5

package/scripts/find-flagged-users.ts

@@ -0,0 +1,216 @@
+import { Connection, Keypair, PublicKey } from '@solana/web3.js';
+import dotenv from 'dotenv';
+import {
+	DriftClient,
+	DriftClientConfig,
+	Wallet,
+	UserMap,
+	DRIFT_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: DriftClientConfig = {
+		connection,
+		wallet,
+		programID: new PublicKey(DRIFT_PROGRAM_ID),
+		accountSubscription: {
+			type: 'polling',
+			accountLoader,
+		},
+		perpMarketIndexes,
+		spotMarketIndexes,
+		oracleInfos,
+		env: 'mainnet-beta',
+	};
+
+	const client = new DriftClient(clientConfig);
+	await client.subscribe();
+
+	const userMap = new UserMap({
+		driftClient: 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()}
+	lpShares=${position.lpShares.toString()}
+	remainderBaseAssetAmount=${position.remainderBaseAssetAmount}
+	lastQuoteAssetAmountPerLp=${position.lastQuoteAssetAmountPerLp.toString()}
+	perLpBase=${position.perLpBase}
+	maxMarginRatio=${position.maxMarginRatio}
+	marketIndex=${position.marketIndex}
+	openOrders=${position.openOrders}
+	positionFlag=${position.positionFlag}
+	isolatedPositionScaledBalance=${position.isolatedPositionScaledBalance.toString()}
+	`;
+
+}
+

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

@@ -48,9 +48,7 @@ async function initializeSingleGrpcClient() {
 	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
-		)
+		[46].includes(val.account.marketIndex)
 	);
 	const perpMarketIndexes = perpMarketProgramAccounts.map(
 		(val) => val.account.marketIndex
@@ -60,7 +58,7 @@ async function initializeSingleGrpcClient() {
 	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)
+		[0].includes(val.account.marketIndex)
 	);
 	const spotMarketIndexes = spotMarketProgramAccounts.map(
 		(val) => val.account.marketIndex
@@ -94,7 +92,9 @@ async function initializeSingleGrpcClient() {
 		}
 	}
 
-	console.log(`📊 Markets: ${perpMarketIndexes.length} perp, ${spotMarketIndexes.length} spot`);
+	console.log(
+		`📊 Markets: ${perpMarketIndexes.length} perp, ${spotMarketIndexes.length} spot`
+	);
 	console.log(`🔮 Oracles: ${oracleInfos.length}`);
 
 
@@ -171,7 +171,9 @@ async function initializeSingleGrpcClient() {
 	await client.subscribe();
 
 	console.log('✅ Client subscribed successfully!');
-	console.log('🚀 Starting high-load testing (50 reads/sec per perp market)...');
+	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 () => {
@@ -179,29 +181,71 @@ async function initializeSingleGrpcClient() {
 			// Test getPerpMarketAccount for each perp market (50 times per second per market)
 			for (const marketIndex of perpMarketIndexes) {
 				const perpMarketAccount = client.getPerpMarketAccount(marketIndex);
-				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()
-				}));
+				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()
-					}));
+					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);
 		}
@@ -211,8 +255,14 @@ async function initializeSingleGrpcClient() {
 	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)`);
+		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

package/scripts/withdraw-isolated-positions.ts

@@ -0,0 +1,174 @@
+import { Connection, Keypair, PublicKey } from '@solana/web3.js';
+import dotenv from 'dotenv';
+import {
+	AnchorProvider,
+	Idl,
+	Program,
+	ProgramAccount,
+} from '@coral-xyz/anchor';
+import driftIDL from '../src/idl/drift.json';
+import {
+	DRIFT_PROGRAM_ID,
+	PerpMarketAccount,
+	SpotMarketAccount,
+	OracleInfo,
+	Wallet,
+	ZERO,
+} from '../src';
+import { DriftClient } from '../src/driftClient';
+import { DriftClientConfig } from '../src/driftClientConfig';
+
+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(DRIFT_PROGRAM_ID);
+	const program = new Program(driftIDL 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 DriftClient with websocket subscription (lightweight)
+	const clientConfig: DriftClientConfig = {
+		connection,
+		wallet,
+		programID: programId,
+		accountSubscription: {
+			type: 'websocket',
+			commitment: 'processed',
+		},
+		perpMarketIndexes,
+		spotMarketIndexes,
+		oracleInfos,
+		env: 'devnet',
+	};
+	const client = new DriftClient(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);
+});

package/src/decode/user.ts

@@ -8,6 +8,7 @@ import {
 	OrderType,
 	PerpPosition,
 	PositionDirection,
+	PositionFlag,
 	SpotBalanceType,
 	SpotPosition,
 	UserAccount,
@@ -83,6 +84,10 @@ export function decodeUser(buffer: Buffer): UserAccount {
 		const baseAssetAmount = readSignedBigInt64LE(buffer, offset + 8);
 		const quoteAssetAmount = readSignedBigInt64LE(buffer, offset + 16);
 		const lpShares = readUnsignedBigInt64LE(buffer, offset + 64);
+		const isolatedPositionScaledBalance = readSignedBigInt64LE(
+			buffer,
+			offset + 72
+		);
 		const openOrders = buffer.readUInt8(offset + 94);
 		const positionFlag = buffer.readUInt8(offset + 95);
 
@@ -90,7 +95,13 @@ export function decodeUser(buffer: Buffer): UserAccount {
 			baseAssetAmount.eq(ZERO) &&
 			openOrders === 0 &&
 			quoteAssetAmount.eq(ZERO) &&
-			lpShares.eq(ZERO)
+			lpShares.eq(ZERO) &&
+			isolatedPositionScaledBalance.eq(ZERO) &&
+			!(
+				(positionFlag &
+					(PositionFlag.BeingLiquidated | PositionFlag.Bankruptcy)) >
+				0
+			)
 		) {
 			offset += 96;
 			continue;
@@ -107,9 +118,7 @@ export function decodeUser(buffer: Buffer): UserAccount {
 		const openAsks = readSignedBigInt64LE(buffer, offset);
 		offset += 8;
 		const settledPnl = readSignedBigInt64LE(buffer, offset);
-		offset += 16;
-		const isolatedPositionScaledBalance = readSignedBigInt64LE(buffer, offset);
-		offset += 8;
+		offset += 24;
 		const lastQuoteAssetAmountPerLp = readSignedBigInt64LE(buffer, offset);
 		offset += 8;
 		const maxMarginRatio = buffer.readUInt16LE(offset);
@@ -118,7 +127,6 @@ export function decodeUser(buffer: Buffer): UserAccount {
 		offset += 3;
 		const perLpBase = buffer.readUInt8(offset);
 		offset += 1;
-
 		perpPositions.push({
 			lastCumulativeFundingRate,
 			baseAssetAmount,
@@ -135,8 +143,8 @@ export function decodeUser(buffer: Buffer): UserAccount {
 			openOrders,
 			perLpBase,
 			maxMarginRatio,
-			isolatedPositionScaledBalance,
 			positionFlag,
+			isolatedPositionScaledBalance,
 		});
 	}