@xcelsior/ui-chat 2.0.2 → 2.0.3

package/dist/index.d.mts

@@ -310,6 +310,9 @@ interface UseWebSocketReturn {
 /**
  * Hook for WebSocket connection in chat widget.
  * Can use an external WebSocket connection (for agents) via the externalWebSocket prop.
+ *
+ * Handles React Strict Mode (dev) gracefully — tracks whether the effect has been
+ * cleaned up so that an aborted connection doesn't trigger reconnection.
  */
 declare function useWebSocket(config: IChatConfig, externalWebSocket?: WebSocket | null): UseWebSocketReturn;
 

package/dist/index.d.ts

@@ -310,6 +310,9 @@ interface UseWebSocketReturn {
 /**
  * Hook for WebSocket connection in chat widget.
  * Can use an external WebSocket connection (for agents) via the externalWebSocket prop.
+ *
+ * Handles React Strict Mode (dev) gracefully — tracks whether the effect has been
+ * cleaned up so that an aborted connection doesn't trigger reconnection.
  */
 declare function useWebSocket(config: IChatConfig, externalWebSocket?: WebSocket | null): UseWebSocketReturn;
 

package/dist/index.js

@@ -62,6 +62,7 @@ function useWebSocket(config, externalWebSocket) {
   const reconnectTimeoutRef = (0, import_react.useRef)(null);
   const reconnectAttemptsRef = (0, import_react.useRef)(0);
   const messageHandlerRef = (0, import_react.useRef)(null);
+  const abortedRef = (0, import_react.useRef)(false);
   const maxReconnectAttempts = 5;
   const reconnectDelay = 3e3;
   const isUsingExternalWs = !!externalWebSocket;
@@ -94,6 +95,7 @@ function useWebSocket(config, externalWebSocket) {
     };
   }, []);
   const connect = (0, import_react.useCallback)(() => {
+    if (abortedRef.current) return;
     console.log("connecting to WebSocket...", config.currentUser, config.conversationId);
     try {
       if (wsRef.current) {
@@ -113,6 +115,10 @@ function useWebSocket(config, externalWebSocket) {
       }
       const ws = new WebSocket(url.toString());
       ws.onopen = () => {
+        if (abortedRef.current) {
+          ws.close(1e3, "Effect cleaned up");
+          return;
+        }
         console.log("WebSocket connected");
         setIsConnected(true);
         setError(null);
@@ -120,12 +126,14 @@ function useWebSocket(config, externalWebSocket) {
         config.onConnectionChange?.(true);
       };
       ws.onerror = (event) => {
+        if (abortedRef.current) return;
         console.error("WebSocket error:", event);
         const err = new Error("WebSocket connection error");
         setError(err);
         config.onError?.(err);
       };
       ws.onclose = (event) => {
+        if (abortedRef.current) return;
         console.log("WebSocket closed:", event.code, event.reason);
         setIsConnected(false);
         config.onConnectionChange?.(false);
@@ -174,6 +182,7 @@ function useWebSocket(config, externalWebSocket) {
   );
   const reconnect = (0, import_react.useCallback)(() => {
     reconnectAttemptsRef.current = 0;
+    abortedRef.current = false;
     connect();
   }, [connect]);
   (0, import_react.useEffect)(() => {
@@ -183,8 +192,10 @@ function useWebSocket(config, externalWebSocket) {
       const cleanup = subscribeToMessage(externalWebSocket);
       return cleanup;
     }
+    abortedRef.current = false;
     connect();
     return () => {
+      abortedRef.current = true;
       if (reconnectTimeoutRef.current) {
         clearTimeout(reconnectTimeoutRef.current);
       }