@drift-labs/vaults-sdk 0.1.0 → 0.1.2

package/lib/vaultClient.js

@@ -3,7 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.VaultClient = void 0;
 const sdk_1 = require("@drift-labs/sdk");
 const anchor_1 = require("@coral-xyz/anchor");
-const addresses_1 = require("./addresses");
+const addresses_1 = require("@drift-labs/competitions-sdk/lib/addresses");
+const addresses_2 = require("./addresses");
 const web3_js_1 = require("@solana/web3.js");
 const spl_token_1 = require("@solana/spl-token");
 const bytes_1 = require("@coral-xyz/anchor/dist/cjs/utils/bytes");
@@ -72,12 +73,12 @@ class VaultClient {
         return netSpotValue.add(unrealizedPnl);
     }
     async initializeVault(params) {
-        const vault = addresses_1.getVaultAddressSync(this.program.programId, params.name);
-        const tokenAccount = addresses_1.getTokenVaultAddressSync(this.program.programId, vault);
+        const vault = (0, addresses_2.getVaultAddressSync)(this.program.programId, params.name);
+        const tokenAccount = (0, addresses_2.getTokenVaultAddressSync)(this.program.programId, vault);
         const driftState = await this.driftClient.getStatePublicKey();
         const spotMarket = this.driftClient.getSpotMarketAccount(params.spotMarketIndex);
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault);
-        const userKey = sdk_1.getUserAccountPublicKeySync(this.driftClient.program.programId, vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault);
+        const userKey = (0, sdk_1.getUserAccountPublicKeySync)(this.driftClient.program.programId, vault);
         const accounts = {
             driftSpotMarket: spotMarket.pubkey,
             driftSpotMarketMint: spotMarket.mint,
@@ -111,6 +112,23 @@ class VaultClient {
         })
             .rpc();
     }
+    /**
+     * Updates the vault margin trading status.
+     * @param vault vault address to update
+     * @param enabeld whether to enable margin trading
+     * @returns
+     */
+    async updateMarginTradingEnabled(vault, enabled) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        return await this.program.methods
+            .updateMarginTradingEnabled(enabled)
+            .accounts({
+            vault: vault,
+            driftUser: vaultAccount.user,
+            driftProgram: this.driftClient.program.programId,
+        })
+            .rpc();
+    }
     /**
      *
      * @param vault vault address to deposit to
@@ -134,12 +152,12 @@ class VaultClient {
             .accounts({
             vault,
             vaultTokenAccount: vaultAccount.tokenAccount,
-            driftUser: await sdk_1.getUserAccountPublicKey(this.driftClient.program.programId, vault),
+            driftUser: await (0, sdk_1.getUserAccountPublicKey)(this.driftClient.program.programId, vault),
             driftProgram: this.driftClient.program.programId,
-            driftUserStats: sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault),
+            driftUserStats: (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault),
             driftState: await this.driftClient.getStatePublicKey(),
             driftSpotMarketVault: driftSpotMarket.vault,
-            userTokenAccount: spl_token_1.getAssociatedTokenAddressSync(driftSpotMarket.mint, this.driftClient.wallet.publicKey),
+            userTokenAccount: (0, spl_token_1.getAssociatedTokenAddressSync)(driftSpotMarket.mint, this.driftClient.wallet.publicKey),
             tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
         })
             .remainingAccounts(remainingAccounts)
@@ -159,7 +177,7 @@ class VaultClient {
             userAccounts: [user.getUserAccount()],
             writableSpotMarketIndexes: [vaultAccount.spotMarketIndex],
         });
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const accounts = {
             vault,
@@ -187,7 +205,7 @@ class VaultClient {
     }
     async managerCancelWithdrawRequest(vault) {
         const vaultAccount = await this.program.account.vault.fetch(vault);
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const accounts = {
             manager: this.driftClient.wallet.publicKey,
@@ -245,12 +263,12 @@ class VaultClient {
                 vault,
                 manager: this.driftClient.wallet.publicKey,
                 vaultTokenAccount: vaultAccount.tokenAccount,
-                driftUser: await sdk_1.getUserAccountPublicKey(this.driftClient.program.programId, vault),
+                driftUser: await (0, sdk_1.getUserAccountPublicKey)(this.driftClient.program.programId, vault),
                 driftProgram: this.driftClient.program.programId,
-                driftUserStats: sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault),
+                driftUserStats: (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault),
                 driftState: await this.driftClient.getStatePublicKey(),
                 driftSpotMarketVault: spotMarket.vault,
-                userTokenAccount: spl_token_1.getAssociatedTokenAddressSync(spotMarket.mint, this.driftClient.wallet.publicKey),
+                userTokenAccount: (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarket.mint, this.driftClient.wallet.publicKey),
                 driftSigner: this.driftClient.getStateAccount().signer,
                 tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
             },
@@ -288,8 +306,8 @@ class VaultClient {
             vault,
             vaultDepositor,
             manager: this.driftClient.wallet.publicKey,
-            driftUserStats: sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vault),
-            driftUser: await sdk_1.getUserAccountPublicKey(this.driftClient.program.programId, vault),
+            driftUserStats: (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vault),
+            driftUser: await (0, sdk_1.getUserAccountPublicKey)(this.driftClient.program.programId, vault),
             driftState: await this.driftClient.getStatePublicKey(),
             driftSigner: this.driftClient.getStateAccount().signer,
             driftProgram: this.driftClient.program.programId,
@@ -302,7 +320,7 @@ class VaultClient {
         });
     }
     createInitVaultDepositorIx(vault, authority) {
-        const vaultDepositor = addresses_1.getVaultDepositorAddressSync(this.program.programId, vault, authority);
+        const vaultDepositor = (0, addresses_2.getVaultDepositorAddressSync)(this.program.programId, vault, authority);
         const accounts = {
             vaultDepositor,
             vault,
@@ -325,7 +343,7 @@ class VaultClient {
      * @returns
      */
     async initializeVaultDepositor(vault, authority) {
-        const vaultDepositor = addresses_1.getVaultDepositorAddressSync(this.program.programId, vault, authority);
+        const vaultDepositor = (0, addresses_2.getVaultDepositorAddressSync)(this.program.programId, vault, authority);
         const accounts = {
             vaultDepositor,
             vault,
@@ -368,7 +386,7 @@ class VaultClient {
             userAccounts: [user.getUserAccount()],
             writableSpotMarketIndexes: [vaultAccount.spotMarketIndex],
         });
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vaultPubKey);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vaultPubKey);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const spotMarket = this.driftClient.getSpotMarketAccount(vaultAccount.spotMarketIndex);
         const accounts = {
@@ -379,7 +397,7 @@ class VaultClient {
             driftUser: vaultAccount.user,
             driftState: driftStateKey,
             driftSpotMarketVault: spotMarket.vault,
-            userTokenAccount: spl_token_1.getAssociatedTokenAddressSync(spotMarket.mint, this.driftClient.wallet.publicKey, true),
+            userTokenAccount: (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarket.mint, this.driftClient.wallet.publicKey, true),
             driftProgram: this.driftClient.program.programId,
             tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
         };
@@ -421,7 +439,7 @@ class VaultClient {
         const remainingAccounts = this.driftClient.getRemainingAccounts({
             userAccounts: [user.getUserAccount()],
         });
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vaultDepositorAccount.vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vaultDepositorAccount.vault);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const accounts = {
             vault: vaultDepositorAccount.vault,
@@ -460,7 +478,7 @@ class VaultClient {
             userAccounts: [user.getUserAccount()],
             writableSpotMarketIndexes: [vaultAccount.spotMarketIndex],
         });
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vaultDepositorAccount.vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vaultDepositorAccount.vault);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const spotMarket = this.driftClient.getSpotMarketAccount(vaultAccount.spotMarketIndex);
         const accounts = {
@@ -472,7 +490,7 @@ class VaultClient {
             driftState: driftStateKey,
             driftSpotMarketVault: spotMarket.vault,
             driftSigner: this.driftClient.getStateAccount().signer,
-            userTokenAccount: spl_token_1.getAssociatedTokenAddressSync(spotMarket.mint, this.driftClient.wallet.publicKey, true),
+            userTokenAccount: (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarket.mint, this.driftClient.wallet.publicKey, true),
             driftProgram: this.driftClient.program.programId,
             tokenProgram: spl_token_1.TOKEN_PROGRAM_ID,
         };
@@ -497,7 +515,7 @@ class VaultClient {
     async cancelRequestWithdraw(vaultDepositor) {
         const vaultDepositorAccount = await this.program.account.vaultDepositor.fetch(vaultDepositor);
         const vaultAccount = await this.program.account.vault.fetch(vaultDepositorAccount.vault);
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vaultDepositorAccount.vault);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vaultDepositorAccount.vault);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const accounts = {
             vault: vaultDepositorAccount.vault,
@@ -551,7 +569,7 @@ class VaultClient {
             userAccounts: [user.getUserAccount()],
             writableSpotMarketIndexes: [vaultAccount.spotMarketIndex],
         });
-        const userStatsKey = sdk_1.getUserStatsAccountPublicKey(this.driftClient.program.programId, vaultPubKey);
+        const userStatsKey = (0, sdk_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, vaultPubKey);
         const driftStateKey = await this.driftClient.getStatePublicKey();
         const accounts = {
             vault: vaultPubKey,
@@ -592,5 +610,48 @@ class VaultClient {
         const { txSig } = await this.driftClient.sendTransaction(tx, [], this.driftClient.opts);
         return txSig;
     }
+    /**
+     * Initializes an insurance fund stake for the vault.
+     * @param vault vault address to update
+     * @param spotMarketIndex spot market index of the insurance fund stake
+     * @returns
+     */
+    async initializeInsuranceFundStake(vault, spotMarketIndex) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const ifStakeAccountPublicKey = (0, sdk_1.getInsuranceFundStakeAccountPublicKey)(this.driftClient.program.programId, vault, spotMarketIndex);
+        return await this.program.methods
+            .initializeInsuranceFundStake(spotMarketIndex)
+            .accounts({
+            vault: vault,
+            driftSpotMarket: this.driftClient.getSpotMarketAccount(spotMarketIndex).pubkey,
+            insuranceFundStake: ifStakeAccountPublicKey,
+            driftUserStats: vaultAccount.userStats,
+            driftState: await this.driftClient.getStatePublicKey(),
+            driftProgram: this.driftClient.program.programId,
+        })
+            .rpc();
+    }
+    /**
+     * Initializes a DriftCompetitions Competitor account for the vault.
+     * @param vault vault address to initialize Competitor for
+     * @param competitionName name of the competition to initialize for
+     * @returns
+     */
+    async initializeCompetitor(vault, competitionsClient, competitionName) {
+        const vaultAccount = await this.program.account.vault.fetch(vault);
+        const encodedName = (0, sdk_1.encodeName)(competitionName);
+        const competitionAddress = (0, addresses_1.getCompetitionAddressSync)(competitionsClient.program.programId, encodedName);
+        const competitorAddress = (0, addresses_1.getCompetitorAddressSync)(competitionsClient.program.programId, competitionAddress, vault);
+        return await this.program.methods
+            .initializeCompetitor()
+            .accounts({
+            vault: vault,
+            competitor: competitorAddress,
+            driftCompetitions: competitionAddress,
+            driftUserStats: vaultAccount.userStats,
+            driftCompetitionsProgram: competitionsClient.program.programId,
+        })
+            .rpc();
+    }
 }
 exports.VaultClient = VaultClient;

package/package.json

@@ -1,16 +1,14 @@
 {
   "name": "@drift-labs/vaults-sdk",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
   "directories": {
     "lib": "lib"
   },
-  "peerDependencies": {
-    "@drift-labs/sdk": "2.40.0-beta.1"
-  },
   "dependencies": {
     "@coral-xyz/anchor": "^0.26.0",
+    "@drift-labs/sdk": "2.43.0-beta.5",
     "@solana/web3.js": "1.73.2",
     "commander": "^11.0.0",
     "dotenv": "^16.3.1",

package/src/idl/drift_vaults.json

@@ -789,6 +789,118 @@
         }
       ],
       "args": []
+    },
+    {
+      "name": "initializeInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "payer",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "rent",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "systemProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftSpotMarket",
+          "isMut": false,
+          "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
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        }
+      ]
+    },
+    {
+      "name": "initializeCompetitor",
+      "accounts": [
+        {
+          "name": "vault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "manager",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "payer",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "rent",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "systemProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "competitor",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftCompetitions",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftUserStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "driftCompetitionsProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": []
     }
   ],
   "accounts": [

package/src/types/drift_vaults.ts

@@ -789,6 +789,118 @@ export type DriftVaults = {
 				}
 			];
 			args: [];
+		},
+		{
+			name: 'initializeInsuranceFundStake';
+			accounts: [
+				{
+					name: 'vault';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'manager';
+					isMut: false;
+					isSigner: true;
+				},
+				{
+					name: 'payer';
+					isMut: true;
+					isSigner: true;
+				},
+				{
+					name: 'rent';
+					isMut: false;
+					isSigner: false;
+				},
+				{
+					name: 'systemProgram';
+					isMut: false;
+					isSigner: false;
+				},
+				{
+					name: 'driftSpotMarket';
+					isMut: false;
+					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;
+				}
+			];
+			args: [
+				{
+					name: 'marketIndex';
+					type: 'u16';
+				}
+			];
+		},
+		{
+			name: 'initializeCompetitor';
+			accounts: [
+				{
+					name: 'vault';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'manager';
+					isMut: false;
+					isSigner: true;
+				},
+				{
+					name: 'payer';
+					isMut: true;
+					isSigner: true;
+				},
+				{
+					name: 'rent';
+					isMut: false;
+					isSigner: false;
+				},
+				{
+					name: 'systemProgram';
+					isMut: false;
+					isSigner: false;
+				},
+				{
+					name: 'competitor';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'driftCompetitions';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'driftUserStats';
+					isMut: true;
+					isSigner: false;
+				},
+				{
+					name: 'driftCompetitionsProgram';
+					isMut: false;
+					isSigner: false;
+				}
+			];
+			args: [];
 		}
 	];
 	accounts: [
@@ -2265,6 +2377,118 @@ export const IDL: DriftVaults = {
 			],
 			args: [],
 		},
+		{
+			name: 'initializeInsuranceFundStake',
+			accounts: [
+				{
+					name: 'vault',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'manager',
+					isMut: false,
+					isSigner: true,
+				},
+				{
+					name: 'payer',
+					isMut: true,
+					isSigner: true,
+				},
+				{
+					name: 'rent',
+					isMut: false,
+					isSigner: false,
+				},
+				{
+					name: 'systemProgram',
+					isMut: false,
+					isSigner: false,
+				},
+				{
+					name: 'driftSpotMarket',
+					isMut: false,
+					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,
+				},
+			],
+			args: [
+				{
+					name: 'marketIndex',
+					type: 'u16',
+				},
+			],
+		},
+		{
+			name: 'initializeCompetitor',
+			accounts: [
+				{
+					name: 'vault',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'manager',
+					isMut: false,
+					isSigner: true,
+				},
+				{
+					name: 'payer',
+					isMut: true,
+					isSigner: true,
+				},
+				{
+					name: 'rent',
+					isMut: false,
+					isSigner: false,
+				},
+				{
+					name: 'systemProgram',
+					isMut: false,
+					isSigner: false,
+				},
+				{
+					name: 'competitor',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'driftCompetitions',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'driftUserStats',
+					isMut: true,
+					isSigner: false,
+				},
+				{
+					name: 'driftCompetitionsProgram',
+					isMut: false,
+					isSigner: false,
+				},
+			],
+			args: [],
+		},
 	],
 	accounts: [
 		{