@replit/river 0.25.0 → 0.25.1

package/dist/{chunk-6NPQ43ZP.js.map → chunk-SZ5NBBX7.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../transport/id.ts","../transport/message.ts","../package.json","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { customAlphabet } from 'nanoid';\n\nconst alphabet = customAlphabet(\n  '1234567890abcdefghijklmnopqrstuvxyzABCDEFGHIJKLMNOPQRSTUVXYZ',\n);\nexport const generateId = () => alphabet(12);\n","import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { PropagationContext } from '../tracing';\nimport { generateId } from './id';\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 * 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  /**\n   * Specifies what the server's expected session state (from the pov of the client). This can be\n   * used by the server to know whether this is a new or a reestablished connection, and whether it\n   * is compatible with what it already has.\n   */\n  expectedSessionState: Type.Object({\n    // what the client expects the server to send next\n    nextExpectedSeq: Type.Integer(),\n    // TODO: remove optional once we know all servers\n    // are nextSentSeq here\n    // what the server expects the client to send next\n    nextSentSeq: Type.Optional(Type.Integer()),\n  }),\n\n  metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const HandshakeErrorRetriableResponseCodes = Type.Union([\n  Type.Literal('SESSION_STATE_MISMATCH'),\n]);\n\nexport const HandshakeErrorFatalResponseCodes = Type.Union([\n  Type.Literal('MALFORMED_HANDSHAKE_META'),\n  Type.Literal('MALFORMED_HANDSHAKE'),\n  Type.Literal('PROTOCOL_VERSION_MISMATCH'),\n  Type.Literal('REJECTED_BY_CUSTOM_HANDLER'),\n]);\n\nexport const HandshakeErrorResponseCodes = Type.Union([\n  HandshakeErrorRetriableResponseCodes,\n  HandshakeErrorFatalResponseCodes,\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      // TODO: remove optional once we know all servers\n      // are sending code here\n      code: Type.Optional(HandshakeErrorResponseCodes),\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?: PropagationContext;\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,\n  to,\n  sessionId,\n  expectedSessionState,\n  metadata,\n  tracing,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  sessionId: string;\n  expectedSessionState: Static<\n    typeof ControlMessageHandshakeRequestSchema\n  >['expectedSessionState'];\n  metadata?: unknown;\n  tracing?: PropagationContext;\n}): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    tracing,\n    payload: {\n      type: 'HANDSHAKE_REQ',\n      protocolVersion: PROTOCOL_VERSION,\n      sessionId,\n      expectedSessionState,\n      metadata,\n    } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n  };\n}\n\n/**\n * This is a reason that can be given during the handshake to indicate that the peer has the wrong\n * session state.\n */\nexport const SESSION_STATE_MISMATCH = 'session state mismatch';\n\nexport function handshakeResponseMessage({\n  from,\n  to,\n  status,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  status: Static<typeof ControlMessageHandshakeResponseSchema>['status'];\n}): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    payload: {\n      type: 'HANDSHAKE_RESP',\n      status,\n    } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n  };\n}\n\nexport function closeStreamMessage(streamId: string): PartialTransportMessage {\n  return {\n    streamId,\n    controlFlags: ControlFlags.StreamClosedBit,\n    payload: {\n      type: 'CLOSE' as const,\n    } satisfies Static<typeof ControlMessagePayloadSchema>,\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","{\n  \"name\": \"@replit/river\",\n  \"description\": \"It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!\",\n  \"version\": \"0.25.0\",\n  \"type\": \"module\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/router/index.js\",\n      \"require\": \"./dist/router/index.cjs\"\n    },\n    \"./logging\": {\n      \"import\": \"./dist/logging/index.js\",\n      \"require\": \"./dist/logging/index.cjs\"\n    },\n    \"./codec\": {\n      \"import\": \"./dist/codec/index.js\",\n      \"require\": \"./dist/codec/index.cjs\"\n    },\n    \"./transport\": {\n      \"import\": \"./dist/transport/index.js\",\n      \"require\": \"./dist/transport/index.cjs\"\n    },\n    \"./transport/ws/client\": {\n      \"import\": \"./dist/transport/impls/ws/client.js\",\n      \"require\": \"./dist/transport/impls/ws/client.cjs\"\n    },\n    \"./transport/ws/server\": {\n      \"import\": \"./dist/transport/impls/ws/server.js\",\n      \"require\": \"./dist/transport/impls/ws/server.cjs\"\n    },\n    \"./transport/uds/client\": {\n      \"import\": \"./dist/transport/impls/uds/client.js\",\n      \"require\": \"./dist/transport/impls/uds/client.cjs\"\n    },\n    \"./transport/uds/server\": {\n      \"import\": \"./dist/transport/impls/uds/server.js\",\n      \"require\": \"./dist/transport/impls/uds/server.cjs\"\n    },\n    \"./test-util\": {\n      \"import\": \"./dist/util/testHelpers.js\",\n      \"require\": \"./dist/util/testHelpers.cjs\"\n    }\n  },\n  \"sideEffects\": [\n    \"./dist/logging/index.js\"\n  ],\n  \"files\": [\n    \"dist\"\n  ],\n  \"dependencies\": {\n    \"@msgpack/msgpack\": \"^3.0.0-beta2\",\n    \"it-pushable\": \"^3.2.3\",\n    \"nanoid\": \"^4.0.2\",\n    \"ws\": \"^8.17.0\"\n  },\n  \"peerDependencies\": {\n    \"@opentelemetry/api\": \"^1.7.0\",\n    \"@sinclair/typebox\": \"~0.32.8\"\n  },\n  \"devDependencies\": {\n    \"@opentelemetry/sdk-trace-base\": \"^1.24.1\",\n    \"@opentelemetry/sdk-trace-web\": \"^1.24.1\",\n    \"@opentelemetry/core\": \"^1.7.0\",\n    \"@types/ws\": \"^8.5.5\",\n    \"@typescript-eslint/eslint-plugin\": \"^7.8.0\",\n    \"@typescript-eslint/parser\": \"^7.8.0\",\n    \"@vitest/ui\": \"^1.3.1\",\n    \"eslint\": \"^8.57.0\",\n    \"eslint-config-prettier\": \"^9.1.0\",\n    \"eslint-plugin-prettier\": \"^5.1.3\",\n    \"prettier\": \"^3.0.0\",\n    \"tsup\": \"^7.2.0\",\n    \"typescript\": \"^5.4.5\",\n    \"vitest\": \"^1.3.1\"\n  },\n  \"scripts\": {\n    \"check\": \"tsc --noEmit && npm run format && npm run lint\",\n    \"format\": \"npx prettier . --check\",\n    \"format:fix\": \"npx prettier . --write\",\n    \"lint\": \"eslint .\",\n    \"lint:fix\": \"eslint . --fix\",\n    \"fix\": \"npm run format:fix && npm run lint:fix\",\n    \"build\": \"rm -rf dist && tsup && du -sh dist\",\n    \"prepack\": \"npm run build\",\n    \"release\": \"npm publish --access public\",\n    \"test:ui\": \"echo \\\"remember to go to /__vitest__ in the webview\\\" && vitest --ui --api.host 0.0.0.0 --api.port 3000\",\n    \"test\": \"vitest\",\n    \"test:single\": \"vitest run --reporter=dot\",\n    \"test:flake\": \"./flake.sh\",\n    \"bench\": \"vitest bench\"\n  },\n  \"engines\": {\n    \"node\": \">=16\"\n  },\n  \"keywords\": [\n    \"rpc\",\n    \"websockets\",\n    \"jsonschema\"\n  ],\n  \"author\": \"Jacky Zhao\",\n  \"license\": \"MIT\"\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 {\n  Context,\n  Span,\n  SpanKind,\n  context,\n  propagation,\n  trace,\n} from '@opentelemetry/api';\nimport { version as RIVER_VERSION } from '../package.json';\nimport { ValidProcType } from '../router';\nimport {\n  ClientTransport,\n  Connection,\n  OpaqueTransportMessage,\n} from '../transport';\n\nexport interface PropagationContext {\n  traceparent: string;\n  tracestate: string;\n}\n\nexport interface TelemetryInfo {\n  span: Span;\n  ctx: Context;\n}\n\nexport function getPropagationContext(\n  ctx: Context,\n): PropagationContext | undefined {\n  const tracing = {\n    traceparent: '',\n    tracestate: '',\n  };\n  propagation.inject(ctx, tracing);\n  return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n  sessionId: string,\n  to: string,\n  from: string,\n  propagationCtx?: PropagationContext,\n): TelemetryInfo {\n  const parentCtx = propagationCtx\n    ? propagation.extract(context.active(), propagationCtx)\n    : context.active();\n\n  const span = tracer.startSpan(\n    `session ${sessionId}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.session.id': sessionId,\n        'river.session.to': to,\n        'river.session.from': from,\n      },\n    },\n    parentCtx,\n  );\n\n  const ctx = trace.setSpan(parentCtx, span);\n\n  return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n  connection: Connection,\n  info: TelemetryInfo,\n): TelemetryInfo {\n  const span = tracer.startSpan(\n    `connection ${connection.id}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.connection.id': connection.id,\n      },\n      links: [{ context: info.span.spanContext() }],\n    },\n    info.ctx,\n  );\n\n  const ctx = trace.setSpan(info.ctx, span);\n\n  return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n  transport: ClientTransport<Connection>,\n  kind: ValidProcType,\n  serviceName: string,\n  procedureName: string,\n  streamId: string,\n): TelemetryInfo {\n  const baseCtx = context.active();\n  const span = tracer.startSpan(\n    `procedure call ${serviceName}.${procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': serviceName,\n        'river.method.name': procedureName,\n        'river.streamId': streamId,\n        'span.kind': 'client',\n      },\n      kind: SpanKind.CLIENT,\n    },\n    baseCtx,\n  );\n\n  const ctx = trace.setSpan(baseCtx, span);\n\n  transport.log?.info(`invoked ${serviceName}.${procedureName}`, {\n    clientId: transport.clientId,\n    transportMessage: {\n      procedureName,\n      serviceName,\n    },\n    telemetry: {\n      traceId: span.spanContext().traceId,\n      spanId: span.spanContext().spanId,\n    },\n  });\n  return { span, ctx };\n}\n\nexport function createHandlerSpan(\n  kind: ValidProcType,\n  message: OpaqueTransportMessage,\n  fn: (span: Span) => Promise<unknown>,\n) {\n  const ctx = message.tracing\n    ? propagation.extract(context.active(), message.tracing)\n    : context.active();\n\n  return tracer.startActiveSpan(\n    `procedure handler ${message.serviceName}.${message.procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': message.serviceName,\n        'river.method.name': message.procedureName,\n        'river.streamId': message.streamId,\n        'span.kind': 'server',\n      },\n      kind: SpanKind.SERVER,\n    },\n    ctx,\n    fn,\n  );\n}\n\nconst tracer = trace.getTracer('river', RIVER_VERSION);\nexport default tracer;\n"],"mappings":";AAAA,SAAS,sBAAsB;AAE/B,IAAM,WAAW;AAAA,EACf;AACF;AACO,IAAM,aAAa,MAAM,SAAS,EAAE;;;ACL3C,SAAS,YAA6B;AAuB/B,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;AAAA;AAAA;AAAA;AAAA;AAAA,EAMvB,sBAAsB,KAAK,OAAO;AAAA;AAAA,IAEhC,iBAAiB,KAAK,QAAQ;AAAA;AAAA;AAAA;AAAA,IAI9B,aAAa,KAAK,SAAS,KAAK,QAAQ,CAAC;AAAA,EAC3C,CAAC;AAAA,EAED,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,uCAAuC,KAAK,MAAM;AAAA,EAC7D,KAAK,QAAQ,wBAAwB;AACvC,CAAC;AAEM,IAAM,mCAAmC,KAAK,MAAM;AAAA,EACzD,KAAK,QAAQ,0BAA0B;AAAA,EACvC,KAAK,QAAQ,qBAAqB;AAAA,EAClC,KAAK,QAAQ,2BAA2B;AAAA,EACxC,KAAK,QAAQ,4BAA4B;AAC3C,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AACF,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;AAAA;AAAA,MAGpB,MAAM,KAAK,SAAS,2BAA2B;AAAA,IACjD,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,wBAAwB;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAS0E;AACxE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQO,SAAS,yBAAyB;AAAA,EACvC;AAAA,EACA;AAAA,EACA;AACF,GAI2E;AACzE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,mBAAmB,UAA2C;AAC5E,SAAO;AAAA,IACL;AAAA,IACA,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,IACR;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;;;ACtRE,cAAW;;;ACHN,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACNA;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAmBA,SAAS,sBACd,KACgC;AAChC,QAAM,UAAU;AAAA,IACd,aAAa;AAAA,IACb,YAAY;AAAA,EACd;AACA,cAAY,OAAO,KAAK,OAAO;AAC/B,SAAO;AACT;AAEO,SAAS,2BACd,WACA,IACA,MACA,gBACe;AACf,QAAM,YAAY,iBACd,YAAY,QAAQ,QAAQ,OAAO,GAAG,cAAc,IACpD,QAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,SAAS;AAAA,IACpB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB;AAAA,QACpB,oBAAoB;AAAA,QACpB,sBAAsB;AAAA,MACxB;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,WAAW,IAAI;AAEzC,SAAO,EAAE,MAAM,IAAI;AACrB;AAuBO,SAAS,wBACd,WACA,MACA,aACA,eACA,UACe;AACf,QAAM,UAAU,QAAQ,OAAO;AAC/B,QAAM,OAAO,OAAO;AAAA,IAClB,kBAAkB,WAAW,IAAI,aAAa;AAAA,IAC9C;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB;AAAA,QACxB,qBAAqB;AAAA,QACrB,kBAAkB;AAAA,QAClB,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,SAAS,IAAI;AAEvC,YAAU,KAAK,KAAK,WAAW,WAAW,IAAI,aAAa,IAAI;AAAA,IAC7D,UAAU,UAAU;AAAA,IACpB,kBAAkB;AAAA,MAChB;AAAA,MACA;AAAA,IACF;AAAA,IACA,WAAW;AAAA,MACT,SAAS,KAAK,YAAY,EAAE;AAAA,MAC5B,QAAQ,KAAK,YAAY,EAAE;AAAA,IAC7B;AAAA,EACF,CAAC;AACD,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,kBACd,MACA,SACA,IACA;AACA,QAAM,MAAM,QAAQ,UAChB,YAAY,QAAQ,QAAQ,OAAO,GAAG,QAAQ,OAAO,IACrD,QAAQ,OAAO;AAEnB,SAAO,OAAO;AAAA,IACZ,qBAAqB,QAAQ,WAAW,IAAI,QAAQ,aAAa;AAAA,IACjE;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB,QAAQ;AAAA,QAChC,qBAAqB,QAAQ;AAAA,QAC7B,kBAAkB,QAAQ;AAAA,QAC1B,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,IAAM,SAAS,MAAM,UAAU,SAAS,OAAa;AACrD,IAAO,kBAAQ;","names":[]}
+{"version":3,"sources":["../transport/id.ts","../transport/message.ts","../package.json","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { customAlphabet } from 'nanoid';\n\nconst alphabet = customAlphabet(\n  '1234567890abcdefghijklmnopqrstuvxyzABCDEFGHIJKLMNOPQRSTUVXYZ',\n);\nexport const generateId = () => alphabet(12);\n","import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { PropagationContext } from '../tracing';\nimport { generateId } from './id';\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 * 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  /**\n   * Specifies what the server's expected session state (from the pov of the client). This can be\n   * used by the server to know whether this is a new or a reestablished connection, and whether it\n   * is compatible with what it already has.\n   */\n  expectedSessionState: Type.Object({\n    // what the client expects the server to send next\n    nextExpectedSeq: Type.Integer(),\n    // TODO: remove optional once we know all servers\n    // are nextSentSeq here\n    // what the server expects the client to send next\n    nextSentSeq: Type.Optional(Type.Integer()),\n  }),\n\n  metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const HandshakeErrorRetriableResponseCodes = Type.Union([\n  Type.Literal('SESSION_STATE_MISMATCH'),\n]);\n\nexport const HandshakeErrorFatalResponseCodes = Type.Union([\n  Type.Literal('MALFORMED_HANDSHAKE_META'),\n  Type.Literal('MALFORMED_HANDSHAKE'),\n  Type.Literal('PROTOCOL_VERSION_MISMATCH'),\n  Type.Literal('REJECTED_BY_CUSTOM_HANDLER'),\n]);\n\nexport const HandshakeErrorResponseCodes = Type.Union([\n  HandshakeErrorRetriableResponseCodes,\n  HandshakeErrorFatalResponseCodes,\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      // TODO: remove optional once we know all servers\n      // are sending code here\n      code: Type.Optional(HandshakeErrorResponseCodes),\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?: PropagationContext;\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,\n  to,\n  sessionId,\n  expectedSessionState,\n  metadata,\n  tracing,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  sessionId: string;\n  expectedSessionState: Static<\n    typeof ControlMessageHandshakeRequestSchema\n  >['expectedSessionState'];\n  metadata?: unknown;\n  tracing?: PropagationContext;\n}): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    tracing,\n    payload: {\n      type: 'HANDSHAKE_REQ',\n      protocolVersion: PROTOCOL_VERSION,\n      sessionId,\n      expectedSessionState,\n      metadata,\n    } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n  };\n}\n\n/**\n * This is a reason that can be given during the handshake to indicate that the peer has the wrong\n * session state.\n */\nexport const SESSION_STATE_MISMATCH = 'session state mismatch';\n\nexport function handshakeResponseMessage({\n  from,\n  to,\n  status,\n}: {\n  from: TransportClientId;\n  to: TransportClientId;\n  status: Static<typeof ControlMessageHandshakeResponseSchema>['status'];\n}): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n  return {\n    id: generateId(),\n    from,\n    to,\n    seq: 0,\n    ack: 0,\n    streamId: generateId(),\n    controlFlags: 0,\n    payload: {\n      type: 'HANDSHAKE_RESP',\n      status,\n    } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n  };\n}\n\nexport function closeStreamMessage(streamId: string): PartialTransportMessage {\n  return {\n    streamId,\n    controlFlags: ControlFlags.StreamClosedBit,\n    payload: {\n      type: 'CLOSE' as const,\n    } satisfies Static<typeof ControlMessagePayloadSchema>,\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","{\n  \"name\": \"@replit/river\",\n  \"description\": \"It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!\",\n  \"version\": \"0.25.1\",\n  \"type\": \"module\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/router/index.js\",\n      \"require\": \"./dist/router/index.cjs\"\n    },\n    \"./logging\": {\n      \"import\": \"./dist/logging/index.js\",\n      \"require\": \"./dist/logging/index.cjs\"\n    },\n    \"./codec\": {\n      \"import\": \"./dist/codec/index.js\",\n      \"require\": \"./dist/codec/index.cjs\"\n    },\n    \"./transport\": {\n      \"import\": \"./dist/transport/index.js\",\n      \"require\": \"./dist/transport/index.cjs\"\n    },\n    \"./transport/ws/client\": {\n      \"import\": \"./dist/transport/impls/ws/client.js\",\n      \"require\": \"./dist/transport/impls/ws/client.cjs\"\n    },\n    \"./transport/ws/server\": {\n      \"import\": \"./dist/transport/impls/ws/server.js\",\n      \"require\": \"./dist/transport/impls/ws/server.cjs\"\n    },\n    \"./transport/uds/client\": {\n      \"import\": \"./dist/transport/impls/uds/client.js\",\n      \"require\": \"./dist/transport/impls/uds/client.cjs\"\n    },\n    \"./transport/uds/server\": {\n      \"import\": \"./dist/transport/impls/uds/server.js\",\n      \"require\": \"./dist/transport/impls/uds/server.cjs\"\n    },\n    \"./test-util\": {\n      \"import\": \"./dist/util/testHelpers.js\",\n      \"require\": \"./dist/util/testHelpers.cjs\"\n    }\n  },\n  \"sideEffects\": [\n    \"./dist/logging/index.js\"\n  ],\n  \"files\": [\n    \"dist\"\n  ],\n  \"dependencies\": {\n    \"@msgpack/msgpack\": \"^3.0.0-beta2\",\n    \"it-pushable\": \"^3.2.3\",\n    \"nanoid\": \"^4.0.2\",\n    \"ws\": \"^8.17.0\"\n  },\n  \"peerDependencies\": {\n    \"@opentelemetry/api\": \"^1.7.0\",\n    \"@sinclair/typebox\": \"~0.32.8\"\n  },\n  \"devDependencies\": {\n    \"@opentelemetry/sdk-trace-base\": \"^1.24.1\",\n    \"@opentelemetry/sdk-trace-web\": \"^1.24.1\",\n    \"@opentelemetry/core\": \"^1.7.0\",\n    \"@types/ws\": \"^8.5.5\",\n    \"@typescript-eslint/eslint-plugin\": \"^7.8.0\",\n    \"@typescript-eslint/parser\": \"^7.8.0\",\n    \"@vitest/ui\": \"^1.3.1\",\n    \"eslint\": \"^8.57.0\",\n    \"eslint-config-prettier\": \"^9.1.0\",\n    \"eslint-plugin-prettier\": \"^5.1.3\",\n    \"prettier\": \"^3.0.0\",\n    \"tsup\": \"^7.2.0\",\n    \"typescript\": \"^5.4.5\",\n    \"vitest\": \"^1.3.1\"\n  },\n  \"scripts\": {\n    \"check\": \"tsc --noEmit && npm run format && npm run lint\",\n    \"format\": \"npx prettier . --check\",\n    \"format:fix\": \"npx prettier . --write\",\n    \"lint\": \"eslint .\",\n    \"lint:fix\": \"eslint . --fix\",\n    \"fix\": \"npm run format:fix && npm run lint:fix\",\n    \"build\": \"rm -rf dist && tsup && du -sh dist\",\n    \"prepack\": \"npm run build\",\n    \"release\": \"npm publish --access public\",\n    \"test:ui\": \"echo \\\"remember to go to /__vitest__ in the webview\\\" && vitest --ui --api.host 0.0.0.0 --api.port 3000\",\n    \"test\": \"vitest\",\n    \"test:single\": \"vitest run --reporter=dot\",\n    \"test:flake\": \"./flake.sh\",\n    \"bench\": \"vitest bench\"\n  },\n  \"engines\": {\n    \"node\": \">=16\"\n  },\n  \"keywords\": [\n    \"rpc\",\n    \"websockets\",\n    \"jsonschema\"\n  ],\n  \"author\": \"Jacky Zhao\",\n  \"license\": \"MIT\"\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 {\n  Context,\n  Span,\n  SpanKind,\n  context,\n  propagation,\n  trace,\n} from '@opentelemetry/api';\nimport { version as RIVER_VERSION } from '../package.json';\nimport { ValidProcType } from '../router';\nimport {\n  ClientTransport,\n  Connection,\n  OpaqueTransportMessage,\n} from '../transport';\n\nexport interface PropagationContext {\n  traceparent: string;\n  tracestate: string;\n}\n\nexport interface TelemetryInfo {\n  span: Span;\n  ctx: Context;\n}\n\nexport function getPropagationContext(\n  ctx: Context,\n): PropagationContext | undefined {\n  const tracing = {\n    traceparent: '',\n    tracestate: '',\n  };\n  propagation.inject(ctx, tracing);\n  return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n  sessionId: string,\n  to: string,\n  from: string,\n  propagationCtx?: PropagationContext,\n): TelemetryInfo {\n  const parentCtx = propagationCtx\n    ? propagation.extract(context.active(), propagationCtx)\n    : context.active();\n\n  const span = tracer.startSpan(\n    `session ${sessionId}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.session.id': sessionId,\n        'river.session.to': to,\n        'river.session.from': from,\n      },\n    },\n    parentCtx,\n  );\n\n  const ctx = trace.setSpan(parentCtx, span);\n\n  return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n  connection: Connection,\n  info: TelemetryInfo,\n): TelemetryInfo {\n  const span = tracer.startSpan(\n    `connection ${connection.id}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.connection.id': connection.id,\n      },\n      links: [{ context: info.span.spanContext() }],\n    },\n    info.ctx,\n  );\n\n  const ctx = trace.setSpan(info.ctx, span);\n\n  return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n  transport: ClientTransport<Connection>,\n  kind: ValidProcType,\n  serviceName: string,\n  procedureName: string,\n  streamId: string,\n): TelemetryInfo {\n  const baseCtx = context.active();\n  const span = tracer.startSpan(\n    `procedure call ${serviceName}.${procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': serviceName,\n        'river.method.name': procedureName,\n        'river.streamId': streamId,\n        'span.kind': 'client',\n      },\n      kind: SpanKind.CLIENT,\n    },\n    baseCtx,\n  );\n\n  const ctx = trace.setSpan(baseCtx, span);\n\n  transport.log?.info(`invoked ${serviceName}.${procedureName}`, {\n    clientId: transport.clientId,\n    transportMessage: {\n      procedureName,\n      serviceName,\n    },\n    telemetry: {\n      traceId: span.spanContext().traceId,\n      spanId: span.spanContext().spanId,\n    },\n  });\n  return { span, ctx };\n}\n\nexport function createHandlerSpan(\n  kind: ValidProcType,\n  message: OpaqueTransportMessage,\n  fn: (span: Span) => Promise<unknown>,\n) {\n  const ctx = message.tracing\n    ? propagation.extract(context.active(), message.tracing)\n    : context.active();\n\n  return tracer.startActiveSpan(\n    `procedure handler ${message.serviceName}.${message.procedureName}`,\n    {\n      attributes: {\n        component: 'river',\n        'river.method.kind': kind,\n        'river.method.service': message.serviceName,\n        'river.method.name': message.procedureName,\n        'river.streamId': message.streamId,\n        'span.kind': 'server',\n      },\n      kind: SpanKind.SERVER,\n    },\n    ctx,\n    fn,\n  );\n}\n\nconst tracer = trace.getTracer('river', RIVER_VERSION);\nexport default tracer;\n"],"mappings":";AAAA,SAAS,sBAAsB;AAE/B,IAAM,WAAW;AAAA,EACf;AACF;AACO,IAAM,aAAa,MAAM,SAAS,EAAE;;;ACL3C,SAAS,YAA6B;AAuB/B,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;AAAA;AAAA;AAAA;AAAA;AAAA,EAMvB,sBAAsB,KAAK,OAAO;AAAA;AAAA,IAEhC,iBAAiB,KAAK,QAAQ;AAAA;AAAA;AAAA;AAAA,IAI9B,aAAa,KAAK,SAAS,KAAK,QAAQ,CAAC;AAAA,EAC3C,CAAC;AAAA,EAED,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,uCAAuC,KAAK,MAAM;AAAA,EAC7D,KAAK,QAAQ,wBAAwB;AACvC,CAAC;AAEM,IAAM,mCAAmC,KAAK,MAAM;AAAA,EACzD,KAAK,QAAQ,0BAA0B;AAAA,EACvC,KAAK,QAAQ,qBAAqB;AAAA,EAClC,KAAK,QAAQ,2BAA2B;AAAA,EACxC,KAAK,QAAQ,4BAA4B;AAC3C,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AACF,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;AAAA;AAAA,MAGpB,MAAM,KAAK,SAAS,2BAA2B;AAAA,IACjD,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,wBAAwB;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAS0E;AACxE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQO,SAAS,yBAAyB;AAAA,EACvC;AAAA,EACA;AAAA,EACA;AACF,GAI2E;AACzE,SAAO;AAAA,IACL,IAAI,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,WAAW;AAAA,IACrB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,mBAAmB,UAA2C;AAC5E,SAAO;AAAA,IACL;AAAA,IACA,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,IACR;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;;;ACtRE,cAAW;;;ACHN,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACNA;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAmBA,SAAS,sBACd,KACgC;AAChC,QAAM,UAAU;AAAA,IACd,aAAa;AAAA,IACb,YAAY;AAAA,EACd;AACA,cAAY,OAAO,KAAK,OAAO;AAC/B,SAAO;AACT;AAEO,SAAS,2BACd,WACA,IACA,MACA,gBACe;AACf,QAAM,YAAY,iBACd,YAAY,QAAQ,QAAQ,OAAO,GAAG,cAAc,IACpD,QAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,SAAS;AAAA,IACpB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB;AAAA,QACpB,oBAAoB;AAAA,QACpB,sBAAsB;AAAA,MACxB;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,WAAW,IAAI;AAEzC,SAAO,EAAE,MAAM,IAAI;AACrB;AAuBO,SAAS,wBACd,WACA,MACA,aACA,eACA,UACe;AACf,QAAM,UAAU,QAAQ,OAAO;AAC/B,QAAM,OAAO,OAAO;AAAA,IAClB,kBAAkB,WAAW,IAAI,aAAa;AAAA,IAC9C;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB;AAAA,QACxB,qBAAqB;AAAA,QACrB,kBAAkB;AAAA,QAClB,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,SAAS,IAAI;AAEvC,YAAU,KAAK,KAAK,WAAW,WAAW,IAAI,aAAa,IAAI;AAAA,IAC7D,UAAU,UAAU;AAAA,IACpB,kBAAkB;AAAA,MAChB;AAAA,MACA;AAAA,IACF;AAAA,IACA,WAAW;AAAA,MACT,SAAS,KAAK,YAAY,EAAE;AAAA,MAC5B,QAAQ,KAAK,YAAY,EAAE;AAAA,IAC7B;AAAA,EACF,CAAC;AACD,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,kBACd,MACA,SACA,IACA;AACA,QAAM,MAAM,QAAQ,UAChB,YAAY,QAAQ,QAAQ,OAAO,GAAG,QAAQ,OAAO,IACrD,QAAQ,OAAO;AAEnB,SAAO,OAAO;AAAA,IACZ,qBAAqB,QAAQ,WAAW,IAAI,QAAQ,aAAa;AAAA,IACjE;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,qBAAqB;AAAA,QACrB,wBAAwB,QAAQ;AAAA,QAChC,qBAAqB,QAAQ;AAAA,QAC7B,kBAAkB,QAAQ;AAAA,QAC1B,aAAa;AAAA,MACf;AAAA,MACA,MAAM,SAAS;AAAA,IACjB;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,IAAM,SAAS,MAAM,UAAU,SAAS,OAAa;AACrD,IAAO,kBAAQ;","names":[]}

package/dist/{chunk-XVGZYVNR.js → chunk-YM5Y4NAT.js}

@@ -3,7 +3,7 @@ import {
   createSessionTelemetryInfo,
   generateId,
   isAck
-} from "./chunk-6NPQ43ZP.js";
+} from "./chunk-SZ5NBBX7.js";
 import {
   NaiveJsonCodec
 } from "./chunk-4PVU7J25.js";
@@ -671,4 +671,4 @@ export {
   SessionState,
   SessionStateGraph
 };
-//# sourceMappingURL=chunk-XVGZYVNR.js.map
+//# sourceMappingURL=chunk-YM5Y4NAT.js.map

package/dist/router/index.cjs

@@ -752,7 +752,7 @@ function Err(error) {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.25.0";
+var version = "0.25.1";
 
 // tracing/index.ts
 function getPropagationContext(ctx) {
@@ -896,6 +896,7 @@ function handleRpc(transport, serverId, input, serviceName, procedureName) {
     tracing: getPropagationContext(ctx),
     controlFlags: 2 /* StreamOpenBit */ | 4 /* StreamClosedBit */
   });
+  let cleanedUp = false;
   const responsePromise = new Promise((resolve) => {
     const onSessionStatus = createSessionDisconnectHandler(serverId, () => {
       cleanup();
@@ -907,6 +908,9 @@ function handleRpc(transport, serverId, input, serviceName, procedureName) {
       );
     });
     function cleanup() {
+      if (cleanedUp)
+        return;
+      cleanedUp = true;
       transport.removeEventListener("message", onMessage);
       transport.removeEventListener("sessionStatus", onSessionStatus);
       span.end();
@@ -936,7 +940,8 @@ function handleStream(transport, serverId, init, serviceName, procedureName) {
   const inputStream = pushable({ objectMode: true });
   const outputStream = pushable({ objectMode: true });
   let firstMessage = true;
-  let healthyClose = true;
+  let sentClose = false;
+  let cleanedUp = false;
   if (init) {
     transport.send(serverId, {
       streamId,
@@ -964,13 +969,14 @@ function handleStream(transport, serverId, init, serviceName, procedureName) {
       }
       transport.send(serverId, m2);
     }
-    if (!healthyClose)
+    if (sentClose)
       return;
+    sentClose = true;
     const m = closeStreamMessage(streamId);
+    m.serviceName = serviceName;
+    m.procedureName = procedureName;
+    m.tracing = getPropagationContext(ctx);
     if (firstMessage) {
-      m.serviceName = serviceName;
-      m.procedureName = procedureName;
-      m.tracing = getPropagationContext(ctx);
       m.controlFlags |= 2 /* StreamOpenBit */;
       firstMessage = false;
     }
@@ -989,6 +995,9 @@ function handleStream(transport, serverId, init, serviceName, procedureName) {
     }
   }
   function cleanup() {
+    if (cleanedUp)
+      return;
+    cleanedUp = true;
     inputStream.end();
     outputStream.end();
     transport.removeEventListener("message", onMessage);
@@ -1002,7 +1011,7 @@ function handleStream(transport, serverId, init, serviceName, procedureName) {
         message: `${serverId} unexpectedly disconnected`
       })
     );
-    healthyClose = false;
+    sentClose = true;
     cleanup();
   });
   transport.addEventListener("message", onMessage);
@@ -1026,7 +1035,8 @@ function handleSubscribe(transport, serverId, input, serviceName, procedureName)
     tracing: getPropagationContext(ctx),
     controlFlags: 2 /* StreamOpenBit */
   });
-  let healthyClose = true;
+  let sentClose = false;
+  let cleanedUp = false;
   const outputStream = pushable({ objectMode: true });
   function onMessage(msg) {
     if (msg.streamId !== streamId)
@@ -1040,6 +1050,9 @@ function handleSubscribe(transport, serverId, input, serviceName, procedureName)
     }
   }
   function cleanup() {
+    if (cleanedUp)
+      return;
+    cleanedUp = true;
     outputStream.end();
     transport.removeEventListener("message", onMessage);
     transport.removeEventListener("sessionStatus", onSessionStatus);
@@ -1047,9 +1060,14 @@ function handleSubscribe(transport, serverId, input, serviceName, procedureName)
   }
   const closeHandler = () => {
     cleanup();
-    if (!healthyClose)
+    if (sentClose)
       return;
-    transport.send(serverId, closeStreamMessage(streamId));
+    sentClose = true;
+    const m = closeStreamMessage(streamId);
+    m.serviceName = serviceName;
+    m.procedureName = procedureName;
+    m.tracing = getPropagationContext(ctx);
+    transport.send(serverId, m);
   };
   const onSessionStatus = createSessionDisconnectHandler(serverId, () => {
     outputStream.push(
@@ -1058,7 +1076,7 @@ function handleSubscribe(transport, serverId, input, serviceName, procedureName)
         message: `${serverId} unexpectedly disconnected`
       })
     );
-    healthyClose = false;
+    sentClose = true;
     cleanup();
   });
   transport.addEventListener("message", onMessage);
@@ -1076,7 +1094,8 @@ function handleUpload(transport, serverId, init, serviceName, procedureName) {
   );
   const inputStream = pushable({ objectMode: true });
   let firstMessage = true;
-  let healthyClose = true;
+  let sentClose = false;
+  let cleanedUp = false;
   if (init) {
     transport.send(serverId, {
       streamId,
@@ -1104,13 +1123,14 @@ function handleUpload(transport, serverId, init, serviceName, procedureName) {
       }
       transport.send(serverId, m2);
     }
-    if (!healthyClose)
+    if (sentClose)
       return;
+    sentClose = true;
     const m = closeStreamMessage(streamId);
+    m.serviceName = serviceName;
+    m.procedureName = procedureName;
+    m.tracing = getPropagationContext(ctx);
     if (firstMessage) {
-      m.serviceName = serviceName;
-      m.procedureName = procedureName;
-      m.tracing = getPropagationContext(ctx);
       m.controlFlags |= 2 /* StreamOpenBit */;
       firstMessage = false;
     }
@@ -1119,7 +1139,7 @@ function handleUpload(transport, serverId, init, serviceName, procedureName) {
   void pipeInputToTransport();
   const responsePromise = new Promise((resolve) => {
     const onSessionStatus = createSessionDisconnectHandler(serverId, () => {
-      healthyClose = false;
+      sentClose = true;
       cleanup();
       resolve(
         Err({
@@ -1129,6 +1149,9 @@ function handleUpload(transport, serverId, init, serviceName, procedureName) {
       );
     });
     function cleanup() {
+      if (cleanedUp)
+        return;
+      cleanedUp = true;
       inputStream.end();
       transport.removeEventListener("message", onMessage);
       transport.removeEventListener("sessionStatus", onSessionStatus);