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

package/VERSION

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

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.1",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

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,