react-realtime-hooks 2.0.2 → 2.0.3

package/dist/index.js

@@ -937,6 +937,7 @@ var useWebSocket = (options) => {
     [protocolsDependency]
   );
   const socketRef = useRef(null);
+  const detachSocketListenersRef = useRef(null);
   const socketKeyRef = useRef(null);
   const activeSocketEpochRef = useRef(null);
   const closingSocketEpochRef = useRef(null);
@@ -1049,6 +1050,10 @@ var useWebSocket = (options) => {
       socketKeyRef.current = null;
       activeSocketEpochRef.current = null;
       closingSocketEpochRef.current = config.trackClose ? socketEpoch : null;
+      if (!config.trackClose) {
+        detachSocketListenersRef.current?.();
+        detachSocketListenersRef.current = null;
+      }
       if (isSocketActive(socket2)) {
         socket2.close(config.code, config.reason);
       }
@@ -1160,6 +1165,8 @@ var useWebSocket = (options) => {
     if (!shouldHandleSocketClose(socketEpoch)) {
       return;
     }
+    detachSocketListenersRef.current?.();
+    detachSocketListenersRef.current = null;
     if (activeSocketEpochRef.current === socketEpoch) {
       socketRef.current = null;
       socketKeyRef.current = null;
@@ -1299,7 +1306,6 @@ var useWebSocket = (options) => {
       status: reconnect.status === "running" || reconnect.status === "scheduled" ? "reconnecting" : "connecting"
     }));
     let cancelled = false;
-    let detachListeners = null;
     queueMicrotask(() => {
       if (cancelled) {
         return;
@@ -1313,6 +1319,8 @@ var useWebSocket = (options) => {
         return;
       }
       const socket2 = new WebSocket(resolvedUrl, protocols);
+      detachSocketListenersRef.current?.();
+      detachSocketListenersRef.current = null;
       const socketEpoch = nextSocketEpochRef.current + 1;
       socketRef.current = socket2;
       socketKeyRef.current = nextSocketKey;
@@ -1346,7 +1354,7 @@ var useWebSocket = (options) => {
       socket2.addEventListener("message", handleSocketMessage);
       socket2.addEventListener("error", handleSocketError);
       socket2.addEventListener("close", handleSocketClose);
-      detachListeners = () => {
+      detachSocketListenersRef.current = () => {
         socket2.removeEventListener("open", handleSocketOpen);
         socket2.removeEventListener("message", handleSocketMessage);
         socket2.removeEventListener("error", handleSocketError);
@@ -1355,7 +1363,6 @@ var useWebSocket = (options) => {
     });
     return () => {
       cancelled = true;
-      detachListeners?.();
     };
   }, [
     closeSocket,
@@ -1376,6 +1383,8 @@ var useWebSocket = (options) => {
     supported
   ]);
   useEffect(() => () => {
+    detachSocketListenersRef.current?.();
+    detachSocketListenersRef.current = null;
     controller.noteEffectInitiatedClose();
     socketKeyRef.current = null;
     activeSocketEpochRef.current = null;
@@ -1520,6 +1529,7 @@ var useEventSource = (options) => {
     [eventsDependency]
   );
   const eventSourceRef = useRef(null);
+  const detachEventSourceListenersRef = useRef(null);
   const eventSourceKeyRef = useRef(null);
   const activeEventSourceEpochRef = useRef(null);
   const nextEventSourceEpochRef = useRef(0);
@@ -1557,6 +1567,8 @@ var useEventSource = (options) => {
     eventSourceRef.current = null;
     eventSourceKeyRef.current = null;
     activeEventSourceEpochRef.current = null;
+    detachEventSourceListenersRef.current?.();
+    detachEventSourceListenersRef.current = null;
     source.close();
   });
   const isActiveEventSourceEvent = useStableCallback((sourceEpoch) => {
@@ -1727,7 +1739,6 @@ var useEventSource = (options) => {
       status: reconnect.status === "running" || reconnect.status === "scheduled" ? "reconnecting" : "connecting"
     }));
     let cancelled = false;
-    let detachListeners = null;
     queueMicrotask(() => {
       if (cancelled) {
         return;
@@ -1777,7 +1788,7 @@ var useEventSource = (options) => {
         source.addEventListener(eventName, handler);
       }
       source.addEventListener("error", handleSourceError);
-      detachListeners = () => {
+      detachEventSourceListenersRef.current = () => {
         source.removeEventListener("open", handleSourceOpen);
         source.removeEventListener("message", handleSourceMessage);
         for (const [eventName, handler] of namedEventHandlers) {
@@ -1788,7 +1799,6 @@ var useEventSource = (options) => {
     });
     return () => {
       cancelled = true;
-      detachListeners?.();
     };
   }, [
     closeEventSource,
@@ -1807,6 +1817,8 @@ var useEventSource = (options) => {
     supported
   ]);
   useEffect(() => () => {
+    detachEventSourceListenersRef.current?.();
+    detachEventSourceListenersRef.current = null;
     suppressReconnectRef.current = true;
     eventSourceKeyRef.current = null;
     activeEventSourceEpochRef.current = null;