@drift-labs/sdk 2.10.0-beta.3 → 2.10.0-beta.4

package/lib/types.d.ts

@@ -317,8 +317,8 @@ export declare enum TradeSide {
     Buy = 1,
     Sell = 2
 }
-export type CandleResolution = '1' | '5' | '15' | '60' | '240' | 'D' | 'W' | 'M';
-export type NewUserRecord = {
+export declare type CandleResolution = '1' | '5' | '15' | '60' | '240' | 'D' | 'W' | 'M';
+export declare type NewUserRecord = {
     ts: BN;
     userAuthority: PublicKey;
     user: PublicKey;
@@ -326,7 +326,7 @@ export type NewUserRecord = {
     name: number[];
     referrer: PublicKey;
 };
-export type DepositRecord = {
+export declare type DepositRecord = {
     ts: BN;
     userAuthority: PublicKey;
     user: PublicKey;
@@ -347,7 +347,7 @@ export type DepositRecord = {
     explanation: DepositExplanation;
     transferUser?: PublicKey;
 };
-export type SpotInterestRecord = {
+export declare type SpotInterestRecord = {
     ts: BN;
     marketIndex: number;
     depositBalance: BN;
@@ -358,7 +358,7 @@ export type SpotInterestRecord = {
     optimalBorrowRate: number;
     maxBorrowRate: number;
 };
-export type CurveRecord = {
+export declare type CurveRecord = {
     ts: BN;
     recordId: BN;
     marketIndex: number;
@@ -406,7 +406,7 @@ export declare type InsuranceFundStakeRecord = {
     userIfSharesAfter: BN;
     totalIfSharesAfter: BN;
 };
-export type LPRecord = {
+export declare type LPRecord = {
     ts: BN;
     user: PublicKey;
     action: LPAction;
@@ -427,7 +427,7 @@ export declare class LPAction {
         settleLiquidity: {};
     };
 }
-export type FundingRateRecord = {
+export declare type FundingRateRecord = {
     ts: BN;
     recordId: BN;
     marketIndex: number;
@@ -442,7 +442,7 @@ export type FundingRateRecord = {
     baseAssetAmountWithAmm: BN;
     baseAssetAmountWithUnsettledLp: BN;
 };
-export type FundingPaymentRecord = {
+export declare type FundingPaymentRecord = {
     ts: BN;
     userAuthority: PublicKey;
     user: PublicKey;
@@ -453,7 +453,7 @@ export type FundingPaymentRecord = {
     ammCumulativeFundingLong: BN;
     ammCumulativeFundingShort: BN;
 };
-export type LiquidationRecord = {
+export declare type LiquidationRecord = {
     ts: BN;
     user: PublicKey;
     liquidator: PublicKey;
@@ -491,7 +491,7 @@ export declare class LiquidationType {
         liquidateSpot: {};
     };
 }
-export type LiquidatePerpRecord = {
+export declare type LiquidatePerpRecord = {
     marketIndex: number;
     oraclePrice: BN;
     baseAssetAmount: BN;
@@ -503,7 +503,7 @@ export type LiquidatePerpRecord = {
     liquidatorFee: BN;
     ifFee: BN;
 };
-export type LiquidateSpotRecord = {
+export declare type LiquidateSpotRecord = {
     assetMarketIndex: number;
     assetPrice: BN;
     assetTransfer: BN;
@@ -512,7 +512,7 @@ export type LiquidateSpotRecord = {
     liabilityTransfer: BN;
     ifFee: BN;
 };
-export type LiquidateBorrowForPerpPnlRecord = {
+export declare type LiquidateBorrowForPerpPnlRecord = {
     perpMarketIndex: number;
     marketOraclePrice: BN;
     pnlTransfer: BN;
@@ -520,7 +520,7 @@ export type LiquidateBorrowForPerpPnlRecord = {
     liabilityPrice: BN;
     liabilityTransfer: BN;
 };
-export type LiquidatePerpPnlForDepositRecord = {
+export declare type LiquidatePerpPnlForDepositRecord = {
     perpMarketIndex: number;
     marketOraclePrice: BN;
     pnlTransfer: BN;
@@ -528,7 +528,7 @@ export type LiquidatePerpPnlForDepositRecord = {
     assetPrice: BN;
     assetTransfer: BN;
 };
-export type PerpBankruptcyRecord = {
+export declare type PerpBankruptcyRecord = {
     marketIndex: number;
     pnl: BN;
     ifPayment: BN;
@@ -536,13 +536,13 @@ export type PerpBankruptcyRecord = {
     clawbackUserPayment: BN | null;
     cumulativeFundingRateDelta: BN;
 };
-export type SpotBankruptcyRecord = {
+export declare type SpotBankruptcyRecord = {
     marketIndex: number;
     borrowAmount: BN;
     cumulativeDepositInterestDelta: BN;
     ifPayment: BN;
 };
-export type SettlePnlRecord = {
+export declare type SettlePnlRecord = {
     ts: BN;
     user: PublicKey;
     marketIndex: number;
@@ -553,12 +553,12 @@ export type SettlePnlRecord = {
     settlePrice: BN;
     explanation: SettlePnlExplanation;
 };
-export type OrderRecord = {
+export declare type OrderRecord = {
     ts: BN;
     user: PublicKey;
     order: Order;
 };
-export type OrderActionRecord = {
+export declare type OrderActionRecord = {
     ts: BN;
     action: OrderAction;
     actionExplanation: OrderActionExplanation;
@@ -588,7 +588,7 @@ export type OrderActionRecord = {
     makerOrderCumulativeQuoteAssetAmountFilled: BN | null;
     oraclePrice: BN;
 };
-export type StateAccount = {
+export declare type StateAccount = {
     admin: PublicKey;
     exchangeStatus: ExchangeStatus;
     whitelistMint: PublicKey;
@@ -612,7 +612,7 @@ export type StateAccount = {
     initialPctToLiquidate: number;
     liquidationDuration: number;
 };
-export type PerpMarketAccount = {
+export declare type PerpMarketAccount = {
     status: MarketStatus;
     contractType: ContractType;
     contractTier: ContractTier;
@@ -645,7 +645,7 @@ export type PerpMarketAccount = {
         quoteMaxInsurance: BN;
     };
 };
-export type HistoricalOracleData = {
+export declare type HistoricalOracleData = {
     lastOraclePrice: BN;
     lastOracleDelay: BN;
     lastOracleConf: BN;
@@ -653,14 +653,14 @@ export type HistoricalOracleData = {
     lastOraclePriceTwap5Min: BN;
     lastOraclePriceTwapTs: BN;
 };
-export type HistoricalIndexData = {
+export declare type HistoricalIndexData = {
     lastIndexBidPrice: BN;
     lastIndexAskPrice: BN;
     lastIndexPriceTwap: BN;
     lastIndexPriceTwap5Min: BN;
     lastIndexPriceTwapTs: BN;
 };
-export type SpotMarketAccount = {
+export declare type SpotMarketAccount = {
     status: MarketStatus;
     assetTier: AssetTier;
     marketIndex: number;
@@ -715,11 +715,11 @@ export type SpotMarketAccount = {
     totalSpotFee: BN;
     ordersEnabled: boolean;
 };
-export type PoolBalance = {
+export declare type PoolBalance = {
     scaledBalance: BN;
     marketIndex: number;
 };
-export type AMM = {
+export declare type AMM = {
     baseAssetReserve: BN;
     sqrtK: BN;
     cumulativeFundingRate: BN;
@@ -797,7 +797,7 @@ export type AMM = {
     askBaseAssetReserve: BN;
     askQuoteAssetReserve: BN;
 };
-export type PerpPosition = {
+export declare type PerpPosition = {
     baseAssetAmount: BN;
     lastCumulativeFundingRate: BN;
     marketIndex: number;
@@ -813,7 +813,7 @@ export type PerpPosition = {
     lastBaseAssetAmountPerLp: BN;
     lastQuoteAssetAmountPerLp: BN;
 };
-export type UserStatsAccount = {
+export declare type UserStatsAccount = {
     numberOfSubAccounts: number;
     numberOfSubAccountsCreated: number;
     makerVolume30D: BN;
@@ -835,7 +835,7 @@ export type UserStatsAccount = {
     authority: PublicKey;
     ifStakedQuoteAssetAmount: BN;
 };
-export type UserAccount = {
+export declare type UserAccount = {
     authority: PublicKey;
     delegate: PublicKey;
     name: number[];
@@ -857,7 +857,7 @@ export type UserAccount = {
     liquidationStartSlot: BN;
     isMarginTradingEnabled: boolean;
 };
-export type SpotPosition = {
+export declare type SpotPosition = {
     marketIndex: number;
     balanceType: SpotBalanceType;
     scaledBalance: BN;
@@ -866,7 +866,7 @@ export type SpotPosition = {
     openAsks: BN;
     cumulativeDeposits: BN;
 };
-export type Order = {
+export declare type Order = {
     status: OrderStatus;
     orderType: OrderType;
     marketType: MarketType;
@@ -892,7 +892,7 @@ export type Order = {
     auctionEndPrice: BN;
     maxTs: BN;
 };
-export type OrderParams = {
+export declare type OrderParams = {
     orderType: OrderType;
     marketType: MarketType;
     userOrderId: number;
@@ -912,29 +912,29 @@ export type OrderParams = {
     auctionStartPrice: BN | null;
     auctionEndPrice: BN | null;
 };
-export type NecessaryOrderParams = {
+export declare type NecessaryOrderParams = {
     orderType: OrderType;
     marketIndex: number;
     baseAssetAmount: BN;
     direction: PositionDirection;
 };
-export type OptionalOrderParams = {
+export declare type OptionalOrderParams = {
     [Property in keyof OrderParams]?: OrderParams[Property];
 } & NecessaryOrderParams;
 export declare const DefaultOrderParams: OrderParams;
-export type MakerInfo = {
+export declare type MakerInfo = {
     maker: PublicKey;
     makerStats: PublicKey;
     makerUserAccount: UserAccount;
     order: Order;
 };
-export type TakerInfo = {
+export declare type TakerInfo = {
     taker: PublicKey;
     takerStats: PublicKey;
     takerUserAccount: UserAccount;
     order: Order;
 };
-export type ReferrerInfo = {
+export declare type ReferrerInfo = {
     referrer: PublicKey;
     referrerStats: PublicKey;
 };
@@ -943,7 +943,7 @@ export interface IWallet {
     signAllTransactions(txs: Transaction[]): Promise<Transaction[]>;
     publicKey: PublicKey;
 }
-export type FeeStructure = {
+export declare type FeeStructure = {
     feeTiers: FeeTier[];
     makerRebateNumerator: BN;
     makerRebateDenominator: BN;
@@ -951,7 +951,7 @@ export type FeeStructure = {
     flatFillerFee: BN;
     referrerRewardEpochUpperBound: BN;
 };
-export type FeeTier = {
+export declare type FeeTier = {
     feeNumerator: number;
     feeDenominator: number;
     makerRebateNumerator: number;
@@ -961,12 +961,12 @@ export type FeeTier = {
     refereeFeeNumerator: number;
     refereeFeeDenominator: number;
 };
-export type OrderFillerRewardStructure = {
+export declare type OrderFillerRewardStructure = {
     rewardNumerator: BN;
     rewardDenominator: BN;
     timeBasedRewardLowerBound: BN;
 };
-export type OracleGuardRails = {
+export declare type OracleGuardRails = {
     priceDivergence: {
         markOracleDivergenceNumerator: BN;
         markOracleDivergenceDenominator: BN;
@@ -978,8 +978,8 @@ export type OracleGuardRails = {
         tooVolatileRatio: BN;
     };
 };
-export type MarginCategory = 'Initial' | 'Maintenance';
-export type InsuranceFundStake = {
+export declare type MarginCategory = 'Initial' | 'Maintenance';
+export declare type InsuranceFundStake = {
     marketIndex: number;
     authority: PublicKey;
     ifShares: BN;
@@ -988,7 +988,7 @@ export type InsuranceFundStake = {
     lastWithdrawRequestValue: BN;
     lastWithdrawRequestTs: BN;
 };
-export type SerumV3FulfillmentConfigAccount = {
+export declare type SerumV3FulfillmentConfigAccount = {
     fulfillmentType: SpotFulfillmentType;
     status: SpotFulfillmentStatus;
     pubkey: PublicKey;

package/lib/user.d.ts

@@ -22,7 +22,7 @@ export declare class User {
      * Subscribe to User state accounts
      * @returns SusbcriptionSuccess result
      */
-    subscribe(): Promise<boolean>;
+    subscribe(userAccount?: UserAccount): Promise<boolean>;
     /**
      *	Forces the accountSubscriber to fetch account updates from rpc
      */

package/lib/user.js

@@ -12,12 +12,6 @@ const pollingUserAccountSubscriber_1 = require("./accounts/pollingUserAccountSub
 const webSocketUserAccountSubscriber_1 = require("./accounts/webSocketUserAccountSubscriber");
 const spotPosition_1 = require("./math/spotPosition");
 class User {
-    get isSubscribed() {
-        return this._isSubscribed && this.accountSubscriber.isSubscribed;
-    }
-    set isSubscribed(val) {
-        this._isSubscribed = val;
-    }
     constructor(config) {
         var _a;
         this._isSubscribed = false;
@@ -31,12 +25,18 @@ class User {
         }
         this.eventEmitter = this.accountSubscriber.eventEmitter;
     }
+    get isSubscribed() {
+        return this._isSubscribed && this.accountSubscriber.isSubscribed;
+    }
+    set isSubscribed(val) {
+        this._isSubscribed = val;
+    }
     /**
      * Subscribe to User state accounts
      * @returns SusbcriptionSuccess result
      */
-    async subscribe() {
-        this.isSubscribed = await this.accountSubscriber.subscribe();
+    async subscribe(userAccount) {
+        this.isSubscribed = await this.accountSubscriber.subscribe(userAccount);
         return this.isSubscribed;
     }
     /**

package/lib/userConfig.d.ts

@@ -1,12 +1,12 @@
 import { DriftClient } from './driftClient';
 import { PublicKey } from '@solana/web3.js';
 import { BulkAccountLoader } from './accounts/bulkAccountLoader';
-export type UserConfig = {
+export declare type UserConfig = {
     accountSubscription?: UserSubscriptionConfig;
     driftClient: DriftClient;
     userAccountPublicKey: PublicKey;
 };
-export type UserSubscriptionConfig = {
+export declare type UserSubscriptionConfig = {
     type: 'websocket';
 } | {
     type: 'polling';

package/lib/userMap/userMap.js

@@ -11,6 +11,7 @@ class UserMap {
     }
     async fetchAllUsers() {
         const userArray = [];
+        const userAccountArray = [];
         const programUserAccounts = (await this.driftClient.program.account.user.all());
         for (const programUserAccount of programUserAccounts) {
             if (this.userMap.has(programUserAccount.publicKey.toString())) {
@@ -22,12 +23,13 @@ class UserMap {
                 accountSubscription: this.accountSubscription,
             });
             userArray.push(user);
+            userAccountArray.push(programUserAccount.account);
         }
         if (this.accountSubscription.type === 'polling') {
             await (0, __1.bulkPollingUserSubscribe)(userArray, this.accountSubscription.accountLoader);
         }
         else {
-            await Promise.all(userArray.map((user) => user.subscribe()));
+            await Promise.all(userArray.map((user, i) => user.subscribe(userAccountArray[i])));
         }
         for (const user of userArray) {
             this.userMap.set(user.getUserAccountPublicKey().toString(), user);

package/lib/userMap/userStatsMap.js

@@ -14,6 +14,7 @@ class UserStatsMap {
     }
     async fetchAllUserStats() {
         const userStatArray = [];
+        const userStatsAccountArray = [];
         const programUserAccounts = (await this.driftClient.program.account.userStats.all());
         for (const programUserAccount of programUserAccounts) {
             const userStat = programUserAccount.account;
@@ -26,12 +27,13 @@ class UserStatsMap {
                 accountSubscription: this.accountSubscription,
             });
             userStatArray.push(chUserStat);
+            userStatsAccountArray.push(userStat);
         }
         if (this.accountSubscription.type === 'polling') {
             await (0, __1.bulkPollingUserStatsSubscribe)(userStatArray, this.accountSubscription.accountLoader);
         }
         else {
-            await Promise.all(userStatArray.map((userStat) => userStat.subscribe()));
+            await Promise.all(userStatArray.map((userStat, i) => userStat.subscribe(userStatsAccountArray[i])));
         }
         for (const userStat of userStatArray) {
             this.userStatsMap.set(userStat.getAccount().authority.toString(), userStat);

package/lib/userStats.d.ts

@@ -9,7 +9,7 @@ export declare class UserStats {
     accountSubscriber: UserStatsAccountSubscriber;
     isSubscribed: boolean;
     constructor(config: UserStatsConfig);
-    subscribe(): Promise<boolean>;
+    subscribe(userStatsAccount?: UserStatsAccount): Promise<boolean>;
     fetchAccounts(): Promise<void>;
     unsubscribe(): Promise<void>;
     getAccountAndSlot(): DataAndSlot<UserStatsAccount>;

package/lib/userStats.js

@@ -17,8 +17,8 @@ class UserStats {
             this.accountSubscriber = new webSocketUserStatsAccountSubsriber_1.WebSocketUserStatsAccountSubscriber(config.driftClient.program, config.userStatsAccountPublicKey);
         }
     }
-    async subscribe() {
-        this.isSubscribed = await this.accountSubscriber.subscribe();
+    async subscribe(userStatsAccount) {
+        this.isSubscribed = await this.accountSubscriber.subscribe(userStatsAccount);
         return this.isSubscribed;
     }
     async fetchAccounts() {

package/lib/userStatsConfig.d.ts

@@ -1,12 +1,12 @@
 import { DriftClient } from './driftClient';
 import { PublicKey } from '@solana/web3.js';
 import { BulkAccountLoader } from './accounts/bulkAccountLoader';
-export type UserStatsConfig = {
+export declare type UserStatsConfig = {
     accountSubscription?: UserStatsSubscriptionConfig;
     driftClient: DriftClient;
     userStatsAccountPublicKey: PublicKey;
 };
-export type UserStatsSubscriptionConfig = {
+export declare type UserStatsSubscriptionConfig = {
     type: 'websocket';
 } | {
     type: 'polling';

package/package.json

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

package/src/accounts/types.ts

@@ -17,6 +17,8 @@ export interface AccountSubscriber<T> {
 	subscribe(onChange: (data: T) => void): Promise<void>;
 	fetch(): Promise<void>;
 	unsubscribe(): Promise<void>;
+
+	setData(userAccount: T): void;
 }
 
 export class NotSubscribedError extends Error {
@@ -69,7 +71,7 @@ export interface UserAccountSubscriber {
 	eventEmitter: StrictEventEmitter<EventEmitter, UserAccountEvents>;
 	isSubscribed: boolean;
 
-	subscribe(): Promise<boolean>;
+	subscribe(userAccount?: UserAccount): Promise<boolean>;
 	fetch(): Promise<void>;
 	unsubscribe(): Promise<void>;
 
@@ -144,7 +146,7 @@ export interface UserStatsAccountSubscriber {
 	eventEmitter: StrictEventEmitter<EventEmitter, UserStatsAccountEvents>;
 	isSubscribed: boolean;
 
-	subscribe(): Promise<boolean>;
+	subscribe(userStatsAccount?: UserStatsAccount): Promise<boolean>;
 	fetch(): Promise<void>;
 	unsubscribe(): Promise<void>;
 

package/src/accounts/webSocketAccountSubscriber.ts

@@ -32,7 +32,9 @@ export class WebSocketAccountSubscriber<T> implements AccountSubscriber<T> {
 		}
 
 		this.onChange = onChange;
-		await this.fetch();
+		if (!this.dataAndSlot) {
+			await this.fetch();
+		}
 
 		this.listenerId = this.program.provider.connection.onAccountChange(
 			this.accountPublicKey,
@@ -43,6 +45,17 @@ export class WebSocketAccountSubscriber<T> implements AccountSubscriber<T> {
 		);
 	}
 
+	setData(data: T): void {
+		if (this.dataAndSlot) {
+			return;
+		}
+
+		this.dataAndSlot = {
+			data,
+			slot: undefined,
+		};
+	}
+
 	async fetch(): Promise<void> {
 		const rpcResponse =
 			await this.program.provider.connection.getAccountInfoAndContext(

package/src/accounts/webSocketUserAccountSubscriber.ts

@@ -27,7 +27,7 @@ export class WebSocketUserAccountSubscriber implements UserAccountSubscriber {
 		this.eventEmitter = new EventEmitter();
 	}
 
-	async subscribe(): Promise<boolean> {
+	async subscribe(userAccount?: UserAccount): Promise<boolean> {
 		if (this.isSubscribed) {
 			return true;
 		}
@@ -37,6 +37,11 @@ export class WebSocketUserAccountSubscriber implements UserAccountSubscriber {
 			this.program,
 			this.userAccountPublicKey
 		);
+
+		if (userAccount) {
+			this.userDataAccountSubscriber.setData(userAccount);
+		}
+
 		await this.userDataAccountSubscriber.subscribe((data: UserAccount) => {
 			this.eventEmitter.emit('userAccountUpdate', data);
 			this.eventEmitter.emit('update');

package/src/accounts/webSocketUserStatsAccountSubsriber.ts

@@ -29,7 +29,7 @@ export class WebSocketUserStatsAccountSubscriber
 		this.eventEmitter = new EventEmitter();
 	}
 
-	async subscribe(): Promise<boolean> {
+	async subscribe(userStatsAccount?: UserStatsAccount): Promise<boolean> {
 		if (this.isSubscribed) {
 			return true;
 		}
@@ -39,6 +39,11 @@ export class WebSocketUserStatsAccountSubscriber
 			this.program,
 			this.userStatsAccountPublicKey
 		);
+
+		if (userStatsAccount) {
+			this.userStatsAccountSubscriber.setData(userStatsAccount);
+		}
+
 		await this.userStatsAccountSubscriber.subscribe(
 			(data: UserStatsAccount) => {
 				this.eventEmitter.emit('userStatsAccountUpdate', data);

package/src/assert/assert.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.assert = void 0;
+function assert(condition, error) {
+    if (!condition) {
+        throw new Error(error || 'Unspecified AssertionError');
+    }
+}
+exports.assert = assert;