@drift-labs/sdk 2.82.0-beta.15 → 2.82.0-beta.17

package/VERSION

@@ -1 +1 @@
-2.82.0-beta.15
+2.82.0-beta.17

package/lib/clock/clockSubscriber.d.ts

@@ -0,0 +1,29 @@
+/// <reference types="node" />
+import { Commitment, Connection } from '@solana/web3.js';
+import { EventEmitter } from 'events';
+import StrictEventEmitter from 'strict-event-emitter-types/types/src';
+type ClockSubscriberConfig = {
+    commitment: Commitment;
+    resubTimeoutMs?: number;
+};
+export interface ClockSubscriberEvent {
+    clockUpdate: (ts: number) => void;
+}
+export declare class ClockSubscriber {
+    private connection;
+    private latestSlot;
+    currentTs: number;
+    private subscriptionId;
+    commitment: Commitment;
+    eventEmitter: StrictEventEmitter<EventEmitter, ClockSubscriberEvent>;
+    private timeoutId?;
+    private resubTimeoutMs?;
+    private isUnsubscribing;
+    private receivingData;
+    constructor(connection: Connection, config?: ClockSubscriberConfig);
+    subscribe(): Promise<void>;
+    private setTimeout;
+    getUnixTs(): number;
+    unsubscribe(onResub?: boolean): Promise<void>;
+}
+export {};

package/lib/clock/clockSubscriber.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClockSubscriber = void 0;
+const web3_js_1 = require("@solana/web3.js");
+const events_1 = require("events");
+const __1 = require("..");
+class ClockSubscriber {
+    constructor(connection, config) {
+        this.connection = connection;
+        this.isUnsubscribing = false;
+        this.receivingData = false;
+        this.eventEmitter = new events_1.EventEmitter();
+        this.resubTimeoutMs = config === null || config === void 0 ? void 0 : config.resubTimeoutMs;
+        this.commitment = (config === null || config === void 0 ? void 0 : config.commitment) || 'confirmed';
+        if (this.resubTimeoutMs < 1000) {
+            console.log('resubTimeoutMs should be at least 1000ms to avoid spamming resub');
+        }
+    }
+    async subscribe() {
+        if (this.subscriptionId != null) {
+            return;
+        }
+        this.subscriptionId = this.connection.onAccountChange(web3_js_1.SYSVAR_CLOCK_PUBKEY, (acctInfo, context) => {
+            if (!this.latestSlot || this.latestSlot < context.slot) {
+                if (this.resubTimeoutMs && !this.isUnsubscribing) {
+                    this.receivingData = true;
+                    clearTimeout(this.timeoutId);
+                    this.setTimeout();
+                }
+                this.latestSlot = context.slot;
+                this.currentTs = new __1.BN(acctInfo.data.subarray(32, 39), undefined, 'le').toNumber();
+                this.eventEmitter.emit('clockUpdate', this.currentTs);
+            }
+        }, this.commitment);
+        if (this.resubTimeoutMs) {
+            this.receivingData = true;
+            this.setTimeout();
+        }
+    }
+    setTimeout() {
+        this.timeoutId = setTimeout(async () => {
+            if (this.isUnsubscribing) {
+                // If we are in the process of unsubscribing, do not attempt to resubscribe
+                return;
+            }
+            if (this.receivingData) {
+                console.log(`No new slot in ${this.resubTimeoutMs}ms, slot subscriber resubscribing`);
+                await this.unsubscribe(true);
+                this.receivingData = false;
+                await this.subscribe();
+            }
+        }, this.resubTimeoutMs);
+    }
+    getUnixTs() {
+        return this.currentTs;
+    }
+    async unsubscribe(onResub = false) {
+        if (!onResub) {
+            this.resubTimeoutMs = undefined;
+        }
+        this.isUnsubscribing = true;
+        clearTimeout(this.timeoutId);
+        this.timeoutId = undefined;
+        if (this.subscriptionId != null) {
+            await this.connection.removeAccountChangeListener(this.subscriptionId);
+            this.subscriptionId = undefined;
+            this.isUnsubscribing = false;
+        }
+        else {
+            this.isUnsubscribing = false;
+        }
+    }
+}
+exports.ClockSubscriber = ClockSubscriber;

package/lib/index.d.ts

@@ -101,4 +101,5 @@ export * from './auctionSubscriber/types';
 export * from './memcmp';
 export * from './decode/user';
 export * from './blockhashSubscriber';
+export * from './clock/clockSubscriber';
 export { BN, PublicKey, pyth };

package/lib/index.js

@@ -124,3 +124,4 @@ __exportStar(require("./auctionSubscriber/types"), exports);
 __exportStar(require("./memcmp"), exports);
 __exportStar(require("./decode/user"), exports);
 __exportStar(require("./blockhashSubscriber"), exports);
+__exportStar(require("./clock/clockSubscriber"), exports);

package/lib/math/exchangeStatus.d.ts

@@ -1,6 +1,6 @@
-import { PerpMarketAccount, PerpOperation, SpotMarketAccount, SpotOperation, StateAccount } from '../types';
+import { PerpMarketAccount, PerpOperation, SpotMarketAccount, SpotOperation, StateAccount, InsuranceFundOperation } from '../types';
 export declare function exchangePaused(state: StateAccount): boolean;
 export declare function fillPaused(state: StateAccount, market: PerpMarketAccount | SpotMarketAccount): boolean;
 export declare function ammPaused(state: StateAccount, market: PerpMarketAccount | SpotMarketAccount): boolean;
-export declare function isOperationPaused(pausedOperations: number, operation: PerpOperation | SpotOperation): boolean;
+export declare function isOperationPaused(pausedOperations: number, operation: PerpOperation | SpotOperation | InsuranceFundOperation): boolean;
 export declare function isAmmDrawdownPause(market: PerpMarketAccount): boolean;

package/lib/types.d.ts

@@ -55,6 +55,12 @@ export declare enum SpotOperation {
     WITHDRAW = 4,
     LIQUIDATION = 8
 }
+export declare enum InsuranceFundOperation {
+    INIT = 1,
+    ADD = 2,
+    REQUEST_REMOVE = 4,
+    REMOVE = 8
+}
 export declare enum UserStatus {
     BEING_LIQUIDATED = 1,
     BANKRUPT = 2,
@@ -780,6 +786,7 @@ export type SpotMarketAccount = {
     flashLoanInitialTokenAmount: BN;
     ordersEnabled: boolean;
     pausedOperations: number;
+    ifPausedOperations: number;
 };
 export type PoolBalance = {
     scaledBalance: BN;

package/lib/types.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SwapReduceOnly = exports.DefaultOrderParams = exports.ModifyOrderPolicy = exports.PostOnlyParams = exports.LiquidationType = exports.LPAction = exports.TradeSide = exports.getVariant = exports.isOneOfVariant = exports.isVariant = exports.StakeAction = exports.SpotFulfillmentConfigStatus = exports.SettlePnlExplanation = exports.DepositExplanation = exports.SpotFulfillmentStatus = exports.SpotFulfillmentType = exports.OrderTriggerCondition = exports.OrderActionExplanation = exports.OrderAction = exports.OrderStatus = exports.MarketType = exports.OrderType = exports.OracleSource = exports.DepositDirection = exports.PositionDirection = exports.SpotBalanceType = exports.SwapDirection = exports.AssetTier = exports.ContractTier = exports.ContractType = exports.UserStatus = exports.SpotOperation = exports.PerpOperation = exports.MarketStatus = exports.ExchangeStatus = void 0;
+exports.SwapReduceOnly = exports.DefaultOrderParams = exports.ModifyOrderPolicy = exports.PostOnlyParams = exports.LiquidationType = exports.LPAction = exports.TradeSide = exports.getVariant = exports.isOneOfVariant = exports.isVariant = exports.StakeAction = exports.SpotFulfillmentConfigStatus = exports.SettlePnlExplanation = exports.DepositExplanation = exports.SpotFulfillmentStatus = exports.SpotFulfillmentType = exports.OrderTriggerCondition = exports.OrderActionExplanation = exports.OrderAction = exports.OrderStatus = exports.MarketType = exports.OrderType = exports.OracleSource = exports.DepositDirection = exports.PositionDirection = exports.SpotBalanceType = exports.SwapDirection = exports.AssetTier = exports.ContractTier = exports.ContractType = exports.UserStatus = exports.InsuranceFundOperation = exports.SpotOperation = exports.PerpOperation = exports.MarketStatus = exports.ExchangeStatus = void 0;
 const _1 = require(".");
 // # Utility Types / Enums / Constants
 var ExchangeStatus;
@@ -43,6 +43,13 @@ var SpotOperation;
     SpotOperation[SpotOperation["WITHDRAW"] = 4] = "WITHDRAW";
     SpotOperation[SpotOperation["LIQUIDATION"] = 8] = "LIQUIDATION";
 })(SpotOperation = exports.SpotOperation || (exports.SpotOperation = {}));
+var InsuranceFundOperation;
+(function (InsuranceFundOperation) {
+    InsuranceFundOperation[InsuranceFundOperation["INIT"] = 1] = "INIT";
+    InsuranceFundOperation[InsuranceFundOperation["ADD"] = 2] = "ADD";
+    InsuranceFundOperation[InsuranceFundOperation["REQUEST_REMOVE"] = 4] = "REQUEST_REMOVE";
+    InsuranceFundOperation[InsuranceFundOperation["REMOVE"] = 8] = "REMOVE";
+})(InsuranceFundOperation = exports.InsuranceFundOperation || (exports.InsuranceFundOperation = {}));
 var UserStatus;
 (function (UserStatus) {
     UserStatus[UserStatus["BEING_LIQUIDATED"] = 1] = "BEING_LIQUIDATED";

package/package.json

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

package/src/clock/clockSubscriber.ts

@@ -0,0 +1,113 @@
+import { Commitment, Connection, SYSVAR_CLOCK_PUBKEY } from '@solana/web3.js';
+import { EventEmitter } from 'events';
+import StrictEventEmitter from 'strict-event-emitter-types/types/src';
+import { BN } from '..';
+
+// eslint-disable-next-line @typescript-eslint/ban-types
+type ClockSubscriberConfig = {
+	commitment: Commitment;
+	resubTimeoutMs?: number;
+};
+
+export interface ClockSubscriberEvent {
+	clockUpdate: (ts: number) => void;
+}
+
+export class ClockSubscriber {
+	private latestSlot: number;
+	currentTs: number;
+	private subscriptionId: number;
+	commitment: Commitment;
+	eventEmitter: StrictEventEmitter<EventEmitter, ClockSubscriberEvent>;
+
+	// Reconnection
+	private timeoutId?: NodeJS.Timeout;
+	private resubTimeoutMs?: number;
+	private isUnsubscribing = false;
+	private receivingData = false;
+
+	public constructor(
+		private connection: Connection,
+		config?: ClockSubscriberConfig
+	) {
+		this.eventEmitter = new EventEmitter();
+		this.resubTimeoutMs = config?.resubTimeoutMs;
+		this.commitment = config?.commitment || 'confirmed';
+		if (this.resubTimeoutMs < 1000) {
+			console.log(
+				'resubTimeoutMs should be at least 1000ms to avoid spamming resub'
+			);
+		}
+	}
+
+	public async subscribe(): Promise<void> {
+		if (this.subscriptionId != null) {
+			return;
+		}
+
+		this.subscriptionId = this.connection.onAccountChange(
+			SYSVAR_CLOCK_PUBKEY,
+			(acctInfo, context) => {
+				if (!this.latestSlot || this.latestSlot < context.slot) {
+					if (this.resubTimeoutMs && !this.isUnsubscribing) {
+						this.receivingData = true;
+						clearTimeout(this.timeoutId);
+						this.setTimeout();
+					}
+					this.latestSlot = context.slot;
+					this.currentTs = new BN(
+						acctInfo.data.subarray(32, 39),
+						undefined,
+						'le'
+					).toNumber();
+					this.eventEmitter.emit('clockUpdate', this.currentTs);
+				}
+			},
+			this.commitment
+		);
+
+		if (this.resubTimeoutMs) {
+			this.receivingData = true;
+			this.setTimeout();
+		}
+	}
+
+	private setTimeout(): void {
+		this.timeoutId = setTimeout(async () => {
+			if (this.isUnsubscribing) {
+				// If we are in the process of unsubscribing, do not attempt to resubscribe
+				return;
+			}
+
+			if (this.receivingData) {
+				console.log(
+					`No new slot in ${this.resubTimeoutMs}ms, slot subscriber resubscribing`
+				);
+				await this.unsubscribe(true);
+				this.receivingData = false;
+				await this.subscribe();
+			}
+		}, this.resubTimeoutMs);
+	}
+
+	public getUnixTs(): number {
+		return this.currentTs;
+	}
+
+	public async unsubscribe(onResub = false): Promise<void> {
+		if (!onResub) {
+			this.resubTimeoutMs = undefined;
+		}
+		this.isUnsubscribing = true;
+		clearTimeout(this.timeoutId);
+		this.timeoutId = undefined;
+
+		if (this.subscriptionId != null) {
+			await this.connection.removeAccountChangeListener(this.subscriptionId);
+			this.subscriptionId = undefined;
+			this.isUnsubscribing = false;
+		} else {
+			this.isUnsubscribing = false;
+		}
+	}
+}

package/src/index.ts

@@ -102,5 +102,6 @@ export * from './auctionSubscriber/types';
 export * from './memcmp';
 export * from './decode/user';
 export * from './blockhashSubscriber';
+export * from './clock/clockSubscriber';
 
 export { BN, PublicKey, pyth };

package/src/math/exchangeStatus.ts

@@ -11,6 +11,7 @@ import {
 	SpotOperation,
 	StateAccount,
 	isVariant,
+	InsuranceFundOperation,
 } from '../types';
 import { BN } from '@coral-xyz/anchor';
 
@@ -65,7 +66,7 @@ export function ammPaused(
 
 export function isOperationPaused(
 	pausedOperations: number,
-	operation: PerpOperation | SpotOperation
+	operation: PerpOperation | SpotOperation | InsuranceFundOperation
 ): boolean {
 	return (pausedOperations & operation) > 0;
 }

package/src/types.ts

@@ -43,6 +43,13 @@ export enum SpotOperation {
 	LIQUIDATION = 8,
 }
 
+export enum InsuranceFundOperation {
+	INIT = 1,
+	ADD = 2,
+	REQUEST_REMOVE = 4,
+	REMOVE = 8,
+}
+
 export enum UserStatus {
 	BEING_LIQUIDATED = 1,
 	BANKRUPT = 2,
@@ -708,6 +715,8 @@ export type SpotMarketAccount = {
 	ordersEnabled: boolean;
 
 	pausedOperations: number;
+
+	ifPausedOperations: number;
 };
 
 export type PoolBalance = {