@agg-build/sdk 4.27.3 → 4.27.4

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { a as loadImmutableHistory, c as orderbookCrc32, i as snapshotToOrderbook, l as projectedOrderbookChecksum, n as applyOrderbookDelta, r as computeChecksum, s as readChartHistoryText, t as AggWebSocket } from "./agg-websocket-DwVwCDvM.mjs";
+import { a as loadImmutableHistory, c as orderbookCrc32, i as snapshotToOrderbook, l as projectedOrderbookChecksum, n as applyOrderbookDelta, r as computeChecksum, s as readChartHistoryText, t as AggWebSocket } from "./agg-websocket-BrkWNijv.mjs";
 import { t as CandleBuilder } from "./candle-builder-BNewIe9K.mjs";
 //#region ../common/src/enums/venue.ts
 /** Use these constants instead of hardcoding "kalshi" or "polymarket". */
@@ -1163,6 +1163,7 @@ function withTimeout(p, ms) {
 var AggClient = class {
 	constructor(options) {
 		this.listeners = /* @__PURE__ */ new Set();
+		this.webSocketAuthCleanups = /* @__PURE__ */ new Set();
 		this._authStatus = "unauthenticated";
 		this.authChannel = null;
 		this.pendingRefresh = null;
@@ -1260,6 +1261,7 @@ var AggClient = class {
 				const { type } = event.data ?? {};
 				if (type === "logout") {
 					this.accessToken = void 0;
+					this.accessTokenUserId = void 0;
 					this.refreshTokenValue = void 0;
 					this.currentUser = void 0;
 					this._authStatus = "unauthenticated";
@@ -1275,6 +1277,7 @@ var AggClient = class {
 	destroy() {
 		this.authChannel?.close();
 		this.authChannel = null;
+		for (const cleanup of this.webSocketAuthCleanups) cleanup();
 	}
 	/**
 	* Generic HTTP method with automatic header injection and 401 retry.
@@ -1381,6 +1384,7 @@ var AggClient = class {
 	}
 	/** Set the user access token (for future user-tier auth). */
 	setAccessToken(token) {
+		if (this.accessToken !== token) this.accessTokenUserId = void 0;
 		this.accessToken = token;
 		this._authStatus = "authenticated";
 		this.persistSession();
@@ -1389,6 +1393,7 @@ var AggClient = class {
 	/** Set the current auth session tokens and optionally seed the user profile. */
 	setSession(session) {
 		const wasAuthenticated = this.isAuthenticated;
+		this.accessTokenUserId = session.user?.id ?? (this.accessToken === session.accessToken ? this.accessTokenUserId : void 0);
 		this.accessToken = session.accessToken;
 		if ("refreshToken" in session) this.refreshTokenValue = session.refreshToken;
 		if (session.user !== void 0) this.currentUser = normalizeSessionUser(session.user, this.currentUser);
@@ -1400,6 +1405,7 @@ var AggClient = class {
 	/** Clear the user access token, refresh token, and user info. Broadcasts cross-tab logout. */
 	clearAccessToken() {
 		this.accessToken = void 0;
+		this.accessTokenUserId = void 0;
 		this.refreshTokenValue = void 0;
 		this.currentUser = void 0;
 		this._authStatus = "unauthenticated";
@@ -1586,6 +1592,7 @@ var AggClient = class {
 	async getCurrentUser() {
 		const user = await this.request("/users/me");
 		this.currentUser = user;
+		this.accessTokenUserId = user.id;
 		this.persistSession();
 		this.notifyListeners();
 		return user;
@@ -2329,7 +2336,8 @@ var AggClient = class {
 		});
 	}
 	/**
-	* Create an AggWebSocket instance configured with this client's appId and optional auth token.
+	* Create an AggWebSocket instance configured with this client's appId and current auth session.
+	* Session changes update this socket until it or the client is destroyed.
 	* Requires `wsUrl` to be set in the client options.
 	*/
 	createWebSocket(callbacks, options) {
@@ -2340,7 +2348,7 @@ var AggClient = class {
 		const watchdogHeaders = {};
 		if (this._appId) watchdogHeaders["x-app-id"] = this._appId;
 		if (this.apiKey) watchdogHeaders["x-app-api-key"] = this.apiKey;
-		return createSocket({
+		const socket = createSocket({
 			url: url.toString(),
 			callbacks,
 			initialToken: this.accessToken,
@@ -2351,6 +2359,101 @@ var AggClient = class {
 				...options?.bookWatchdog
 			}
 		});
+		this.bindWebSocketAuth(socket, callbacks);
+		return socket;
+	}
+	bindWebSocketAuth(socket, callbacks) {
+		let token = this.accessToken;
+		let userId = this.accessTokenUserId;
+		let authenticatedUserId;
+		let awaitingDisconnect = false;
+		let connectRequested = false;
+		let manuallyDisconnected = false;
+		let currentCallbacks = callbacks ?? {};
+		const originalSetCallbacks = socket.setCallbacks;
+		const originalConnect = socket.connect;
+		const originalDisconnect = socket.disconnect;
+		const originalDestroy = socket.destroy;
+		const canReceiveUserEvent = (eventUserId) => !!this.accessToken && !awaitingDisconnect && authenticatedUserId !== void 0 && (this.accessTokenUserId === void 0 || authenticatedUserId === this.accessTokenUserId) && (eventUserId === void 0 || eventUserId === authenticatedUserId);
+		const wrapCallbacks = (handlers) => ({
+			...handlers,
+			onConnectionStateChange: (connected) => {
+				if (!connected) {
+					authenticatedUserId = void 0;
+					awaitingDisconnect = false;
+				}
+				handlers.onConnectionStateChange?.(connected);
+			},
+			onAuthenticated: (message) => {
+				if (awaitingDisconnect || !this.accessToken || this.accessTokenUserId !== void 0 && this.accessTokenUserId !== message.userId) return;
+				authenticatedUserId = message.userId;
+				handlers.onAuthenticated?.(message);
+			},
+			onOrderEvent: (message) => {
+				if (canReceiveUserEvent(message.userId)) handlers.onOrderEvent?.(message);
+			},
+			onOrderEventReceipt: (message) => {
+				if (canReceiveUserEvent()) handlers.onOrderEventReceipt?.(message);
+			},
+			onOrderSubmitted: (message) => {
+				if (canReceiveUserEvent()) handlers.onOrderSubmitted?.(message);
+			},
+			onBalanceUpdate: (message) => {
+				if (canReceiveUserEvent()) handlers.onBalanceUpdate?.(message);
+			},
+			onRedeemEvent: (message) => {
+				if (canReceiveUserEvent()) handlers.onRedeemEvent?.(message);
+			},
+			onWithdrawalLifecycle: (message) => {
+				if (canReceiveUserEvent(message.userId)) handlers.onWithdrawalLifecycle?.(message);
+			}
+		});
+		socket.setCallbacks = (handlers) => {
+			currentCallbacks = handlers;
+			originalSetCallbacks.call(socket, wrapCallbacks(handlers));
+		};
+		socket.setCallbacks(currentCallbacks);
+		socket.connect = () => {
+			connectRequested = true;
+			manuallyDisconnected = false;
+			originalConnect.call(socket);
+		};
+		socket.disconnect = () => {
+			connectRequested = false;
+			manuallyDisconnected = true;
+			originalDisconnect.call(socket);
+		};
+		const unsubscribe = this.onAuthStateChange(() => {
+			const nextToken = this.accessToken;
+			const nextUserId = this.accessTokenUserId;
+			if (token === nextToken && userId === nextUserId) return;
+			const resetConnection = !nextToken || !!token && (userId !== nextUserId || nextUserId === void 0);
+			token = nextToken;
+			userId = nextUserId;
+			if (resetConnection) {
+				const diagnostics = socket.getDiagnostics();
+				const reconnect = connectRequested || socket.connected || !manuallyDisconnected && (diagnostics.orderbookOutcomeIds.length > 0 || diagnostics.tradeOutcomeIds.length > 0 || diagnostics.arbMarketIds.length > 0 || diagnostics.arbFeedSubscribers > 0);
+				authenticatedUserId = void 0;
+				awaitingDisconnect = true;
+				originalDisconnect.call(socket);
+				socket.authenticate(nextToken ?? "");
+				if (reconnect) originalConnect.call(socket);
+			} else if (nextToken) socket.authenticate(nextToken);
+		});
+		const cleanup = () => {
+			unsubscribe();
+			this.webSocketAuthCleanups.delete(cleanup);
+			socket.setCallbacks = originalSetCallbacks;
+			socket.connect = originalConnect;
+			socket.disconnect = originalDisconnect;
+			socket.destroy = originalDestroy;
+			socket.setCallbacks(currentCallbacks);
+		};
+		this.webSocketAuthCleanups.add(cleanup);
+		socket.destroy = () => {
+			cleanup();
+			originalDestroy.call(socket);
+		};
 	}
 	/** Execute a previously quoted managed trade. Returns pending order IDs. */
 	async executeManaged(params) {