@drift-labs/sdk 0.1.25-master.0 → 0.1.28

package/src/accounts/pollingOracleSubscriber.ts

@@ -0,0 +1,103 @@
+import { NotSubscribedError, OracleEvents, OracleSubscriber } from './types';
+import { Program } from '@project-serum/anchor';
+import StrictEventEmitter from 'strict-event-emitter-types';
+import { EventEmitter } from 'events';
+import { PublicKey } from '@solana/web3.js';
+import { BulkAccountLoader } from './bulkAccountLoader';
+import { OracleClient, OraclePriceData } from '../oracles/types';
+
+export class PollingOracleSubscriber implements OracleSubscriber {
+	isSubscribed: boolean;
+	program: Program;
+	eventEmitter: StrictEventEmitter<EventEmitter, OracleEvents>;
+	publicKey: PublicKey;
+
+	accountLoader: BulkAccountLoader;
+	oracleClient: OracleClient;
+	callbackId?: string;
+	errorCallbackId?: string;
+
+	oraclePriceData?: OraclePriceData;
+
+	public constructor(
+		publicKey: PublicKey,
+		oracleClient: OracleClient,
+		accountLoader: BulkAccountLoader
+	) {
+		this.isSubscribed = false;
+		this.publicKey = publicKey;
+		this.oracleClient = oracleClient;
+		this.accountLoader = accountLoader;
+		this.eventEmitter = new EventEmitter();
+	}
+
+	async subscribe(): Promise<boolean> {
+		if (this.isSubscribed) {
+			return true;
+		}
+
+		this.addToAccountLoader();
+		await this.fetch();
+		this.eventEmitter.emit('update');
+
+		this.isSubscribed = true;
+		return true;
+	}
+
+	addToAccountLoader(): void {
+		if (this.callbackId) {
+			return;
+		}
+
+		this.callbackId = this.accountLoader.addAccount(
+			this.publicKey,
+			async (buffer) => {
+				const oraclePriceData =
+					await this.oracleClient.getOraclePriceDataFromBuffer(buffer);
+				this.oraclePriceData = oraclePriceData;
+				// @ts-ignore
+				this.eventEmitter.emit('oracleUpdate', oraclePriceData);
+				this.eventEmitter.emit('update');
+			}
+		);
+
+		this.errorCallbackId = this.accountLoader.addErrorCallbacks((error) => {
+			this.eventEmitter.emit('error', error);
+		});
+	}
+
+	async fetch(): Promise<void> {
+		await this.accountLoader.load();
+		const buffer = this.accountLoader.getAccountData(this.publicKey);
+		this.oraclePriceData = await this.oracleClient.getOraclePriceDataFromBuffer(
+			buffer
+		);
+	}
+
+	async unsubscribe(): Promise<void> {
+		if (!this.isSubscribed) {
+			return;
+		}
+
+		this.accountLoader.removeAccount(this.publicKey, this.callbackId);
+		this.callbackId = undefined;
+
+		this.accountLoader.removeErrorCallbacks(this.errorCallbackId);
+		this.errorCallbackId = undefined;
+
+		this.isSubscribed = false;
+	}
+
+	assertIsSubscribed(): void {
+		if (!this.isSubscribed) {
+			throw new NotSubscribedError(
+				'You must call `subscribe` before using this function'
+			);
+		}
+	}
+
+	public getOraclePriceData(): OraclePriceData {
+		this.assertIsSubscribed();
+		return this.oraclePriceData;
+	}
+}

package/src/accounts/pollingTokenAccountSubscriber.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PollingTokenAccountSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+const token_1 = require("../token");
+class PollingTokenAccountSubscriber {
+    constructor(publicKey, accountLoader) {
+        this.isSubscribed = false;
+        this.publicKey = publicKey;
+        this.accountLoader = accountLoader;
+        this.eventEmitter = new events_1.EventEmitter();
+    }
+    async subscribe() {
+        if (this.isSubscribed) {
+            return true;
+        }
+        this.addToAccountLoader();
+        await this.fetch();
+        this.eventEmitter.emit('update');
+        this.isSubscribed = true;
+        return true;
+    }
+    addToAccountLoader() {
+        if (this.callbackId) {
+            return;
+        }
+        this.callbackId = this.accountLoader.addAccount(this.publicKey, (buffer) => {
+            const tokenAccount = (0, token_1.parseTokenAccount)(buffer);
+            this.tokenAccount = tokenAccount;
+            // @ts-ignore
+            this.eventEmitter.emit('tokenAccountUpdate', tokenAccount);
+            this.eventEmitter.emit('update');
+        });
+        this.errorCallbackId = this.accountLoader.addErrorCallbacks((error) => {
+            this.eventEmitter.emit('error', error);
+        });
+    }
+    async fetch() {
+        await this.accountLoader.load();
+        const buffer = this.accountLoader.getAccountData(this.publicKey);
+        this.tokenAccount = (0, token_1.parseTokenAccount)(buffer);
+    }
+    async unsubscribe() {
+        if (!this.isSubscribed) {
+            return;
+        }
+        this.accountLoader.removeAccount(this.publicKey, this.callbackId);
+        this.callbackId = undefined;
+        this.accountLoader.removeErrorCallbacks(this.errorCallbackId);
+        this.errorCallbackId = undefined;
+        this.isSubscribed = false;
+    }
+    assertIsSubscribed() {
+        if (!this.isSubscribed) {
+            throw new types_1.NotSubscribedError('You must call `subscribe` before using this function');
+        }
+    }
+    getTokenAccount() {
+        this.assertIsSubscribed();
+        return this.tokenAccount;
+    }
+}
+exports.PollingTokenAccountSubscriber = PollingTokenAccountSubscriber;
+//# sourceMappingURL=pollingTokenAccountSubscriber.js.map

package/src/accounts/pollingTokenAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pollingTokenAccountSubscriber.js","sourceRoot":"","sources":["pollingTokenAccountSubscriber.ts"],"names":[],"mappings":";;;AAAA,mCAIiB;AAGjB,mCAAsC;AAItC,oCAA6C;AAE7C,MAAa,6BAA6B;IAYzC,YAAmB,SAAoB,EAAE,aAAgC;QACxE,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,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,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,kBAAkB;QACjB,IAAI,IAAI,CAAC,UAAU,EAAE;YACpB,OAAO;SACP;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAC9C,IAAI,CAAC,SAAS,EACd,CAAC,MAAM,EAAE,EAAE;YACV,MAAM,YAAY,GAAG,IAAA,yBAAiB,EAAC,MAAM,CAAC,CAAC;YAC/C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;YACjC,aAAa;YACb,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;YAC3D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CACD,CAAC;QAEF,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,KAAK;QACV,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,GAAG,IAAA,yBAAiB,EAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACvB,OAAO;SACP;QAED,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAClE,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAE5B,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QAEjC,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,eAAe;QACrB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,YAAY,CAAC;IAC1B,CAAC;CACD;AArFD,sEAqFC"}

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"}