dynamodb-reactive 0.2.2 → 0.2.4

package/dist/react.cjs

@@ -132,6 +132,7 @@ var WebSocketManager = class {
   connectionWaiters = /* @__PURE__ */ new Set();
   socketGeneration = 0;
   intentionalDisconnect = false;
+  lastConnectionError;
   constructor(config) {
     this.config = withDefaults(config);
   }
@@ -172,7 +173,11 @@ var WebSocketManager = class {
       this.connectionWaiters.add(waiter);
       options.signal?.addEventListener("abort", abort, { once: true });
       timeout = setTimeout(
-        () => finish(new Error(`Connection timed out after ${timeoutMs}ms`)),
+        () => finish(
+          new Error(`Connection timed out after ${timeoutMs}ms`, {
+            cause: this.lastConnectionError
+          })
+        ),
         timeoutMs
       );
     });
@@ -186,16 +191,30 @@ var WebSocketManager = class {
       await this.waitForConnectionId();
       return;
     }
+    if (this.state !== "reconnecting") this.lastConnectionError = void 0;
     this.intentionalDisconnect = false;
     const generation = ++this.socketGeneration;
     this.setState("connecting");
     let socket;
+    let retryable = true;
     try {
+      if (!this.config.webSocketFactory && typeof globalThis.WebSocket !== "function") {
+        retryable = false;
+        throw new Error(
+          "WebSocket is unavailable in this runtime. Enable the native WebSocket API (Node 20: --experimental-websocket), use Node 22 or newer, or configure webSocketFactory."
+        );
+      }
       const url = await this.buildUrl();
       if (generation !== this.socketGeneration || this.intentionalDisconnect) {
         return;
       }
-      const nextSocket = new WebSocket(url);
+      let nextSocket;
+      try {
+        nextSocket = this.config.webSocketFactory ? this.config.webSocketFactory(url) : new globalThis.WebSocket(url);
+      } catch (cause) {
+        retryable = false;
+        throw cause;
+      }
       socket = nextSocket;
       this.ws = nextSocket;
       this.startHandshakeTimer(nextSocket, generation);
@@ -209,10 +228,7 @@ var WebSocketManager = class {
         this.ws = null;
         this._connectionId = null;
         this.setState("disconnected");
-        this.invokeLifecycleCallback(
-          "onDisconnect",
-          this.config.onDisconnect
-        );
+        this.invokeLifecycleCallback("onDisconnect", this.config.onDisconnect);
         if (this.config.autoReconnect && !this.intentionalDisconnect) {
           this.scheduleReconnect();
         } else {
@@ -221,11 +237,9 @@ var WebSocketManager = class {
       };
       nextSocket.onerror = () => {
         if (!this.isCurrent(nextSocket, generation)) return;
-        this.invokeLifecycleCallback(
-          "onError",
-          this.config.onError,
-          new Error("WebSocket error")
-        );
+        const failure = new Error("WebSocket error");
+        this.lastConnectionError = failure;
+        this.invokeLifecycleCallback("onError", this.config.onError, failure);
       };
       nextSocket.onmessage = (event) => {
         if (!this.isCurrent(nextSocket, generation)) return;
@@ -234,13 +248,18 @@ var WebSocketManager = class {
     } catch (cause) {
       if (generation !== this.socketGeneration) return;
       const failure = cause instanceof Error ? cause : new Error(String(cause));
+      this.lastConnectionError = failure;
       this.clearConnectionTimers();
+      if (!retryable) {
+        this.clearTimers();
+        this.reconnectAttempts = 0;
+      }
       if (socket && this.ws === socket) socket.close();
       this.ws = null;
       this._connectionId = null;
       this.setState("disconnected");
       this.invokeLifecycleCallback("onError", this.config.onError, failure);
-      if (this.config.autoReconnect && !this.intentionalDisconnect) {
+      if (retryable && this.config.autoReconnect && !this.intentionalDisconnect) {
         this.scheduleReconnect();
         await this.waitForConnectionId();
         return;
@@ -305,6 +324,7 @@ var WebSocketManager = class {
       if (message.type === "pong") {
         if (!this._connectionId || this.state === "connected") return;
         this.clearHandshakeTimer();
+        this.lastConnectionError = void 0;
         this.reconnectAttempts = 0;
         this.setState("connected");
         this.startHeartbeat(socket, generation);
@@ -325,6 +345,7 @@ var WebSocketManager = class {
         return;
       }
       const failure = new Error("WebSocket handshake timed out");
+      this.lastConnectionError = failure;
       this.invokeLifecycleCallback("onError", this.config.onError, failure);
       socket.close(4e3, "Handshake timeout");
     }, this.config.handshakeTimeout);
@@ -332,7 +353,7 @@ var WebSocketManager = class {
   startHeartbeat(socket, generation) {
     this.clearHeartbeatTimer();
     this.heartbeatTimer = setInterval(() => {
-      if (!this.isCurrent(socket, generation) || this.state !== "connected" || socket.readyState !== WebSocket.OPEN) {
+      if (!this.isCurrent(socket, generation) || this.state !== "connected" || socket.readyState !== socket.OPEN) {
         return;
       }
       socket.send(JSON.stringify({ type: "ping" }));
@@ -349,7 +370,9 @@ var WebSocketManager = class {
   scheduleReconnect() {
     if (this.reconnectTimer || this.intentionalDisconnect) return;
     if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
-      const failure = new Error("Max reconnection attempts reached");
+      const failure = new Error("Max reconnection attempts reached", {
+        cause: this.lastConnectionError
+      });
       error(failure.message);
       this.setState("disconnected");
       this.rejectWaiters(failure);
@@ -358,10 +381,7 @@ var WebSocketManager = class {
     }
     this.setState("reconnecting");
     this.reconnectAttempts++;
-    const delay = Math.min(
-      this.config.reconnectDelay * 2 ** (this.reconnectAttempts - 1),
-      3e4
-    ) * (0.5 + Math.random() * 0.5);
+    const delay = Math.min(this.config.reconnectDelay * 2 ** (this.reconnectAttempts - 1), 3e4) * (0.5 + Math.random() * 0.5);
     this.reconnectTimer = setTimeout(() => {
       this.reconnectTimer = null;
       void this.connect().catch(() => void 0);