@openhands/agent-canvas 1.14.0 → 1.16.0

package/dist/hooks/use-websocket.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"use-websocket.cjs","names":[],"sources":["../../src/hooks/use-websocket.ts"],"sourcesContent":["import React from \"react\";\nimport { sendWebSocketAuth } from \"#/utils/websocket-auth\";\n\nexport interface WebSocketHookOptions {\n  queryParams?: Record<string, string | boolean>;\n  sessionApiKey?: string | null;\n  onOpen?: (event: Event) => void;\n  onClose?: (event: CloseEvent) => void;\n  onMessage?: (event: MessageEvent) => void;\n  onError?: (event: Event) => void;\n  reconnect?: {\n    enabled?: boolean;\n    maxAttempts?: number;\n  };\n}\n\nexport const useWebSocket = <T = string>(\n  url: string,\n  options?: WebSocketHookOptions,\n) => {\n  const [isConnected, setIsConnected] = React.useState(false);\n  const [lastMessage, setLastMessage] = React.useState<T | null>(null);\n  const [error, setError] = React.useState<Error | null>(null);\n  const [isReconnecting, setIsReconnecting] = React.useState(false);\n  const wsRef = React.useRef<WebSocket | null>(null);\n  const attemptCountRef = React.useRef(0);\n  const reconnectTimeoutRef = React.useRef<NodeJS.Timeout | null>(null);\n  const shouldReconnectRef = React.useRef(true); // Only set to false by disconnect()\n  // Track which WebSocket instances are allowed to reconnect using a WeakSet\n  const allowedToReconnectRef = React.useRef<WeakSet<WebSocket>>(new WeakSet());\n\n  // Store options in a ref to avoid reconnecting when callbacks change\n  const optionsRef = React.useRef(options);\n  React.useEffect(() => {\n    optionsRef.current = options;\n  }, [options]);\n\n  const connectWebSocket = React.useCallback(() => {\n    // Build URL with query parameters if provided\n    let wsUrl = url;\n    if (optionsRef.current?.queryParams) {\n      const stringParams = Object.entries(\n        optionsRef.current.queryParams,\n      ).reduce(\n        (acc, [key, value]) => {\n          acc[key] = String(value);\n          return acc;\n        },\n        {} as Record<string, string>,\n      );\n      const params = new URLSearchParams(stringParams);\n      wsUrl = `${url}?${params.toString()}`;\n    }\n\n    const ws = new WebSocket(wsUrl);\n    wsRef.current = ws;\n    // Mark this WebSocket instance as allowed to reconnect\n    allowedToReconnectRef.current.add(ws);\n\n    ws.onopen = (event) => {\n      sendWebSocketAuth(ws, optionsRef.current?.sessionApiKey);\n      setIsConnected(true);\n      setError(null); // Clear any previous errors\n      setIsReconnecting(false);\n      attemptCountRef.current = 0; // Reset attempt count on successful connection\n      optionsRef.current?.onOpen?.(event);\n    };\n\n    ws.onmessage = (event) => {\n      setLastMessage(event.data);\n      optionsRef.current?.onMessage?.(event);\n    };\n\n    ws.onclose = (event) => {\n      // Check if this specific WebSocket instance is allowed to reconnect\n      const canReconnect = allowedToReconnectRef.current.has(ws);\n      setIsConnected(false);\n      // If the connection closes with an error code, treat it as an error\n      if (event.code !== 1000) {\n        // 1000 is normal closure\n        setError(\n          new Error(\n            `WebSocket closed with code ${event.code}: ${event.reason || \"Connection closed unexpectedly\"}`,\n          ),\n        );\n        // Also call onError handler for error closures (only if allowed to reconnect)\n        if (canReconnect) {\n          optionsRef.current?.onError?.(event);\n        }\n      }\n      optionsRef.current?.onClose?.(event);\n\n      // Attempt reconnection if enabled and allowed\n      // IMPORTANT: Only reconnect if this specific instance is allowed to reconnect\n      const reconnectEnabled = optionsRef.current?.reconnect?.enabled ?? false;\n      const maxAttempts =\n        optionsRef.current?.reconnect?.maxAttempts ?? Infinity;\n\n      if (\n        reconnectEnabled &&\n        canReconnect &&\n        shouldReconnectRef.current &&\n        attemptCountRef.current < maxAttempts\n      ) {\n        setIsReconnecting(true);\n        attemptCountRef.current += 1;\n\n        reconnectTimeoutRef.current = setTimeout(() => {\n          connectWebSocket();\n        }, 3000); // 3 second delay\n      } else {\n        setIsReconnecting(false);\n      }\n    };\n\n    ws.onerror = (event) => {\n      setIsConnected(false);\n      optionsRef.current?.onError?.(event);\n    };\n  }, [url]);\n\n  React.useEffect(() => {\n    // Reset shouldReconnect flag and attempt count when creating a new connection\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n\n    // Only attempt connection if we have a valid URL\n    if (url && url.trim() !== \"\") {\n      connectWebSocket();\n    }\n\n    return () => {\n      // Disable reconnection on unmount to prevent reconnection attempts\n      // This must be set BEFORE closing the socket, so the onclose handler sees it\n      shouldReconnectRef.current = false;\n      // Clear any pending reconnection timeouts\n      if (reconnectTimeoutRef.current) {\n        clearTimeout(reconnectTimeoutRef.current);\n        reconnectTimeoutRef.current = null;\n      }\n      // Close the WebSocket connection\n      if (wsRef.current) {\n        const { readyState } = wsRef.current;\n        // Remove this WebSocket from the allowed list BEFORE closing\n        // so its onclose handler won't try to reconnect\n        allowedToReconnectRef.current.delete(wsRef.current);\n        // Only close if not already closed/closing\n        if (\n          readyState === WebSocket.CONNECTING ||\n          readyState === WebSocket.OPEN\n        ) {\n          wsRef.current.close();\n        }\n        wsRef.current = null;\n      }\n    };\n  }, [url, connectWebSocket]);\n\n  const sendMessage = React.useCallback(\n    (data: string | Blob | BufferSource) => {\n      if (wsRef.current?.readyState === WebSocket.OPEN) {\n        wsRef.current.send(data);\n      }\n    },\n    [],\n  );\n\n  const disconnect = React.useCallback(() => {\n    shouldReconnectRef.current = false;\n    setIsReconnecting(false);\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n    if (wsRef.current) {\n      // Remove from allowed list before closing\n      allowedToReconnectRef.current.delete(wsRef.current);\n      wsRef.current.close();\n    }\n  }, []);\n\n  const reconnect = React.useCallback(() => {\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n    setIsReconnecting(true);\n    setError(null);\n\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n\n    if (wsRef.current) {\n      const ws = wsRef.current;\n      allowedToReconnectRef.current.delete(ws);\n      wsRef.current = null;\n      if (\n        ws.readyState === WebSocket.CONNECTING ||\n        ws.readyState === WebSocket.OPEN\n      ) {\n        ws.close();\n      }\n    }\n\n    connectWebSocket();\n  }, [connectWebSocket]);\n\n  return {\n    isConnected,\n    lastMessage,\n    error,\n    socket: wsRef.current,\n    sendMessage,\n    isReconnecting,\n    attemptCount: attemptCountRef.current,\n    disconnect,\n    reconnect,\n  };\n};\n"],"mappings":"sIAgBA,IAAa,GACX,EACA,IACG,CACH,GAAM,CAAC,EAAa,GAAkB,EAAA,QAAM,SAAS,EAAK,EACpD,CAAC,EAAa,GAAkB,EAAA,QAAM,SAAmB,IAAI,EAC7D,CAAC,EAAO,GAAY,EAAA,QAAM,SAAuB,IAAI,EACrD,CAAC,EAAgB,GAAqB,EAAA,QAAM,SAAS,EAAK,EAC1D,EAAQ,EAAA,QAAM,OAAyB,IAAI,EAC3C,EAAkB,EAAA,QAAM,OAAO,CAAC,EAChC,EAAsB,EAAA,QAAM,OAA8B,IAAI,EAC9D,EAAqB,EAAA,QAAM,OAAO,EAAI,EAEtC,EAAwB,EAAA,QAAM,OAA2B,IAAI,OAAS,EAGtE,EAAa,EAAA,QAAM,OAAO,CAAO,EACvC,EAAA,QAAM,cAAgB,CACpB,EAAW,QAAU,CACvB,EAAG,CAAC,CAAO,CAAC,EAEZ,IAAM,EAAmB,EAAA,QAAM,gBAAkB,CAE/C,IAAI,EAAQ,EACZ,GAAI,EAAW,SAAS,YAAa,CACnC,IAAM,EAAe,OAAO,QAC1B,EAAW,QAAQ,WACrB,EAAE,QACC,EAAK,CAAC,EAAK,MACV,EAAI,GAAO,OAAO,CAAK,EAChB,GAET,CAAC,CACH,EAEA,EAAQ,GAAG,EAAI,GAAG,IADC,gBAAgB,CACjB,EAAO,SAAS,GACpC,CAEA,IAAM,EAAK,IAAI,UAAU,CAAK,EAC9B,EAAM,QAAU,EAEhB,EAAsB,QAAQ,IAAI,CAAE,EAEpC,EAAG,OAAU,GAAU,CACrB,EAAA,kBAAkB,EAAI,EAAW,SAAS,aAAa,EACvD,EAAe,EAAI,EACnB,EAAS,IAAI,EACb,EAAkB,EAAK,EACvB,EAAgB,QAAU,EAC1B,EAAW,SAAS,SAAS,CAAK,CACpC,EAEA,EAAG,UAAa,GAAU,CACxB,EAAe,EAAM,IAAI,EACzB,EAAW,SAAS,YAAY,CAAK,CACvC,EAEA,EAAG,QAAW,GAAU,CAEtB,IAAM,EAAe,EAAsB,QAAQ,IAAI,CAAE,EACzD,EAAe,EAAK,EAEhB,EAAM,OAAS,MAEjB,EACM,MACF,8BAA8B,EAAM,KAAK,IAAI,EAAM,QAAU,kCAC/D,CACF,EAEI,GACF,EAAW,SAAS,UAAU,CAAK,GAGvC,EAAW,SAAS,UAAU,CAAK,EAInC,IAAM,EAAmB,EAAW,SAAS,WAAW,SAAW,GAC7D,EACJ,EAAW,SAAS,WAAW,aAAe,IAG9C,GACA,GACA,EAAmB,SACnB,EAAgB,QAAU,GAE1B,EAAkB,EAAI,EACtB,EAAgB,SAAW,EAE3B,EAAoB,QAAU,eAAiB,CAC7C,EAAiB,CACnB,EAAG,GAAI,GAEP,EAAkB,EAAK,CAE3B,EAEA,EAAG,QAAW,GAAU,CACtB,EAAe,EAAK,EACpB,EAAW,SAAS,UAAU,CAAK,CACrC,CACF,EAAG,CAAC,CAAG,CAAC,EAER,EAAA,QAAM,eAEJ,EAAmB,QAAU,GAC7B,EAAgB,QAAU,EAGtB,GAAO,EAAI,KAAK,IAAM,IACxB,EAAiB,MAGN,CAUX,GAPA,EAAmB,QAAU,GAE7B,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAG5B,EAAM,QAAS,CACjB,GAAM,CAAE,cAAe,EAAM,QAG7B,EAAsB,QAAQ,OAAO,EAAM,OAAO,GAGhD,IAAe,UAAU,YACzB,IAAe,UAAU,OAEzB,EAAM,QAAQ,MAAM,EAEtB,EAAM,QAAU,IAClB,CACF,GACC,CAAC,EAAK,CAAgB,CAAC,EAE1B,IAAM,EAAc,EAAA,QAAM,YACvB,GAAuC,CAClC,EAAM,SAAS,aAAe,UAAU,MAC1C,EAAM,QAAQ,KAAK,CAAI,CAE3B,EACA,CAAC,CACH,EAEM,EAAa,EAAA,QAAM,gBAAkB,CACzC,EAAmB,QAAU,GAC7B,EAAkB,EAAK,EACvB,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAE5B,EAAM,UAER,EAAsB,QAAQ,OAAO,EAAM,OAAO,EAClD,EAAM,QAAQ,MAAM,EAExB,EAAG,CAAC,CAAC,EAEC,EAAY,EAAA,QAAM,gBAAkB,CAWxC,GAVA,EAAmB,QAAU,GAC7B,EAAgB,QAAU,EAC1B,EAAkB,EAAI,EACtB,EAAS,IAAI,EAEb,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAG5B,EAAM,QAAS,CACjB,IAAM,EAAK,EAAM,QACjB,EAAsB,QAAQ,OAAO,CAAE,EACvC,EAAM,QAAU,MAEd,EAAG,aAAe,UAAU,YAC5B,EAAG,aAAe,UAAU,OAE5B,EAAG,MAAM,CAEb,CAEA,EAAiB,CACnB,EAAG,CAAC,CAAgB,CAAC,EAErB,MAAO,CACL,cACA,cACA,QACA,OAAQ,EAAM,QACd,cACA,iBACA,aAAc,EAAgB,QAC9B,aACA,WACF,CACF"}
+{"version":3,"file":"use-websocket.cjs","names":[],"sources":["../../src/hooks/use-websocket.ts"],"sourcesContent":["import React from \"react\";\nimport { sendWebSocketAuth } from \"#/utils/websocket-auth\";\nimport { startHandshakeWatchdog } from \"#/utils/websocket-handshake\";\n\nexport interface WebSocketHookOptions {\n  queryParams?: Record<string, string | boolean>;\n  sessionApiKey?: string | null;\n  onOpen?: (event: Event) => void;\n  onClose?: (event: CloseEvent) => void;\n  onMessage?: (event: MessageEvent) => void;\n  onError?: (event: Event) => void;\n  reconnect?: {\n    enabled?: boolean;\n    maxAttempts?: number;\n  };\n}\n\n// Reconnect backoff bounds: 1s, 2s, 4s, … capped at 30s.\nconst RECONNECT_BASE_DELAY_MS = 1_000;\nconst RECONNECT_MAX_DELAY_MS = 30_000;\n\nexport const useWebSocket = (url: string, options?: WebSocketHookOptions) => {\n  const [isConnected, setIsConnected] = React.useState(false);\n  const [error, setError] = React.useState<Error | null>(null);\n  const [isReconnecting, setIsReconnecting] = React.useState(false);\n  const wsRef = React.useRef<WebSocket | null>(null);\n  const attemptCountRef = React.useRef(0);\n  const reconnectTimeoutRef = React.useRef<NodeJS.Timeout | null>(null);\n  const shouldReconnectRef = React.useRef(true); // Only set to false by disconnect()\n  // Track which WebSocket instances are allowed to reconnect using a WeakSet\n  const allowedToReconnectRef = React.useRef<WeakSet<WebSocket>>(new WeakSet());\n\n  // Store options in a ref to avoid reconnecting when callbacks change\n  const optionsRef = React.useRef(options);\n  React.useEffect(() => {\n    optionsRef.current = options;\n  }, [options]);\n\n  const connectWebSocket = React.useCallback(() => {\n    // Build URL with query parameters if provided\n    let wsUrl = url;\n    if (optionsRef.current?.queryParams) {\n      const stringParams = Object.entries(\n        optionsRef.current.queryParams,\n      ).reduce(\n        (acc, [key, value]) => {\n          acc[key] = String(value);\n          return acc;\n        },\n        {} as Record<string, string>,\n      );\n      const params = new URLSearchParams(stringParams);\n      wsUrl = `${url}?${params.toString()}`;\n    }\n\n    const ws = new WebSocket(wsUrl);\n    wsRef.current = ws;\n    // Mark this WebSocket instance as allowed to reconnect\n    allowedToReconnectRef.current.add(ws);\n\n    // Abort a socket stuck in CONNECTING so it can't hold Chrome's per-host\n    // handshake lock indefinitely; its close flows into the reconnect path\n    // below.\n    const cancelHandshakeWatchdog = startHandshakeWatchdog(ws);\n\n    ws.onopen = (event) => {\n      cancelHandshakeWatchdog();\n      sendWebSocketAuth(ws, optionsRef.current?.sessionApiKey);\n      setIsConnected(true);\n      setError(null); // Clear any previous errors\n      setIsReconnecting(false);\n      attemptCountRef.current = 0; // Reset attempt count on successful connection\n      optionsRef.current?.onOpen?.(event);\n    };\n\n    ws.onmessage = (event) => {\n      // Deliberately no `lastMessage` state here: nothing reads it, and a\n      // React state write per frame re-renders this hook's owner on every\n      // streamed token. Consumers subscribe via `onMessage`.\n      optionsRef.current?.onMessage?.(event);\n    };\n\n    ws.onclose = (event) => {\n      cancelHandshakeWatchdog();\n      // Check if this specific WebSocket instance is allowed to reconnect\n      const canReconnect = allowedToReconnectRef.current.has(ws);\n      setIsConnected(false);\n      // If the connection closes with an error code, treat it as an error\n      if (event.code !== 1000) {\n        // 1000 is normal closure\n        setError(\n          new Error(\n            `WebSocket closed with code ${event.code}: ${event.reason || \"Connection closed unexpectedly\"}`,\n          ),\n        );\n        // Also call onError handler for error closures (only if allowed to reconnect)\n        if (canReconnect) {\n          optionsRef.current?.onError?.(event);\n        }\n      }\n      // Notify the consumer unless this socket was deliberately replaced by a\n      // newer one — a replaced socket's close event arrives late and must not\n      // clobber the replacement's OPEN state in the consumer. Final closes\n      // (disconnect/unmount, nothing replacing the socket) still notify.\n      const wasReplaced = wsRef.current !== null && wsRef.current !== ws;\n      if (!wasReplaced) {\n        optionsRef.current?.onClose?.(event);\n      }\n\n      // Attempt reconnection if enabled and allowed\n      // IMPORTANT: Only reconnect if this specific instance is allowed to reconnect\n      const reconnectEnabled = optionsRef.current?.reconnect?.enabled ?? false;\n      const maxAttempts =\n        optionsRef.current?.reconnect?.maxAttempts ?? Infinity;\n\n      if (\n        reconnectEnabled &&\n        canReconnect &&\n        shouldReconnectRef.current &&\n        attemptCountRef.current < maxAttempts\n      ) {\n        setIsReconnecting(true);\n        attemptCountRef.current += 1;\n\n        // Exponential backoff with up to 30% random jitter so parallel\n        // sockets (main + planning) don't retry in lockstep and hammer an\n        // already-struggling server every few seconds forever.\n        const baseDelay = Math.min(\n          RECONNECT_BASE_DELAY_MS * 2 ** (attemptCountRef.current - 1),\n          RECONNECT_MAX_DELAY_MS,\n        );\n        const delay = baseDelay + Math.random() * baseDelay * 0.3;\n\n        reconnectTimeoutRef.current = setTimeout(() => {\n          connectWebSocket();\n        }, delay);\n      } else {\n        setIsReconnecting(false);\n      }\n    };\n\n    ws.onerror = (event) => {\n      // Ignore errors from sockets we've deliberately replaced or closed —\n      // aborting a mid-handshake socket fires `error`, and it must not\n      // surface as a connection failure for the replacement socket.\n      if (!allowedToReconnectRef.current.has(ws)) {\n        return;\n      }\n      setIsConnected(false);\n      optionsRef.current?.onError?.(event);\n    };\n  }, [url]);\n\n  React.useEffect(() => {\n    // Reset shouldReconnect flag and attempt count when creating a new connection\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n\n    // Only attempt connection if we have a valid URL\n    if (url && url.trim() !== \"\") {\n      connectWebSocket();\n    }\n\n    return () => {\n      // Disable reconnection on unmount to prevent reconnection attempts\n      // This must be set BEFORE closing the socket, so the onclose handler sees it\n      shouldReconnectRef.current = false;\n      // Clear any pending reconnection timeouts\n      if (reconnectTimeoutRef.current) {\n        clearTimeout(reconnectTimeoutRef.current);\n        reconnectTimeoutRef.current = null;\n      }\n      // Close the WebSocket connection\n      if (wsRef.current) {\n        const { readyState } = wsRef.current;\n        // Remove this WebSocket from the allowed list BEFORE closing\n        // so its onclose handler won't try to reconnect\n        allowedToReconnectRef.current.delete(wsRef.current);\n        // Only close if not already closed/closing\n        if (\n          readyState === WebSocket.CONNECTING ||\n          readyState === WebSocket.OPEN\n        ) {\n          wsRef.current.close();\n        }\n        wsRef.current = null;\n      }\n    };\n  }, [url, connectWebSocket]);\n\n  const sendMessage = React.useCallback(\n    (data: string | Blob | BufferSource) => {\n      if (wsRef.current?.readyState === WebSocket.OPEN) {\n        wsRef.current.send(data);\n      }\n    },\n    [],\n  );\n\n  const disconnect = React.useCallback(() => {\n    shouldReconnectRef.current = false;\n    setIsReconnecting(false);\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n    if (wsRef.current) {\n      // Remove from allowed list before closing\n      allowedToReconnectRef.current.delete(wsRef.current);\n      wsRef.current.close();\n    }\n  }, []);\n\n  const reconnect = React.useCallback(() => {\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n    setIsReconnecting(true);\n    setError(null);\n\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n\n    if (wsRef.current) {\n      const ws = wsRef.current;\n      allowedToReconnectRef.current.delete(ws);\n      wsRef.current = null;\n      if (\n        ws.readyState === WebSocket.CONNECTING ||\n        ws.readyState === WebSocket.OPEN\n      ) {\n        ws.close();\n      }\n    }\n\n    connectWebSocket();\n  }, [connectWebSocket]);\n\n  return {\n    isConnected,\n    error,\n    socket: wsRef.current,\n    sendMessage,\n    isReconnecting,\n    attemptCount: attemptCountRef.current,\n    disconnect,\n    reconnect,\n  };\n};\n"],"mappings":"oLAkBA,IAAM,EAA0B,IAC1B,EAAyB,IAElB,GAAgB,EAAa,IAAmC,CAC3E,GAAM,CAAC,EAAa,GAAkB,EAAA,QAAM,SAAS,EAAK,EACpD,CAAC,EAAO,GAAY,EAAA,QAAM,SAAuB,IAAI,EACrD,CAAC,EAAgB,GAAqB,EAAA,QAAM,SAAS,EAAK,EAC1D,EAAQ,EAAA,QAAM,OAAyB,IAAI,EAC3C,EAAkB,EAAA,QAAM,OAAO,CAAC,EAChC,EAAsB,EAAA,QAAM,OAA8B,IAAI,EAC9D,EAAqB,EAAA,QAAM,OAAO,EAAI,EAEtC,EAAwB,EAAA,QAAM,OAA2B,IAAI,OAAS,EAGtE,EAAa,EAAA,QAAM,OAAO,CAAO,EACvC,EAAA,QAAM,cAAgB,CACpB,EAAW,QAAU,CACvB,EAAG,CAAC,CAAO,CAAC,EAEZ,IAAM,EAAmB,EAAA,QAAM,gBAAkB,CAE/C,IAAI,EAAQ,EACZ,GAAI,EAAW,SAAS,YAAa,CACnC,IAAM,EAAe,OAAO,QAC1B,EAAW,QAAQ,WACrB,EAAE,QACC,EAAK,CAAC,EAAK,MACV,EAAI,GAAO,OAAO,CAAK,EAChB,GAET,CAAC,CACH,EAEA,EAAQ,GAAG,EAAI,GAAG,IADC,gBAAgB,CACjB,EAAO,SAAS,GACpC,CAEA,IAAM,EAAK,IAAI,UAAU,CAAK,EAC9B,EAAM,QAAU,EAEhB,EAAsB,QAAQ,IAAI,CAAE,EAKpC,IAAM,EAA0B,EAAA,uBAAuB,CAAE,EAEzD,EAAG,OAAU,GAAU,CACrB,EAAwB,EACxB,EAAA,kBAAkB,EAAI,EAAW,SAAS,aAAa,EACvD,EAAe,EAAI,EACnB,EAAS,IAAI,EACb,EAAkB,EAAK,EACvB,EAAgB,QAAU,EAC1B,EAAW,SAAS,SAAS,CAAK,CACpC,EAEA,EAAG,UAAa,GAAU,CAIxB,EAAW,SAAS,YAAY,CAAK,CACvC,EAEA,EAAG,QAAW,GAAU,CACtB,EAAwB,EAExB,IAAM,EAAe,EAAsB,QAAQ,IAAI,CAAE,EACzD,EAAe,EAAK,EAEhB,EAAM,OAAS,MAEjB,EACM,MACF,8BAA8B,EAAM,KAAK,IAAI,EAAM,QAAU,kCAC/D,CACF,EAEI,GACF,EAAW,SAAS,UAAU,CAAK,GAOnB,EAAM,UAAY,MAAQ,EAAM,UAAY,GAE9D,EAAW,SAAS,UAAU,CAAK,EAKrC,IAAM,EAAmB,EAAW,SAAS,WAAW,SAAW,GAC7D,EACJ,EAAW,SAAS,WAAW,aAAe,IAEhD,GACE,GACA,GACA,EAAmB,SACnB,EAAgB,QAAU,EAC1B,CACA,EAAkB,EAAI,EACtB,EAAgB,SAAW,EAK3B,IAAM,EAAY,KAAK,IACrB,EAA0B,IAAM,EAAgB,QAAU,GAC1D,CACF,EACM,EAAQ,EAAY,KAAK,OAAO,EAAI,EAAY,GAEtD,EAAoB,QAAU,eAAiB,CAC7C,EAAiB,CACnB,EAAG,CAAK,CACV,MACE,EAAkB,EAAK,CAE3B,EAEA,EAAG,QAAW,GAAU,CAIjB,EAAsB,QAAQ,IAAI,CAAE,IAGzC,EAAe,EAAK,EACpB,EAAW,SAAS,UAAU,CAAK,EACrC,CACF,EAAG,CAAC,CAAG,CAAC,EAER,EAAA,QAAM,eAEJ,EAAmB,QAAU,GAC7B,EAAgB,QAAU,EAGtB,GAAO,EAAI,KAAK,IAAM,IACxB,EAAiB,MAGN,CAUX,GAPA,EAAmB,QAAU,GAE7B,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAG5B,EAAM,QAAS,CACjB,GAAM,CAAE,cAAe,EAAM,QAG7B,EAAsB,QAAQ,OAAO,EAAM,OAAO,GAGhD,IAAe,UAAU,YACzB,IAAe,UAAU,OAEzB,EAAM,QAAQ,MAAM,EAEtB,EAAM,QAAU,IAClB,CACF,GACC,CAAC,EAAK,CAAgB,CAAC,EAE1B,IAAM,EAAc,EAAA,QAAM,YACvB,GAAuC,CAClC,EAAM,SAAS,aAAe,UAAU,MAC1C,EAAM,QAAQ,KAAK,CAAI,CAE3B,EACA,CAAC,CACH,EAEM,EAAa,EAAA,QAAM,gBAAkB,CACzC,EAAmB,QAAU,GAC7B,EAAkB,EAAK,EACvB,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAE5B,EAAM,UAER,EAAsB,QAAQ,OAAO,EAAM,OAAO,EAClD,EAAM,QAAQ,MAAM,EAExB,EAAG,CAAC,CAAC,EAEC,EAAY,EAAA,QAAM,gBAAkB,CAWxC,GAVA,EAAmB,QAAU,GAC7B,EAAgB,QAAU,EAC1B,EAAkB,EAAI,EACtB,EAAS,IAAI,EAEb,AAEE,EAAoB,WADpB,aAAa,EAAoB,OAAO,EACV,MAG5B,EAAM,QAAS,CACjB,IAAM,EAAK,EAAM,QACjB,EAAsB,QAAQ,OAAO,CAAE,EACvC,EAAM,QAAU,MAEd,EAAG,aAAe,UAAU,YAC5B,EAAG,aAAe,UAAU,OAE5B,EAAG,MAAM,CAEb,CAEA,EAAiB,CACnB,EAAG,CAAC,CAAgB,CAAC,EAErB,MAAO,CACL,cACA,QACA,OAAQ,EAAM,QACd,cACA,iBACA,aAAc,EAAgB,QAC9B,aACA,WACF,CACF"}

package/dist/hooks/use-websocket.d.ts

@@ -10,9 +10,8 @@ export interface WebSocketHookOptions {
         maxAttempts?: number;
     };
 }
-export declare const useWebSocket: <T = string>(url: string, options?: WebSocketHookOptions) => {
+export declare const useWebSocket: (url: string, options?: WebSocketHookOptions) => {
     isConnected: boolean;
-    lastMessage: T | null;
     error: Error | null;
     socket: WebSocket | null;
     sendMessage: (data: string | Blob | BufferSource) => void;

package/dist/hooks/use-websocket.js

@@ -1,63 +1,70 @@
 import { sendWebSocketAuth as e } from "../utils/websocket-auth.js";
-import t from "react";
+import { startHandshakeWatchdog as t } from "../utils/websocket-handshake.js";
+import n from "react";
 //#region src/hooks/use-websocket.ts
-var n = (n, r) => {
-	let [i, a] = t.useState(!1), [o, s] = t.useState(null), [c, l] = t.useState(null), [u, d] = t.useState(!1), f = t.useRef(null), p = t.useRef(0), m = t.useRef(null), h = t.useRef(!0), g = t.useRef(/* @__PURE__ */ new WeakSet()), _ = t.useRef(r);
-	t.useEffect(() => {
-		_.current = r;
-	}, [r]);
-	let v = t.useCallback(() => {
-		let t = n;
-		if (_.current?.queryParams) {
-			let e = Object.entries(_.current.queryParams).reduce((e, [t, n]) => (e[t] = String(n), e), {});
-			t = `${n}?${new URLSearchParams(e).toString()}`;
+var r = 1e3, i = 3e4, a = (a, o) => {
+	let [s, c] = n.useState(!1), [l, u] = n.useState(null), [d, f] = n.useState(!1), p = n.useRef(null), m = n.useRef(0), h = n.useRef(null), g = n.useRef(!0), _ = n.useRef(/* @__PURE__ */ new WeakSet()), v = n.useRef(o);
+	n.useEffect(() => {
+		v.current = o;
+	}, [o]);
+	let y = n.useCallback(() => {
+		let n = a;
+		if (v.current?.queryParams) {
+			let e = Object.entries(v.current.queryParams).reduce((e, [t, n]) => (e[t] = String(n), e), {});
+			n = `${a}?${new URLSearchParams(e).toString()}`;
 		}
-		let r = new WebSocket(t);
-		f.current = r, g.current.add(r), r.onopen = (t) => {
-			e(r, _.current?.sessionApiKey), a(!0), l(null), d(!1), p.current = 0, _.current?.onOpen?.(t);
-		}, r.onmessage = (e) => {
-			s(e.data), _.current?.onMessage?.(e);
-		}, r.onclose = (e) => {
-			let t = g.current.has(r);
-			a(!1), e.code !== 1e3 && (l(/* @__PURE__ */ Error(`WebSocket closed with code ${e.code}: ${e.reason || "Connection closed unexpectedly"}`)), t && _.current?.onError?.(e)), _.current?.onClose?.(e);
-			let n = _.current?.reconnect?.enabled ?? !1, i = _.current?.reconnect?.maxAttempts ?? Infinity;
-			n && t && h.current && p.current < i ? (d(!0), p.current += 1, m.current = setTimeout(() => {
-				v();
-			}, 3e3)) : d(!1);
-		}, r.onerror = (e) => {
-			a(!1), _.current?.onError?.(e);
+		let o = new WebSocket(n);
+		p.current = o, _.current.add(o);
+		let s = t(o);
+		o.onopen = (t) => {
+			s(), e(o, v.current?.sessionApiKey), c(!0), u(null), f(!1), m.current = 0, v.current?.onOpen?.(t);
+		}, o.onmessage = (e) => {
+			v.current?.onMessage?.(e);
+		}, o.onclose = (e) => {
+			s();
+			let t = _.current.has(o);
+			c(!1), e.code !== 1e3 && (u(/* @__PURE__ */ Error(`WebSocket closed with code ${e.code}: ${e.reason || "Connection closed unexpectedly"}`)), t && v.current?.onError?.(e)), p.current !== null && p.current !== o || v.current?.onClose?.(e);
+			let n = v.current?.reconnect?.enabled ?? !1, a = v.current?.reconnect?.maxAttempts ?? Infinity;
+			if (n && t && g.current && m.current < a) {
+				f(!0), m.current += 1;
+				let e = Math.min(r * 2 ** (m.current - 1), i), t = e + Math.random() * e * .3;
+				h.current = setTimeout(() => {
+					y();
+				}, t);
+			} else f(!1);
+		}, o.onerror = (e) => {
+			_.current.has(o) && (c(!1), v.current?.onError?.(e));
 		};
-	}, [n]);
-	t.useEffect(() => (h.current = !0, p.current = 0, n && n.trim() !== "" && v(), () => {
-		if (h.current = !1, m.current &&= (clearTimeout(m.current), null), f.current) {
-			let { readyState: e } = f.current;
-			g.current.delete(f.current), (e === WebSocket.CONNECTING || e === WebSocket.OPEN) && f.current.close(), f.current = null;
+	}, [a]);
+	n.useEffect(() => (g.current = !0, m.current = 0, a && a.trim() !== "" && y(), () => {
+		if (g.current = !1, h.current &&= (clearTimeout(h.current), null), p.current) {
+			let { readyState: e } = p.current;
+			_.current.delete(p.current), (e === WebSocket.CONNECTING || e === WebSocket.OPEN) && p.current.close(), p.current = null;
 		}
-	}), [n, v]);
-	let y = t.useCallback((e) => {
-		f.current?.readyState === WebSocket.OPEN && f.current.send(e);
-	}, []), b = t.useCallback(() => {
-		h.current = !1, d(!1), m.current &&= (clearTimeout(m.current), null), f.current && (g.current.delete(f.current), f.current.close());
-	}, []), x = t.useCallback(() => {
-		if (h.current = !0, p.current = 0, d(!0), l(null), m.current &&= (clearTimeout(m.current), null), f.current) {
-			let e = f.current;
-			g.current.delete(e), f.current = null, (e.readyState === WebSocket.CONNECTING || e.readyState === WebSocket.OPEN) && e.close();
+	}), [a, y]);
+	let b = n.useCallback((e) => {
+		p.current?.readyState === WebSocket.OPEN && p.current.send(e);
+	}, []), x = n.useCallback(() => {
+		g.current = !1, f(!1), h.current &&= (clearTimeout(h.current), null), p.current && (_.current.delete(p.current), p.current.close());
+	}, []), S = n.useCallback(() => {
+		if (g.current = !0, m.current = 0, f(!0), u(null), h.current &&= (clearTimeout(h.current), null), p.current) {
+			let e = p.current;
+			_.current.delete(e), p.current = null, (e.readyState === WebSocket.CONNECTING || e.readyState === WebSocket.OPEN) && e.close();
 		}
-		v();
-	}, [v]);
+		y();
+	}, [y]);
 	return {
-		isConnected: i,
-		lastMessage: o,
-		error: c,
-		socket: f.current,
-		sendMessage: y,
-		isReconnecting: u,
-		attemptCount: p.current,
-		disconnect: b,
-		reconnect: x
+		isConnected: s,
+		error: l,
+		socket: p.current,
+		sendMessage: b,
+		isReconnecting: d,
+		attemptCount: m.current,
+		disconnect: x,
+		reconnect: S
 	};
 };
 //#endregion
-export { n as useWebSocket };
+export { a as useWebSocket };
 
 //# sourceMappingURL=use-websocket.js.map

package/dist/hooks/use-websocket.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-websocket.js","names":[],"sources":["../../src/hooks/use-websocket.ts"],"sourcesContent":["import React from \"react\";\nimport { sendWebSocketAuth } from \"#/utils/websocket-auth\";\n\nexport interface WebSocketHookOptions {\n  queryParams?: Record<string, string | boolean>;\n  sessionApiKey?: string | null;\n  onOpen?: (event: Event) => void;\n  onClose?: (event: CloseEvent) => void;\n  onMessage?: (event: MessageEvent) => void;\n  onError?: (event: Event) => void;\n  reconnect?: {\n    enabled?: boolean;\n    maxAttempts?: number;\n  };\n}\n\nexport const useWebSocket = <T = string>(\n  url: string,\n  options?: WebSocketHookOptions,\n) => {\n  const [isConnected, setIsConnected] = React.useState(false);\n  const [lastMessage, setLastMessage] = React.useState<T | null>(null);\n  const [error, setError] = React.useState<Error | null>(null);\n  const [isReconnecting, setIsReconnecting] = React.useState(false);\n  const wsRef = React.useRef<WebSocket | null>(null);\n  const attemptCountRef = React.useRef(0);\n  const reconnectTimeoutRef = React.useRef<NodeJS.Timeout | null>(null);\n  const shouldReconnectRef = React.useRef(true); // Only set to false by disconnect()\n  // Track which WebSocket instances are allowed to reconnect using a WeakSet\n  const allowedToReconnectRef = React.useRef<WeakSet<WebSocket>>(new WeakSet());\n\n  // Store options in a ref to avoid reconnecting when callbacks change\n  const optionsRef = React.useRef(options);\n  React.useEffect(() => {\n    optionsRef.current = options;\n  }, [options]);\n\n  const connectWebSocket = React.useCallback(() => {\n    // Build URL with query parameters if provided\n    let wsUrl = url;\n    if (optionsRef.current?.queryParams) {\n      const stringParams = Object.entries(\n        optionsRef.current.queryParams,\n      ).reduce(\n        (acc, [key, value]) => {\n          acc[key] = String(value);\n          return acc;\n        },\n        {} as Record<string, string>,\n      );\n      const params = new URLSearchParams(stringParams);\n      wsUrl = `${url}?${params.toString()}`;\n    }\n\n    const ws = new WebSocket(wsUrl);\n    wsRef.current = ws;\n    // Mark this WebSocket instance as allowed to reconnect\n    allowedToReconnectRef.current.add(ws);\n\n    ws.onopen = (event) => {\n      sendWebSocketAuth(ws, optionsRef.current?.sessionApiKey);\n      setIsConnected(true);\n      setError(null); // Clear any previous errors\n      setIsReconnecting(false);\n      attemptCountRef.current = 0; // Reset attempt count on successful connection\n      optionsRef.current?.onOpen?.(event);\n    };\n\n    ws.onmessage = (event) => {\n      setLastMessage(event.data);\n      optionsRef.current?.onMessage?.(event);\n    };\n\n    ws.onclose = (event) => {\n      // Check if this specific WebSocket instance is allowed to reconnect\n      const canReconnect = allowedToReconnectRef.current.has(ws);\n      setIsConnected(false);\n      // If the connection closes with an error code, treat it as an error\n      if (event.code !== 1000) {\n        // 1000 is normal closure\n        setError(\n          new Error(\n            `WebSocket closed with code ${event.code}: ${event.reason || \"Connection closed unexpectedly\"}`,\n          ),\n        );\n        // Also call onError handler for error closures (only if allowed to reconnect)\n        if (canReconnect) {\n          optionsRef.current?.onError?.(event);\n        }\n      }\n      optionsRef.current?.onClose?.(event);\n\n      // Attempt reconnection if enabled and allowed\n      // IMPORTANT: Only reconnect if this specific instance is allowed to reconnect\n      const reconnectEnabled = optionsRef.current?.reconnect?.enabled ?? false;\n      const maxAttempts =\n        optionsRef.current?.reconnect?.maxAttempts ?? Infinity;\n\n      if (\n        reconnectEnabled &&\n        canReconnect &&\n        shouldReconnectRef.current &&\n        attemptCountRef.current < maxAttempts\n      ) {\n        setIsReconnecting(true);\n        attemptCountRef.current += 1;\n\n        reconnectTimeoutRef.current = setTimeout(() => {\n          connectWebSocket();\n        }, 3000); // 3 second delay\n      } else {\n        setIsReconnecting(false);\n      }\n    };\n\n    ws.onerror = (event) => {\n      setIsConnected(false);\n      optionsRef.current?.onError?.(event);\n    };\n  }, [url]);\n\n  React.useEffect(() => {\n    // Reset shouldReconnect flag and attempt count when creating a new connection\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n\n    // Only attempt connection if we have a valid URL\n    if (url && url.trim() !== \"\") {\n      connectWebSocket();\n    }\n\n    return () => {\n      // Disable reconnection on unmount to prevent reconnection attempts\n      // This must be set BEFORE closing the socket, so the onclose handler sees it\n      shouldReconnectRef.current = false;\n      // Clear any pending reconnection timeouts\n      if (reconnectTimeoutRef.current) {\n        clearTimeout(reconnectTimeoutRef.current);\n        reconnectTimeoutRef.current = null;\n      }\n      // Close the WebSocket connection\n      if (wsRef.current) {\n        const { readyState } = wsRef.current;\n        // Remove this WebSocket from the allowed list BEFORE closing\n        // so its onclose handler won't try to reconnect\n        allowedToReconnectRef.current.delete(wsRef.current);\n        // Only close if not already closed/closing\n        if (\n          readyState === WebSocket.CONNECTING ||\n          readyState === WebSocket.OPEN\n        ) {\n          wsRef.current.close();\n        }\n        wsRef.current = null;\n      }\n    };\n  }, [url, connectWebSocket]);\n\n  const sendMessage = React.useCallback(\n    (data: string | Blob | BufferSource) => {\n      if (wsRef.current?.readyState === WebSocket.OPEN) {\n        wsRef.current.send(data);\n      }\n    },\n    [],\n  );\n\n  const disconnect = React.useCallback(() => {\n    shouldReconnectRef.current = false;\n    setIsReconnecting(false);\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n    if (wsRef.current) {\n      // Remove from allowed list before closing\n      allowedToReconnectRef.current.delete(wsRef.current);\n      wsRef.current.close();\n    }\n  }, []);\n\n  const reconnect = React.useCallback(() => {\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n    setIsReconnecting(true);\n    setError(null);\n\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n\n    if (wsRef.current) {\n      const ws = wsRef.current;\n      allowedToReconnectRef.current.delete(ws);\n      wsRef.current = null;\n      if (\n        ws.readyState === WebSocket.CONNECTING ||\n        ws.readyState === WebSocket.OPEN\n      ) {\n        ws.close();\n      }\n    }\n\n    connectWebSocket();\n  }, [connectWebSocket]);\n\n  return {\n    isConnected,\n    lastMessage,\n    error,\n    socket: wsRef.current,\n    sendMessage,\n    isReconnecting,\n    attemptCount: attemptCountRef.current,\n    disconnect,\n    reconnect,\n  };\n};\n"],"mappings":";;;AAgBA,IAAa,KACX,GACA,MACG;CACH,IAAM,CAAC,GAAa,KAAkB,EAAM,SAAS,EAAK,GACpD,CAAC,GAAa,KAAkB,EAAM,SAAmB,IAAI,GAC7D,CAAC,GAAO,KAAY,EAAM,SAAuB,IAAI,GACrD,CAAC,GAAgB,KAAqB,EAAM,SAAS,EAAK,GAC1D,IAAQ,EAAM,OAAyB,IAAI,GAC3C,IAAkB,EAAM,OAAO,CAAC,GAChC,IAAsB,EAAM,OAA8B,IAAI,GAC9D,IAAqB,EAAM,OAAO,EAAI,GAEtC,IAAwB,EAAM,uBAA2B,IAAI,QAAQ,CAAC,GAGtE,IAAa,EAAM,OAAO,CAAO;CACvC,EAAM,gBAAgB;EACpB,EAAW,UAAU;CACvB,GAAG,CAAC,CAAO,CAAC;CAEZ,IAAM,IAAmB,EAAM,kBAAkB;EAE/C,IAAI,IAAQ;EACZ,IAAI,EAAW,SAAS,aAAa;GACnC,IAAM,IAAe,OAAO,QAC1B,EAAW,QAAQ,WACrB,EAAE,QACC,GAAK,CAAC,GAAK,QACV,EAAI,KAAO,OAAO,CAAK,GAChB,IAET,CAAC,CACH;GAEA,IAAQ,GAAG,EAAI,GAAG,IADC,gBAAgB,CACjB,EAAO,SAAS;EACpC;EAEA,IAAM,IAAK,IAAI,UAAU,CAAK;EA6D9B,AA5DA,EAAM,UAAU,GAEhB,EAAsB,QAAQ,IAAI,CAAE,GAEpC,EAAG,UAAU,MAAU;GAMrB,AALA,EAAkB,GAAI,EAAW,SAAS,aAAa,GACvD,EAAe,EAAI,GACnB,EAAS,IAAI,GACb,EAAkB,EAAK,GACvB,EAAgB,UAAU,GAC1B,EAAW,SAAS,SAAS,CAAK;EACpC,GAEA,EAAG,aAAa,MAAU;GAExB,AADA,EAAe,EAAM,IAAI,GACzB,EAAW,SAAS,YAAY,CAAK;EACvC,GAEA,EAAG,WAAW,MAAU;GAEtB,IAAM,IAAe,EAAsB,QAAQ,IAAI,CAAE;GAezD,AAdA,EAAe,EAAK,GAEhB,EAAM,SAAS,QAEjB,EACE,gBAAI,MACF,8BAA8B,EAAM,KAAK,IAAI,EAAM,UAAU,kCAC/D,CACF,GAEI,KACF,EAAW,SAAS,UAAU,CAAK,IAGvC,EAAW,SAAS,UAAU,CAAK;GAInC,IAAM,IAAmB,EAAW,SAAS,WAAW,WAAW,IAC7D,IACJ,EAAW,SAAS,WAAW,eAAe;GAEhD,AACE,KACA,KACA,EAAmB,WACnB,EAAgB,UAAU,KAE1B,EAAkB,EAAI,GACtB,EAAgB,WAAW,GAE3B,EAAoB,UAAU,iBAAiB;IAC7C,EAAiB;GACnB,GAAG,GAAI,KAEP,EAAkB,EAAK;EAE3B,GAEA,EAAG,WAAW,MAAU;GAEtB,AADA,EAAe,EAAK,GACpB,EAAW,SAAS,UAAU,CAAK;EACrC;CACF,GAAG,CAAC,CAAG,CAAC;CAER,EAAM,iBAEJ,EAAmB,UAAU,IAC7B,EAAgB,UAAU,GAGtB,KAAO,EAAI,KAAK,MAAM,MACxB,EAAiB,SAGN;EAUX,IAPA,EAAmB,UAAU,IAE7B,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAG5B,EAAM,SAAS;GACjB,IAAM,EAAE,kBAAe,EAAM;GAW7B,AARA,EAAsB,QAAQ,OAAO,EAAM,OAAO,IAGhD,MAAe,UAAU,cACzB,MAAe,UAAU,SAEzB,EAAM,QAAQ,MAAM,GAEtB,EAAM,UAAU;EAClB;CACF,IACC,CAAC,GAAK,CAAgB,CAAC;CAE1B,IAAM,IAAc,EAAM,aACvB,MAAuC;EACtC,AAAI,EAAM,SAAS,eAAe,UAAU,QAC1C,EAAM,QAAQ,KAAK,CAAI;CAE3B,GACA,CAAC,CACH,GAEM,IAAa,EAAM,kBAAkB;EAOzC,AANA,EAAmB,UAAU,IAC7B,EAAkB,EAAK,GACvB,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAE5B,EAAM,YAER,EAAsB,QAAQ,OAAO,EAAM,OAAO,GAClD,EAAM,QAAQ,MAAM;CAExB,GAAG,CAAC,CAAC,GAEC,IAAY,EAAM,kBAAkB;EAWxC,IAVA,EAAmB,UAAU,IAC7B,EAAgB,UAAU,GAC1B,EAAkB,EAAI,GACtB,EAAS,IAAI,GAEb,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAG5B,EAAM,SAAS;GACjB,IAAM,IAAK,EAAM;GAGjB,AAFA,EAAsB,QAAQ,OAAO,CAAE,GACvC,EAAM,UAAU,OAEd,EAAG,eAAe,UAAU,cAC5B,EAAG,eAAe,UAAU,SAE5B,EAAG,MAAM;EAEb;EAEA,EAAiB;CACnB,GAAG,CAAC,CAAgB,CAAC;CAErB,OAAO;EACL;EACA;EACA;EACA,QAAQ,EAAM;EACd;EACA;EACA,cAAc,EAAgB;EAC9B;EACA;CACF;AACF"}
+{"version":3,"file":"use-websocket.js","names":[],"sources":["../../src/hooks/use-websocket.ts"],"sourcesContent":["import React from \"react\";\nimport { sendWebSocketAuth } from \"#/utils/websocket-auth\";\nimport { startHandshakeWatchdog } from \"#/utils/websocket-handshake\";\n\nexport interface WebSocketHookOptions {\n  queryParams?: Record<string, string | boolean>;\n  sessionApiKey?: string | null;\n  onOpen?: (event: Event) => void;\n  onClose?: (event: CloseEvent) => void;\n  onMessage?: (event: MessageEvent) => void;\n  onError?: (event: Event) => void;\n  reconnect?: {\n    enabled?: boolean;\n    maxAttempts?: number;\n  };\n}\n\n// Reconnect backoff bounds: 1s, 2s, 4s, … capped at 30s.\nconst RECONNECT_BASE_DELAY_MS = 1_000;\nconst RECONNECT_MAX_DELAY_MS = 30_000;\n\nexport const useWebSocket = (url: string, options?: WebSocketHookOptions) => {\n  const [isConnected, setIsConnected] = React.useState(false);\n  const [error, setError] = React.useState<Error | null>(null);\n  const [isReconnecting, setIsReconnecting] = React.useState(false);\n  const wsRef = React.useRef<WebSocket | null>(null);\n  const attemptCountRef = React.useRef(0);\n  const reconnectTimeoutRef = React.useRef<NodeJS.Timeout | null>(null);\n  const shouldReconnectRef = React.useRef(true); // Only set to false by disconnect()\n  // Track which WebSocket instances are allowed to reconnect using a WeakSet\n  const allowedToReconnectRef = React.useRef<WeakSet<WebSocket>>(new WeakSet());\n\n  // Store options in a ref to avoid reconnecting when callbacks change\n  const optionsRef = React.useRef(options);\n  React.useEffect(() => {\n    optionsRef.current = options;\n  }, [options]);\n\n  const connectWebSocket = React.useCallback(() => {\n    // Build URL with query parameters if provided\n    let wsUrl = url;\n    if (optionsRef.current?.queryParams) {\n      const stringParams = Object.entries(\n        optionsRef.current.queryParams,\n      ).reduce(\n        (acc, [key, value]) => {\n          acc[key] = String(value);\n          return acc;\n        },\n        {} as Record<string, string>,\n      );\n      const params = new URLSearchParams(stringParams);\n      wsUrl = `${url}?${params.toString()}`;\n    }\n\n    const ws = new WebSocket(wsUrl);\n    wsRef.current = ws;\n    // Mark this WebSocket instance as allowed to reconnect\n    allowedToReconnectRef.current.add(ws);\n\n    // Abort a socket stuck in CONNECTING so it can't hold Chrome's per-host\n    // handshake lock indefinitely; its close flows into the reconnect path\n    // below.\n    const cancelHandshakeWatchdog = startHandshakeWatchdog(ws);\n\n    ws.onopen = (event) => {\n      cancelHandshakeWatchdog();\n      sendWebSocketAuth(ws, optionsRef.current?.sessionApiKey);\n      setIsConnected(true);\n      setError(null); // Clear any previous errors\n      setIsReconnecting(false);\n      attemptCountRef.current = 0; // Reset attempt count on successful connection\n      optionsRef.current?.onOpen?.(event);\n    };\n\n    ws.onmessage = (event) => {\n      // Deliberately no `lastMessage` state here: nothing reads it, and a\n      // React state write per frame re-renders this hook's owner on every\n      // streamed token. Consumers subscribe via `onMessage`.\n      optionsRef.current?.onMessage?.(event);\n    };\n\n    ws.onclose = (event) => {\n      cancelHandshakeWatchdog();\n      // Check if this specific WebSocket instance is allowed to reconnect\n      const canReconnect = allowedToReconnectRef.current.has(ws);\n      setIsConnected(false);\n      // If the connection closes with an error code, treat it as an error\n      if (event.code !== 1000) {\n        // 1000 is normal closure\n        setError(\n          new Error(\n            `WebSocket closed with code ${event.code}: ${event.reason || \"Connection closed unexpectedly\"}`,\n          ),\n        );\n        // Also call onError handler for error closures (only if allowed to reconnect)\n        if (canReconnect) {\n          optionsRef.current?.onError?.(event);\n        }\n      }\n      // Notify the consumer unless this socket was deliberately replaced by a\n      // newer one — a replaced socket's close event arrives late and must not\n      // clobber the replacement's OPEN state in the consumer. Final closes\n      // (disconnect/unmount, nothing replacing the socket) still notify.\n      const wasReplaced = wsRef.current !== null && wsRef.current !== ws;\n      if (!wasReplaced) {\n        optionsRef.current?.onClose?.(event);\n      }\n\n      // Attempt reconnection if enabled and allowed\n      // IMPORTANT: Only reconnect if this specific instance is allowed to reconnect\n      const reconnectEnabled = optionsRef.current?.reconnect?.enabled ?? false;\n      const maxAttempts =\n        optionsRef.current?.reconnect?.maxAttempts ?? Infinity;\n\n      if (\n        reconnectEnabled &&\n        canReconnect &&\n        shouldReconnectRef.current &&\n        attemptCountRef.current < maxAttempts\n      ) {\n        setIsReconnecting(true);\n        attemptCountRef.current += 1;\n\n        // Exponential backoff with up to 30% random jitter so parallel\n        // sockets (main + planning) don't retry in lockstep and hammer an\n        // already-struggling server every few seconds forever.\n        const baseDelay = Math.min(\n          RECONNECT_BASE_DELAY_MS * 2 ** (attemptCountRef.current - 1),\n          RECONNECT_MAX_DELAY_MS,\n        );\n        const delay = baseDelay + Math.random() * baseDelay * 0.3;\n\n        reconnectTimeoutRef.current = setTimeout(() => {\n          connectWebSocket();\n        }, delay);\n      } else {\n        setIsReconnecting(false);\n      }\n    };\n\n    ws.onerror = (event) => {\n      // Ignore errors from sockets we've deliberately replaced or closed —\n      // aborting a mid-handshake socket fires `error`, and it must not\n      // surface as a connection failure for the replacement socket.\n      if (!allowedToReconnectRef.current.has(ws)) {\n        return;\n      }\n      setIsConnected(false);\n      optionsRef.current?.onError?.(event);\n    };\n  }, [url]);\n\n  React.useEffect(() => {\n    // Reset shouldReconnect flag and attempt count when creating a new connection\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n\n    // Only attempt connection if we have a valid URL\n    if (url && url.trim() !== \"\") {\n      connectWebSocket();\n    }\n\n    return () => {\n      // Disable reconnection on unmount to prevent reconnection attempts\n      // This must be set BEFORE closing the socket, so the onclose handler sees it\n      shouldReconnectRef.current = false;\n      // Clear any pending reconnection timeouts\n      if (reconnectTimeoutRef.current) {\n        clearTimeout(reconnectTimeoutRef.current);\n        reconnectTimeoutRef.current = null;\n      }\n      // Close the WebSocket connection\n      if (wsRef.current) {\n        const { readyState } = wsRef.current;\n        // Remove this WebSocket from the allowed list BEFORE closing\n        // so its onclose handler won't try to reconnect\n        allowedToReconnectRef.current.delete(wsRef.current);\n        // Only close if not already closed/closing\n        if (\n          readyState === WebSocket.CONNECTING ||\n          readyState === WebSocket.OPEN\n        ) {\n          wsRef.current.close();\n        }\n        wsRef.current = null;\n      }\n    };\n  }, [url, connectWebSocket]);\n\n  const sendMessage = React.useCallback(\n    (data: string | Blob | BufferSource) => {\n      if (wsRef.current?.readyState === WebSocket.OPEN) {\n        wsRef.current.send(data);\n      }\n    },\n    [],\n  );\n\n  const disconnect = React.useCallback(() => {\n    shouldReconnectRef.current = false;\n    setIsReconnecting(false);\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n    if (wsRef.current) {\n      // Remove from allowed list before closing\n      allowedToReconnectRef.current.delete(wsRef.current);\n      wsRef.current.close();\n    }\n  }, []);\n\n  const reconnect = React.useCallback(() => {\n    shouldReconnectRef.current = true;\n    attemptCountRef.current = 0;\n    setIsReconnecting(true);\n    setError(null);\n\n    if (reconnectTimeoutRef.current) {\n      clearTimeout(reconnectTimeoutRef.current);\n      reconnectTimeoutRef.current = null;\n    }\n\n    if (wsRef.current) {\n      const ws = wsRef.current;\n      allowedToReconnectRef.current.delete(ws);\n      wsRef.current = null;\n      if (\n        ws.readyState === WebSocket.CONNECTING ||\n        ws.readyState === WebSocket.OPEN\n      ) {\n        ws.close();\n      }\n    }\n\n    connectWebSocket();\n  }, [connectWebSocket]);\n\n  return {\n    isConnected,\n    error,\n    socket: wsRef.current,\n    sendMessage,\n    isReconnecting,\n    attemptCount: attemptCountRef.current,\n    disconnect,\n    reconnect,\n  };\n};\n"],"mappings":";;;;AAkBA,IAAM,IAA0B,KAC1B,IAAyB,KAElB,KAAgB,GAAa,MAAmC;CAC3E,IAAM,CAAC,GAAa,KAAkB,EAAM,SAAS,EAAK,GACpD,CAAC,GAAO,KAAY,EAAM,SAAuB,IAAI,GACrD,CAAC,GAAgB,KAAqB,EAAM,SAAS,EAAK,GAC1D,IAAQ,EAAM,OAAyB,IAAI,GAC3C,IAAkB,EAAM,OAAO,CAAC,GAChC,IAAsB,EAAM,OAA8B,IAAI,GAC9D,IAAqB,EAAM,OAAO,EAAI,GAEtC,IAAwB,EAAM,uBAA2B,IAAI,QAAQ,CAAC,GAGtE,IAAa,EAAM,OAAO,CAAO;CACvC,EAAM,gBAAgB;EACpB,EAAW,UAAU;CACvB,GAAG,CAAC,CAAO,CAAC;CAEZ,IAAM,IAAmB,EAAM,kBAAkB;EAE/C,IAAI,IAAQ;EACZ,IAAI,EAAW,SAAS,aAAa;GACnC,IAAM,IAAe,OAAO,QAC1B,EAAW,QAAQ,WACrB,EAAE,QACC,GAAK,CAAC,GAAK,QACV,EAAI,KAAO,OAAO,CAAK,GAChB,IAET,CAAC,CACH;GAEA,IAAQ,GAAG,EAAI,GAAG,IADC,gBAAgB,CACjB,EAAO,SAAS;EACpC;EAEA,IAAM,IAAK,IAAI,UAAU,CAAK;EAG9B,AAFA,EAAM,UAAU,GAEhB,EAAsB,QAAQ,IAAI,CAAE;EAKpC,IAAM,IAA0B,EAAuB,CAAE;EA8EzD,AA5EA,EAAG,UAAU,MAAU;GAOrB,AANA,EAAwB,GACxB,EAAkB,GAAI,EAAW,SAAS,aAAa,GACvD,EAAe,EAAI,GACnB,EAAS,IAAI,GACb,EAAkB,EAAK,GACvB,EAAgB,UAAU,GAC1B,EAAW,SAAS,SAAS,CAAK;EACpC,GAEA,EAAG,aAAa,MAAU;GAIxB,EAAW,SAAS,YAAY,CAAK;EACvC,GAEA,EAAG,WAAW,MAAU;GACtB,EAAwB;GAExB,IAAM,IAAe,EAAsB,QAAQ,IAAI,CAAE;GAoBzD,AAnBA,EAAe,EAAK,GAEhB,EAAM,SAAS,QAEjB,EACE,gBAAI,MACF,8BAA8B,EAAM,KAAK,IAAI,EAAM,UAAU,kCAC/D,CACF,GAEI,KACF,EAAW,SAAS,UAAU,CAAK,IAOnB,EAAM,YAAY,QAAQ,EAAM,YAAY,KAE9D,EAAW,SAAS,UAAU,CAAK;GAKrC,IAAM,IAAmB,EAAW,SAAS,WAAW,WAAW,IAC7D,IACJ,EAAW,SAAS,WAAW,eAAe;GAEhD,IACE,KACA,KACA,EAAmB,WACnB,EAAgB,UAAU,GAC1B;IAEA,AADA,EAAkB,EAAI,GACtB,EAAgB,WAAW;IAK3B,IAAM,IAAY,KAAK,IACrB,IAA0B,MAAM,EAAgB,UAAU,IAC1D,CACF,GACM,IAAQ,IAAY,KAAK,OAAO,IAAI,IAAY;IAEtD,EAAoB,UAAU,iBAAiB;KAC7C,EAAiB;IACnB,GAAG,CAAK;GACV,OACE,EAAkB,EAAK;EAE3B,GAEA,EAAG,WAAW,MAAU;GAIjB,EAAsB,QAAQ,IAAI,CAAE,MAGzC,EAAe,EAAK,GACpB,EAAW,SAAS,UAAU,CAAK;EACrC;CACF,GAAG,CAAC,CAAG,CAAC;CAER,EAAM,iBAEJ,EAAmB,UAAU,IAC7B,EAAgB,UAAU,GAGtB,KAAO,EAAI,KAAK,MAAM,MACxB,EAAiB,SAGN;EAUX,IAPA,EAAmB,UAAU,IAE7B,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAG5B,EAAM,SAAS;GACjB,IAAM,EAAE,kBAAe,EAAM;GAW7B,AARA,EAAsB,QAAQ,OAAO,EAAM,OAAO,IAGhD,MAAe,UAAU,cACzB,MAAe,UAAU,SAEzB,EAAM,QAAQ,MAAM,GAEtB,EAAM,UAAU;EAClB;CACF,IACC,CAAC,GAAK,CAAgB,CAAC;CAE1B,IAAM,IAAc,EAAM,aACvB,MAAuC;EACtC,AAAI,EAAM,SAAS,eAAe,UAAU,QAC1C,EAAM,QAAQ,KAAK,CAAI;CAE3B,GACA,CAAC,CACH,GAEM,IAAa,EAAM,kBAAkB;EAOzC,AANA,EAAmB,UAAU,IAC7B,EAAkB,EAAK,GACvB,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAE5B,EAAM,YAER,EAAsB,QAAQ,OAAO,EAAM,OAAO,GAClD,EAAM,QAAQ,MAAM;CAExB,GAAG,CAAC,CAAC,GAEC,IAAY,EAAM,kBAAkB;EAWxC,IAVA,EAAmB,UAAU,IAC7B,EAAgB,UAAU,GAC1B,EAAkB,EAAI,GACtB,EAAS,IAAI,GAEb,AAEE,EAAoB,aADpB,aAAa,EAAoB,OAAO,GACV,OAG5B,EAAM,SAAS;GACjB,IAAM,IAAK,EAAM;GAGjB,AAFA,EAAsB,QAAQ,OAAO,CAAE,GACvC,EAAM,UAAU,OAEd,EAAG,eAAe,UAAU,cAC5B,EAAG,eAAe,UAAU,SAE5B,EAAG,MAAM;EAEb;EAEA,EAAiB;CACnB,GAAG,CAAC,CAAgB,CAAC;CAErB,OAAO;EACL;EACA;EACA,QAAQ,EAAM;EACd;EACA;EACA,cAAc,EAAgB;EAC9B;EACA;CACF;AACF"}