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

package/lib/orderParams.js

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getMarketOrderParams = exports.getTriggerLimitOrderParams = exports.getTriggerMarketOrderParams = exports.getLimitOrderParams = void 0;
 const types_1 = require("./types");
 const numericConstants_1 = require("./constants/numericConstants");
-function getLimitOrderParams(marketIndex, direction, baseAssetAmount, price, reduceOnly, discountToken = false, referrer = false, userOrderId = 0, postOnly = false) {
+function getLimitOrderParams(marketIndex, direction, baseAssetAmount, price, reduceOnly, discountToken = false, referrer = false, userOrderId = 0, postOnly = false, oraclePriceOffset = numericConstants_1.ZERO) {
     return {
         orderType: types_1.OrderType.LIMIT,
         userOrderId,
@@ -24,7 +24,7 @@ function getLimitOrderParams(marketIndex, direction, baseAssetAmount, price, red
         },
         triggerCondition: types_1.OrderTriggerCondition.ABOVE,
         triggerPrice: numericConstants_1.ZERO,
-        oraclePriceOffset: numericConstants_1.ZERO,
+        oraclePriceOffset,
     };
 }
 exports.getLimitOrderParams = getLimitOrderParams;

package/lib/orders.d.ts

@@ -1,8 +1,9 @@
 /// <reference types="bn.js" />
 import { Market, Order, UserAccount, UserPosition } from './types';
 import { BN, ClearingHouseUser } from '.';
+import { OraclePriceData } from '.';
 export declare function calculateNewStateAfterOrder(userAccount: UserAccount, userPosition: UserPosition, market: Market, order: Order): [UserAccount, UserPosition, Market] | null;
-export declare function calculateBaseAssetAmountMarketCanExecute(market: Market, order: Order): BN;
-export declare function calculateAmountToTradeForLimit(market: Market, order: Order): BN;
+export declare function calculateBaseAssetAmountMarketCanExecute(market: Market, order: Order, oraclePriceData?: OraclePriceData): BN;
+export declare function calculateAmountToTradeForLimit(market: Market, order: Order, oraclePriceData?: OraclePriceData): BN;
 export declare function calculateAmountToTradeForTriggerLimit(market: Market, order: Order): BN;
 export declare function calculateBaseAssetAmountUserCanExecute(market: Market, order: Order, user: ClearingHouseUser): BN;

package/lib/orders.js

@@ -80,9 +80,9 @@ function calculateAmountSwapped(marketBefore, marketAfter) {
         baseAssetAmountSwapped: marketBefore.amm.baseAssetReserve.sub(marketAfter.amm.baseAssetReserve),
     };
 }
-function calculateBaseAssetAmountMarketCanExecute(market, order) {
+function calculateBaseAssetAmountMarketCanExecute(market, order, oraclePriceData) {
     if ((0, types_1.isVariant)(order.orderType, 'limit')) {
-        return calculateAmountToTradeForLimit(market, order);
+        return calculateAmountToTradeForLimit(market, order, oraclePriceData);
     }
     else if ((0, types_1.isVariant)(order.orderType, 'triggerLimit')) {
         return calculateAmountToTradeForTriggerLimit(market, order);
@@ -96,8 +96,15 @@ function calculateBaseAssetAmountMarketCanExecute(market, order) {
     }
 }
 exports.calculateBaseAssetAmountMarketCanExecute = calculateBaseAssetAmountMarketCanExecute;
-function calculateAmountToTradeForLimit(market, order) {
-    const [maxAmountToTrade, direction] = (0, amm_1.calculateMaxBaseAssetAmountToTrade)(market.amm, order.price);
+function calculateAmountToTradeForLimit(market, order, oraclePriceData) {
+    let limitPrice = order.price;
+    if (!order.oraclePriceOffset.eq(numericConstants_1.ZERO)) {
+        if (!oraclePriceData) {
+            throw Error('Cant calculate limit price for oracle offset oracle without OraclePriceData');
+        }
+        limitPrice = oraclePriceData.price.add(order.oraclePriceOffset);
+    }
+    const [maxAmountToTrade, direction] = (0, amm_1.calculateMaxBaseAssetAmountToTrade)(market.amm, limitPrice);
     // Check that directions are the same
     const sameDirection = isSameDirection(direction, order.direction);
     if (!sameDirection) {

package/lib/types.d.ts

@@ -71,6 +71,9 @@ export declare class OrderAction {
     static readonly CANCEL: {
         cancel: {};
     };
+    static readonly EXPIRE: {
+        expire: {};
+    };
     static readonly FILL: {
         fill: {};
     };

package/lib/types.js

@@ -42,6 +42,7 @@ class OrderAction {
 exports.OrderAction = OrderAction;
 OrderAction.PLACE = { place: {} };
 OrderAction.CANCEL = { cancel: {} };
+OrderAction.EXPIRE = { expire: {} };
 OrderAction.FILL = { fill: {} };
 class OrderTriggerCondition {
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "0.1.26",
+	"version": "0.1.29-master.0",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/accounts/bulkAccountLoader.ts

@@ -9,6 +9,7 @@ type AccountToLoad = {
 
 const GET_MULTIPLE_ACCOUNTS_CHUNK_SIZE = 99;
 
+const oneMinute = 60 * 1000;
 const fiveMinutes = 5 * 60 * 1000;
 
 export class BulkAccountLoader {
@@ -104,10 +105,6 @@ export class BulkAccountLoader {
 			this.loadPromiseResolver = resolver;
 		});
 
-		if (this.loggingEnabled) {
-			console.log('Loading accounts');
-		}
-
 		try {
 			const chunks = this.chunks(
 				Array.from(this.accountsToLoad.values()),
@@ -163,6 +160,11 @@ export class BulkAccountLoader {
 			args
 		);
 
+		const oneMinuteSinceLastUpdate = Date.now() - this.lastUpdate > oneMinute;
+		if (this.loggingEnabled && oneMinuteSinceLastUpdate) {
+			console.log('rpcResponse', JSON.stringify(rpcResponse));
+		}
+
 		const newSlot = rpcResponse.result.context.slot;
 
 		for (const i in accountsToLoad) {
@@ -177,6 +179,10 @@ export class BulkAccountLoader {
 				newBuffer = Buffer.from(raw, dataType);
 			}
 
+			if (this.loggingEnabled && oneMinuteSinceLastUpdate) {
+				console.log('oldRPCResponse', oldRPCResponse);
+			}
+
 			if (!oldRPCResponse) {
 				this.accountData.set(key, {
 					slot: newSlot,

package/src/accounts/bulkUserSubscription.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.bulkPollingUserSubscribe = void 0;
+/**
+ * @param users
+ * @param accountLoader
+ */
+async function bulkPollingUserSubscribe(users, accountLoader) {
+    if (users.length === 0) {
+        await accountLoader.load();
+        return;
+    }
+    // Fetch all the accounts first
+    const program = users[0].clearingHouse.program;
+    let userProgramAccounts;
+    let orderProgramAccounts;
+    await Promise.all([
+        (async () => {
+            userProgramAccounts = await program.account.user.all();
+        })(),
+        (async () => {
+            orderProgramAccounts = await program.account.userOrders.all();
+        })(),
+    ]);
+    // Create a map of the authority to keys
+    const authorityToKeys = new Map();
+    const userToAuthority = new Map();
+    for (const userProgramAccount of userProgramAccounts) {
+        const userAccountPublicKey = userProgramAccount.publicKey;
+        const userAccount = userProgramAccount.account;
+        authorityToKeys.set(userAccount.authority.toString(), {
+            user: userAccountPublicKey,
+            userPositions: userAccount.positions,
+            userOrders: undefined,
+        });
+        userToAuthority.set(userAccountPublicKey.toString(), userAccount.authority.toString());
+    }
+    for (const orderProgramAccount of orderProgramAccounts) {
+        const userOrderAccountPublicKey = orderProgramAccount.publicKey;
+        const userOrderAccount = orderProgramAccount.account;
+        const authority = userToAuthority.get(userOrderAccount.user.toString());
+        const userPublicKeys = authorityToKeys.get(authority);
+        userPublicKeys.userOrders = userOrderAccountPublicKey;
+    }
+    await Promise.all(users.map((user) => {
+        // Pull the keys from the authority map so we can skip fetching them in addToAccountLoader
+        const userPublicKeys = authorityToKeys.get(user.authority.toString());
+        return user.accountSubscriber.addToAccountLoader(userPublicKeys);
+    }));
+    await accountLoader.load();
+    await Promise.all(users.map(async (user) => {
+        return user.subscribe();
+    }));
+}
+exports.bulkPollingUserSubscribe = bulkPollingUserSubscribe;
+//# sourceMappingURL=bulkUserSubscription.js.map

package/src/accounts/bulkUserSubscription.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bulkUserSubscription.js","sourceRoot":"","sources":["bulkUserSubscription.ts"],"names":[],"mappings":";;;AAOA;;;GAGG;AACI,KAAK,UAAU,wBAAwB,CAC7C,KAA0B,EAC1B,aAAgC;IAEhC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;QACvB,MAAM,aAAa,CAAC,IAAI,EAAE,CAAC;QAC3B,OAAO;KACP;IAED,+BAA+B;IAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC;IAC/C,IAAI,mBAAqC,CAAC;IAC1C,IAAI,oBAAsC,CAAC;IAC3C,MAAM,OAAO,CAAC,GAAG,CAAC;QACjB,CAAC,KAAK,IAAI,EAAE;YACX,mBAAmB,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QACxD,CAAC,CAAC,EAAE;QACJ,CAAC,KAAK,IAAI,EAAE;YACX,oBAAoB,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;QAC/D,CAAC,CAAC,EAAE;KACJ,CAAC,CAAC;IAEH,wCAAwC;IACxC,MAAM,eAAe,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC1D,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;IAClD,KAAK,MAAM,kBAAkB,IAAI,mBAAmB,EAAE;QACrD,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,SAAS,CAAC;QAC1D,MAAM,WAAW,GAAG,kBAAkB,CAAC,OAAsB,CAAC;QAE9D,eAAe,CAAC,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE;YACrD,IAAI,EAAE,oBAAoB;YAC1B,aAAa,EAAE,WAAW,CAAC,SAAS;YACpC,UAAU,EAAE,SAAS;SACrB,CAAC,CAAC;QAEH,eAAe,CAAC,GAAG,CAClB,oBAAoB,CAAC,QAAQ,EAAE,EAC/B,WAAW,CAAC,SAAS,CAAC,QAAQ,EAAE,CAChC,CAAC;KACF;IACD,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,EAAE;QACvD,MAAM,yBAAyB,GAAG,mBAAmB,CAAC,SAAS,CAAC;QAChE,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,OAA4B,CAAC;QAE1E,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxE,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,cAAc,CAAC,UAAU,GAAG,yBAAyB,CAAC;KACtD;IAED,MAAM,OAAO,CAAC,GAAG,CAChB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAClB,0FAA0F;QAC1F,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtE,OACC,IAAI,CAAC,iBACL,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC,CAAC,CACF,CAAC;IAEF,MAAM,aAAa,CAAC,IAAI,EAAE,CAAC;IAE3B,MAAM,OAAO,CAAC,GAAG,CAChB,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACxB,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC,CAAC,CACF,CAAC;AACH,CAAC;AAlED,4DAkEC"}

package/src/accounts/pollingClearingHouseAccountSubscriber.js

@@ -0,0 +1,210 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PollingClearingHouseAccountSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+const addresses_1 = require("../addresses");
+const utils_1 = require("./utils");
+class PollingClearingHouseAccountSubscriber {
+    constructor(program, accountLoader) {
+        this.accountsToPoll = new Map();
+        this.optionalExtraSubscriptions = [];
+        this.type = 'polling';
+        this.isSubscribing = false;
+        this.isSubscribed = false;
+        this.program = program;
+        this.eventEmitter = new events_1.EventEmitter();
+        this.accountLoader = accountLoader;
+    }
+    async subscribe(optionalSubscriptions) {
+        if (this.isSubscribed) {
+            return true;
+        }
+        if (this.isSubscribing) {
+            return await this.subscriptionPromise;
+        }
+        this.optionalExtraSubscriptions = optionalSubscriptions;
+        this.isSubscribing = true;
+        this.subscriptionPromise = new Promise((res) => {
+            this.subscriptionPromiseResolver = res;
+        });
+        await this.updateAccountsToPoll();
+        await this.addToAccountLoader();
+        await this.fetch();
+        this.eventEmitter.emit('update');
+        this.isSubscribing = false;
+        this.isSubscribed = true;
+        this.subscriptionPromiseResolver(true);
+        return true;
+    }
+    async updateAccountsToPoll() {
+        var _a, _b, _c, _d, _e, _f, _g;
+        if (this.accountsToPoll.size > 0) {
+            return;
+        }
+        const statePublicKey = await (0, addresses_1.getClearingHouseStateAccountPublicKey)(this.program.programId);
+        const state = (await this.program.account.state.fetch(statePublicKey));
+        this.accountsToPoll.set(statePublicKey.toString(), {
+            key: 'state',
+            publicKey: statePublicKey,
+            eventType: 'stateAccountUpdate',
+        });
+        this.accountsToPoll.set(state.markets.toString(), {
+            key: 'markets',
+            publicKey: state.markets,
+            eventType: 'marketsAccountUpdate',
+        });
+        this.accountsToPoll.set(state.orderState.toString(), {
+            key: 'orderState',
+            publicKey: state.orderState,
+            eventType: 'orderStateAccountUpdate',
+        });
+        if ((_a = this.optionalExtraSubscriptions) === null || _a === void 0 ? void 0 : _a.includes('tradeHistoryAccount')) {
+            this.accountsToPoll.set(state.tradeHistory.toString(), {
+                key: 'tradeHistory',
+                publicKey: state.tradeHistory,
+                eventType: 'tradeHistoryAccountUpdate',
+            });
+        }
+        if ((_b = this.optionalExtraSubscriptions) === null || _b === void 0 ? void 0 : _b.includes('depositHistoryAccount')) {
+            this.accountsToPoll.set(state.depositHistory.toString(), {
+                key: 'depositHistory',
+                publicKey: state.depositHistory,
+                eventType: 'depositHistoryAccountUpdate',
+            });
+        }
+        if ((_c = this.optionalExtraSubscriptions) === null || _c === void 0 ? void 0 : _c.includes('fundingPaymentHistoryAccount')) {
+            this.accountsToPoll.set(state.fundingPaymentHistory.toString(), {
+                key: 'fundingPaymentHistory',
+                publicKey: state.fundingPaymentHistory,
+                eventType: 'fundingPaymentHistoryAccountUpdate',
+            });
+        }
+        if ((_d = this.optionalExtraSubscriptions) === null || _d === void 0 ? void 0 : _d.includes('fundingRateHistoryAccount')) {
+            this.accountsToPoll.set(state.fundingRateHistory.toString(), {
+                key: 'fundingRateHistory',
+                publicKey: state.fundingRateHistory,
+                eventType: 'fundingRateHistoryAccountUpdate',
+            });
+        }
+        if ((_e = this.optionalExtraSubscriptions) === null || _e === void 0 ? void 0 : _e.includes('curveHistoryAccount')) {
+            this.accountsToPoll.set(state.extendedCurveHistory.toString(), {
+                key: 'extendedCurveHistory',
+                publicKey: state.extendedCurveHistory,
+                eventType: 'curveHistoryAccountUpdate',
+            });
+        }
+        if ((_f = this.optionalExtraSubscriptions) === null || _f === void 0 ? void 0 : _f.includes('liquidationHistoryAccount')) {
+            this.accountsToPoll.set(state.liquidationHistory.toString(), {
+                key: 'liquidationHistory',
+                publicKey: state.liquidationHistory,
+                eventType: 'liquidationHistoryAccountUpdate',
+            });
+        }
+        if ((_g = this.optionalExtraSubscriptions) === null || _g === void 0 ? void 0 : _g.includes('orderHistoryAccount')) {
+            const orderState = (await this.program.account.orderState.fetch(state.orderState));
+            this.accountsToPoll.set(orderState.orderHistory.toString(), {
+                key: 'orderHistory',
+                publicKey: orderState.orderHistory,
+                eventType: 'orderHistoryAccountUpdate',
+            });
+        }
+    }
+    async addToAccountLoader() {
+        for (const [_, accountToPoll] of this.accountsToPoll) {
+            accountToPoll.callbackId = this.accountLoader.addAccount(accountToPoll.publicKey, (buffer) => {
+                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 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');
+        }
+    }
+    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.state;
+    }
+    getMarketsAccount() {
+        this.assertIsSubscribed();
+        return this.markets;
+    }
+    getOrderStateAccount() {
+        this.assertIsSubscribed();
+        return this.orderState;
+    }
+    getTradeHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('tradeHistoryAccount');
+        return this.tradeHistory;
+    }
+    getDepositHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('depositHistoryAccount');
+        return this.depositHistory;
+    }
+    getFundingPaymentHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('fundingPaymentHistoryAccount');
+        return this.fundingPaymentHistory;
+    }
+    getFundingRateHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('fundingRateHistoryAccount');
+        return this.fundingRateHistory;
+    }
+    getCurveHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('curveHistoryAccount');
+        return this.extendedCurveHistory;
+    }
+    getLiquidationHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('liquidationHistoryAccount');
+        return this.liquidationHistory;
+    }
+    getOrderHistoryAccount() {
+        this.assertIsSubscribed();
+        this.assertOptionalIsSubscribed('orderHistoryAccount');
+        return this.orderHistory;
+    }
+}
+exports.PollingClearingHouseAccountSubscriber = PollingClearingHouseAccountSubscriber;
+//# sourceMappingURL=pollingClearingHouseAccountSubscriber.js.map

package/src/accounts/pollingClearingHouseAccountSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pollingClearingHouseAccountSubscriber.js","sourceRoot":"","sources":["pollingClearingHouseAccountSubscriber.ts"],"names":[],"mappings":";;;AAAA,mCAMiB;AAGjB,mCAAsC;AAatC,4CAAqE;AAErE,mCAAqC;AAGrC,MAAa,qCAAqC;IA8BjD,YAAmB,OAAgB,EAAE,aAAgC;QAtBrE,mBAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;QAclD,+BAA0B,GAAgC,EAAE,CAAC;QAE7D,SAAI,GAA4B,SAAS,CAAC;QAElC,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;QACvC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACpC,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,0BAA0B,GAAG,qBAAqB,CAAC;QAExD,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,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAClC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,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;IAED,KAAK,CAAC,oBAAoB;;QACzB,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,GAAG,CAAC,EAAE;YACjC,OAAO;SACP;QAED,MAAM,cAAc,GAAG,MAAM,IAAA,iDAAqC,EACjE,IAAI,CAAC,OAAO,CAAC,SAAS,CACtB,CAAC;QAEF,MAAM,KAAK,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CACpD,cAAc,CACd,CAAiB,CAAC;QAEnB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,QAAQ,EAAE,EAAE;YAClD,GAAG,EAAE,OAAO;YACZ,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,oBAAoB;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE;YACjD,GAAG,EAAE,SAAS;YACd,SAAS,EAAE,KAAK,CAAC,OAAO;YACxB,SAAS,EAAE,sBAAsB;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE;YACpD,GAAG,EAAE,YAAY;YACjB,SAAS,EAAE,KAAK,CAAC,UAAU;YAC3B,SAAS,EAAE,yBAAyB;SACpC,CAAC,CAAC;QAEH,IAAI,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACrE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,EAAE,EAAE;gBACtD,GAAG,EAAE,cAAc;gBACnB,SAAS,EAAE,KAAK,CAAC,YAAY;gBAC7B,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;SACH;QAED,IAAI,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,uBAAuB,CAAC,EAAE;YACvE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,QAAQ,EAAE,EAAE;gBACxD,GAAG,EAAE,gBAAgB;gBACrB,SAAS,EAAE,KAAK,CAAC,cAAc;gBAC/B,SAAS,EAAE,6BAA6B;aACxC,CAAC,CAAC;SACH;QAED,IACC,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,8BAA8B,CAAC,EACxE;YACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,QAAQ,EAAE,EAAE;gBAC/D,GAAG,EAAE,uBAAuB;gBAC5B,SAAS,EAAE,KAAK,CAAC,qBAAqB;gBACtC,SAAS,EAAE,oCAAoC;aAC/C,CAAC,CAAC;SACH;QAED,IACC,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,2BAA2B,CAAC,EACrE;YACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,EAAE;gBAC5D,GAAG,EAAE,oBAAoB;gBACzB,SAAS,EAAE,KAAK,CAAC,kBAAkB;gBACnC,SAAS,EAAE,iCAAiC;aAC5C,CAAC,CAAC;SACH;QAED,IAAI,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACrE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE;gBAC9D,GAAG,EAAE,sBAAsB;gBAC3B,SAAS,EAAE,KAAK,CAAC,oBAAoB;gBACrC,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;SACH;QAED,IACC,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,2BAA2B,CAAC,EACrE;YACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,EAAE;gBAC5D,GAAG,EAAE,oBAAoB;gBACzB,SAAS,EAAE,KAAK,CAAC,kBAAkB;gBACnC,SAAS,EAAE,iCAAiC;aAC5C,CAAC,CAAC;SACH;QAED,IAAI,MAAA,IAAI,CAAC,0BAA0B,0CAAE,QAAQ,CAAC,qBAAqB,CAAC,EAAE;YACrE,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAC9D,KAAK,CAAC,UAAU,CAChB,CAAsB,CAAC;YAExB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,EAAE,EAAE;gBAC3D,GAAG,EAAE,cAAc;gBACnB,SAAS,EAAE,UAAU,CAAC,YAAY;gBAClC,SAAS,EAAE,2BAA2B;aACtC,CAAC,CAAC;SACH;IACF,CAAC;IAED,KAAK,CAAC,kBAAkB;QACvB,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,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;IAEM,KAAK,CAAC,KAAK;QACjB,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;IAEM,KAAK,CAAC,WAAW;QACvB,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;QAC5B,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,KAAK,CAAC;IACnB,CAAC;IAEM,iBAAiB;QACvB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC;IACrB,CAAC;IAEM,oBAAoB;QAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC;IACxB,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC;IAC1B,CAAC;IAEM,wBAAwB;QAC9B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,uBAAuB,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,cAAc,CAAC;IAC5B,CAAC;IAEM,+BAA+B;QACrC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,8BAA8B,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACnC,CAAC;IAEM,4BAA4B;QAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC,kBAAkB,CAAC;IAChC,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,oBAAoB,CAAC;IAClC,CAAC;IAEM,4BAA4B;QAClC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,2BAA2B,CAAC,CAAC;QAC7D,OAAO,IAAI,CAAC,kBAAkB,CAAC;IAChC,CAAC;IAEM,sBAAsB;QAC5B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,0BAA0B,CAAC,qBAAqB,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC;IAC1B,CAAC;CACD;AA1SD,sFA0SC"}

package/src/accounts/pollingOracleSubscriber.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PollingOracleSubscriber = void 0;
+const types_1 = require("./types");
+const events_1 = require("events");
+class PollingOracleSubscriber {
+    constructor(publicKey, oracleClient, accountLoader) {
+        this.isSubscribed = false;
+        this.publicKey = publicKey;
+        this.oracleClient = oracleClient;
+        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, 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() {
+        await this.accountLoader.load();
+        const buffer = this.accountLoader.getAccountData(this.publicKey);
+        this.oraclePriceData = await this.oracleClient.getOraclePriceDataFromBuffer(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');
+        }
+    }
+    getOraclePriceData() {
+        this.assertIsSubscribed();
+        return this.oraclePriceData;
+    }
+}
+exports.PollingOracleSubscriber = PollingOracleSubscriber;
+//# sourceMappingURL=pollingOracleSubscriber.js.map

package/src/accounts/pollingOracleSubscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pollingOracleSubscriber.js","sourceRoot":"","sources":["pollingOracleSubscriber.ts"],"names":[],"mappings":";;;AAAA,mCAA6E;AAG7E,mCAAsC;AAKtC,MAAa,uBAAuB;IAanC,YACC,SAAoB,EACpB,YAA0B,EAC1B,aAAgC;QAEhC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,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,KAAK,EAAE,MAAM,EAAE,EAAE;YAChB,MAAM,eAAe,GACpB,MAAM,IAAI,CAAC,YAAY,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC9D,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;YACvC,aAAa;YACb,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC;YACxD,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,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,4BAA4B,CAC1E,MAAM,CACN,CAAC;IACH,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,kBAAkB;QACxB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC,eAAe,CAAC;IAC7B,CAAC;CACD;AA9FD,0DA8FC"}

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