@replit/river 0.19.2 → 0.20.0

package/dist/{chunk-D5PVGZPQ.js → chunk-5WFL722S.js}

@@ -11,6 +11,12 @@ var TransportMessageSchema = (t) => Type.Object({
   procedureName: Type.Optional(Type.String()),
   streamId: Type.String(),
   controlFlags: Type.Integer(),
+  tracing: Type.Optional(
+    Type.Object({
+      traceparent: Type.String(),
+      tracestate: Type.String()
+    })
+  ),
   payload: t
 });
 var ControlMessageAckSchema = Type.Object({
@@ -48,7 +54,7 @@ var ControlMessagePayloadSchema = Type.Union([
 var OpaqueTransportMessageSchema = TransportMessageSchema(
   Type.Unknown()
 );
-function handshakeRequestMessage(from, to, sessionId, metadata) {
+function handshakeRequestMessage(from, to, sessionId, metadata, tracing) {
   return {
     id: nanoid(),
     from,
@@ -57,6 +63,7 @@ function handshakeRequestMessage(from, to, sessionId, metadata) {
     ack: 0,
     streamId: nanoid(),
     controlFlags: 0,
+    tracing,
     payload: {
       type: "HANDSHAKE_REQ",
       protocolVersion: PROTOCOL_VERSION,
@@ -104,6 +111,11 @@ function coerceErrorString(err) {
   return `[coerced to error] ${String(err)}`;
 }
 
+// tracing/index.ts
+import { trace } from "@opentelemetry/api";
+var tracer = trace.getTracer("river");
+var tracing_default = tracer;
+
 export {
   TransportMessageSchema,
   PROTOCOL_VERSION,
@@ -116,5 +128,7 @@ export {
   isAck,
   isStreamOpen,
   isStreamClose,
+  tracing_default,
   coerceErrorString
 };
+//# sourceMappingURL=chunk-5WFL722S.js.map

package/dist/chunk-5WFL722S.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/message.ts","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { Connection, Session } from './session';\n\n/**\n * Control flags for transport messages.\n * An RPC message is coded with StreamOpenBit | StreamClosedBit.\n * Streams are expected to start with StreamOpenBit sent and the client SHOULD send an empty\n * message with StreamClosedBit to close the stream handler on the server, indicating that\n * it will not be using the stream anymore.\n */\nexport const enum ControlFlags {\n  AckBit = 0b0001,\n  StreamOpenBit = 0b0010,\n  StreamClosedBit = 0b0100,\n}\n\n/**\n * Metadata associated with a handshake request, as sent by the client.\n *\n * You should use declaration merging to extend this interface\n * with whatever you need. For example, if you need to store an\n * identifier for the client, you could do:\n * ```\n * declare module '@replit/river' {\n *   interface HandshakeMetadataClient {\n *     id: string;\n *   }\n * }\n * ```\n */\n// eslint-disable-next-line @typescript-eslint/no-empty-interface\nexport interface HandshakeRequestMetadata {}\n\n/**\n * Metadata associated with a handshake response, after the server\n * has processed the data in {@link HandshakeRequestMetadata}. This\n * is a separate interface for multiple reasons, but one of the main\n * ones is that the server should remove any sensitive data from the\n * client's request metadata before storing it in the session, that\n * way no secrets are persisted in memory.\n *\n * You should use declaration merging to extend this interface\n * with whatever you need. For example, if you need to store an\n * identifier for the client, you could do:\n * ```\n * declare module '@replit/river' {\n *   interface HandshakeMetadataServer {\n *     id: string;\n *   }\n * }\n * ```\n */\n// eslint-disable-next-line @typescript-eslint/no-empty-interface\nexport interface ParsedHandshakeMetadata {}\n\n/**\n * Options for extending the client handshake process.\n */\nexport interface ClientHandshakeOptions {\n  /**\n   * Schema for the metadata that the client sends to the server\n   * during the handshake.\n   *\n   * Needs to match {@link HandshakeRequestMetadata}.\n   */\n  schema: TSchema;\n\n  /**\n   * Gets the {@link HandshakeRequestMetadata} to send to the server.\n   */\n  get: () => HandshakeRequestMetadata | Promise<HandshakeRequestMetadata>;\n}\n\n/**\n * Options for extending the server handshake process.\n */\nexport interface ServerHandshakeOptions {\n  /**\n   * Schema for the metadata that the server receives from the client\n   * during the handshake.\n   *\n   * Needs to match {@link HandshakeRequestMetadata}.\n   */\n  requestSchema: TSchema;\n\n  /**\n   * Schema for the transformed metadata that is then associated with the\n   * client's session.\n   *\n   * Needs to match {@link ParsedHandshakeMetadata}.\n   */\n  parsedSchema: TSchema;\n\n  /**\n   * Parses the {@link HandshakeRequestMetadata} sent by the client, transforming\n   * it into {@link ParsedHandshakeMetadata}.\n   *\n   * May return `false` if the client should be rejected.\n   *\n   * @param metadata - The metadata sent by the client.\n   * @param session - The session that the client would be associated with.\n   * @param isReconnect - Whether the client is reconnecting to the session,\n   *                      or if this is a new session.\n   */\n  parse: (\n    metadata: HandshakeRequestMetadata,\n    session: Session<Connection>,\n    isReconnect: boolean,\n  ) =>\n    | false\n    | ParsedHandshakeMetadata\n    | Promise<false | ParsedHandshakeMetadata>;\n}\n\n/**\n * Generic Typebox schema for a transport message.\n * @template T The type of the payload.\n * @param {T} t The payload schema.\n * @returns The transport message schema.\n */\nexport const TransportMessageSchema = <T extends TSchema>(t: T) =>\n  Type.Object({\n    id: Type.String(),\n    from: Type.String(),\n    to: Type.String(),\n    seq: Type.Integer(),\n    ack: Type.Integer(),\n    serviceName: Type.Optional(Type.String()),\n    procedureName: Type.Optional(Type.String()),\n    streamId: Type.String(),\n    controlFlags: Type.Integer(),\n    tracing: Type.Optional(\n      Type.Object({\n        traceparent: Type.String(),\n        tracestate: Type.String(),\n      }),\n    ),\n    payload: t,\n  });\n\n/**\n * Defines the schema for a transport acknowledgement message. This is never constructed manually\n * and is only used internally by the library for tracking inflight messages.\n * @returns The transport message schema.\n */\nexport const ControlMessageAckSchema = Type.Object({\n  type: Type.Literal('ACK'),\n});\n\n/**\n * Defines the schema for a transport close message. This is never constructed manually and is only\n * used internally by the library for closing and cleaning up streams.\n */\nexport const ControlMessageCloseSchema = Type.Object({\n  type: Type.Literal('CLOSE'),\n});\n\nexport const PROTOCOL_VERSION = 'v1.1';\nexport const ControlMessageHandshakeRequestSchema = Type.Object({\n  type: Type.Literal('HANDSHAKE_REQ'),\n  protocolVersion: Type.String(),\n  sessionId: Type.String(),\n  metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const ControlMessageHandshakeResponseSchema = Type.Object({\n  type: Type.Literal('HANDSHAKE_RESP'),\n  status: Type.Union([\n    Type.Object({\n      ok: Type.Literal(true),\n      sessionId: Type.String(),\n    }),\n    Type.Object({\n      ok: Type.Literal(false),\n      reason: Type.String(),\n    }),\n  ]),\n});\n\nexport const ControlMessagePayloadSchema = Type.Union([\n  ControlMessageCloseSchema,\n  ControlMessageAckSchema,\n  ControlMessageHandshakeRequestSchema,\n  ControlMessageHandshakeResponseSchema,\n]);\n\n/**\n * Defines the schema for an opaque transport message that is agnostic to any\n * procedure/service.\n * @returns The transport message schema.\n */\nexport const OpaqueTransportMessageSchema = TransportMessageSchema(\n  Type.Unknown(),\n);\n\n/**\n * Represents a transport message. This is the same type as {@link TransportMessageSchema} but\n * we can't statically infer generics from generic Typebox schemas so we have to define it again here.\n *\n * TypeScript can't enforce types when a bitmask is involved, so these are the semantics of\n * `controlFlags`:\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `streamId` must be set to a unique value\n *   (suggestion: use `nanoid`).\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `serviceName` and `procedureName` must be set.\n * * If `controlFlags & StreamClosedBit == StreamClosedBit` and the kind is `stream` or `subscription`,\n *   `payload` should be discarded (usually contains a control message).\n * * If `controlFlags & AckBit == AckBit`, the message is an explicit acknowledgement message and doesn't\n *   contain any payload that is relevant to the application so should not be delivered.\n * @template Payload The type of the payload.\n */\nexport interface TransportMessage<Payload = unknown> {\n  id: string;\n  from: string;\n  to: string;\n  seq: number;\n  ack: number;\n  serviceName?: string;\n  procedureName?: string;\n  streamId: string;\n  controlFlags: number;\n  tracing?: { traceparent: string; tracestate: string };\n  payload: Payload;\n}\n\nexport type PartialTransportMessage<Payload = unknown> = Omit<\n  TransportMessage<Payload>,\n  'id' | 'from' | 'to' | 'seq' | 'ack'\n>;\n\nexport function handshakeRequestMessage(\n  from: TransportClientId,\n  to: TransportClientId,\n  sessionId: string,\n  metadata?: HandshakeRequestMetadata,\n  tracing?: { traceparent: string; tracestate: string },\n): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n  return {\n    id: nanoid(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: nanoid(),\n    controlFlags: 0,\n    tracing,\n    payload: {\n      type: 'HANDSHAKE_REQ',\n      protocolVersion: PROTOCOL_VERSION,\n      sessionId,\n      metadata,\n    } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n  };\n}\n\nexport function handshakeResponseMessage(\n  from: TransportClientId,\n  to: TransportClientId,\n  status: Static<typeof ControlMessageHandshakeResponseSchema>['status'],\n): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n  return {\n    id: nanoid(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: nanoid(),\n    controlFlags: 0,\n    payload: {\n      type: 'HANDSHAKE_RESP',\n      status,\n    } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n  };\n}\n\n/**\n * A type alias for a transport message with an opaque payload.\n * @template T - The type of the opaque payload.\n */\nexport type OpaqueTransportMessage = TransportMessage;\nexport type TransportClientId = string;\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an ack message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the AckBit, false otherwise.\n */\nexport function isAck(controlFlag: number): boolean {\n  /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n  return (controlFlag & ControlFlags.AckBit) === ControlFlags.AckBit;\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream open message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamOpenBit, false otherwise.\n */\nexport function isStreamOpen(controlFlag: number): boolean {\n  return (\n    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n    (controlFlag & ControlFlags.StreamOpenBit) === ControlFlags.StreamOpenBit\n  );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream close message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamCloseBit, false otherwise.\n */\nexport function isStreamClose(controlFlag: number): boolean {\n  return (\n    /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n    (controlFlag & ControlFlags.StreamClosedBit) ===\n    ControlFlags.StreamClosedBit\n  );\n}\n","export function coerceErrorString(err: unknown): string {\n  if (err instanceof Error) {\n    return err.message || 'unknown reason';\n  }\n\n  return `[coerced to error] ${String(err)}`;\n}\n","import { trace } from '@opentelemetry/api';\n\nconst tracer = trace.getTracer('river');\nexport default tracer;\n"],"mappings":";AAAA,SAAS,YAA6B;AACtC,SAAS,cAAc;AAwHhB,IAAM,yBAAyB,CAAoB,MACxD,KAAK,OAAO;AAAA,EACV,IAAI,KAAK,OAAO;AAAA,EAChB,MAAM,KAAK,OAAO;AAAA,EAClB,IAAI,KAAK,OAAO;AAAA,EAChB,KAAK,KAAK,QAAQ;AAAA,EAClB,KAAK,KAAK,QAAQ;AAAA,EAClB,aAAa,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EACxC,eAAe,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EAC1C,UAAU,KAAK,OAAO;AAAA,EACtB,cAAc,KAAK,QAAQ;AAAA,EAC3B,SAAS,KAAK;AAAA,IACZ,KAAK,OAAO;AAAA,MACV,aAAa,KAAK,OAAO;AAAA,MACzB,YAAY,KAAK,OAAO;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EACA,SAAS;AACX,CAAC;AAOI,IAAM,0BAA0B,KAAK,OAAO;AAAA,EACjD,MAAM,KAAK,QAAQ,KAAK;AAC1B,CAAC;AAMM,IAAM,4BAA4B,KAAK,OAAO;AAAA,EACnD,MAAM,KAAK,QAAQ,OAAO;AAC5B,CAAC;AAEM,IAAM,mBAAmB;AACzB,IAAM,uCAAuC,KAAK,OAAO;AAAA,EAC9D,MAAM,KAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,KAAK,OAAO;AAAA,EAC7B,WAAW,KAAK,OAAO;AAAA,EACvB,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,wCAAwC,KAAK,OAAO;AAAA,EAC/D,MAAM,KAAK,QAAQ,gBAAgB;AAAA,EACnC,QAAQ,KAAK,MAAM;AAAA,IACjB,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,IAAI;AAAA,MACrB,WAAW,KAAK,OAAO;AAAA,IACzB,CAAC;AAAA,IACD,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,KAAK;AAAA,MACtB,QAAQ,KAAK,OAAO;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AACH,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,+BAA+B;AAAA,EAC1C,KAAK,QAAQ;AACf;AAoCO,SAAS,wBACd,MACA,IACA,WACA,UACA,SACuE;AACvE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,yBACd,MACA,IACA,QACwE;AACxE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAcO,SAAS,MAAM,aAA8B;AAElD,UAAQ,cAAc,oBAAyB;AACjD;AAOO,SAAS,aAAa,aAA8B;AACzD;AAAA;AAAA,KAEG,cAAc,2BAAgC;AAAA;AAEnD;AAOO,SAAS,cAAc,aAA8B;AAC1D;AAAA;AAAA,KAEG,cAAc,6BACf;AAAA;AAEJ;;;AC3TO,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACNA,SAAS,aAAa;AAEtB,IAAM,SAAS,MAAM,UAAU,OAAO;AACtC,IAAO,kBAAQ;","names":[]}

package/dist/{chunk-NBE3D667.js → chunk-DT5JS6TM.js}

@@ -66,3 +66,4 @@ export {
   log,
   bindLogger
 };
+//# sourceMappingURL=chunk-DT5JS6TM.js.map

package/dist/chunk-DT5JS6TM.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../logging/log.ts"],"sourcesContent":["import { OpaqueTransportMessage } from '../transport';\nimport { PartialTransportMessage } from '../transport/message';\n\nconst LoggingLevels = {\n  debug: -1,\n  info: 0,\n  warn: 1,\n  error: 2,\n} as const;\ntype LoggingLevel = keyof typeof LoggingLevels;\n\nexport type LogFn = (\n  msg: string,\n  ctx?: MessageMetadata,\n  level?: LoggingLevel,\n) => void;\nexport type Logger = {\n  [key in LoggingLevel]: (msg: string, metadata?: MessageMetadata) => void;\n};\n\nexport type MessageMetadata = Record<string, unknown> &\n  Partial<{\n    protocolVersion: string;\n    clientId: string;\n    connectedTo: string;\n    sessionId: string;\n    connId: string;\n    fullTransportMessage: OpaqueTransportMessage;\n    partialTransportMessage: Partial<PartialTransportMessage>;\n  }>;\n\nclass BaseLogger implements Logger {\n  minLevel: LoggingLevel;\n  private output: LogFn;\n\n  constructor(output: LogFn, minLevel: LoggingLevel = 'info') {\n    this.minLevel = minLevel;\n    this.output = output;\n  }\n\n  debug(msg: string, metadata?: MessageMetadata) {\n    if (LoggingLevels[this.minLevel] <= LoggingLevels.debug) {\n      this.output(msg, metadata ?? {}, 'debug');\n    }\n  }\n\n  info(msg: string, metadata?: MessageMetadata) {\n    if (LoggingLevels[this.minLevel] <= LoggingLevels.info) {\n      this.output(msg, metadata ?? {}, 'info');\n    }\n  }\n\n  warn(msg: string, metadata?: MessageMetadata) {\n    if (LoggingLevels[this.minLevel] <= LoggingLevels.warn) {\n      this.output(msg, metadata ?? {}, 'warn');\n    }\n  }\n\n  error(msg: string, metadata?: MessageMetadata) {\n    if (LoggingLevels[this.minLevel] <= LoggingLevels.error) {\n      this.output(msg, metadata ?? {}, 'error');\n    }\n  }\n}\n\nexport const stringLogger: LogFn = (msg, _ctx, level = 'info') => {\n  console.log(`[river:${level}] ${msg}`);\n};\n\nconst colorMap = {\n  debug: '\\u001b[34m',\n  info: '\\u001b[32m',\n  warn: '\\u001b[33m',\n  error: '\\u001b[31m',\n};\n\nexport const coloredStringLogger: LogFn = (msg, _ctx, level = 'info') => {\n  const color = colorMap[level];\n  console.log(`[river:${color}${level}\\u001b[0m] ${msg}`);\n};\n\nexport const jsonLogger: LogFn = (msg, ctx, level) => {\n  console.log(JSON.stringify({ msg, ctx, level }));\n};\n\nexport let log: Logger | undefined = undefined;\n\nexport function bindLogger(fn: undefined, level?: LoggingLevel): undefined;\nexport function bindLogger(fn: LogFn | Logger, level?: LoggingLevel): Logger;\nexport function bindLogger(\n  fn: LogFn | Logger | undefined,\n  level?: LoggingLevel,\n): Logger | undefined {\n  if (typeof fn === 'function') {\n    log = new BaseLogger(fn, level);\n    return log;\n  }\n\n  log = fn;\n  return fn;\n}\n"],"mappings":";AAGA,IAAM,gBAAgB;AAAA,EACpB,OAAO;AAAA,EACP,MAAM;AAAA,EACN,MAAM;AAAA,EACN,OAAO;AACT;AAuBA,IAAM,aAAN,MAAmC;AAAA,EACjC;AAAA,EACQ;AAAA,EAER,YAAY,QAAe,WAAyB,QAAQ;AAC1D,SAAK,WAAW;AAChB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAM,KAAa,UAA4B;AAC7C,QAAI,cAAc,KAAK,QAAQ,KAAK,cAAc,OAAO;AACvD,WAAK,OAAO,KAAK,YAAY,CAAC,GAAG,OAAO;AAAA,IAC1C;AAAA,EACF;AAAA,EAEA,KAAK,KAAa,UAA4B;AAC5C,QAAI,cAAc,KAAK,QAAQ,KAAK,cAAc,MAAM;AACtD,WAAK,OAAO,KAAK,YAAY,CAAC,GAAG,MAAM;AAAA,IACzC;AAAA,EACF;AAAA,EAEA,KAAK,KAAa,UAA4B;AAC5C,QAAI,cAAc,KAAK,QAAQ,KAAK,cAAc,MAAM;AACtD,WAAK,OAAO,KAAK,YAAY,CAAC,GAAG,MAAM;AAAA,IACzC;AAAA,EACF;AAAA,EAEA,MAAM,KAAa,UAA4B;AAC7C,QAAI,cAAc,KAAK,QAAQ,KAAK,cAAc,OAAO;AACvD,WAAK,OAAO,KAAK,YAAY,CAAC,GAAG,OAAO;AAAA,IAC1C;AAAA,EACF;AACF;AAEO,IAAM,eAAsB,CAAC,KAAK,MAAM,QAAQ,WAAW;AAChE,UAAQ,IAAI,UAAU,KAAK,KAAK,GAAG,EAAE;AACvC;AAEA,IAAM,WAAW;AAAA,EACf,OAAO;AAAA,EACP,MAAM;AAAA,EACN,MAAM;AAAA,EACN,OAAO;AACT;AAEO,IAAM,sBAA6B,CAAC,KAAK,MAAM,QAAQ,WAAW;AACvE,QAAM,QAAQ,SAAS,KAAK;AAC5B,UAAQ,IAAI,UAAU,KAAK,GAAG,KAAK,YAAc,GAAG,EAAE;AACxD;AAEO,IAAM,aAAoB,CAAC,KAAK,KAAK,UAAU;AACpD,UAAQ,IAAI,KAAK,UAAU,EAAE,KAAK,KAAK,MAAM,CAAC,CAAC;AACjD;AAEO,IAAI,MAA0B;AAI9B,SAAS,WACd,IACA,OACoB;AACpB,MAAI,OAAO,OAAO,YAAY;AAC5B,UAAM,IAAI,WAAW,IAAI,KAAK;AAC9B,WAAO;AAAA,EACT;AAEA,QAAM;AACN,SAAO;AACT;","names":[]}

package/dist/{chunk-ZWPEZS27.js → chunk-HGBAUTIW.js}

@@ -1,6 +1,6 @@
 import {
   Connection
-} from "./chunk-YFPVQTWL.js";
+} from "./chunk-NYOK5VKK.js";
 
 // transport/transforms/messageFraming.ts
 import { Transform } from "node:stream";
@@ -98,3 +98,4 @@ var UdsConnection = class extends Connection {
 export {
   UdsConnection
 };
+//# sourceMappingURL=chunk-HGBAUTIW.js.map

package/dist/chunk-HGBAUTIW.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/transforms/messageFraming.ts","../transport/impls/uds/connection.ts"],"sourcesContent":["import { Transform, TransformCallback, TransformOptions } from 'node:stream';\n\nexport interface LengthEncodedOptions extends TransformOptions {\n  /** Maximum in-memory buffer size before we throw */\n  maxBufferSizeBytes: number;\n}\n\n/**\n * A transform stream that emits data each time a message with a network/BigEndian uint32 length prefix is received.\n * @extends Transform\n */\nexport class Uint32LengthPrefixFraming extends Transform {\n  receivedBuffer: Buffer;\n  maxBufferSizeBytes: number;\n\n  constructor({ maxBufferSizeBytes, ...options }: LengthEncodedOptions) {\n    super(options);\n    this.maxBufferSizeBytes = maxBufferSizeBytes;\n    this.receivedBuffer = Buffer.alloc(0);\n  }\n\n  _transform(chunk: Buffer, _encoding: BufferEncoding, cb: TransformCallback) {\n    if (\n      this.receivedBuffer.byteLength + chunk.byteLength >\n      this.maxBufferSizeBytes\n    ) {\n      const err = new Error(\n        `buffer overflow: ${this.receivedBuffer.byteLength}B > ${this.maxBufferSizeBytes}B`,\n      );\n\n      this.emit('error', err);\n      cb(err);\n      return;\n    }\n\n    this.receivedBuffer = Buffer.concat([this.receivedBuffer, chunk]);\n\n    // ensure there's enough for a length prefix\n    while (this.receivedBuffer.length > 4) {\n      // read length from buffer (accounting for uint32 prefix)\n      const claimedMessageLength = this.receivedBuffer.readUInt32BE(0) + 4;\n      if (this.receivedBuffer.length >= claimedMessageLength) {\n        // slice the buffer to extract the message\n        const message = this.receivedBuffer.subarray(4, claimedMessageLength);\n        this.push(message);\n        this.receivedBuffer =\n          this.receivedBuffer.subarray(claimedMessageLength);\n      } else {\n        // not enough data for a complete message, wait for more data\n        break;\n      }\n    }\n\n    cb();\n  }\n\n  _flush(cb: TransformCallback) {\n    // if there's any leftover data that doesn't form a complete message\n    if (this.receivedBuffer.length) {\n      this.emit('error', new Error('got incomplete message while flushing'));\n    }\n\n    this.receivedBuffer = Buffer.alloc(0);\n    cb();\n  }\n\n  _destroy(error: Error | null, callback: (error: Error | null) => void): void {\n    this.receivedBuffer = Buffer.alloc(0);\n    super._destroy(error, callback);\n  }\n}\n\nfunction createLengthEncodedStream(options?: Partial<LengthEncodedOptions>) {\n  return new Uint32LengthPrefixFraming({\n    maxBufferSizeBytes: options?.maxBufferSizeBytes ?? 16 * 1024 * 1024, // 16MB\n  });\n}\n\nexport const MessageFramer = {\n  createFramedStream: createLengthEncodedStream,\n  write: (buf: Uint8Array) => {\n    const lengthPrefix = Buffer.alloc(4);\n    lengthPrefix.writeUInt32BE(buf.length, 0);\n    return Buffer.concat([lengthPrefix, buf]);\n  },\n};\n","import { Connection } from '../../session';\nimport { type Socket } from 'node:net';\nimport stream from 'node:stream';\nimport {\n  MessageFramer,\n  Uint32LengthPrefixFraming,\n} from '../../transforms/messageFraming';\n\nexport class UdsConnection extends Connection {\n  sock: Socket;\n  input: stream.Readable;\n  framer: Uint32LengthPrefixFraming;\n\n  constructor(sock: Socket) {\n    super();\n    this.framer = MessageFramer.createFramedStream();\n    this.sock = sock;\n    this.input = sock.pipe(this.framer);\n  }\n\n  addDataListener(cb: (msg: Uint8Array) => void) {\n    this.input.on('data', cb);\n  }\n\n  removeDataListener(cb: (msg: Uint8Array) => void): void {\n    this.input.off('data', cb);\n  }\n\n  addCloseListener(cb: () => void): void {\n    this.sock.on('close', cb);\n  }\n\n  addErrorListener(cb: (err: Error) => void): void {\n    this.sock.on('error', cb);\n  }\n\n  send(payload: Uint8Array) {\n    if (this.framer.destroyed || !this.sock.writable) return false;\n    return this.sock.write(MessageFramer.write(payload));\n  }\n\n  close() {\n    this.sock.destroy();\n    this.framer.destroy();\n  }\n}\n"],"mappings":";;;;;AAAA,SAAS,iBAAsD;AAWxD,IAAM,4BAAN,cAAwC,UAAU;AAAA,EACvD;AAAA,EACA;AAAA,EAEA,YAAY,EAAE,oBAAoB,GAAG,QAAQ,GAAyB;AACpE,UAAM,OAAO;AACb,SAAK,qBAAqB;AAC1B,SAAK,iBAAiB,OAAO,MAAM,CAAC;AAAA,EACtC;AAAA,EAEA,WAAW,OAAe,WAA2B,IAAuB;AAC1E,QACE,KAAK,eAAe,aAAa,MAAM,aACvC,KAAK,oBACL;AACA,YAAM,MAAM,IAAI;AAAA,QACd,oBAAoB,KAAK,eAAe,UAAU,OAAO,KAAK,kBAAkB;AAAA,MAClF;AAEA,WAAK,KAAK,SAAS,GAAG;AACtB,SAAG,GAAG;AACN;AAAA,IACF;AAEA,SAAK,iBAAiB,OAAO,OAAO,CAAC,KAAK,gBAAgB,KAAK,CAAC;AAGhE,WAAO,KAAK,eAAe,SAAS,GAAG;AAErC,YAAM,uBAAuB,KAAK,eAAe,aAAa,CAAC,IAAI;AACnE,UAAI,KAAK,eAAe,UAAU,sBAAsB;AAEtD,cAAM,UAAU,KAAK,eAAe,SAAS,GAAG,oBAAoB;AACpE,aAAK,KAAK,OAAO;AACjB,aAAK,iBACH,KAAK,eAAe,SAAS,oBAAoB;AAAA,MACrD,OAAO;AAEL;AAAA,MACF;AAAA,IACF;AAEA,OAAG;AAAA,EACL;AAAA,EAEA,OAAO,IAAuB;AAE5B,QAAI,KAAK,eAAe,QAAQ;AAC9B,WAAK,KAAK,SAAS,IAAI,MAAM,uCAAuC,CAAC;AAAA,IACvE;AAEA,SAAK,iBAAiB,OAAO,MAAM,CAAC;AACpC,OAAG;AAAA,EACL;AAAA,EAEA,SAAS,OAAqB,UAA+C;AAC3E,SAAK,iBAAiB,OAAO,MAAM,CAAC;AACpC,UAAM,SAAS,OAAO,QAAQ;AAAA,EAChC;AACF;AAEA,SAAS,0BAA0B,SAAyC;AAC1E,SAAO,IAAI,0BAA0B;AAAA,IACnC,oBAAoB,SAAS,sBAAsB,KAAK,OAAO;AAAA;AAAA,EACjE,CAAC;AACH;AAEO,IAAM,gBAAgB;AAAA,EAC3B,oBAAoB;AAAA,EACpB,OAAO,CAAC,QAAoB;AAC1B,UAAM,eAAe,OAAO,MAAM,CAAC;AACnC,iBAAa,cAAc,IAAI,QAAQ,CAAC;AACxC,WAAO,OAAO,OAAO,CAAC,cAAc,GAAG,CAAC;AAAA,EAC1C;AACF;;;AC7EO,IAAM,gBAAN,cAA4B,WAAW;AAAA,EAC5C;AAAA,EACA;AAAA,EACA;AAAA,EAEA,YAAY,MAAc;AACxB,UAAM;AACN,SAAK,SAAS,cAAc,mBAAmB;AAC/C,SAAK,OAAO;AACZ,SAAK,QAAQ,KAAK,KAAK,KAAK,MAAM;AAAA,EACpC;AAAA,EAEA,gBAAgB,IAA+B;AAC7C,SAAK,MAAM,GAAG,QAAQ,EAAE;AAAA,EAC1B;AAAA,EAEA,mBAAmB,IAAqC;AACtD,SAAK,MAAM,IAAI,QAAQ,EAAE;AAAA,EAC3B;AAAA,EAEA,iBAAiB,IAAsB;AACrC,SAAK,KAAK,GAAG,SAAS,EAAE;AAAA,EAC1B;AAAA,EAEA,iBAAiB,IAAgC;AAC/C,SAAK,KAAK,GAAG,SAAS,EAAE;AAAA,EAC1B;AAAA,EAEA,KAAK,SAAqB;AACxB,QAAI,KAAK,OAAO,aAAa,CAAC,KAAK,KAAK;AAAU,aAAO;AACzD,WAAO,KAAK,KAAK,MAAM,cAAc,MAAM,OAAO,CAAC;AAAA,EACrD;AAAA,EAEA,QAAQ;AACN,SAAK,KAAK,QAAQ;AAClB,SAAK,OAAO,QAAQ;AAAA,EACtB;AACF;","names":[]}