@replit/river 0.214.0 → 0.215.1

package/dist/router/index.d.cts

@@ -1,14 +1,14 @@
-import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-DqYQvm_L.cjs';
-export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-DqYQvm_L.cjs';
-import { TSchema, Static } from '@sinclair/typebox';
-import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-Dlsh5WDF.cjs';
-import { C as Connection, q as ServerHandshakeOptions } from '../transport-CxT7y8Qk.cjs';
-export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-CxT7y8Qk.cjs';
-import { S as ServerTransport } from '../server-BDSYa-CO.cjs';
+import { P as ProcedureHandlerContext, a as PayloadType, M as MaybeDisposable, A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-zaansuuR.cjs';
+export { B as BaseErrorSchemaType, C as CANCEL_CODE, c as Client, E as Err, d as ErrResult, e as INVALID_REQUEST_CODE, O as Ok, f as OkResult, g as ProcErrors, h as ProcHandler, i as ProcInit, j as ProcRequest, k as ProcResponse, l as ProcType, m as Procedure, n as ProcedureErrorSchemaType, o as ProcedureMap, R as RPCProcedure, p as RawReadable, q as Readable, r as ReadableBrokenError, s as ReadableResult, t as ReaderErrorSchema, u as ResponseData, v as Result, w as ResultUnwrapErr, x as ResultUnwrapOk, S as SerializedProcedureSchema, y as SerializedProcedureSchemaProtocolv1, z as SerializedServerSchema, D as SerializedServerSchemaProtocolv1, F as SerializedServiceSchema, G as SerializedServiceSchemaProtocolv1, H as Service, J as ServiceConfiguration, K as StreamProcedure, L as SubscriptionProcedure, U as UNCAUGHT_ERROR_CODE, N as UNEXPECTED_DISCONNECT_CODE, Q as UploadProcedure, V as ValidProcType, W as Writable, T as createClient, X as createServiceSchema, Y as flattenErrorType, Z as serializeSchema, _ as serializeSchemaV1Compat } from '../services-zaansuuR.cjs';
+import { Static, TSchema } from '@sinclair/typebox';
+import { T as TransportClientId, O as OpaqueTransportMessage } from '../message-DL74OqsX.cjs';
+import { C as Connection, q as ServerHandshakeOptions } from '../transport-CCBNESLA.cjs';
+export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-CCBNESLA.cjs';
+import { S as ServerTransport } from '../server-BkEzDYIv.cjs';
 import '@opentelemetry/api';
-import '../client-DXJRow2s.cjs';
-import '../index-D9R6UTMl.cjs';
-import '../adapter-DjiEwOYi.cjs';
+import '../client-B9aKi9Li.cjs';
+import '../index-Bf9PGbS4.cjs';
+import '../adapter-CjgmjtUJ.cjs';
 
 type StreamId = string;
 /**
@@ -74,6 +74,6 @@ declare function createServer<MetadataSchema extends TSchema, ParsedMetadata ext
     middlewares?: Array<Middleware>;
 }>): Server<Context, ParsedMetadata, Services>;
 
-var version = "0.214.0";
+var version = "0.215.1";
 
 export { MaybeDisposable, type Middleware, type MiddlewareContext, type MiddlewareParam, PayloadType, ProcedureHandlerContext, version as RIVER_VERSION, type Server, createServer };

package/dist/router/index.d.ts

@@ -1,14 +1,14 @@
-import { A as AnyServiceSchemaMap, M as MaybeDisposable, P as ProcedureHandlerContext, a as PayloadType, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-D47_GPCH.js';
-export { J as BaseErrorSchemaType, G as CANCEL_CODE, L as Client, N as Err, T as ErrResult, F as INVALID_REQUEST_CODE, O as Ok, X as OkResult, h as ProcErrors, d as ProcHandler, e as ProcInit, f as ProcRequest, g as ProcResponse, i as ProcType, z as Procedure, B as ProcedureErrorSchemaType, r as ProcedureMap, R as RPCProcedure, y as RawReadable, v as Readable, x as ReadableBrokenError, w as ReadableResult, H as ReaderErrorSchema, _ as ResponseData, Q as Result, Z as ResultUnwrapErr, Y as ResultUnwrapOk, m as SerializedProcedureSchema, q as SerializedProcedureSchemaProtocolv1, k as SerializedServerSchema, o as SerializedServerSchemaProtocolv1, l as SerializedServiceSchema, p as SerializedServiceSchemaProtocolv1, S as Service, c as ServiceConfiguration, u as StreamProcedure, t as SubscriptionProcedure, D as UNCAUGHT_ERROR_CODE, E as UNEXPECTED_DISCONNECT_CODE, U as UploadProcedure, V as ValidProcType, W as Writable, K as createClient, j as createServiceSchema, C as flattenErrorType, s as serializeSchema, n as serializeSchemaV1Compat } from '../services-D47_GPCH.js';
-import { TSchema, Static } from '@sinclair/typebox';
-import { c as TransportClientId, b as OpaqueTransportMessage } from '../message-Dlsh5WDF.js';
-import { C as Connection, q as ServerHandshakeOptions } from '../transport-pdbkDzmJ.js';
-export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-pdbkDzmJ.js';
-import { S as ServerTransport } from '../server-DFOzjvLh.js';
+import { P as ProcedureHandlerContext, a as PayloadType, M as MaybeDisposable, A as AnyServiceSchemaMap, I as InstantiatedServiceSchemaMap, b as AnyProcedure } from '../services-KdKBWdJr.js';
+export { B as BaseErrorSchemaType, C as CANCEL_CODE, c as Client, E as Err, d as ErrResult, e as INVALID_REQUEST_CODE, O as Ok, f as OkResult, g as ProcErrors, h as ProcHandler, i as ProcInit, j as ProcRequest, k as ProcResponse, l as ProcType, m as Procedure, n as ProcedureErrorSchemaType, o as ProcedureMap, R as RPCProcedure, p as RawReadable, q as Readable, r as ReadableBrokenError, s as ReadableResult, t as ReaderErrorSchema, u as ResponseData, v as Result, w as ResultUnwrapErr, x as ResultUnwrapOk, S as SerializedProcedureSchema, y as SerializedProcedureSchemaProtocolv1, z as SerializedServerSchema, D as SerializedServerSchemaProtocolv1, F as SerializedServiceSchema, G as SerializedServiceSchemaProtocolv1, H as Service, J as ServiceConfiguration, K as StreamProcedure, L as SubscriptionProcedure, U as UNCAUGHT_ERROR_CODE, N as UNEXPECTED_DISCONNECT_CODE, Q as UploadProcedure, V as ValidProcType, W as Writable, T as createClient, X as createServiceSchema, Y as flattenErrorType, Z as serializeSchema, _ as serializeSchemaV1Compat } from '../services-KdKBWdJr.js';
+import { Static, TSchema } from '@sinclair/typebox';
+import { T as TransportClientId, O as OpaqueTransportMessage } from '../message-DL74OqsX.js';
+import { C as Connection, q as ServerHandshakeOptions } from '../transport-kW92H6x-.js';
+export { r as createClientHandshakeOptions, s as createServerHandshakeOptions } from '../transport-kW92H6x-.js';
+import { S as ServerTransport } from '../server-_bfE7LYc.js';
 import '@opentelemetry/api';
-import '../client-Dw0JBBs3.js';
-import '../index-CSM8soK7.js';
-import '../adapter-Cp7_gIVA.js';
+import '../client-BOc8blGj.js';
+import '../index-DiAq34gk.js';
+import '../adapter-Dtt4bYL-.js';
 
 type StreamId = string;
 /**
@@ -74,6 +74,6 @@ declare function createServer<MetadataSchema extends TSchema, ParsedMetadata ext
     middlewares?: Array<Middleware>;
 }>): Server<Context, ParsedMetadata, Services>;
 
-var version = "0.214.0";
+var version = "0.215.1";
 
 export { MaybeDisposable, type Middleware, type MiddlewareContext, type MiddlewareParam, PayloadType, ProcedureHandlerContext, version as RIVER_VERSION, type Server, createServer };

package/dist/router/index.js

@@ -18,7 +18,7 @@ import {
   serializeSchema,
   serializeSchemaV1Compat,
   version
-} from "../chunk-LECVFB2G.js";
+} from "../chunk-NUGV5QWU.js";
 export {
   CANCEL_CODE,
   Err,

package/dist/{server-BDSYa-CO.d.cts → server-BkEzDYIv.d.cts}

@@ -1,8 +1,8 @@
-import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-CxT7y8Qk.cjs';
-import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-Dlsh5WDF.cjs';
+import { C as Connection, t as CommonSession, j as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, e as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-CCBNESLA.cjs';
+import { O as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, b as TransportMessage, T as TransportClientId } from './message-DL74OqsX.cjs';
 import { Static, TSchema } from '@sinclair/typebox';
-import { T as Tags } from './index-D9R6UTMl.cjs';
-import { S as SendResult } from './adapter-DjiEwOYi.cjs';
+import { T as Tags } from './index-Bf9PGbS4.cjs';
+import { S as SendResult } from './adapter-CjgmjtUJ.cjs';
 
 interface SessionWaitingForHandshakeListeners {
     onConnectionErrored: (err: unknown) => void;

package/dist/{server-DFOzjvLh.d.ts → server-_bfE7LYc.d.ts}

@@ -1,8 +1,8 @@
-import { C as Connection, t as CommonSession, d as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, c as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-pdbkDzmJ.js';
-import { b as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, a as TransportMessage, c as TransportClientId } from './message-Dlsh5WDF.js';
+import { C as Connection, t as CommonSession, j as SessionState, u as CommonSessionProps, T as Transport, v as ServerTransportOptions, q as ServerHandshakeOptions, w as ServerSession, e as ProvidedServerTransportOptions, D as DeleteSessionOptions } from './transport-kW92H6x-.js';
+import { O as OpaqueTransportMessage, H as HandshakeErrorResponseCodes, P as ProtocolVersion, b as TransportMessage, T as TransportClientId } from './message-DL74OqsX.js';
 import { Static, TSchema } from '@sinclair/typebox';
-import { T as Tags } from './index-CSM8soK7.js';
-import { S as SendResult } from './adapter-Cp7_gIVA.js';
+import { T as Tags } from './index-DiAq34gk.js';
+import { S as SendResult } from './adapter-Dtt4bYL-.js';
 
 interface SessionWaitingForHandshakeListeners {
     onConnectionErrored: (err: unknown) => void;

package/dist/{services-D47_GPCH.d.ts → services-KdKBWdJr.d.ts}

@@ -1,9 +1,9 @@
 import * as _sinclair_typebox from '@sinclair/typebox';
 import { TObject, TLiteral, TEnum, TString, TSchema, TNever, TUnion, Static } from '@sinclair/typebox';
 import { Span } from '@opentelemetry/api';
-import { c as TransportClientId } from './message-Dlsh5WDF.js';
-import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-pdbkDzmJ.js';
-import { C as ClientTransport } from './client-Dw0JBBs3.js';
+import { T as TransportClientId } from './message-DL74OqsX.js';
+import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-kW92H6x-.js';
+import { C as ClientTransport } from './client-BOc8blGj.js';
 
 /**
  * {@link UNCAUGHT_ERROR_CODE} is the code that is used when an error is thrown
@@ -1111,4 +1111,4 @@ declare class ServiceScaffold<Context extends MaybeDisposable, State extends obj
     };
 }
 
-export { type ReadableIterator as $, type AnyServiceSchemaMap as A, type ProcedureErrorSchemaType as B, flattenErrorType as C, UNCAUGHT_ERROR_CODE as D, UNEXPECTED_DISCONNECT_CODE as E, INVALID_REQUEST_CODE as F, CANCEL_CODE as G, ReaderErrorSchema as H, type InstantiatedServiceSchemaMap as I, type BaseErrorSchemaType as J, createClient as K, type Client as L, type MaybeDisposable as M, Err as N, Ok as O, type ProcedureHandlerContext as P, type Result as Q, type RpcProcedure as R, type Service as S, type ErrResult as T, type UploadProcedure as U, type ValidProcType as V, type Writable as W, type OkResult as X, type ResultUnwrapOk as Y, type ResultUnwrapErr as Z, type ResponseData as _, type PayloadType as a, type AnyProcedure as b, type ServiceConfiguration as c, type ProcHandler as d, type ProcInit as e, type ProcRequest as f, type ProcResponse as g, type ProcErrors as h, type ProcType as i, createServiceSchema as j, type SerializedServerSchema as k, type SerializedServiceSchema as l, type SerializedProcedureSchema as m, serializeSchemaV1Compat as n, type SerializedServerSchemaProtocolv1 as o, type SerializedServiceSchemaProtocolv1 as p, type SerializedProcedureSchemaProtocolv1 as q, type ProcedureMap as r, serializeSchema as s, type SubscriptionProcedure as t, type StreamProcedure as u, type Readable as v, type ReadableResult as w, ReadableBrokenError as x, ReadableImpl as y, Procedure as z };
+export { type ReadableIterator as $, type AnyServiceSchemaMap as A, type BaseErrorSchemaType as B, CANCEL_CODE as C, type SerializedServerSchemaProtocolv1 as D, Err as E, type SerializedServiceSchema as F, type SerializedServiceSchemaProtocolv1 as G, type Service as H, type InstantiatedServiceSchemaMap as I, type ServiceConfiguration as J, type StreamProcedure as K, type SubscriptionProcedure as L, type MaybeDisposable as M, UNEXPECTED_DISCONNECT_CODE as N, Ok as O, type ProcedureHandlerContext as P, type UploadProcedure as Q, type RpcProcedure as R, type SerializedProcedureSchema as S, createClient as T, UNCAUGHT_ERROR_CODE as U, type ValidProcType as V, type Writable as W, createServiceSchema as X, flattenErrorType as Y, serializeSchema as Z, serializeSchemaV1Compat as _, type PayloadType as a, type AnyProcedure as b, type Client as c, type ErrResult as d, INVALID_REQUEST_CODE as e, type OkResult as f, type ProcErrors as g, type ProcHandler as h, type ProcInit as i, type ProcRequest as j, type ProcResponse as k, type ProcType as l, Procedure as m, type ProcedureErrorSchemaType as n, type ProcedureMap as o, ReadableImpl as p, type Readable as q, ReadableBrokenError as r, type ReadableResult as s, ReaderErrorSchema as t, type ResponseData as u, type Result as v, type ResultUnwrapErr as w, type ResultUnwrapOk as x, type SerializedProcedureSchemaProtocolv1 as y, type SerializedServerSchema as z };

package/dist/{services-DqYQvm_L.d.cts → services-zaansuuR.d.cts}

@@ -1,9 +1,9 @@
 import * as _sinclair_typebox from '@sinclair/typebox';
 import { TObject, TLiteral, TEnum, TString, TSchema, TNever, TUnion, Static } from '@sinclair/typebox';
 import { Span } from '@opentelemetry/api';
-import { c as TransportClientId } from './message-Dlsh5WDF.cjs';
-import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-CxT7y8Qk.cjs';
-import { C as ClientTransport } from './client-DXJRow2s.cjs';
+import { T as TransportClientId } from './message-DL74OqsX.cjs';
+import { C as Connection, n as ClientHandshakeOptions, z as SessionId } from './transport-CCBNESLA.cjs';
+import { C as ClientTransport } from './client-B9aKi9Li.cjs';
 
 /**
  * {@link UNCAUGHT_ERROR_CODE} is the code that is used when an error is thrown
@@ -1111,4 +1111,4 @@ declare class ServiceScaffold<Context extends MaybeDisposable, State extends obj
     };
 }
 
-export { type ReadableIterator as $, type AnyServiceSchemaMap as A, type ProcedureErrorSchemaType as B, flattenErrorType as C, UNCAUGHT_ERROR_CODE as D, UNEXPECTED_DISCONNECT_CODE as E, INVALID_REQUEST_CODE as F, CANCEL_CODE as G, ReaderErrorSchema as H, type InstantiatedServiceSchemaMap as I, type BaseErrorSchemaType as J, createClient as K, type Client as L, type MaybeDisposable as M, Err as N, Ok as O, type ProcedureHandlerContext as P, type Result as Q, type RpcProcedure as R, type Service as S, type ErrResult as T, type UploadProcedure as U, type ValidProcType as V, type Writable as W, type OkResult as X, type ResultUnwrapOk as Y, type ResultUnwrapErr as Z, type ResponseData as _, type PayloadType as a, type AnyProcedure as b, type ServiceConfiguration as c, type ProcHandler as d, type ProcInit as e, type ProcRequest as f, type ProcResponse as g, type ProcErrors as h, type ProcType as i, createServiceSchema as j, type SerializedServerSchema as k, type SerializedServiceSchema as l, type SerializedProcedureSchema as m, serializeSchemaV1Compat as n, type SerializedServerSchemaProtocolv1 as o, type SerializedServiceSchemaProtocolv1 as p, type SerializedProcedureSchemaProtocolv1 as q, type ProcedureMap as r, serializeSchema as s, type SubscriptionProcedure as t, type StreamProcedure as u, type Readable as v, type ReadableResult as w, ReadableBrokenError as x, ReadableImpl as y, Procedure as z };
+export { type ReadableIterator as $, type AnyServiceSchemaMap as A, type BaseErrorSchemaType as B, CANCEL_CODE as C, type SerializedServerSchemaProtocolv1 as D, Err as E, type SerializedServiceSchema as F, type SerializedServiceSchemaProtocolv1 as G, type Service as H, type InstantiatedServiceSchemaMap as I, type ServiceConfiguration as J, type StreamProcedure as K, type SubscriptionProcedure as L, type MaybeDisposable as M, UNEXPECTED_DISCONNECT_CODE as N, Ok as O, type ProcedureHandlerContext as P, type UploadProcedure as Q, type RpcProcedure as R, type SerializedProcedureSchema as S, createClient as T, UNCAUGHT_ERROR_CODE as U, type ValidProcType as V, type Writable as W, createServiceSchema as X, flattenErrorType as Y, serializeSchema as Z, serializeSchemaV1Compat as _, type PayloadType as a, type AnyProcedure as b, type Client as c, type ErrResult as d, INVALID_REQUEST_CODE as e, type OkResult as f, type ProcErrors as g, type ProcHandler as h, type ProcInit as i, type ProcRequest as j, type ProcResponse as k, type ProcType as l, Procedure as m, type ProcedureErrorSchemaType as n, type ProcedureMap as o, ReadableImpl as p, type Readable as q, ReadableBrokenError as r, type ReadableResult as s, ReaderErrorSchema as t, type ResponseData as u, type Result as v, type ResultUnwrapErr as w, type ResultUnwrapOk as x, type SerializedProcedureSchemaProtocolv1 as y, type SerializedServerSchema as z };

package/dist/testUtil/index.cjs

@@ -360,6 +360,7 @@ var IdentifiedSession = class extends CommonSession {
   telemetry;
   to;
   protocolVersion;
+  listeners;
   /**
    * Index of the message we will send next (excluding handshake)
    */
@@ -383,7 +384,8 @@ var IdentifiedSession = class extends CommonSession {
       telemetry,
       log,
       protocolVersion,
-      seqSent: messagesSent
+      seqSent: messagesSent,
+      listeners
     } = props;
     super(props);
     this.id = id;
@@ -395,6 +397,7 @@ var IdentifiedSession = class extends CommonSession {
     this.log = log;
     this.protocolVersion = protocolVersion;
     this.seqSent = messagesSent;
+    this.listeners = listeners;
   }
   get loggingMetadata() {
     const metadata = {
@@ -411,7 +414,7 @@ var IdentifiedSession = class extends CommonSession {
     }
     return metadata;
   }
-  constructMsg(partialMsg) {
+  encodeMsg(partialMsg) {
     const msg = {
       ...partialMsg,
       id: generateId(),
@@ -420,18 +423,37 @@ var IdentifiedSession = class extends CommonSession {
       seq: this.seq,
       ack: this.ack
     };
+    const encoded = this.codec.toBuffer(msg);
+    if (!encoded.ok) {
+      this.listeners.onMessageSendFailure(
+        { ...partialMsg, seq: this.seq },
+        encoded.reason
+      );
+      return encoded;
+    }
     this.seq++;
-    return msg;
+    return {
+      ok: true,
+      value: {
+        id: msg.id,
+        seq: msg.seq,
+        msg: partialMsg,
+        data: encoded.value
+      }
+    };
   }
   nextSeq() {
     return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.sendBuffer.push(constructedMsg);
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    this.sendBuffer.push(encodeResult.value);
     return {
       ok: true,
-      value: constructedMsg.id
+      value: encodeResult.value.id
     };
   }
   _handleStateExit() {
@@ -464,23 +486,6 @@ var IdentifiedSessionWithGracePeriod = class extends IdentifiedSession {
     super._handleClose();
   }
 };
-function sendMessage(conn, codec, msg) {
-  const buff = codec.toBuffer(msg);
-  if (!buff.ok) {
-    return buff;
-  }
-  const sent = conn.send(buff.value);
-  if (!sent) {
-    return {
-      ok: false,
-      reason: "failed to send message"
-    };
-  }
-  return {
-    ok: true,
-    value: msg.id
-  };
-}
 
 // transport/sessionStateMachine/SessionConnecting.ts
 var SessionConnecting = class extends IdentifiedSessionWithGracePeriod {
@@ -559,7 +564,7 @@ function coerceErrorString(err) {
 }
 
 // package.json
-var version = "0.214.0";
+var version = "0.215.1";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -642,7 +647,21 @@ var SessionWaitingForHandshake = class extends CommonSession {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     this.conn.removeDataListener();
@@ -691,7 +710,21 @@ var SessionHandshaking = class extends IdentifiedSessionWithGracePeriod {
     this.listeners.onHandshake(parsedMsgRes.value);
   };
   sendHandshake(msg) {
-    return sendMessage(this.conn, this.codec, msg);
+    const buff = this.codec.toBuffer(msg);
+    if (!buff.ok) {
+      return buff;
+    }
+    const sent = this.conn.send(buff.value);
+    if (!sent) {
+      return {
+        ok: false,
+        reason: "failed to send handshake"
+      };
+    }
+    return {
+      ok: true,
+      value: msg.id
+    };
   }
   _handleStateExit() {
     super._handleStateExit();
@@ -726,28 +759,35 @@ var SessionConnected = class extends IdentifiedSession {
     }
     this.startMissingHeartbeatTimeout();
   }
-  assertSendOrdering(constructedMsg) {
-    if (constructedMsg.seq > this.seqSent + 1) {
-      const msg = `invariant violation: would have sent out of order msg (seq: ${constructedMsg.seq}, expected: ${this.seqSent} + 1)`;
+  assertSendOrdering(encodedMsg) {
+    if (encodedMsg.seq > this.seqSent + 1) {
+      const msg = `invariant violation: would have sent out of order msg (seq: ${encodedMsg.seq}, expected: ${this.seqSent} + 1)`;
       this.log?.error(msg, {
         ...this.loggingMetadata,
-        transportMessage: constructedMsg,
         tags: ["invariant-violation"]
       });
       throw new Error(msg);
     }
   }
   send(msg) {
-    const constructedMsg = this.constructMsg(msg);
-    this.assertSendOrdering(constructedMsg);
-    this.sendBuffer.push(constructedMsg);
-    const res = sendMessage(this.conn, this.codec, constructedMsg);
-    if (!res.ok) {
-      this.listeners.onMessageSendFailure(constructedMsg, res.reason);
-      return res;
+    const encodeResult = this.encodeMsg(msg);
+    if (!encodeResult.ok) {
+      return encodeResult;
+    }
+    const encodedMsg = encodeResult.value;
+    this.assertSendOrdering(encodedMsg);
+    this.sendBuffer.push(encodedMsg);
+    const sent = this.conn.send(encodedMsg.data);
+    if (!sent) {
+      const reason = "failed to send message";
+      this.listeners.onMessageSendFailure(
+        { ...encodedMsg.msg, seq: encodedMsg.seq },
+        reason
+      );
+      return { ok: false, reason };
     }
-    this.seqSent = constructedMsg.seq;
-    return res;
+    this.seqSent = encodedMsg.seq;
+    return { ok: true, value: encodedMsg.id };
   }
   constructor(props) {
     super(props);
@@ -765,10 +805,14 @@ var SessionConnected = class extends IdentifiedSession {
       );
       for (const msg of this.sendBuffer) {
         this.assertSendOrdering(msg);
-        const res = sendMessage(this.conn, this.codec, msg);
-        if (!res.ok) {
-          this.listeners.onMessageSendFailure(msg, res.reason);
-          return res;
+        const sent = this.conn.send(msg.data);
+        if (!sent) {
+          const reason = "failed to send buffered message";
+          this.listeners.onMessageSendFailure(
+            { ...msg.msg, seq: msg.seq },
+            reason
+          );
+          return { ok: false, reason };
         }
         this.seqSent = msg.seq;
       }
@@ -1176,6 +1220,17 @@ var Transport = class {
     const noConnectionSession = SessionStateGraph.transition.ConnectingToNoConnection(session, {
       onSessionGracePeriodElapsed: () => {
         this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
       }
     });
     this.updateSession(noConnectionSession);
@@ -1183,18 +1238,32 @@ var Transport = class {
   }
   onConnClosed(session) {
     let noConnectionSession;
+    const listeners = {
+      onSessionGracePeriodElapsed: () => {
+        this.onSessionGracePeriodElapsed(noConnectionSession);
+      },
+      onMessageSendFailure: (msg, reason) => {
+        this.log?.error(`failed to send message: ${reason}`, {
+          ...noConnectionSession.loggingMetadata,
+          transportMessage: msg
+        });
+        this.protocolError({
+          type: ProtocolError.MessageSendFailure,
+          message: reason
+        });
+        this.deleteSession(noConnectionSession, { unhealthy: true });
+      }
+    };
     if (session.state === "Handshaking" /* Handshaking */) {
-      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.HandshakingToNoConnection(
+        session,
+        listeners
+      );
     } else {
-      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(session, {
-        onSessionGracePeriodElapsed: () => {
-          this.onSessionGracePeriodElapsed(noConnectionSession);
-        }
-      });
+      noConnectionSession = SessionStateGraph.transition.ConnectedToNoConnection(
+        session,
+        listeners
+      );
     }
     this.updateSession(noConnectionSession);
     return noConnectionSession;
@@ -1298,6 +1367,10 @@ var LeakyBucketRateLimit = class {
     clearInterval(this.intervalHandle);
     this.intervalHandle = void 0;
   }
+  resetBudget() {
+    this.stopLeak();
+    this.budgetConsumed = 0;
+  }
   close() {
     this.stopLeak();
   }
@@ -1355,6 +1428,17 @@ var ClientTransport = class extends Transport {
       {
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(session);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...session.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(session, { unhealthy: true });
         }
       },
       this.options,
@@ -1419,6 +1503,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(handshakingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...handshakingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(handshakingSession, { unhealthy: true });
         }
       }
     );
@@ -1580,6 +1675,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(backingOffSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...backingOffSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(backingOffSession, { unhealthy: true });
         }
       }
     );
@@ -1643,6 +1749,17 @@ var ClientTransport = class extends Transport {
         },
         onSessionGracePeriodElapsed: () => {
           this.onSessionGracePeriodElapsed(connectingSession);
+        },
+        onMessageSendFailure: (msg, reason) => {
+          this.log?.error(`failed to send message: ${reason}`, {
+            ...connectingSession.loggingMetadata,
+            transportMessage: msg
+          });
+          this.protocolError({
+            type: ProtocolError.MessageSendFailure,
+            message: reason
+          });
+          this.deleteSession(connectingSession, { unhealthy: true });
         }
       }
     );
@@ -1973,6 +2090,17 @@ var ServerTransport = class extends Transport {
           {
             onSessionGracePeriodElapsed: () => {
               this.onSessionGracePeriodElapsed(noConnectionSession);
+            },
+            onMessageSendFailure: (msg2, reason) => {
+              this.log?.error(`failed to send message: ${reason}`, {
+                ...noConnectionSession.loggingMetadata,
+                transportMessage: msg2
+              });
+              this.protocolError({
+                type: ProtocolError.MessageSendFailure,
+                message: reason
+              });
+              this.deleteSession(noConnectionSession, { unhealthy: true });
             }
           }
         );
@@ -2823,6 +2951,8 @@ function dummySession() {
     "server",
     {
       onSessionGracePeriodElapsed: () => {
+      },
+      onMessageSendFailure: () => {
       }
     },
     testingSessionOptions,