@replit/river 0.12.5 → 0.12.6

package/dist/{chunk-2PKONGC3.js → chunk-4SDJ5VN4.js}

@@ -204,9 +204,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -222,8 +220,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -314,15 +312,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -392,13 +381,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -437,7 +423,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/{chunk-7J66WOIC.js → chunk-PBPXYLI6.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IIBVKYDB.js";
 import {
   Connection
-} from "./chunk-2PKONGC3.js";
+} from "./chunk-4SDJ5VN4.js";
 
 // transport/impls/uds/connection.ts
 var UdsConnection = class extends Connection {

package/dist/{chunk-H4QR2K3T.js → chunk-Q7GL34DZ.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IIBVKYDB.js";
 import {
   Connection
-} from "./chunk-2PKONGC3.js";
+} from "./chunk-4SDJ5VN4.js";
 
 // transport/impls/stdio/connection.ts
 var StreamConnection = class extends Connection {

package/dist/{chunk-6OAKPO5R.js → chunk-VLBVQX5H.js}

@@ -1,6 +1,6 @@
 import {
   Connection
-} from "./chunk-2PKONGC3.js";
+} from "./chunk-4SDJ5VN4.js";
 
 // transport/impls/ws/connection.ts
 var WebSocketConnection = class extends Connection {

package/dist/{connection-64b9e0ff.d.ts → connection-1f9971d8.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection } from './index-1afc5580.js';
+import { C as Connection } from './index-54e0f99c.js';
 import { U as Uint32LengthPrefixFraming } from './messageFraming-b200ef25.js';
 
 declare class StreamConnection extends Connection {

package/dist/{connection-a6cc48e8.d.ts → connection-24d878ac.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection } from './index-1afc5580.js';
+import { C as Connection } from './index-54e0f99c.js';
 import { Socket } from 'node:net';
 import { U as Uint32LengthPrefixFraming } from './messageFraming-b200ef25.js';
 

package/dist/{connection-4a839b9a.d.ts → connection-ba37d174.d.ts}

@@ -1,5 +1,5 @@
 import WebSocket from 'isomorphic-ws';
-import { C as Connection } from './index-1afc5580.js';
+import { C as Connection } from './index-54e0f99c.js';
 
 declare class WebSocketConnection extends Connection {
     ws: WebSocket;

package/dist/{index-1afc5580.d.ts → index-54e0f99c.d.ts}

@@ -214,7 +214,6 @@ declare class Session<ConnType extends Connection> {
     addToSendBuff(msg: TransportMessage): void;
     closeStaleConnection(conn?: ConnType): void;
     replaceWithNewConnection(newConn: ConnType): void;
-    graceCb?: () => unknown;
     beginGrace(cb: () => void): void;
     cancelGrace(): void;
     close(): void;
@@ -333,7 +332,6 @@ declare abstract class Transport<ConnType extends Connection> {
      * @param clientId The client ID of this transport.
      */
     constructor(clientId: TransportClientId, providedOptions?: Partial<TransportOptions>);
-    private sessionByClientId;
     /**
      * This is called immediately after a new connection is established and we
      * may or may not know the identity of the connected client.
@@ -356,7 +354,7 @@ declare abstract class Transport<ConnType extends Connection> {
      * @param conn The connection object.
      * @param connectedTo The peer we are connected to.
      */
-    onDisconnect(conn: ConnType, connectedTo: TransportClientId): void;
+    protected onDisconnect(conn: ConnType, connectedTo: TransportClientId): void;
     /**
      * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.
      * @param msg The message to parse.
@@ -368,7 +366,7 @@ declare abstract class Transport<ConnType extends Connection> {
      * You generally shouldn't need to override this in downstream transport implementations.
      * @param msg The received message.
      */
-    handleMsg(msg: OpaqueTransportMessage): void;
+    protected handleMsg(msg: OpaqueTransportMessage): void;
     /**
      * Adds a listener to this transport.
      * @param the type of event to listen for
@@ -424,13 +422,13 @@ declare abstract class ClientTransport<ConnType extends Connection> extends Tran
      * @param to The client ID of the node to connect to.
      */
     connect(to: TransportClientId, attempt?: number): Promise<void>;
-    receiveWithBootSequence(conn: ConnType, sessionCb: (sess: Session<ConnType>) => void): (data: Uint8Array) => void;
-    onDisconnect(conn: ConnType, connectedTo: string): void;
+    private receiveWithBootSequence;
+    protected onDisconnect(conn: ConnType, connectedTo: string): void;
 }
 declare abstract class ServerTransport<ConnType extends Connection> extends Transport<ConnType> {
     constructor(clientId: TransportClientId, providedOptions?: Partial<TransportOptions>);
     protected handleConnection(conn: ConnType): void;
-    receiveWithBootSequence(conn: ConnType, sessionCb: (sess: Session<ConnType>) => void): (data: Uint8Array) => void;
+    protected receiveWithBootSequence(conn: ConnType, sessionCb: (sess: Session<ConnType>) => void): (data: Uint8Array) => void;
 }
 
 export { Connection as C, EventMap as E, OpaqueTransportMessage as O, PartialTransportMessage as P, ServerTransport as S, Transport as T, ClientTransport as a, TransportClientId as b, TransportOptions as c, Session as d, TransportMessageSchema as e, OpaqueTransportMessageSchema as f, TransportMessage as g, isStreamClose as h, isStreamOpen as i, EventTypes as j, EventHandler as k };

package/dist/router/index.d.cts

@@ -1,6 +1,6 @@
 import { A as AnyService, P as PayloadType, b as Result, R as RiverError, S as ServiceContext, d as ProcType, e as ProcInput, f as ProcOutput, g as ProcErrors, h as ProcHasInit, i as ProcInit } from '../builder-c593de11.js';
 export { E as Err, O as Ok, m as ProcHandler, k as ProcListing, a as Procedure, o as RiverErrorSchema, c as RiverUncaughtSchema, l as Service, j as ServiceBuilder, n as ServiceContextWithState, U as UNCAUGHT_ERROR, V as ValidProcType, s as serializeService } from '../builder-c593de11.js';
-import { T as Transport, C as Connection, b as TransportClientId } from '../index-1afc5580.js';
+import { T as Transport, C as Connection, b as TransportClientId } from '../index-54e0f99c.js';
 import { Pushable } from 'it-pushable';
 import { Static } from '@sinclair/typebox';
 import '../types-3e5768ec.js';

package/dist/router/index.d.ts

@@ -1,6 +1,6 @@
 import { A as AnyService, P as PayloadType, b as Result, R as RiverError, S as ServiceContext, d as ProcType, e as ProcInput, f as ProcOutput, g as ProcErrors, h as ProcHasInit, i as ProcInit } from '../builder-c593de11.js';
 export { E as Err, O as Ok, m as ProcHandler, k as ProcListing, a as Procedure, o as RiverErrorSchema, c as RiverUncaughtSchema, l as Service, j as ServiceBuilder, n as ServiceContextWithState, U as UNCAUGHT_ERROR, V as ValidProcType, s as serializeService } from '../builder-c593de11.js';
-import { T as Transport, C as Connection, b as TransportClientId } from '../index-1afc5580.js';
+import { T as Transport, C as Connection, b as TransportClientId } from '../index-54e0f99c.js';
 import { Pushable } from 'it-pushable';
 import { Static } from '@sinclair/typebox';
 import '../types-3e5768ec.js';

package/dist/transport/impls/stdio/client.cjs

@@ -288,9 +288,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -306,8 +304,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -459,15 +457,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -537,13 +526,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -582,7 +568,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/stdio/client.d.cts

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { S as StreamConnection } from '../../../connection-1f9971d8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/client.d.ts

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { S as StreamConnection } from '../../../connection-1f9971d8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/client.js

@@ -1,11 +1,11 @@
 import {
   StreamConnection
-} from "../../../chunk-H4QR2K3T.js";
+} from "../../../chunk-Q7GL34DZ.js";
 import "../../../chunk-5IZ2UHWV.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ClientTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

package/dist/transport/impls/stdio/server.cjs

@@ -296,9 +296,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -314,8 +312,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -467,15 +465,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -545,13 +534,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -590,7 +576,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/stdio/server.d.cts

@@ -1,5 +1,5 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { S as StreamConnection } from '../../../connection-1f9971d8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/server.d.ts

@@ -1,5 +1,5 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { S as StreamConnection } from '../../../connection-64b9e0ff.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { S as StreamConnection } from '../../../connection-1f9971d8.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/stdio/server.js

@@ -1,11 +1,11 @@
 import {
   StreamConnection
-} from "../../../chunk-H4QR2K3T.js";
+} from "../../../chunk-Q7GL34DZ.js";
 import "../../../chunk-5IZ2UHWV.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ServerTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import "../../../chunk-H4BYJELI.js";

package/dist/transport/impls/uds/client.cjs

@@ -260,9 +260,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -278,8 +276,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -555,15 +553,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -633,13 +622,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -678,7 +664,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/uds/client.d.cts

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { U as UdsConnection } from '../../../connection-24d878ac.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import 'node:net';

package/dist/transport/impls/uds/client.d.ts

@@ -1,5 +1,5 @@
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { U as UdsConnection } from '../../../connection-24d878ac.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import 'node:net';

package/dist/transport/impls/uds/client.js

@@ -1,10 +1,10 @@
 import {
   UdsConnection
-} from "../../../chunk-7J66WOIC.js";
+} from "../../../chunk-PBPXYLI6.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ClientTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

package/dist/transport/impls/uds/server.cjs

@@ -296,9 +296,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -314,8 +312,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -467,15 +465,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -545,13 +534,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -590,7 +576,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/uds/server.d.cts

@@ -1,6 +1,6 @@
 import { Server, Socket } from 'node:net';
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { U as UdsConnection } from '../../../connection-24d878ac.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/uds/server.d.ts

@@ -1,6 +1,6 @@
 import { Server, Socket } from 'node:net';
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { U as UdsConnection } from '../../../connection-a6cc48e8.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { U as UdsConnection } from '../../../connection-24d878ac.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 import '../../../messageFraming-b200ef25.js';

package/dist/transport/impls/uds/server.js

@@ -1,10 +1,10 @@
 import {
   UdsConnection
-} from "../../../chunk-7J66WOIC.js";
+} from "../../../chunk-PBPXYLI6.js";
 import "../../../chunk-IIBVKYDB.js";
 import {
   ServerTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

package/dist/transport/impls/ws/client.cjs

@@ -288,9 +288,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -306,8 +304,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -459,15 +457,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -537,13 +526,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -582,7 +568,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/ws/client.d.cts

@@ -1,6 +1,6 @@
 import WebSocket from 'isomorphic-ws';
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { W as WebSocketConnection } from '../../../connection-4a839b9a.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { W as WebSocketConnection } from '../../../connection-ba37d174.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 

package/dist/transport/impls/ws/client.d.ts

@@ -1,6 +1,6 @@
 import WebSocket from 'isomorphic-ws';
-import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
-import { W as WebSocketConnection } from '../../../connection-4a839b9a.js';
+import { a as ClientTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
+import { W as WebSocketConnection } from '../../../connection-ba37d174.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 

package/dist/transport/impls/ws/client.js

@@ -1,9 +1,9 @@
 import {
   WebSocketConnection
-} from "../../../chunk-6OAKPO5R.js";
+} from "../../../chunk-VLBVQX5H.js";
 import {
   ClientTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

package/dist/transport/impls/ws/server.cjs

@@ -296,9 +296,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -314,8 +312,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -467,15 +465,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -545,13 +534,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -590,7 +576,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/impls/ws/server.d.cts

@@ -1,7 +1,7 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
 import { WebSocketServer } from 'ws';
 import { WebSocket } from 'isomorphic-ws';
-import { W as WebSocketConnection } from '../../../connection-4a839b9a.js';
+import { W as WebSocketConnection } from '../../../connection-ba37d174.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 

package/dist/transport/impls/ws/server.d.ts

@@ -1,7 +1,7 @@
-import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-1afc5580.js';
+import { S as ServerTransport, b as TransportClientId, c as TransportOptions } from '../../../index-54e0f99c.js';
 import { WebSocketServer } from 'ws';
 import { WebSocket } from 'isomorphic-ws';
-import { W as WebSocketConnection } from '../../../connection-4a839b9a.js';
+import { W as WebSocketConnection } from '../../../connection-ba37d174.js';
 import '../../../types-3e5768ec.js';
 import '@sinclair/typebox';
 

package/dist/transport/impls/ws/server.js

@@ -1,9 +1,9 @@
 import {
   WebSocketConnection
-} from "../../../chunk-6OAKPO5R.js";
+} from "../../../chunk-VLBVQX5H.js";
 import {
   ServerTransport
-} from "../../../chunk-2PKONGC3.js";
+} from "../../../chunk-4SDJ5VN4.js";
 import "../../../chunk-GZ7HCLLM.js";
 import "../../../chunk-XFFS4UOD.js";
 import {

package/dist/transport/index.cjs

@@ -318,9 +318,7 @@ var Session = class {
     this.cancelGrace();
     this.connection = newConn;
   }
-  graceCb;
   beginGrace(cb) {
-    this.graceCb = cb;
     this.disconnectionGrace = setTimeout(() => {
       this.close();
       cb();
@@ -336,8 +334,8 @@ var Session = class {
   close() {
     this.closeStaleConnection(this.connection);
     this.cancelGrace();
-    clearInterval(this.heartbeat);
     this.resetBufferedMessages();
+    clearInterval(this.heartbeat);
   }
   get connected() {
     return this.connection !== void 0;
@@ -489,15 +487,6 @@ var Transport = class {
     this.clientId = clientId;
     this.state = "open";
   }
-  sessionByClientId(clientId) {
-    const session = this.sessions.get(clientId);
-    if (!session) {
-      const err = `${this.clientId} -- (invariant violation) no existing session for ${clientId}`;
-      log?.error(err);
-      throw new Error(err);
-    }
-    return session;
-  }
   /**
    * Called when a new connection is established
    * and we know the identity of the connected client.
@@ -567,13 +556,10 @@ var Transport = class {
       status: "disconnect",
       conn
     });
-    if (this.state !== "open")
+    const session = this.sessions.get(connectedTo);
+    if (!session)
       return;
-    const session = this.sessionByClientId(connectedTo);
-    log?.info(
-      `${this.clientId} -- connection (id: ${conn.debugId}) disconnect from ${connectedTo}, ${SESSION_DISCONNECT_GRACE_MS}ms until session (id: ${session.debugId}) disconnect`
-    );
-    session.closeStaleConnection(conn);
+    session.connection = void 0;
     session.beginGrace(() => this.deleteSession(session));
   }
   /**
@@ -612,7 +598,12 @@ var Transport = class {
   handleMsg(msg) {
     if (this.state !== "open")
       return;
-    const session = this.sessionByClientId(msg.from);
+    const session = this.sessions.get(msg.from);
+    if (!session) {
+      const err = `${this.clientId} -- (invariant violation) no existing session for ${msg.from}`;
+      log?.error(err);
+      return;
+    }
     session.cancelGrace();
     log?.debug(`${this.clientId} -- received msg: ${JSON.stringify(msg)}`);
     if (msg.seq !== session.nextExpectedSeq) {

package/dist/transport/index.d.cts

@@ -1,3 +1,3 @@
-export { a as ClientTransport, C as Connection, k as EventHandler, E as EventMap, j as EventTypes, O as OpaqueTransportMessage, f as OpaqueTransportMessageSchema, S as ServerTransport, d as Session, T as Transport, b as TransportClientId, g as TransportMessage, e as TransportMessageSchema, h as isStreamClose, i as isStreamOpen } from '../index-1afc5580.js';
+export { a as ClientTransport, C as Connection, k as EventHandler, E as EventMap, j as EventTypes, O as OpaqueTransportMessage, f as OpaqueTransportMessageSchema, S as ServerTransport, d as Session, T as Transport, b as TransportClientId, g as TransportMessage, e as TransportMessageSchema, h as isStreamClose, i as isStreamOpen } from '../index-54e0f99c.js';
 import '../types-3e5768ec.js';
 import '@sinclair/typebox';

package/dist/transport/index.d.ts

@@ -1,3 +1,3 @@
-export { a as ClientTransport, C as Connection, k as EventHandler, E as EventMap, j as EventTypes, O as OpaqueTransportMessage, f as OpaqueTransportMessageSchema, S as ServerTransport, d as Session, T as Transport, b as TransportClientId, g as TransportMessage, e as TransportMessageSchema, h as isStreamClose, i as isStreamOpen } from '../index-1afc5580.js';
+export { a as ClientTransport, C as Connection, k as EventHandler, E as EventMap, j as EventTypes, O as OpaqueTransportMessage, f as OpaqueTransportMessageSchema, S as ServerTransport, d as Session, T as Transport, b as TransportClientId, g as TransportMessage, e as TransportMessageSchema, h as isStreamClose, i as isStreamOpen } from '../index-54e0f99c.js';
 import '../types-3e5768ec.js';
 import '@sinclair/typebox';

package/dist/transport/index.js

@@ -5,7 +5,7 @@ import {
   ServerTransport,
   Session,
   Transport
-} from "../chunk-2PKONGC3.js";
+} from "../chunk-4SDJ5VN4.js";
 import "../chunk-GZ7HCLLM.js";
 import {
   OpaqueTransportMessageSchema,

package/dist/util/testHelpers.d.cts

@@ -1,7 +1,7 @@
 import * as it_pushable from 'it-pushable';
 import WebSocket from 'isomorphic-ws';
 import http from 'node:http';
-import { P as PartialTransportMessage, T as Transport, C as Connection, O as OpaqueTransportMessage } from '../index-1afc5580.js';
+import { P as PartialTransportMessage, T as Transport, C as Connection, O as OpaqueTransportMessage } from '../index-54e0f99c.js';
 import { P as PayloadType, R as RiverError, a as Procedure, S as ServiceContext, b as Result, c as RiverUncaughtSchema } from '../builder-c593de11.js';
 import { Static } from '@sinclair/typebox';
 import net from 'node:net';

package/dist/util/testHelpers.d.ts

@@ -1,7 +1,7 @@
 import * as it_pushable from 'it-pushable';
 import WebSocket from 'isomorphic-ws';
 import http from 'node:http';
-import { P as PartialTransportMessage, T as Transport, C as Connection, O as OpaqueTransportMessage } from '../index-1afc5580.js';
+import { P as PartialTransportMessage, T as Transport, C as Connection, O as OpaqueTransportMessage } from '../index-54e0f99c.js';
 import { P as PayloadType, R as RiverError, a as Procedure, S as ServiceContext, b as Result, c as RiverUncaughtSchema } from '../builder-c593de11.js';
 import { Static } from '@sinclair/typebox';
 import net from 'node:net';

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@replit/river",
   "description": "It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!",
-  "version": "0.12.5",
+  "version": "0.12.6",
   "type": "module",
   "exports": {
     ".": {