@drift-labs/sdk 2.94.0-beta.0 → 2.94.0-beta.2

package/VERSION

@@ -1 +1 @@
-2.94.0-beta.0
+2.94.0-beta.2

package/lib/bankrun/bankrunConnection.d.ts

@@ -19,7 +19,7 @@ export declare class BankrunContextWrapper {
     readonly context: ProgramTestContext;
     readonly provider: BankrunProvider;
     readonly commitment: Commitment;
-    constructor(context: ProgramTestContext);
+    constructor(context: ProgramTestContext, verifySignatures?: boolean);
     sendTransaction(tx: Transaction, additionalSigners?: Keypair[]): Promise<TransactionSignature>;
     getMinimumBalanceForRentExemption(_: number): Promise<number>;
     fundKeypair(keypair: Keypair | Wallet, lamports: number | bigint): Promise<TransactionSignature>;
@@ -36,7 +36,8 @@ export declare class BankrunConnection {
     private nextClientSubscriptionId;
     private onLogCallbacks;
     private onAccountChangeCallbacks;
-    constructor(banksClient: BanksClient, context: ProgramTestContext);
+    private verifySignatures;
+    constructor(banksClient: BanksClient, context: ProgramTestContext, verifySignatures?: boolean);
     getSlot(): Promise<bigint>;
     toConnection(): SolanaConnection;
     getTokenAccount(publicKey: PublicKey): Promise<Account>;

package/lib/bankrun/bankrunConnection.js

@@ -11,11 +11,11 @@ const bs58_1 = __importDefault(require("bs58"));
 const anchor_1 = require("@coral-xyz/anchor");
 const spl_token_1 = require("@solana/spl-token");
 class BankrunContextWrapper {
-    constructor(context) {
+    constructor(context, verifySignatures = true) {
         this.commitment = 'confirmed';
         this.context = context;
         this.provider = new anchor_bankrun_1.BankrunProvider(context);
-        this.connection = new BankrunConnection(this.context.banksClient, this.context);
+        this.connection = new BankrunConnection(this.context.banksClient, this.context, verifySignatures);
     }
     async sendTransaction(tx, additionalSigners) {
         tx.recentBlockhash = (await this.getLatestBlockhash()).toString();
@@ -62,13 +62,14 @@ class BankrunContextWrapper {
 }
 exports.BankrunContextWrapper = BankrunContextWrapper;
 class BankrunConnection {
-    constructor(banksClient, context) {
+    constructor(banksClient, context, verifySignatures = true) {
         this.transactionToMeta = new Map();
         this.nextClientSubscriptionId = 0;
         this.onLogCallbacks = new Map();
         this.onAccountChangeCallbacks = new Map();
         this._banksClient = banksClient;
         this.context = context;
+        this.verifySignatures = verifySignatures;
     }
     getSlot() {
         return this._banksClient.getSlot();
@@ -103,7 +104,15 @@ class BankrunConnection {
         return signature;
     }
     async sendTransaction(tx) {
-        const banksTransactionMeta = await this._banksClient.tryProcessTransaction(tx);
+        const serialized = tx.serialize({
+            verifySignatures: this.verifySignatures,
+        });
+        // @ts-ignore
+        const internal = this._banksClient.inner;
+        const inner = tx instanceof web3_js_1.Transaction
+            ? await internal.tryProcessLegacyTransaction(serialized)
+            : await internal.tryProcessVersionedTransaction(serialized);
+        const banksTransactionMeta = new solana_bankrun_1.BanksTransactionResultWithMeta(inner);
         if (banksTransactionMeta.result) {
             throw new Error(banksTransactionMeta.result);
         }

package/lib/driftClient.d.ts

@@ -121,12 +121,19 @@ export declare class DriftClient {
      * @param includeDelegates
      */
     updateWallet(newWallet: IWallet, subAccountIds?: number[], activeSubAccountId?: number, includeDelegates?: boolean, authoritySubaccountMap?: Map<string, number[]>): Promise<boolean>;
+    /**
+     * Update the subscribed accounts to a given authority, while leaving the
+     * connected wallet intact. This allows a user to emulate another user's
+     * account on the UI and sign permissionless transactions with their own wallet.
+     * @param emulateAuthority
+     */
+    emulateAccount(emulateAuthority: PublicKey): Promise<boolean>;
     switchActiveUser(subAccountId: number, authority?: PublicKey): Promise<void>;
     addUser(subAccountId: number, authority?: PublicKey, userAccount?: UserAccount): Promise<boolean>;
     /**
      * Adds and subscribes to users based on params set by the constructor or by updateWallet.
      */
-    addAndSubscribeToUsers(): Promise<boolean>;
+    addAndSubscribeToUsers(authority?: PublicKey): Promise<boolean>;
     initializeUserAccount(subAccountId?: number, name?: string, referrerInfo?: ReferrerInfo, txParams?: TxParams): Promise<[TransactionSignature, PublicKey]>;
     getInitializeUserInstructions(subAccountId?: number, name?: string, referrerInfo?: ReferrerInfo): Promise<[PublicKey, TransactionInstruction]>;
     getInitializeUserStatsIx(): Promise<TransactionInstruction>;

package/lib/driftClient.js

@@ -407,6 +407,43 @@ class DriftClient {
         }
         return success;
     }
+    /**
+     * Update the subscribed accounts to a given authority, while leaving the
+     * connected wallet intact. This allows a user to emulate another user's
+     * account on the UI and sign permissionless transactions with their own wallet.
+     * @param emulateAuthority
+     */
+    async emulateAccount(emulateAuthority) {
+        var _a, _b, _c;
+        this.skipLoadUsers = false;
+        // Update provider for txSender with new wallet details
+        this.authority = emulateAuthority;
+        this.userStatsAccountPublicKey = undefined;
+        this.includeDelegates = true;
+        const walletSupportsVersionedTxns = 
+        //@ts-ignore
+        (_b = (_a = this.wallet.supportedTransactionVersions) === null || _a === void 0 ? void 0 : _a.size) !== null && _b !== void 0 ? _b : 0 > 1;
+        this.txVersion = walletSupportsVersionedTxns ? 0 : 'legacy';
+        this.authoritySubAccountMap = new Map();
+        /* Reset user stats account */
+        if ((_c = this.userStats) === null || _c === void 0 ? void 0 : _c.isSubscribed) {
+            await this.userStats.unsubscribe();
+        }
+        this.userStats = undefined;
+        this.userStats = new userStats_1.UserStats({
+            driftClient: this,
+            userStatsAccountPublicKey: this.getUserStatsAccountPublicKey(),
+            accountSubscription: this.userStatsAccountSubscriptionConfig,
+        });
+        await this.userStats.subscribe();
+        let success = true;
+        if (this.isSubscribed) {
+            await Promise.all(this.unsubscribeUsers());
+            this.users.clear();
+            success = await this.addAndSubscribeToUsers(emulateAuthority);
+        }
+        return success;
+    }
     async switchActiveUser(subAccountId, authority) {
         var _a;
         const authorityChanged = authority && !((_a = this.authority) === null || _a === void 0 ? void 0 : _a.equals(authority));
@@ -445,7 +482,7 @@ class DriftClient {
     /**
      * Adds and subscribes to users based on params set by the constructor or by updateWallet.
      */
-    async addAndSubscribeToUsers() {
+    async addAndSubscribeToUsers(authority) {
         var _a, _b, _c, _d, _e, _f, _g;
         // save the rpc calls if driftclient is initialized without a real wallet
         if (this.skipLoadUsers)
@@ -465,9 +502,9 @@ class DriftClient {
         else {
             let userAccounts = [];
             let delegatedAccounts = [];
-            const userAccountsPromise = this.getUserAccountsForAuthority(this.wallet.publicKey);
+            const userAccountsPromise = this.getUserAccountsForAuthority(authority !== null && authority !== void 0 ? authority : this.wallet.publicKey);
             if (this.includeDelegates) {
-                const delegatedAccountsPromise = this.getUserAccountsForDelegate(this.wallet.publicKey);
+                const delegatedAccountsPromise = this.getUserAccountsForDelegate(authority !== null && authority !== void 0 ? authority : this.wallet.publicKey);
                 [userAccounts, delegatedAccounts] = await Promise.all([
                     userAccountsPromise,
                     delegatedAccountsPromise,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.94.0-beta.0",
+	"version": "2.94.0-beta.2",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/bankrun/bankrunConnection.ts

@@ -55,12 +55,13 @@ export class BankrunContextWrapper {
 	public readonly provider: BankrunProvider;
 	public readonly commitment: Commitment = 'confirmed';
 
-	constructor(context: ProgramTestContext) {
+	constructor(context: ProgramTestContext, verifySignatures = true) {
 		this.context = context;
 		this.provider = new BankrunProvider(context);
 		this.connection = new BankrunConnection(
 			this.context.banksClient,
-			this.context
+			this.context,
+			verifySignatures
 		);
 	}
 
@@ -149,9 +150,16 @@ export class BankrunConnection {
 		[PublicKey, AccountChangeCallback]
 	>();
 
-	constructor(banksClient: BanksClient, context: ProgramTestContext) {
+	private verifySignatures: boolean;
+
+	constructor(
+		banksClient: BanksClient,
+		context: ProgramTestContext,
+		verifySignatures = true
+	) {
 		this._banksClient = banksClient;
 		this.context = context;
+		this.verifySignatures = verifySignatures;
 	}
 
 	getSlot(): Promise<bigint> {
@@ -205,9 +213,17 @@ export class BankrunConnection {
 	}
 
 	async sendTransaction(tx: Transaction): Promise<TransactionSignature> {
-		const banksTransactionMeta = await this._banksClient.tryProcessTransaction(
-			tx
-		);
+		const serialized = tx.serialize({
+			verifySignatures: this.verifySignatures,
+		});
+		// @ts-ignore
+		const internal = this._banksClient.inner;
+		const inner =
+			tx instanceof Transaction
+				? await internal.tryProcessLegacyTransaction(serialized)
+				: await internal.tryProcessVersionedTransaction(serialized);
+		const banksTransactionMeta = new BanksTransactionResultWithMeta(inner);
+
 		if (banksTransactionMeta.result) {
 			throw new Error(banksTransactionMeta.result);
 		}

package/src/driftClient.ts

@@ -714,6 +714,51 @@ export class DriftClient {
 		return success;
 	}
 
+	/**
+	 * Update the subscribed accounts to a given authority, while leaving the
+	 * connected wallet intact. This allows a user to emulate another user's
+	 * account on the UI and sign permissionless transactions with their own wallet.
+	 * @param emulateAuthority
+	 */
+	public async emulateAccount(emulateAuthority: PublicKey): Promise<boolean> {
+		this.skipLoadUsers = false;
+		// Update provider for txSender with new wallet details
+		this.authority = emulateAuthority;
+		this.userStatsAccountPublicKey = undefined;
+		this.includeDelegates = true;
+		const walletSupportsVersionedTxns =
+			//@ts-ignore
+			this.wallet.supportedTransactionVersions?.size ?? 0 > 1;
+		this.txVersion = walletSupportsVersionedTxns ? 0 : 'legacy';
+
+		this.authoritySubAccountMap = new Map<string, number[]>();
+
+		/* Reset user stats account */
+		if (this.userStats?.isSubscribed) {
+			await this.userStats.unsubscribe();
+		}
+
+		this.userStats = undefined;
+
+		this.userStats = new UserStats({
+			driftClient: this,
+			userStatsAccountPublicKey: this.getUserStatsAccountPublicKey(),
+			accountSubscription: this.userStatsAccountSubscriptionConfig,
+		});
+
+		await this.userStats.subscribe();
+
+		let success = true;
+
+		if (this.isSubscribed) {
+			await Promise.all(this.unsubscribeUsers());
+			this.users.clear();
+			success = await this.addAndSubscribeToUsers(emulateAuthority);
+		}
+
+		return success;
+	}
+
 	public async switchActiveUser(subAccountId: number, authority?: PublicKey) {
 		const authorityChanged = authority && !this.authority?.equals(authority);
 
@@ -771,7 +816,7 @@ export class DriftClient {
 	/**
 	 * Adds and subscribes to users based on params set by the constructor or by updateWallet.
 	 */
-	public async addAndSubscribeToUsers(): Promise<boolean> {
+	public async addAndSubscribeToUsers(authority?: PublicKey): Promise<boolean> {
 		// save the rpc calls if driftclient is initialized without a real wallet
 		if (this.skipLoadUsers) return true;
 
@@ -799,12 +844,12 @@ export class DriftClient {
 			let delegatedAccounts = [];
 
 			const userAccountsPromise = this.getUserAccountsForAuthority(
-				this.wallet.publicKey
+				authority ?? this.wallet.publicKey
 			);
 
 			if (this.includeDelegates) {
 				const delegatedAccountsPromise = this.getUserAccountsForDelegate(
-					this.wallet.publicKey
+					authority ?? this.wallet.publicKey
 				);
 				[userAccounts, delegatedAccounts] = await Promise.all([
 					userAccountsPromise,