@temple-digital-group/temple-canton-js 2.0.2 → 2.0.3-beta.0

package/src/websocket/index.ts

@@ -1,341 +1,341 @@
-import config from "../../src/config/index.js";
-import NodeWebSocket from "ws";
-
-// Environment-aware WebSocket constructor
-const WS: typeof globalThis.WebSocket =
-	typeof WebSocket !== "undefined" ? WebSocket : (NodeWebSocket as unknown as typeof WebSocket);
-
-// Whether we're in Node.js (can set headers on upgrade) vs browser (must auth via message)
-const isNode = typeof process !== "undefined" && typeof process.versions?.node === "string";
-
-type MessageCallback = (data: unknown) => void;
-
-export type Granularity = 60 | 300 | 900 | 3600 | 14400 | 86400;
-
-/** Normalize CC → Amulet in symbol strings */
-function normalizeSymbol(symbol: string): string {
-	return symbol.replace(/\bCC\b/g, "Amulet");
-}
-
-export class TempleWebSocket {
-	private ws: InstanceType<typeof WS> | null = null;
-	/** Channel subscriptions — require a subscribe message to the server */
-	private subscriptions: Map<string, Set<MessageCallback>> = new Map();
-	/** User event subscriptions — auto-delivered after auth, no subscribe message needed */
-	private userEventSubscribers: Map<string, Set<MessageCallback>> = new Map();
-	private reconnectTimer: ReturnType<typeof setTimeout> | null = null;
-	private pingInterval: ReturnType<typeof setInterval> | null = null;
-	private reconnectAttempts = 0;
-	private maxReconnectDelay = 30_000;
-	private authenticated = false;
-
-	/** Whether to automatically reconnect on close/error */
-	autoReconnect = true;
-
-	/** Event callbacks */
-	onConnect: (() => void) | null = null;
-	onDisconnect: ((code: number, reason: string) => void) | null = null;
-	onError: ((error: unknown) => void) | null = null;
-	onAuth: ((success: boolean, userId?: number) => void) | null = null;
-
-	get connected(): boolean {
-		return this.ws?.readyState === WS.OPEN;
-	}
-
-	/**
-	 * Connect to the Temple WebSocket server.
-	 * Auth is handled automatically via header (Node.js) or message (browser).
-	 */
-	connect(): void {
-		if (this.ws && (this.ws.readyState === WS.OPEN || this.ws.readyState === WS.CONNECTING)) {
-			return;
-		}
-
-		const url = config.WS_URL;
-		const apiKey = config.API_KEY;
-
-		if (isNode && apiKey) {
-			// Node.js: auth via HTTP upgrade header
-			this.ws = new WS(url, { headers: { "X-API-Key": apiKey } } as unknown as string[]);
-		} else {
-			this.ws = new WS(url);
-		}
-
-		this.ws.onopen = () => {
-			this.reconnectAttempts = 0;
-
-			// Browser: auth via message after connect
-			if (!isNode && apiKey) {
-				this.send({ type: "auth", api_key: apiKey });
-			} else {
-				this.authenticated = true;
-			}
-
-			// Re-subscribe to all active channels (market data only, not user events)
-			const channels = [...this.subscriptions.keys()];
-			if (channels.length > 0) {
-				this.send({ type: "subscribe", channels });
-			}
-
-			// Keepalive ping every 30s
-			this.pingInterval = setInterval(() => {
-				if (this.connected) {
-					this.send({ type: "ping" });
-				}
-			}, 30_000);
-
-			this.onConnect?.();
-		};
-
-		this.ws.onmessage = (event: MessageEvent) => {
-			this.handleMessage(typeof event.data === "string" ? event.data : String(event.data));
-		};
-
-		this.ws.onclose = (event: CloseEvent) => {
-			this.cleanup();
-			this.onDisconnect?.(event.code, event.reason);
-			if (this.autoReconnect) {
-				this.scheduleReconnect();
-			}
-		};
-
-		this.ws.onerror = (event: Event) => {
-			this.onError?.(event);
-		};
-	}
-
-	/** Disconnect and stop auto-reconnect. */
-	disconnect(): void {
-		this.autoReconnect = false;
-		this.cleanup();
-		if (this.ws) {
-			this.ws.close(1000, "client disconnect");
-			this.ws = null;
-		}
-	}
-
-	/**
-	 * Subscribe to a market data channel. Sends a subscribe message to the server.
-	 * Returns an unsubscribe function.
-	 *
-	 * @param channel - Channel pattern (e.g. "orderbook:Amulet/USDCx", "trades:Amulet/USDCx")
-	 * @param callback - Called with parsed message data for this channel
-	 * @returns Unsubscribe function
-	 */
-	subscribe(channel: string, callback: MessageCallback): () => void {
-		if (!this.subscriptions.has(channel)) {
-			this.subscriptions.set(channel, new Set());
-			// Send subscribe if already connected
-			if (this.connected) {
-				this.send({ type: "subscribe", channels: [channel] });
-			}
-		}
-		this.subscriptions.get(channel)!.add(callback);
-
-		// Return unsubscribe function
-		return () => this.unsubscribe(channel, callback);
-	}
-
-	/**
-	 * Unsubscribe a callback (or all callbacks) from a market data channel.
-	 */
-	unsubscribe(channel: string, callback?: MessageCallback): void {
-		const callbacks = this.subscriptions.get(channel);
-		if (!callbacks) return;
-
-		if (callback) {
-			callbacks.delete(callback);
-		}
-
-		if (!callback || callbacks.size === 0) {
-			this.subscriptions.delete(channel);
-			if (this.connected) {
-				this.send({ type: "unsubscribe", channels: [channel] });
-			}
-		}
-	}
-
-	/**
-	 * Subscribe to user events (auto-delivered after auth, no subscribe message sent).
-	 * The server pushes these automatically once authenticated.
-	 * Returns an unsubscribe function.
-	 *
-	 * @param event - Event name (e.g. "trade", "order", "balance")
-	 * @param callback - Called with the event data
-	 * @returns Unsubscribe function
-	 */
-	onUserEvent(event: string, callback: MessageCallback): () => void {
-		if (!this.userEventSubscribers.has(event)) {
-			this.userEventSubscribers.set(event, new Set());
-		}
-		this.userEventSubscribers.get(event)!.add(callback);
-
-		return () => {
-			const handlers = this.userEventSubscribers.get(event);
-			if (handlers) {
-				handlers.delete(callback);
-				if (handlers.size === 0) {
-					this.userEventSubscribers.delete(event);
-				}
-			}
-		};
-	}
-
-	private send(msg: Record<string, unknown>): void {
-		if (this.ws && this.ws.readyState === WS.OPEN) {
-			this.ws.send(JSON.stringify(msg));
-		}
-	}
-
-	private handleMessage(raw: string): void {
-		let parsed: Record<string, unknown>;
-		try {
-			parsed = JSON.parse(raw);
-		} catch {
-			return;
-		}
-
-		const type = parsed.type as string | undefined;
-
-		// Handle auth response
-		if (type === "authenticated") {
-			this.authenticated = true;
-			this.onAuth?.(true, parsed.user_id as number | undefined);
-			return;
-		}
-		if (type === "auth_expired") {
-			this.authenticated = false;
-			this.onAuth?.(false);
-			return;
-		}
-
-		// Handle pong
-		if (type === "pong") return;
-
-		// Handle subscribed/unsubscribed confirmations
-		if (type === "subscribed" || type === "unsubscribed") return;
-
-		// Handle server errors
-		if (type === "error") {
-			this.onError?.(parsed);
-			return;
-		}
-
-		// Route user data (auto-pushed after auth, keyed by event name)
-		if (type === "user_data") {
-			const event = parsed.event as string | undefined;
-			if (event && this.userEventSubscribers.has(event)) {
-				const handlers = this.userEventSubscribers.get(event)!;
-				for (const cb of handlers) {
-					try {
-						cb(parsed.data ?? parsed);
-					} catch {
-						// Don't let one bad callback kill the others
-					}
-				}
-			}
-			return;
-		}
-
-		// Route channel data (market data from subscriptions)
-		if (type === "data") {
-			const channel = parsed.channel as string | undefined;
-			if (channel && this.subscriptions.has(channel)) {
-				const callbacks = this.subscriptions.get(channel)!;
-				for (const cb of callbacks) {
-					try {
-						cb(parsed.data ?? parsed);
-					} catch {
-						// Don't let one bad callback kill the others
-					}
-				}
-			}
-			return;
-		}
-	}
-
-	private scheduleReconnect(): void {
-		if (this.reconnectTimer) return;
-		const delay = Math.min(1000 * Math.pow(2, this.reconnectAttempts), this.maxReconnectDelay);
-		this.reconnectAttempts++;
-		this.reconnectTimer = setTimeout(() => {
-			this.reconnectTimer = null;
-			this.connect();
-		}, delay);
-	}
-
-	private cleanup(): void {
-		if (this.pingInterval) {
-			clearInterval(this.pingInterval);
-			this.pingInterval = null;
-		}
-		if (this.reconnectTimer) {
-			clearTimeout(this.reconnectTimer);
-			this.reconnectTimer = null;
-		}
-		this.authenticated = false;
-	}
-}
-
-// ─── Singleton & Convenience API ─────────────────────────────────────────────
-
-let instance: TempleWebSocket | null = null;
-
-/** Get or create the shared WebSocket instance. Auto-connects if not already. */
-export function createWebSocket(): TempleWebSocket {
-	if (!instance) {
-		instance = new TempleWebSocket();
-	}
-	if (!instance.connected) {
-		instance.connect();
-	}
-	return instance;
-}
-
-/** Disconnect and destroy the shared WebSocket instance. */
-export function disconnectWebSocket(): void {
-	if (instance) {
-		instance.disconnect();
-		instance = null;
-	}
-}
-
-// ─── Market Data Subscriptions ───────────────────────────────────────────────
-
-export function subscribeOrderbook(symbol: string, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`orderbook:${normalizeSymbol(symbol)}`, cb);
-}
-
-export function subscribeTrades(symbol: string, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`trades:${normalizeSymbol(symbol)}`, cb);
-}
-
-export function subscribeTicker(symbol: string, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`ticker:${normalizeSymbol(symbol)}`, cb);
-}
-
-export function subscribeCandles(symbol: string, granularity: Granularity, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`candles:${normalizeSymbol(symbol)}:${granularity}`, cb);
-}
-
-export function subscribeOracle(symbol: string, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`oracle:${symbol}`, cb);
-}
-
-export function subscribeOracleVolume(symbol: string, cb: MessageCallback): () => void {
-	return createWebSocket().subscribe(`oracle_volume:${symbol}`, cb);
-}
-
-// ─── User Data Subscriptions (auto-delivered after auth) ─────────────────────
-
-export function subscribeUserOrders(cb: MessageCallback): () => void {
-	return createWebSocket().onUserEvent("user_order", cb);
-}
-
-export function subscribeUserTrades(cb: MessageCallback): () => void {
-	return createWebSocket().onUserEvent("user_trade", cb);
-}
-
-export function subscribeUserBalances(cb: MessageCallback): () => void {
-	return createWebSocket().onUserEvent("user_balance", cb);
-}
+import config from "../../src/config/index.js";
+import NodeWebSocket from "ws";
+
+// Environment-aware WebSocket constructor
+const WS: typeof globalThis.WebSocket =
+	typeof WebSocket !== "undefined" ? WebSocket : (NodeWebSocket as unknown as typeof WebSocket);
+
+// Whether we're in Node.js (can set headers on upgrade) vs browser (must auth via message)
+const isNode = typeof process !== "undefined" && typeof process.versions?.node === "string";
+
+type MessageCallback = (data: unknown) => void;
+
+export type Granularity = 60 | 300 | 900 | 3600 | 14400 | 86400;
+
+/** Normalize CC → Amulet in symbol strings */
+function normalizeSymbol(symbol: string): string {
+	return symbol.replace(/\bCC\b/g, "Amulet");
+}
+
+export class TempleWebSocket {
+	private ws: InstanceType<typeof WS> | null = null;
+	/** Channel subscriptions — require a subscribe message to the server */
+	private subscriptions: Map<string, Set<MessageCallback>> = new Map();
+	/** User event subscriptions — auto-delivered after auth, no subscribe message needed */
+	private userEventSubscribers: Map<string, Set<MessageCallback>> = new Map();
+	private reconnectTimer: ReturnType<typeof setTimeout> | null = null;
+	private pingInterval: ReturnType<typeof setInterval> | null = null;
+	private reconnectAttempts = 0;
+	private maxReconnectDelay = 30_000;
+	private authenticated = false;
+
+	/** Whether to automatically reconnect on close/error */
+	autoReconnect = true;
+
+	/** Event callbacks */
+	onConnect: (() => void) | null = null;
+	onDisconnect: ((code: number, reason: string) => void) | null = null;
+	onError: ((error: unknown) => void) | null = null;
+	onAuth: ((success: boolean, userId?: number) => void) | null = null;
+
+	get connected(): boolean {
+		return this.ws?.readyState === WS.OPEN;
+	}
+
+	/**
+	 * Connect to the Temple WebSocket server.
+	 * Auth is handled automatically via header (Node.js) or message (browser).
+	 */
+	connect(): void {
+		if (this.ws && (this.ws.readyState === WS.OPEN || this.ws.readyState === WS.CONNECTING)) {
+			return;
+		}
+
+		const url = config.WS_URL;
+		const apiKey = config.API_KEY;
+
+		if (isNode && apiKey) {
+			// Node.js: auth via HTTP upgrade header
+			this.ws = new WS(url, { headers: { "X-API-Key": apiKey } } as unknown as string[]);
+		} else {
+			this.ws = new WS(url);
+		}
+
+		this.ws.onopen = () => {
+			this.reconnectAttempts = 0;
+
+			// Browser: auth via message after connect
+			if (!isNode && apiKey) {
+				this.send({ type: "auth", api_key: apiKey });
+			} else {
+				this.authenticated = true;
+			}
+
+			// Re-subscribe to all active channels (market data only, not user events)
+			const channels = [...this.subscriptions.keys()];
+			if (channels.length > 0) {
+				this.send({ type: "subscribe", channels });
+			}
+
+			// Keepalive ping every 30s
+			this.pingInterval = setInterval(() => {
+				if (this.connected) {
+					this.send({ type: "ping" });
+				}
+			}, 30_000);
+
+			this.onConnect?.();
+		};
+
+		this.ws.onmessage = (event: MessageEvent) => {
+			this.handleMessage(typeof event.data === "string" ? event.data : String(event.data));
+		};
+
+		this.ws.onclose = (event: CloseEvent) => {
+			this.cleanup();
+			this.onDisconnect?.(event.code, event.reason);
+			if (this.autoReconnect) {
+				this.scheduleReconnect();
+			}
+		};
+
+		this.ws.onerror = (event: Event) => {
+			this.onError?.(event);
+		};
+	}
+
+	/** Disconnect and stop auto-reconnect. */
+	disconnect(): void {
+		this.autoReconnect = false;
+		this.cleanup();
+		if (this.ws) {
+			this.ws.close(1000, "client disconnect");
+			this.ws = null;
+		}
+	}
+
+	/**
+	 * Subscribe to a market data channel. Sends a subscribe message to the server.
+	 * Returns an unsubscribe function.
+	 *
+	 * @param channel - Channel pattern (e.g. "orderbook:Amulet/USDCx", "trades:Amulet/USDCx")
+	 * @param callback - Called with parsed message data for this channel
+	 * @returns Unsubscribe function
+	 */
+	subscribe(channel: string, callback: MessageCallback): () => void {
+		if (!this.subscriptions.has(channel)) {
+			this.subscriptions.set(channel, new Set());
+			// Send subscribe if already connected
+			if (this.connected) {
+				this.send({ type: "subscribe", channels: [channel] });
+			}
+		}
+		this.subscriptions.get(channel)!.add(callback);
+
+		// Return unsubscribe function
+		return () => this.unsubscribe(channel, callback);
+	}
+
+	/**
+	 * Unsubscribe a callback (or all callbacks) from a market data channel.
+	 */
+	unsubscribe(channel: string, callback?: MessageCallback): void {
+		const callbacks = this.subscriptions.get(channel);
+		if (!callbacks) return;
+
+		if (callback) {
+			callbacks.delete(callback);
+		}
+
+		if (!callback || callbacks.size === 0) {
+			this.subscriptions.delete(channel);
+			if (this.connected) {
+				this.send({ type: "unsubscribe", channels: [channel] });
+			}
+		}
+	}
+
+	/**
+	 * Subscribe to user events (auto-delivered after auth, no subscribe message sent).
+	 * The server pushes these automatically once authenticated.
+	 * Returns an unsubscribe function.
+	 *
+	 * @param event - Event name (e.g. "trade", "order", "balance")
+	 * @param callback - Called with the event data
+	 * @returns Unsubscribe function
+	 */
+	onUserEvent(event: string, callback: MessageCallback): () => void {
+		if (!this.userEventSubscribers.has(event)) {
+			this.userEventSubscribers.set(event, new Set());
+		}
+		this.userEventSubscribers.get(event)!.add(callback);
+
+		return () => {
+			const handlers = this.userEventSubscribers.get(event);
+			if (handlers) {
+				handlers.delete(callback);
+				if (handlers.size === 0) {
+					this.userEventSubscribers.delete(event);
+				}
+			}
+		};
+	}
+
+	private send(msg: Record<string, unknown>): void {
+		if (this.ws && this.ws.readyState === WS.OPEN) {
+			this.ws.send(JSON.stringify(msg));
+		}
+	}
+
+	private handleMessage(raw: string): void {
+		let parsed: Record<string, unknown>;
+		try {
+			parsed = JSON.parse(raw);
+		} catch {
+			return;
+		}
+
+		const type = parsed.type as string | undefined;
+
+		// Handle auth response
+		if (type === "authenticated") {
+			this.authenticated = true;
+			this.onAuth?.(true, parsed.user_id as number | undefined);
+			return;
+		}
+		if (type === "auth_expired") {
+			this.authenticated = false;
+			this.onAuth?.(false);
+			return;
+		}
+
+		// Handle pong
+		if (type === "pong") return;
+
+		// Handle subscribed/unsubscribed confirmations
+		if (type === "subscribed" || type === "unsubscribed") return;
+
+		// Handle server errors
+		if (type === "error") {
+			this.onError?.(parsed);
+			return;
+		}
+
+		// Route user data (auto-pushed after auth, keyed by event name)
+		if (type === "user_data") {
+			const event = parsed.event as string | undefined;
+			if (event && this.userEventSubscribers.has(event)) {
+				const handlers = this.userEventSubscribers.get(event)!;
+				for (const cb of handlers) {
+					try {
+						cb(parsed.data ?? parsed);
+					} catch {
+						// Don't let one bad callback kill the others
+					}
+				}
+			}
+			return;
+		}
+
+		// Route channel data (market data from subscriptions)
+		if (type === "data") {
+			const channel = parsed.channel as string | undefined;
+			if (channel && this.subscriptions.has(channel)) {
+				const callbacks = this.subscriptions.get(channel)!;
+				for (const cb of callbacks) {
+					try {
+						cb(parsed.data ?? parsed);
+					} catch {
+						// Don't let one bad callback kill the others
+					}
+				}
+			}
+			return;
+		}
+	}
+
+	private scheduleReconnect(): void {
+		if (this.reconnectTimer) return;
+		const delay = Math.min(1000 * Math.pow(2, this.reconnectAttempts), this.maxReconnectDelay);
+		this.reconnectAttempts++;
+		this.reconnectTimer = setTimeout(() => {
+			this.reconnectTimer = null;
+			this.connect();
+		}, delay);
+	}
+
+	private cleanup(): void {
+		if (this.pingInterval) {
+			clearInterval(this.pingInterval);
+			this.pingInterval = null;
+		}
+		if (this.reconnectTimer) {
+			clearTimeout(this.reconnectTimer);
+			this.reconnectTimer = null;
+		}
+		this.authenticated = false;
+	}
+}
+
+// ─── Singleton & Convenience API ─────────────────────────────────────────────
+
+let instance: TempleWebSocket | null = null;
+
+/** Get or create the shared WebSocket instance. Auto-connects if not already. */
+export function createWebSocket(): TempleWebSocket {
+	if (!instance) {
+		instance = new TempleWebSocket();
+	}
+	if (!instance.connected) {
+		instance.connect();
+	}
+	return instance;
+}
+
+/** Disconnect and destroy the shared WebSocket instance. */
+export function disconnectWebSocket(): void {
+	if (instance) {
+		instance.disconnect();
+		instance = null;
+	}
+}
+
+// ─── Market Data Subscriptions ───────────────────────────────────────────────
+
+export function subscribeOrderbook(symbol: string, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`orderbook:${normalizeSymbol(symbol)}`, cb);
+}
+
+export function subscribeTrades(symbol: string, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`trades:${normalizeSymbol(symbol)}`, cb);
+}
+
+export function subscribeTicker(symbol: string, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`ticker:${normalizeSymbol(symbol)}`, cb);
+}
+
+export function subscribeCandles(symbol: string, granularity: Granularity, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`candles:${normalizeSymbol(symbol)}:${granularity}`, cb);
+}
+
+export function subscribeOracle(symbol: string, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`oracle:${symbol}`, cb);
+}
+
+export function subscribeOracleVolume(symbol: string, cb: MessageCallback): () => void {
+	return createWebSocket().subscribe(`oracle_volume:${symbol}`, cb);
+}
+
+// ─── User Data Subscriptions (auto-delivered after auth) ─────────────────────
+
+export function subscribeUserOrders(cb: MessageCallback): () => void {
+	return createWebSocket().onUserEvent("user_order", cb);
+}
+
+export function subscribeUserTrades(cb: MessageCallback): () => void {
+	return createWebSocket().onUserEvent("user_trade", cb);
+}
+
+export function subscribeUserBalances(cb: MessageCallback): () => void {
+	return createWebSocket().onUserEvent("user_balance", cb);
+}