fix(ui): map sensing websocket port for docker (#572)

ui/services/sensing.service.js

@@ -9,11 +9,25 @@
  * emit simulated frames so the UI can clearly distinguish live vs. fallback data.
  */
 
-// Derive WebSocket URL from the page origin so it works on any port.
-// The /ws/sensing endpoint is available on the same HTTP port (3000).
-const _wsProto = (typeof window !== 'undefined' && window.location.protocol === 'https:') ? 'wss:' : 'ws:';
-const _wsHost  = (typeof window !== 'undefined' && window.location.host) ? window.location.host : 'localhost:3000';
-const SENSING_WS_URL = `${_wsProto}//${_wsHost}/ws/sensing`;
+const SENSING_WS_PORT_BY_HTTP_PORT = {
+  // Docker image: HTTP UI/API on 3000, sensing stream on 3001.
+  '3000': '3001',
+  // Python sensing stack: UI on 8080, sensing stream on 8765.
+  '8080': '8765',
+};
+
+export function buildSensingWsUrl(locationLike = (typeof window !== 'undefined' ? window.location : null)) {
+  const protocol = locationLike && locationLike.protocol === 'https:' ? 'wss:' : 'ws:';
+  const host = locationLike && locationLike.host ? locationLike.host : 'localhost:3001';
+  const hostname = locationLike && locationLike.hostname ? locationLike.hostname : host.split(':')[0];
+  const port = locationLike && locationLike.port ? locationLike.port : '';
+  const wsPort = SENSING_WS_PORT_BY_HTTP_PORT[port];
+  const wsHost = wsPort ? `${hostname}:${wsPort}` : host;
+
+  return `${protocol}//${wsHost}/ws/sensing`;
+}
+
+const SENSING_WS_URL = buildSensingWsUrl();
 const RECONNECT_DELAYS = [1000, 2000, 4000, 8000, 16000];
 const MAX_RECONNECT_ATTEMPTS = 20;
 // Number of failed attempts that must occur before simulation starts.

ui/services/websocket.service.js

@@ -136,9 +136,22 @@ export class WebSocketService {
 
   // Set up WebSocket event handlers
   setupEventHandlers(url, ws, handlers) {
-    const connection = this.connections.get(url);
+    const getConnection = (eventName) => {
+      const connection = this.connections.get(url);
+      if (!connection) {
+        this.logger.warn(`Ignoring WebSocket ${eventName} for unregistered connection`, {
+          url,
+          readyState: ws.readyState
+        });
+        return null;
+      }
+      return connection;
+    };
 
     ws.onopen = (event) => {
+      const connection = getConnection('open');
+      if (!connection) return;
+
       const connectionTime = Date.now() - connection.connectionStartTime;
       this.logger.info(`WebSocket connected successfully`, { url, connectionTime });
       
@@ -158,6 +171,9 @@ export class WebSocketService {
     };
 
     ws.onmessage = (event) => {
+      const connection = getConnection('message');
+      if (!connection) return;
+
       connection.lastActivity = Date.now();
       connection.messageCount++;
       
@@ -188,6 +204,9 @@ export class WebSocketService {
     };
 
     ws.onerror = (event) => {
+      const connection = getConnection('error');
+      if (!connection) return;
+
       connection.errorCount++;
       this.logger.error(`WebSocket error occurred`, { 
         url, 
@@ -208,6 +227,9 @@ export class WebSocketService {
     };
 
     ws.onclose = (event) => {
+      const connection = getConnection('close');
+      if (!connection) return;
+
       const { code, reason, wasClean } = event;
       this.logger.info(`WebSocket closed`, { url, code, reason, wasClean });
       
@@ -607,4 +629,4 @@ export class WebSocketService {
 }
 
 // Create singleton instance
-export const wsService = new WebSocketService();
+export const wsService = new WebSocketService();