@drift-labs/common 1.0.0

package/lib/drift/Drift/clients/AuthorityDrift/DriftOperations/types.d.ts

@@ -0,0 +1,90 @@
+import { BigNum, PositionDirection, PostOnlyParams, QuoteResponse } from '@drift-labs/sdk';
+import { AuctionParamsRequestOptions } from '../../../../base/actions/trade/openPerpOrder/openPerpMarketOrder';
+/**
+ * Interface for deposit operation parameters.
+ */
+export interface DepositParams {
+    subAccountId: number;
+    amount: BigNum;
+    spotMarketIndex: number;
+    isMaxBorrowRepayment?: boolean;
+}
+/**
+ * Interface for withdraw operation parameters.
+ */
+export interface WithdrawParams {
+    subAccountId: number;
+    amount: BigNum;
+    spotMarketIndex: number;
+    isBorrow?: boolean;
+    isMax?: boolean;
+}
+/**
+ * Interface for create user and deposit operation parameters.
+ */
+export interface CreateUserAndDepositParams {
+    depositAmount: BigNum;
+    depositSpotMarketIndex: number;
+    newAccountName?: string;
+    maxLeverage?: number;
+    poolId?: number;
+    subAccountId?: number;
+    referrerName?: string;
+}
+/**
+ * Interface for perp market order parameters.
+ */
+export type PerpOrderParams = {
+    subAccountId: number;
+    marketIndex: number;
+    direction: PositionDirection;
+    assetType: 'base' | 'quote';
+    size: BigNum;
+    reduceOnly?: boolean;
+    postOnly?: PostOnlyParams;
+    orderConfig: {
+        orderType: 'market';
+        disableSwift?: boolean;
+        auctionParamsOptions?: AuctionParamsRequestOptions;
+        postOnly?: never;
+        bracketOrders?: {
+            takeProfitPrice?: BigNum;
+            stopLossPrice?: BigNum;
+        };
+    } | {
+        orderType: 'limit';
+        limitPrice: BigNum;
+        disableSwift?: boolean;
+        bracketOrders?: {
+            takeProfitPrice?: BigNum;
+            stopLossPrice?: BigNum;
+        };
+    } | {
+        orderType: 'takeProfit' | 'stopLoss';
+        triggerPrice: BigNum;
+        limitPrice?: BigNum;
+    } | {
+        orderType: 'oracleLimit';
+        oraclePriceOffset: BigNum;
+    };
+};
+/**
+ * Interface for swap operation parameters.
+ */
+export interface SwapParams {
+    fromMarketIndex: number;
+    toMarketIndex: number;
+    amount: BigNum;
+    subAccountId: number;
+    jupiterQuote?: QuoteResponse;
+}
+/**
+ * Interface for settle account P&Ls.
+ */
+export interface SettleAccountPnlParams {
+    subAccountId: number;
+}
+export interface CancelOrdersParams {
+    subAccountId: number;
+    orderIds: number[];
+}

package/lib/drift/Drift/clients/AuthorityDrift/DriftOperations/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/lib/drift/Drift/clients/AuthorityDrift/DriftOperations/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../../src/drift/Drift/clients/AuthorityDrift/DriftOperations/types.ts"],"names":[],"mappings":"","sourcesContent":["import {\n\tBigNum,\n\tPositionDirection,\n\tPostOnlyParams,\n\tQuoteResponse,\n} from '@drift-labs/sdk';\nimport { AuctionParamsRequestOptions } from '../../../../base/actions/trade/openPerpOrder/openPerpMarketOrder';\n\n/**\n * Interface for deposit operation parameters.\n */\nexport interface DepositParams {\n\tsubAccountId: number;\n\tamount: BigNum;\n\tspotMarketIndex: number;\n\tisMaxBorrowRepayment?: boolean;\n}\n\n/**\n * Interface for withdraw operation parameters.\n */\nexport interface WithdrawParams {\n\tsubAccountId: number;\n\tamount: BigNum;\n\tspotMarketIndex: number;\n\tisBorrow?: boolean;\n\tisMax?: boolean;\n}\n\n/**\n * Interface for create user and deposit operation parameters.\n */\nexport interface CreateUserAndDepositParams {\n\tdepositAmount: BigNum;\n\tdepositSpotMarketIndex: number;\n\tnewAccountName?: string;\n\tmaxLeverage?: number;\n\tpoolId?: number;\n\tsubAccountId?: number;\n\treferrerName?: string;\n}\n\n/**\n * Interface for perp market order parameters.\n */\nexport type PerpOrderParams = {\n\tsubAccountId: number;\n\tmarketIndex: number;\n\tdirection: PositionDirection;\n\tassetType: 'base' | 'quote';\n\tsize: BigNum;\n\treduceOnly?: boolean;\n\tpostOnly?: PostOnlyParams;\n\torderConfig:\n\t\t| {\n\t\t\t\torderType: 'market';\n\t\t\t\tdisableSwift?: boolean;\n\t\t\t\tauctionParamsOptions?: AuctionParamsRequestOptions;\n\t\t\t\tpostOnly?: never;\n\t\t\t\tbracketOrders?: {\n\t\t\t\t\ttakeProfitPrice?: BigNum;\n\t\t\t\t\tstopLossPrice?: BigNum;\n\t\t\t\t};\n\t\t\t\t// TODO: isMaxLeverage?\n\t\t  }\n\t\t| {\n\t\t\t\torderType: 'limit';\n\t\t\t\tlimitPrice: BigNum;\n\t\t\t\tdisableSwift?: boolean;\n\t\t\t\tbracketOrders?: {\n\t\t\t\t\ttakeProfitPrice?: BigNum;\n\t\t\t\t\tstopLossPrice?: BigNum;\n\t\t\t\t};\n\t\t\t\t// TODO: isMaxLeverage?\n\t\t  }\n\t\t| {\n\t\t\t\torderType: 'takeProfit' | 'stopLoss';\n\t\t\t\ttriggerPrice: BigNum;\n\t\t\t\tlimitPrice?: BigNum;\n\t\t  }\n\t\t| {\n\t\t\t\torderType: 'oracleLimit';\n\t\t\t\toraclePriceOffset: BigNum;\n\t\t  };\n};\n\n/**\n * Interface for swap operation parameters.\n */\nexport interface SwapParams {\n\tfromMarketIndex: number;\n\ttoMarketIndex: number;\n\tamount: BigNum;\n\tsubAccountId: number;\n\tjupiterQuote?: QuoteResponse;\n}\n\n/**\n * Interface for settle account P&Ls.\n */\nexport interface SettleAccountPnlParams {\n\tsubAccountId: number;\n}\n\nexport interface CancelOrdersParams {\n\tsubAccountId: number;\n\torderIds: number[];\n}\n"]}

package/lib/drift/Drift/clients/AuthorityDrift/SubscriptionManager.d.ts

@@ -0,0 +1,139 @@
+import { CustomizedCadenceBulkAccountLoader, DriftClient, IWallet, User } from '@drift-labs/sdk';
+import { MarketId } from '../../../../types';
+import { PollingDlob } from '../../data/PollingDlob';
+import { UserAccountCache } from '../../stores/UserAccountCache';
+import { DriftL2OrderbookManager } from './DriftL2OrderbookManager';
+/**
+ * Comprehensive subscription and market operations manager for the Drift protocol.
+ *
+ * This class handles all aspects of subscription management including:
+ * - User account subscriptions and event listeners
+ * - Market account and oracle polling optimization
+ * 	- Markets are categorized based on user involvement, and polling cadences are updated accordingly
+ * - DLOB server polling configuration (for lesser used markets)
+ * - DLOB server websocket subscription configuration (for selected trade market). Used to get orderbook data too.
+ * - Authority (wallet) management with full resubscription
+ *
+ * By combining subscription and market operations, this provides a unified
+ * interface for all subscription-related optimizations.
+ */
+export declare class SubscriptionManager {
+    private driftClient;
+    private accountLoader;
+    private pollingDlob;
+    private orderbookManager;
+    private userAccountCache;
+    private tradableMarkets;
+    private selectedTradeMarket;
+    /**
+     * Creates a new SubscriptionManager instance.
+     *
+     * @param driftClient - The DriftClient instance for managing subscriptions
+     * @param accountLoader - Handles bulk account loading and polling frequency management
+     * @param pollingDlob - Manages DLOB server polling for market data
+     * @param userAccountCache - Cache for user account data updates
+     * @param tradableMarkets - Array of markets that are available for trading
+     * @param selectedTradeMarket - The market that is currently being traded
+     */
+    constructor(driftClient: DriftClient, accountLoader: CustomizedCadenceBulkAccountLoader, pollingDlob: PollingDlob, orderbookManager: DriftL2OrderbookManager, userAccountCache: UserAccountCache, tradableMarkets: MarketId[], selectedTradeMarket: MarketId | null);
+    /**
+     * Updates the tradable markets list.
+     *
+     * @param tradableMarkets - The new tradable markets list
+     */
+    updateTradableMarkets(tradableMarkets: MarketId[]): void;
+    /**
+     * Subscribes to updates for all user accounts under the current authority.
+     *
+     * This method sets up event listeners for user account updates and ensures
+     * that market subscriptions are properly configured based on user positions.
+     * When a user's account data changes, it automatically triggers subscription
+     * updates to maintain optimal polling frequencies.
+     */
+    subscribeToAllUsersUpdates(): void;
+    /**
+     * Categorizes markets based on user involvement (active positions).
+     *
+     * Markets are classified as "user-involved" if any of the provided users
+     * have active positions (perp or spot) in those markets. This categorization
+     * is used to optimize polling frequencies - user-involved markets are polled
+     * more frequently to ensure accurate close to real-time data for user positions.
+     */
+    categorizeMarketsByUserInvolvement(users: User[]): {
+        userInvolvedMarkets: MarketId[];
+        userNotInvolvedMarkets: MarketId[];
+    };
+    /**
+     * Updates the polling cadence for a market account and its associated oracle account.
+     *
+     * This method optimizes data fetching by adjusting how frequently market and oracle
+     * accounts are polled from the RPC. Higher cadences (lower numbers) mean more frequent polling.
+     */
+    updateMarketAccountCadence(market: MarketId, newCadence: number): void;
+    /**
+     * Updates account loader polling cadences for multiple markets based on their categorization.
+     *
+     * This method applies different polling frequencies based on market importance:
+     * - Selected trade market: Highest frequency (most important for active trading)
+     * - User-involved markets: Medium frequency (user has positions)
+     * - User-not-involved markets: Lowest frequency (background monitoring)
+     */
+    private updateAccountLoaderCadenceForMarkets;
+    /**
+     * Updates DLOB (Decentralized Limit Order Book) polling intervals for market data.
+     *
+     * This method configures how frequently the DLOB server is polled for market data
+     * including mark prices, oracle prices, and orderbook data. Different markets are
+     * assigned to different polling intervals based on their importance.
+     */
+    private updatePollingDlobIntervals;
+    /**
+     * Handles comprehensive subscription updates when user account data changes.
+     *
+     * This is the main orchestration method that ensures optimal polling frequencies
+     * are maintained when user positions change. It automatically:
+     *
+     * 1. Categorizes markets based on current user involvement
+     * 2. Updates account loader cadences for market and oracle accounts
+     * 3. Updates DLOB polling intervals for price data
+     *
+     * The method ensures that markets with user positions get higher priority polling
+     * to maintain accurate real-time data for position management.
+     */
+    handleSubscriptionUpdatesOnUserUpdates(users: User[]): void;
+    /**
+     * Subscribes to markets that users are involved in but aren't in the default tradable markets list.
+     *
+     * This method ensures that if users have positions in markets that weren't initially
+     * included in the tradable markets list, those markets are still properly subscribed to.
+     * This is important for maintaining accurate account data even for edge cases.
+     */
+    subscribeToNonWhitelistedButUserInvolvedMarkets(users: User[]): Promise<void>;
+    private checkIsUserInvolvedInMarket;
+    /**
+     * Updates the selected trade market and optimizes all subscription polling accordingly.
+     *
+     * This method handles:
+     * 1. Removing the previous market from high-frequency polling
+     * 2. Adding the new market to high-frequency polling
+     * 3. Updating market account cadences for optimal performance
+     * 4. Categorizing the previous market based on user involvement
+     * 5. Assigning appropriate polling intervals to the previous market
+     *
+     * @param newSelectedTradeMarket - The new market to prioritize for trading
+     */
+    updateSelectedTradeMarket(newSelectedTradeMarket: MarketId | null): void;
+    /**
+     * Updates the authority (wallet) for the drift client and reestablishes subscriptions.
+     *
+     * This method handles the complete process of switching to a new wallet:
+     * 1. Checks if the wallet is actually different
+     * 2. Unsubscribes from current user accounts
+     * 3. Resets the user account cache
+     * 4. Updates the DriftClient with the new wallet
+     * 5. Resubscribes to markets based on new user positions
+     * 6. Switches to the specified subaccount if provided
+     * 7. Reestablishes user account subscriptions
+     */
+    updateAuthority(wallet: IWallet, activeSubAccountId?: number): Promise<void>;
+}

package/lib/drift/Drift/clients/AuthorityDrift/SubscriptionManager.js

@@ -0,0 +1,287 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SubscriptionManager = void 0;
+const types_1 = require("../../../../types");
+const constants_1 = require("../../constants");
+const orderbook_1 = require("../../constants/orderbook");
+/**
+ * Comprehensive subscription and market operations manager for the Drift protocol.
+ *
+ * This class handles all aspects of subscription management including:
+ * - User account subscriptions and event listeners
+ * - Market account and oracle polling optimization
+ * 	- Markets are categorized based on user involvement, and polling cadences are updated accordingly
+ * - DLOB server polling configuration (for lesser used markets)
+ * - DLOB server websocket subscription configuration (for selected trade market). Used to get orderbook data too.
+ * - Authority (wallet) management with full resubscription
+ *
+ * By combining subscription and market operations, this provides a unified
+ * interface for all subscription-related optimizations.
+ */
+class SubscriptionManager {
+    /**
+     * Creates a new SubscriptionManager instance.
+     *
+     * @param driftClient - The DriftClient instance for managing subscriptions
+     * @param accountLoader - Handles bulk account loading and polling frequency management
+     * @param pollingDlob - Manages DLOB server polling for market data
+     * @param userAccountCache - Cache for user account data updates
+     * @param tradableMarkets - Array of markets that are available for trading
+     * @param selectedTradeMarket - The market that is currently being traded
+     */
+    constructor(driftClient, accountLoader, pollingDlob, orderbookManager, userAccountCache, tradableMarkets, selectedTradeMarket) {
+        this.driftClient = driftClient;
+        this.accountLoader = accountLoader;
+        this.pollingDlob = pollingDlob;
+        this.orderbookManager = orderbookManager;
+        this.userAccountCache = userAccountCache;
+        this.tradableMarkets = tradableMarkets;
+        this.selectedTradeMarket = selectedTradeMarket;
+    }
+    /**
+     * Updates the tradable markets list.
+     *
+     * @param tradableMarkets - The new tradable markets list
+     */
+    updateTradableMarkets(tradableMarkets) {
+        this.tradableMarkets = tradableMarkets;
+    }
+    /**
+     * Subscribes to updates for all user accounts under the current authority.
+     *
+     * This method sets up event listeners for user account updates and ensures
+     * that market subscriptions are properly configured based on user positions.
+     * When a user's account data changes, it automatically triggers subscription
+     * updates to maintain optimal polling frequencies.
+     */
+    subscribeToAllUsersUpdates() {
+        const users = this.driftClient.getUsers();
+        this.handleSubscriptionUpdatesOnUserUpdates(users);
+        users.forEach((user) => {
+            this.userAccountCache.updateUserAccount(user); // initial hydration of user account data in the store
+            user.eventEmitter.on('update', () => {
+                this.handleSubscriptionUpdatesOnUserUpdates(users);
+                this.userAccountCache.updateUserAccount(user);
+            });
+        });
+    }
+    /**
+     * Categorizes markets based on user involvement (active positions).
+     *
+     * Markets are classified as "user-involved" if any of the provided users
+     * have active positions (perp or spot) in those markets. This categorization
+     * is used to optimize polling frequencies - user-involved markets are polled
+     * more frequently to ensure accurate close to real-time data for user positions.
+     */
+    categorizeMarketsByUserInvolvement(users) {
+        const perpMarketIndexesSet = new Set();
+        const spotMarketIndexesSet = new Set();
+        users.forEach((user) => {
+            const { activePerpPositions, activeSpotPositions } = user.getActivePositions();
+            activePerpPositions.forEach((marketIndex) => perpMarketIndexesSet.add(marketIndex));
+            activeSpotPositions.forEach((marketIndex) => spotMarketIndexesSet.add(marketIndex));
+        });
+        const userInvolvedMarkets = Array.from(perpMarketIndexesSet)
+            .map((index) => types_1.MarketId.createPerpMarket(index))
+            .concat(Array.from(spotMarketIndexesSet).map((index) => types_1.MarketId.createSpotMarket(index)));
+        const userInvolvedMarketKeys = userInvolvedMarkets.map((market) => market.key);
+        const userNotInvolvedMarkets = this.tradableMarkets.filter((market) => !userInvolvedMarketKeys.includes(market.key));
+        return { userInvolvedMarkets, userNotInvolvedMarkets };
+    }
+    /**
+     * Updates the polling cadence for a market account and its associated oracle account.
+     *
+     * This method optimizes data fetching by adjusting how frequently market and oracle
+     * accounts are polled from the RPC. Higher cadences (lower numbers) mean more frequent polling.
+     */
+    updateMarketAccountCadence(market, newCadence) {
+        const marketAccount = market.isPerp
+            ? this.driftClient.getPerpMarketAccount(market.marketIndex)
+            : this.driftClient.getSpotMarketAccount(market.marketIndex);
+        if (!marketAccount) {
+            throw new Error(`Market account not found for market ${market.key}`);
+        }
+        const currentMarketCadence = this.accountLoader.getAccountCadence(marketAccount.pubkey);
+        if (currentMarketCadence !== newCadence) {
+            this.accountLoader.setCustomPollingFrequency(marketAccount.pubkey, newCadence);
+        }
+        const oracleAccountPubKey = market.isPerp
+            ? marketAccount.amm.oracle
+            : marketAccount.oracle;
+        const currentOracleCadence = this.accountLoader.getAccountCadence(oracleAccountPubKey);
+        if (currentOracleCadence !== newCadence) {
+            this.accountLoader.setCustomPollingFrequency(oracleAccountPubKey, newCadence);
+        }
+    }
+    /**
+     * Updates account loader polling cadences for multiple markets based on their categorization.
+     *
+     * This method applies different polling frequencies based on market importance:
+     * - Selected trade market: Highest frequency (most important for active trading)
+     * - User-involved markets: Medium frequency (user has positions)
+     * - User-not-involved markets: Lowest frequency (background monitoring)
+     */
+    updateAccountLoaderCadenceForMarkets(userInvolvedMarkets, userNotInvolvedMarkets, selectedTradeMarket) {
+        if (selectedTradeMarket) {
+            this.updateMarketAccountCadence(selectedTradeMarket, constants_1.SELECTED_MARKET_ACCOUNT_POLLING_CADENCE);
+        }
+        userInvolvedMarkets.forEach((market) => {
+            this.updateMarketAccountCadence(market, constants_1.USER_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE);
+        });
+        userNotInvolvedMarkets.forEach((market) => {
+            this.updateMarketAccountCadence(market, constants_1.USER_NOT_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE);
+        });
+    }
+    /**
+     * Updates DLOB (Decentralized Limit Order Book) polling intervals for market data.
+     *
+     * This method configures how frequently the DLOB server is polled for market data
+     * including mark prices, oracle prices, and orderbook data. Different markets are
+     * assigned to different polling intervals based on their importance.
+     */
+    updatePollingDlobIntervals(userInvolvedMarkets, userNotInvolvedMarkets) {
+        this.pollingDlob.addMarketsToInterval(constants_1.PollingCategory.USER_INVOLVED, userInvolvedMarkets);
+        this.pollingDlob.addMarketsToInterval(constants_1.PollingCategory.USER_NOT_INVOLVED, userNotInvolvedMarkets);
+    }
+    /**
+     * Handles comprehensive subscription updates when user account data changes.
+     *
+     * This is the main orchestration method that ensures optimal polling frequencies
+     * are maintained when user positions change. It automatically:
+     *
+     * 1. Categorizes markets based on current user involvement
+     * 2. Updates account loader cadences for market and oracle accounts
+     * 3. Updates DLOB polling intervals for price data
+     *
+     * The method ensures that markets with user positions get higher priority polling
+     * to maintain accurate real-time data for position management.
+     */
+    handleSubscriptionUpdatesOnUserUpdates(users) {
+        const { userInvolvedMarkets, userNotInvolvedMarkets } = this.categorizeMarketsByUserInvolvement(users);
+        // Filter out selected trade market from the other categories
+        const filteredUserInvolvedMarkets = userInvolvedMarkets.filter((market) => { var _a; return market.key !== ((_a = this.selectedTradeMarket) === null || _a === void 0 ? void 0 : _a.key); });
+        const filteredUserNotInvolvedMarkets = userNotInvolvedMarkets.filter((market) => { var _a; return market.key !== ((_a = this.selectedTradeMarket) === null || _a === void 0 ? void 0 : _a.key); });
+        // Update market account cadences
+        this.updateAccountLoaderCadenceForMarkets(filteredUserInvolvedMarkets, filteredUserNotInvolvedMarkets, this.selectedTradeMarket);
+        // Handle polling dlob polling intervals
+        this.updatePollingDlobIntervals(filteredUserInvolvedMarkets.map((market) => market.key), filteredUserNotInvolvedMarkets.map((market) => market.key)
+        // this.selectedTradeMarket
+        );
+    }
+    /**
+     * Subscribes to markets that users are involved in but aren't in the default tradable markets list.
+     *
+     * This method ensures that if users have positions in markets that weren't initially
+     * included in the tradable markets list, those markets are still properly subscribed to.
+     * This is important for maintaining accurate account data even for edge cases.
+     */
+    async subscribeToNonWhitelistedButUserInvolvedMarkets(users) {
+        const perpMarketIndexesSet = new Set(this.driftClient
+            .accountSubscriber.perpMarketIndexes);
+        const spotMarketIndexesSet = new Set(this.driftClient
+            .accountSubscriber.spotMarketIndexes);
+        const { userInvolvedMarkets } = this.categorizeMarketsByUserInvolvement(users);
+        userInvolvedMarkets.forEach((market) => {
+            if (market.isPerp) {
+                perpMarketIndexesSet.add(market.marketIndex);
+            }
+            else {
+                spotMarketIndexesSet.add(market.marketIndex);
+            }
+        });
+        this.driftClient.accountSubscriber.perpMarketIndexes = Array.from(perpMarketIndexesSet);
+        this.driftClient.accountSubscriber.spotMarketIndexes = Array.from(spotMarketIndexesSet);
+        // TODO: see if this can be optimized - instead of unsubscribing and resubscribing, find a way to add the new markets to the existing subscription
+        await this.driftClient.accountSubscriber.unsubscribe();
+        await this.driftClient.accountSubscriber.subscribe();
+    }
+    checkIsUserInvolvedInMarket(marketId) {
+        let isUserInvolvedInPreviousSelectedTradeMarket = false;
+        const allUsers = this.userAccountCache.allUsers;
+        for (const user of allUsers) {
+            const { activePerpPositions, activeSpotPositions } = user.userClient.getActivePositions();
+            if (marketId.isPerp &&
+                activePerpPositions.includes(marketId.marketIndex)) {
+                isUserInvolvedInPreviousSelectedTradeMarket = true;
+                break;
+            }
+            if (marketId.isSpot &&
+                activeSpotPositions.includes(marketId.marketIndex)) {
+                isUserInvolvedInPreviousSelectedTradeMarket = true;
+                break;
+            }
+        }
+        return isUserInvolvedInPreviousSelectedTradeMarket;
+    }
+    /**
+     * Updates the selected trade market and optimizes all subscription polling accordingly.
+     *
+     * This method handles:
+     * 1. Removing the previous market from high-frequency polling
+     * 2. Adding the new market to high-frequency polling
+     * 3. Updating market account cadences for optimal performance
+     * 4. Categorizing the previous market based on user involvement
+     * 5. Assigning appropriate polling intervals to the previous market
+     *
+     * @param newSelectedTradeMarket - The new market to prioritize for trading
+     */
+    updateSelectedTradeMarket(newSelectedTradeMarket) {
+        const previousSelectedTradeMarket = this.selectedTradeMarket;
+        this.orderbookManager.unsubscribe();
+        if (previousSelectedTradeMarket) {
+            const isUserInvolvedInPreviousSelectedTradeMarket = this.checkIsUserInvolvedInMarket(previousSelectedTradeMarket);
+            const pollingCategoryOfPreviousSelectedMarket = isUserInvolvedInPreviousSelectedTradeMarket
+                ? constants_1.PollingCategory.USER_INVOLVED
+                : constants_1.PollingCategory.USER_NOT_INVOLVED;
+            this.pollingDlob.addMarketToInterval(pollingCategoryOfPreviousSelectedMarket, previousSelectedTradeMarket.key);
+            const pollingCadenceOfPreviousSelectedMarket = isUserInvolvedInPreviousSelectedTradeMarket
+                ? constants_1.USER_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE
+                : constants_1.USER_NOT_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE;
+            this.updateMarketAccountCadence(previousSelectedTradeMarket, pollingCadenceOfPreviousSelectedMarket);
+        }
+        // Update the selected trade market
+        this.selectedTradeMarket = newSelectedTradeMarket;
+        if (newSelectedTradeMarket) {
+            const isUserInvolvedInNewSelectedTradeMarket = this.checkIsUserInvolvedInMarket(newSelectedTradeMarket);
+            const pollingCategoryOfNewsSelectedMarket = isUserInvolvedInNewSelectedTradeMarket
+                ? constants_1.PollingCategory.USER_INVOLVED
+                : constants_1.PollingCategory.USER_NOT_INVOLVED;
+            this.pollingDlob.removeMarketFromInterval(pollingCategoryOfNewsSelectedMarket, newSelectedTradeMarket.key);
+            this.updateMarketAccountCadence(newSelectedTradeMarket, constants_1.SELECTED_MARKET_ACCOUNT_POLLING_CADENCE);
+            this.orderbookManager.updateSubscription({
+                ...orderbook_1.DEFAULT_ORDERBOOK_SUBSCRIPTION_CONFIG,
+                marketId: newSelectedTradeMarket,
+            });
+        }
+    }
+    /**
+     * Updates the authority (wallet) for the drift client and reestablishes subscriptions.
+     *
+     * This method handles the complete process of switching to a new wallet:
+     * 1. Checks if the wallet is actually different
+     * 2. Unsubscribes from current user accounts
+     * 3. Resets the user account cache
+     * 4. Updates the DriftClient with the new wallet
+     * 5. Resubscribes to markets based on new user positions
+     * 6. Switches to the specified subaccount if provided
+     * 7. Reestablishes user account subscriptions
+     */
+    async updateAuthority(wallet, activeSubAccountId) {
+        if (this.driftClient.wallet.publicKey.equals(wallet.publicKey)) {
+            return;
+        }
+        await Promise.all(this.driftClient.unsubscribeUsers());
+        this.userAccountCache.reset();
+        const updateWalletResult = await this.driftClient.updateWallet(wallet, undefined, activeSubAccountId, true, undefined);
+        await this.subscribeToNonWhitelistedButUserInvolvedMarkets(this.driftClient.getUsers());
+        if (!updateWalletResult) {
+            throw new Error('Failed to update wallet');
+        }
+        if (activeSubAccountId) {
+            await this.driftClient.switchActiveUser(activeSubAccountId);
+        }
+        this.subscribeToAllUsersUpdates();
+    }
+}
+exports.SubscriptionManager = SubscriptionManager;
+//# sourceMappingURL=SubscriptionManager.js.map

package/lib/drift/Drift/clients/AuthorityDrift/SubscriptionManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SubscriptionManager.js","sourceRoot":"","sources":["../../../../../src/drift/Drift/clients/AuthorityDrift/SubscriptionManager.ts"],"names":[],"mappings":";;;AASA,6CAAwD;AAExD,+CAKyB;AAGzB,yDAAkF;AAElF;;;;;;;;;;;;;GAaG;AACH,MAAa,mBAAmB;IAC/B;;;;;;;;;OASG;IACH,YACS,WAAwB,EACxB,aAAiD,EACjD,WAAwB,EACxB,gBAAyC,EACzC,gBAAkC,EAClC,eAA2B,EAC3B,mBAAoC;QANpC,gBAAW,GAAX,WAAW,CAAa;QACxB,kBAAa,GAAb,aAAa,CAAoC;QACjD,gBAAW,GAAX,WAAW,CAAa;QACxB,qBAAgB,GAAhB,gBAAgB,CAAyB;QACzC,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,oBAAe,GAAf,eAAe,CAAY;QAC3B,wBAAmB,GAAnB,mBAAmB,CAAiB;IAC1C,CAAC;IAEJ;;;;OAIG;IACH,qBAAqB,CAAC,eAA2B;QAChD,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACxC,CAAC;IAED;;;;;;;OAOG;IACH,0BAA0B;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAE1C,IAAI,CAAC,sCAAsC,CAAC,KAAK,CAAC,CAAC;QAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACtB,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,sDAAsD;YAErG,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;gBACnC,IAAI,CAAC,sCAAsC,CAAC,KAAK,CAAC,CAAC;gBACnD,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,kCAAkC,CAAC,KAAa;QAI/C,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/C,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACtB,MAAM,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,GACjD,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC3B,mBAAmB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAC3C,oBAAoB,CAAC,GAAG,CAAC,WAAW,CAAC,CACrC,CAAC;YACF,mBAAmB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAC3C,oBAAoB,CAAC,GAAG,CAAC,WAAW,CAAC,CACrC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC;aAC1D,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,gBAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;aAChD,MAAM,CACN,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC9C,gBAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAChC,CACD,CAAC;QAEH,MAAM,sBAAsB,GAAG,mBAAmB,CAAC,GAAG,CACrD,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CACtB,CAAC;QACF,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CACzD,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,sBAAsB,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CACxD,CAAC;QAEF,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,CAAC;IACxD,CAAC;IAED;;;;;OAKG;IACH,0BAA0B,CAAC,MAAgB,EAAE,UAAkB;QAC9D,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM;YAClC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC;YAC3D,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAE7D,IAAI,CAAC,aAAa,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,uCAAuC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,oBAAoB,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAChE,aAAa,CAAC,MAAM,CACpB,CAAC;QAEF,IAAI,oBAAoB,KAAK,UAAU,EAAE,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,yBAAyB,CAC3C,aAAa,CAAC,MAAM,EACpB,UAAU,CACV,CAAC;QACH,CAAC;QAED,MAAM,mBAAmB,GAAG,MAAM,CAAC,MAAM;YACxC,CAAC,CAAE,aAAmC,CAAC,GAAG,CAAC,MAAM;YACjD,CAAC,CAAE,aAAmC,CAAC,MAAM,CAAC;QAE/C,MAAM,oBAAoB,GACzB,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;QAE3D,IAAI,oBAAoB,KAAK,UAAU,EAAE,CAAC;YACzC,IAAI,CAAC,aAAa,CAAC,yBAAyB,CAC3C,mBAAmB,EACnB,UAAU,CACV,CAAC;QACH,CAAC;IACF,CAAC;IAED;;;;;;;OAOG;IACK,oCAAoC,CAC3C,mBAA+B,EAC/B,sBAAkC,EAClC,mBAAqC;QAErC,IAAI,mBAAmB,EAAE,CAAC;YACzB,IAAI,CAAC,0BAA0B,CAC9B,mBAAmB,EACnB,mDAAuC,CACvC,CAAC;QACH,CAAC;QAED,mBAAmB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACtC,IAAI,CAAC,0BAA0B,CAC9B,MAAM,EACN,wDAA4C,CAC5C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,sBAAsB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACzC,IAAI,CAAC,0BAA0B,CAC9B,MAAM,EACN,4DAAgD,CAChD,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACK,0BAA0B,CACjC,mBAAgC,EAChC,sBAAmC;QAEnC,IAAI,CAAC,WAAW,CAAC,oBAAoB,CACpC,2BAAe,CAAC,aAAa,EAC7B,mBAAmB,CACnB,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,oBAAoB,CACpC,2BAAe,CAAC,iBAAiB,EACjC,sBAAsB,CACtB,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,sCAAsC,CAAC,KAAa;QACnD,MAAM,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,GACpD,IAAI,CAAC,kCAAkC,CAAC,KAAK,CAAC,CAAC;QAEhD,6DAA6D;QAC7D,MAAM,2BAA2B,GAAG,mBAAmB,CAAC,MAAM,CAC7D,CAAC,MAAM,EAAE,EAAE,WAAC,OAAA,MAAM,CAAC,GAAG,MAAK,MAAA,IAAI,CAAC,mBAAmB,0CAAE,GAAG,CAAA,CAAA,EAAA,CACxD,CAAC;QACF,MAAM,8BAA8B,GAAG,sBAAsB,CAAC,MAAM,CACnE,CAAC,MAAM,EAAE,EAAE,WAAC,OAAA,MAAM,CAAC,GAAG,MAAK,MAAA,IAAI,CAAC,mBAAmB,0CAAE,GAAG,CAAA,CAAA,EAAA,CACxD,CAAC;QAEF,iCAAiC;QACjC,IAAI,CAAC,oCAAoC,CACxC,2BAA2B,EAC3B,8BAA8B,EAC9B,IAAI,CAAC,mBAAmB,CACxB,CAAC;QAEF,wCAAwC;QACxC,IAAI,CAAC,0BAA0B,CAC9B,2BAA2B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EACvD,8BAA8B,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC;QAC1D,2BAA2B;SAC3B,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,+CAA+C,CACpD,KAAa;QAEb,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAElC,IAAI,CAAC,WAAW;aACd,iBACF,CAAC,iBAAiB,CACnB,CAAC;QACF,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAElC,IAAI,CAAC,WAAW;aACd,iBACF,CAAC,iBAAiB,CACnB,CAAC;QAEF,MAAM,EAAE,mBAAmB,EAAE,GAC5B,IAAI,CAAC,kCAAkC,CAAC,KAAK,CAAC,CAAC;QAEhD,mBAAmB,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACtC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnB,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACP,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC9C,CAAC;QACF,CAAC,CAAC,CAAC;QAGF,IAAI,CAAC,WAAW,CAAC,iBACjB,CAAC,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAEtD,IAAI,CAAC,WAAW,CAAC,iBACjB,CAAC,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAEvD,kJAAkJ;QAClJ,MACC,IAAI,CAAC,WAAW,CAAC,iBACjB,CAAC,WAAW,EAAE,CAAC;QAChB,MACC,IAAI,CAAC,WAAW,CAAC,iBACjB,CAAC,SAAS,EAAE,CAAC;IACf,CAAC;IAEO,2BAA2B,CAAC,QAAkB;QACrD,IAAI,2CAA2C,GAAG,KAAK,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;QAChD,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,GACjD,IAAI,CAAC,UAAU,CAAC,kBAAkB,EAAE,CAAC;YAEtC,IACC,QAAQ,CAAC,MAAM;gBACf,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,EACjD,CAAC;gBACF,2CAA2C,GAAG,IAAI,CAAC;gBACnD,MAAM;YACP,CAAC;YACD,IACC,QAAQ,CAAC,MAAM;gBACf,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,EACjD,CAAC;gBACF,2CAA2C,GAAG,IAAI,CAAC;gBACnD,MAAM;YACP,CAAC;QACF,CAAC;QAED,OAAO,2CAA2C,CAAC;IACpD,CAAC;IAED;;;;;;;;;;;OAWG;IACH,yBAAyB,CAAC,sBAAuC;QAChE,MAAM,2BAA2B,GAAG,IAAI,CAAC,mBAAmB,CAAC;QAE7D,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC;QAEpC,IAAI,2BAA2B,EAAE,CAAC;YACjC,MAAM,2CAA2C,GAChD,IAAI,CAAC,2BAA2B,CAAC,2BAA2B,CAAC,CAAC;YAE/D,MAAM,uCAAuC,GAC5C,2CAA2C;gBAC1C,CAAC,CAAC,2BAAe,CAAC,aAAa;gBAC/B,CAAC,CAAC,2BAAe,CAAC,iBAAiB,CAAC;YACtC,IAAI,CAAC,WAAW,CAAC,mBAAmB,CACnC,uCAAuC,EACvC,2BAA2B,CAAC,GAAG,CAC/B,CAAC;YAEF,MAAM,sCAAsC,GAC3C,2CAA2C;gBAC1C,CAAC,CAAC,wDAA4C;gBAC9C,CAAC,CAAC,4DAAgD,CAAC;YACrD,IAAI,CAAC,0BAA0B,CAC9B,2BAA2B,EAC3B,sCAAsC,CACtC,CAAC;QACH,CAAC;QAED,mCAAmC;QACnC,IAAI,CAAC,mBAAmB,GAAG,sBAAsB,CAAC;QAElD,IAAI,sBAAsB,EAAE,CAAC;YAC5B,MAAM,sCAAsC,GAC3C,IAAI,CAAC,2BAA2B,CAAC,sBAAsB,CAAC,CAAC;YAE1D,MAAM,mCAAmC,GACxC,sCAAsC;gBACrC,CAAC,CAAC,2BAAe,CAAC,aAAa;gBAC/B,CAAC,CAAC,2BAAe,CAAC,iBAAiB,CAAC;YACtC,IAAI,CAAC,WAAW,CAAC,wBAAwB,CACxC,mCAAmC,EACnC,sBAAsB,CAAC,GAAG,CAC1B,CAAC;YAEF,IAAI,CAAC,0BAA0B,CAC9B,sBAAsB,EACtB,mDAAuC,CACvC,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC;gBACxC,GAAG,iDAAqC;gBACxC,QAAQ,EAAE,sBAAsB;aAChC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,eAAe,CACpB,MAAe,EACf,kBAA2B;QAE3B,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAChE,OAAO;QACR,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACvD,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAE9B,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAC7D,MAAM,EACN,SAAS,EACT,kBAAkB,EAClB,IAAI,EACJ,SAAS,CACT,CAAC;QAEF,MAAM,IAAI,CAAC,+CAA+C,CACzD,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAC3B,CAAC;QAEF,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,kBAAkB,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,0BAA0B,EAAE,CAAC;IACnC,CAAC;CACD;AAtaD,kDAsaC","sourcesContent":["import {\n\tCustomizedCadenceBulkAccountLoader,\n\tDriftClient,\n\tIWallet,\n\tPerpMarketAccount,\n\tPollingDriftClientAccountSubscriber,\n\tSpotMarketAccount,\n\tUser,\n} from '@drift-labs/sdk';\nimport { MarketId, MarketKey } from '../../../../types';\nimport { PollingDlob } from '../../data/PollingDlob';\nimport {\n\tPollingCategory,\n\tSELECTED_MARKET_ACCOUNT_POLLING_CADENCE,\n\tUSER_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE,\n\tUSER_NOT_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE,\n} from '../../constants';\nimport { UserAccountCache } from '../../stores/UserAccountCache';\nimport { DriftL2OrderbookManager } from './DriftL2OrderbookManager';\nimport { DEFAULT_ORDERBOOK_SUBSCRIPTION_CONFIG } from '../../constants/orderbook';\n\n/**\n * Comprehensive subscription and market operations manager for the Drift protocol.\n *\n * This class handles all aspects of subscription management including:\n * - User account subscriptions and event listeners\n * - Market account and oracle polling optimization\n * \t- Markets are categorized based on user involvement, and polling cadences are updated accordingly\n * - DLOB server polling configuration (for lesser used markets)\n * - DLOB server websocket subscription configuration (for selected trade market). Used to get orderbook data too.\n * - Authority (wallet) management with full resubscription\n *\n * By combining subscription and market operations, this provides a unified\n * interface for all subscription-related optimizations.\n */\nexport class SubscriptionManager {\n\t/**\n\t * Creates a new SubscriptionManager instance.\n\t *\n\t * @param driftClient - The DriftClient instance for managing subscriptions\n\t * @param accountLoader - Handles bulk account loading and polling frequency management\n\t * @param pollingDlob - Manages DLOB server polling for market data\n\t * @param userAccountCache - Cache for user account data updates\n\t * @param tradableMarkets - Array of markets that are available for trading\n\t * @param selectedTradeMarket - The market that is currently being traded\n\t */\n\tconstructor(\n\t\tprivate driftClient: DriftClient,\n\t\tprivate accountLoader: CustomizedCadenceBulkAccountLoader,\n\t\tprivate pollingDlob: PollingDlob,\n\t\tprivate orderbookManager: DriftL2OrderbookManager,\n\t\tprivate userAccountCache: UserAccountCache,\n\t\tprivate tradableMarkets: MarketId[],\n\t\tprivate selectedTradeMarket: MarketId | null\n\t) {}\n\n\t/**\n\t * Updates the tradable markets list.\n\t *\n\t * @param tradableMarkets - The new tradable markets list\n\t */\n\tupdateTradableMarkets(tradableMarkets: MarketId[]): void {\n\t\tthis.tradableMarkets = tradableMarkets;\n\t}\n\n\t/**\n\t * Subscribes to updates for all user accounts under the current authority.\n\t *\n\t * This method sets up event listeners for user account updates and ensures\n\t * that market subscriptions are properly configured based on user positions.\n\t * When a user's account data changes, it automatically triggers subscription\n\t * updates to maintain optimal polling frequencies.\n\t */\n\tsubscribeToAllUsersUpdates(): void {\n\t\tconst users = this.driftClient.getUsers();\n\n\t\tthis.handleSubscriptionUpdatesOnUserUpdates(users);\n\n\t\tusers.forEach((user) => {\n\t\t\tthis.userAccountCache.updateUserAccount(user); // initial hydration of user account data in the store\n\n\t\t\tuser.eventEmitter.on('update', () => {\n\t\t\t\tthis.handleSubscriptionUpdatesOnUserUpdates(users);\n\t\t\t\tthis.userAccountCache.updateUserAccount(user);\n\t\t\t});\n\t\t});\n\t}\n\n\t/**\n\t * Categorizes markets based on user involvement (active positions).\n\t *\n\t * Markets are classified as \"user-involved\" if any of the provided users\n\t * have active positions (perp or spot) in those markets. This categorization\n\t * is used to optimize polling frequencies - user-involved markets are polled\n\t * more frequently to ensure accurate close to real-time data for user positions.\n\t */\n\tcategorizeMarketsByUserInvolvement(users: User[]): {\n\t\tuserInvolvedMarkets: MarketId[];\n\t\tuserNotInvolvedMarkets: MarketId[];\n\t} {\n\t\tconst perpMarketIndexesSet = new Set<number>();\n\t\tconst spotMarketIndexesSet = new Set<number>();\n\t\tusers.forEach((user) => {\n\t\t\tconst { activePerpPositions, activeSpotPositions } =\n\t\t\t\tuser.getActivePositions();\n\t\t\tactivePerpPositions.forEach((marketIndex) =>\n\t\t\t\tperpMarketIndexesSet.add(marketIndex)\n\t\t\t);\n\t\t\tactiveSpotPositions.forEach((marketIndex) =>\n\t\t\t\tspotMarketIndexesSet.add(marketIndex)\n\t\t\t);\n\t\t});\n\n\t\tconst userInvolvedMarkets = Array.from(perpMarketIndexesSet)\n\t\t\t.map((index) => MarketId.createPerpMarket(index))\n\t\t\t.concat(\n\t\t\t\tArray.from(spotMarketIndexesSet).map((index) =>\n\t\t\t\t\tMarketId.createSpotMarket(index)\n\t\t\t\t)\n\t\t\t);\n\n\t\tconst userInvolvedMarketKeys = userInvolvedMarkets.map(\n\t\t\t(market) => market.key\n\t\t);\n\t\tconst userNotInvolvedMarkets = this.tradableMarkets.filter(\n\t\t\t(market) => !userInvolvedMarketKeys.includes(market.key)\n\t\t);\n\n\t\treturn { userInvolvedMarkets, userNotInvolvedMarkets };\n\t}\n\n\t/**\n\t * Updates the polling cadence for a market account and its associated oracle account.\n\t *\n\t * This method optimizes data fetching by adjusting how frequently market and oracle\n\t * accounts are polled from the RPC. Higher cadences (lower numbers) mean more frequent polling.\n\t */\n\tupdateMarketAccountCadence(market: MarketId, newCadence: number): void {\n\t\tconst marketAccount = market.isPerp\n\t\t\t? this.driftClient.getPerpMarketAccount(market.marketIndex)\n\t\t\t: this.driftClient.getSpotMarketAccount(market.marketIndex);\n\n\t\tif (!marketAccount) {\n\t\t\tthrow new Error(`Market account not found for market ${market.key}`);\n\t\t}\n\n\t\tconst currentMarketCadence = this.accountLoader.getAccountCadence(\n\t\t\tmarketAccount.pubkey\n\t\t);\n\n\t\tif (currentMarketCadence !== newCadence) {\n\t\t\tthis.accountLoader.setCustomPollingFrequency(\n\t\t\t\tmarketAccount.pubkey,\n\t\t\t\tnewCadence\n\t\t\t);\n\t\t}\n\n\t\tconst oracleAccountPubKey = market.isPerp\n\t\t\t? (marketAccount as PerpMarketAccount).amm.oracle\n\t\t\t: (marketAccount as SpotMarketAccount).oracle;\n\n\t\tconst currentOracleCadence =\n\t\t\tthis.accountLoader.getAccountCadence(oracleAccountPubKey);\n\n\t\tif (currentOracleCadence !== newCadence) {\n\t\t\tthis.accountLoader.setCustomPollingFrequency(\n\t\t\t\toracleAccountPubKey,\n\t\t\t\tnewCadence\n\t\t\t);\n\t\t}\n\t}\n\n\t/**\n\t * Updates account loader polling cadences for multiple markets based on their categorization.\n\t *\n\t * This method applies different polling frequencies based on market importance:\n\t * - Selected trade market: Highest frequency (most important for active trading)\n\t * - User-involved markets: Medium frequency (user has positions)\n\t * - User-not-involved markets: Lowest frequency (background monitoring)\n\t */\n\tprivate updateAccountLoaderCadenceForMarkets(\n\t\tuserInvolvedMarkets: MarketId[],\n\t\tuserNotInvolvedMarkets: MarketId[],\n\t\tselectedTradeMarket?: MarketId | null\n\t): void {\n\t\tif (selectedTradeMarket) {\n\t\t\tthis.updateMarketAccountCadence(\n\t\t\t\tselectedTradeMarket,\n\t\t\t\tSELECTED_MARKET_ACCOUNT_POLLING_CADENCE\n\t\t\t);\n\t\t}\n\n\t\tuserInvolvedMarkets.forEach((market) => {\n\t\t\tthis.updateMarketAccountCadence(\n\t\t\t\tmarket,\n\t\t\t\tUSER_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE\n\t\t\t);\n\t\t});\n\n\t\tuserNotInvolvedMarkets.forEach((market) => {\n\t\t\tthis.updateMarketAccountCadence(\n\t\t\t\tmarket,\n\t\t\t\tUSER_NOT_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE\n\t\t\t);\n\t\t});\n\t}\n\n\t/**\n\t * Updates DLOB (Decentralized Limit Order Book) polling intervals for market data.\n\t *\n\t * This method configures how frequently the DLOB server is polled for market data\n\t * including mark prices, oracle prices, and orderbook data. Different markets are\n\t * assigned to different polling intervals based on their importance.\n\t */\n\tprivate updatePollingDlobIntervals(\n\t\tuserInvolvedMarkets: MarketKey[],\n\t\tuserNotInvolvedMarkets: MarketKey[]\n\t): void {\n\t\tthis.pollingDlob.addMarketsToInterval(\n\t\t\tPollingCategory.USER_INVOLVED,\n\t\t\tuserInvolvedMarkets\n\t\t);\n\t\tthis.pollingDlob.addMarketsToInterval(\n\t\t\tPollingCategory.USER_NOT_INVOLVED,\n\t\t\tuserNotInvolvedMarkets\n\t\t);\n\t}\n\n\t/**\n\t * Handles comprehensive subscription updates when user account data changes.\n\t *\n\t * This is the main orchestration method that ensures optimal polling frequencies\n\t * are maintained when user positions change. It automatically:\n\t *\n\t * 1. Categorizes markets based on current user involvement\n\t * 2. Updates account loader cadences for market and oracle accounts\n\t * 3. Updates DLOB polling intervals for price data\n\t *\n\t * The method ensures that markets with user positions get higher priority polling\n\t * to maintain accurate real-time data for position management.\n\t */\n\thandleSubscriptionUpdatesOnUserUpdates(users: User[]): void {\n\t\tconst { userInvolvedMarkets, userNotInvolvedMarkets } =\n\t\t\tthis.categorizeMarketsByUserInvolvement(users);\n\n\t\t// Filter out selected trade market from the other categories\n\t\tconst filteredUserInvolvedMarkets = userInvolvedMarkets.filter(\n\t\t\t(market) => market.key !== this.selectedTradeMarket?.key\n\t\t);\n\t\tconst filteredUserNotInvolvedMarkets = userNotInvolvedMarkets.filter(\n\t\t\t(market) => market.key !== this.selectedTradeMarket?.key\n\t\t);\n\n\t\t// Update market account cadences\n\t\tthis.updateAccountLoaderCadenceForMarkets(\n\t\t\tfilteredUserInvolvedMarkets,\n\t\t\tfilteredUserNotInvolvedMarkets,\n\t\t\tthis.selectedTradeMarket\n\t\t);\n\n\t\t// Handle polling dlob polling intervals\n\t\tthis.updatePollingDlobIntervals(\n\t\t\tfilteredUserInvolvedMarkets.map((market) => market.key),\n\t\t\tfilteredUserNotInvolvedMarkets.map((market) => market.key)\n\t\t\t// this.selectedTradeMarket\n\t\t);\n\t}\n\n\t/**\n\t * Subscribes to markets that users are involved in but aren't in the default tradable markets list.\n\t *\n\t * This method ensures that if users have positions in markets that weren't initially\n\t * included in the tradable markets list, those markets are still properly subscribed to.\n\t * This is important for maintaining accurate account data even for edge cases.\n\t */\n\tasync subscribeToNonWhitelistedButUserInvolvedMarkets(\n\t\tusers: User[]\n\t): Promise<void> {\n\t\tconst perpMarketIndexesSet = new Set(\n\t\t\t(\n\t\t\t\tthis.driftClient\n\t\t\t\t\t.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t\t).perpMarketIndexes\n\t\t);\n\t\tconst spotMarketIndexesSet = new Set(\n\t\t\t(\n\t\t\t\tthis.driftClient\n\t\t\t\t\t.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t\t).spotMarketIndexes\n\t\t);\n\n\t\tconst { userInvolvedMarkets } =\n\t\t\tthis.categorizeMarketsByUserInvolvement(users);\n\n\t\tuserInvolvedMarkets.forEach((market) => {\n\t\t\tif (market.isPerp) {\n\t\t\t\tperpMarketIndexesSet.add(market.marketIndex);\n\t\t\t} else {\n\t\t\t\tspotMarketIndexesSet.add(market.marketIndex);\n\t\t\t}\n\t\t});\n\n\t\t(\n\t\t\tthis.driftClient.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t).perpMarketIndexes = Array.from(perpMarketIndexesSet);\n\t\t(\n\t\t\tthis.driftClient.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t).spotMarketIndexes = Array.from(spotMarketIndexesSet);\n\n\t\t// TODO: see if this can be optimized - instead of unsubscribing and resubscribing, find a way to add the new markets to the existing subscription\n\t\tawait (\n\t\t\tthis.driftClient.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t).unsubscribe();\n\t\tawait (\n\t\t\tthis.driftClient.accountSubscriber as PollingDriftClientAccountSubscriber\n\t\t).subscribe();\n\t}\n\n\tprivate checkIsUserInvolvedInMarket(marketId: MarketId): boolean {\n\t\tlet isUserInvolvedInPreviousSelectedTradeMarket = false;\n\t\tconst allUsers = this.userAccountCache.allUsers;\n\t\tfor (const user of allUsers) {\n\t\t\tconst { activePerpPositions, activeSpotPositions } =\n\t\t\t\tuser.userClient.getActivePositions();\n\n\t\t\tif (\n\t\t\t\tmarketId.isPerp &&\n\t\t\t\tactivePerpPositions.includes(marketId.marketIndex)\n\t\t\t) {\n\t\t\t\tisUserInvolvedInPreviousSelectedTradeMarket = true;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tif (\n\t\t\t\tmarketId.isSpot &&\n\t\t\t\tactiveSpotPositions.includes(marketId.marketIndex)\n\t\t\t) {\n\t\t\t\tisUserInvolvedInPreviousSelectedTradeMarket = true;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\treturn isUserInvolvedInPreviousSelectedTradeMarket;\n\t}\n\n\t/**\n\t * Updates the selected trade market and optimizes all subscription polling accordingly.\n\t *\n\t * This method handles:\n\t * 1. Removing the previous market from high-frequency polling\n\t * 2. Adding the new market to high-frequency polling\n\t * 3. Updating market account cadences for optimal performance\n\t * 4. Categorizing the previous market based on user involvement\n\t * 5. Assigning appropriate polling intervals to the previous market\n\t *\n\t * @param newSelectedTradeMarket - The new market to prioritize for trading\n\t */\n\tupdateSelectedTradeMarket(newSelectedTradeMarket: MarketId | null): void {\n\t\tconst previousSelectedTradeMarket = this.selectedTradeMarket;\n\n\t\tthis.orderbookManager.unsubscribe();\n\n\t\tif (previousSelectedTradeMarket) {\n\t\t\tconst isUserInvolvedInPreviousSelectedTradeMarket =\n\t\t\t\tthis.checkIsUserInvolvedInMarket(previousSelectedTradeMarket);\n\n\t\t\tconst pollingCategoryOfPreviousSelectedMarket =\n\t\t\t\tisUserInvolvedInPreviousSelectedTradeMarket\n\t\t\t\t\t? PollingCategory.USER_INVOLVED\n\t\t\t\t\t: PollingCategory.USER_NOT_INVOLVED;\n\t\t\tthis.pollingDlob.addMarketToInterval(\n\t\t\t\tpollingCategoryOfPreviousSelectedMarket,\n\t\t\t\tpreviousSelectedTradeMarket.key\n\t\t\t);\n\n\t\t\tconst pollingCadenceOfPreviousSelectedMarket =\n\t\t\t\tisUserInvolvedInPreviousSelectedTradeMarket\n\t\t\t\t\t? USER_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE\n\t\t\t\t\t: USER_NOT_INVOLVED_MARKET_ACCOUNT_POLLING_CADENCE;\n\t\t\tthis.updateMarketAccountCadence(\n\t\t\t\tpreviousSelectedTradeMarket,\n\t\t\t\tpollingCadenceOfPreviousSelectedMarket\n\t\t\t);\n\t\t}\n\n\t\t// Update the selected trade market\n\t\tthis.selectedTradeMarket = newSelectedTradeMarket;\n\n\t\tif (newSelectedTradeMarket) {\n\t\t\tconst isUserInvolvedInNewSelectedTradeMarket =\n\t\t\t\tthis.checkIsUserInvolvedInMarket(newSelectedTradeMarket);\n\n\t\t\tconst pollingCategoryOfNewsSelectedMarket =\n\t\t\t\tisUserInvolvedInNewSelectedTradeMarket\n\t\t\t\t\t? PollingCategory.USER_INVOLVED\n\t\t\t\t\t: PollingCategory.USER_NOT_INVOLVED;\n\t\t\tthis.pollingDlob.removeMarketFromInterval(\n\t\t\t\tpollingCategoryOfNewsSelectedMarket,\n\t\t\t\tnewSelectedTradeMarket.key\n\t\t\t);\n\n\t\t\tthis.updateMarketAccountCadence(\n\t\t\t\tnewSelectedTradeMarket,\n\t\t\t\tSELECTED_MARKET_ACCOUNT_POLLING_CADENCE\n\t\t\t);\n\n\t\t\tthis.orderbookManager.updateSubscription({\n\t\t\t\t...DEFAULT_ORDERBOOK_SUBSCRIPTION_CONFIG,\n\t\t\t\tmarketId: newSelectedTradeMarket,\n\t\t\t});\n\t\t}\n\t}\n\n\t/**\n\t * Updates the authority (wallet) for the drift client and reestablishes subscriptions.\n\t *\n\t * This method handles the complete process of switching to a new wallet:\n\t * 1. Checks if the wallet is actually different\n\t * 2. Unsubscribes from current user accounts\n\t * 3. Resets the user account cache\n\t * 4. Updates the DriftClient with the new wallet\n\t * 5. Resubscribes to markets based on new user positions\n\t * 6. Switches to the specified subaccount if provided\n\t * 7. Reestablishes user account subscriptions\n\t */\n\tasync updateAuthority(\n\t\twallet: IWallet,\n\t\tactiveSubAccountId?: number\n\t): Promise<void> {\n\t\tif (this.driftClient.wallet.publicKey.equals(wallet.publicKey)) {\n\t\t\treturn;\n\t\t}\n\n\t\tawait Promise.all(this.driftClient.unsubscribeUsers());\n\t\tthis.userAccountCache.reset();\n\n\t\tconst updateWalletResult = await this.driftClient.updateWallet(\n\t\t\twallet,\n\t\t\tundefined,\n\t\t\tactiveSubAccountId,\n\t\t\ttrue,\n\t\t\tundefined\n\t\t);\n\n\t\tawait this.subscribeToNonWhitelistedButUserInvolvedMarkets(\n\t\t\tthis.driftClient.getUsers()\n\t\t);\n\n\t\tif (!updateWalletResult) {\n\t\t\tthrow new Error('Failed to update wallet');\n\t\t}\n\n\t\tif (activeSubAccountId) {\n\t\t\tawait this.driftClient.switchActiveUser(activeSubAccountId);\n\t\t}\n\n\t\tthis.subscribeToAllUsersUpdates();\n\t}\n}\n"]}