drift-labs
diff --git a/‎sdk/VERSION‎
Lines changed: 1 addition & 1 deletion b/‎sdk/VERSION‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎sdk/scripts/deposit-isolated-positions.ts‎
Lines changed: 110 additions & 0 deletions b/‎sdk/scripts/deposit-isolated-positions.ts‎
Lines changed: 110 additions & 0 deletions
diff --git a/‎sdk/scripts/find-flagged-users.ts‎
Lines changed: 216 additions & 0 deletions b/‎sdk/scripts/find-flagged-users.ts‎
Lines changed: 216 additions & 0 deletions
@@ -1 +1 @@
-2.155.0-beta.4
+2.155.0-beta.4
@@ -0,0 +1,110 @@
+import { Connection, Keypair, PublicKey } from '@solana/web3.js';
+import dotenv from 'dotenv';
+import { AnchorProvider, Idl, Program, ProgramAccount, BN } from '@coral-xyz/anchor';
+import driftIDL from '../src/idl/drift.json';
+import {
+	DRIFT_PROGRAM_ID,
+	PerpMarketAccount,
+	SpotMarketAccount,
+	OracleInfo,
+	Wallet,
+	numberToSafeBN,
+} from '../src';
+import { DriftClient } from '../src/driftClient';
+import { DriftClientConfig } from '../src/driftClientConfig';
+
+async function main() {
+	dotenv.config({ path: '../' });
+
+	const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
+	if (!RPC_ENDPOINT) throw new Error('RPC_ENDPOINT env var required');
+
+	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 to use a real wallet.');
+	}
+	const wallet = new Wallet(keypair);
+
+	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);
+
+	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 });
+		}
+	}
+
+	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();
+
+	const candidates = perpMarketIndexes.filter((i) => i >= 0 && i <= 5);
+	const targetMarketIndex = candidates.length
+		? candidates[Math.floor(Math.random() * candidates.length)]
+		: perpMarketIndexes[0];
+
+	const perpMarketAccount = client.getPerpMarketAccount(targetMarketIndex);
+	const quoteSpotMarketIndex = perpMarketAccount.quoteSpotMarketIndex;
+	const spotMarketAccount = client.getSpotMarketAccount(quoteSpotMarketIndex);
+
+	const precision = new BN(10).pow(new BN(spotMarketAccount.decimals));
+	const amount = numberToSafeBN(0.01, precision);
+
+	const userTokenAccount = await client.getAssociatedTokenAccount(quoteSpotMarketIndex);
+	const ix = await client.getDepositIntoIsolatedPerpPositionIx(
+		amount,
+		targetMarketIndex,
+		userTokenAccount,
+		0
+	);
+
+	const tx = await client.buildTransaction([ix]);
+	const { txSig } = await client.sendTransaction(tx);
+	console.log(`Deposited into isolated perp market ${targetMarketIndex}: ${txSig}`);
+
+	await client.getUser().unsubscribe();
+	await client.unsubscribe();
+}
+
+main().catch((e) => {
+	console.error(e);
+	process.exit(1);
+});
+
+
@@ -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()}
+	`;
+
+}
+