@replit/river 0.22.0 → 0.23.1

package/dist/transport/index.d.ts

@@ -1,7 +1,7 @@
 import { C as Codec } from '../types-3e5768ec.js';
-import { O as OpaqueTransportMessage, C as Connection, a as Session, T as TransportClientId, j as PropagationContext, P as PartialTransportMessage, S as SessionOptions, f as ClientHandshakeOptions, g as ServerHandshakeOptions } from '../index-2ece5234.js';
-export { H as HandshakeRequestMetadata, d as OpaqueTransportMessageSchema, b as ParsedHandshakeMetadata, e as TransportMessage, c as TransportMessageSchema, h as isStreamClose, i as isStreamOpen } from '../index-2ece5234.js';
-import '@sinclair/typebox';
+import { O as OpaqueTransportMessage, C as Connection, b as Session, T as TransportClientId, l as PropagationContext, P as PartialTransportMessage, f as ClientHandshakeOptions, c as ServerHandshakeOptions, m as ParsedMetadata, S as SessionOptions } from '../index-a6fe0edd.js';
+export { h as OpaqueTransportMessageSchema, i as TransportMessage, g as TransportMessageSchema, k as isStreamClose, j as isStreamOpen } from '../index-a6fe0edd.js';
+import { TSchema } from '@sinclair/typebox';
 import '@opentelemetry/api';
 
 type ConnectionStatus = 'connect' | 'disconnect';
@@ -98,13 +98,9 @@ declare class LeakyBucketRateLimit {
 type TransportStatus = 'open' | 'closed' | 'destroyed';
 type TransportOptions = SessionOptions;
 type ProvidedTransportOptions = Partial<TransportOptions>;
-type ClientTransportOptions = TransportOptions & ConnectionRetryOptions & {
-    handshake?: ClientHandshakeOptions;
-};
+type ClientTransportOptions = TransportOptions & ConnectionRetryOptions;
 type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;
-type ServerTransportOptions = TransportOptions & {
-    handshake?: ServerHandshakeOptions;
-};
+type ServerTransportOptions = TransportOptions;
 type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;
 /**
  * Transports manage the lifecycle (creation/deletion) of sessions and connections. Its responsibilities include:
@@ -246,7 +242,7 @@ declare abstract class Transport<ConnType extends Connection> {
      */
     destroy(): void;
 }
-declare abstract class ClientTransport<ConnType extends Connection> extends Transport<ConnType> {
+declare abstract class ClientTransport<ConnType extends Connection, HandshakeMetadata extends TSchema = TSchema> extends Transport<ConnType> {
     /**
      * The options for this transport.
      */
@@ -263,7 +259,12 @@ declare abstract class ClientTransport<ConnType extends Connection> extends Tran
      * tests or a special case where you don't want to reconnect.
      */
     reconnectOnConnectionDrop: boolean;
+    /**
+     * Optional handshake options for this client.
+     */
+    handshakeExtensions?: ClientHandshakeOptions<HandshakeMetadata>;
     constructor(clientId: TransportClientId, providedOptions?: ProvidedClientTransportOptions);
+    extendHandshake(options: ClientHandshakeOptions<HandshakeMetadata>): void;
     protected handleConnection(conn: ConnType, to: TransportClientId): void;
     receiveHandshakeResponseMessage(data: Uint8Array, conn: ConnType): Session<ConnType> | false;
     /**
@@ -284,13 +285,23 @@ declare abstract class ClientTransport<ConnType extends Connection> extends Tran
     protected sendHandshake(to: TransportClientId, conn: ConnType): Promise<boolean>;
     close(): void;
 }
-declare abstract class ServerTransport<ConnType extends Connection> extends Transport<ConnType> {
+declare abstract class ServerTransport<ConnType extends Connection, CustomMetadataSchema extends TSchema = TSchema> extends Transport<ConnType> {
     /**
      * The options for this transport.
      */
     protected options: ServerTransportOptions;
+    /**
+     * Optional handshake options for the server.
+     */
+    handshakeExtensions?: ServerHandshakeOptions<CustomMetadataSchema>;
+    /**
+     * A map of session handshake data for each session.
+     */
+    sessionHandshakeMetadata: WeakMap<Session<ConnType>, ParsedMetadata>;
     constructor(clientId: TransportClientId, providedOptions?: ProvidedServerTransportOptions);
+    extendHandshake(options: ServerHandshakeOptions<CustomMetadataSchema>): void;
     protected handleConnection(conn: ConnType): void;
+    private validateHandshakeMetadata;
     receiveHandshakeRequestMessage(data: Uint8Array, conn: ConnType): Promise<Session<ConnType> | false>;
 }
 

package/dist/transport/index.js

@@ -5,11 +5,11 @@ import {
   ServerTransport,
   Session,
   Transport
-} from "../chunk-ZPBWKBM5.js";
+} from "../chunk-AUH7GO52.js";
 import {
   OpaqueTransportMessageSchema,
   TransportMessageSchema
-} from "../chunk-3MFX6NXA.js";
+} from "../chunk-RBH47IKH.js";
 import "../chunk-OTQNCLFH.js";
 import "../chunk-3AW3IXVD.js";
 export {

package/dist/util/testHelpers.cjs

@@ -355,7 +355,7 @@ var log = void 0;
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.22.0";
+var version = "0.23.1";
 
 // tracing/index.ts
 function createSessionTelemetryInfo(session, propagationCtx) {
@@ -415,12 +415,6 @@ var Session = class {
    * for this session.
    */
   advertisedSessionId;
-  /**
-   * The metadata for this session, as parsed from the handshake.
-   *
-   * Will only ever be populated on the server side.
-   */
-  metadata;
   /**
    * Number of messages we've sent along this session (excluding handshake and acks)
    */
@@ -578,8 +572,8 @@ var Session = class {
   replaceWithNewConnection(newConn) {
     this.closeStaleConnection(newConn);
     this.cancelGrace();
-    this.connection = newConn;
     this.sendBufferedMessages(newConn);
+    this.connection = newConn;
   }
   beginGrace(cb) {
     log?.info(
@@ -786,7 +780,8 @@ function dummyCtx(state, session, extendedContext) {
     to: session.to,
     from: session.from,
     streamId: (0, import_nanoid2.nanoid)(),
-    session
+    session,
+    metadata: {}
   };
 }
 function asClientRpc(state, proc, extendedContext, session = dummySession()) {