react-realtime-hooks 1.0.2 → 1.0.3

package/dist/index.js

@@ -402,6 +402,7 @@ var useHeartbeat = (options) => {
   const startOnMount = options.startOnMount ?? true;
   const intervalRef = useRef(createManagedInterval());
   const timeoutRef = useRef(createManagedTimeout());
+  const generationRef = useRef(0);
   const [state, setState] = useState(
     () => createInitialState2(enabled && startOnMount)
   );
@@ -428,8 +429,7 @@ var useHeartbeat = (options) => {
       handleTimeout();
     }, options.timeoutMs);
   });
-  const runBeat = useEffectEvent(() => {
-    const performedAt = Date.now();
+  const handleBeatSuccess = useEffectEvent((performedAt) => {
     commitState((current) => ({
       ...current,
       hasTimedOut: false,
@@ -437,7 +437,35 @@ var useHeartbeat = (options) => {
     }));
     scheduleTimeout();
     options.onBeat?.();
-    void options.beat?.();
+  });
+  const handleBeatError = useEffectEvent((error) => {
+    timeoutRef.current.cancel();
+    options.onError?.(error);
+  });
+  const runBeat = useEffectEvent(() => {
+    const generation = generationRef.current;
+    const completeBeat = (result) => {
+      if (generation !== generationRef.current || result === false) {
+        return;
+      }
+      handleBeatSuccess(Date.now());
+    };
+    const failBeat = (error) => {
+      if (generation !== generationRef.current) {
+        return;
+      }
+      handleBeatError(error);
+    };
+    try {
+      const result = options.beat?.();
+      if (result !== null && typeof result === "object" && "then" in result) {
+        void Promise.resolve(result).then(completeBeat, failBeat);
+        return;
+      }
+      completeBeat(result);
+    } catch (error) {
+      failBeat(error);
+    }
   });
   const start = () => {
     if (!enabled) {
@@ -455,6 +483,7 @@ var useHeartbeat = (options) => {
     }));
   };
   const stop = () => {
+    generationRef.current += 1;
     intervalRef.current.cancel();
     timeoutRef.current.cancel();
     commitState((current) => ({
@@ -500,6 +529,7 @@ var useHeartbeat = (options) => {
     return void 0;
   }, [enabled, options.intervalMs, startOnMount]);
   useEffect(() => () => {
+    generationRef.current += 1;
     intervalRef.current.cancel();
     timeoutRef.current.cancel();
   }, []);
@@ -605,6 +635,7 @@ var toProtocolsDependency = (protocols) => {
   return Array.isArray(protocols) ? protocols.join("|") : protocols;
 };
 var toHeartbeatConfig = (heartbeat) => heartbeat === void 0 || heartbeat === false ? null : heartbeat;
+var isSocketActive = (socket) => socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING;
 var useWebSocket = (options) => {
   const connect = options.connect ?? true;
   const supported = isWebSocketSupported();
@@ -616,6 +647,7 @@ var useWebSocket = (options) => {
   const manualOpenRef = useRef(false);
   const skipCloseReconnectRef = useRef(false);
   const suppressReconnectRef = useRef(false);
+  const pendingCloseActionRef = useRef(null);
   const terminalErrorRef = useRef(null);
   const [openNonce, setOpenNonce] = useState(0);
   const [state, setState] = useState(
@@ -634,6 +666,7 @@ var useWebSocket = (options) => {
   const heartbeatConfig = toHeartbeatConfig(
     options.heartbeat
   );
+  const defaultHeartbeatAction = options.reconnect === false ? "close" : "reconnect";
   const heartbeatHookOptions = heartbeatConfig === null ? {
     enabled: false,
     intervalMs: 1e3,
@@ -669,6 +702,27 @@ var useWebSocket = (options) => {
   if (heartbeatConfig !== null && heartbeatConfig.onTimeout !== void 0) {
     heartbeatHookOptions.onTimeout = heartbeatConfig.onTimeout;
   }
+  if (heartbeatConfig !== null) {
+    const onTimeout = heartbeatHookOptions.onTimeout;
+    heartbeatHookOptions.onTimeout = () => {
+      applyHeartbeatAction(
+        heartbeatConfig.timeoutAction ?? defaultHeartbeatAction,
+        new Event("heartbeat-timeout"),
+        "heartbeat-timeout"
+      );
+      onTimeout?.();
+    };
+    const onError = heartbeatConfig.onError;
+    heartbeatHookOptions.onError = (error) => {
+      const event = error instanceof Event ? error : new Event("heartbeat-error");
+      applyHeartbeatAction(
+        heartbeatConfig.errorAction ?? heartbeatConfig.timeoutAction ?? defaultHeartbeatAction,
+        event,
+        "error"
+      );
+      onError?.(error);
+    };
+  }
   const heartbeat = useHeartbeat(
     heartbeatHookOptions
   );
@@ -684,10 +738,46 @@ var useWebSocket = (options) => {
     }
     socketRef.current = null;
     socketKeyRef.current = null;
-    if (socket2.readyState === WebSocket.OPEN || socket2.readyState === WebSocket.CONNECTING) {
+    if (isSocketActive(socket2)) {
       socket2.close(code, reason);
     }
   });
+  const applyHeartbeatAction = useEffectEvent(
+    (action, error, reconnectTrigger) => {
+      heartbeat.stop();
+      if (action === "none") {
+        commitState((current) => ({
+          ...current,
+          lastChangedAt: Date.now(),
+          lastError: error
+        }));
+        return;
+      }
+      const shouldReconnect = action === "reconnect" && reconnectEnabled && (options.shouldReconnect?.(error) ?? true);
+      manualOpenRef.current = false;
+      terminalErrorRef.current = shouldReconnect ? null : error;
+      const socket2 = socketRef.current;
+      if (socket2 === null || !isSocketActive(socket2)) {
+        commitState((current) => ({
+          ...current,
+          lastChangedAt: Date.now(),
+          lastError: error,
+          status: shouldReconnect ? "reconnecting" : "error"
+        }));
+        if (shouldReconnect) {
+          reconnect.schedule(reconnectTrigger);
+        }
+        return;
+      }
+      pendingCloseActionRef.current = {
+        error,
+        reconnectTrigger: shouldReconnect ? reconnectTrigger : null
+      };
+      skipCloseReconnectRef.current = true;
+      suppressReconnectRef.current = true;
+      closeSocket();
+    }
+  );
   const parseMessage = useEffectEvent((event) => {
     const parser = options.parseMessage ?? defaultParseMessage;
     return parser(event);
@@ -755,9 +845,26 @@ var useWebSocket = (options) => {
     socketKeyRef.current = null;
     heartbeat.stop();
     updateBufferedAmount();
+    const pendingCloseAction = pendingCloseActionRef.current;
+    pendingCloseActionRef.current = null;
     const terminalError = terminalErrorRef.current;
     const skipCloseReconnect = skipCloseReconnectRef.current;
     skipCloseReconnectRef.current = false;
+    if (pendingCloseAction !== null) {
+      suppressReconnectRef.current = false;
+      commitState((current) => ({
+        ...current,
+        lastChangedAt: Date.now(),
+        lastCloseEvent: event,
+        lastError: pendingCloseAction.error ?? current.lastError,
+        status: pendingCloseAction.reconnectTrigger === null ? "error" : "reconnecting"
+      }));
+      options.onClose?.(event);
+      if (pendingCloseAction.reconnectTrigger !== null) {
+        reconnect.schedule(pendingCloseAction.reconnectTrigger);
+      }
+      return;
+    }
     if (terminalError !== null) {
       suppressReconnectRef.current = false;
       commitState((current) => ({
@@ -916,7 +1023,7 @@ var useWebSocket = (options) => {
     if (socket2 === null) {
       return;
     }
-    if (socket2.readyState === WebSocket.OPEN || socket2.readyState === WebSocket.CONNECTING) {
+    if (isSocketActive(socket2)) {
       socket2.close();
     }
   }, []);