@drift-labs/vaults-sdk 0.1.532 → 0.1.534

package/lib/vaultClient.js

@@ -1262,15 +1262,129 @@ class VaultClient {
         if (!spotMarket) {
             throw new Error(`Spot market ${spotMarketIndex} not found on driftClient`);
         }
+        const ifVaultTokenAccount = (0, addresses_1.getInsuranceFundTokenVaultAddressSync)(this.program.programId, vault, spotMarketIndex);
         return await this.program.methods
             .initializeInsuranceFundStake(spotMarketIndex)
             .accounts({
             vault: vault,
             driftSpotMarket: spotMarket.pubkey,
+            driftSpotMarketMint: spotMarket.mint,
+            vaultTokenAccount: ifVaultTokenAccount,
+            insuranceFundStake: ifStakeAccountPublicKey,
+            driftUserStats: vaultAccount.userStats,
+            driftState: await this.driftClient.getStatePublicKey(),
+            driftProgram: this.driftClient.program.programId,
+        })
+            .rpc();
+    }
+    /**
+     * Adds an amount to an insurance fund stake for the vault.
+     * @param vault vault address to update
+     * @param spotMarketIndex spot market index of the insurance fund stake
+     * @param amount amount to add to the insurance fund stake, in spotMarketIndex precision
+     * @returns
+     */
+    async addToInsuranceFundStake(vault, spotMarketIndex, amount, managerTokenAccount) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        if (!vaultAccount.manager.equals(this.driftClient.wallet.publicKey)) {
+            throw new Error(`Only the manager of the vault can add to the insurance fund stake.`);
+        }
+        const ifStakeAccountPublicKey = (0, sdk_1.getInsuranceFundStakeAccountPublicKey)(this.driftClient.program.programId, vault, spotMarketIndex);
+        const ifVaultPublicKey = await (0, sdk_1.getInsuranceFundVaultPublicKey)(this.driftClient.program.programId, spotMarketIndex);
+        const spotMarket = this.driftClient.getSpotMarketAccount(spotMarketIndex);
+        if (!spotMarket) {
+            throw new Error(`Spot market ${spotMarketIndex} not found on driftClient`);
+        }
+        if (!managerTokenAccount) {
+            managerTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarket.mint, this.driftClient.wallet.publicKey);
+        }
+        const ifVaultTokenAccount = (0, addresses_1.getInsuranceFundTokenVaultAddressSync)(this.program.programId, vault, spotMarketIndex);
+        return await this.program.methods
+            .addInsuranceFundStake(spotMarketIndex, amount)
+            .accounts({
+            vault: vault,
+            driftSpotMarket: spotMarket.pubkey,
+            driftSpotMarketVault: spotMarket.vault,
+            insuranceFundStake: ifStakeAccountPublicKey,
+            insuranceFundVault: ifVaultPublicKey,
+            managerTokenAccount,
+            vaultIfTokenAccount: ifVaultTokenAccount,
+            driftUserStats: vaultAccount.userStats,
+            driftState: await this.driftClient.getStatePublicKey(),
+            driftProgram: this.driftClient.program.programId,
+            driftSigner: this.driftClient.getStateAccount().signer,
+            tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
+        })
+            .rpc();
+    }
+    async requestRemoveInsuranceFundStake(vault, spotMarketIndex, amount) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const ifStakeAccountPublicKey = (0, sdk_1.getInsuranceFundStakeAccountPublicKey)(this.driftClient.program.programId, vault, spotMarketIndex);
+        const ifVaultPublicKey = await (0, sdk_1.getInsuranceFundVaultPublicKey)(this.driftClient.program.programId, spotMarketIndex);
+        const spotMarket = this.driftClient.getSpotMarketAccount(spotMarketIndex);
+        if (!spotMarket) {
+            throw new Error(`Spot market ${spotMarketIndex} not found on driftClient`);
+        }
+        return await this.program.methods
+            .requestRemoveInsuranceFundStake(spotMarketIndex, amount)
+            .accounts({
+            vault,
+            manager: this.driftClient.wallet.publicKey,
+            driftSpotMarket: spotMarket.pubkey,
             insuranceFundStake: ifStakeAccountPublicKey,
+            insuranceFundVault: ifVaultPublicKey,
             driftUserStats: vaultAccount.userStats,
+            driftProgram: this.driftClient.program.programId,
+        })
+            .rpc();
+    }
+    async cancelRequestRemoveInsuranceFundStake(vault, spotMarketIndex) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const ifStakeAccountPublicKey = (0, sdk_1.getInsuranceFundStakeAccountPublicKey)(this.driftClient.program.programId, vault, spotMarketIndex);
+        const ifVaultPublicKey = await (0, sdk_1.getInsuranceFundVaultPublicKey)(this.driftClient.program.programId, spotMarketIndex);
+        const spotMarket = this.driftClient.getSpotMarketAccount(spotMarketIndex);
+        if (!spotMarket) {
+            throw new Error(`Spot market ${spotMarketIndex} not found on driftClient`);
+        }
+        return await this.program.methods
+            .cancelRequestRemoveInsuranceFundStake(spotMarketIndex)
+            .accounts({
+            vault: vault,
+            manager: this.driftClient.wallet.publicKey,
+            driftSpotMarket: spotMarket.pubkey,
+            insuranceFundStake: ifStakeAccountPublicKey,
+            insuranceFundVault: ifVaultPublicKey,
+            driftUserStats: vaultAccount.userStats,
+            driftProgram: this.driftClient.program.programId,
+        })
+            .rpc();
+    }
+    async removeInsuranceFundStake(vault, spotMarketIndex, managerTokenAccount) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const ifStakeAccountPublicKey = (0, sdk_1.getInsuranceFundStakeAccountPublicKey)(this.driftClient.program.programId, vault, spotMarketIndex);
+        const ifVaultPublicKey = await (0, sdk_1.getInsuranceFundVaultPublicKey)(this.driftClient.program.programId, spotMarketIndex);
+        const spotMarket = this.driftClient.getSpotMarketAccount(spotMarketIndex);
+        if (!spotMarket) {
+            throw new Error(`Spot market ${spotMarketIndex} not found on driftClient`);
+        }
+        if (!managerTokenAccount) {
+            managerTokenAccount = (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarket.mint, this.driftClient.wallet.publicKey);
+        }
+        const ifVaultTokenAccount = (0, addresses_1.getInsuranceFundTokenVaultAddressSync)(this.program.programId, vault, spotMarketIndex);
+        return await this.program.methods
+            .removeInsuranceFundStake(spotMarketIndex)
+            .accounts({
+            vault: vault,
+            driftSpotMarket: spotMarket.pubkey,
+            insuranceFundStake: ifStakeAccountPublicKey,
+            insuranceFundVault: ifVaultPublicKey,
+            managerTokenAccount,
+            vaultIfTokenAccount: ifVaultTokenAccount,
             driftState: await this.driftClient.getStatePublicKey(),
+            driftUserStats: vaultAccount.userStats,
+            driftSigner: this.driftClient.getStateAccount().signer,
             driftProgram: this.driftClient.program.programId,
+            tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
         })
             .rpc();
     }

package/package.json

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

package/src/addresses.ts

@@ -42,6 +42,21 @@ export function getTokenVaultAddressSync(
 	)[0];
 }
 
+export function getInsuranceFundTokenVaultAddressSync(
+	programId: PublicKey,
+	vault: PublicKey,
+	marketIndex: number
+): PublicKey {
+	return PublicKey.findProgramAddressSync(
+		[
+			Buffer.from(anchor.utils.bytes.utf8.encode('vault_token_account')),
+			vault.toBuffer(),
+			new anchor.BN(marketIndex).toArrayLike(Buffer, 'le', 2),
+		],
+		programId
+	)[0];
+}
+
 export function getVaultProtocolAddressSync(
 	programId: PublicKey,
 	vault: PublicKey

package/src/idl/drift_vaults.json

@@ -1137,21 +1137,227 @@
           "isMut": false,
           "isSigner": false
         },
+        {
+          "name": "driftSpotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftSpotMarketMint",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "vaultTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftUserStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftState",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "tokenProgram",
+          "isMut": false,
+          "isSigner": false
+        },
         {
           "name": "systemProgram",
           "isMut": false,
           "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        }
+      ]
+    },
+    {
+      "name": "addInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "driftSpotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftSpotMarketVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "managerTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "vaultIfTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftUserStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftState",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftSigner",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "tokenProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        },
+        {
+          "name": "amount",
+          "type": "u64"
+        }
+      ]
+    },
+    {
+      "name": "requestRemoveInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
         },
         {
           "name": "driftSpotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftUserStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftProgram",
           "isMut": false,
           "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        },
+        {
+          "name": "amount",
+          "type": "u64"
+        }
+      ]
+    },
+    {
+      "name": "removeInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "driftSpotMarket",
+          "isMut": true,
+          "isSigner": false
         },
         {
           "name": "insuranceFundStake",
           "isMut": true,
           "isSigner": false
         },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "managerTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "vaultIfTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
         {
           "name": "driftUserStats",
           "isMut": true,
@@ -1162,6 +1368,62 @@
           "isMut": false,
           "isSigner": false
         },
+        {
+          "name": "driftSigner",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "tokenProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        }
+      ]
+    },
+    {
+      "name": "cancelRequestRemoveInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "driftSpotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftUserStats",
+          "isMut": true,
+          "isSigner": false
+        },
         {
           "name": "driftProgram",
           "isMut": false,