@drift-labs/sdk 0.1.26 → 0.1.29-master.0

package/src/accounts/pollingUserAccountSubscriber.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PollingUserAccountSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+const addresses_1 = require("../addresses");
+const utils_1 = require("./utils");
+class PollingUserAccountSubscriber {
+    constructor(program, authority, accountLoader) {
+        this.accountsToPoll = new Map();
+        this.type = 'polling';
+        this.isSubscribed = false;
+        this.program = program;
+        this.authority = authority;
+        this.accountLoader = accountLoader;
+        this.eventEmitter = new events_1.EventEmitter();
+    }
+    async subscribe() {
+        if (this.isSubscribed) {
+            return true;
+        }
+        await this.addToAccountLoader();
+        await this.fetchIfUnloaded();
+        this.eventEmitter.emit('update');
+        this.isSubscribed = true;
+        return true;
+    }
+    async addToAccountLoader(userPublicKeys) {
+        if (this.accountsToPoll.size > 0) {
+            return;
+        }
+        if (!userPublicKeys) {
+            const userPublicKey = await (0, addresses_1.getUserAccountPublicKey)(this.program.programId, this.authority);
+            const userAccount = (await this.program.account.user.fetch(userPublicKey));
+            this.accountsToPoll.set(userPublicKey.toString(), {
+                key: 'user',
+                publicKey: userPublicKey,
+                eventType: 'userAccountData',
+            });
+            this.accountsToPoll.set(userAccount.positions.toString(), {
+                key: 'userPositions',
+                publicKey: userAccount.positions,
+                eventType: 'userPositionsData',
+            });
+            const userOrdersPublicKey = await (0, addresses_1.getUserOrdersAccountPublicKey)(this.program.programId, userPublicKey);
+            this.accountsToPoll.set(userOrdersPublicKey.toString(), {
+                key: 'userOrders',
+                publicKey: userOrdersPublicKey,
+                eventType: 'userOrdersData',
+            });
+        }
+        else {
+            this.accountsToPoll.set(userPublicKeys.user.toString(), {
+                key: 'user',
+                publicKey: userPublicKeys.user,
+                eventType: 'userAccountData',
+            });
+            this.accountsToPoll.set(userPublicKeys.userPositions.toString(), {
+                key: 'userPositions',
+                publicKey: userPublicKeys.userPositions,
+                eventType: 'userPositionsData',
+            });
+            if (userPublicKeys.userOrders) {
+                this.accountsToPoll.set(userPublicKeys.userOrders.toString(), {
+                    key: 'userOrders',
+                    publicKey: userPublicKeys.userOrders,
+                    eventType: 'userOrdersData',
+                });
+            }
+        }
+        for (const [_, accountToPoll] of this.accountsToPoll) {
+            accountToPoll.callbackId = this.accountLoader.addAccount(accountToPoll.publicKey, (buffer) => {
+                if (!buffer) {
+                    return;
+                }
+                const account = this.program.account[accountToPoll.key].coder.accounts.decode((0, utils_1.capitalize)(accountToPoll.key), buffer);
+                this[accountToPoll.key] = account;
+                // @ts-ignore
+                this.eventEmitter.emit(accountToPoll.eventType, account);
+                this.eventEmitter.emit('update');
+            });
+        }
+        this.errorCallbackId = this.accountLoader.addErrorCallbacks((error) => {
+            this.eventEmitter.emit('error', error);
+        });
+    }
+    async fetchIfUnloaded() {
+        let shouldFetch = false;
+        for (const [_, accountToPoll] of this.accountsToPoll) {
+            if (this[accountToPoll.key] === undefined) {
+                shouldFetch = true;
+                break;
+            }
+        }
+        if (shouldFetch) {
+            await this.fetch();
+        }
+    }
+    async fetch() {
+        await this.accountLoader.load();
+        for (const [_, accountToPoll] of this.accountsToPoll) {
+            const buffer = this.accountLoader.getAccountData(accountToPoll.publicKey);
+            if (buffer) {
+                this[accountToPoll.key] = this.program.account[accountToPoll.key].coder.accounts.decode((0, utils_1.capitalize)(accountToPoll.key), buffer);
+            }
+        }
+    }
+    async unsubscribe() {
+        if (!this.isSubscribed) {
+            return;
+        }
+        for (const [_, accountToPoll] of this.accountsToPoll) {
+            this.accountLoader.removeAccount(accountToPoll.publicKey, accountToPoll.callbackId);
+        }
+        this.accountLoader.removeErrorCallbacks(this.errorCallbackId);
+        this.errorCallbackId = undefined;
+        this.accountsToPoll.clear();
+        this.isSubscribed = false;
+    }
+    assertIsSubscribed() {
+        if (!this.isSubscribed) {
+            throw new types_1.NotSubscribedError('You must call `subscribe` before using this function');
+        }
+    }
+    getUserAccount() {
+        this.assertIsSubscribed();
+        return this.user;
+    }
+    getUserPositionsAccount() {
+        this.assertIsSubscribed();
+        return this.userPositions;
+    }
+    getUserOrdersAccount() {
+        this.assertIsSubscribed();
+        return this.userOrders;
+    }
+}
+exports.PollingUserAccountSubscriber = PollingUserAccountSubscriber;
+//# sourceMappingURL=pollingUserAccountSubscriber.js.map

package/src/accounts/pollingUserAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pollingUserAccountSubscriber.js","sourceRoot":"","sources":["pollingUserAccountSubscriber.ts"],"names":[],"mappings":";;;AAAA,mCAMiB;AAGjB,mCAAsC;AAEtC,4CAGsB;AAGtB,mCAAqC;AAGrC,MAAa,4BAA4B;IAgBxC,YACC,OAAgB,EAChB,SAAoB,EACpB,aAAgC;QAZjC,mBAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;QAOlD,SAAI,GAA4B,SAAS,CAAC;QAOzC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,IAAI,qBAAY,EAAE,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,SAAS;QACd,IAAI,IAAI,CAAC,YAAY,EAAE;YACtB,OAAO,IAAI,CAAC;SACZ;QAED,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,cAA+B;QACvD,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE;YACjC,OAAO;SACP;QAED,IAAI,CAAC,cAAc,EAAE;YACpB,MAAM,aAAa,GAAG,MAAM,IAAA,mCAAuB,EAClD,IAAI,CAAC,OAAO,CAAC,SAAS,EACtB,IAAI,CAAC,SAAS,CACd,CAAC;YAEF,MAAM,WAAW,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CACzD,aAAa,CACb,CAAgB,CAAC;YAElB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE;gBACjD,GAAG,EAAE,MAAM;gBACX,SAAS,EAAE,aAAa;gBACxB,SAAS,EAAE,iBAAiB;aAC5B,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE;gBACzD,GAAG,EAAE,eAAe;gBACpB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAChC,SAAS,EAAE,mBAAmB;aAC9B,CAAC,CAAC;YAEH,MAAM,mBAAmB,GAAG,MAAM,IAAA,yCAA6B,EAC9D,IAAI,CAAC,OAAO,CAAC,SAAS,EACtB,aAAa,CACb,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,mBAAmB,CAAC,QAAQ,EAAE,EAAE;gBACvD,GAAG,EAAE,YAAY;gBACjB,SAAS,EAAE,mBAAmB;gBAC9B,SAAS,EAAE,gBAAgB;aAC3B,CAAC,CAAC;SACH;aAAM;YACN,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE;gBACvD,GAAG,EAAE,MAAM;gBACX,SAAS,EAAE,cAAc,CAAC,IAAI;gBAC9B,SAAS,EAAE,iBAAiB;aAC5B,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE;gBAChE,GAAG,EAAE,eAAe;gBACpB,SAAS,EAAE,cAAc,CAAC,aAAa;gBACvC,SAAS,EAAE,mBAAmB;aAC9B,CAAC,CAAC;YAEH,IAAI,cAAc,CAAC,UAAU,EAAE;gBAC9B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE;oBAC7D,GAAG,EAAE,YAAY;oBACjB,SAAS,EAAE,cAAc,CAAC,UAAU;oBACpC,SAAS,EAAE,gBAAgB;iBAC3B,CAAC,CAAC;aACH;SACD;QAED,KAAK,MAAM,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;YACrD,aAAa,CAAC,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CACvD,aAAa,CAAC,SAAS,EACvB,CAAC,MAAM,EAAE,EAAE;gBACV,IAAI,CAAC,MAAM,EAAE;oBACZ,OAAO;iBACP;gBAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CACnC,aAAa,CAAC,GAAG,CACjB,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAA,kBAAU,EAAC,aAAa,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;gBAC/D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;gBAClC,aAAa;gBACb,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBACzD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC,CACD,CAAC;SACF;QAED,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC,KAAK,EAAE,EAAE;YACrE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe;QACpB,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,KAAK,MAAM,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;YACrD,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE;gBAC1C,WAAW,GAAG,IAAI,CAAC;gBACnB,MAAM;aACN;SACD;QAED,IAAI,WAAW,EAAE;YAChB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;SACnB;IACF,CAAC;IAED,KAAK,CAAC,KAAK;QACV,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAChC,KAAK,MAAM,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;YACrD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAC1E,IAAI,MAAM,EAAE;gBACX,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAC7C,aAAa,CAAC,GAAG,CACjB,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAA,kBAAU,EAAC,aAAa,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;aAC/D;SACD;IACF,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,OAAO;SACP;QAED,KAAK,MAAM,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE;YACrD,IAAI,CAAC,aAAa,CAAC,aAAa,CAC/B,aAAa,CAAC,SAAS,EACvB,aAAa,CAAC,UAAU,CACxB,CAAC;SACF;QAED,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QAEjC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAE5B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,MAAM,IAAI,0BAAkB,CAC3B,sDAAsD,CACtD,CAAC;SACF;IACF,CAAC;IAEM,cAAc;QACpB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,IAAI,CAAC;IAClB,CAAC;IAEM,uBAAuB;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,aAAa,CAAC;IAC3B,CAAC;IAEM,oBAAoB;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC;IACxB,CAAC;CACD;AAhMD,oEAgMC"}

package/src/accounts/types.js

@@ -8,3 +8,4 @@ class NotSubscribedError extends Error {
     }
 }
 exports.NotSubscribedError = NotSubscribedError;
+//# sourceMappingURL=types.js.map

package/src/accounts/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":";;;AAgCA,MAAa,kBAAmB,SAAQ,KAAK;IAA7C;;QACC,SAAI,GAAG,oBAAoB,CAAC;IAC7B,CAAC;CAAA;AAFD,gDAEC"}

package/src/accounts/types.ts

@@ -17,7 +17,11 @@ import StrictEventEmitter from 'strict-event-emitter-types';
 import { EventEmitter } from 'events';
 import { PublicKey } from '@solana/web3.js';
 import { AccountInfo } from '@solana/spl-token';
-import { ClearingHouseConfigType, ClearingHouseUserConfigType } from '..';
+import {
+	ClearingHouseConfigType,
+	ClearingHouseUserConfigType,
+	OraclePriceData,
+} from '..';
 
 export interface AccountSubscriber<T> {
 	data?: T;
@@ -127,6 +131,23 @@ export interface TokenAccountSubscriber {
 	getTokenAccount(): AccountInfo;
 }
 
+export interface OracleEvents {
+	oracleUpdate: (payload: OraclePriceData) => void;
+	update: void;
+	error: (e: Error) => void;
+}
+
+export interface OracleSubscriber {
+	eventEmitter: StrictEventEmitter<EventEmitter, OracleEvents>;
+	isSubscribed: boolean;
+
+	subscribe(): Promise<boolean>;
+	fetch(): Promise<void>;
+	unsubscribe(): Promise<void>;
+
+	getOraclePriceData(): OraclePriceData;
+}
+
 export type AccountToPoll = {
 	key: string;
 	publicKey: PublicKey;

package/src/accounts/utils.js

@@ -5,3 +5,4 @@ function capitalize(value) {
     return value[0].toUpperCase() + value.slice(1);
 }
 exports.capitalize = capitalize;
+//# sourceMappingURL=utils.js.map

package/src/accounts/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["utils.ts"],"names":[],"mappings":";;;AAAA,SAAgB,UAAU,CAAC,KAAa;IACvC,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAChD,CAAC;AAFD,gCAEC"}

package/src/accounts/webSocketAccountSubscriber.js

@@ -1,13 +1,4 @@
 "use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WebSocketAccountSubscriber = void 0;
 const utils_1 = require("./utils");
@@ -17,23 +8,19 @@ class WebSocketAccountSubscriber {
         this.program = program;
         this.accountPublicKey = accountPublicKey;
     }
-    subscribe(onChange) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (this.listenerId) {
-                return;
-            }
-            this.onChange = onChange;
-            yield this.fetch();
-            this.listenerId = this.program.provider.connection.onAccountChange(this.accountPublicKey, (accountInfo, context) => {
-                this.handleRpcResponse(context, accountInfo);
-            }, this.program.provider.opts.commitment);
-        });
+    async subscribe(onChange) {
+        if (this.listenerId) {
+            return;
+        }
+        this.onChange = onChange;
+        await this.fetch();
+        this.listenerId = this.program.provider.connection.onAccountChange(this.accountPublicKey, (accountInfo, context) => {
+            this.handleRpcResponse(context, accountInfo);
+        }, this.program.provider.opts.commitment);
     }
-    fetch() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const rpcResponse = yield this.program.provider.connection.getAccountInfoAndContext(this.accountPublicKey, this.program.provider.opts.commitment);
-            this.handleRpcResponse(rpcResponse.context, rpcResponse === null || rpcResponse === void 0 ? void 0 : rpcResponse.value);
-        });
+    async fetch() {
+        const rpcResponse = await this.program.provider.connection.getAccountInfoAndContext(this.accountPublicKey, this.program.provider.opts.commitment);
+        this.handleRpcResponse(rpcResponse.context, rpcResponse === null || rpcResponse === void 0 ? void 0 : rpcResponse.value);
     }
     handleRpcResponse(context, accountInfo) {
         const newSlot = context.slot;
@@ -47,7 +34,7 @@ class WebSocketAccountSubscriber {
                 slot: newSlot,
             };
             if (newBuffer) {
-                this.data = this.program.account[this.accountName].coder.accounts.decode(utils_1.capitalize(this.accountName), newBuffer);
+                this.data = this.program.account[this.accountName].coder.accounts.decode((0, utils_1.capitalize)(this.accountName), newBuffer);
                 this.onChange(this.data);
             }
             return;
@@ -61,7 +48,7 @@ class WebSocketAccountSubscriber {
                 buffer: newBuffer,
                 slot: newSlot,
             };
-            this.data = this.program.account[this.accountName].coder.accounts.decode(utils_1.capitalize(this.accountName), newBuffer);
+            this.data = this.program.account[this.accountName].coder.accounts.decode((0, utils_1.capitalize)(this.accountName), newBuffer);
             this.onChange(this.data);
         }
     }
@@ -74,3 +61,4 @@ class WebSocketAccountSubscriber {
     }
 }
 exports.WebSocketAccountSubscriber = WebSocketAccountSubscriber;
+//# sourceMappingURL=webSocketAccountSubscriber.js.map

package/src/accounts/webSocketAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webSocketAccountSubscriber.js","sourceRoot":"","sources":["webSocketAccountSubscriber.ts"],"names":[],"mappings":";;;AAGA,mCAAqC;AAGrC,MAAa,0BAA0B;IAStC,YACC,WAAmB,EACnB,OAAgB,EAChB,gBAA2B;QAE3B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAA2B;QAC1C,IAAI,IAAI,CAAC,UAAU,EAAE;YACpB,OAAO;SACP;QAED,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAEnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,eAAe,CACjE,IAAI,CAAC,gBAAgB,EACrB,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE;YACxB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC9C,CAAC,EACD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CACrC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACV,MAAM,WAAW,GAChB,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,wBAAwB,CAC9D,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CACrC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,KAAK,CAAC,CAAC;IACjE,CAAC;IAED,iBAAiB,CAAC,OAAgB,EAAE,WAAiC;QACpE,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;QAC7B,IAAI,SAAS,GAAuB,SAAS,CAAC;QAC9C,IAAI,WAAW,EAAE;YAChB,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC;SAC7B;QAED,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;YACtB,IAAI,CAAC,WAAW,GAAG;gBAClB,MAAM,EAAE,SAAS;gBACjB,IAAI,EAAE,OAAO;aACb,CAAC;YACF,IAAI,SAAS,EAAE;gBACd,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAC/B,IAAI,CAAC,WAAW,CAChB,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAA,kBAAU,EAAC,IAAI,CAAC,WAAW,CAAC,EAAE,SAAS,CAAC,CAAC;gBACjE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aACzB;YACD,OAAO;SACP;QAED,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE;YACrC,OAAO;SACP;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QAC1C,IAAI,SAAS,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE;YAC9D,IAAI,CAAC,WAAW,GAAG;gBAClB,MAAM,EAAE,SAAS;gBACjB,IAAI,EAAE,OAAO;aACb,CAAC;YACF,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CACvE,IAAA,kBAAU,EAAC,IAAI,CAAC,WAAW,CAAC,EAC5B,SAAS,CACT,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SACzB;IACF,CAAC;IAED,WAAW;QACV,IAAI,IAAI,CAAC,UAAU,EAAE;YACpB,MAAM,OAAO,GACZ,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,2BAA2B,CAC3D,IAAI,CAAC,UAAU,CACf,CAAC;YACH,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAC5B,OAAO,OAAO,CAAC;SACf;IACF,CAAC;CACD;AA9FD,gEA8FC"}

package/src/accounts/webSocketClearingHouseAccountSubscriber.js

@@ -0,0 +1,212 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketClearingHouseAccountSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+const addresses_1 = require("../addresses");
+const webSocketAccountSubscriber_1 = require("./webSocketAccountSubscriber");
+class WebSocketClearingHouseAccountSubscriber {
+    constructor(program) {
+        this.optionalExtraSubscriptions = [];
+        this.type = 'websocket';
+        this.isSubscribing = false;
+        this.isSubscribed = false;
+        this.program = program;
+        this.eventEmitter = new events_1.EventEmitter();
+    }
+    async subscribe(optionalSubscriptions) {
+        if (this.isSubscribed) {
+            return true;
+        }
+        if (this.isSubscribing) {
+            return await this.subscriptionPromise;
+        }
+        this.isSubscribing = true;
+        this.subscriptionPromise = new Promise((res) => {
+            this.subscriptionPromiseResolver = res;
+        });
+        const statePublicKey = await (0, addresses_1.getClearingHouseStateAccountPublicKey)(this.program.programId);
+        // create and activate main state account subscription
+        this.stateAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('state', this.program, statePublicKey);
+        await this.stateAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('stateAccountUpdate', data);
+            this.eventEmitter.emit('update');
+        });
+        const state = this.stateAccountSubscriber.data;
+        this.marketsAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('markets', this.program, state.markets);
+        await this.marketsAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('marketsAccountUpdate', data);
+            this.eventEmitter.emit('update');
+        });
+        this.orderStateAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('orderState', this.program, state.orderState);
+        await this.orderStateAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('orderStateAccountUpdate', data);
+            this.eventEmitter.emit('update');
+        });
+        const orderState = this.orderStateAccountSubscriber.data;
+        // create subscribers for other state accounts
+        this.tradeHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('tradeHistory', this.program, state.tradeHistory);
+        this.depositHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('depositHistory', this.program, state.depositHistory);
+        this.fundingPaymentHistoryAccountSubscriber =
+            new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('fundingPaymentHistory', this.program, state.fundingPaymentHistory);
+        this.fundingRateHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('fundingRateHistory', this.program, state.fundingRateHistory);
+        this.liquidationHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('liquidationHistory', this.program, state.liquidationHistory);
+        this.curveHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('extendedCurveHistory', this.program, state.extendedCurveHistory);
+        this.orderHistoryAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('orderHistory', this.program, orderState.orderHistory);
+        const extraSusbcribersToUse = [];
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('tradeHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.tradeHistoryAccountSubscriber,
+                eventType: 'tradeHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('depositHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.depositHistoryAccountSubscriber,
+                eventType: 'depositHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('fundingPaymentHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.fundingPaymentHistoryAccountSubscriber,
+                eventType: 'fundingPaymentHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('fundingRateHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.fundingRateHistoryAccountSubscriber,
+                eventType: 'fundingRateHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('liquidationHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.liquidationHistoryAccountSubscriber,
+                eventType: 'liquidationHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('curveHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.curveHistoryAccountSubscriber,
+                eventType: 'curveHistoryAccountUpdate',
+            });
+        if (optionalSubscriptions === null || optionalSubscriptions === void 0 ? void 0 : optionalSubscriptions.includes('orderHistoryAccount'))
+            extraSusbcribersToUse.push({
+                subscriber: this.orderHistoryAccountSubscriber,
+                eventType: 'orderHistoryAccountUpdate',
+            });
+        this.optionalExtraSubscriptions = optionalSubscriptions !== null && optionalSubscriptions !== void 0 ? optionalSubscriptions : [];
+        // await all subcriptions in parallel to boost performance
+        //// the state account subscription above can't happen in here, because some of these susbcriptions are dependent on clearing house state being available
+        await Promise.all(extraSusbcribersToUse.map(({ subscriber, eventType }) => subscriber.subscribe((data) => {
+            this.eventEmitter.emit(eventType, data);
+            this.eventEmitter.emit('update');
+        })));
+        this.eventEmitter.emit('update');
+        this.isSubscribing = false;
+        this.isSubscribed = true;
+        this.subscriptionPromiseResolver(true);
+        return true;
+    }
+    async fetch() {
+        if (!this.isSubscribed) {
+            return;
+        }
+        const promises = this.optionalExtraSubscriptions
+            .map((optionalSubscription) => {
+            const subscriber = `${optionalSubscription}Subscriber`;
+            return this[subscriber].fetch();
+        })
+            .concat([
+            this.stateAccountSubscriber.fetch(),
+            this.marketsAccountSubscriber.fetch(),
+        ]);
+        await Promise.all(promises);
+    }
+    async unsubscribe() {
+        if (!this.isSubscribed) {
+            return;
+        }
+        await this.stateAccountSubscriber.unsubscribe();
+        await this.marketsAccountSubscriber.unsubscribe();
+        await this.orderStateAccountSubscriber.unsubscribe();
+        if (this.optionalExtraSubscriptions.includes('tradeHistoryAccount')) {
+            await this.tradeHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('fundingRateHistoryAccount')) {
+            await this.fundingRateHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('fundingPaymentHistoryAccount')) {
+            await this.fundingPaymentHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('depositHistoryAccount')) {
+            await this.depositHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('curveHistoryAccount')) {
+            await this.curveHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('liquidationHistoryAccount')) {
+            await this.liquidationHistoryAccountSubscriber.unsubscribe();
+        }
+        if (this.optionalExtraSubscriptions.includes('orderHistoryAccount')) {
+            await this.orderHistoryAccountSubscriber.unsubscribe();
+        }
+        this.isSubscribed = false;
+    }
+    assertIsSubscribed() {
+        if (!this.isSubscribed) {
+            throw new types_1.NotSubscribedError('You must call `subscribe` before using this function');
+        }
+    }
+    assertOptionalIsSubscribed(optionalSubscription) {
+        if (!this.isSubscribed) {
+            throw new types_1.NotSubscribedError('You must call `subscribe` before using this function');
+        }
+        if (!this.optionalExtraSubscriptions.includes(optionalSubscription)) {
+            throw new types_1.NotSubscribedError(`You need to subscribe to the optional Clearing House account "${optionalSubscription}" to use this method`);
+        }
+    }
+    getStateAccount() {
+        this.assertIsSubscribed();
+        return this.stateAccountSubscriber.data;
+    }
+    getMarketsAccount() {
+        this.assertIsSubscribed();
+        return this.marketsAccountSubscriber.data;
+    }
+    getTradeHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('tradeHistoryAccount');
+        return this.tradeHistoryAccountSubscriber.data;
+    }
+    getDepositHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('depositHistoryAccount');
+        return this.depositHistoryAccountSubscriber.data;
+    }
+    getFundingPaymentHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('fundingPaymentHistoryAccount');
+        return this.fundingPaymentHistoryAccountSubscriber.data;
+    }
+    getFundingRateHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('fundingRateHistoryAccount');
+        return this.fundingRateHistoryAccountSubscriber.data;
+    }
+    getCurveHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('curveHistoryAccount');
+        return this.curveHistoryAccountSubscriber.data;
+    }
+    getLiquidationHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('liquidationHistoryAccount');
+        return this.liquidationHistoryAccountSubscriber.data;
+    }
+    getOrderHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('orderHistoryAccount');
+        return this.orderHistoryAccountSubscriber.data;
+    }
+    getOrderStateAccount() {
+        this.assertIsSubscribed();
+        return this.orderStateAccountSubscriber.data;
+    }
+}
+exports.WebSocketClearingHouseAccountSubscriber = WebSocketClearingHouseAccountSubscriber;
+//# sourceMappingURL=webSocketClearingHouseAccountSubscriber.js.map

package/src/accounts/webSocketClearingHouseAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webSocketClearingHouseAccountSubscriber.js","sourceRoot":"","sources":["webSocketClearingHouseAccountSubscriber.ts"],"names":[],"mappings":";;;AAKA,mCAAgE;AAehE,mCAAsC;AACtC,4CAAqE;AACrE,6EAA0E;AAG1E,MAAa,uCAAuC;IAyBnD,YAAmB,OAAgB;QARnC,+BAA0B,GAAgC,EAAE,CAAC;QAE7D,SAAI,GAA4B,WAAW,CAAC;QAEpC,kBAAa,GAAG,KAAK,CAAC;QAK7B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,IAAI,qBAAY,EAAE,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,SAAS,CACrB,qBAAmD;QAEnD,IAAI,IAAI,CAAC,YAAY,EAAE;YACtB,OAAO,IAAI,CAAC;SACZ;QAED,IAAI,IAAI,CAAC,aAAa,EAAE;YACvB,OAAO,MAAM,IAAI,CAAC,mBAAmB,CAAC;SACtC;QAED,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAE1B,IAAI,CAAC,mBAAmB,GAAG,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAC9C,IAAI,CAAC,2BAA2B,GAAG,GAAG,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,MAAM,IAAA,iDAAqC,EACjE,IAAI,CAAC,OAAO,CAAC,SAAS,CACtB,CAAC;QAEF,sDAAsD;QACtD,IAAI,CAAC,sBAAsB,GAAG,IAAI,uDAA0B,CAC3D,OAAO,EACP,IAAI,CAAC,OAAO,EACZ,cAAc,CACd,CAAC;QACF,MAAM,IAAI,CAAC,sBAAsB,CAAC,SAAS,CAAC,CAAC,IAAkB,EAAE,EAAE;YAClE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;YACnD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;QAE/C,IAAI,CAAC,wBAAwB,GAAG,IAAI,uDAA0B,CAC7D,SAAS,EACT,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,OAAO,CACb,CAAC;QAEF,MAAM,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,CAAC,IAAoB,EAAE,EAAE;YACtE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,2BAA2B,GAAG,IAAI,uDAA0B,CAChE,YAAY,EACZ,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,UAAU,CAChB,CAAC;QAEF,MAAM,IAAI,CAAC,2BAA2B,CAAC,SAAS,CAC/C,CAAC,IAAuB,EAAE,EAAE;YAC3B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;YACxD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CACD,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC;QAEzD,8CAA8C;QAE9C,IAAI,CAAC,6BAA6B,GAAG,IAAI,uDAA0B,CAClE,cAAc,EACd,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,YAAY,CAClB,CAAC;QAEF,IAAI,CAAC,+BAA+B,GAAG,IAAI,uDAA0B,CACpE,gBAAgB,EAChB,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,cAAc,CACpB,CAAC;QAEF,IAAI,CAAC,sCAAsC;YAC1C,IAAI,uDAA0B,CAC7B,uBAAuB,EACvB,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,qBAAqB,CAC3B,CAAC;QAEH,IAAI,CAAC,mCAAmC,GAAG,IAAI,uDAA0B,CACxE,oBAAoB,EACpB,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,kBAAkB,CACxB,CAAC;QAEF,IAAI,CAAC,mCAAmC,GAAG,IAAI,uDAA0B,CACxE,oBAAoB,EACpB,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,kBAAkB,CACxB,CAAC;QAEF,IAAI,CAAC,6BAA6B,GAAG,IAAI,uDAA0B,CAClE,sBAAsB,EACtB,IAAI,CAAC,OAAO,EACZ,KAAK,CAAC,oBAAoB,CAC1B,CAAC;QAEF,IAAI,CAAC,6BAA6B,GAAG,IAAI,uDAA0B,CAClE,cAAc,EACd,IAAI,CAAC,OAAO,EACZ,UAAU,CAAC,YAAY,CACvB,CAAC;QAEF,MAAM,qBAAqB,GAGrB,EAAE,CAAC;QAET,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,qBAAqB,CAAC;YACzD,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,6BAA6B;gBAC9C,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,uBAAuB,CAAC;YAC3D,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,+BAA+B;gBAChD,SAAS,EAAE,6BAA6B;aACxC,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,8BAA8B,CAAC;YAClE,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,sCAAsC;gBACvD,SAAS,EAAE,oCAAoC;aAC/C,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,2BAA2B,CAAC;YAC/D,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,mCAAmC;gBACpD,SAAS,EAAE,iCAAiC;aAC5C,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,2BAA2B,CAAC;YAC/D,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,mCAAmC;gBACpD,SAAS,EAAE,iCAAiC;aAC5C,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,qBAAqB,CAAC;YACzD,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,6BAA6B;gBAC9C,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;QAEJ,IAAI,qBAAqB,aAArB,qBAAqB,uBAArB,qBAAqB,CAAE,QAAQ,CAAC,qBAAqB,CAAC;YACzD,qBAAqB,CAAC,IAAI,CAAC;gBAC1B,UAAU,EAAE,IAAI,CAAC,6BAA6B;gBAC9C,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;QAEJ,IAAI,CAAC,0BAA0B,GAAG,qBAAqB,aAArB,qBAAqB,cAArB,qBAAqB,GAAI,EAAE,CAAC;QAE9D,0DAA0D;QAC1D,yJAAyJ;QACzJ,MAAM,OAAO,CAAC,GAAG,CAChB,qBAAqB,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,CACvD,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACxC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CACF,CACD,CAAC;QAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEjC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC;QAEvC,OAAO,IAAI,CAAC;IACb,CAAC;IAEM,KAAK,CAAC,KAAK;QACjB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,OAAO;SACP;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,0BAA0B;aAC9C,GAAG,CAAC,CAAC,oBAAoB,EAAE,EAAE;YAC7B,MAAM,UAAU,GAAG,GAAG,oBAAoB,YAAY,CAAC;YACvD,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC,CAAC;aACD,MAAM,CAAC;YACP,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE;YACnC,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE;SACrC,CAAC,CAAC;QAEJ,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,KAAK,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,OAAO;SACP;QAED,MAAM,IAAI,CAAC,sBAAsB,CAAC,WAAW,EAAE,CAAC;QAChD,MAAM,IAAI,CAAC,wBAAwB,CAAC,WAAW,EAAE,CAAC;QAClD,MAAM,IAAI,CAAC,2BAA2B,CAAC,WAAW,EAAE,CAAC;QAErD,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACpE,MAAM,IAAI,CAAC,6BAA6B,CAAC,WAAW,EAAE,CAAC;SACvD;QAED,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE;YAC1E,MAAM,IAAI,CAAC,mCAAmC,CAAC,WAAW,EAAE,CAAC;SAC7D;QAED,IACC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,8BAA8B,CAAC,EACvE;YACD,MAAM,IAAI,CAAC,sCAAsC,CAAC,WAAW,EAAE,CAAC;SAChE;QAED,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE;YACtE,MAAM,IAAI,CAAC,+BAA+B,CAAC,WAAW,EAAE,CAAC;SACzD;QAED,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACpE,MAAM,IAAI,CAAC,6BAA6B,CAAC,WAAW,EAAE,CAAC;SACvD;QAED,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,2BAA2B,CAAC,EAAE;YAC1E,MAAM,IAAI,CAAC,mCAAmC,CAAC,WAAW,EAAE,CAAC;SAC7D;QAED,IAAI,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACpE,MAAM,IAAI,CAAC,6BAA6B,CAAC,WAAW,EAAE,CAAC;SACvD;QAED,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,MAAM,IAAI,0BAAkB,CAC3B,sDAAsD,CACtD,CAAC;SACF;IACF,CAAC;IAED,0BAA0B,CACzB,oBAA+C;QAE/C,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,MAAM,IAAI,0BAAkB,CAC3B,sDAAsD,CACtD,CAAC;SACF;QAED,IAAI,CAAC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE;YACpE,MAAM,IAAI,0BAAkB,CAC3B,iEAAiE,oBAAoB,sBAAsB,CAC3G,CAAC;SACF;IACF,CAAC;IAEM,eAAe;QACrB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;IACzC,CAAC;IAEM,iBAAiB;QACvB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC;IAC3C,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC;IAChD,CAAC;IAEM,wBAAwB;QAC9B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,uBAAuB,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC;IAClD,CAAC;IAEM,+BAA+B;QACrC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,8BAA8B,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC,sCAAsC,CAAC,IAAI,CAAC;IACzD,CAAC;IAEM,4BAA4B;QAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,CAAC;IACtD,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC;IAChD,CAAC;IAEM,4BAA4B;QAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC,mCAAmC,CAAC,IAAI,CAAC;IACtD,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC;IAChD,CAAC;IAEM,oBAAoB;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC;IAC9C,CAAC;CACD;AAzVD,0FAyVC"}

package/src/accounts/webSocketUserAccountSubscriber.js

@@ -0,0 +1,78 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketUserAccountSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+const addresses_1 = require("../addresses");
+const webSocketAccountSubscriber_1 = require("./webSocketAccountSubscriber");
+class WebSocketUserAccountSubscriber {
+    constructor(program, authority) {
+        this.type = 'websocket';
+        this.isSubscribed = false;
+        this.program = program;
+        this.authority = authority;
+        this.eventEmitter = new events_1.EventEmitter();
+    }
+    async subscribe() {
+        if (this.isSubscribed) {
+            return true;
+        }
+        const userPublicKey = await (0, addresses_1.getUserAccountPublicKey)(this.program.programId, this.authority);
+        this.userDataAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('user', this.program, userPublicKey);
+        await this.userDataAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('userAccountData', data);
+            this.eventEmitter.emit('update');
+        });
+        const userAccountData = this.userDataAccountSubscriber.data;
+        this.userPositionsAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('userPositions', this.program, userAccountData.positions);
+        await this.userPositionsAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('userPositionsData', data);
+            this.eventEmitter.emit('update');
+        });
+        const userOrdersPublicKey = await (0, addresses_1.getUserOrdersAccountPublicKey)(this.program.programId, userPublicKey);
+        this.userOrdersAccountSubscriber = new webSocketAccountSubscriber_1.WebSocketAccountSubscriber('userOrders', this.program, userOrdersPublicKey);
+        await this.userOrdersAccountSubscriber.subscribe((data) => {
+            this.eventEmitter.emit('userOrdersData', data);
+            this.eventEmitter.emit('update');
+        });
+        this.eventEmitter.emit('update');
+        this.isSubscribed = true;
+        return true;
+    }
+    async fetch() {
+        await Promise.all([
+            this.userDataAccountSubscriber.fetch(),
+            this.userPositionsAccountSubscriber.fetch(),
+            this.userOrdersAccountSubscriber.fetch(),
+        ]);
+    }
+    async unsubscribe() {
+        if (!this.isSubscribed) {
+            return;
+        }
+        await Promise.all([
+            this.userDataAccountSubscriber.unsubscribe(),
+            this.userPositionsAccountSubscriber.unsubscribe(),
+            await this.userOrdersAccountSubscriber.unsubscribe(),
+        ]);
+        this.isSubscribed = false;
+    }
+    assertIsSubscribed() {
+        if (!this.isSubscribed) {
+            throw new types_1.NotSubscribedError('You must call `subscribe` before using this function');
+        }
+    }
+    getUserAccount() {
+        this.assertIsSubscribed();
+        return this.userDataAccountSubscriber.data;
+    }
+    getUserPositionsAccount() {
+        this.assertIsSubscribed();
+        return this.userPositionsAccountSubscriber.data;
+    }
+    getUserOrdersAccount() {
+        return this.userOrdersAccountSubscriber.data;
+    }
+}
+exports.WebSocketUserAccountSubscriber = WebSocketUserAccountSubscriber;
+//# sourceMappingURL=webSocketUserAccountSubscriber.js.map

package/src/accounts/webSocketUserAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"webSocketUserAccountSubscriber.js","sourceRoot":"","sources":["webSocketUserAccountSubscriber.ts"],"names":[],"mappings":";;;AAAA,mCAKiB;AAGjB,mCAAsC;AAEtC,4CAGsB;AACtB,6EAA0E;AAI1E,MAAa,8BAA8B;IAY1C,YAAmB,OAAgB,EAAE,SAAoB;QAFzD,SAAI,GAA4B,WAAW,CAAC;QAG3C,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,IAAI,qBAAY,EAAE,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,SAAS;QACd,IAAI,IAAI,CAAC,YAAY,EAAE;YACtB,OAAO,IAAI,CAAC;SACZ;QAED,MAAM,aAAa,GAAG,MAAM,IAAA,mCAAuB,EAClD,IAAI,CAAC,OAAO,CAAC,SAAS,EACtB,IAAI,CAAC,SAAS,CACd,CAAC;QACF,IAAI,CAAC,yBAAyB,GAAG,IAAI,uDAA0B,CAC9D,MAAM,EACN,IAAI,CAAC,OAAO,EACZ,aAAa,CACb,CAAC;QACF,MAAM,IAAI,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,IAAiB,EAAE,EAAE;YACpE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC;QAC5D,IAAI,CAAC,8BAA8B,GAAG,IAAI,uDAA0B,CACnE,eAAe,EACf,IAAI,CAAC,OAAO,EACZ,eAAe,CAAC,SAAS,CACzB,CAAC;QAEF,MAAM,IAAI,CAAC,8BAA8B,CAAC,SAAS,CAClD,CAAC,IAA0B,EAAE,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;YAClD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,MAAM,IAAA,yCAA6B,EAC9D,IAAI,CAAC,OAAO,CAAC,SAAS,EACtB,aAAa,CACb,CAAC;QAEF,IAAI,CAAC,2BAA2B,GAAG,IAAI,uDAA0B,CAChE,YAAY,EACZ,IAAI,CAAC,OAAO,EACZ,mBAAmB,CACnB,CAAC;QACF,MAAM,IAAI,CAAC,2BAA2B,CAAC,SAAS,CAC/C,CAAC,IAAuB,EAAE,EAAE;YAC3B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;YAC/C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CACD,CAAC;QAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,KAAK,CAAC,KAAK;QACV,MAAM,OAAO,CAAC,GAAG,CAAC;YACjB,IAAI,CAAC,yBAAyB,CAAC,KAAK,EAAE;YACtC,IAAI,CAAC,8BAA8B,CAAC,KAAK,EAAE;YAC3C,IAAI,CAAC,2BAA2B,CAAC,KAAK,EAAE;SACxC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,OAAO;SACP;QAED,MAAM,OAAO,CAAC,GAAG,CAAC;YACjB,IAAI,CAAC,yBAAyB,CAAC,WAAW,EAAE;YAC5C,IAAI,CAAC,8BAA8B,CAAC,WAAW,EAAE;YACjD,MAAM,IAAI,CAAC,2BAA2B,CAAC,WAAW,EAAE;SACpD,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC3B,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,MAAM,IAAI,0BAAkB,CAC3B,sDAAsD,CACtD,CAAC;SACF;IACF,CAAC;IAEM,cAAc;QACpB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC;IAC5C,CAAC;IAEM,uBAAuB;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC;IACjD,CAAC;IAEM,oBAAoB;QAC1B,OAAO,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC;IAC9C,CAAC;CACD;AArHD,wEAqHC"}