import type { WsPush, WsRequest, WsResponse } from "@t3tools/contracts"; type PushListener = (data: unknown) => void; interface PendingRequest { resolve: (result: unknown) => void; reject: (error: Error) => void; timeout: ReturnType<typeof setTimeout>;} const REQUEST_TIMEOUT_MS = 60_000;const RECONNECT_DELAYS_MS = [500, 1_000, 2_000, 4_000, 8_000]; interface WsUrlResolutionInput { explicitUrl?: string; bridgeUrl?: string | null; envUrl?: string; pageUrl: string; pageProtocol: string; pageHostname: string; pagePort: string;} function normalizeUrl(value: string | null | undefined): string | undefined { if (!value) return undefined; const trimmed = value.trim(); return trimmed.length > 0 ? trimmed : undefined;} function defaultWsUrlForPage(protocol: string, hostname: string, port: string): string { const wsProtocol = protocol === "https:" ? "wss" : "ws"; const authority = port.length > 0 ? `${hostname}:${port}` : hostname; return `${wsProtocol}://${authority}`;} export function appendTokenFromPageUrl(wsUrl: string, pageUrl: string): string { try { const page = new URL(pageUrl); const token = page.searchParams.get("token")?.trim(); if (!token) return wsUrl; const resolvedWsUrl = new URL(wsUrl, page.origin); if (!resolvedWsUrl.searchParams.has("token")) { resolvedWsUrl.searchParams.set("token", token); } return resolvedWsUrl.toString(); } catch { return wsUrl; }} export function resolveWsUrl(input: WsUrlResolutionInput): string { const explicitUrl = normalizeUrl(input.explicitUrl); if (explicitUrl) return explicitUrl; const baseUrl = normalizeUrl(input.bridgeUrl) ?? normalizeUrl(input.envUrl) ?? defaultWsUrlForPage(input.pageProtocol, input.pageHostname, input.pagePort); return appendTokenFromPageUrl(baseUrl, input.pageUrl);} export class WsTransport { private ws: WebSocket | null = null; private nextId = 1; private readonly pending = new Map<string, PendingRequest>(); private readonly listeners = new Map<string, Set<PushListener>>(); private reconnectAttempt = 0; private reconnectTimer: ReturnType<typeof setTimeout> | null = null; private disposed = false; private readonly url: string; constructor(url?: string) { const bridgeUrl = window.desktopBridge?.getWsUrl(); // In dev mode, VITE_WS_URL points to the server's WebSocket endpoint. // In production, the page is served by the WS server on the same host:port. const envUrl = import.meta.env.VITE_WS_URL as string | undefined; this.url = resolveWsUrl({ explicitUrl: url, bridgeUrl, envUrl, pageUrl: window.location.href, pageProtocol: window.location.protocol, pageHostname: window.location.hostname, pagePort: window.location.port, }); this.connect(); } async request<T = unknown>(method: string, params?: unknown): Promise<T> { if (typeof method !== "string" || method.length === 0) { throw new Error("Request method is required"); } const id = String(this.nextId++); const message: WsRequest = { id, method, ...(params !== undefined ? { params } : {}) }; return new Promise<T>((resolve, reject) => { const timeout = setTimeout(() => { this.pending.delete(id); reject(new Error(`Request timed out: ${method}`)); }, REQUEST_TIMEOUT_MS); this.pending.set(id, { resolve: resolve as (result: unknown) => void, reject, timeout, }); this.send(message); }); } subscribe(channel: string, listener: PushListener): () => void { let channelListeners = this.listeners.get(channel); if (!channelListeners) { channelListeners = new Set(); this.listeners.set(channel, channelListeners); } channelListeners.add(listener); return () => { channelListeners!.delete(listener); if (channelListeners!.size === 0) { this.listeners.delete(channel); } }; } dispose() { this.disposed = true; if (this.reconnectTimer !== null) { clearTimeout(this.reconnectTimer); this.reconnectTimer = null; } for (const pending of this.pending.values()) { clearTimeout(pending.timeout); pending.reject(new Error("Transport disposed")); } this.pending.clear(); this.ws?.close(); this.ws = null; } private connect() { if (this.disposed) return; const ws = new WebSocket(this.url); ws.addEventListener("open", () => { this.ws = ws; this.reconnectAttempt = 0; }); ws.addEventListener("message", (event) => { this.handleMessage(event.data); }); ws.addEventListener("close", () => { this.ws = null; this.scheduleReconnect(); }); ws.addEventListener("error", () => { // close event will fire after error }); } private handleMessage(raw: unknown) { let parsed: unknown; try { parsed = JSON.parse(String(raw)); } catch { return; } if (!parsed || typeof parsed !== "object") return; const message = parsed as Record<string, unknown>; // Push event if (message.type === "push") { const push = message as unknown as WsPush; const channelListeners = this.listeners.get(push.channel); if (channelListeners) { for (const listener of channelListeners) { try { listener(push.data); } catch { // Swallow listener errors } } } return; } // Response to a request if (typeof message.id === "string") { const response = message as unknown as WsResponse; const pending = this.pending.get(response.id); if (!pending) return; clearTimeout(pending.timeout); this.pending.delete(response.id); if (response.error) { pending.reject(new Error(response.error.message)); } else { pending.resolve(response.result); } } } private send(message: WsRequest) { if (this.ws?.readyState === WebSocket.OPEN) { this.ws.send(JSON.stringify(message)); return; } // If not connected, wait for connection const waitForOpen = () => { const check = setInterval(() => { if (this.disposed) { clearInterval(check); return; } if (this.ws?.readyState === WebSocket.OPEN) { clearInterval(check); this.ws.send(JSON.stringify(message)); } }, 50); // Give up after timeout (the pending request will time out on its own) setTimeout(() => clearInterval(check), REQUEST_TIMEOUT_MS); }; waitForOpen(); } private scheduleReconnect() { if (this.disposed) return; const delay = RECONNECT_DELAYS_MS[Math.min(this.reconnectAttempt, RECONNECT_DELAYS_MS.length - 1)] ?? RECONNECT_DELAYS_MS[0]!; this.reconnectAttempt++; this.reconnectTimer = setTimeout(() => { this.reconnectTimer = null; this.connect(); }, delay); }}
import type { WsPush, WsRequest, WsResponse } from "@t3tools/contracts";
type PushListener = (data: unknown) => void;
interface PendingRequest {
resolve: (result: unknown) => void;
reject: (error: Error) => void;
timeout: ReturnType<typeof setTimeout>;
}
const REQUEST_TIMEOUT_MS = 60_000;
const RECONNECT_DELAYS_MS = [500, 1_000, 2_000, 4_000, 8_000];
interface WsUrlResolutionInput {
explicitUrl?: string;
bridgeUrl?: string | null;
envUrl?: string;
pageUrl: string;
pageProtocol: string;
pageHostname: string;
pagePort: string;
}
function normalizeUrl(value: string | null | undefined): string | undefined {
if (!value) return undefined;
const trimmed = value.trim();
return trimmed.length > 0 ? trimmed : undefined;
}
function defaultWsUrlForPage(protocol: string, hostname: string, port: string): string {
const wsProtocol = protocol === "https:" ? "wss" : "ws";
const authority = port.length > 0 ? `${hostname}:${port}` : hostname;
return `${wsProtocol}://${authority}`;
}
export function appendTokenFromPageUrl(wsUrl: string, pageUrl: string): string {
try {
const page = new URL(pageUrl);
const token = page.searchParams.get("token")?.trim();
if (!token) return wsUrl;
const resolvedWsUrl = new URL(wsUrl, page.origin);
if (!resolvedWsUrl.searchParams.has("token")) {
resolvedWsUrl.searchParams.set("token", token);
}
return resolvedWsUrl.toString();
} catch {
return wsUrl;
}
}
export function resolveWsUrl(input: WsUrlResolutionInput): string {
const explicitUrl = normalizeUrl(input.explicitUrl);
if (explicitUrl) return explicitUrl;
const baseUrl =
normalizeUrl(input.bridgeUrl) ??
normalizeUrl(input.envUrl) ??
defaultWsUrlForPage(input.pageProtocol, input.pageHostname, input.pagePort);
return appendTokenFromPageUrl(baseUrl, input.pageUrl);
}
export class WsTransport {
private ws: WebSocket | null = null;
private nextId = 1;
private readonly pending = new Map<string, PendingRequest>();
private readonly listeners = new Map<string, Set<PushListener>>();
private reconnectAttempt = 0;
private reconnectTimer: ReturnType<typeof setTimeout> | null = null;
private disposed = false;
private readonly url: string;
constructor(url?: string) {
const bridgeUrl = window.desktopBridge?.getWsUrl();
// In dev mode, VITE_WS_URL points to the server's WebSocket endpoint.
// In production, the page is served by the WS server on the same host:port.
const envUrl = import.meta.env.VITE_WS_URL as string | undefined;
this.url = resolveWsUrl({
explicitUrl: url,
bridgeUrl,
envUrl,
pageUrl: window.location.href,
pageProtocol: window.location.protocol,
pageHostname: window.location.hostname,
pagePort: window.location.port,
});
this.connect();
}
async request<T = unknown>(method: string, params?: unknown): Promise<T> {
if (typeof method !== "string" || method.length === 0) {
throw new Error("Request method is required");
}
const id = String(this.nextId++);
const message: WsRequest = { id, method, ...(params !== undefined ? { params } : {}) };
return new Promise<T>((resolve, reject) => {
const timeout = setTimeout(() => {
this.pending.delete(id);
reject(new Error(`Request timed out: ${method}`));
}, REQUEST_TIMEOUT_MS);
this.pending.set(id, {
resolve: resolve as (result: unknown) => void,
reject,
timeout,
});
this.send(message);
});
}
subscribe(channel: string, listener: PushListener): () => void {
let channelListeners = this.listeners.get(channel);
if (!channelListeners) {
channelListeners = new Set();
this.listeners.set(channel, channelListeners);
}
channelListeners.add(listener);
return () => {
channelListeners!.delete(listener);
if (channelListeners!.size === 0) {
this.listeners.delete(channel);
}
};
}
dispose() {
this.disposed = true;
if (this.reconnectTimer !== null) {
clearTimeout(this.reconnectTimer);
this.reconnectTimer = null;
}
for (const pending of this.pending.values()) {
clearTimeout(pending.timeout);
pending.reject(new Error("Transport disposed"));
}
this.pending.clear();
this.ws?.close();
this.ws = null;
}
private connect() {
if (this.disposed) return;
const ws = new WebSocket(this.url);
ws.addEventListener("open", () => {
this.ws = ws;
this.reconnectAttempt = 0;
});
ws.addEventListener("message", (event) => {
this.handleMessage(event.data);
});
ws.addEventListener("close", () => {
this.ws = null;
this.scheduleReconnect();
});
ws.addEventListener("error", () => {
// close event will fire after error
});
}
private handleMessage(raw: unknown) {
let parsed: unknown;
try {
parsed = JSON.parse(String(raw));
} catch {
return;
}
if (!parsed || typeof parsed !== "object") return;
const message = parsed as Record<string, unknown>;
// Push event
if (message.type === "push") {
const push = message as unknown as WsPush;
const channelListeners = this.listeners.get(push.channel);
if (channelListeners) {
for (const listener of channelListeners) {
try {
listener(push.data);
} catch {
// Swallow listener errors
}
}
}
return;
}
// Response to a request
if (typeof message.id === "string") {
const response = message as unknown as WsResponse;
const pending = this.pending.get(response.id);
if (!pending) return;
clearTimeout(pending.timeout);
this.pending.delete(response.id);
if (response.error) {
pending.reject(new Error(response.error.message));
} else {
pending.resolve(response.result);
}
}
}
private send(message: WsRequest) {
if (this.ws?.readyState === WebSocket.OPEN) {
this.ws.send(JSON.stringify(message));
return;
}
// If not connected, wait for connection
const waitForOpen = () => {
const check = setInterval(() => {
if (this.disposed) {
clearInterval(check);
return;
}
if (this.ws?.readyState === WebSocket.OPEN) {
clearInterval(check);
this.ws.send(JSON.stringify(message));
}
}, 50);
// Give up after timeout (the pending request will time out on its own)
setTimeout(() => clearInterval(check), REQUEST_TIMEOUT_MS);
};
waitForOpen();
}
private scheduleReconnect() {
if (this.disposed) return;
const delay =
RECONNECT_DELAYS_MS[Math.min(this.reconnectAttempt, RECONNECT_DELAYS_MS.length - 1)] ??
RECONNECT_DELAYS_MS[0]!;
this.reconnectAttempt++;
this.reconnectTimer = setTimeout(() => {
this.reconnectTimer = null;
this.connect();
}, delay);
}
}