@herdctl/web 0.0.1 → 0.1.1

package/dist/server/ws/fleet-bridge.js

@@ -0,0 +1,184 @@
+/**
+ * Fleet Bridge - Connects FleetManager events to WebSocket broadcasts
+ *
+ * This module subscribes to FleetManager events and broadcasts them to
+ * connected WebSocket clients. It handles event filtering so that high-volume
+ * events like job:output are only sent to subscribed clients.
+ */
+import { createLogger, } from "@herdctl/core";
+const logger = createLogger("web:ws");
+/**
+ * FleetBridge connects FleetManager events to WebSocket clients
+ */
+export class FleetBridge {
+    fleetManager;
+    wsHandler;
+    isStarted = false;
+    // Store bound handlers so we can remove them later
+    handlers = {
+        agentStarted: this.onAgentStarted.bind(this),
+        agentStopped: this.onAgentStopped.bind(this),
+        jobCreated: this.onJobCreated.bind(this),
+        jobOutput: this.onJobOutput.bind(this),
+        jobCompleted: this.onJobCompleted.bind(this),
+        jobFailed: this.onJobFailed.bind(this),
+        jobCancelled: this.onJobCancelled.bind(this),
+        scheduleTriggered: this.onScheduleTriggered.bind(this),
+    };
+    constructor(fleetManager, wsHandler) {
+        this.fleetManager = fleetManager;
+        this.wsHandler = wsHandler;
+    }
+    /**
+     * Start listening to FleetManager events and broadcasting to WebSocket clients
+     */
+    start() {
+        if (this.isStarted) {
+            logger.debug("FleetBridge already started");
+            return;
+        }
+        logger.debug("Starting FleetBridge event listeners");
+        // Subscribe to FleetManager events
+        this.fleetManager.on("agent:started", this.handlers.agentStarted);
+        this.fleetManager.on("agent:stopped", this.handlers.agentStopped);
+        this.fleetManager.on("job:created", this.handlers.jobCreated);
+        this.fleetManager.on("job:output", this.handlers.jobOutput);
+        this.fleetManager.on("job:completed", this.handlers.jobCompleted);
+        this.fleetManager.on("job:failed", this.handlers.jobFailed);
+        this.fleetManager.on("job:cancelled", this.handlers.jobCancelled);
+        this.fleetManager.on("schedule:triggered", this.handlers.scheduleTriggered);
+        this.isStarted = true;
+        logger.info("FleetBridge started - broadcasting events to WebSocket clients");
+    }
+    /**
+     * Stop listening to FleetManager events
+     */
+    stop() {
+        if (!this.isStarted) {
+            logger.debug("FleetBridge not started");
+            return;
+        }
+        logger.debug("Stopping FleetBridge event listeners");
+        // Unsubscribe from FleetManager events
+        this.fleetManager.off("agent:started", this.handlers.agentStarted);
+        this.fleetManager.off("agent:stopped", this.handlers.agentStopped);
+        this.fleetManager.off("job:created", this.handlers.jobCreated);
+        this.fleetManager.off("job:output", this.handlers.jobOutput);
+        this.fleetManager.off("job:completed", this.handlers.jobCompleted);
+        this.fleetManager.off("job:failed", this.handlers.jobFailed);
+        this.fleetManager.off("job:cancelled", this.handlers.jobCancelled);
+        this.fleetManager.off("schedule:triggered", this.handlers.scheduleTriggered);
+        this.isStarted = false;
+        logger.info("FleetBridge stopped");
+    }
+    /**
+     * Check if the bridge is currently running
+     */
+    isRunning() {
+        return this.isStarted;
+    }
+    // ==========================================================================
+    // Event Handlers
+    // ==========================================================================
+    /**
+     * Handle agent:started event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onAgentStarted(payload) {
+        logger.debug(`Broadcasting agent:started for ${payload.agent.name}`);
+        this.wsHandler.broadcast({
+            type: "agent:updated",
+            payload,
+        });
+    }
+    /**
+     * Handle agent:stopped event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onAgentStopped(payload) {
+        logger.debug(`Broadcasting agent:stopped for ${payload.agentName}`);
+        this.wsHandler.broadcast({
+            type: "agent:updated",
+            payload,
+        });
+    }
+    /**
+     * Handle job:created event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onJobCreated(payload) {
+        logger.debug(`Broadcasting job:created for job ${payload.job.id} (agent: ${payload.agentName})`);
+        this.wsHandler.broadcast({
+            type: "job:created",
+            payload,
+        });
+    }
+    /**
+     * Handle job:output event
+     *
+     * Broadcast ONLY to clients subscribed to this agent (high volume event)
+     */
+    onJobOutput(payload) {
+        // Only log at debug level since this can be very high volume
+        const subscriberCount = this.wsHandler.getSubscriberCount(payload.agentName);
+        if (subscriberCount > 0) {
+            logger.debug(`Broadcasting job:output for job ${payload.jobId} to ${subscriberCount} subscribers`);
+            this.wsHandler.broadcastToSubscribers(payload.agentName, {
+                type: "job:output",
+                payload,
+            });
+        }
+    }
+    /**
+     * Handle job:completed event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onJobCompleted(payload) {
+        logger.debug(`Broadcasting job:completed for job ${payload.job.id} (agent: ${payload.agentName})`);
+        this.wsHandler.broadcast({
+            type: "job:completed",
+            payload,
+        });
+    }
+    /**
+     * Handle job:failed event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onJobFailed(payload) {
+        logger.debug(`Broadcasting job:failed for job ${payload.job.id} (agent: ${payload.agentName})`);
+        this.wsHandler.broadcast({
+            type: "job:failed",
+            payload,
+        });
+    }
+    /**
+     * Handle job:cancelled event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onJobCancelled(payload) {
+        logger.debug(`Broadcasting job:cancelled for job ${payload.job.id} (agent: ${payload.agentName})`);
+        this.wsHandler.broadcast({
+            type: "job:cancelled",
+            payload,
+        });
+    }
+    /**
+     * Handle schedule:triggered event
+     *
+     * Broadcast to ALL clients (low volume event)
+     */
+    onScheduleTriggered(payload) {
+        logger.debug(`Broadcasting schedule:triggered for ${payload.agentName}/${payload.scheduleName}`);
+        this.wsHandler.broadcast({
+            type: "schedule:triggered",
+            payload,
+        });
+    }
+}
+//# sourceMappingURL=fleet-bridge.js.map

package/dist/server/ws/fleet-bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fleet-bridge.js","sourceRoot":"","sources":["../../../src/server/ws/fleet-bridge.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EACL,YAAY,GAUb,MAAM,eAAe,CAAC;AAGvB,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;AAEtC;;GAEG;AACH,MAAM,OAAO,WAAW;IACd,YAAY,CAAe;IAC3B,SAAS,CAAmB;IAC5B,SAAS,GAAG,KAAK,CAAC;IAE1B,mDAAmD;IAC3C,QAAQ,GAAG;QACjB,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC;QACxC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;QACtC,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;QACtC,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,iBAAiB,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC;KACvD,CAAC;IAEF,YAAY,YAA0B,EAAE,SAA2B;QACjE,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAErD,mCAAmC;QACnC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC5D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,oBAAoB,EAAE,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;QAE5E,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,gEAAgE,CAAC,CAAC;IAChF,CAAC;IAED;;OAEG;IACH,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;YACxC,OAAO;QACT,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAErD,uCAAuC;QACvC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC7D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC7D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,oBAAoB,EAAE,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;QAE7E,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,6EAA6E;IAC7E,iBAAiB;IACjB,6EAA6E;IAE7E;;;;OAIG;IACK,cAAc,CAAC,OAA4B;QACjD,MAAM,CAAC,KAAK,CAAC,kCAAkC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,eAAe;YACrB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,cAAc,CAAC,OAA4B;QACjD,MAAM,CAAC,KAAK,CAAC,kCAAkC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,eAAe;YACrB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,YAAY,CAAC,OAA0B;QAC7C,MAAM,CAAC,KAAK,CAAC,oCAAoC,OAAO,CAAC,GAAG,CAAC,EAAE,YAAY,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;QACjG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,aAAa;YACnB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,WAAW,CAAC,OAAyB;QAC3C,6DAA6D;QAC7D,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC7E,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,mCAAmC,OAAO,CAAC,KAAK,OAAO,eAAe,cAAc,CAAC,CAAC;YACnG,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,OAAO,CAAC,SAAS,EAAE;gBACvD,IAAI,EAAE,YAAY;gBAClB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,cAAc,CAAC,OAA4B;QACjD,MAAM,CAAC,KAAK,CAAC,sCAAsC,OAAO,CAAC,GAAG,CAAC,EAAE,YAAY,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;QACnG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,eAAe;YACrB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,WAAW,CAAC,OAAyB;QAC3C,MAAM,CAAC,KAAK,CAAC,mCAAmC,OAAO,CAAC,GAAG,CAAC,EAAE,YAAY,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;QAChG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,YAAY;YAClB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,cAAc,CAAC,OAA4B;QACjD,MAAM,CAAC,KAAK,CAAC,sCAAsC,OAAO,CAAC,GAAG,CAAC,EAAE,YAAY,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;QACnG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,eAAe;YACrB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,mBAAmB,CAAC,OAAiC;QAC3D,MAAM,CAAC,KAAK,CAAC,uCAAuC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QACjG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,oBAAoB;YAC1B,OAAO;SACR,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/server/ws/handler.d.ts

@@ -0,0 +1,93 @@
+/**
+ * WebSocket connection handler for @herdctl/web
+ *
+ * Manages connected WebSocket clients, handles incoming messages,
+ * and provides broadcast functions for the fleet bridge.
+ */
+import type { WebSocket } from "ws";
+import { type FleetManager } from "@herdctl/core";
+import { type ServerMessage } from "./types.js";
+import type { WebChatManager } from "../chat/index.js";
+/**
+ * Represents a connected WebSocket client with its subscriptions
+ */
+export interface WebSocketClient {
+    /** The WebSocket connection */
+    socket: WebSocket;
+    /** Set of agent names this client is subscribed to for output events */
+    subscriptions: Set<string>;
+    /** Unique client ID for logging */
+    clientId: string;
+}
+/**
+ * WebSocket handler manages all connected clients and message routing
+ */
+export declare class WebSocketHandler {
+    /** All connected clients */
+    private clients;
+    /** Counter for generating unique client IDs */
+    private clientIdCounter;
+    /** FleetManager for getting initial status */
+    private fleetManager;
+    /** WebChatManager for handling chat messages */
+    private chatManager;
+    constructor(fleetManager: FleetManager);
+    /**
+     * Set the chat manager for handling chat messages
+     *
+     * @param chatManager - WebChatManager instance
+     */
+    setChatManager(chatManager: WebChatManager): void;
+    /**
+     * Handle a new WebSocket connection
+     *
+     * @param socket - The WebSocket connection
+     */
+    handleConnection(socket: WebSocket): Promise<void>;
+    /**
+     * Handle an incoming message from a client
+     */
+    private handleMessage;
+    /**
+     * Process a validated client message
+     */
+    private processClientMessage;
+    /**
+     * Handle a chat:send message from a client
+     */
+    private handleChatSend;
+    /**
+     * Handle client disconnect
+     */
+    private handleDisconnect;
+    /**
+     * Send a message to a specific client
+     */
+    private sendToClient;
+    /**
+     * Broadcast a message to ALL connected clients
+     *
+     * Use this for low-volume events like job:created, job:completed, etc.
+     */
+    broadcast(message: ServerMessage): void;
+    /**
+     * Broadcast a message only to clients subscribed to a specific agent
+     *
+     * Use this for high-volume events like job:output to avoid overwhelming
+     * clients that aren't viewing that agent.
+     */
+    broadcastToSubscribers(agentName: string, message: ServerMessage): void;
+    /**
+     * Get the number of connected clients
+     */
+    getConnectedCount(): number;
+    /**
+     * Get the number of clients subscribed to a specific agent
+     */
+    getSubscriberCount(agentName: string): number;
+    /**
+     * Close all connections (for shutdown)
+     */
+    closeAll(): void;
+}
+//# sourceMappingURL=handler.d.ts.map

package/dist/server/ws/handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../../src/server/ws/handler.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAW,MAAM,IAAI,CAAC;AAC7C,OAAO,EAAgB,KAAK,YAAY,EAAE,MAAM,eAAe,CAAC;AAChE,OAAO,EAIL,KAAK,aAAa,EAEnB,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAIvD;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,+BAA+B;IAC/B,MAAM,EAAE,SAAS,CAAC;IAClB,wEAAwE;IACxE,aAAa,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3B,mCAAmC;IACnC,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,4BAA4B;IAC5B,OAAO,CAAC,OAAO,CAAmC;IAClD,+CAA+C;IAC/C,OAAO,CAAC,eAAe,CAAK;IAC5B,8CAA8C;IAC9C,OAAO,CAAC,YAAY,CAAe;IACnC,gDAAgD;IAChD,OAAO,CAAC,WAAW,CAA+B;gBAEtC,YAAY,EAAE,YAAY;IAItC;;;;OAIG;IACH,cAAc,CAAC,WAAW,EAAE,cAAc,GAAG,IAAI;IAIjD;;;;OAIG;IACG,gBAAgB,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAmCxD;;OAEG;IACH,OAAO,CAAC,aAAa;IA0BrB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA8B5B;;OAEG;YACW,cAAc;IA4E5B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAKxB;;OAEG;IACH,OAAO,CAAC,YAAY;IAUpB;;;;OAIG;IACH,SAAS,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAMvC;;;;;OAKG;IACH,sBAAsB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,GAAG,IAAI;IAQvE;;OAEG;IACH,iBAAiB,IAAI,MAAM;IAI3B;;OAEG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAU7C;;OAEG;IACH,QAAQ,IAAI,IAAI;CAUjB"}

package/dist/server/ws/handler.js

@@ -0,0 +1,270 @@
+/**
+ * WebSocket connection handler for @herdctl/web
+ *
+ * Manages connected WebSocket clients, handles incoming messages,
+ * and provides broadcast functions for the fleet bridge.
+ */
+import { createLogger } from "@herdctl/core";
+import { isClientMessage, } from "./types.js";
+const logger = createLogger("web:ws");
+/**
+ * WebSocket handler manages all connected clients and message routing
+ */
+export class WebSocketHandler {
+    /** All connected clients */
+    clients = new Set();
+    /** Counter for generating unique client IDs */
+    clientIdCounter = 0;
+    /** FleetManager for getting initial status */
+    fleetManager;
+    /** WebChatManager for handling chat messages */
+    chatManager = null;
+    constructor(fleetManager) {
+        this.fleetManager = fleetManager;
+    }
+    /**
+     * Set the chat manager for handling chat messages
+     *
+     * @param chatManager - WebChatManager instance
+     */
+    setChatManager(chatManager) {
+        this.chatManager = chatManager;
+    }
+    /**
+     * Handle a new WebSocket connection
+     *
+     * @param socket - The WebSocket connection
+     */
+    async handleConnection(socket) {
+        const clientId = `client-${++this.clientIdCounter}`;
+        const client = {
+            socket,
+            subscriptions: new Set(),
+            clientId,
+        };
+        this.clients.add(client);
+        logger.info(`WebSocket client connected: ${clientId} (total: ${this.clients.size})`);
+        // Send initial fleet status snapshot
+        try {
+            const status = await this.fleetManager.getFleetStatus();
+            this.sendToClient(client, { type: "fleet:status", payload: status });
+        }
+        catch (error) {
+            logger.warn(`Failed to send initial fleet status to ${clientId}: ${error.message}`);
+        }
+        // Handle incoming messages
+        socket.on("message", (data) => {
+            this.handleMessage(client, data);
+        });
+        // Handle disconnect
+        socket.on("close", () => {
+            this.handleDisconnect(client);
+        });
+        // Handle errors
+        socket.on("error", (error) => {
+            logger.warn(`WebSocket error for ${clientId}: ${error.message}`);
+        });
+    }
+    /**
+     * Handle an incoming message from a client
+     */
+    handleMessage(client, data) {
+        try {
+            // Parse the message
+            let message;
+            if (typeof data === "string") {
+                message = JSON.parse(data);
+            }
+            else if (Buffer.isBuffer(data)) {
+                message = JSON.parse(data.toString("utf-8"));
+            }
+            else {
+                logger.debug(`Received non-string/buffer message from ${client.clientId}`);
+                return;
+            }
+            // Validate the message
+            if (!isClientMessage(message)) {
+                logger.debug(`Invalid message from ${client.clientId}: ${JSON.stringify(message)}`);
+                return;
+            }
+            // Handle based on message type
+            this.processClientMessage(client, message);
+        }
+        catch (error) {
+            logger.debug(`Failed to parse message from ${client.clientId}: ${error.message}`);
+        }
+    }
+    /**
+     * Process a validated client message
+     */
+    processClientMessage(client, message) {
+        switch (message.type) {
+            case "subscribe": {
+                const { agentName } = message.payload;
+                client.subscriptions.add(agentName);
+                logger.debug(`${client.clientId} subscribed to agent: ${agentName}`);
+                break;
+            }
+            case "unsubscribe": {
+                const { agentName } = message.payload;
+                client.subscriptions.delete(agentName);
+                logger.debug(`${client.clientId} unsubscribed from agent: ${agentName}`);
+                break;
+            }
+            case "ping": {
+                this.sendToClient(client, { type: "pong" });
+                break;
+            }
+            case "chat:send": {
+                this.handleChatSend(client, message).catch((error) => {
+                    logger.error(`Error handling chat:send: ${error.message}`);
+                });
+                break;
+            }
+        }
+    }
+    /**
+     * Handle a chat:send message from a client
+     */
+    async handleChatSend(client, message) {
+        const { agentName, sessionId, message: userMessage } = message.payload;
+        if (!this.chatManager) {
+            this.sendToClient(client, {
+                type: "chat:error",
+                payload: {
+                    agentName,
+                    sessionId,
+                    error: "Chat manager not available",
+                },
+            });
+            return;
+        }
+        logger.debug(`${client.clientId} sending chat message`, { agentName, sessionId });
+        let jobId = "";
+        try {
+            const result = await this.chatManager.sendMessage(agentName, sessionId, userMessage, async (chunk) => {
+                // Stream chunks back to the client
+                this.sendToClient(client, {
+                    type: "chat:response",
+                    payload: {
+                        agentName,
+                        sessionId,
+                        jobId,
+                        chunk,
+                    },
+                });
+            });
+            jobId = result.jobId;
+            if (result.success) {
+                // Send completion message
+                this.sendToClient(client, {
+                    type: "chat:complete",
+                    payload: {
+                        agentName,
+                        sessionId,
+                        jobId: result.jobId,
+                    },
+                });
+            }
+            else {
+                // Send error message
+                this.sendToClient(client, {
+                    type: "chat:error",
+                    payload: {
+                        agentName,
+                        sessionId,
+                        error: result.error ?? "Unknown error",
+                    },
+                });
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            logger.error(`Chat send failed`, { agentName, sessionId, error: errorMessage });
+            this.sendToClient(client, {
+                type: "chat:error",
+                payload: {
+                    agentName,
+                    sessionId,
+                    error: errorMessage,
+                },
+            });
+        }
+    }
+    /**
+     * Handle client disconnect
+     */
+    handleDisconnect(client) {
+        this.clients.delete(client);
+        logger.info(`WebSocket client disconnected: ${client.clientId} (remaining: ${this.clients.size})`);
+    }
+    /**
+     * Send a message to a specific client
+     */
+    sendToClient(client, message) {
+        if (client.socket.readyState === client.socket.OPEN) {
+            try {
+                client.socket.send(JSON.stringify(message));
+            }
+            catch (error) {
+                logger.warn(`Failed to send message to ${client.clientId}: ${error.message}`);
+            }
+        }
+    }
+    /**
+     * Broadcast a message to ALL connected clients
+     *
+     * Use this for low-volume events like job:created, job:completed, etc.
+     */
+    broadcast(message) {
+        for (const client of this.clients) {
+            this.sendToClient(client, message);
+        }
+    }
+    /**
+     * Broadcast a message only to clients subscribed to a specific agent
+     *
+     * Use this for high-volume events like job:output to avoid overwhelming
+     * clients that aren't viewing that agent.
+     */
+    broadcastToSubscribers(agentName, message) {
+        for (const client of this.clients) {
+            if (client.subscriptions.has(agentName)) {
+                this.sendToClient(client, message);
+            }
+        }
+    }
+    /**
+     * Get the number of connected clients
+     */
+    getConnectedCount() {
+        return this.clients.size;
+    }
+    /**
+     * Get the number of clients subscribed to a specific agent
+     */
+    getSubscriberCount(agentName) {
+        let count = 0;
+        for (const client of this.clients) {
+            if (client.subscriptions.has(agentName)) {
+                count++;
+            }
+        }
+        return count;
+    }
+    /**
+     * Close all connections (for shutdown)
+     */
+    closeAll() {
+        for (const client of this.clients) {
+            try {
+                client.socket.close(1000, "Server shutting down");
+            }
+            catch {
+                // Ignore errors during shutdown
+            }
+        }
+        this.clients.clear();
+    }
+}
+//# sourceMappingURL=handler.js.map

package/dist/server/ws/handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.js","sourceRoot":"","sources":["../../../src/server/ws/handler.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,YAAY,EAAqB,MAAM,eAAe,CAAC;AAChE,OAAO,EACL,eAAe,GAKhB,MAAM,YAAY,CAAC;AAGpB,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;AActC;;GAEG;AACH,MAAM,OAAO,gBAAgB;IAC3B,4BAA4B;IACpB,OAAO,GAAyB,IAAI,GAAG,EAAE,CAAC;IAClD,+CAA+C;IACvC,eAAe,GAAG,CAAC,CAAC;IAC5B,8CAA8C;IACtC,YAAY,CAAe;IACnC,gDAAgD;IACxC,WAAW,GAA0B,IAAI,CAAC;IAElD,YAAY,YAA0B;QACpC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED;;;;OAIG;IACH,cAAc,CAAC,WAA2B;QACxC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAAiB;QACtC,MAAM,QAAQ,GAAG,UAAU,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC;QACpD,MAAM,MAAM,GAAoB;YAC9B,MAAM;YACN,aAAa,EAAE,IAAI,GAAG,EAAE;YACxB,QAAQ;SACT,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzB,MAAM,CAAC,IAAI,CAAC,+BAA+B,QAAQ,YAAY,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;QAErF,qCAAqC;QACrC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC;YACxD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QACvE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,0CAA0C,QAAQ,KAAM,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;QACjG,CAAC;QAED,2BAA2B;QAC3B,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAa,EAAE,EAAE;YACrC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,oBAAoB;QACpB,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,gBAAgB;QAChB,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;YAClC,MAAM,CAAC,IAAI,CAAC,uBAAuB,QAAQ,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,MAAuB,EAAE,IAAa;QAC1D,IAAI,CAAC;YACH,oBAAoB;YACpB,IAAI,OAAgB,CAAC;YACrB,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7B,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACjC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;YAC/C,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,KAAK,CAAC,2CAA2C,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC3E,OAAO;YACT,CAAC;YAED,uBAAuB;YACvB,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9B,MAAM,CAAC,KAAK,CAAC,wBAAwB,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACpF,OAAO;YACT,CAAC;YAED,+BAA+B;YAC/B,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,gCAAgC,MAAM,CAAC,QAAQ,KAAM,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/F,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,MAAuB,EAAE,OAAsB;QAC1E,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;gBACtC,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACpC,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,QAAQ,yBAAyB,SAAS,EAAE,CAAC,CAAC;gBACrE,MAAM;YACR,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;gBACtC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACvC,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,QAAQ,6BAA6B,SAAS,EAAE,CAAC,CAAC;gBACzE,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC5C,MAAM;YACR,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACnD,MAAM,CAAC,KAAK,CAAC,6BAA8B,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxE,CAAC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,MAAuB,EAAE,OAAwB;QAC5E,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;QAEvE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,YAAY;gBAClB,OAAO,EAAE;oBACP,SAAS;oBACT,SAAS;oBACT,KAAK,EAAE,4BAA4B;iBACpC;aACF,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,QAAQ,uBAAuB,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QAElF,IAAI,KAAK,GAAG,EAAE,CAAC;QAEf,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAC/C,SAAS,EACT,SAAS,EACT,WAAW,EACX,KAAK,EAAE,KAAK,EAAE,EAAE;gBACd,mCAAmC;gBACnC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;oBACxB,IAAI,EAAE,eAAe;oBACrB,OAAO,EAAE;wBACP,SAAS;wBACT,SAAS;wBACT,KAAK;wBACL,KAAK;qBACN;iBACF,CAAC,CAAC;YACL,CAAC,CACF,CAAC;YAEF,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YAErB,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,0BAA0B;gBAC1B,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;oBACxB,IAAI,EAAE,eAAe;oBACrB,OAAO,EAAE;wBACP,SAAS;wBACT,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,KAAK;qBACpB;iBACF,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,qBAAqB;gBACrB,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;oBACxB,IAAI,EAAE,YAAY;oBAClB,OAAO,EAAE;wBACP,SAAS;wBACT,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe;qBACvC;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;YAEhF,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,IAAI,EAAE,YAAY;gBAClB,OAAO,EAAE;oBACP,SAAS;oBACT,SAAS;oBACT,KAAK,EAAE,YAAY;iBACpB;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,MAAuB;QAC9C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,kCAAkC,MAAM,CAAC,QAAQ,gBAAgB,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;IACrG,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAAuB,EAAE,OAAsB;QAClE,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,KAAK,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACpD,IAAI,CAAC;gBACH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;YAC9C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,6BAA6B,MAAM,CAAC,QAAQ,KAAM,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3F,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,OAAsB;QAC9B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,sBAAsB,CAAC,SAAiB,EAAE,OAAsB;QAC9D,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBACxC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,SAAiB;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBACxC,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;YACpD,CAAC;YAAC,MAAM,CAAC;gBACP,gCAAgC;YAClC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF"}

package/dist/server/ws/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * WebSocket module exports for @herdctl/web
+ */
+export * from "./types.js";
+export { isChatSendMessage } from "./types.js";
+export { WebSocketHandler, type WebSocketClient } from "./handler.js";
+export { FleetBridge } from "./fleet-bridge.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/server/ws/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/server/ws/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,KAAK,eAAe,EAAE,MAAM,cAAc,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/server/ws/index.js

@@ -0,0 +1,8 @@
+/**
+ * WebSocket module exports for @herdctl/web
+ */
+export * from "./types.js";
+export { isChatSendMessage } from "./types.js";
+export { WebSocketHandler } from "./handler.js";
+export { FleetBridge } from "./fleet-bridge.js";
+//# sourceMappingURL=index.js.map

package/dist/server/ws/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/server/ws/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAwB,MAAM,cAAc,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}