@replit/river 0.24.3 → 0.25.1

package/dist/{chunk-DZNP3EI5.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.24.3\",\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-KVLCQ24J.js → chunk-YM5Y4NAT.js}

@@ -3,7 +3,7 @@ import {
   createSessionTelemetryInfo,
   generateId,
   isAck
-} from "./chunk-DZNP3EI5.js";
+} from "./chunk-SZ5NBBX7.js";
 import {
   NaiveJsonCodec
 } from "./chunk-4PVU7J25.js";
@@ -344,15 +344,16 @@ var SessionConnected = class extends IdentifiedSession {
   state = "Connected" /* Connected */;
   conn;
   listeners;
-  heartbeatHandle;
-  heartbeatMisses = 0;
+  activeHeartbeatHandle;
+  activeHeartbeatMisses = 0;
+  passiveHeartbeatHandle;
   get isActivelyHeartbeating() {
-    return this.heartbeatHandle !== void 0;
+    return this.activeHeartbeatHandle !== void 0;
   }
   updateBookkeeping(ack, seq) {
     this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);
     this.ack = seq + 1;
-    this.heartbeatMisses = 0;
+    this.activeHeartbeatMisses = 0;
   }
   send(msg) {
     const constructedMsg = this.constructMsg(msg);
@@ -378,8 +379,8 @@ var SessionConnected = class extends IdentifiedSession {
     }
   }
   startActiveHeartbeat() {
-    this.heartbeatHandle = setInterval(() => {
-      const misses = this.heartbeatMisses;
+    this.activeHeartbeatHandle = setInterval(() => {
+      const misses = this.activeHeartbeatMisses;
       const missDuration = misses * this.options.heartbeatIntervalMs;
       if (misses >= this.options.heartbeatsUntilDead) {
         this.log?.info(
@@ -388,15 +389,32 @@ var SessionConnected = class extends IdentifiedSession {
         );
         this.telemetry.span.addEvent("closing connection due to inactivity");
         this.conn.close();
-        clearInterval(this.heartbeatHandle);
-        this.heartbeatHandle = void 0;
+        clearInterval(this.activeHeartbeatHandle);
+        this.activeHeartbeatHandle = void 0;
         return;
       }
       this.sendHeartbeat();
-      this.heartbeatMisses++;
+      this.activeHeartbeatMisses++;
     }, this.options.heartbeatIntervalMs);
   }
+  waitForNextHeartbeat() {
+    const duration = this.options.heartbeatsUntilDead * this.options.heartbeatIntervalMs;
+    if (this.passiveHeartbeatHandle) {
+      clearTimeout(this.passiveHeartbeatHandle);
+      this.passiveHeartbeatHandle = void 0;
+    }
+    this.passiveHeartbeatHandle = setTimeout(() => {
+      this.log?.info(
+        `closing connection to ${this.to} due to not receiving a heartbeat in the last ${duration}ms`,
+        this.loggingMetadata
+      );
+      this.telemetry.span.addEvent("closing connection due to inactivity");
+      this.conn.close();
+      this.passiveHeartbeatHandle = void 0;
+    }, duration);
+  }
   sendHeartbeat() {
+    this.log?.debug("sending heartbeat", this.loggingMetadata);
     this.send({
       streamId: "heartbeat",
       controlFlags: 1 /* AckBit */,
@@ -448,6 +466,7 @@ var SessionConnected = class extends IdentifiedSession {
     });
     if (!this.isActivelyHeartbeating) {
       this.sendHeartbeat();
+      this.waitForNextHeartbeat();
     }
   };
   _handleStateExit() {
@@ -455,8 +474,10 @@ var SessionConnected = class extends IdentifiedSession {
     this.conn.removeDataListener(this.onMessageData);
     this.conn.removeCloseListener(this.listeners.onConnectionClosed);
     this.conn.removeErrorListener(this.listeners.onConnectionErrored);
-    clearInterval(this.heartbeatHandle);
-    this.heartbeatHandle = void 0;
+    clearInterval(this.activeHeartbeatHandle);
+    clearTimeout(this.passiveHeartbeatHandle);
+    this.activeHeartbeatHandle = void 0;
+    this.passiveHeartbeatHandle = void 0;
   }
   _handleClose() {
     super._handleClose();
@@ -650,4 +671,4 @@ export {
   SessionState,
   SessionStateGraph
 };
-//# sourceMappingURL=chunk-KVLCQ24J.js.map
+//# sourceMappingURL=chunk-YM5Y4NAT.js.map

package/dist/chunk-YM5Y4NAT.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../transport/options.ts","../transport/sessionStateMachine/common.ts","../transport/sessionStateMachine/SessionConnecting.ts","../transport/sessionStateMachine/SessionNoConnection.ts","../transport/sessionStateMachine/SessionWaitingForHandshake.ts","../transport/sessionStateMachine/SessionHandshaking.ts","../transport/sessionStateMachine/SessionConnected.ts","../transport/sessionStateMachine/transitions.ts"],"sourcesContent":["import { NaiveJsonCodec } from '../codec/json';\nimport { ConnectionRetryOptions } from './rateLimit';\nimport { SessionOptions } from './sessionStateMachine/common';\n\nexport type TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n  heartbeatIntervalMs: 1_000,\n  heartbeatsUntilDead: 2,\n  sessionDisconnectGraceMs: 5_000,\n  connectionTimeoutMs: 2_000,\n  handshakeTimeoutMs: 1_000,\n  codec: NaiveJsonCodec,\n};\n\nexport type ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n  baseIntervalMs: 250,\n  maxJitterMs: 200,\n  maxBackoffMs: 32_000,\n  attemptBudgetCapacity: 5,\n  budgetRestoreIntervalMs: 200,\n};\n\nexport const defaultClientTransportOptions: ClientTransportOptions = {\n  ...defaultTransportOptions,\n  ...defaultConnectionRetryOptions,\n};\n\nexport type ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nexport const defaultServerTransportOptions: ServerTransportOptions = {\n  ...defaultTransportOptions,\n};\n","import { Logger, MessageMetadata } from '../../logging';\nimport { TelemetryInfo } from '../../tracing';\nimport {\n  OpaqueTransportMessage,\n  OpaqueTransportMessageSchema,\n  PartialTransportMessage,\n  TransportClientId,\n  TransportMessage,\n} from '../message';\nimport { Value } from '@sinclair/typebox/value';\nimport { SessionNoConnection } from './SessionNoConnection';\nimport { SessionConnecting } from './SessionConnecting';\nimport { SessionHandshaking } from './SessionHandshaking';\nimport { SessionConnected } from './SessionConnected';\nimport { Codec } from '../../codec';\nimport { Connection } from '../connection';\nimport { generateId } from '../id';\n\nexport const enum SessionState {\n  NoConnection = 'NoConnection',\n  Connecting = 'Connecting',\n  Handshaking = 'Handshaking',\n  Connected = 'Connected',\n  WaitingForHandshake = 'WaitingForHandshake',\n}\n\nexport type Session<ConnType extends Connection> =\n  | SessionNoConnection\n  | SessionConnecting<ConnType>\n  | SessionHandshaking<ConnType>\n  | SessionConnected<ConnType>;\n\nexport const ERR_CONSUMED = `session state has been consumed and is no longer valid`;\n\nabstract class StateMachineState {\n  abstract readonly state: SessionState;\n\n  /*\n   * Whether this state has been consumed\n   * and we've moved on to another state\n   */\n  _isConsumed: boolean;\n\n  // called when we're transitioning to another state\n  // note that this is internal and should not be called directly\n  // by consumers, the proxy will call this when the state is consumed\n  // and we're transitioning to another state\n  abstract _handleStateExit(): void;\n\n  // called when we exit the state machine entirely\n  // note that this is internal and should not be called directly\n  // by consumers, the proxy will call this when .close is closed\n  abstract _handleClose(): void;\n\n  close(): void {\n    this._handleClose();\n  }\n\n  constructor() {\n    this._isConsumed = false;\n\n    // proxy helps us prevent access to properties after the state has been consumed\n    // e.g. if we hold a reference to a state and try to access it after it's been consumed\n    // we intercept the access and throw an error to help catch bugs\n    return new Proxy(this, {\n      get(target, prop) {\n        // always allow access to _isConsumed, id, and state\n        if (prop === '_isConsumed' || prop === 'id' || prop === 'state') {\n          return Reflect.get(target, prop);\n        }\n\n        // modify _handleStateExit\n        if (prop === '_handleStateExit') {\n          return () => {\n            target._isConsumed = true;\n            target._handleStateExit();\n          };\n        }\n\n        // modify _handleClose\n        if (prop === '_handleClose') {\n          return () => {\n            target._handleStateExit();\n            target._handleClose();\n          };\n        }\n\n        if (target._isConsumed) {\n          throw new Error(\n            `${ERR_CONSUMED}: getting ${prop.toString()} on consumed state`,\n          );\n        }\n\n        return Reflect.get(target, prop);\n      },\n      set(target, prop, value) {\n        if (target._isConsumed) {\n          throw new Error(\n            `${ERR_CONSUMED}: setting ${prop.toString()} on consumed state`,\n          );\n        }\n\n        return Reflect.set(target, prop, value);\n      },\n    });\n  }\n}\n\nexport interface SessionOptions {\n  /**\n   * Frequency at which to send heartbeat acknowledgements\n   */\n  heartbeatIntervalMs: number;\n  /**\n   * Number of elapsed heartbeats without a response message before we consider\n   * the connection dead.\n   */\n  heartbeatsUntilDead: number;\n  /**\n   * Duration to wait between connection disconnect and actual session disconnect\n   */\n  sessionDisconnectGraceMs: number;\n  /**\n   * Connection timeout in milliseconds\n   */\n  connectionTimeoutMs: number;\n  /**\n   * Handshake timeout in milliseconds\n   */\n  handshakeTimeoutMs: number;\n  /**\n   * The codec to use for encoding/decoding messages over the wire\n   */\n  codec: Codec;\n}\n\n// all session states have a from and options\nexport abstract class CommonSession extends StateMachineState {\n  readonly from: TransportClientId;\n  readonly options: SessionOptions;\n\n  log?: Logger;\n  abstract get loggingMetadata(): MessageMetadata;\n\n  constructor(\n    from: TransportClientId,\n    options: SessionOptions,\n    log: Logger | undefined,\n  ) {\n    super();\n    this.from = from;\n    this.options = options;\n    this.log = log;\n  }\n\n  parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n    const parsedMsg = this.options.codec.fromBuffer(msg);\n\n    if (parsedMsg === null) {\n      const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n      this.log?.error(\n        `received malformed msg: ${decodedBuffer}`,\n        this.loggingMetadata,\n      );\n      return null;\n    }\n\n    if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n      this.log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n        ...this.loggingMetadata,\n        validationErrors: [\n          ...Value.Errors(OpaqueTransportMessageSchema, parsedMsg),\n        ],\n      });\n\n      return null;\n    }\n\n    return parsedMsg;\n  }\n}\n\nexport type InheritedProperties = Pick<\n  IdentifiedSession,\n  'id' | 'from' | 'to' | 'seq' | 'ack' | 'sendBuffer' | 'telemetry' | 'options'\n>;\n\nexport type SessionId = string;\n\n// all sessions where we know the other side's client id\nexport abstract class IdentifiedSession extends CommonSession {\n  readonly id: SessionId;\n  readonly telemetry: TelemetryInfo;\n  readonly to: TransportClientId;\n\n  /**\n   * Index of the message we will send next (excluding handshake)\n   */\n  seq: number;\n\n  /**\n   * Number of unique messages we've received this session (excluding handshake)\n   */\n  ack: number;\n  sendBuffer: Array<OpaqueTransportMessage>;\n\n  constructor(\n    id: SessionId,\n    from: TransportClientId,\n    to: TransportClientId,\n    seq: number,\n    ack: number,\n    sendBuffer: Array<OpaqueTransportMessage>,\n    telemetry: TelemetryInfo,\n    options: SessionOptions,\n    log: Logger | undefined,\n  ) {\n    super(from, options, log);\n    this.id = id;\n    this.to = to;\n    this.seq = seq;\n    this.ack = ack;\n    this.sendBuffer = sendBuffer;\n    this.telemetry = telemetry;\n    this.log = log;\n  }\n\n  get loggingMetadata(): MessageMetadata {\n    const spanContext = this.telemetry.span.spanContext();\n\n    return {\n      clientId: this.from,\n      connectedTo: this.to,\n      sessionId: this.id,\n      telemetry: {\n        traceId: spanContext.traceId,\n        spanId: spanContext.spanId,\n      },\n    };\n  }\n\n  constructMsg<Payload>(\n    partialMsg: PartialTransportMessage<Payload>,\n  ): TransportMessage<Payload> {\n    const msg = {\n      ...partialMsg,\n      id: generateId(),\n      to: this.to,\n      from: this.from,\n      seq: this.seq,\n      ack: this.ack,\n    };\n\n    this.seq++;\n    return msg;\n  }\n\n  nextSeq(): number {\n    return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;\n  }\n\n  send(msg: PartialTransportMessage): string {\n    const constructedMsg = this.constructMsg(msg);\n    this.sendBuffer.push(constructedMsg);\n    return constructedMsg.id;\n  }\n\n  _handleStateExit(): void {\n    // noop\n  }\n\n  _handleClose(): void {\n    // zero out the buffer\n    this.sendBuffer.length = 0;\n    this.telemetry.span.end();\n  }\n}\n","import { Connection } from '../connection';\nimport { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionConnectingListeners {\n  onConnectionEstablished: (conn: Connection) => void;\n  onConnectionFailed: (err: unknown) => void;\n\n  // timeout related\n  onConnectionTimeout: () => void;\n}\n\n/*\n * A session that is connecting but we don't have access to the raw connection\n * yet.\n *\n * Valid transitions:\n * - Connecting -> NoConnection (timeout)\n * - Connecting -> Handshaking (on connection established)\n */\nexport class SessionConnecting<\n  ConnType extends Connection,\n> extends IdentifiedSession {\n  readonly state = SessionState.Connecting as const;\n  connPromise: Promise<ConnType>;\n  listeners: SessionConnectingListeners;\n\n  connectionTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(\n    connPromise: Promise<ConnType>,\n    listeners: SessionConnectingListeners,\n    ...args: ConstructorParameters<typeof IdentifiedSession>\n  ) {\n    super(...args);\n    this.connPromise = connPromise;\n    this.listeners = listeners;\n\n    this.connectionTimeout = setTimeout(() => {\n      listeners.onConnectionTimeout();\n    }, this.options.connectionTimeoutMs);\n\n    connPromise.then(\n      (conn) => {\n        if (this._isConsumed) return;\n        listeners.onConnectionEstablished(conn);\n      },\n      (err) => {\n        if (this._isConsumed) return;\n        listeners.onConnectionFailed(err);\n      },\n    );\n  }\n\n  // close a pending connection if it resolves, ignore errors if the promise\n  // ends up rejected anyways\n  bestEffortClose() {\n    void this.connPromise\n      .then((conn) => conn.close())\n      .catch(() => {\n        // ignore errors\n      });\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n    clearTimeout(this.connectionTimeout);\n    this.connectionTimeout = undefined;\n  }\n\n  _handleClose(): void {\n    // close the pending connection if it resolves\n    this.bestEffortClose();\n    super._handleClose();\n  }\n}\n","import { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionNoConnectionListeners {\n  // timeout related\n  onSessionGracePeriodElapsed: () => void;\n}\n\n/*\n * A session that is not connected and cannot send or receive messages.\n *\n * Valid transitions:\n * - NoConnection -> Connecting (on connect)\n */\nexport class SessionNoConnection extends IdentifiedSession {\n  readonly state = SessionState.NoConnection as const;\n  listeners: SessionNoConnectionListeners;\n\n  gracePeriodTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(\n    listeners: SessionNoConnectionListeners,\n    ...args: ConstructorParameters<typeof IdentifiedSession>\n  ) {\n    super(...args);\n    this.listeners = listeners;\n\n    this.gracePeriodTimeout = setTimeout(() => {\n      this.listeners.onSessionGracePeriodElapsed();\n    }, this.options.sessionDisconnectGraceMs);\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n\n    if (this.gracePeriodTimeout) {\n      clearTimeout(this.gracePeriodTimeout);\n      this.gracePeriodTimeout = undefined;\n    }\n  }\n}\n","import { MessageMetadata } from '../../logging';\nimport { Connection } from '../connection';\nimport { TransportMessage } from '../message';\nimport { SessionHandshakingListeners } from './SessionHandshaking';\nimport { CommonSession, SessionState } from './common';\n\n/*\n * Server-side session that has a connection but is waiting for the client to identify itself.\n *\n * Valid transitions:\n * - WaitingForHandshake -> NoConnection (on close)\n * - WaitingForHandshake -> Connected (on handshake)\n */\nexport class SessionWaitingForHandshake<\n  ConnType extends Connection,\n> extends CommonSession {\n  readonly state = SessionState.WaitingForHandshake as const;\n  conn: ConnType;\n  listeners: SessionHandshakingListeners;\n\n  handshakeTimeout?: ReturnType<typeof setTimeout>;\n\n  constructor(\n    conn: ConnType,\n    listeners: SessionHandshakingListeners,\n    ...args: ConstructorParameters<typeof CommonSession>\n  ) {\n    super(...args);\n    this.conn = conn;\n    this.listeners = listeners;\n\n    this.handshakeTimeout = setTimeout(() => {\n      listeners.onHandshakeTimeout();\n    }, this.options.handshakeTimeoutMs);\n\n    this.conn.addDataListener(this.onHandshakeData);\n    this.conn.addErrorListener(listeners.onConnectionErrored);\n    this.conn.addCloseListener(listeners.onConnectionClosed);\n  }\n\n  onHandshakeData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) {\n      this.listeners.onInvalidHandshake('could not parse message');\n      return;\n    }\n\n    // after this fires, the listener is responsible for transitioning the session\n    // and thus removing the handshake timeout\n    this.listeners.onHandshake(parsedMsg);\n  };\n\n  get loggingMetadata(): MessageMetadata {\n    return {\n      clientId: this.from,\n      connId: this.conn.id,\n    };\n  }\n\n  sendHandshake(msg: TransportMessage): boolean {\n    return this.conn.send(this.options.codec.toBuffer(msg));\n  }\n\n  _handleStateExit(): void {\n    this.conn.removeDataListener(this.onHandshakeData);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    clearTimeout(this.handshakeTimeout);\n    this.handshakeTimeout = undefined;\n  }\n\n  _handleClose(): void {\n    this.conn.close();\n  }\n}\n","import { Connection } from '../connection';\nimport { OpaqueTransportMessage, TransportMessage } from '../message';\nimport { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionHandshakingListeners {\n  onConnectionErrored: (err: unknown) => void;\n  onConnectionClosed: () => void;\n  onHandshake: (msg: OpaqueTransportMessage) => void;\n  onInvalidHandshake: (reason: string) => void;\n\n  // timeout related\n  onHandshakeTimeout: () => void;\n}\n\n/*\n * A session that is handshaking and waiting for the other side to identify itself.\n *\n * Valid transitions:\n * - Handshaking -> NoConnection (on close)\n * - Handshaking -> Connected (on handshake)\n */\nexport class SessionHandshaking<\n  ConnType extends Connection,\n> extends IdentifiedSession {\n  readonly state = SessionState.Handshaking as const;\n  conn: ConnType;\n  listeners: SessionHandshakingListeners;\n\n  handshakeTimeout: ReturnType<typeof setTimeout>;\n\n  constructor(\n    conn: ConnType,\n    listeners: SessionHandshakingListeners,\n    ...args: ConstructorParameters<typeof IdentifiedSession>\n  ) {\n    super(...args);\n    this.conn = conn;\n    this.listeners = listeners;\n\n    this.handshakeTimeout = setTimeout(() => {\n      listeners.onHandshakeTimeout();\n    }, this.options.handshakeTimeoutMs);\n\n    this.conn.addDataListener(this.onHandshakeData);\n    this.conn.addErrorListener(listeners.onConnectionErrored);\n    this.conn.addCloseListener(listeners.onConnectionClosed);\n  }\n\n  onHandshakeData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) {\n      this.listeners.onInvalidHandshake('could not parse message');\n      return;\n    }\n\n    this.listeners.onHandshake(parsedMsg);\n  };\n\n  sendHandshake(msg: TransportMessage): boolean {\n    return this.conn.send(this.options.codec.toBuffer(msg));\n  }\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n    this.conn.removeDataListener(this.onHandshakeData);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    clearTimeout(this.handshakeTimeout);\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n    this.conn.close();\n  }\n}\n","import { Static } from '@sinclair/typebox';\nimport {\n  ControlFlags,\n  ControlMessageAckSchema,\n  OpaqueTransportMessage,\n  PartialTransportMessage,\n  isAck,\n} from '../message';\nimport { IdentifiedSession, SessionState } from './common';\nimport { Connection } from '../connection';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nexport interface SessionConnectedListeners {\n  onConnectionErrored: (err: unknown) => void;\n  onConnectionClosed: () => void;\n  onMessage: (msg: OpaqueTransportMessage) => void;\n  onInvalidMessage: (reason: string) => void;\n}\n\n/*\n * A session that is connected and can send and receive messages.\n *\n * Valid transitions:\n * - Connected -> NoConnection (on close)\n */\nexport class SessionConnected<\n  ConnType extends Connection,\n> extends IdentifiedSession {\n  readonly state = SessionState.Connected as const;\n  conn: ConnType;\n  listeners: SessionConnectedListeners;\n\n  private activeHeartbeatHandle?: ReturnType<typeof setInterval> | undefined;\n  private activeHeartbeatMisses = 0;\n\n  private passiveHeartbeatHandle?: ReturnType<typeof setTimeout> | undefined;\n\n  get isActivelyHeartbeating() {\n    return this.activeHeartbeatHandle !== undefined;\n  }\n\n  updateBookkeeping(ack: number, seq: number) {\n    this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n    this.ack = seq + 1;\n    this.activeHeartbeatMisses = 0;\n  }\n\n  send(msg: PartialTransportMessage): string {\n    const constructedMsg = this.constructMsg(msg);\n    this.sendBuffer.push(constructedMsg);\n    this.conn.send(this.options.codec.toBuffer(constructedMsg));\n    return constructedMsg.id;\n  }\n\n  constructor(\n    conn: ConnType,\n    listeners: SessionConnectedListeners,\n    ...args: ConstructorParameters<typeof IdentifiedSession>\n  ) {\n    super(...args);\n    this.conn = conn;\n    this.listeners = listeners;\n\n    this.conn.addDataListener(this.onMessageData);\n    this.conn.addCloseListener(listeners.onConnectionClosed);\n    this.conn.addErrorListener(listeners.onConnectionErrored);\n\n    // send any buffered messages\n    if (this.sendBuffer.length > 0) {\n      this.log?.debug(\n        `sending ${this.sendBuffer.length} buffered messages`,\n        this.loggingMetadata,\n      );\n    }\n\n    for (const msg of this.sendBuffer) {\n      conn.send(this.options.codec.toBuffer(msg));\n    }\n\n    // dont explicity clear the buffer, we'll just filter out old messages\n    // when we receive an ack\n  }\n\n  startActiveHeartbeat() {\n    this.activeHeartbeatHandle = setInterval(() => {\n      const misses = this.activeHeartbeatMisses;\n      const missDuration = misses * this.options.heartbeatIntervalMs;\n      if (misses >= this.options.heartbeatsUntilDead) {\n        this.log?.info(\n          `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n          this.loggingMetadata,\n        );\n        this.telemetry.span.addEvent('closing connection due to inactivity');\n        this.conn.close();\n        clearInterval(this.activeHeartbeatHandle);\n        this.activeHeartbeatHandle = undefined;\n        return;\n      }\n\n      this.sendHeartbeat();\n      this.activeHeartbeatMisses++;\n    }, this.options.heartbeatIntervalMs);\n  }\n\n  waitForNextHeartbeat() {\n    const duration =\n      this.options.heartbeatsUntilDead * this.options.heartbeatIntervalMs;\n\n    if (this.passiveHeartbeatHandle) {\n      clearTimeout(this.passiveHeartbeatHandle);\n      this.passiveHeartbeatHandle = undefined;\n    }\n\n    this.passiveHeartbeatHandle = setTimeout(() => {\n      this.log?.info(\n        `closing connection to ${this.to} due to not receiving a heartbeat in the last ${duration}ms`,\n        this.loggingMetadata,\n      );\n      this.telemetry.span.addEvent('closing connection due to inactivity');\n      this.conn.close();\n      this.passiveHeartbeatHandle = undefined;\n    }, duration);\n  }\n\n  private sendHeartbeat() {\n    this.log?.debug('sending heartbeat', this.loggingMetadata);\n    this.send({\n      streamId: 'heartbeat',\n      controlFlags: ControlFlags.AckBit,\n      payload: {\n        type: 'ACK',\n      } satisfies Static<typeof ControlMessageAckSchema>,\n    });\n  }\n\n  onMessageData = (msg: Uint8Array) => {\n    const parsedMsg = this.parseMsg(msg);\n    if (parsedMsg === null) return;\n\n    // check message ordering here\n    if (parsedMsg.seq !== this.ack) {\n      if (parsedMsg.seq < this.ack) {\n        this.log?.debug(\n          `received duplicate msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack}), discarding`,\n          {\n            ...this.loggingMetadata,\n            transportMessage: parsedMsg,\n          },\n        );\n      } else {\n        const reason = `received out-of-order msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;\n        this.log?.error(reason, {\n          ...this.loggingMetadata,\n          transportMessage: parsedMsg,\n          tags: ['invariant-violation'],\n        });\n        this.telemetry.span.setStatus({\n          code: SpanStatusCode.ERROR,\n          message: reason,\n        });\n\n        this.listeners.onInvalidMessage(reason);\n      }\n\n      return;\n    }\n\n    // message is ok to update bookkeeping with\n    this.log?.debug(`received msg`, {\n      ...this.loggingMetadata,\n      transportMessage: parsedMsg,\n    });\n\n    this.updateBookkeeping(parsedMsg.ack, parsedMsg.seq);\n\n    // dispatch directly if its not an explicit ack\n    if (!isAck(parsedMsg.controlFlags)) {\n      this.listeners.onMessage(parsedMsg);\n      return;\n    }\n\n    // discard acks (unless we aren't heartbeating in which case just respond)\n    this.log?.debug(`discarding msg (ack bit set)`, {\n      ...this.loggingMetadata,\n      transportMessage: parsedMsg,\n    });\n\n    // if we are not actively heartbeating, we are in passive\n    // heartbeat mode and should send a response to the ack\n    if (!this.isActivelyHeartbeating) {\n      this.sendHeartbeat();\n      this.waitForNextHeartbeat();\n    }\n  };\n\n  _handleStateExit(): void {\n    super._handleStateExit();\n    this.conn.removeDataListener(this.onMessageData);\n    this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n    this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n    clearInterval(this.activeHeartbeatHandle);\n    clearTimeout(this.passiveHeartbeatHandle);\n    this.activeHeartbeatHandle = undefined;\n    this.passiveHeartbeatHandle = undefined;\n  }\n\n  _handleClose(): void {\n    super._handleClose();\n    this.conn.close();\n  }\n}\n","import { OpaqueTransportMessage, TransportClientId } from '..';\nimport {\n  SessionConnecting,\n  SessionConnectingListeners,\n} from './SessionConnecting';\nimport {\n  SessionNoConnection,\n  SessionNoConnectionListeners,\n} from './SessionNoConnection';\nimport { IdentifiedSession, SessionOptions } from './common';\nimport { PropagationContext, createSessionTelemetryInfo } from '../../tracing';\nimport { SessionWaitingForHandshake } from './SessionWaitingForHandshake';\nimport {\n  SessionHandshaking,\n  SessionHandshakingListeners,\n} from './SessionHandshaking';\nimport {\n  SessionConnected,\n  SessionConnectedListeners,\n} from './SessionConnected';\nimport { generateId } from '../id';\nimport { Connection } from '../connection';\nimport { Logger } from '../../logging';\n\nfunction inheritSharedSession(\n  session: IdentifiedSession,\n): ConstructorParameters<typeof IdentifiedSession> {\n  return [\n    session.id,\n    session.from,\n    session.to,\n    session.seq,\n    session.ack,\n    session.sendBuffer,\n    session.telemetry,\n    session.options,\n    session.log,\n  ];\n}\n\n/*\n * Session state machine:\n * 1. SessionNoConnection is the client entrypoint as\n *    we know who the other side is already, we just need to connect\n * 5. SessionWaitingForHandshake is the server entrypoint\n *    as we have a connection but don't know who the other side is yet\n *\n *                           1. SessionNoConnection         ◄──┐\n *                           │  reconnect / connect attempt    │\n *                           ▼                                 │\n *                           2. SessionConnecting              │\n *                           │  connect success  ──────────────┤ connect failure\n *                           ▼                                 │\n *                           3. SessionHandshaking             │\n *                           │  handshake success       ┌──────┤ connection drop\n * 5. WaitingForHandshake    │  handshake failure  ─────┤      │\n * │  handshake success      ▼                          │      │ connection drop\n * ├───────────────────────► 4. SessionConnected        │      │ heartbeat misses\n * │                         │  invalid message  ───────┼──────┘\n * │                         ▼                          │\n * └───────────────────────► x. Destroy Session   ◄─────┘\n *   handshake failure\n */\nexport const SessionStateGraph = {\n  entrypoints: {\n    NoConnection(\n      to: TransportClientId,\n      from: TransportClientId,\n      listeners: SessionNoConnectionListeners,\n      options: SessionOptions,\n      log?: Logger,\n    ) {\n      const id = `session-${generateId()}`;\n      const telemetry = createSessionTelemetryInfo(id, to, from);\n      const sendBuffer: Array<OpaqueTransportMessage> = [];\n\n      const session = new SessionNoConnection(\n        listeners,\n        id,\n        from,\n        to,\n        0,\n        0,\n        sendBuffer,\n        telemetry,\n        options,\n        log,\n      );\n\n      session.log?.info(`session ${session.id} created in NoConnection state`, {\n        ...session.loggingMetadata,\n        tags: ['state-transition'],\n      });\n\n      return session;\n    },\n    WaitingForHandshake<ConnType extends Connection>(\n      from: TransportClientId,\n      conn: ConnType,\n      listeners: SessionHandshakingListeners,\n      options: SessionOptions,\n      log?: Logger,\n    ): SessionWaitingForHandshake<ConnType> {\n      const session = new SessionWaitingForHandshake(\n        conn,\n        listeners,\n        from,\n        options,\n        log,\n      );\n\n      session.log?.info(`session created in WaitingForHandshake state`, {\n        ...session.loggingMetadata,\n        tags: ['state-transition'],\n      });\n\n      return session;\n    },\n  },\n  // All of the transitions 'move'/'consume' the old session and return a new one.\n  // After a session is transitioned, any usage of the old session will throw.\n  transition: {\n    // happy path transitions\n    NoConnectionToConnecting<ConnType extends Connection>(\n      oldSession: SessionNoConnection,\n      connPromise: Promise<ConnType>,\n      listeners: SessionConnectingListeners,\n    ): SessionConnecting<ConnType> {\n      const carriedState = inheritSharedSession(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionConnecting(\n        connPromise,\n        listeners,\n        ...carriedState,\n      );\n      session.log?.info(\n        `session ${session.id} transition from NoConnection to Connecting`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n      return session;\n    },\n    ConnectingToHandshaking<ConnType extends Connection>(\n      oldSession: SessionConnecting<ConnType>,\n      conn: ConnType,\n      listeners: SessionHandshakingListeners,\n    ): SessionHandshaking<ConnType> {\n      const carriedState = inheritSharedSession(oldSession);\n      oldSession._handleStateExit();\n\n      const session = new SessionHandshaking(conn, listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from Connecting to Handshaking`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    HandshakingToConnected<ConnType extends Connection>(\n      oldSession: SessionHandshaking<ConnType>,\n      listeners: SessionConnectedListeners,\n    ): SessionConnected<ConnType> {\n      const carriedState = inheritSharedSession(oldSession);\n      const conn = oldSession.conn;\n      oldSession._handleStateExit();\n\n      const session = new SessionConnected(conn, listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from Handshaking to Connected`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    WaitingForHandshakeToConnected<ConnType extends Connection>(\n      pendingSession: SessionWaitingForHandshake<ConnType>,\n      oldSession: SessionNoConnection | undefined,\n      sessionId: string,\n      to: TransportClientId,\n      propagationCtx: PropagationContext | undefined,\n      listeners: SessionConnectedListeners,\n    ): SessionConnected<ConnType> {\n      const conn = pendingSession.conn;\n      const { from, options } = pendingSession;\n      const carriedState: ConstructorParameters<typeof IdentifiedSession> =\n        oldSession\n          ? // old session exists, inherit state\n            inheritSharedSession(oldSession)\n          : // old session does not exist, create new state\n            [\n              sessionId,\n              from,\n              to,\n              0,\n              0,\n              [],\n              createSessionTelemetryInfo(sessionId, to, from, propagationCtx),\n              options,\n              pendingSession.log,\n            ];\n\n      pendingSession._handleStateExit();\n      oldSession?._handleStateExit();\n\n      const session = new SessionConnected(conn, listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from WaitingForHandshake to Connected`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    // disconnect paths\n    ConnectingToNoConnection<ConnType extends Connection>(\n      oldSession: SessionConnecting<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection {\n      const carriedState = inheritSharedSession(oldSession);\n      oldSession.bestEffortClose();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection(listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from Connecting to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    HandshakingToNoConnection<ConnType extends Connection>(\n      oldSession: SessionHandshaking<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection {\n      const carriedState = inheritSharedSession(oldSession);\n      oldSession.conn.close();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection(listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from Handshaking to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n    ConnectedToNoConnection<ConnType extends Connection>(\n      oldSession: SessionConnected<ConnType>,\n      listeners: SessionNoConnectionListeners,\n    ): SessionNoConnection {\n      const carriedState = inheritSharedSession(oldSession);\n      oldSession.conn.close();\n      oldSession._handleStateExit();\n\n      const session = new SessionNoConnection(listeners, ...carriedState);\n      session.log?.info(\n        `session ${session.id} transition from Connected to NoConnection`,\n        {\n          ...session.loggingMetadata,\n          tags: ['state-transition'],\n        },\n      );\n\n      return session;\n    },\n  },\n} as const;\n"],"mappings":";;;;;;;;;;;AAQO,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,qBAAqB;AAAA,EACrB,oBAAoB;AAAA,EACpB,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AAAA,EACH,GAAG;AACL;AAMO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AACL;;;AC/BA,SAAS,aAAa;AASf,IAAW,eAAX,kBAAWA,kBAAX;AACL,EAAAA,cAAA,kBAAe;AACf,EAAAA,cAAA,gBAAa;AACb,EAAAA,cAAA,iBAAc;AACd,EAAAA,cAAA,eAAY;AACZ,EAAAA,cAAA,yBAAsB;AALN,SAAAA;AAAA,GAAA;AAcX,IAAM,eAAe;AAE5B,IAAe,oBAAf,MAAiC;AAAA;AAAA;AAAA;AAAA;AAAA,EAO/B;AAAA,EAaA,QAAc;AACZ,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,cAAc;AACZ,SAAK,cAAc;AAKnB,WAAO,IAAI,MAAM,MAAM;AAAA,MACrB,IAAI,QAAQ,MAAM;AAEhB,YAAI,SAAS,iBAAiB,SAAS,QAAQ,SAAS,SAAS;AAC/D,iBAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,QACjC;AAGA,YAAI,SAAS,oBAAoB;AAC/B,iBAAO,MAAM;AACX,mBAAO,cAAc;AACrB,mBAAO,iBAAiB;AAAA,UAC1B;AAAA,QACF;AAGA,YAAI,SAAS,gBAAgB;AAC3B,iBAAO,MAAM;AACX,mBAAO,iBAAiB;AACxB,mBAAO,aAAa;AAAA,UACtB;AAAA,QACF;AAEA,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,MACjC;AAAA,MACA,IAAI,QAAQ,MAAM,OAAO;AACvB,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,MAAM,KAAK;AAAA,MACxC;AAAA,IACF,CAAC;AAAA,EACH;AACF;AA+BO,IAAe,gBAAf,cAAqC,kBAAkB;AAAA,EACnD;AAAA,EACA;AAAA,EAET;AAAA,EAGA,YACE,MACA,SACA,KACA;AACA,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,UAAU;AACf,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,SAAS,KAAgD;AACvD,UAAM,YAAY,KAAK,QAAQ,MAAM,WAAW,GAAG;AAEnD,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,KAAK;AAAA,QACR,2BAA2B,aAAa;AAAA,QACxC,KAAK;AAAA,MACP;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,KAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QACpE,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,8BAA8B,SAAS;AAAA,QACzD;AAAA,MACF,CAAC;AAED,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AACF;AAUO,IAAe,oBAAf,cAAyC,cAAc;AAAA,EACnD;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EACA;AAAA,EAEA,YACE,IACA,MACA,IACA,KACA,KACA,YACA,WACA,SACA,KACA;AACA,UAAM,MAAM,SAAS,GAAG;AACxB,SAAK,KAAK;AACV,SAAK,KAAK;AACV,SAAK,MAAM;AACX,SAAK,MAAM;AACX,SAAK,aAAa;AAClB,SAAK,YAAY;AACjB,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,cAAc,KAAK,UAAU,KAAK,YAAY;AAEpD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,WAAW;AAAA,QACT,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,WAAW;AAAA,MACf,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,WAAO;AAAA,EACT;AAAA,EAEA,UAAkB;AAChB,WAAO,KAAK,WAAW,SAAS,IAAI,KAAK,WAAW,CAAC,EAAE,MAAM,KAAK;AAAA,EACpE;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AACnC,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,mBAAyB;AAAA,EAEzB;AAAA,EAEA,eAAqB;AAEnB,SAAK,WAAW,SAAS;AACzB,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AACF;;;ACjQO,IAAM,oBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,aACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,cAAc;AACnB,SAAK,YAAY;AAEjB,SAAK,oBAAoB,WAAW,MAAM;AACxC,gBAAU,oBAAoB;AAAA,IAChC,GAAG,KAAK,QAAQ,mBAAmB;AAEnC,gBAAY;AAAA,MACV,CAAC,SAAS;AACR,YAAI,KAAK;AAAa;AACtB,kBAAU,wBAAwB,IAAI;AAAA,MACxC;AAAA,MACA,CAAC,QAAQ;AACP,YAAI,KAAK;AAAa;AACtB,kBAAU,mBAAmB,GAAG;AAAA,MAClC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA,EAIA,kBAAkB;AAChB,SAAK,KAAK,YACP,KAAK,CAAC,SAAS,KAAK,MAAM,CAAC,EAC3B,MAAM,MAAM;AAAA,IAEb,CAAC;AAAA,EACL;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,iBAAa,KAAK,iBAAiB;AACnC,SAAK,oBAAoB;AAAA,EAC3B;AAAA,EAEA,eAAqB;AAEnB,SAAK,gBAAgB;AACrB,UAAM,aAAa;AAAA,EACrB;AACF;;;AC7DO,IAAM,sBAAN,cAAkC,kBAAkB;AAAA,EAChD;AAAA,EACT;AAAA,EAEA;AAAA,EAEA,YACE,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,YAAY;AAEjB,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,UAAU,4BAA4B;AAAA,IAC7C,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AAAA,EACrB;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAEvB,QAAI,KAAK,oBAAoB;AAC3B,mBAAa,KAAK,kBAAkB;AACpC,WAAK,qBAAqB;AAAA,IAC5B;AAAA,EACF;AACF;;;AC9BO,IAAM,6BAAN,cAEG,cAAc;AAAA,EACb;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,mBAAmB,WAAW,MAAM;AACvC,gBAAU,mBAAmB;AAAA,IAC/B,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AACxD,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AAAA,EACzD;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU,mBAAmB,yBAAyB;AAC3D;AAAA,IACF;AAIA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,IAAI,kBAAmC;AACrC,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK,KAAK;AAAA,IACpB;AAAA,EACF;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,iBAAa,KAAK,gBAAgB;AAClC,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,eAAqB;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;ACrDO,IAAM,qBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,mBAAmB,WAAW,MAAM;AACvC,gBAAU,mBAAmB;AAAA,IAC/B,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AACxD,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AAAA,EACzD;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU,mBAAmB,yBAAyB;AAC3D;AAAA,IACF;AAEA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,iBAAa,KAAK,gBAAgB;AAAA,EACpC;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;AChEA,SAAS,sBAAsB;AAexB,IAAM,mBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEQ;AAAA,EACA,wBAAwB;AAAA,EAExB;AAAA,EAER,IAAI,yBAAyB;AAC3B,WAAO,KAAK,0BAA0B;AAAA,EACxC;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AACjB,SAAK,wBAAwB;AAAA,EAC/B;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AACnC,SAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,cAAc,CAAC;AAC1D,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,KAAK,gBAAgB,KAAK,aAAa;AAC5C,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AACvD,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AAGxD,QAAI,KAAK,WAAW,SAAS,GAAG;AAC9B,WAAK,KAAK;AAAA,QACR,WAAW,KAAK,WAAW,MAAM;AAAA,QACjC,KAAK;AAAA,MACP;AAAA,IACF;AAEA,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,IAC5C;AAAA,EAIF;AAAA,EAEA,uBAAuB;AACrB,SAAK,wBAAwB,YAAY,MAAM;AAC7C,YAAM,SAAS,KAAK;AACpB,YAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,UAAI,UAAU,KAAK,QAAQ,qBAAqB;AAC9C,aAAK,KAAK;AAAA,UACR,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,KAAK,MAAM;AAChB,sBAAc,KAAK,qBAAqB;AACxC,aAAK,wBAAwB;AAC7B;AAAA,MACF;AAEA,WAAK,cAAc;AACnB,WAAK;AAAA,IACP,GAAG,KAAK,QAAQ,mBAAmB;AAAA,EACrC;AAAA,EAEA,uBAAuB;AACrB,UAAM,WACJ,KAAK,QAAQ,sBAAsB,KAAK,QAAQ;AAElD,QAAI,KAAK,wBAAwB;AAC/B,mBAAa,KAAK,sBAAsB;AACxC,WAAK,yBAAyB;AAAA,IAChC;AAEA,SAAK,yBAAyB,WAAW,MAAM;AAC7C,WAAK,KAAK;AAAA,QACR,yBAAyB,KAAK,EAAE,iDAAiD,QAAQ;AAAA,QACzF,KAAK;AAAA,MACP;AACA,WAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,WAAK,KAAK,MAAM;AAChB,WAAK,yBAAyB;AAAA,IAChC,GAAG,QAAQ;AAAA,EACb;AAAA,EAEQ,gBAAgB;AACtB,SAAK,KAAK,MAAM,qBAAqB,KAAK,eAAe;AACzD,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,CAAC,QAAoB;AACnC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc;AAAM;AAGxB,QAAI,UAAU,QAAQ,KAAK,KAAK;AAC9B,UAAI,UAAU,MAAM,KAAK,KAAK;AAC5B,aAAK,KAAK;AAAA,UACR,oCAAoC,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAAA,UAC1E;AAAA,YACE,GAAG,KAAK;AAAA,YACR,kBAAkB;AAAA,UACpB;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAC5F,aAAK,KAAK,MAAM,QAAQ;AAAA,UACtB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,UAAU,KAAK,UAAU;AAAA,UAC5B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,UAAU,iBAAiB,MAAM;AAAA,MACxC;AAEA;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gBAAgB;AAAA,MAC9B,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,SAAK,kBAAkB,UAAU,KAAK,UAAU,GAAG;AAGnD,QAAI,CAAC,MAAM,UAAU,YAAY,GAAG;AAClC,WAAK,UAAU,UAAU,SAAS;AAClC;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gCAAgC;AAAA,MAC9C,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAID,QAAI,CAAC,KAAK,wBAAwB;AAChC,WAAK,cAAc;AACnB,WAAK,qBAAqB;AAAA,IAC5B;AAAA,EACF;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,aAAa;AAC/C,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,kBAAc,KAAK,qBAAqB;AACxC,iBAAa,KAAK,sBAAsB;AACxC,SAAK,wBAAwB;AAC7B,SAAK,yBAAyB;AAAA,EAChC;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;AC1LA,SAAS,qBACP,SACiD;AACjD,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,EACV;AACF;AAyBO,IAAM,oBAAoB;AAAA,EAC/B,aAAa;AAAA,IACX,aACE,IACA,MACA,WACA,SACA,KACA;AACA,YAAM,KAAK,WAAW,WAAW,CAAC;AAClC,YAAM,YAAY,2BAA2B,IAAI,IAAI,IAAI;AACzD,YAAM,aAA4C,CAAC;AAEnD,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAEA,cAAQ,KAAK,KAAK,WAAW,QAAQ,EAAE,kCAAkC;AAAA,QACvE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,oBACE,MACA,MACA,WACA,SACA,KACsC;AACtC,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAEA,cAAQ,KAAK,KAAK,gDAAgD;AAAA,QAChE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA,EAGA,YAAY;AAAA;AAAA,IAEV,yBACE,YACA,aACA,WAC6B;AAC7B,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL;AACA,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,wBACE,YACA,MACA,WAC8B;AAC9B,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,mBAAmB,MAAM,WAAW,GAAG,YAAY;AACvE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,uBACE,YACA,WAC4B;AAC5B,YAAM,eAAe,qBAAqB,UAAU;AACpD,YAAM,OAAO,WAAW;AACxB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,iBAAiB,MAAM,WAAW,GAAG,YAAY;AACrE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,+BACE,gBACA,YACA,WACA,IACA,gBACA,WAC4B;AAC5B,YAAM,OAAO,eAAe;AAC5B,YAAM,EAAE,MAAM,QAAQ,IAAI;AAC1B,YAAM,eACJ;AAAA;AAAA,QAEI,qBAAqB,UAAU;AAAA;AAAA;AAAA,QAE/B;AAAA,UACE;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,CAAC;AAAA,UACD,2BAA2B,WAAW,IAAI,MAAM,cAAc;AAAA,UAC9D;AAAA,UACA,eAAe;AAAA,QACjB;AAAA;AAEN,qBAAe,iBAAiB;AAChC,kBAAY,iBAAiB;AAE7B,YAAM,UAAU,IAAI,iBAAiB,MAAM,WAAW,GAAG,YAAY;AACrE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,IAEA,yBACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,gBAAgB;AAC3B,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,0BACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,wBACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,EACF;AACF;","names":["SessionState"]}

package/dist/{client-2ba72e89.d.ts → client-1321630c.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection, T as Transport, o as ClientTransportOptions, L as LeakyBucketRateLimit, p as ClientHandshakeOptions, b as ProvidedClientTransportOptions, f as SessionConnecting, e as SessionNoConnection, g as SessionHandshaking, h as SessionConnected } from './handshake-0b88e8fc.js';
+import { C as Connection, T as Transport, o as ClientTransportOptions, L as LeakyBucketRateLimit, p as ClientHandshakeOptions, b as ProvidedClientTransportOptions, f as SessionConnecting, e as SessionNoConnection, g as SessionHandshaking, h as SessionConnected } from './handshake-3772d7ca.js';
 import { c as TransportClientId, P as PartialTransportMessage, b as OpaqueTransportMessage } from './message-e6c560fd.js';
 
 declare abstract class ClientTransport<ConnType extends Connection> extends Transport<ConnType> {

package/dist/{connection-55cba970.d.ts → connection-bd907ca6.d.ts}

@@ -1,4 +1,4 @@
-import { C as Connection } from './handshake-0b88e8fc.js';
+import { C as Connection } from './handshake-3772d7ca.js';
 import { W as WsLike } from './wslike-e0b32dd5.js';
 
 declare class WebSocketConnection extends Connection {

package/dist/{handshake-0b88e8fc.d.ts → handshake-3772d7ca.d.ts}

@@ -114,13 +114,15 @@ declare class SessionConnected<ConnType extends Connection> extends IdentifiedSe
     readonly state: SessionState.Connected;
     conn: ConnType;
     listeners: SessionConnectedListeners;
-    heartbeatHandle?: ReturnType<typeof setInterval> | undefined;
-    heartbeatMisses: number;
+    private activeHeartbeatHandle?;
+    private activeHeartbeatMisses;
+    private passiveHeartbeatHandle?;
     get isActivelyHeartbeating(): boolean;
     updateBookkeeping(ack: number, seq: number): void;
     send(msg: PartialTransportMessage): string;
     constructor(conn: ConnType, listeners: SessionConnectedListeners, ...args: ConstructorParameters<typeof IdentifiedSession>);
     startActiveHeartbeat(): void;
+    waitForNextHeartbeat(): void;
     private sendHeartbeat;
     onMessageData: (msg: Uint8Array) => void;
     _handleStateExit(): void;

package/dist/router/index.cjs

@@ -752,7 +752,7 @@ function Err(error) {
 var import_api = require("@opentelemetry/api");
 
 // package.json
-var version = "0.24.3";
+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);