@drift-labs/vaults-sdk 0.4.41 → 0.4.43

package/lib/types/drift_vaults.d.ts

@@ -225,6 +225,37 @@ export type DriftVaults = {
                 }
             ];
         },
+        {
+            name: 'updateUserPoolId';
+            accounts: [
+                {
+                    name: 'vault';
+                    isMut: true;
+                    isSigner: false;
+                },
+                {
+                    name: 'manager';
+                    isMut: false;
+                    isSigner: true;
+                },
+                {
+                    name: 'driftUser';
+                    isMut: true;
+                    isSigner: false;
+                },
+                {
+                    name: 'driftProgram';
+                    isMut: false;
+                    isSigner: false;
+                }
+            ];
+            args: [
+                {
+                    name: 'poolId';
+                    type: 'u8';
+                }
+            ];
+        },
         {
             name: 'updateVaultProtocol';
             accounts: [

package/lib/types/drift_vaults.js

@@ -228,6 +228,37 @@ exports.IDL = {
                 },
             ],
         },
+        {
+            name: 'updateUserPoolId',
+            accounts: [
+                {
+                    name: 'vault',
+                    isMut: true,
+                    isSigner: false,
+                },
+                {
+                    name: 'manager',
+                    isMut: false,
+                    isSigner: true,
+                },
+                {
+                    name: 'driftUser',
+                    isMut: true,
+                    isSigner: false,
+                },
+                {
+                    name: 'driftProgram',
+                    isMut: false,
+                    isSigner: false,
+                },
+            ],
+            args: [
+                {
+                    name: 'poolId',
+                    type: 'u8',
+                },
+            ],
+        },
         {
             name: 'updateVaultProtocol',
             accounts: [

package/lib/vaultClient.d.ts

@@ -130,6 +130,21 @@ export declare class VaultClient {
      * @returns
      */
     updateMarginTradingEnabled(vault: PublicKey, enabled: boolean, uiTxParams?: TxParams): Promise<TransactionSignature>;
+    /**
+     * Updates the vault's pool id (for isolated pools).
+     * @param vault vault address to update
+     * @param poolId pool id to update to
+     * @returns
+     */
+    updateUserPoolId(vault: PublicKey, poolId: number, uiTxParams?: TxParams): Promise<TransactionSignature>;
+    /**
+     * Gets the instruction to update the pool id for a vault.
+     * @param vault vault address to update
+     * @param vaultAccount vault account data (optional, will be fetched if not provided)
+     * @param poolId pool id to update to
+     * @returns instruction to update pool id
+     */
+    getUpdatePoolIdIx(vault: PublicKey, poolId: number, vaultAccount?: any): Promise<TransactionInstruction>;
     private handleWSolMovement;
     /**
      *

package/lib/vaultClient.js

@@ -446,6 +446,50 @@ class VaultClient {
             .instruction();
         return await this.createAndSendTxn([updateMarginTradingEnabledIx], uiTxParams);
     }
+    /**
+     * Updates the vault's pool id (for isolated pools).
+     * @param vault vault address to update
+     * @param poolId pool id to update to
+     * @returns
+     */
+    async updateUserPoolId(vault, poolId, uiTxParams) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const updatePoolIdIx = await this.getUpdatePoolIdIx(vault, poolId, vaultAccount);
+        return await this.createAndSendTxn([updatePoolIdIx], uiTxParams);
+    }
+    /**
+     * Gets the instruction to update the pool id for a vault.
+     * @param vault vault address to update
+     * @param vaultAccount vault account data (optional, will be fetched if not provided)
+     * @param poolId pool id to update to
+     * @returns instruction to update pool id
+     */
+    async getUpdatePoolIdIx(vault, poolId, vaultAccount) {
+        if (!vaultAccount) {
+            vaultAccount = await this.program.account.vault.fetch(vault);
+        }
+        const accounts = {
+            vault: vault,
+            driftUser: vaultAccount.user,
+            driftProgram: this.driftClient.program.programId,
+        };
+        const user = await this.getSubscribedVaultUser(vaultAccount.user);
+        const remainingAccounts = [];
+        try {
+            const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault);
+            const userStats = (await this.driftClient.program.account.userStats.fetch(userStatsKey));
+            remainingAccounts.push(...this.getRemainingAccountsForUser([user.getUserAccount()], [], vaultAccount, userStats));
+        }
+        catch (err) {
+            console.error('failed to get remaining accounts', err);
+            // do nothing
+        }
+        return await this.program.methods
+            .updateUserPoolId(poolId)
+            .accounts({ ...accounts, manager: this.driftClient.wallet.publicKey })
+            .remainingAccounts(remainingAccounts)
+            .instruction();
+    }
     async handleWSolMovement(amount, driftSpotMarket, userTokenAccount) {
         const isSolDeposit = driftSpotMarket.mint.equals(sdk_1.WRAPPED_SOL_MINT);
         const preIxs = [];

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/vaults-sdk",
-	"version": "0.4.41",
+	"version": "0.4.43",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"directories": {

package/src/idl/drift_vaults.json

@@ -225,6 +225,37 @@
         }
       ]
     },
+    {
+      "name": "updateUserPoolId",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "driftUser",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "poolId",
+          "type": "u8"
+        }
+      ]
+    },
     {
       "name": "updateVaultProtocol",
       "accounts": [

package/src/types/drift_vaults.ts

@@ -225,6 +225,37 @@ export type DriftVaults = {
 				}
 			];
 		},
+		{
+			name: 'updateUserPoolId';
+			accounts: [
+				{
+					name: 'vault';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'manager';
+					isMut: false;
+					isSigner: true;
+				},
+				{
+					name: 'driftUser';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'driftProgram';
+					isMut: false;
+					isSigner: false;
+				}
+			];
+			args: [
+				{
+					name: 'poolId';
+					type: 'u8';
+				}
+			];
+		},
 		{
 			name: 'updateVaultProtocol';
 			accounts: [
@@ -3150,6 +3181,37 @@ export const IDL: DriftVaults = {
 				},
 			],
 		},
+		{
+			name: 'updateUserPoolId',
+			accounts: [
+				{
+					name: 'vault',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'manager',
+					isMut: false,
+					isSigner: true,
+				},
+				{
+					name: 'driftUser',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'driftProgram',
+					isMut: false,
+					isSigner: false,
+				},
+			],
+			args: [
+				{
+					name: 'poolId',
+					type: 'u8',
+				},
+			],
+		},
 		{
 			name: 'updateVaultProtocol',
 			accounts: [

package/src/vaultClient.ts

@@ -712,6 +712,79 @@ export class VaultClient {
 		);
 	}
 
+	/**
+	 * Updates the vault's pool id (for isolated pools).
+	 * @param vault vault address to update
+	 * @param poolId pool id to update to
+	 * @returns
+	 */
+	public async updateUserPoolId(
+		vault: PublicKey,
+		poolId: number,
+		uiTxParams?: TxParams
+	): Promise<TransactionSignature> {
+		const vaultAccount = await this.program.account.vault.fetch(vault);
+		const updatePoolIdIx = await this.getUpdatePoolIdIx(
+			vault,
+			poolId,
+			vaultAccount
+		);
+		return await this.createAndSendTxn([updatePoolIdIx], uiTxParams);
+	}
+
+	/**
+	 * Gets the instruction to update the pool id for a vault.
+	 * @param vault vault address to update
+	 * @param vaultAccount vault account data (optional, will be fetched if not provided)
+	 * @param poolId pool id to update to
+	 * @returns instruction to update pool id
+	 */
+	public async getUpdatePoolIdIx(
+		vault: PublicKey,
+		poolId: number,
+		vaultAccount?: any
+	): Promise<TransactionInstruction> {
+		if (!vaultAccount) {
+			vaultAccount = await this.program.account.vault.fetch(vault);
+		}
+
+		const accounts = {
+			vault: vault,
+			driftUser: vaultAccount.user,
+			driftProgram: this.driftClient.program.programId,
+		};
+
+		const user = await this.getSubscribedVaultUser(vaultAccount.user);
+
+		const remainingAccounts: AccountMeta[] = [];
+		try {
+			const userStatsKey = getUserStatsAccountPublicKey(
+				this.driftClient.program.programId,
+				vault
+			);
+			const userStats = (await this.driftClient.program.account.userStats.fetch(
+				userStatsKey
+			)) as UserStatsAccount;
+			remainingAccounts.push(
+				...this.getRemainingAccountsForUser(
+					[user.getUserAccount()],
+					[],
+					vaultAccount,
+					userStats
+				)
+			);
+		} catch (err) {
+			console.error('failed to get remaining accounts', err);
+			// do nothing
+		}
+
+		return await this.program.methods
+			.updateUserPoolId(poolId)
+			.accounts({ ...accounts, manager: this.driftClient.wallet.publicKey })
+			.remainingAccounts(remainingAccounts)
+			.instruction();
+	}
+
 	private async handleWSolMovement(
 		amount: BN,
 		driftSpotMarket: SpotMarketAccount,