@replit/river 0.23.12 → 0.23.13

package/dist/transport/index.cjs

@@ -224,7 +224,7 @@ var import_nanoid2 = require("nanoid");
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.23.12";
+var version = "0.23.13";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -559,84 +559,8 @@ var Session = class {
   }
 };
 
-// util/stringify.ts
-function coerceErrorString(err) {
-  if (err instanceof Error) {
-    return err.message || "unknown reason";
-  }
-  return `[coerced to error] ${String(err)}`;
-}
-
-// transport/rateLimit.ts
-var LeakyBucketRateLimit = class {
-  budgetConsumed;
-  intervalHandles;
-  options;
-  constructor(options) {
-    this.options = options;
-    this.budgetConsumed = /* @__PURE__ */ new Map();
-    this.intervalHandles = /* @__PURE__ */ new Map();
-  }
-  getBackoffMs(user) {
-    if (!this.budgetConsumed.has(user))
-      return 0;
-    const exponent = Math.max(0, this.getBudgetConsumed(user) - 1);
-    const jitter = Math.floor(Math.random() * this.options.maxJitterMs);
-    const backoffMs = Math.min(
-      this.options.baseIntervalMs * 2 ** exponent,
-      this.options.maxBackoffMs
-    );
-    return backoffMs + jitter;
-  }
-  get totalBudgetRestoreTime() {
-    return this.options.budgetRestoreIntervalMs * this.options.attemptBudgetCapacity;
-  }
-  consumeBudget(user) {
-    this.stopLeak(user);
-    this.budgetConsumed.set(user, this.getBudgetConsumed(user) + 1);
-  }
-  getBudgetConsumed(user) {
-    return this.budgetConsumed.get(user) ?? 0;
-  }
-  hasBudget(user) {
-    return this.getBudgetConsumed(user) < this.options.attemptBudgetCapacity;
-  }
-  startRestoringBudget(user) {
-    if (this.intervalHandles.has(user)) {
-      return;
-    }
-    const restoreBudgetForUser = () => {
-      const currentBudget = this.budgetConsumed.get(user);
-      if (!currentBudget) {
-        this.stopLeak(user);
-        return;
-      }
-      const newBudget = currentBudget - 1;
-      if (newBudget === 0) {
-        this.budgetConsumed.delete(user);
-        return;
-      }
-      this.budgetConsumed.set(user, newBudget);
-    };
-    const intervalHandle = setInterval(
-      restoreBudgetForUser,
-      this.options.budgetRestoreIntervalMs
-    );
-    this.intervalHandles.set(user, intervalHandle);
-  }
-  stopLeak(user) {
-    if (!this.intervalHandles.has(user)) {
-      return;
-    }
-    clearInterval(this.intervalHandles.get(user));
-    this.intervalHandles.delete(user);
-  }
-  close() {
-    for (const user of this.intervalHandles.keys()) {
-      this.stopLeak(user);
-    }
-  }
-};
+// transport/transport.ts
+var import_api3 = require("@opentelemetry/api");
 
 // codec/json.ts
 var encoder = new TextEncoder();
@@ -690,8 +614,7 @@ var NaiveJsonCodec = {
   }
 };
 
-// transport/transport.ts
-var import_api3 = require("@opentelemetry/api");
+// transport/options.ts
 var defaultTransportOptions = {
   heartbeatIntervalMs: 1e3,
   heartbeatsUntilDead: 2,
@@ -712,6 +635,8 @@ var defaultClientTransportOptions = {
 var defaultServerTransportOptions = {
   ...defaultTransportOptions
 };
+
+// transport/transport.ts
 var Transport = class {
   /**
    * The status of the transport.
@@ -866,6 +791,16 @@ var Transport = class {
    * @param connectedTo The peer we are connected to.
    */
   onDisconnect(conn, session) {
+    if (session.connection !== void 0 && session.connection.id !== conn.id) {
+      session.telemetry.span.addEvent("onDisconnect race");
+      this.log?.warn("onDisconnect race", {
+        clientId: this.clientId,
+        ...session.loggingMetadata,
+        ...conn.loggingMetadata,
+        tags: ["invariant-violation"]
+      });
+      return;
+    }
     conn.telemetry?.span.end();
     this.eventDispatcher.dispatchEvent("connectionStatus", {
       status: "disconnect",
@@ -873,6 +808,16 @@ var Transport = class {
     });
     session.connection = void 0;
     session.beginGrace(() => {
+      if (session.connection !== void 0) {
+        session.telemetry.span.addEvent("session grace period race");
+        this.log?.warn("session grace period race", {
+          clientId: this.clientId,
+          ...session.loggingMetadata,
+          ...conn.loggingMetadata,
+          tags: ["invariant-violation"]
+        });
+        return;
+      }
       session.telemetry.span.addEvent("session grace period expired");
       this.deleteSession({
         session,
@@ -1040,6 +985,91 @@ var Transport = class {
     return this.status;
   }
 };
+
+// transport/client.ts
+var import_api4 = require("@opentelemetry/api");
+
+// transport/rateLimit.ts
+var LeakyBucketRateLimit = class {
+  budgetConsumed;
+  intervalHandles;
+  options;
+  constructor(options) {
+    this.options = options;
+    this.budgetConsumed = /* @__PURE__ */ new Map();
+    this.intervalHandles = /* @__PURE__ */ new Map();
+  }
+  getBackoffMs(user) {
+    if (!this.budgetConsumed.has(user))
+      return 0;
+    const exponent = Math.max(0, this.getBudgetConsumed(user) - 1);
+    const jitter = Math.floor(Math.random() * this.options.maxJitterMs);
+    const backoffMs = Math.min(
+      this.options.baseIntervalMs * 2 ** exponent,
+      this.options.maxBackoffMs
+    );
+    return backoffMs + jitter;
+  }
+  get totalBudgetRestoreTime() {
+    return this.options.budgetRestoreIntervalMs * this.options.attemptBudgetCapacity;
+  }
+  consumeBudget(user) {
+    this.stopLeak(user);
+    this.budgetConsumed.set(user, this.getBudgetConsumed(user) + 1);
+  }
+  getBudgetConsumed(user) {
+    return this.budgetConsumed.get(user) ?? 0;
+  }
+  hasBudget(user) {
+    return this.getBudgetConsumed(user) < this.options.attemptBudgetCapacity;
+  }
+  startRestoringBudget(user) {
+    if (this.intervalHandles.has(user)) {
+      return;
+    }
+    const restoreBudgetForUser = () => {
+      const currentBudget = this.budgetConsumed.get(user);
+      if (!currentBudget) {
+        this.stopLeak(user);
+        return;
+      }
+      const newBudget = currentBudget - 1;
+      if (newBudget === 0) {
+        this.budgetConsumed.delete(user);
+        return;
+      }
+      this.budgetConsumed.set(user, newBudget);
+    };
+    const intervalHandle = setInterval(
+      restoreBudgetForUser,
+      this.options.budgetRestoreIntervalMs
+    );
+    this.intervalHandles.set(user, intervalHandle);
+  }
+  stopLeak(user) {
+    if (!this.intervalHandles.has(user)) {
+      return;
+    }
+    clearInterval(this.intervalHandles.get(user));
+    this.intervalHandles.delete(user);
+  }
+  close() {
+    for (const user of this.intervalHandles.keys()) {
+      this.stopLeak(user);
+    }
+  }
+};
+
+// util/stringify.ts
+function coerceErrorString(err) {
+  if (err instanceof Error) {
+    return err.message || "unknown reason";
+  }
+  return `[coerced to error] ${String(err)}`;
+}
+
+// transport/client.ts
+var import_value2 = require("@sinclair/typebox/value");
 var ClientTransport = class extends Transport {
   /**
    * The options for this transport.
@@ -1100,7 +1130,7 @@ var ClientTransport = class extends Transport {
         const parsed = this.parseMsg(data2, conn);
         if (!parsed) {
           conn.telemetry?.span.setStatus({
-            code: import_api3.SpanStatusCode.ERROR,
+            code: import_api4.SpanStatusCode.ERROR,
             message: "message parse failure"
           });
           conn.close();
@@ -1131,7 +1161,7 @@ var ClientTransport = class extends Transport {
     });
     conn.addErrorListener((err) => {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api4.SpanStatusCode.ERROR,
         message: "connection error"
       });
       this.log?.warn(
@@ -1149,7 +1179,7 @@ var ClientTransport = class extends Transport {
     const parsed = this.parseMsg(data, conn);
     if (!parsed) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api4.SpanStatusCode.ERROR,
         message: "non-transport message"
       });
       this.protocolError(
@@ -1158,9 +1188,9 @@ var ClientTransport = class extends Transport {
       );
       return false;
     }
-    if (!import_value.Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {
+    if (!import_value2.Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api4.SpanStatusCode.ERROR,
         message: "invalid handshake response"
       });
       this.log?.warn(`received invalid handshake resp`, {
@@ -1169,7 +1199,7 @@ var ClientTransport = class extends Transport {
         connectedTo: parsed.from,
         transportMessage: parsed,
         validationErrors: [
-          ...import_value.Value.Errors(
+          ...import_value2.Value.Errors(
             ControlMessageHandshakeResponseSchema,
             parsed.payload
           )
@@ -1183,7 +1213,7 @@ var ClientTransport = class extends Transport {
     }
     if (!parsed.payload.status.ok) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api4.SpanStatusCode.ERROR,
         message: "handshake rejected"
       });
       this.log?.warn(`received handshake rejection`, {
@@ -1205,7 +1235,7 @@ var ClientTransport = class extends Transport {
         closeHandshakingConnection: true
       });
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api4.SpanStatusCode.ERROR,
         message: "session id mismatch"
       });
       this.log?.warn(`handshake from ${parsed.from} session id mismatch`, {
@@ -1305,7 +1335,7 @@ var ClientTransport = class extends Transport {
         } catch (err) {
           const errStr = coerceErrorString(err);
           span.recordException(errStr);
-          span.setStatus({ code: import_api3.SpanStatusCode.ERROR });
+          span.setStatus({ code: import_api4.SpanStatusCode.ERROR });
           throw err;
         } finally {
           span.end();
@@ -1356,13 +1386,13 @@ var ClientTransport = class extends Transport {
     let metadata = void 0;
     if (this.handshakeExtensions) {
       metadata = await this.handshakeExtensions.construct();
-      if (!import_value.Value.Check(this.handshakeExtensions.schema, metadata)) {
+      if (!import_value2.Value.Check(this.handshakeExtensions.schema, metadata)) {
         this.log?.error(`constructed handshake metadata did not match schema`, {
           ...conn.loggingMetadata,
           clientId: this.clientId,
           connectedTo: to,
           validationErrors: [
-            ...import_value.Value.Errors(this.handshakeExtensions.schema, metadata)
+            ...import_value2.Value.Errors(this.handshakeExtensions.schema, metadata)
           ],
           tags: ["invariant-violation"]
         });
@@ -1371,7 +1401,7 @@ var ClientTransport = class extends Transport {
           "handshake metadata did not match schema"
         );
         conn.telemetry?.span.setStatus({
-          code: import_api3.SpanStatusCode.ERROR,
+          code: import_api4.SpanStatusCode.ERROR,
           message: "handshake meta mismatch"
         });
         return false;
@@ -1399,6 +1429,10 @@ var ClientTransport = class extends Transport {
     super.close();
   }
 };
+
+// transport/server.ts
+var import_api5 = require("@opentelemetry/api");
+var import_value3 = require("@sinclair/typebox/value");
 var ServerTransport = class extends Transport {
   /**
    * The options for this transport.
@@ -1447,7 +1481,7 @@ var ServerTransport = class extends Transport {
           }
         );
         conn.telemetry?.span.setStatus({
-          code: import_api3.SpanStatusCode.ERROR,
+          code: import_api5.SpanStatusCode.ERROR,
           message: "handshake timeout"
         });
         conn.close();
@@ -1498,7 +1532,7 @@ var ServerTransport = class extends Transport {
     });
     conn.addErrorListener((err) => {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api5.SpanStatusCode.ERROR,
         message: "connection error"
       });
       if (!session)
@@ -1512,9 +1546,9 @@ var ServerTransport = class extends Transport {
   async validateHandshakeMetadata(conn, session, rawMetadata, from) {
     let parsedMetadata = {};
     if (this.handshakeExtensions) {
-      if (!import_value.Value.Check(this.handshakeExtensions.schema, rawMetadata)) {
+      if (!import_value3.Value.Check(this.handshakeExtensions.schema, rawMetadata)) {
         conn.telemetry?.span.setStatus({
-          code: import_api3.SpanStatusCode.ERROR,
+          code: import_api5.SpanStatusCode.ERROR,
           message: "malformed handshake meta"
         });
         const reason = "received malformed handshake metadata";
@@ -1527,7 +1561,7 @@ var ServerTransport = class extends Transport {
           ...conn.loggingMetadata,
           clientId: this.clientId,
           validationErrors: [
-            ...import_value.Value.Errors(this.handshakeExtensions.schema, rawMetadata)
+            ...import_value3.Value.Errors(this.handshakeExtensions.schema, rawMetadata)
           ]
         });
         this.protocolError(ProtocolError.HandshakeFailed, reason);
@@ -1541,7 +1575,7 @@ var ServerTransport = class extends Transport {
       if (parsedMetadata === false) {
         const reason = "rejected by handshake handler";
         conn.telemetry?.span.setStatus({
-          code: import_api3.SpanStatusCode.ERROR,
+          code: import_api5.SpanStatusCode.ERROR,
           message: reason
         });
         const responseMsg = handshakeResponseMessage(this.clientId, from, {
@@ -1563,7 +1597,7 @@ var ServerTransport = class extends Transport {
     const parsed = this.parseMsg(data, conn);
     if (!parsed) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api5.SpanStatusCode.ERROR,
         message: "non-transport message"
       });
       this.protocolError(
@@ -1572,9 +1606,9 @@ var ServerTransport = class extends Transport {
       );
       return false;
     }
-    if (!import_value.Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {
+    if (!import_value3.Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api5.SpanStatusCode.ERROR,
         message: "invalid handshake request"
       });
       const reason = "received invalid handshake msg";
@@ -1590,7 +1624,7 @@ var ServerTransport = class extends Transport {
         // before passing it to user-land
         transportMessage: parsed,
         validationErrors: [
-          ...import_value.Value.Errors(ControlMessageHandshakeRequestSchema, parsed.payload)
+          ...import_value3.Value.Errors(ControlMessageHandshakeRequestSchema, parsed.payload)
         ]
       });
       this.protocolError(
@@ -1602,7 +1636,7 @@ var ServerTransport = class extends Transport {
     const gotVersion = parsed.payload.protocolVersion;
     if (gotVersion !== PROTOCOL_VERSION) {
       conn.telemetry?.span.setStatus({
-        code: import_api3.SpanStatusCode.ERROR,
+        code: import_api5.SpanStatusCode.ERROR,
         message: "incorrect protocol version"
       });
       const reason = `incorrect version (got: ${gotVersion} wanted ${PROTOCOL_VERSION})`;