@getpaseo/server 0.1.15 → 0.1.17

package/dist/server/server/websocket-server.js

@@ -4,7 +4,7 @@ import { hostname as getHostname } from "node:os";
 import { WSInboundMessageSchema, wrapSessionMessage, } from "./messages.js";
 import { asUint8Array, decodeBinaryMuxFrame, encodeBinaryMuxFrame, } from "../shared/binary-mux.js";
 import { isHostAllowed } from "./allowed-hosts.js";
-import { Session } from "./session.js";
+import { Session, } from "./session.js";
 import { PushTokenStore } from "./push/token-store.js";
 import { PushService } from "./push/push-service.js";
 import { computeShouldNotifyClient, computeShouldSendPush, } from "./agent-attention-policy.js";
@@ -68,6 +68,12 @@ function bufferFromWsData(data) {
     return Buffer.from(data);
 }
 const EXTERNAL_SESSION_DISCONNECT_GRACE_MS = 90000;
+const HELLO_TIMEOUT_MS = 15000;
+const WS_CLOSE_HELLO_TIMEOUT = 4001;
+const WS_CLOSE_INVALID_HELLO = 4002;
+const WS_CLOSE_INCOMPATIBLE_PROTOCOL = 4003;
+const WS_PROTOCOL_VERSION = 1;
+const WS_RUNTIME_METRICS_FLUSH_MS = 30000;
 export class MissingDaemonVersionError extends Error {
     constructor() {
         super("VoiceAssistantWebSocketServer requires a non-empty daemonVersion.");
@@ -78,12 +84,33 @@ export class MissingDaemonVersionError extends Error {
  * WebSocket server that only accepts sockets + parses/forwards messages to the session layer.
  */
 export class VoiceAssistantWebSocketServer {
-    constructor(server, logger, serverId, agentManager, agentStorage, downloadTokenStore, paseoHome, createAgentMcpTransport, wsConfig, speech, terminalManager, voice, dictation, agentProviderRuntimeSettings, daemonVersion) {
+    constructor(server, logger, serverId, agentManager, agentStorage, downloadTokenStore, paseoHome, createAgentMcpTransport, wsConfig, speech, terminalManager, voice, dictation, agentProviderRuntimeSettings, daemonVersion, onLifecycleIntent) {
+        this.pendingConnections = new Map();
         this.sessions = new Map();
         this.externalSessionsByKey = new Map();
-        this.clientIdCounter = 0;
         this.voiceSpeakHandlers = new Map();
         this.voiceCallerContexts = new Map();
+        this.runtimeWindowStartedAt = Date.now();
+        this.runtimeCounters = {
+            connectedAwaitingHello: 0,
+            helloResumed: 0,
+            helloNew: 0,
+            pendingDisconnected: 0,
+            sessionDisconnectedWaitingReconnect: 0,
+            sessionSocketDisconnectedAttached: 0,
+            sessionCleanup: 0,
+            validationFailed: 0,
+            binaryBeforeHelloRejected: 0,
+            pendingMessageRejectedBeforeHello: 0,
+            missingConnectionForMessage: 0,
+            unexpectedHelloOnActiveConnection: 0,
+            relayExternalSocketAttached: 0,
+            originRejected: 0,
+            hostRejected: 0,
+        };
+        this.inboundMessageCounts = new Map();
+        this.inboundSessionRequestCounts = new Map();
+        this.runtimeMetricsInterval = null;
         this.ACTIVITY_THRESHOLD_MS = 120000;
         this.logger = logger.child({ module: "websocket-server" });
         this.serverId = serverId;
@@ -102,6 +129,7 @@ export class VoiceAssistantWebSocketServer {
         this.voice = voice ?? null;
         this.dictation = dictation ?? null;
         this.agentProviderRuntimeSettings = agentProviderRuntimeSettings;
+        this.onLifecycleIntent = onLifecycleIntent ?? null;
         this.serverCapabilities = buildServerCapabilities({
             readiness: this.dictation?.getSpeechReadiness?.() ?? null,
         });
@@ -120,6 +148,7 @@ export class VoiceAssistantWebSocketServer {
                 const origin = requestMetadata.origin;
                 const requestHost = requestMetadata.host ?? null;
                 if (requestHost && !isHostAllowed(requestHost, allowedHosts)) {
+                    this.incrementRuntimeCounter("hostRejected");
                     this.logger.warn({ ...requestMetadata, host: requestHost }, "Rejected connection from disallowed host");
                     callback(false, 403, "Host not allowed");
                     return;
@@ -131,6 +160,7 @@ export class VoiceAssistantWebSocketServer {
                     callback(true);
                 }
                 else {
+                    this.incrementRuntimeCounter("originRejected");
                     this.logger.warn({ ...requestMetadata, origin }, "Rejected connection from origin");
                     callback(false, 403, "Origin not allowed");
                 }
@@ -139,6 +169,11 @@ export class VoiceAssistantWebSocketServer {
         this.wss.on("connection", (ws, request) => {
             void this.attachSocket(ws, request);
         });
+        const runtimeMetricsInterval = setInterval(() => {
+            this.flushRuntimeMetrics();
+        }, WS_RUNTIME_METRICS_FLUSH_MS);
+        this.runtimeMetricsInterval = runtimeMetricsInterval;
+        runtimeMetricsInterval.unref?.();
         this.logger.info("WebSocket server initialized on /ws");
     }
     broadcast(message) {
@@ -159,26 +194,52 @@ export class VoiceAssistantWebSocketServer {
             return;
         }
         this.serverCapabilities = next;
-        this.broadcastServerInfo();
+        this.broadcastCapabilitiesUpdate();
     }
     async attachExternalSocket(ws, metadata) {
+        if (metadata?.transport === "relay") {
+            this.incrementRuntimeCounter("relayExternalSocketAttached");
+        }
         await this.attachSocket(ws, undefined, metadata);
     }
     async close() {
+        if (this.runtimeMetricsInterval) {
+            clearInterval(this.runtimeMetricsInterval);
+            this.runtimeMetricsInterval = null;
+        }
+        this.flushRuntimeMetrics({ final: true });
         const uniqueConnections = new Set([
             ...this.sessions.values(),
             ...this.externalSessionsByKey.values(),
         ]);
+        const pendingSockets = new Set(this.pendingConnections.keys());
+        for (const pending of this.pendingConnections.values()) {
+            if (pending.helloTimeout) {
+                clearTimeout(pending.helloTimeout);
+                pending.helloTimeout = null;
+            }
+        }
         const cleanupPromises = [];
         for (const connection of uniqueConnections) {
             if (connection.externalDisconnectCleanupTimeout) {
                 clearTimeout(connection.externalDisconnectCleanupTimeout);
                 connection.externalDisconnectCleanupTimeout = null;
             }
-            const ws = connection.socketRef.current;
             cleanupPromises.push(connection.session.cleanup());
+            for (const ws of connection.sockets) {
+                cleanupPromises.push(new Promise((resolve) => {
+                    // WebSocket.CLOSED = 3
+                    if (ws.readyState === 3) {
+                        resolve();
+                        return;
+                    }
+                    ws.once("close", () => resolve());
+                    ws.close();
+                }));
+            }
+        }
+        for (const ws of pendingSockets) {
             cleanupPromises.push(new Promise((resolve) => {
-                // WebSocket.CLOSED = 3
                 if (ws.readyState === 3) {
                     resolve();
                     return;
@@ -188,6 +249,7 @@ export class VoiceAssistantWebSocketServer {
             }));
         }
         await Promise.all(cleanupPromises);
+        this.pendingConnections.clear();
         this.sessions.clear();
         this.externalSessionsByKey.clear();
         this.wss.close();
@@ -204,38 +266,20 @@ export class VoiceAssistantWebSocketServer {
         }
         ws.send(encodeBinaryMuxFrame(frame));
     }
-    async attachSocket(ws, request, metadata) {
-        const externalSessionKey = metadata?.transport === "relay" && metadata.externalSessionKey.trim().length > 0
-            ? metadata.externalSessionKey
-            : null;
-        if (externalSessionKey) {
-            const existing = this.externalSessionsByKey.get(externalSessionKey);
-            if (existing) {
-                if (existing.externalDisconnectCleanupTimeout) {
-                    clearTimeout(existing.externalDisconnectCleanupTimeout);
-                    existing.externalDisconnectCleanupTimeout = null;
-                }
-                const previousSocket = existing.socketRef.current;
-                if (previousSocket !== ws) {
-                    this.sessions.delete(previousSocket);
-                    existing.socketRef.current = ws;
-                }
-                this.sessions.set(ws, existing);
-                this.sendServerInfo(ws);
-                existing.connectionLogger.trace({
-                    clientId: existing.clientId,
-                    externalSessionKey,
-                    totalSessions: this.sessions.size,
-                }, "Client reconnected");
-                this.bindSocketHandlers(ws, existing);
-                return;
-            }
+    sendToConnection(connection, message) {
+        for (const ws of connection.sockets) {
+            this.sendToClient(ws, message);
+        }
+    }
+    sendBinaryToConnection(connection, frame) {
+        for (const ws of connection.sockets) {
+            this.sendBinaryToClient(ws, frame);
         }
-        const clientId = `client-${++this.clientIdCounter}`;
+    }
+    async attachSocket(ws, request, metadata) {
         const requestMetadata = extractSocketRequestMetadata(request);
         const connectionLoggerFields = {
-            clientId,
-            transport: externalSessionKey ? "relay" : "direct",
+            transport: metadata?.transport === "relay" ? "relay" : "direct",
         };
         if (requestMetadata.host) {
             connectionLoggerFields.host = requestMetadata.host;
@@ -250,14 +294,52 @@ export class VoiceAssistantWebSocketServer {
             connectionLoggerFields.remoteAddress = requestMetadata.remoteAddress;
         }
         const connectionLogger = this.logger.child(connectionLoggerFields);
-        const socketRef = { current: ws };
+        const pending = {
+            connectionLogger,
+            helloTimeout: null,
+        };
+        const timeout = setTimeout(() => {
+            if (this.pendingConnections.get(ws) !== pending) {
+                return;
+            }
+            pending.helloTimeout = null;
+            this.pendingConnections.delete(ws);
+            pending.connectionLogger.warn({ timeoutMs: HELLO_TIMEOUT_MS }, "Closing connection due to missing hello");
+            try {
+                ws.close(WS_CLOSE_HELLO_TIMEOUT, "Hello timeout");
+            }
+            catch {
+                // ignore close errors
+            }
+        }, HELLO_TIMEOUT_MS);
+        pending.helloTimeout = timeout;
+        timeout.unref?.();
+        this.pendingConnections.set(ws, pending);
+        this.incrementRuntimeCounter("connectedAwaitingHello");
+        this.bindSocketHandlers(ws);
+        pending.connectionLogger.trace({
+            totalPendingConnections: this.pendingConnections.size,
+        }, "Client connected; awaiting hello");
+    }
+    createSessionConnection(params) {
+        const { ws, clientId, connectionLogger } = params;
+        let connection = null;
         const session = new Session({
             clientId,
             onMessage: (msg) => {
-                this.sendToClient(socketRef.current, wrapSessionMessage(msg));
+                if (!connection) {
+                    return;
+                }
+                this.sendToConnection(connection, wrapSessionMessage(msg));
             },
             onBinaryMessage: (frame) => {
-                this.sendBinaryToClient(socketRef.current, frame);
+                if (!connection) {
+                    return;
+                }
+                this.sendBinaryToConnection(connection, frame);
+            },
+            onLifecycleIntent: (intent) => {
+                this.onLifecycleIntent?.(intent);
             },
             logger: connectionLogger.child({ module: "session" }),
             downloadTokenStore: this.downloadTokenStore,
@@ -289,21 +371,98 @@ export class VoiceAssistantWebSocketServer {
             dictation: this.dictation ?? undefined,
             agentProviderRuntimeSettings: this.agentProviderRuntimeSettings,
         });
-        const connection = {
+        connection = {
             session,
             clientId,
             connectionLogger,
-            socketRef,
-            externalSessionKey,
+            sockets: new Set([ws]),
             externalDisconnectCleanupTimeout: null,
         };
-        this.sessions.set(ws, connection);
-        if (externalSessionKey) {
-            this.externalSessionsByKey.set(externalSessionKey, connection);
+        return connection;
+    }
+    clearPendingConnection(ws) {
+        const pending = this.pendingConnections.get(ws);
+        if (!pending) {
+            return null;
+        }
+        if (pending.helloTimeout) {
+            clearTimeout(pending.helloTimeout);
+            pending.helloTimeout = null;
+        }
+        this.pendingConnections.delete(ws);
+        return pending;
+    }
+    buildWelcomeMessage(params) {
+        return {
+            type: "welcome",
+            serverId: this.serverId,
+            hostname: getHostname(),
+            version: this.daemonVersion,
+            resumed: params.resumed,
+            ...(this.serverCapabilities ? { capabilities: this.serverCapabilities } : {}),
+        };
+    }
+    handleHello(params) {
+        const { ws, message, pending } = params;
+        if (message.protocolVersion !== WS_PROTOCOL_VERSION) {
+            this.clearPendingConnection(ws);
+            pending.connectionLogger.warn({
+                receivedProtocolVersion: message.protocolVersion,
+                expectedProtocolVersion: WS_PROTOCOL_VERSION,
+            }, "Rejected hello due to protocol version mismatch");
+            try {
+                ws.close(WS_CLOSE_INCOMPATIBLE_PROTOCOL, "Incompatible protocol version");
+            }
+            catch {
+                // ignore close errors
+            }
+            return;
+        }
+        const clientId = message.clientId.trim();
+        if (clientId.length === 0) {
+            this.clearPendingConnection(ws);
+            pending.connectionLogger.warn("Rejected hello with empty clientId");
+            try {
+                ws.close(WS_CLOSE_INVALID_HELLO, "Invalid hello");
+            }
+            catch {
+                // ignore close errors
+            }
+            return;
+        }
+        this.clearPendingConnection(ws);
+        const existing = this.externalSessionsByKey.get(clientId);
+        if (existing) {
+            this.incrementRuntimeCounter("helloResumed");
+            if (existing.externalDisconnectCleanupTimeout) {
+                clearTimeout(existing.externalDisconnectCleanupTimeout);
+                existing.externalDisconnectCleanupTimeout = null;
+            }
+            existing.sockets.add(ws);
+            this.sessions.set(ws, existing);
+            this.sendToClient(ws, this.buildWelcomeMessage({ resumed: true }));
+            existing.connectionLogger.trace({
+                clientId,
+                resumed: true,
+                totalSessions: this.sessions.size,
+            }, "Client connected via hello");
+            return;
         }
-        this.sendServerInfo(ws);
-        connectionLogger.trace({ clientId, externalSessionKey, totalSessions: this.sessions.size }, "Client connected");
-        this.bindSocketHandlers(ws, connection);
+        const connectionLogger = pending.connectionLogger.child({ clientId });
+        this.incrementRuntimeCounter("helloNew");
+        const connection = this.createSessionConnection({
+            ws,
+            clientId,
+            connectionLogger,
+        });
+        this.sessions.set(ws, connection);
+        this.externalSessionsByKey.set(clientId, connection);
+        this.sendToClient(ws, this.buildWelcomeMessage({ resumed: false }));
+        connection.connectionLogger.trace({
+            clientId,
+            resumed: false,
+            totalSessions: this.sessions.size,
+        }, "Client connected via hello");
     }
     buildServerInfoStatusPayload() {
         return {
@@ -314,33 +473,29 @@ export class VoiceAssistantWebSocketServer {
             ...(this.serverCapabilities ? { capabilities: this.serverCapabilities } : {}),
         };
     }
-    broadcastServerInfo() {
+    broadcastCapabilitiesUpdate() {
         this.broadcast(wrapSessionMessage({
             type: "status",
             payload: this.buildServerInfoStatusPayload(),
         }));
     }
-    sendServerInfo(ws) {
-        // Advertise stable server identity immediately on connect (used for URL/shareable IDs).
-        this.sendToClient(ws, wrapSessionMessage({
-            type: "status",
-            payload: this.buildServerInfoStatusPayload(),
-        }));
-    }
-    bindSocketHandlers(ws, connection) {
+    bindSocketHandlers(ws) {
         ws.on("message", (data) => {
             void this.handleRawMessage(ws, data);
         });
         ws.on("close", async (code, reason) => {
-            await this.detachSocket(ws, connection, {
+            await this.detachSocket(ws, {
                 code: typeof code === "number" ? code : undefined,
                 reason,
             });
         });
         ws.on("error", async (error) => {
             const err = error instanceof Error ? error : new Error(String(error));
-            connection.connectionLogger.error({ err }, "Client error");
-            await this.detachSocket(ws, connection, { error: err });
+            const active = this.sessions.get(ws);
+            const pending = this.pendingConnections.get(ws);
+            const log = active?.connectionLogger ?? pending?.connectionLogger ?? this.logger;
+            log.error({ err }, "Client error");
+            await this.detachSocket(ws, { error: err });
         });
     }
     resolveVoiceSpeakHandler(callerAgentId) {
@@ -349,13 +504,24 @@ export class VoiceAssistantWebSocketServer {
     resolveVoiceCallerContext(callerAgentId) {
         return this.voiceCallerContexts.get(callerAgentId) ?? null;
     }
-    async detachSocket(ws, connection, details) {
-        const activeConnection = this.sessions.get(ws);
-        if (activeConnection !== connection)
+    async detachSocket(ws, details) {
+        const pending = this.clearPendingConnection(ws);
+        if (pending) {
+            this.incrementRuntimeCounter("pendingDisconnected");
+            pending.connectionLogger.trace({
+                code: details.code,
+                reason: stringifyCloseReason(details.reason),
+            }, "Pending client disconnected");
             return;
+        }
+        const connection = this.sessions.get(ws);
+        if (!connection) {
+            return;
+        }
         this.sessions.delete(ws);
-        if (connection.externalSessionKey &&
-            connection.socketRef.current === ws) {
+        connection.sockets.delete(ws);
+        if (connection.sockets.size === 0) {
+            this.incrementRuntimeCounter("sessionDisconnectedWaitingReconnect");
             if (connection.externalDisconnectCleanupTimeout) {
                 clearTimeout(connection.externalDisconnectCleanupTimeout);
             }
@@ -369,41 +535,61 @@ export class VoiceAssistantWebSocketServer {
             connection.externalDisconnectCleanupTimeout = timeout;
             connection.connectionLogger.trace({
                 clientId: connection.clientId,
-                externalSessionKey: connection.externalSessionKey,
                 code: details.code,
                 reason: stringifyCloseReason(details.reason),
                 reconnectGraceMs: EXTERNAL_SESSION_DISCONNECT_GRACE_MS,
             }, "Client disconnected; waiting for reconnect");
             return;
         }
+        if (connection.sockets.size > 0) {
+            this.incrementRuntimeCounter("sessionSocketDisconnectedAttached");
+            connection.connectionLogger.trace({
+                clientId: connection.clientId,
+                remainingSockets: connection.sockets.size,
+                code: details.code,
+                reason: stringifyCloseReason(details.reason),
+            }, "Client socket disconnected; session remains attached");
+            return;
+        }
         await this.cleanupConnection(connection, "Client disconnected");
     }
     async cleanupConnection(connection, logMessage) {
+        this.incrementRuntimeCounter("sessionCleanup");
         if (connection.externalDisconnectCleanupTimeout) {
             clearTimeout(connection.externalDisconnectCleanupTimeout);
             connection.externalDisconnectCleanupTimeout = null;
         }
-        const currentSocket = connection.socketRef.current;
-        this.sessions.delete(currentSocket);
-        if (connection.externalSessionKey) {
-            const existing = this.externalSessionsByKey.get(connection.externalSessionKey);
-            if (existing === connection) {
-                this.externalSessionsByKey.delete(connection.externalSessionKey);
-            }
+        for (const socket of connection.sockets) {
+            this.sessions.delete(socket);
+        }
+        connection.sockets.clear();
+        const existing = this.externalSessionsByKey.get(connection.clientId);
+        if (existing === connection) {
+            this.externalSessionsByKey.delete(connection.clientId);
         }
         connection.connectionLogger.trace({ clientId: connection.clientId, totalSessions: this.sessions.size }, logMessage);
         await connection.session.cleanup();
     }
     async handleRawMessage(ws, data) {
+        const activeConnection = this.sessions.get(ws);
+        const pendingConnection = this.pendingConnections.get(ws);
+        const log = activeConnection?.connectionLogger ?? pendingConnection?.connectionLogger ?? this.logger;
         try {
-            const activeConnection = this.sessions.get(ws);
             const buffer = bufferFromWsData(data);
             const asBytes = asUint8Array(buffer);
             if (asBytes) {
                 const frame = decodeBinaryMuxFrame(asBytes);
                 if (frame) {
                     if (!activeConnection) {
-                        this.logger.error("No session found for client");
+                        this.incrementRuntimeCounter("binaryBeforeHelloRejected");
+                        log.warn("Rejected binary frame before hello");
+                        this.clearPendingConnection(ws);
+                        try {
+                            ws.close(WS_CLOSE_INVALID_HELLO, "Session message before hello");
+                        }
+                        catch {
+                            // ignore close errors
+                        }
                         return;
                     }
                     activeConnection.session.handleBinaryFrame(frame);
@@ -413,13 +599,26 @@ export class VoiceAssistantWebSocketServer {
             const parsed = JSON.parse(buffer.toString());
             const parsedMessage = WSInboundMessageSchema.safeParse(parsed);
             if (!parsedMessage.success) {
+                this.incrementRuntimeCounter("validationFailed");
+                if (pendingConnection) {
+                    pendingConnection.connectionLogger.warn({
+                        error: parsedMessage.error.message,
+                    }, "Rejected pending message before hello");
+                    this.clearPendingConnection(ws);
+                    try {
+                        ws.close(WS_CLOSE_INVALID_HELLO, "Invalid hello");
+                    }
+                    catch {
+                        // ignore close errors
+                    }
+                    return;
+                }
                 const requestInfo = extractRequestInfoFromUnknownWsInbound(parsed);
                 const isUnknownSchema = requestInfo?.requestId != null &&
                     typeof parsed === "object" &&
                     parsed != null &&
                     "type" in parsed &&
                     parsed.type === "session";
-                const log = activeConnection?.connectionLogger ?? this.logger;
                 log.warn({
                     clientId: activeConnection?.clientId,
                     requestId: requestInfo?.requestId,
@@ -449,6 +648,7 @@ export class VoiceAssistantWebSocketServer {
                 return;
             }
             const message = parsedMessage.data;
+            this.recordInboundMessageType(message.type);
             if (message.type === "ping") {
                 this.sendToClient(ws, { type: "pong" });
                 return;
@@ -456,11 +656,46 @@ export class VoiceAssistantWebSocketServer {
             if (message.type === "recording_state") {
                 return;
             }
+            if (pendingConnection) {
+                if (message.type === "hello") {
+                    this.handleHello({
+                        ws,
+                        message,
+                        pending: pendingConnection,
+                    });
+                    return;
+                }
+                pendingConnection.connectionLogger.warn({
+                    messageType: message.type,
+                }, "Rejected pending message before hello");
+                this.incrementRuntimeCounter("pendingMessageRejectedBeforeHello");
+                this.clearPendingConnection(ws);
+                try {
+                    ws.close(WS_CLOSE_INVALID_HELLO, "Session message before hello");
+                }
+                catch {
+                    // ignore close errors
+                }
+                return;
+            }
             if (!activeConnection) {
-                this.logger.error("No session found for client");
+                this.incrementRuntimeCounter("missingConnectionForMessage");
+                this.logger.error("No connection found for websocket");
+                return;
+            }
+            if (message.type === "hello") {
+                this.incrementRuntimeCounter("unexpectedHelloOnActiveConnection");
+                activeConnection.connectionLogger.warn("Received hello on active connection");
+                try {
+                    ws.close(WS_CLOSE_INVALID_HELLO, "Unexpected hello");
+                }
+                catch {
+                    // ignore close errors
+                }
                 return;
             }
             if (message.type === "session") {
+                this.recordInboundSessionRequestType(message.message.type);
                 await activeConnection.session.handleMessage(message.message);
             }
         }
@@ -482,11 +717,21 @@ export class VoiceAssistantWebSocketServer {
             const trimmedRawPayload = typeof rawPayload === "string" && rawPayload.length > 2000
                 ? `${rawPayload.slice(0, 2000)}... (truncated)`
                 : rawPayload;
-            this.logger.error({
+            log.error({
                 err,
                 rawPayload: trimmedRawPayload,
                 parsedPayload,
             }, "Failed to parse/handle message");
+            if (this.pendingConnections.has(ws)) {
+                this.clearPendingConnection(ws);
+                try {
+                    ws.close(WS_CLOSE_INVALID_HELLO, "Invalid hello");
+                }
+                catch {
+                    // ignore close errors
+                }
+                return;
+            }
             const requestInfo = extractRequestInfoFromUnknownWsInbound(parsedPayload);
             if (requestInfo) {
                 this.sendToClient(ws, wrapSessionMessage({
@@ -509,6 +754,84 @@ export class VoiceAssistantWebSocketServer {
             }));
         }
     }
+    incrementRuntimeCounter(counter) {
+        this.runtimeCounters[counter] += 1;
+    }
+    incrementCount(map, key) {
+        map.set(key, (map.get(key) ?? 0) + 1);
+    }
+    recordInboundMessageType(type) {
+        this.incrementCount(this.inboundMessageCounts, type);
+    }
+    recordInboundSessionRequestType(type) {
+        this.incrementCount(this.inboundSessionRequestCounts, type);
+    }
+    getTopCounts(map, limit) {
+        return [...map.entries()].sort((a, b) => b[1] - a[1]).slice(0, limit);
+    }
+    collectSessionRuntimeMetrics() {
+        const uniqueConnections = new Set(this.externalSessionsByKey.values());
+        let checkoutDiffTargetCount = 0;
+        let checkoutDiffSubscriptionCount = 0;
+        let checkoutDiffWatcherCount = 0;
+        let checkoutDiffFallbackRefreshTargetCount = 0;
+        let terminalDirectorySubscriptionCount = 0;
+        let terminalSubscriptionCount = 0;
+        let terminalStreamCount = 0;
+        for (const connection of uniqueConnections) {
+            const sessionMetrics = connection.session.getRuntimeMetrics();
+            checkoutDiffTargetCount += sessionMetrics.checkoutDiffTargetCount;
+            checkoutDiffSubscriptionCount += sessionMetrics.checkoutDiffSubscriptionCount;
+            checkoutDiffWatcherCount += sessionMetrics.checkoutDiffWatcherCount;
+            checkoutDiffFallbackRefreshTargetCount +=
+                sessionMetrics.checkoutDiffFallbackRefreshTargetCount;
+            terminalDirectorySubscriptionCount += sessionMetrics.terminalDirectorySubscriptionCount;
+            terminalSubscriptionCount += sessionMetrics.terminalSubscriptionCount;
+            terminalStreamCount += sessionMetrics.terminalStreamCount;
+        }
+        return {
+            checkoutDiffTargetCount,
+            checkoutDiffSubscriptionCount,
+            checkoutDiffWatcherCount,
+            checkoutDiffFallbackRefreshTargetCount,
+            terminalDirectorySubscriptionCount,
+            terminalSubscriptionCount,
+            terminalStreamCount,
+        };
+    }
+    flushRuntimeMetrics(options) {
+        const now = Date.now();
+        const windowMs = Math.max(0, now - this.runtimeWindowStartedAt);
+        const activeConnections = new Set(this.sessions.values()).size;
+        const activeSockets = this.sessions.size;
+        const pendingConnections = this.pendingConnections.size;
+        const reconnectGraceSessions = [...this.externalSessionsByKey.values()].filter((connection) => connection.sockets.size === 0 &&
+            connection.externalDisconnectCleanupTimeout !== null).length;
+        const sessionMetrics = this.collectSessionRuntimeMetrics();
+        this.logger.info({
+            windowMs,
+            final: Boolean(options?.final),
+            sessions: {
+                activeConnections,
+                externalSessionKeys: this.externalSessionsByKey.size,
+                reconnectGraceSessions,
+            },
+            sockets: {
+                activeSockets,
+                pendingConnections,
+            },
+            counters: { ...this.runtimeCounters },
+            inboundMessageTypesTop: this.getTopCounts(this.inboundMessageCounts, 12),
+            inboundSessionRequestTypesTop: this.getTopCounts(this.inboundSessionRequestCounts, 20),
+            runtime: sessionMetrics,
+        }, "ws_runtime_metrics");
+        for (const counter of Object.keys(this.runtimeCounters)) {
+            this.runtimeCounters[counter] = 0;
+        }
+        this.inboundMessageCounts.clear();
+        this.inboundSessionRequestCounts.clear();
+        this.runtimeWindowStartedAt = now;
+    }
     getClientActivityState(session) {
         const activity = session.getClientActivity();
         if (!activity) {