@replit/river 0.23.8 → 0.23.10
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/{chunk-WQRQFAP6.js → chunk-B323CECK.js} +473 -15
- package/dist/chunk-B323CECK.js.map +1 -0
- package/dist/{chunk-ELZRZNA5.js → chunk-FRICSBDW.js} +2 -2
- package/dist/chunk-FRICSBDW.js.map +1 -0
- package/dist/{chunk-RJ6CXPBM.js → chunk-KBAZ5TWE.js} +129 -65
- package/dist/chunk-KBAZ5TWE.js.map +1 -0
- package/dist/{chunk-DXTG3E3B.js → chunk-UXQMGZKP.js} +2 -2
- package/dist/{chunk-CBNCT4B3.js → chunk-Z4G27Y2I.js} +2 -2
- package/dist/{connection-6ce35bd5.d.ts → connection-700340c4.d.ts} +1 -1
- package/dist/{connection-a3fdfa3a.d.ts → connection-efcd4e1a.d.ts} +1 -1
- package/dist/router/index.cjs +473 -14
- package/dist/router/index.cjs.map +1 -1
- package/dist/router/index.d.cts +41 -7
- package/dist/router/index.d.ts +41 -7
- package/dist/router/index.js +4 -2
- package/dist/{services-fd8a9894.d.ts → services-409c5545.d.ts} +10 -9
- package/dist/transport/impls/uds/client.cjs +122 -58
- package/dist/transport/impls/uds/client.cjs.map +1 -1
- package/dist/transport/impls/uds/client.d.cts +2 -2
- package/dist/transport/impls/uds/client.d.ts +2 -2
- package/dist/transport/impls/uds/client.js +3 -3
- package/dist/transport/impls/uds/server.cjs +103 -47
- package/dist/transport/impls/uds/server.cjs.map +1 -1
- package/dist/transport/impls/uds/server.d.cts +2 -2
- package/dist/transport/impls/uds/server.d.ts +2 -2
- package/dist/transport/impls/uds/server.js +3 -3
- package/dist/transport/impls/ws/client.cjs +125 -58
- package/dist/transport/impls/ws/client.cjs.map +1 -1
- package/dist/transport/impls/ws/client.d.cts +2 -2
- package/dist/transport/impls/ws/client.d.ts +2 -2
- package/dist/transport/impls/ws/client.js +6 -3
- package/dist/transport/impls/ws/client.js.map +1 -1
- package/dist/transport/impls/ws/server.cjs +103 -47
- package/dist/transport/impls/ws/server.cjs.map +1 -1
- package/dist/transport/impls/ws/server.d.cts +2 -2
- package/dist/transport/impls/ws/server.d.ts +2 -2
- package/dist/transport/impls/ws/server.js +3 -3
- package/dist/transport/index.cjs +128 -64
- package/dist/transport/index.cjs.map +1 -1
- package/dist/transport/index.d.cts +1 -1
- package/dist/transport/index.d.ts +1 -1
- package/dist/transport/index.js +2 -2
- package/dist/{transport-3d34f714.d.ts → transport-cf856c41.d.ts} +46 -24
- package/dist/util/testHelpers.cjs +46 -4
- package/dist/util/testHelpers.cjs.map +1 -1
- package/dist/util/testHelpers.d.cts +2 -2
- package/dist/util/testHelpers.d.ts +2 -2
- package/dist/util/testHelpers.js +3 -3
- package/package.json +3 -3
- package/dist/chunk-ELZRZNA5.js.map +0 -1
- package/dist/chunk-RJ6CXPBM.js.map +0 -1
- package/dist/chunk-WQRQFAP6.js.map +0 -1
- /package/dist/{chunk-DXTG3E3B.js.map → chunk-UXQMGZKP.js.map} +0 -0
- /package/dist/{chunk-CBNCT4B3.js.map → chunk-Z4G27Y2I.js.map} +0 -0
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../transport/message.ts","../package.json","../util/stringify.ts","../tracing/index.ts"],"sourcesContent":["import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { PropagationContext } from '../tracing';\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 metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const ControlMessageHandshakeResponseSchema = Type.Object({\n type: Type.Literal('HANDSHAKE_RESP'),\n status: Type.Union([\n Type.Object({\n ok: Type.Literal(true),\n sessionId: Type.String(),\n }),\n Type.Object({\n ok: Type.Literal(false),\n reason: Type.String(),\n }),\n ]),\n});\n\nexport const ControlMessagePayloadSchema = Type.Union([\n ControlMessageCloseSchema,\n ControlMessageAckSchema,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n]);\n\n/**\n * Defines the schema for an opaque transport message that is agnostic to any\n * procedure/service.\n * @returns The transport message schema.\n */\nexport const OpaqueTransportMessageSchema = TransportMessageSchema(\n Type.Unknown(),\n);\n\n/**\n * Represents a transport message. This is the same type as {@link TransportMessageSchema} but\n * we can't statically infer generics from generic Typebox schemas so we have to define it again here.\n *\n * TypeScript can't enforce types when a bitmask is involved, so these are the semantics of\n * `controlFlags`:\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `streamId` must be set to a unique value\n * (suggestion: use `nanoid`).\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `serviceName` and `procedureName` must be set.\n * * If `controlFlags & StreamClosedBit == StreamClosedBit` and the kind is `stream` or `subscription`,\n * `payload` should be discarded (usually contains a control message).\n * * If `controlFlags & AckBit == AckBit`, the message is an explicit acknowledgement message and doesn't\n * contain any payload that is relevant to the application so should not be delivered.\n * @template Payload The type of the payload.\n */\nexport interface TransportMessage<Payload = unknown> {\n id: string;\n from: string;\n to: string;\n seq: number;\n ack: number;\n serviceName?: string;\n procedureName?: string;\n streamId: string;\n controlFlags: number;\n tracing?: 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: TransportClientId,\n to: TransportClientId,\n sessionId: string,\n metadata?: unknown,\n tracing?: PropagationContext,\n): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n tracing,\n payload: {\n type: 'HANDSHAKE_REQ',\n protocolVersion: PROTOCOL_VERSION,\n sessionId,\n metadata,\n } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n };\n}\n\nexport function handshakeResponseMessage(\n from: TransportClientId,\n to: TransportClientId,\n status: Static<typeof ControlMessageHandshakeResponseSchema>['status'],\n): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n payload: {\n type: 'HANDSHAKE_RESP',\n status,\n } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n };\n}\n\n/**\n * A type alias for a transport message with an opaque payload.\n * @template T - The type of the opaque payload.\n */\nexport type OpaqueTransportMessage = TransportMessage;\nexport type TransportClientId = string;\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an ack message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the AckBit, false otherwise.\n */\nexport function isAck(controlFlag: number): boolean {\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n return (controlFlag & ControlFlags.AckBit) === ControlFlags.AckBit;\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream open message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamOpenBit, false otherwise.\n */\nexport function isStreamOpen(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamOpenBit) === ControlFlags.StreamOpenBit\n );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream close message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamCloseBit, false otherwise.\n */\nexport function isStreamClose(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamClosedBit) ===\n ControlFlags.StreamClosedBit\n );\n}\n","{\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.23.8\",\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 --test-timeout=500\",\n \"test:single\": \"vitest run --test-timeout=500 --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 Session,\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 session: Session<Connection>,\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 ${session.id}`,\n {\n attributes: {\n component: 'river',\n 'river.session.id': session.id,\n 'river.session.to': session.to,\n 'river.session.from': session.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,YAA6B;AACtC,SAAS,cAAc;AAsBhB,IAAM,yBAAyB,CAAoB,MACxD,KAAK,OAAO;AAAA,EACV,IAAI,KAAK,OAAO;AAAA,EAChB,MAAM,KAAK,OAAO;AAAA,EAClB,IAAI,KAAK,OAAO;AAAA,EAChB,KAAK,KAAK,QAAQ;AAAA,EAClB,KAAK,KAAK,QAAQ;AAAA,EAClB,aAAa,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EACxC,eAAe,KAAK,SAAS,KAAK,OAAO,CAAC;AAAA,EAC1C,UAAU,KAAK,OAAO;AAAA,EACtB,cAAc,KAAK,QAAQ;AAAA,EAC3B,SAAS,KAAK;AAAA,IACZ,KAAK,OAAO;AAAA,MACV,aAAa,KAAK,OAAO;AAAA,MACzB,YAAY,KAAK,OAAO;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EACA,SAAS;AACX,CAAC;AAOI,IAAM,0BAA0B,KAAK,OAAO;AAAA,EACjD,MAAM,KAAK,QAAQ,KAAK;AAC1B,CAAC;AAMM,IAAM,4BAA4B,KAAK,OAAO;AAAA,EACnD,MAAM,KAAK,QAAQ,OAAO;AAC5B,CAAC;AAEM,IAAM,mBAAmB;AACzB,IAAM,uCAAuC,KAAK,OAAO;AAAA,EAC9D,MAAM,KAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,KAAK,OAAO;AAAA,EAC7B,WAAW,KAAK,OAAO;AAAA,EACvB,UAAU,KAAK,SAAS,KAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,wCAAwC,KAAK,OAAO;AAAA,EAC/D,MAAM,KAAK,QAAQ,gBAAgB;AAAA,EACnC,QAAQ,KAAK,MAAM;AAAA,IACjB,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,IAAI;AAAA,MACrB,WAAW,KAAK,OAAO;AAAA,IACzB,CAAC;AAAA,IACD,KAAK,OAAO;AAAA,MACV,IAAI,KAAK,QAAQ,KAAK;AAAA,MACtB,QAAQ,KAAK,OAAO;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AACH,CAAC;AAEM,IAAM,8BAA8B,KAAK,MAAM;AAAA,EACpD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,+BAA+B;AAAA,EAC1C,KAAK,QAAQ;AACf;AAoCO,SAAS,wBACd,MACA,IACA,WACA,UACA,SACuE;AACvE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd;AAAA,IACA,SAAS;AAAA,MACP,MAAM;AAAA,MACN,iBAAiB;AAAA,MACjB;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,yBACd,MACA,IACA,QACwE;AACxE,SAAO;AAAA,IACL,IAAI,OAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAcO,SAAS,MAAM,aAA8B;AAElD,UAAQ,cAAc,oBAAyB;AACjD;AAOO,SAAS,aAAa,aAA8B;AACzD;AAAA;AAAA,KAEG,cAAc,2BAAgC;AAAA;AAEnD;AAOO,SAAS,cAAc,aAA8B;AAC1D;AAAA;AAAA,KAEG,cAAc,6BACf;AAAA;AAEJ;;;ACtNE,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;AAoBA,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,SACA,gBACe;AACf,QAAM,YAAY,iBACd,YAAY,QAAQ,QAAQ,OAAO,GAAG,cAAc,IACpD,QAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,QAAQ,EAAE;AAAA,IACrB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB,QAAQ;AAAA,QAC5B,oBAAoB,QAAQ;AAAA,QAC5B,sBAAsB,QAAQ;AAAA,MAChC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,QAAM,MAAM,MAAM,QAAQ,WAAW,IAAI;AAEzC,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,8BACd,YACA,MACe;AACf,QAAM,OAAO,OAAO;AAAA,IAClB,cAAc,WAAW,EAAE;AAAA,IAC3B;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,uBAAuB,WAAW;AAAA,MACpC;AAAA,MACA,OAAO,CAAC,EAAE,SAAS,KAAK,KAAK,YAAY,EAAE,CAAC;AAAA,IAC9C;AAAA,IACA,KAAK;AAAA,EACP;AAEA,QAAM,MAAM,MAAM,QAAQ,KAAK,KAAK,IAAI;AAExC,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,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":[]}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../transport/session.ts","../transport/events.ts","../transport/transport.ts","../transport/rateLimit.ts"],"sourcesContent":["import { customAlphabet } from 'nanoid';\nimport {\n ControlFlags,\n ControlMessageAckSchema,\n OpaqueTransportMessage,\n PartialTransportMessage,\n TransportClientId,\n TransportMessage,\n} from './message';\nimport { Codec } from '../codec';\nimport { Logger, MessageMetadata } from '../logging/log';\nimport { Static } from '@sinclair/typebox';\nimport {\n PropagationContext,\n TelemetryInfo,\n createSessionTelemetryInfo,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nconst nanoid = customAlphabet('1234567890abcdefghijklmnopqrstuvxyz', 6);\nexport const unsafeId = () => nanoid();\n\ntype SequenceNumber = number;\n\n/**\n * A connection is the actual raw underlying transport connection.\n * It’s responsible for dispatching to/from the actual connection itself\n * This should be instantiated as soon as the client/server has a connection\n * It’s tied to the lifecycle of the underlying transport connection (i.e. if the WS drops, this connection should be deleted)\n */\nexport abstract class Connection {\n id: string;\n telemetry?: TelemetryInfo;\n constructor() {\n this.id = `conn-${nanoid(12)}`; // for debugging, no collision safety needed\n }\n\n get loggingMetadata(): MessageMetadata {\n const metadata: MessageMetadata = { connId: this.id };\n const spanContext = this.telemetry?.span.spanContext();\n if (spanContext) {\n metadata.telemetry = {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n };\n }\n\n return metadata;\n }\n\n /**\n * Handle adding a callback for when a message is received.\n * @param msg The message that was received.\n */\n abstract addDataListener(cb: (msg: Uint8Array) => void): void;\n abstract removeDataListener(cb: (msg: Uint8Array) => void): void;\n\n /**\n * Handle adding a callback for when the connection is closed.\n * This should also be called if an error happens.\n * @param cb The callback to call when the connection is closed.\n */\n abstract addCloseListener(cb: () => void): void;\n\n /**\n * Handle adding a callback for when an error is received.\n * This should only be used for this.logging errors, all cleanup\n * should be delegated to addCloseListener.\n *\n * The implementer should take care such that the implemented\n * connection will call both the close and error callbacks\n * on an error.\n *\n * @param cb The callback to call when an error is received.\n */\n abstract addErrorListener(cb: (err: Error) => void): void;\n\n /**\n * Sends a message over the connection.\n * @param msg The message to send.\n * @returns true if the message was sent, false otherwise.\n */\n abstract send(msg: Uint8Array): boolean;\n\n /**\n * Closes the connection.\n */\n abstract close(): void;\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 * The codec to use for encoding/decoding messages over the wire\n */\n codec: Codec;\n}\n\n/**\n * A session is a higher-level abstraction that operates over the span of potentially multiple transport-level connections\n * - It’s responsible for tracking any metadata for a particular client that might need to be persisted across connections (i.e. the sendBuffer, ack, seq)\n * - This will only be considered disconnected if\n * - the server tells the client that we’ve reconnected but it doesn’t recognize us anymore (server definitely died) or\n * - we hit a grace period after a connection disconnect\n */\nexport class Session<ConnType extends Connection> {\n private codec: Codec;\n private options: SessionOptions;\n telemetry: TelemetryInfo;\n\n /**\n * The buffer of messages that have been sent but not yet acknowledged.\n */\n private sendBuffer: Array<OpaqueTransportMessage> = [];\n\n /**\n * The active connection associated with this session\n */\n connection?: ConnType;\n readonly from: TransportClientId;\n readonly to: TransportClientId;\n\n /**\n * The unique ID of this session.\n */\n id: string;\n\n /**\n * What the other side advertised as their session ID\n * for this session.\n */\n advertisedSessionId?: string;\n\n /**\n * Number of messages we've sent along this session (excluding handshake and acks)\n */\n private seq: SequenceNumber = 0;\n\n /**\n * Number of unique messages we've received this session (excluding handshake and acks)\n */\n private ack: SequenceNumber = 0;\n\n /**\n * The grace period between when the inner connection is disconnected\n * and when we should consider the entire session disconnected.\n */\n private disconnectionGrace?: ReturnType<typeof setTimeout>;\n\n /**\n * Number of heartbeats we've sent without a response.\n */\n private heartbeatMisses: number;\n\n /**\n * The interval for sending heartbeats.\n */\n private heartbeat: ReturnType<typeof setInterval>;\n private log?: Logger;\n\n constructor(\n conn: ConnType | undefined,\n from: TransportClientId,\n to: TransportClientId,\n options: SessionOptions,\n propagationCtx?: PropagationContext,\n ) {\n this.id = `session-${nanoid(12)}`;\n this.options = options;\n this.from = from;\n this.to = to;\n this.connection = conn;\n this.codec = options.codec;\n\n // setup heartbeat\n this.heartbeatMisses = 0;\n this.heartbeat = setInterval(\n () => this.sendHeartbeat(),\n options.heartbeatIntervalMs,\n );\n this.telemetry = createSessionTelemetryInfo(this, propagationCtx);\n }\n\n bindLogger(log: Logger) {\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 connId: this.connection?.id,\n telemetry: {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n },\n };\n }\n\n /**\n * Sends a message over the session's connection.\n * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.\n *\n * @param msg The partial message to be sent, which will be constructed into a full message.\n * @param addToSendBuff Whether to add the message to the send buffer for retry.\n * @returns The full transport ID of the message that was attempted to be sent.\n */\n send(msg: PartialTransportMessage): string {\n const fullMsg: TransportMessage = this.constructMsg(msg);\n this.log?.debug(`sending msg`, {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n });\n\n if (this.connection) {\n const ok = this.connection.send(this.codec.toBuffer(fullMsg));\n if (ok) return fullMsg.id;\n this.log?.info(\n `failed to send msg to ${fullMsg.to}, connection is probably dead`,\n {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n },\n );\n } else {\n this.log?.debug(\n `buffering msg to ${fullMsg.to}, connection not ready yet`,\n { ...this.loggingMetadata, transportMessage: fullMsg },\n );\n }\n\n return fullMsg.id;\n }\n\n sendHeartbeat() {\n const misses = this.heartbeatMisses;\n const missDuration = misses * this.options.heartbeatIntervalMs;\n if (misses > this.options.heartbeatsUntilDead) {\n if (this.connection) {\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.closeStaleConnection();\n }\n return;\n }\n\n this.send({\n streamId: 'heartbeat',\n controlFlags: ControlFlags.AckBit,\n payload: {\n type: 'ACK',\n } satisfies Static<typeof ControlMessageAckSchema>,\n });\n this.heartbeatMisses++;\n }\n\n resetBufferedMessages() {\n this.sendBuffer = [];\n this.seq = 0;\n this.ack = 0;\n }\n\n sendBufferedMessages(conn: ConnType) {\n this.log?.info(`resending ${this.sendBuffer.length} buffered messages`, {\n ...this.loggingMetadata,\n connId: conn.id,\n });\n for (const msg of this.sendBuffer) {\n this.log?.debug(`resending msg`, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n });\n const ok = conn.send(this.codec.toBuffer(msg));\n if (!ok) {\n // this should never happen unless the transport has an\n // incorrect implementation of `createNewOutgoingConnection`\n const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: errMsg,\n });\n\n this.log?.error(errMsg, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n tags: ['invariant-violation'],\n });\n conn.close();\n return;\n }\n }\n }\n\n updateBookkeeping(ack: number, seq: number) {\n if (seq + 1 < this.ack) {\n this.log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {\n ...this.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n this.ack = seq + 1;\n }\n\n closeStaleConnection(conn?: ConnType) {\n if (this.connection === undefined || this.connection === conn) return;\n this.log?.info(\n `closing old inner connection from session to ${this.to}`,\n this.loggingMetadata,\n );\n this.connection.close();\n this.connection = undefined;\n }\n\n replaceWithNewConnection(newConn: ConnType) {\n this.closeStaleConnection(newConn);\n this.cancelGrace();\n this.sendBufferedMessages(newConn);\n this.connection = newConn;\n }\n\n beginGrace(cb: () => void) {\n this.log?.info(\n `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,\n this.loggingMetadata,\n );\n this.disconnectionGrace = setTimeout(() => {\n this.close();\n cb();\n }, this.options.sessionDisconnectGraceMs);\n }\n\n // called on reconnect of the underlying session\n cancelGrace() {\n this.heartbeatMisses = 0;\n clearTimeout(this.disconnectionGrace);\n this.disconnectionGrace = undefined;\n }\n\n // closed when we want to discard the whole session\n // (i.e. shutdown or session disconnect)\n close() {\n this.closeStaleConnection();\n this.cancelGrace();\n this.resetBufferedMessages();\n clearInterval(this.heartbeat);\n }\n\n get connected() {\n return this.connection !== undefined;\n }\n\n get nextExpectedSeq() {\n return this.ack;\n }\n\n constructMsg<Payload>(\n partialMsg: PartialTransportMessage<Payload>,\n ): TransportMessage<Payload> {\n const msg = {\n ...partialMsg,\n id: unsafeId(),\n to: this.to,\n from: this.from,\n seq: this.seq,\n ack: this.ack,\n };\n\n this.seq++;\n this.sendBuffer.push(msg);\n return msg;\n }\n\n inspectSendBuffer(): ReadonlyArray<OpaqueTransportMessage> {\n return this.sendBuffer;\n }\n}\n","import { OpaqueTransportMessage } from './message';\nimport { Connection, Session } from './session';\n\ntype ConnectionStatus = 'connect' | 'disconnect';\nexport const ProtocolError = {\n RetriesExceeded: 'conn_retry_exceeded',\n HandshakeFailed: 'handshake_failed',\n UseAfterDestroy: 'use_after_destroy',\n MessageOrderingViolated: 'message_ordering_violated',\n} as const;\n\nexport type ProtocolErrorType =\n (typeof ProtocolError)[keyof typeof ProtocolError];\n\nexport interface EventMap {\n message: OpaqueTransportMessage;\n connectionStatus: {\n status: ConnectionStatus;\n conn: Connection;\n };\n sessionStatus: {\n status: ConnectionStatus;\n session: Session<Connection>;\n };\n protocolError: {\n type: ProtocolErrorType;\n message: string;\n };\n}\n\nexport type EventTypes = keyof EventMap;\nexport type EventHandler<K extends EventTypes> = (\n event: EventMap[K],\n) => unknown;\n\nexport class EventDispatcher<T extends EventTypes> {\n private eventListeners: { [K in T]?: Set<EventHandler<K>> } = {};\n\n numberOfListeners<K extends T>(eventType: K) {\n return this.eventListeners[eventType]?.size ?? 0;\n }\n\n addEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n if (!this.eventListeners[eventType]) {\n this.eventListeners[eventType] = new Set();\n }\n\n this.eventListeners[eventType]?.add(handler);\n }\n\n removeEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n this.eventListeners[eventType]?.delete(handler);\n }\n }\n\n dispatchEvent<K extends T>(eventType: K, event: EventMap[K]) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n for (const handler of handlers) {\n handler(event);\n }\n }\n }\n}\n","import { Codec } from '../codec/types';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n OpaqueTransportMessage,\n OpaqueTransportMessageSchema,\n TransportClientId,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n handshakeRequestMessage,\n handshakeResponseMessage,\n PartialTransportMessage,\n ControlFlags,\n ControlMessagePayloadSchema,\n isAck,\n PROTOCOL_VERSION,\n} from './message';\nimport {\n BaseLogger,\n LogFn,\n Logger,\n LoggingLevel,\n createLogProxy,\n} from '../logging/log';\nimport {\n EventDispatcher,\n EventHandler,\n EventTypes,\n ProtocolError,\n ProtocolErrorType,\n} from './events';\nimport { Connection, Session, SessionOptions } from './session';\nimport { Static } from '@sinclair/typebox';\nimport { coerceErrorString } from '../util/stringify';\nimport { ConnectionRetryOptions, LeakyBucketRateLimit } from './rateLimit';\nimport { NaiveJsonCodec } from '../codec';\nimport tracer, {\n PropagationContext,\n createConnectionTelemetryInfo,\n getPropagationContext,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\nimport {\n ClientHandshakeOptions,\n ServerHandshakeOptions,\n} from '../router/handshake';\n\n/**\n * Represents the possible states of a transport.\n * @property {'open'} open - The transport is open and operational (note that this doesn't mean it is actively connected)\n * @property {'closed'} closed - The transport is closed and not operational, but can be reopened.\n * @property {'destroyed'} destroyed - The transport is permanently destroyed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed' | 'destroyed';\n\ntype 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 codec: NaiveJsonCodec,\n};\n\ntype 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\nconst defaultClientTransportOptions: ClientTransportOptions = {\n ...defaultTransportOptions,\n ...defaultConnectionRetryOptions,\n};\n\ntype ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nconst defaultServerTransportOptions: ServerTransportOptions = {\n ...defaultTransportOptions,\n};\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions and connections. Its responsibilities include:\n *\n * 1) Constructing a new {@link Session} and {@link Connection} on {@link TransportMessage}s from new clients.\n * After constructing the {@link Connection}, {@link onConnect} is called which adds it to the connection map.\n * 2) Delegating message listening of the connection to the newly created {@link Connection}.\n * From this point on, the {@link Connection} is responsible for *reading* and *writing*\n * messages from the connection.\n * 3) When a connection is closed, the {@link Transport} calls {@link onDisconnect} which closes the\n * connection via {@link Connection.close} and removes it from the {@link connections} map.\n\n *\n * ```plaintext\n * ▲\n * incoming │\n * messages │\n * ▼\n * ┌─────────────┐ 1:N ┌───────────┐ 1:1* ┌────────────┐\n * │ Transport │ ◄─────► │ Session │ ◄─────► │ Connection │\n * └─────────────┘ └───────────┘ └────────────┘\n * ▲ * (may or may not be initialized yet)\n * │\n * ▼\n * ┌───────────┐\n * │ Message │\n * │ Listeners │\n * └───────────┘\n * ```\n * @abstract\n */\nexport abstract class Transport<ConnType extends Connection> {\n /**\n * A flag indicating whether the transport has been destroyed.\n * A destroyed transport will not attempt to reconnect and cannot be used again.\n */\n state: TransportStatus;\n\n /**\n * The {@link Codec} used to encode and decode messages.\n */\n codec: Codec;\n\n /**\n * The client ID of this transport.\n */\n clientId: TransportClientId;\n\n /**\n * The map of {@link Session}s managed by this transport.\n */\n sessions: Map<TransportClientId, Session<ConnType>>;\n\n /**\n * The map of {@link Connection}s managed by this transport.\n */\n get connections() {\n return new Map(\n [...this.sessions]\n .map(([client, session]) => [client, session.connection])\n .filter((entry): entry is [string, ConnType] => entry[1] !== undefined),\n );\n }\n\n /**\n * The event dispatcher for handling events of type EventTypes.\n */\n eventDispatcher: EventDispatcher<EventTypes>;\n\n /**\n * The options for this transport.\n */\n protected options: TransportOptions;\n log?: Logger;\n\n /**\n * Creates a new Transport instance.\n * This should also set up {@link onConnect}, and {@link onDisconnect} listeners.\n * @param codec The codec used to encode and decode messages.\n * @param clientId The client ID of this transport.\n */\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedTransportOptions,\n ) {\n this.options = { ...defaultTransportOptions, ...providedOptions };\n this.eventDispatcher = new EventDispatcher();\n this.sessions = new Map();\n this.codec = this.options.codec;\n this.clientId = clientId;\n this.state = 'open';\n }\n\n bindLogger(fn: LogFn | Logger, level?: LoggingLevel) {\n // construct logger from fn\n if (typeof fn === 'function') {\n this.log = createLogProxy(new BaseLogger(fn, level));\n return;\n }\n\n // object case, just assign\n this.log = createLogProxy(fn);\n }\n\n /**\n * This is called immediately after a new connection is established and we\n * may or may not know the identity of the connected client.\n * It should attach all the necessary listeners to the connection for lifecycle\n * events (i.e. data, close, error)\n *\n * This method is implemented by {@link ClientTransport} and {@link ServerTransport}.\n */\n protected abstract handleConnection(\n conn: ConnType,\n to: TransportClientId,\n ): void;\n\n /**\n * Called when a new connection is established\n * and we know the identity of the connected client.\n * @param conn The connection object.\n */\n protected onConnect(\n conn: ConnType,\n connectedTo: TransportClientId,\n session: Session<ConnType>,\n isReconnect: boolean,\n ) {\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'connect',\n conn,\n });\n\n conn.telemetry = createConnectionTelemetryInfo(conn, session.telemetry);\n\n if (isReconnect) {\n session.replaceWithNewConnection(conn);\n this.log?.info(`reconnected to ${connectedTo}`, {\n ...conn.loggingMetadata,\n ...session.loggingMetadata,\n clientId: this.clientId,\n connectedTo,\n });\n }\n }\n\n protected createSession(\n to: TransportClientId,\n conn?: ConnType,\n propagationCtx?: PropagationContext,\n ) {\n const session = new Session<ConnType>(\n conn,\n this.clientId,\n to,\n this.options,\n propagationCtx,\n );\n\n if (this.log) {\n session.bindLogger(this.log);\n }\n\n this.sessions.set(session.to, session);\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'connect',\n session,\n });\n return session;\n }\n\n protected getOrCreateSession(\n to: TransportClientId,\n conn?: ConnType,\n sessionId?: string,\n propagationCtx?: PropagationContext,\n ) {\n let session = this.sessions.get(to);\n let isReconnect = session !== undefined;\n\n if (\n session?.advertisedSessionId !== undefined &&\n sessionId !== undefined &&\n session.advertisedSessionId !== sessionId\n ) {\n this.log?.info(\n `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,\n session.loggingMetadata,\n );\n this.deleteSession(session);\n isReconnect = false;\n session = undefined;\n }\n\n if (!session) {\n session = this.createSession(to, conn, propagationCtx);\n this.log?.info(\n `no session for ${to}, created a new one`,\n session.loggingMetadata,\n );\n }\n\n if (sessionId !== undefined) {\n session.advertisedSessionId = sessionId;\n }\n\n return { session, isReconnect };\n }\n\n protected deleteSession(session: Session<ConnType>) {\n session.close();\n session.telemetry.span.end();\n this.sessions.delete(session.to);\n this.log?.info(\n `session ${session.id} disconnect from ${session.to}`,\n session.loggingMetadata,\n );\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'disconnect',\n session,\n });\n }\n\n /**\n * The downstream implementation needs to call this when a connection is closed.\n * @param conn The connection object.\n * @param connectedTo The peer we are connected to.\n */\n protected onDisconnect(conn: ConnType, session: Session<ConnType>) {\n conn.telemetry?.span.end();\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'disconnect',\n conn,\n });\n\n session.connection = undefined;\n session.beginGrace(() => {\n session.telemetry.span.addEvent('session grace period expired');\n this.deleteSession(session);\n });\n }\n\n /**\n * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.\n * @param msg The message to parse.\n * @returns The parsed message, or null if the message is malformed or invalid.\n */\n protected parseMsg(\n msg: Uint8Array,\n conn: ConnType,\n ): OpaqueTransportMessage | null {\n const parsedMsg = this.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, killing conn: ${decodedBuffer}`,\n {\n clientId: this.clientId,\n ...conn.loggingMetadata,\n },\n );\n return null;\n }\n\n if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n this.log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n clientId: this.clientId,\n ...conn.loggingMetadata,\n validationErrors: [\n ...Value.Errors(OpaqueTransportMessageSchema, parsedMsg),\n ],\n });\n return null;\n }\n\n return parsedMsg;\n }\n\n /**\n * Called when a message is received by this transport.\n * You generally shouldn't need to override this in downstream transport implementations.\n * @param msg The received message.\n */\n protected handleMsg(msg: OpaqueTransportMessage, conn: ConnType) {\n if (this.state !== 'open') return;\n const session = this.sessions.get(msg.from);\n if (!session) {\n this.log?.error(`received message for unknown session from ${msg.from}`, {\n clientId: this.clientId,\n transportMessage: msg,\n ...conn.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n // got a msg so we know the other end is alive, reset the grace period\n session.cancelGrace();\n\n this.log?.debug(`received msg`, {\n clientId: this.clientId,\n transportMessage: msg,\n ...conn.loggingMetadata,\n });\n if (msg.seq !== session.nextExpectedSeq) {\n if (msg.seq < session.nextExpectedSeq) {\n this.log?.debug(\n `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,\n {\n clientId: this.clientId,\n transportMessage: msg,\n ...conn.loggingMetadata,\n },\n );\n } else {\n const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;\n this.log?.error(`${errMsg}, marking connection as dead`, {\n clientId: this.clientId,\n transportMessage: msg,\n ...conn.loggingMetadata,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);\n session.telemetry.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message order violated',\n });\n session.close();\n }\n\n return;\n }\n\n session.updateBookkeeping(msg.ack, msg.seq);\n\n // don't dispatch explicit acks\n if (!isAck(msg.controlFlags)) {\n this.eventDispatcher.dispatchEvent('message', msg);\n } else {\n this.log?.debug(`discarding msg (ack bit set)`, {\n clientId: this.clientId,\n transportMessage: msg,\n ...conn.loggingMetadata,\n });\n }\n }\n\n /**\n * Adds a listener to this transport.\n * @param the type of event to listen for\n * @param handler The message handler to add.\n */\n addEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.addEventListener(type, handler);\n }\n\n /**\n * Removes a listener from this transport.\n * @param the type of event to un-listen on\n * @param handler The message handler to remove.\n */\n removeEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.removeEventListener(type, handler);\n }\n\n /**\n * Sends a message over this transport, delegating to the appropriate connection to actually\n * send the message.\n * @param msg The message to send.\n * @returns The ID of the sent message or undefined if it wasn't sent\n */\n send(\n to: TransportClientId,\n msg: PartialTransportMessage,\n ): string | undefined {\n if (this.state === 'destroyed') {\n const err = 'transport is destroyed, cant send';\n this.log?.error(err, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.UseAfterDestroy, err);\n return undefined;\n } else if (this.state === 'closed') {\n this.log?.info(`transport closed when sending, discarding`, {\n clientId: this.clientId,\n transportMessage: msg,\n });\n return undefined;\n }\n\n return this.getOrCreateSession(to).session.send(msg);\n }\n\n // control helpers\n sendCloseStream(to: TransportClientId, streamId: string) {\n return this.send(to, {\n streamId: streamId,\n controlFlags: ControlFlags.StreamClosedBit,\n payload: {\n type: 'CLOSE' as const,\n } satisfies Static<typeof ControlMessagePayloadSchema>,\n });\n }\n\n protected protocolError(type: ProtocolErrorType, message: string) {\n this.eventDispatcher.dispatchEvent('protocolError', { type, message });\n }\n\n /**\n * Default close implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.close() at the end.\n * Closes the transport. Any messages sent while the transport is closed will be silently discarded.\n */\n close() {\n this.state = 'closed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n this.log?.info(`manually closed transport`, { clientId: this.clientId });\n }\n\n /**\n * Default destroy implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.destroy() at the end.\n * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.\n */\n destroy() {\n this.state = 'destroyed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n this.log?.info(`manually destroyed transport`, { clientId: this.clientId });\n }\n}\n\nexport abstract class ClientTransport<\n ConnType extends Connection,\n> extends Transport<ConnType> {\n /**\n * The options for this transport.\n */\n protected options: ClientTransportOptions;\n\n /**\n * The map of reconnect promises for each client ID.\n */\n inflightConnectionPromises: Map<TransportClientId, Promise<ConnType>>;\n retryBudget: LeakyBucketRateLimit;\n\n /**\n * A flag indicating whether the transport should automatically reconnect\n * when a connection is dropped.\n * Realistically, this should always be true for clients unless you are writing\n * tests or a special case where you don't want to reconnect.\n */\n reconnectOnConnectionDrop = true;\n\n /**\n * Optional handshake options for this client.\n */\n handshakeExtensions?: ClientHandshakeOptions;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedClientTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultClientTransportOptions,\n ...providedOptions,\n };\n this.inflightConnectionPromises = new Map();\n this.retryBudget = new LeakyBucketRateLimit(this.options);\n }\n\n extendHandshake(options: ClientHandshakeOptions) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType, to: TransportClientId): void {\n if (this.state !== 'open') return;\n let session: Session<ConnType> | undefined = undefined;\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n this.log?.warn(\n `connection to ${to} timed out waiting for handshake, closing`,\n { ...conn.loggingMetadata, clientId: this.clientId, connectedTo: to },\n );\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const handshakeHandler = (data: Uint8Array) => {\n const maybeSession = this.receiveHandshakeResponseMessage(data, conn);\n if (!maybeSession) {\n conn.close();\n return;\n } else {\n session = maybeSession;\n clearTimeout(handshakeTimeout);\n }\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener((data) => {\n const parsed = this.parseMsg(data, conn);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message parse failure',\n });\n conn.close();\n return;\n }\n\n this.handleMsg(parsed, conn);\n });\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (session) {\n this.onDisconnect(conn, session);\n }\n this.log?.info(`connection to ${to} disconnected`, {\n ...conn.loggingMetadata,\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n });\n this.inflightConnectionPromises.delete(to);\n if (this.reconnectOnConnectionDrop) {\n void this.connect(to);\n }\n });\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n this.log?.warn(\n `error in connection to ${to}: ${coerceErrorString(err)}`,\n {\n ...conn.loggingMetadata,\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n },\n );\n });\n }\n\n receiveHandshakeResponseMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Session<ConnType> | false {\n const parsed = this.parseMsg(data, conn);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake response',\n });\n this.log?.warn(`received invalid handshake resp`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n validationErrors: [\n ...Value.Errors(\n ControlMessageHandshakeResponseSchema,\n parsed.payload,\n ),\n ],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake resp',\n );\n return false;\n }\n\n if (!parsed.payload.status.ok) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake rejected',\n });\n this.log?.warn(`received handshake rejection`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n parsed.payload.status.reason,\n );\n return false;\n }\n\n this.log?.debug(`handshake from ${parsed.from} ok`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n });\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.status.sessionId,\n );\n\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n // After a successful connection, we start restoring the budget\n // so that the next time we try to connect, we don't hit the client\n // with backoff forever.\n this.retryBudget.startRestoringBudget(parsed.from);\n return session;\n }\n\n /**\n * Abstract method that creates a new {@link Connection} object.\n * This should call {@link handleConnection} when the connection is created.\n * The downstream client implementation needs to implement this.\n *\n * @param to The client ID of the node to connect to.\n * @returns The new connection object.\n */\n protected abstract createNewOutgoingConnection(\n to: TransportClientId,\n ): Promise<ConnType>;\n\n /**\n * Manually attempts to connect to a client.\n * @param to The client ID of the node to connect to.\n */\n async connect(to: TransportClientId): Promise<void> {\n const canProceedWithConnection = () => this.state === 'open';\n if (!canProceedWithConnection()) {\n this.log?.info(\n `transport state is no longer open, cancelling attempt to connect to ${to}`,\n { clientId: this.clientId, connectedTo: to },\n );\n return;\n }\n\n let reconnectPromise = this.inflightConnectionPromises.get(to);\n if (!reconnectPromise) {\n // check budget\n const budgetConsumed = this.retryBudget.getBudgetConsumed(to);\n if (!this.retryBudget.hasBudget(to)) {\n const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;\n this.log?.error(errMsg, { clientId: this.clientId, connectedTo: to });\n this.protocolError(ProtocolError.RetriesExceeded, errMsg);\n return;\n }\n\n let sleep = Promise.resolve();\n const backoffMs = this.retryBudget.getBackoffMs(to);\n if (backoffMs > 0) {\n sleep = new Promise((resolve) => setTimeout(resolve, backoffMs));\n }\n\n this.log?.info(\n `attempting connection to ${to} (${backoffMs}ms backoff)`,\n {\n clientId: this.clientId,\n connectedTo: to,\n },\n );\n this.retryBudget.consumeBudget(to);\n reconnectPromise = tracer.startActiveSpan('connect', async (span) => {\n try {\n span.addEvent('backoff', { backoffMs });\n await sleep;\n if (!canProceedWithConnection()) {\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('connecting');\n const conn = await this.createNewOutgoingConnection(to);\n if (!canProceedWithConnection()) {\n this.log?.info(\n `transport state is no longer open, closing pre-handshake connection to ${to}`,\n {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n },\n );\n conn.close();\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('sending handshake');\n const ok = await this.sendHandshake(to, conn);\n if (!ok) {\n conn.close();\n throw new Error('failed to send handshake');\n }\n\n return conn;\n } catch (err) {\n // rethrow the error so that the promise is rejected\n // as it was before we wrapped it in a span\n const errStr = coerceErrorString(err);\n span.recordException(errStr);\n span.setStatus({ code: SpanStatusCode.ERROR });\n throw err;\n } finally {\n span.end();\n }\n });\n\n this.inflightConnectionPromises.set(to, reconnectPromise);\n } else {\n this.log?.info(\n `attempting connection to ${to} (reusing previous attempt)`,\n {\n clientId: this.clientId,\n connectedTo: to,\n },\n );\n }\n\n try {\n await reconnectPromise;\n } catch (error: unknown) {\n this.inflightConnectionPromises.delete(to);\n const errStr = coerceErrorString(error);\n\n if (!this.reconnectOnConnectionDrop || !canProceedWithConnection()) {\n this.log?.warn(`connection to ${to} failed (${errStr})`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n } else {\n this.log?.warn(`connection to ${to} failed (${errStr}), retrying`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n return this.connect(to);\n }\n }\n }\n\n protected deleteSession(session: Session<ConnType>) {\n this.inflightConnectionPromises.delete(session.to);\n super.deleteSession(session);\n }\n\n protected async sendHandshake(to: TransportClientId, conn: ConnType) {\n let metadata: unknown = undefined;\n\n if (this.handshakeExtensions) {\n metadata = await this.handshakeExtensions.construct();\n if (!Value.Check(this.handshakeExtensions.schema, metadata)) {\n this.log?.error(`constructed handshake metadata did not match schema`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n validationErrors: [\n ...Value.Errors(this.handshakeExtensions.schema, metadata),\n ],\n tags: ['invariant-violation'],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'handshake metadata did not match schema',\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake meta mismatch',\n });\n return false;\n }\n }\n\n // dont pass conn here as we dont want the session to start using the conn\n // until we have finished the handshake\n const { session } = this.getOrCreateSession(to);\n const requestMsg = handshakeRequestMessage(\n this.clientId,\n to,\n session.id,\n metadata,\n getPropagationContext(session.telemetry.ctx),\n );\n this.log?.debug(`sending handshake request to ${to}`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n transportMessage: requestMsg,\n });\n conn.send(this.codec.toBuffer(requestMsg));\n return true;\n }\n\n close() {\n this.retryBudget.close();\n super.close();\n }\n}\n\nexport abstract class ServerTransport<\n ConnType extends Connection,\n> extends Transport<ConnType> {\n /**\n * The options for this transport.\n */\n protected options: ServerTransportOptions;\n\n /**\n * Optional handshake options for the server.\n */\n handshakeExtensions?: ServerHandshakeOptions;\n\n /**\n * A map of session handshake data for each session.\n */\n sessionHandshakeMetadata: WeakMap<Session<ConnType>, ParsedMetadata>;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultServerTransportOptions,\n ...providedOptions,\n };\n this.sessionHandshakeMetadata = new WeakMap();\n this.log?.info(`initiated server transport`, {\n clientId: this.clientId,\n protocolVersion: PROTOCOL_VERSION,\n });\n }\n\n extendHandshake(options: ServerHandshakeOptions) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType) {\n if (this.state !== 'open') return;\n\n this.log?.info(`new incoming connection`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n });\n\n let session: Session<ConnType> | undefined = undefined;\n const client = () => session?.to ?? 'unknown';\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n this.log?.warn(\n `connection to ${client()} timed out waiting for handshake, closing`,\n {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n connectedTo: client(),\n },\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake timeout',\n });\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const buffer: Array<Uint8Array> = [];\n let receivedHandshakeMessage = false;\n\n const handshakeHandler = (data: Uint8Array) => {\n // if we've already received, just buffer the data\n if (receivedHandshakeMessage) {\n buffer.push(data);\n return;\n }\n\n receivedHandshakeMessage = true;\n clearTimeout(handshakeTimeout);\n\n void this.receiveHandshakeRequestMessage(data, conn).then(\n (maybeSession) => {\n if (!maybeSession) {\n conn.close();\n return;\n }\n\n session = maybeSession;\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n const dataHandler = (data: Uint8Array) => {\n const parsed = this.parseMsg(data, conn);\n if (!parsed) {\n conn.close();\n return;\n }\n\n this.handleMsg(parsed, conn);\n };\n\n // process any data we missed\n for (const data of buffer) {\n dataHandler(data);\n }\n\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener(dataHandler);\n buffer.length = 0;\n },\n );\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (!session) return;\n this.log?.info(`connection to ${client()} disconnected`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n });\n this.onDisconnect(conn, session);\n });\n\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n if (!session) return;\n this.log?.warn(\n `connection to ${client()} got an error: ${coerceErrorString(err)}`,\n { ...conn.loggingMetadata, clientId: this.clientId },\n );\n });\n }\n\n private async validateHandshakeMetadata(\n conn: ConnType,\n session: Session<ConnType> | undefined,\n rawMetadata: Static<\n typeof ControlMessageHandshakeRequestSchema\n >['metadata'],\n from: TransportClientId,\n ): Promise<ParsedMetadata | false> {\n let parsedMetadata: ParsedMetadata = {};\n if (this.handshakeExtensions) {\n // check that the metadata that was sent is the correct shape\n if (!Value.Check(this.handshakeExtensions.schema, rawMetadata)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'malformed handshake meta',\n });\n const reason = 'received malformed handshake metadata';\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.log?.warn(`received malformed handshake metadata from ${from}`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n validationErrors: [\n ...Value.Errors(this.handshakeExtensions.schema, rawMetadata),\n ],\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n const previousParsedMetadata = session\n ? this.sessionHandshakeMetadata.get(session)\n : undefined;\n\n parsedMetadata = await this.handshakeExtensions.validate(\n rawMetadata,\n previousParsedMetadata,\n );\n\n // handler rejected the connection\n if (parsedMetadata === false) {\n const reason = 'rejected by handshake handler';\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.log?.warn(`rejected handshake from ${from}`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n }\n\n return parsedMetadata;\n }\n\n async receiveHandshakeRequestMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Promise<Session<ConnType> | false> {\n const parsed = this.parseMsg(data, conn);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake request',\n });\n const reason = 'received invalid handshake msg';\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.log?.warn(reason, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n // safe to this.log metadata here as we remove the payload\n // before passing it to user-land\n transportMessage: parsed,\n validationErrors: [\n ...Value.Errors(ControlMessageHandshakeRequestSchema, parsed.payload),\n ],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake request',\n );\n return false;\n }\n\n // double check protocol version here\n const gotVersion = parsed.payload.protocolVersion;\n if (gotVersion !== PROTOCOL_VERSION) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'incorrect protocol version',\n });\n\n const reason = `incorrect version (got: ${gotVersion} wanted ${PROTOCOL_VERSION})`;\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.log?.warn(\n `received handshake msg with incompatible protocol version (got: ${gotVersion}, expected: ${PROTOCOL_VERSION})`,\n { ...conn.loggingMetadata, clientId: this.clientId },\n );\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n const oldSession = this.sessions.get(parsed.from);\n const parsedMetadata = await this.validateHandshakeMetadata(\n conn,\n oldSession,\n parsed.payload.metadata,\n parsed.from,\n );\n\n if (parsedMetadata === false) {\n return false;\n }\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.sessionId,\n parsed.tracing,\n );\n\n this.sessionHandshakeMetadata.set(session, parsedMetadata);\n\n this.log?.debug(\n `handshake from ${parsed.from} ok, responding with handshake success`,\n conn.loggingMetadata,\n );\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: true,\n sessionId: session.id,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n return session;\n }\n}\n","import { TransportClientId } from './message';\n\n/**\n * Options to control the backoff and retry behavior of the client transport's connection behaviour.\n *\n * River implements exponential backoff with jitter to prevent flooding the server\n * when there's an issue with connection establishment.\n *\n * The backoff is calculated via the following:\n * backOff = min(jitter + {@link baseIntervalMs} * 2 ^ budget_consumed, {@link maxBackoffMs})\n *\n * We use a leaky bucket rate limit with a budget of {@link attemptBudgetCapacity} reconnection attempts.\n * Budget only starts to restore after a successful handshake at a rate of one budget per {@link budgetRestoreIntervalMs}.\n */\nexport interface ConnectionRetryOptions {\n /**\n * The base interval to wait before retrying a connection.\n */\n baseIntervalMs: number;\n\n /**\n * The maximum random jitter to add to the total backoff time.\n */\n maxJitterMs: number;\n\n /**\n * The maximum amount of time to wait before retrying a connection.\n * This does not include the jitter.\n */\n maxBackoffMs: number;\n\n /**\n * The max number of times to attempt a connection before a successful handshake.\n * This persists across connections but starts restoring budget after a successful handshake.\n * The restoration interval depends on {@link budgetRestoreIntervalMs}\n */\n attemptBudgetCapacity: number;\n\n /**\n * After a successful connection attempt, how long to wait before we restore a single budget.\n */\n budgetRestoreIntervalMs: number;\n}\n\nexport class LeakyBucketRateLimit {\n private budgetConsumed: Map<TransportClientId, number>;\n private intervalHandles: Map<\n TransportClientId,\n ReturnType<typeof setInterval>\n >;\n private readonly options: ConnectionRetryOptions;\n\n constructor(options: ConnectionRetryOptions) {\n this.options = options;\n this.budgetConsumed = new Map();\n this.intervalHandles = new Map();\n }\n\n getBackoffMs(user: TransportClientId) {\n if (!this.budgetConsumed.has(user)) return 0;\n\n const exponent = Math.max(0, this.getBudgetConsumed(user) - 1);\n const jitter = Math.floor(Math.random() * this.options.maxJitterMs);\n const backoffMs = Math.min(\n this.options.baseIntervalMs * 2 ** exponent,\n this.options.maxBackoffMs,\n );\n\n return backoffMs + jitter;\n }\n\n get totalBudgetRestoreTime() {\n return (\n this.options.budgetRestoreIntervalMs * this.options.attemptBudgetCapacity\n );\n }\n\n consumeBudget(user: TransportClientId) {\n // If we're consuming again, let's ensure that we're not leaking\n this.stopLeak(user);\n this.budgetConsumed.set(user, this.getBudgetConsumed(user) + 1);\n }\n\n getBudgetConsumed(user: TransportClientId) {\n return this.budgetConsumed.get(user) ?? 0;\n }\n\n hasBudget(user: TransportClientId) {\n return this.getBudgetConsumed(user) < this.options.attemptBudgetCapacity;\n }\n\n startRestoringBudget(user: TransportClientId) {\n if (this.intervalHandles.has(user)) {\n return;\n }\n\n const restoreBudgetForUser = () => {\n const currentBudget = this.budgetConsumed.get(user);\n if (!currentBudget) {\n this.stopLeak(user);\n return;\n }\n\n const newBudget = currentBudget - 1;\n if (newBudget === 0) {\n this.budgetConsumed.delete(user);\n return;\n }\n\n this.budgetConsumed.set(user, newBudget);\n };\n\n const intervalHandle = setInterval(\n restoreBudgetForUser,\n this.options.budgetRestoreIntervalMs,\n );\n\n this.intervalHandles.set(user, intervalHandle);\n }\n\n private stopLeak(user: TransportClientId) {\n if (!this.intervalHandles.has(user)) {\n return;\n }\n\n clearInterval(this.intervalHandles.get(user));\n this.intervalHandles.delete(user);\n }\n\n close() {\n for (const user of this.intervalHandles.keys()) {\n this.stopLeak(user);\n }\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,sBAAsB;AAiB/B,SAAS,sBAAsB;AAE/B,IAAM,SAAS,eAAe,uCAAuC,CAAC;AAC/D,IAAM,WAAW,MAAM,OAAO;AAU9B,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQ,OAAO,EAAE,CAAC;AAAA,EAC9B;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,WAA4B,EAAE,QAAQ,KAAK,GAAG;AACpD,UAAM,cAAc,KAAK,WAAW,KAAK,YAAY;AACrD,QAAI,aAAa;AACf,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAwCF;AA6BO,IAAM,UAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKQ,aAA4C,CAAC;AAAA;AAAA;AAAA;AAAA,EAKrD;AAAA,EACS;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA;AAAA;AAAA;AAAA;AAAA,EAKQ,MAAsB;AAAA;AAAA;AAAA;AAAA,EAKtB,MAAsB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMtB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EACA;AAAA,EAER,YACE,MACA,MACA,IACA,SACA,gBACA;AACA,SAAK,KAAK,WAAW,OAAO,EAAE,CAAC;AAC/B,SAAK,UAAU;AACf,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,aAAa;AAClB,SAAK,QAAQ,QAAQ;AAGrB,SAAK,kBAAkB;AACvB,SAAK,YAAY;AAAA,MACf,MAAM,KAAK,cAAc;AAAA,MACzB,QAAQ;AAAA,IACV;AACA,SAAK,YAAY,2BAA2B,MAAM,cAAc;AAAA,EAClE;AAAA,EAEA,WAAW,KAAa;AACtB,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,QAAQ,KAAK,YAAY;AAAA,MACzB,WAAW;AAAA,QACT,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,KAAK,KAAsC;AACzC,UAAM,UAA4B,KAAK,aAAa,GAAG;AACvD,SAAK,KAAK,MAAM,eAAe;AAAA,MAC7B,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,KAAK,WAAW,KAAK,KAAK,MAAM,SAAS,OAAO,CAAC;AAC5D,UAAI;AAAI,eAAO,QAAQ;AACvB,WAAK,KAAK;AAAA,QACR,yBAAyB,QAAQ,EAAE;AAAA,QACnC;AAAA,UACE,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,QACpB;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK,KAAK;AAAA,QACR,oBAAoB,QAAQ,EAAE;AAAA,QAC9B,EAAE,GAAG,KAAK,iBAAiB,kBAAkB,QAAQ;AAAA,MACvD;AAAA,IACF;AAEA,WAAO,QAAQ;AAAA,EACjB;AAAA,EAEA,gBAAgB;AACd,UAAM,SAAS,KAAK;AACpB,UAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,QAAI,SAAS,KAAK,QAAQ,qBAAqB;AAC7C,UAAI,KAAK,YAAY;AACnB,aAAK,KAAK;AAAA,UACR,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,qBAAqB;AAAA,MAC5B;AACA;AAAA,IACF;AAEA,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AACD,SAAK;AAAA,EACP;AAAA,EAEA,wBAAwB;AACtB,SAAK,aAAa,CAAC;AACnB,SAAK,MAAM;AACX,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,qBAAqB,MAAgB;AACnC,SAAK,KAAK,KAAK,aAAa,KAAK,WAAW,MAAM,sBAAsB;AAAA,MACtE,GAAG,KAAK;AAAA,MACR,QAAQ,KAAK;AAAA,IACf,CAAC;AACD,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,KAAK,MAAM,iBAAiB;AAAA,QAC/B,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,QAClB,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,YAAM,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG,CAAC;AAC7C,UAAI,CAAC,IAAI;AAGP,cAAM,SAAS,sCAAsC,KAAK,EAAE;AAC5D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,KAAK,MAAM,QAAQ;AAAA,UACtB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,QAAQ,KAAK;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM;AACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,QAAI,MAAM,IAAI,KAAK,KAAK;AACtB,WAAK,KAAK,MAAM,sBAAsB,GAAG,UAAU,KAAK,GAAG,IAAI;AAAA,QAC7D,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAEA,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AAAA,EACnB;AAAA,EAEA,qBAAqB,MAAiB;AACpC,QAAI,KAAK,eAAe,UAAa,KAAK,eAAe;AAAM;AAC/D,SAAK,KAAK;AAAA,MACR,gDAAgD,KAAK,EAAE;AAAA,MACvD,KAAK;AAAA,IACP;AACA,SAAK,WAAW,MAAM;AACtB,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,yBAAyB,SAAmB;AAC1C,SAAK,qBAAqB,OAAO;AACjC,SAAK,YAAY;AACjB,SAAK,qBAAqB,OAAO;AACjC,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,WAAW,IAAgB;AACzB,SAAK,KAAK;AAAA,MACR,YAAY,KAAK,QAAQ,wBAAwB,oCAAoC,KAAK,EAAE;AAAA,MAC5F,KAAK;AAAA,IACP;AACA,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,MAAM;AACX,SAAG;AAAA,IACL,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA;AAAA,EAGA,cAAc;AACZ,SAAK,kBAAkB;AACvB,iBAAa,KAAK,kBAAkB;AACpC,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA,EAIA,QAAQ;AACN,SAAK,qBAAqB;AAC1B,SAAK,YAAY;AACjB,SAAK,sBAAsB;AAC3B,kBAAc,KAAK,SAAS;AAAA,EAC9B;AAAA,EAEA,IAAI,YAAY;AACd,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,SAAS;AAAA,MACb,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,SAAK,WAAW,KAAK,GAAG;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,oBAA2D;AACzD,WAAO,KAAK;AAAA,EACd;AACF;;;AC1YO,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AA0BO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,kBAA+B,WAAc;AAC3C,WAAO,KAAK,eAAe,SAAS,GAAG,QAAQ;AAAA,EACjD;AAAA,EAEA,iBAA8B,WAAc,SAA0B;AACpE,QAAI,CAAC,KAAK,eAAe,SAAS,GAAG;AACnC,WAAK,eAAe,SAAS,IAAI,oBAAI,IAAI;AAAA,IAC3C;AAEA,SAAK,eAAe,SAAS,GAAG,IAAI,OAAO;AAAA,EAC7C;AAAA,EAEA,oBAAiC,WAAc,SAA0B;AACvE,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,WAAK,eAAe,SAAS,GAAG,OAAO,OAAO;AAAA,IAChD;AAAA,EACF;AAAA,EAEA,cAA2B,WAAc,OAAoB;AAC3D,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,iBAAW,WAAW,UAAU;AAC9B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;AChEA,SAAS,aAAa;;;AC2Cf,IAAM,uBAAN,MAA2B;AAAA,EACxB;AAAA,EACA;AAAA,EAIS;AAAA,EAEjB,YAAY,SAAiC;AAC3C,SAAK,UAAU;AACf,SAAK,iBAAiB,oBAAI,IAAI;AAC9B,SAAK,kBAAkB,oBAAI,IAAI;AAAA,EACjC;AAAA,EAEA,aAAa,MAAyB;AACpC,QAAI,CAAC,KAAK,eAAe,IAAI,IAAI;AAAG,aAAO;AAE3C,UAAM,WAAW,KAAK,IAAI,GAAG,KAAK,kBAAkB,IAAI,IAAI,CAAC;AAC7D,UAAM,SAAS,KAAK,MAAM,KAAK,OAAO,IAAI,KAAK,QAAQ,WAAW;AAClE,UAAM,YAAY,KAAK;AAAA,MACrB,KAAK,QAAQ,iBAAiB,KAAK;AAAA,MACnC,KAAK,QAAQ;AAAA,IACf;AAEA,WAAO,YAAY;AAAA,EACrB;AAAA,EAEA,IAAI,yBAAyB;AAC3B,WACE,KAAK,QAAQ,0BAA0B,KAAK,QAAQ;AAAA,EAExD;AAAA,EAEA,cAAc,MAAyB;AAErC,SAAK,SAAS,IAAI;AAClB,SAAK,eAAe,IAAI,MAAM,KAAK,kBAAkB,IAAI,IAAI,CAAC;AAAA,EAChE;AAAA,EAEA,kBAAkB,MAAyB;AACzC,WAAO,KAAK,eAAe,IAAI,IAAI,KAAK;AAAA,EAC1C;AAAA,EAEA,UAAU,MAAyB;AACjC,WAAO,KAAK,kBAAkB,IAAI,IAAI,KAAK,QAAQ;AAAA,EACrD;AAAA,EAEA,qBAAqB,MAAyB;AAC5C,QAAI,KAAK,gBAAgB,IAAI,IAAI,GAAG;AAClC;AAAA,IACF;AAEA,UAAM,uBAAuB,MAAM;AACjC,YAAM,gBAAgB,KAAK,eAAe,IAAI,IAAI;AAClD,UAAI,CAAC,eAAe;AAClB,aAAK,SAAS,IAAI;AAClB;AAAA,MACF;AAEA,YAAM,YAAY,gBAAgB;AAClC,UAAI,cAAc,GAAG;AACnB,aAAK,eAAe,OAAO,IAAI;AAC/B;AAAA,MACF;AAEA,WAAK,eAAe,IAAI,MAAM,SAAS;AAAA,IACzC;AAEA,UAAM,iBAAiB;AAAA,MACrB;AAAA,MACA,KAAK,QAAQ;AAAA,IACf;AAEA,SAAK,gBAAgB,IAAI,MAAM,cAAc;AAAA,EAC/C;AAAA,EAEQ,SAAS,MAAyB;AACxC,QAAI,CAAC,KAAK,gBAAgB,IAAI,IAAI,GAAG;AACnC;AAAA,IACF;AAEA,kBAAc,KAAK,gBAAgB,IAAI,IAAI,CAAC;AAC5C,SAAK,gBAAgB,OAAO,IAAI;AAAA,EAClC;AAAA,EAEA,QAAQ;AACN,eAAW,QAAQ,KAAK,gBAAgB,KAAK,GAAG;AAC9C,WAAK,SAAS,IAAI;AAAA,IACpB;AAAA,EACF;AACF;;;AD9FA,SAAS,kBAAAA,uBAAsB;AAmBxB,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AAAA,EACH,GAAG;AACL;AAMA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AACL;AAgCO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA;AAAA,EAK3D;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,cAAc;AAChB,WAAO,IAAI;AAAA,MACT,CAAC,GAAG,KAAK,QAAQ,EACd,IAAI,CAAC,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,QAAQ,UAAU,CAAC,EACvD,OAAO,CAAC,UAAuC,MAAM,CAAC,MAAM,MAAS;AAAA,IAC1E;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA,EACV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW,oBAAI,IAAI;AACxB,SAAK,QAAQ,KAAK,QAAQ;AAC1B,SAAK,WAAW;AAChB,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,WAAW,IAAoB,OAAsB;AAEnD,QAAI,OAAO,OAAO,YAAY;AAC5B,WAAK,MAAM,eAAe,IAAI,WAAW,IAAI,KAAK,CAAC;AACnD;AAAA,IACF;AAGA,SAAK,MAAM,eAAe,EAAE;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAoBU,UACR,MACA,aACA,SACA,aACA;AACA,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,YAAY,8BAA8B,MAAM,QAAQ,SAAS;AAEtE,QAAI,aAAa;AACf,cAAQ,yBAAyB,IAAI;AACrC,WAAK,KAAK,KAAK,kBAAkB,WAAW,IAAI;AAAA,QAC9C,GAAG,KAAK;AAAA,QACR,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK;AAAA,QACf;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAEU,cACR,IACA,MACA,gBACA;AACA,UAAM,UAAU,IAAI;AAAA,MAClB;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACF;AAEA,QAAI,KAAK,KAAK;AACZ,cAAQ,WAAW,KAAK,GAAG;AAAA,IAC7B;AAEA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AACrC,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEU,mBACR,IACA,MACA,WACA,gBACA;AACA,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,QAAI,cAAc,YAAY;AAE9B,QACE,SAAS,wBAAwB,UACjC,cAAc,UACd,QAAQ,wBAAwB,WAChC;AACA,WAAK,KAAK;AAAA,QACR,eAAe,EAAE,6DAA6D,QAAQ,mBAAmB,UAAU,SAAS;AAAA,QAC5H,QAAQ;AAAA,MACV;AACA,WAAK,cAAc,OAAO;AAC1B,oBAAc;AACd,gBAAU;AAAA,IACZ;AAEA,QAAI,CAAC,SAAS;AACZ,gBAAU,KAAK,cAAc,IAAI,MAAM,cAAc;AACrD,WAAK,KAAK;AAAA,QACR,kBAAkB,EAAE;AAAA,QACpB,QAAQ;AAAA,MACV;AAAA,IACF;AAEA,QAAI,cAAc,QAAW;AAC3B,cAAQ,sBAAsB;AAAA,IAChC;AAEA,WAAO,EAAE,SAAS,YAAY;AAAA,EAChC;AAAA,EAEU,cAAc,SAA4B;AAClD,YAAQ,MAAM;AACd,YAAQ,UAAU,KAAK,IAAI;AAC3B,SAAK,SAAS,OAAO,QAAQ,EAAE;AAC/B,SAAK,KAAK;AAAA,MACR,WAAW,QAAQ,EAAE,oBAAoB,QAAQ,EAAE;AAAA,MACnD,QAAQ;AAAA,IACV;AACA,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,aAAa,MAAgB,SAA4B;AACjE,SAAK,WAAW,KAAK,IAAI;AACzB,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,aAAa;AACrB,YAAQ,WAAW,MAAM;AACvB,cAAQ,UAAU,KAAK,SAAS,8BAA8B;AAC9D,WAAK,cAAc,OAAO;AAAA,IAC5B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,SACR,KACA,MAC+B;AAC/B,UAAM,YAAY,KAAK,MAAM,WAAW,GAAG;AAE3C,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,KAAK;AAAA,QACR,yCAAyC,aAAa;AAAA,QACtD;AAAA,UACE,UAAU,KAAK;AAAA,UACf,GAAG,KAAK;AAAA,QACV;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,KAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QACpE,UAAU,KAAK;AAAA,QACf,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,8BAA8B,SAAS;AAAA,QACzD;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B,MAAgB;AAC/D,QAAI,KAAK,UAAU;AAAQ;AAC3B,UAAM,UAAU,KAAK,SAAS,IAAI,IAAI,IAAI;AAC1C,QAAI,CAAC,SAAS;AACZ,WAAK,KAAK,MAAM,6CAA6C,IAAI,IAAI,IAAI;AAAA,QACvE,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAGA,YAAQ,YAAY;AAEpB,SAAK,KAAK,MAAM,gBAAgB;AAAA,MAC9B,UAAU,KAAK;AAAA,MACf,kBAAkB;AAAA,MAClB,GAAG,KAAK;AAAA,IACV,CAAC;AACD,QAAI,IAAI,QAAQ,QAAQ,iBAAiB;AACvC,UAAI,IAAI,MAAM,QAAQ,iBAAiB;AACrC,aAAK,KAAK;AAAA,UACR,oCAAoC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AAAA,UACnF;AAAA,YACE,UAAU,KAAK;AAAA,YACf,kBAAkB;AAAA,YAClB,GAAG,KAAK;AAAA,UACV;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AACrG,aAAK,KAAK,MAAM,GAAG,MAAM,gCAAgC;AAAA,UACvD,UAAU,KAAK;AAAA,UACf,kBAAkB;AAAA,UAClB,GAAG,KAAK;AAAA,UACR,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,gBAAQ,UAAU,KAAK,UAAU;AAAA,UAC/B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,gBAAQ,MAAM;AAAA,MAChB;AAEA;AAAA,IACF;AAEA,YAAQ,kBAAkB,IAAI,KAAK,IAAI,GAAG;AAG1C,QAAI,CAAC,MAAM,IAAI,YAAY,GAAG;AAC5B,WAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,IACnD,OAAO;AACL,WAAK,KAAK,MAAM,gCAAgC;AAAA,QAC9C,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,GAAG,KAAK;AAAA,MACV,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBACE,MACA,SACM;AACN,SAAK,gBAAgB,iBAAiB,MAAM,OAAO;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,oBACE,MACA,SACM;AACN,SAAK,gBAAgB,oBAAoB,MAAM,OAAO;AAAA,EACxD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,KACE,IACA,KACoB;AACpB,QAAI,KAAK,UAAU,aAAa;AAC9B,YAAM,MAAM;AACZ,WAAK,KAAK,MAAM,KAAK;AAAA,QACnB,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD,WAAK,cAAc,cAAc,iBAAiB,GAAG;AACrD,aAAO;AAAA,IACT,WAAW,KAAK,UAAU,UAAU;AAClC,WAAK,KAAK,KAAK,6CAA6C;AAAA,QAC1D,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,MACpB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO,KAAK,mBAAmB,EAAE,EAAE,QAAQ,KAAK,GAAG;AAAA,EACrD;AAAA;AAAA,EAGA,gBAAgB,IAAuB,UAAkB;AACvD,WAAO,KAAK,KAAK,IAAI;AAAA,MACnB;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEU,cAAc,MAAyB,SAAiB;AAChE,SAAK,gBAAgB,cAAc,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAAA,EACvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACzE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAU;AACR,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,KAAK,gCAAgC,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EAC5E;AACF;AAEO,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,4BAA4B;AAAA;AAAA;AAAA;AAAA,EAK5B;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,6BAA6B,oBAAI,IAAI;AAC1C,SAAK,cAAc,IAAI,qBAAqB,KAAK,OAAO;AAAA,EAC1D;AAAA,EAEA,gBAAgB,SAAiC;AAC/C,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB,IAA6B;AACtE,QAAI,KAAK,UAAU;AAAQ;AAC3B,QAAI,UAAyC;AAG7C,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK,KAAK;AAAA,UACR,iBAAiB,EAAE;AAAA,UACnB,EAAE,GAAG,KAAK,iBAAiB,UAAU,KAAK,UAAU,aAAa,GAAG;AAAA,QACtE;AACA,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,mBAAmB,CAAC,SAAqB;AAC7C,YAAM,eAAe,KAAK,gCAAgC,MAAM,IAAI;AACpE,UAAI,CAAC,cAAc;AACjB,aAAK,MAAM;AACX;AAAA,MACF,OAAO;AACL,kBAAU;AACV,qBAAa,gBAAgB;AAAA,MAC/B;AAIA,WAAK,mBAAmB,gBAAgB;AACxC,WAAK,gBAAgB,CAACC,UAAS;AAC7B,cAAM,SAAS,KAAK,SAASA,OAAM,IAAI;AACvC,YAAI,CAAC,QAAQ;AACX,eAAK,WAAW,KAAK,UAAU;AAAA,YAC7B,MAAMD,gBAAe;AAAA,YACrB,SAAS;AAAA,UACX,CAAC;AACD,eAAK,MAAM;AACX;AAAA,QACF;AAEA,aAAK,UAAU,QAAQ,IAAI;AAAA,MAC7B,CAAC;AAAA,IACH;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,SAAS;AACX,aAAK,aAAa,MAAM,OAAO;AAAA,MACjC;AACA,WAAK,KAAK,KAAK,iBAAiB,EAAE,iBAAiB;AAAA,QACjD,GAAG,KAAK;AAAA,QACR,GAAG,SAAS;AAAA,QACZ,UAAU,KAAK;AAAA,QACf,aAAa;AAAA,MACf,CAAC;AACD,WAAK,2BAA2B,OAAO,EAAE;AACzC,UAAI,KAAK,2BAA2B;AAClC,aAAK,KAAK,QAAQ,EAAE;AAAA,MACtB;AAAA,IACF,CAAC;AACD,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK;AAAA,QACR,0BAA0B,EAAE,KAAK,kBAAkB,GAAG,CAAC;AAAA,QACvD;AAAA,UACE,GAAG,KAAK;AAAA,UACR,GAAG,SAAS;AAAA,UACZ,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,gCACE,MACA,MAC2B;AAC3B,UAAM,SAAS,KAAK,SAAS,MAAM,IAAI;AACvC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,uCAAuC,OAAO,OAAO,GAAG;AACvE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK,KAAK,mCAAmC;AAAA,QAChD,GAAG,KAAK;AAAA,QACR,UAAU,KAAK;AAAA,QACf,aAAa,OAAO;AAAA,QACpB,kBAAkB;AAAA,QAClB,kBAAkB;AAAA,UAChB,GAAG,MAAM;AAAA,YACP;AAAA,YACA,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,OAAO,QAAQ,OAAO,IAAI;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK,KAAK,KAAK,gCAAgC;AAAA,QAC7C,GAAG,KAAK;AAAA,QACR,UAAU,KAAK;AAAA,QACf,aAAa,OAAO;AAAA,QACpB,kBAAkB;AAAA,MACpB,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd,OAAO,QAAQ,OAAO;AAAA,MACxB;AACA,aAAO;AAAA,IACT;AAEA,SAAK,KAAK,MAAM,kBAAkB,OAAO,IAAI,OAAO;AAAA,MAClD,GAAG,KAAK;AAAA,MACR,UAAU,KAAK;AAAA,MACf,aAAa,OAAO;AAAA,MACpB,kBAAkB;AAAA,IACpB,CAAC;AAED,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ,OAAO;AAAA,IACxB;AAEA,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAKtD,SAAK,YAAY,qBAAqB,OAAO,IAAI;AACjD,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAM,QAAQ,IAAsC;AAClD,UAAM,2BAA2B,MAAM,KAAK,UAAU;AACtD,QAAI,CAAC,yBAAyB,GAAG;AAC/B,WAAK,KAAK;AAAA,QACR,uEAAuE,EAAE;AAAA,QACzE,EAAE,UAAU,KAAK,UAAU,aAAa,GAAG;AAAA,MAC7C;AACA;AAAA,IACF;AAEA,QAAI,mBAAmB,KAAK,2BAA2B,IAAI,EAAE;AAC7D,QAAI,CAAC,kBAAkB;AAErB,YAAM,iBAAiB,KAAK,YAAY,kBAAkB,EAAE;AAC5D,UAAI,CAAC,KAAK,YAAY,UAAU,EAAE,GAAG;AACnC,cAAM,SAAS,uBAAuB,EAAE,yCAAyC,cAAc,yBAAyB,KAAK,YAAY,sBAAsB;AAC/J,aAAK,KAAK,MAAM,QAAQ,EAAE,UAAU,KAAK,UAAU,aAAa,GAAG,CAAC;AACpE,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD;AAAA,MACF;AAEA,UAAI,QAAQ,QAAQ,QAAQ;AAC5B,YAAM,YAAY,KAAK,YAAY,aAAa,EAAE;AAClD,UAAI,YAAY,GAAG;AACjB,gBAAQ,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,SAAS,CAAC;AAAA,MACjE;AAEA,WAAK,KAAK;AAAA,QACR,4BAA4B,EAAE,KAAK,SAAS;AAAA,QAC5C;AAAA,UACE,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf;AAAA,MACF;AACA,WAAK,YAAY,cAAc,EAAE;AACjC,yBAAmB,gBAAO,gBAAgB,WAAW,OAAO,SAAS;AACnE,YAAI;AACF,eAAK,SAAS,WAAW,EAAE,UAAU,CAAC;AACtC,gBAAM;AACN,cAAI,CAAC,yBAAyB,GAAG;AAC/B,kBAAM,IAAI,MAAM,mCAAmC;AAAA,UACrD;AAEA,eAAK,SAAS,YAAY;AAC1B,gBAAM,OAAO,MAAM,KAAK,4BAA4B,EAAE;AACtD,cAAI,CAAC,yBAAyB,GAAG;AAC/B,iBAAK,KAAK;AAAA,cACR,0EAA0E,EAAE;AAAA,cAC5E;AAAA,gBACE,GAAG,KAAK;AAAA,gBACR,UAAU,KAAK;AAAA,gBACf,aAAa;AAAA,cACf;AAAA,YACF;AACA,iBAAK,MAAM;AACX,kBAAM,IAAI,MAAM,mCAAmC;AAAA,UACrD;AAEA,eAAK,SAAS,mBAAmB;AACjC,gBAAM,KAAK,MAAM,KAAK,cAAc,IAAI,IAAI;AAC5C,cAAI,CAAC,IAAI;AACP,iBAAK,MAAM;AACX,kBAAM,IAAI,MAAM,0BAA0B;AAAA,UAC5C;AAEA,iBAAO;AAAA,QACT,SAAS,KAAK;AAGZ,gBAAM,SAAS,kBAAkB,GAAG;AACpC,eAAK,gBAAgB,MAAM;AAC3B,eAAK,UAAU,EAAE,MAAMA,gBAAe,MAAM,CAAC;AAC7C,gBAAM;AAAA,QACR,UAAE;AACA,eAAK,IAAI;AAAA,QACX;AAAA,MACF,CAAC;AAED,WAAK,2BAA2B,IAAI,IAAI,gBAAgB;AAAA,IAC1D,OAAO;AACL,WAAK,KAAK;AAAA,QACR,4BAA4B,EAAE;AAAA,QAC9B;AAAA,UACE,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf;AAAA,MACF;AAAA,IACF;AAEA,QAAI;AACF,YAAM;AAAA,IACR,SAAS,OAAgB;AACvB,WAAK,2BAA2B,OAAO,EAAE;AACzC,YAAM,SAAS,kBAAkB,KAAK;AAEtC,UAAI,CAAC,KAAK,6BAA6B,CAAC,yBAAyB,GAAG;AAClE,aAAK,KAAK,KAAK,iBAAiB,EAAE,YAAY,MAAM,KAAK;AAAA,UACvD,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf,CAAC;AAAA,MACH,OAAO;AACL,aAAK,KAAK,KAAK,iBAAiB,EAAE,YAAY,MAAM,eAAe;AAAA,UACjE,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,QACf,CAAC;AACD,eAAO,KAAK,QAAQ,EAAE;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AAAA,EAEU,cAAc,SAA4B;AAClD,SAAK,2BAA2B,OAAO,QAAQ,EAAE;AACjD,UAAM,cAAc,OAAO;AAAA,EAC7B;AAAA,EAEA,MAAgB,cAAc,IAAuB,MAAgB;AACnE,QAAI,WAAoB;AAExB,QAAI,KAAK,qBAAqB;AAC5B,iBAAW,MAAM,KAAK,oBAAoB,UAAU;AACpD,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,QAAQ,GAAG;AAC3D,aAAK,KAAK,MAAM,uDAAuD;AAAA,UACrE,GAAG,KAAK;AAAA,UACR,UAAU,KAAK;AAAA,UACf,aAAa;AAAA,UACb,kBAAkB;AAAA,YAChB,GAAG,MAAM,OAAO,KAAK,oBAAoB,QAAQ,QAAQ;AAAA,UAC3D;AAAA,UACA,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK;AAAA,UACH,cAAc;AAAA,UACd;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,eAAO;AAAA,MACT;AAAA,IACF;AAIA,UAAM,EAAE,QAAQ,IAAI,KAAK,mBAAmB,EAAE;AAC9C,UAAM,aAAa;AAAA,MACjB,KAAK;AAAA,MACL;AAAA,MACA,QAAQ;AAAA,MACR;AAAA,MACA,sBAAsB,QAAQ,UAAU,GAAG;AAAA,IAC7C;AACA,SAAK,KAAK,MAAM,gCAAgC,EAAE,IAAI;AAAA,MACpD,GAAG,KAAK;AAAA,MACR,UAAU,KAAK;AAAA,MACf,aAAa;AAAA,MACb,kBAAkB;AAAA,IACpB,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,UAAU,CAAC;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ;AACN,SAAK,YAAY,MAAM;AACvB,UAAM,MAAM;AAAA,EACd;AACF;AAEO,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,2BAA2B,oBAAI,QAAQ;AAC5C,SAAK,KAAK,KAAK,8BAA8B;AAAA,MAC3C,UAAU,KAAK;AAAA,MACf,iBAAiB;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,SAAiC;AAC/C,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU;AAAQ;AAE3B,SAAK,KAAK,KAAK,2BAA2B;AAAA,MACxC,GAAG,KAAK;AAAA,MACR,UAAU,KAAK;AAAA,IACjB,CAAC;AAED,QAAI,UAAyC;AAC7C,UAAM,SAAS,MAAM,SAAS,MAAM;AAGpC,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK,KAAK;AAAA,UACR,iBAAiB,OAAO,CAAC;AAAA,UACzB;AAAA,YACE,GAAG,KAAK;AAAA,YACR,UAAU,KAAK;AAAA,YACf,aAAa,OAAO;AAAA,UACtB;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,SAA4B,CAAC;AACnC,QAAI,2BAA2B;AAE/B,UAAM,mBAAmB,CAAC,SAAqB;AAE7C,UAAI,0BAA0B;AAC5B,eAAO,KAAK,IAAI;AAChB;AAAA,MACF;AAEA,iCAA2B;AAC3B,mBAAa,gBAAgB;AAE7B,WAAK,KAAK,+BAA+B,MAAM,IAAI,EAAE;AAAA,QACnD,CAAC,iBAAiB;AAChB,cAAI,CAAC,cAAc;AACjB,iBAAK,MAAM;AACX;AAAA,UACF;AAEA,oBAAU;AAIV,gBAAM,cAAc,CAACC,UAAqB;AACxC,kBAAM,SAAS,KAAK,SAASA,OAAM,IAAI;AACvC,gBAAI,CAAC,QAAQ;AACX,mBAAK,MAAM;AACX;AAAA,YACF;AAEA,iBAAK,UAAU,QAAQ,IAAI;AAAA,UAC7B;AAGA,qBAAWA,SAAQ,QAAQ;AACzB,wBAAYA,KAAI;AAAA,UAClB;AAEA,eAAK,mBAAmB,gBAAgB;AACxC,eAAK,gBAAgB,WAAW;AAChC,iBAAO,SAAS;AAAA,QAClB;AAAA,MACF;AAAA,IACF;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,CAAC;AAAS;AACd,WAAK,KAAK,KAAK,iBAAiB,OAAO,CAAC,iBAAiB;AAAA,QACvD,GAAG,KAAK;AAAA,QACR,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,WAAK,aAAa,MAAM,OAAO;AAAA,IACjC,CAAC;AAED,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMD,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,UAAI,CAAC;AAAS;AACd,WAAK,KAAK;AAAA,QACR,iBAAiB,OAAO,CAAC,kBAAkB,kBAAkB,GAAG,CAAC;AAAA,QACjE,EAAE,GAAG,KAAK,iBAAiB,UAAU,KAAK,SAAS;AAAA,MACrD;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,0BACZ,MACA,SACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,SAAS;AACf,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,aAAK,KAAK,KAAK,8CAA8C,IAAI,IAAI;AAAA,UACnE,GAAG,KAAK;AAAA,UACR,UAAU,KAAK;AAAA,UACf,kBAAkB;AAAA,YAChB,GAAG,MAAM,OAAO,KAAK,oBAAoB,QAAQ,WAAW;AAAA,UAC9D;AAAA,QACF,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAEA,YAAM,yBAAyB,UAC3B,KAAK,yBAAyB,IAAI,OAAO,IACzC;AAEJ,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,cAAM,SAAS;AACf,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAMA,gBAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,aAAK,KAAK,KAAK,2BAA2B,IAAI,IAAI;AAAA,UAChD,GAAG,KAAK;AAAA,UACR,UAAU,KAAK;AAAA,QACjB,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,+BACJ,MACA,MACoC;AACpC,UAAM,SAAS,KAAK,SAAS,MAAM,IAAI;AACvC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,sCAAsC,OAAO,OAAO,GAAG;AACtE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMA,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,YAAM,SAAS;AACf,YAAME,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK,KAAK,KAAK,QAAQ;AAAA,QACrB,GAAG,KAAK;AAAA,QACR,UAAU,KAAK;AAAA;AAAA;AAAA,QAGf,kBAAkB;AAAA,QAClB,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,sCAAsC,OAAO,OAAO;AAAA,QACtE;AAAA,MACF,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAGA,UAAM,aAAa,OAAO,QAAQ;AAClC,QAAI,eAAe,kBAAkB;AACnC,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAMF,gBAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AAED,YAAM,SAAS,2BAA2B,UAAU,WAAW,gBAAgB;AAC/E,YAAME,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK,KAAK;AAAA,QACR,mEAAmE,UAAU,eAAe,gBAAgB;AAAA,QAC5G,EAAE,GAAG,KAAK,iBAAiB,UAAU,KAAK,SAAS;AAAA,MACrD;AACA,WAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,aAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,SAAS,IAAI,OAAO,IAAI;AAChD,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,QAAI,mBAAmB,OAAO;AAC5B,aAAO;AAAA,IACT;AAEA,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,SAAK,yBAAyB,IAAI,SAAS,cAAc;AAEzD,SAAK,KAAK;AAAA,MACR,kBAAkB,OAAO,IAAI;AAAA,MAC7B,KAAK;AAAA,IACP;AACA,UAAM,cAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,MACvE,IAAI;AAAA,MACJ,WAAW,QAAQ;AAAA,IACrB,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAEtD,WAAO;AAAA,EACT;AACF;","names":["SpanStatusCode","data","responseMsg"]}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../router/services.ts","../router/procedures.ts","../node_modules/p-defer/index.js","../node_modules/it-pushable/src/fifo.ts","../node_modules/it-pushable/src/index.ts","../router/client.ts","../router/result.ts","../router/server.ts","../router/handshake.ts"],"sourcesContent":["import { Type, TUnion, TSchema } from '@sinclair/typebox';\nimport { RiverError, RiverUncaughtSchema } from './result';\nimport {\n Branded,\n ProcedureMap,\n Unbranded,\n AnyProcedure,\n PayloadType,\n} from './procedures';\n\n/**\n * An instantiated service, probably from a {@link ServiceSchema}.\n *\n * You shouldn't construct these directly, use {@link ServiceSchema} instead.\n */\nexport interface Service<\n State extends object,\n Procs extends ProcedureMap<State>,\n> {\n readonly state: State;\n readonly procedures: Procs;\n}\n\n/**\n * Represents any {@link Service} object.\n */\nexport type AnyService = Service<object, ProcedureMap>;\n\n/**\n * Represents any {@link ServiceSchema} object.\n */\nexport type AnyServiceSchema = ServiceSchema<object, ProcedureMap>;\n\n/**\n * A dictionary of {@link ServiceSchema}s, where the key is the service name.\n */\nexport type AnyServiceSchemaMap = Record<string, AnyServiceSchema>;\n\n// This has the secret sauce to keep go to definition working, the structure is\n// somewhat delicate, so be careful when modifying it. Would be nice to add a\n// static test for this.\n/**\n * Takes a {@link AnyServiceSchemaMap} and returns a dictionary of instantiated\n * services.\n */\nexport type InstantiatedServiceSchemaMap<T extends AnyServiceSchemaMap> = {\n [K in keyof T]: T[K] extends ServiceSchema<infer S, infer P>\n ? Service<S, P>\n : never;\n};\n\n/**\n * Helper to get the type definition for a specific handler of a procedure in a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcHandler<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName]['handler'];\n\n/**\n * Helper to get whether the type definition for the procedure contains an init type.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcHasInit<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName] extends { init: PayloadType } ? true : false;\n\n/**\n * Helper to get the type definition for the procedure init type of a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcInit<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName] extends { init: PayloadType }\n ? S['procedures'][ProcName]['init']\n : never;\n\n/**\n * Helper to get the type definition for the procedure input of a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcInput<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName]['input'];\n\n/**\n * Helper to get the type definition for the procedure output of a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcOutput<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName]['output'];\n\n/**\n * Helper to get the type definition for the procedure errors of a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcErrors<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = TUnion<[S['procedures'][ProcName]['errors'], typeof RiverUncaughtSchema]>;\n\n/**\n * Helper to get the type of procedure in a service.\n * @template S - The service.\n * @template ProcName - The name of the procedure.\n */\nexport type ProcType<\n S extends AnyService,\n ProcName extends keyof S['procedures'],\n> = S['procedures'][ProcName]['type'];\n\n/**\n * A list of procedures where every procedure is \"branded\", as-in the procedure\n * was created via the {@link Procedure} constructors.\n */\ntype BrandedProcedureMap<State> = Record<string, Branded<AnyProcedure<State>>>;\n\n/**\n * The configuration for a service.\n */\nexport interface ServiceConfiguration<State extends object> {\n /**\n * A factory function for creating a fresh state.\n */\n initializeState: () => State;\n}\n\nexport interface SerializedServiceSchema {\n procedures: Record<\n string,\n {\n input: PayloadType;\n output: PayloadType;\n errors?: RiverError;\n type: 'rpc' | 'subscription' | 'upload' | 'stream';\n init?: PayloadType;\n }\n >;\n}\n\nexport interface SerializedServerSchema {\n handshakeSchema?: TSchema;\n services: Record<string, SerializedServiceSchema>;\n}\n\nexport function serializeSchema(\n services: AnyServiceSchemaMap,\n handshakeSchema?: TSchema,\n): SerializedServerSchema {\n const serializedServiceObject = Object.entries(services).reduce<\n Record<string, SerializedServiceSchema>\n >((acc, [name, value]) => {\n acc[name] = value.serialize();\n return acc;\n }, {});\n\n const schema: SerializedServerSchema = {\n services: serializedServiceObject,\n };\n\n if (handshakeSchema) {\n schema.handshakeSchema = Type.Strict(handshakeSchema);\n }\n\n return schema;\n}\n\n/**\n * The schema for a {@link Service}. This is used to define a service, specifically\n * its initial state and procedures.\n *\n * There are two ways to define a service:\n * 1. the {@link ServiceSchema.define} static method, which takes a configuration and\n * a list of procedures directly. Use this to ergonomically define a service schema\n * in one go. Good for smaller services, especially if they're stateless.\n * 2. the {@link ServiceSchema.scaffold} static method, which creates a scaffold that\n * can be used to define procedures separately from the configuration. Use this to\n * better organize your service's definition, especially if it's a large service.\n * You can also use it in a builder pattern to define the service in a more\n * fluent way.\n *\n * See the static methods for more information and examples.\n *\n * When defining procedures, use the {@link Procedure} constructors to create them.\n */\nexport class ServiceSchema<\n State extends object,\n Procedures extends ProcedureMap<State>,\n> {\n /**\n * Factory function for creating a fresh state.\n */\n protected readonly initializeState: () => State;\n\n /**\n * The procedures for this service.\n */\n readonly procedures: Procedures;\n\n /**\n * @param config - The configuration for this service.\n * @param procedures - The procedures for this service.\n */\n protected constructor(\n config: ServiceConfiguration<State>,\n procedures: Procedures,\n ) {\n this.initializeState = config.initializeState;\n this.procedures = procedures;\n }\n\n /**\n * Creates a {@link ServiceScaffold}, which can be used to define procedures\n * that can then be merged into a {@link ServiceSchema}, via the scaffold's\n * `finalize` method.\n *\n * There are two patterns that work well with this method. The first is using\n * it to separate the definition of procedures from the definition of the\n * service's configuration:\n * ```ts\n * const MyServiceScaffold = ServiceSchema.scaffold({\n * initializeState: () => ({ count: 0 }),\n * });\n *\n * const incrementProcedures = MyServiceScaffold.procedures({\n * increment: Procedure.rpc({\n * input: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, input) {\n * ctx.state.count += input.amount;\n * return Ok({ current: ctx.state.count });\n * }\n * }),\n * })\n *\n * const MyService = MyServiceScaffold.finalize({\n * ...incrementProcedures,\n * // you can also directly define procedures here\n * });\n * ```\n * This might be really handy if you have a very large service and you're\n * wanting to split it over multiple files. You can define the scaffold\n * in one file, and then import that scaffold in other files where you\n * define procedures - and then finally import the scaffolds and your\n * procedure objects in a final file where you finalize the scaffold into\n * a service schema.\n *\n * The other way is to use it like in a builder pattern:\n * ```ts\n * const MyService = ServiceSchema\n * .scaffold({ initializeState: () => ({ count: 0 }) })\n * .finalize({\n * increment: Procedure.rpc({\n * input: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, input) {\n * ctx.state.count += input.amount;\n * return Ok({ current: ctx.state.count });\n * }\n * }),\n * })\n * ```\n * Depending on your preferences, this may be a more appealing way to define\n * a schema versus using the {@link ServiceSchema.define} method.\n */\n static scaffold<State extends object>(config: ServiceConfiguration<State>) {\n return new ServiceScaffold(config);\n }\n\n /**\n * Creates a new {@link ServiceSchema} with the given configuration and procedures.\n *\n * All procedures must be created with the {@link Procedure} constructors.\n *\n * NOTE: There is an overload that lets you just provide the procedures alone if your\n * service has no state.\n *\n * @param config - The configuration for this service.\n * @param procedures - The procedures for this service.\n *\n * @example\n * ```\n * const service = ServiceSchema.define(\n * { initializeState: () => ({ count: 0 }) },\n * {\n * increment: Procedure.rpc({\n * input: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, input) {\n * ctx.state.count += input.amount;\n * return Ok({ current: ctx.state.count });\n * }\n * }),\n * },\n * );\n * ```\n */\n static define<\n State extends object,\n Procedures extends BrandedProcedureMap<State>,\n >(\n config: ServiceConfiguration<State>,\n procedures: Procedures,\n ): ServiceSchema<\n State,\n { [K in keyof Procedures]: Unbranded<Procedures[K]> }\n >;\n /**\n * Creates a new {@link ServiceSchema} with the given procedures.\n *\n * All procedures must be created with the {@link Procedure} constructors.\n *\n * NOTE: There is an overload that lets you provide configuration as well,\n * if your service has extra configuration like a state.\n *\n * @param procedures - The procedures for this service.\n *\n * @example\n * ```\n * const service = ServiceSchema.define({\n * add: Procedure.rpc({\n * input: Type.Object({ a: Type.Number(), b: Type.Number() }),\n * output: Type.Object({ result: Type.Number() }),\n * async handler(ctx, input) {\n * return Ok({ result: input.a + input.b });\n * }\n * }),\n * });\n */\n static define<Procedures extends BrandedProcedureMap<Record<string, never>>>(\n procedures: Procedures,\n ): ServiceSchema<\n Record<string, never>,\n { [K in keyof Procedures]: Unbranded<Procedures[K]> }\n >;\n // actual implementation\n static define(\n configOrProcedures:\n | ServiceConfiguration<object>\n | BrandedProcedureMap<object>,\n maybeProcedures?: BrandedProcedureMap<object>,\n ): ServiceSchema<object, ProcedureMap> {\n let config: ServiceConfiguration<object>;\n let procedures: BrandedProcedureMap<object>;\n\n if (\n 'initializeState' in configOrProcedures &&\n typeof configOrProcedures.initializeState === 'function'\n ) {\n if (!maybeProcedures) {\n throw new Error('Expected procedures to be defined');\n }\n\n config = configOrProcedures as ServiceConfiguration<object>;\n procedures = maybeProcedures;\n } else {\n config = { initializeState: () => ({}) };\n procedures = configOrProcedures as BrandedProcedureMap<object>;\n }\n\n return new ServiceSchema(config, procedures);\n }\n\n /**\n * Serializes this schema's procedures into a plain object that is JSON compatible.\n */\n serialize(): SerializedServiceSchema {\n return {\n procedures: Object.fromEntries(\n Object.entries(this.procedures).map(([procName, procDef]) => [\n procName,\n {\n input: Type.Strict(procDef.input),\n output: Type.Strict(procDef.output),\n // Only add `description` field if the type declares it.\n ...('description' in procDef\n ? { description: procDef.description }\n : {}),\n // Only add the `errors` field if the type declares it.\n ...('errors' in procDef\n ? {\n errors: Type.Strict(procDef.errors),\n }\n : {}),\n type: procDef.type,\n // Only add the `init` field if the type declares it.\n ...('init' in procDef\n ? {\n init: Type.Strict(procDef.init),\n }\n : {}),\n },\n ]),\n ),\n };\n }\n\n /**\n * Instantiates this schema into a {@link Service} object.\n *\n * You probably don't need this, usually the River server will handle this\n * for you.\n */\n instantiate(): Service<State, Procedures> {\n return Object.freeze({\n state: this.initializeState(),\n procedures: this.procedures,\n });\n }\n}\n\n/**\n * A scaffold for defining a service's procedures.\n *\n * @see {@link ServiceSchema.scaffold}\n */\n// note that this isn't exported\nclass ServiceScaffold<State extends object> {\n /**\n * The configuration for this service.\n */\n protected readonly config: ServiceConfiguration<State>;\n\n /**\n * @param config - The configuration for this service.\n */\n constructor(config: ServiceConfiguration<State>) {\n this.config = config;\n }\n\n /**\n * Define procedures for this service. Use the {@link Procedure} constructors\n * to create them. This returns the procedures object, which can then be\n * passed to {@link ServiceSchema.finalize} to create a {@link ServiceSchema}.\n *\n * @example\n * ```\n * const myProcedures = MyServiceScaffold.procedures({\n * myRPC: Procedure.rpc({\n * // ...\n * }),\n * });\n *\n * const MyService = MyServiceScaffold.finalize({\n * ...myProcedures,\n * });\n * ```\n *\n * @param procedures - The procedures for this service.\n */\n procedures<T extends BrandedProcedureMap<State>>(procedures: T): T {\n return procedures;\n }\n\n /**\n * Finalizes the scaffold into a {@link ServiceSchema}. This is where you\n * provide the service's procedures and get a {@link ServiceSchema} in return.\n *\n * You can directly define procedures here, or you can define them separately\n * with the {@link ServiceScaffold.procedures} method, and then pass them here.\n *\n * @example\n * ```\n * const MyService = MyServiceScaffold.finalize({\n * myRPC: Procedure.rpc({\n * // ...\n * }),\n * // e.g. from the procedures method\n * ...myOtherProcedures,\n * });\n * ```\n */\n finalize<T extends BrandedProcedureMap<State>>(\n procedures: T,\n ): ServiceSchema<State, { [K in keyof T]: Unbranded<T[K]> }> {\n return ServiceSchema.define(this.config, procedures);\n }\n}\n","import { Static, TNever, Type, TSchema } from '@sinclair/typebox';\nimport type { Pushable } from 'it-pushable';\nimport { ServiceContextWithTransportInfo } from './context';\nimport { Result, RiverError, RiverUncaughtSchema } from './result';\n\n/**\n * Brands a type to prevent it from being directly constructed.\n */\nexport type Branded<T> = T & { readonly __BRAND_DO_NOT_USE: unique symbol };\n\n/**\n * Unbrands a {@link Branded} type.\n */\nexport type Unbranded<T> = T extends Branded<infer U> ? U : never;\n\n/**\n * The valid {@link Procedure} types. The `stream` and `upload` types can optionally have a\n * different type for the very first initialization message. The suffixless types correspond to\n * gRPC's four combinations of stream / non-stream in each direction.\n */\nexport type ValidProcType =\n // Single message in both directions (1:1).\n | 'rpc'\n // Client-stream (potentially preceded by an initialization message), single message from server (n:1).\n | 'upload'\n // Single message from client, stream from server (1:n).\n | 'subscription'\n // Bidirectional stream (potentially preceded by an initialization message) (n:n).\n | 'stream';\n\n/**\n * Represents the payload type for {@link Procedure}s.\n */\nexport type PayloadType = TSchema;\n\n/**\n * Represents results from a {@link Procedure}. Might come from inside a stream or\n * from a single message.\n */\nexport type ProcedureResult<\n O extends PayloadType,\n E extends RiverError,\n> = Result<Static<O>, Static<E> | Static<typeof RiverUncaughtSchema>>;\n\n/**\n * Procedure for a single message in both directions (1:1).\n *\n * @template State - The context state object.\n * @template I - The TypeBox schema of the input object.\n * @template O - The TypeBox schema of the output object.\n * @template E - The TypeBox schema of the error object.\n */\nexport interface RpcProcedure<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n> {\n type: 'rpc';\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n input: Static<I>,\n ): Promise<ProcedureResult<O, E>>;\n}\n\n/**\n * Procedure for a client-stream (potentially preceded by an initialization message),\n * single message from server (n:1).\n *\n * @template State - The context state object.\n * @template I - The TypeBox schema of the input object.\n * @template O - The TypeBox schema of the output object.\n * @template E - The TypeBox schema of the error object.\n * @template Init - The TypeBox schema of the input initialization object, if any.\n */\nexport type UploadProcedure<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n Init extends PayloadType | null = null,\n> = Init extends PayloadType\n ? {\n type: 'upload';\n init: Init;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n init: Static<Init>,\n input: AsyncIterableIterator<Static<I>>,\n ): Promise<ProcedureResult<O, E>>;\n }\n : {\n type: 'upload';\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n input: AsyncIterableIterator<Static<I>>,\n ): Promise<ProcedureResult<O, E>>;\n };\n\n/**\n * Procedure for a single message from client, stream from server (1:n).\n *\n * @template State - The context state object.\n * @template I - The TypeBox schema of the input object.\n * @template O - The TypeBox schema of the output object.\n * @template E - The TypeBox schema of the error object.\n */\nexport interface SubscriptionProcedure<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n> {\n type: 'subscription';\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n input: Static<I>,\n output: Pushable<ProcedureResult<O, E>>,\n ): Promise<(() => void) | void>;\n}\n\n/**\n * Procedure for a bidirectional stream (potentially preceded by an initialization message),\n * (n:n).\n *\n * @template State - The context state object.\n * @template I - The TypeBox schema of the input object.\n * @template O - The TypeBox schema of the output object.\n * @template E - The TypeBox schema of the error object.\n * @template Init - The TypeBox schema of the input initialization object, if any.\n */\nexport type StreamProcedure<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n Init extends PayloadType | null = null,\n> = Init extends PayloadType\n ? {\n type: 'stream';\n init: Init;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n init: Static<Init>,\n input: AsyncIterableIterator<Static<I>>,\n output: Pushable<ProcedureResult<O, E>>,\n ): Promise<(() => void) | void>;\n }\n : {\n type: 'stream';\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler(\n context: ServiceContextWithTransportInfo<State>,\n input: AsyncIterableIterator<Static<I>>,\n output: Pushable<ProcedureResult<O, E>>,\n ): Promise<(() => void) | void>;\n };\n\n/**\n * Defines a Procedure type that can be a:\n * - {@link RpcProcedure} for a single message in both directions (1:1)\n * - {@link UploadProcedure} for a client-stream (potentially preceded by an\n * initialization message)\n * - {@link SubscriptionProcedure} for a single message from client, stream from server (1:n)\n * - {@link StreamProcedure} for a bidirectional stream (potentially preceded by an\n * initialization message)\n *\n * @template State - The TypeBox schema of the state object.\n * @template Ty - The type of the procedure.\n * @template I - The TypeBox schema of the input object.\n * @template O - The TypeBox schema of the output object.\n * @template Init - The TypeBox schema of the input initialization object, if any.\n */\n// prettier-ignore\nexport type Procedure<\n State,\n Ty extends ValidProcType,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n Init extends PayloadType | null = null,\n> = { type: Ty } & (\n Init extends PayloadType\n ? Ty extends 'upload' ? UploadProcedure<State, I, O, E, Init>\n : Ty extends 'stream' ? StreamProcedure<State, I, O, E, Init>\n : never\n : Ty extends 'rpc' ? RpcProcedure<State, I, O, E>\n : Ty extends 'upload' ? UploadProcedure<State, I, O, E>\n : Ty extends 'subscription' ? SubscriptionProcedure<State, I, O, E>\n : Ty extends 'stream' ? StreamProcedure<State, I, O, E>\n : never\n);\n\n/**\n * Represents any {@link Procedure} type.\n *\n * @template State - The context state object. You can provide this to constrain\n * the type of procedures.\n */\nexport type AnyProcedure<State = object> = Procedure<\n State,\n ValidProcType,\n PayloadType,\n PayloadType,\n RiverError,\n PayloadType | null\n>;\n\n/**\n * Represents a map of {@link Procedure}s.\n *\n * @template State - The context state object. You can provide this to constrain\n * the type of procedures.\n */\nexport type ProcedureMap<State = object> = Record<string, AnyProcedure<State>>;\n\n// typescript is funky so with these upcoming procedure constructors, the overloads\n// which handle the `init` case _must_ come first, otherwise the `init` property\n// is not recognized as optional, for some reason\n\n/**\n * Creates an {@link RpcProcedure}.\n */\n// signature: default errors\nfunction rpc<State, I extends PayloadType, O extends PayloadType>(def: {\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: RpcProcedure<State, I, O, TNever>['handler'];\n}): Branded<RpcProcedure<State, I, O, TNever>>;\n\n// signature: explicit errors\nfunction rpc<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n>(def: {\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: RpcProcedure<State, I, O, E>['handler'];\n}): Branded<RpcProcedure<State, I, O, E>>;\n\n// implementation\nfunction rpc({\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n input: PayloadType;\n output: PayloadType;\n errors?: RiverError;\n description?: string;\n handler: RpcProcedure<\n object,\n PayloadType,\n PayloadType,\n RiverError\n >['handler'];\n}) {\n return {\n ...(description ? { description } : {}),\n type: 'rpc',\n input,\n output,\n errors,\n handler,\n };\n}\n\n/**\n * Creates an {@link UploadProcedure}, optionally with an initialization message.\n */\n// signature: init with default errors\nfunction upload<\n State,\n I extends PayloadType,\n O extends PayloadType,\n Init extends PayloadType,\n>(def: {\n init: Init;\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: UploadProcedure<State, I, O, TNever, Init>['handler'];\n}): Branded<UploadProcedure<State, I, O, TNever, Init>>;\n\n// signature: init with explicit errors\nfunction upload<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n Init extends PayloadType,\n>(def: {\n init: Init;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: UploadProcedure<State, I, O, E, Init>['handler'];\n}): Branded<UploadProcedure<State, I, O, E, Init>>;\n\n// signature: no init with default errors\nfunction upload<State, I extends PayloadType, O extends PayloadType>(def: {\n init?: never;\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: UploadProcedure<State, I, O, TNever>['handler'];\n}): Branded<UploadProcedure<State, I, O, TNever>>;\n\n// signature: no init with explicit errors\nfunction upload<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n>(def: {\n init?: never;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: UploadProcedure<State, I, O, E>['handler'];\n}): Branded<UploadProcedure<State, I, O, E>>;\n\n// implementation\nfunction upload({\n init,\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init?: PayloadType | null;\n input: PayloadType;\n output: PayloadType;\n errors?: RiverError;\n description?: string;\n handler: UploadProcedure<\n object,\n PayloadType,\n PayloadType,\n RiverError,\n PayloadType | null\n >['handler'];\n}) {\n return init !== undefined && init !== null\n ? {\n type: 'upload',\n ...(description ? { description } : {}),\n init,\n input,\n output,\n errors,\n handler,\n }\n : {\n type: 'upload',\n ...(description ? { description } : {}),\n input,\n output,\n errors,\n handler,\n };\n}\n\n/**\n * Creates a {@link SubscriptionProcedure}.\n */\n// signature: default errors\nfunction subscription<\n State,\n I extends PayloadType,\n O extends PayloadType,\n>(def: {\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: SubscriptionProcedure<State, I, O, TNever>['handler'];\n}): Branded<SubscriptionProcedure<State, I, O, TNever>>;\n\n// signature: explicit errors\nfunction subscription<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n>(def: {\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: SubscriptionProcedure<State, I, O, E>['handler'];\n}): Branded<SubscriptionProcedure<State, I, O, E>>;\n\n// implementation\nfunction subscription({\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n input: PayloadType;\n output: PayloadType;\n errors?: RiverError;\n description?: string;\n handler: SubscriptionProcedure<\n object,\n PayloadType,\n PayloadType,\n RiverError\n >['handler'];\n}) {\n return {\n type: 'subscription',\n ...(description ? { description } : {}),\n input,\n output,\n errors,\n handler,\n };\n}\n\n/**\n * Creates a {@link StreamProcedure}, optionally with an initialization message.\n */\n// signature: init with default errors\nfunction stream<\n State,\n I extends PayloadType,\n O extends PayloadType,\n Init extends PayloadType,\n>(def: {\n init: Init;\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: StreamProcedure<State, I, O, TNever, Init>['handler'];\n}): Branded<StreamProcedure<State, I, O, TNever, Init>>;\n\n// signature: init with explicit errors\nfunction stream<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n Init extends PayloadType,\n>(def: {\n init: Init;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: StreamProcedure<State, I, O, E, Init>['handler'];\n}): Branded<StreamProcedure<State, I, O, E, Init>>;\n\n// signature: no init with default errors\nfunction stream<State, I extends PayloadType, O extends PayloadType>(def: {\n init?: never;\n input: I;\n output: O;\n errors?: never;\n description?: string;\n handler: StreamProcedure<State, I, O, TNever>['handler'];\n}): Branded<StreamProcedure<State, I, O, TNever>>;\n\n// signature: no init with explicit errors\nfunction stream<\n State,\n I extends PayloadType,\n O extends PayloadType,\n E extends RiverError,\n>(def: {\n init?: never;\n input: I;\n output: O;\n errors: E;\n description?: string;\n handler: StreamProcedure<State, I, O, E>['handler'];\n}): Branded<StreamProcedure<State, I, O, E>>;\n\n// implementation\nfunction stream({\n init,\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init?: PayloadType | null;\n input: PayloadType;\n output: PayloadType;\n errors?: RiverError;\n description?: string;\n handler: StreamProcedure<\n object,\n PayloadType,\n PayloadType,\n RiverError,\n PayloadType | null\n >['handler'];\n}) {\n return init !== undefined && init !== null\n ? {\n type: 'stream',\n ...(description ? { description } : {}),\n init,\n input,\n output,\n errors,\n handler,\n }\n : {\n type: 'stream',\n ...(description ? { description } : {}),\n input,\n output,\n errors,\n handler,\n };\n}\n\n/**\n * Holds the {@link Procedure} creation functions. Use these to create\n * procedures for services. You aren't allowed to create procedures directly.\n */\nexport const Procedure = {\n rpc,\n upload,\n subscription,\n stream,\n};\n","export default function pDefer() {\n\tconst deferred = {};\n\n\tdeferred.promise = new Promise((resolve, reject) => {\n\t\tdeferred.resolve = resolve;\n\t\tdeferred.reject = reject;\n\t});\n\n\treturn deferred;\n}\n","// ported from https://www.npmjs.com/package/fast-fifo\n\nexport interface Next<T> {\n done?: boolean\n error?: Error\n value?: T\n}\n\nclass FixedFIFO<T> {\n public buffer: Array<Next<T> | undefined>\n private readonly mask: number\n private top: number\n private btm: number\n public next: FixedFIFO<T> | null\n\n constructor (hwm: number) {\n if (!(hwm > 0) || ((hwm - 1) & hwm) !== 0) {\n throw new Error('Max size for a FixedFIFO should be a power of two')\n }\n\n this.buffer = new Array(hwm)\n this.mask = hwm - 1\n this.top = 0\n this.btm = 0\n this.next = null\n }\n\n push (data: Next<T>): boolean {\n if (this.buffer[this.top] !== undefined) {\n return false\n }\n\n this.buffer[this.top] = data\n this.top = (this.top + 1) & this.mask\n\n return true\n }\n\n shift (): Next<T> | undefined {\n const last = this.buffer[this.btm]\n\n if (last === undefined) {\n return undefined\n }\n\n this.buffer[this.btm] = undefined\n this.btm = (this.btm + 1) & this.mask\n return last\n }\n\n isEmpty (): boolean {\n return this.buffer[this.btm] === undefined\n }\n}\n\nexport interface FIFOOptions {\n /**\n * When the queue reaches this size, it will be split into head/tail parts\n */\n splitLimit?: number\n}\n\nexport class FIFO<T> {\n public size: number\n private readonly hwm: number\n private head: FixedFIFO<T>\n private tail: FixedFIFO<T>\n\n constructor (options: FIFOOptions = {}) {\n this.hwm = options.splitLimit ?? 16\n this.head = new FixedFIFO<T>(this.hwm)\n this.tail = this.head\n this.size = 0\n }\n\n calculateSize (obj: any): number {\n if (obj?.byteLength != null) {\n return obj.byteLength\n }\n\n return 1\n }\n\n push (val: Next<T>): void {\n if (val?.value != null) {\n this.size += this.calculateSize(val.value)\n }\n\n if (!this.head.push(val)) {\n const prev = this.head\n this.head = prev.next = new FixedFIFO<T>(2 * this.head.buffer.length)\n this.head.push(val)\n }\n }\n\n shift (): Next<T> | undefined {\n let val = this.tail.shift()\n\n if (val === undefined && (this.tail.next != null)) {\n const next = this.tail.next\n this.tail.next = null\n this.tail = next\n val = this.tail.shift()\n }\n\n if (val?.value != null) {\n this.size -= this.calculateSize(val.value)\n }\n\n return val\n }\n\n isEmpty (): boolean {\n return this.head.isEmpty()\n }\n}\n","/**\n * @packageDocumentation\n *\n * An iterable that you can push values into.\n *\n * @example\n *\n * ```js\n * import { pushable } from 'it-pushable'\n *\n * const source = pushable()\n *\n * setTimeout(() => source.push('hello'), 100)\n * setTimeout(() => source.push('world'), 200)\n * setTimeout(() => source.end(), 300)\n *\n * const start = Date.now()\n *\n * for await (const value of source) {\n * console.log(`got \"${value}\" after ${Date.now() - start}ms`)\n * }\n * console.log(`done after ${Date.now() - start}ms`)\n *\n * // Output:\n * // got \"hello\" after 105ms\n * // got \"world\" after 207ms\n * // done after 309ms\n * ```\n *\n * @example\n *\n * ```js\n * import { pushableV } from 'it-pushable'\n * import all from 'it-all'\n *\n * const source = pushableV()\n *\n * source.push(1)\n * source.push(2)\n * source.push(3)\n * source.end()\n *\n * console.info(await all(source))\n *\n * // Output:\n * // [ [1, 2, 3] ]\n * ```\n */\n\nimport deferred from 'p-defer'\nimport { FIFO, type Next } from './fifo.js'\n\nexport class AbortError extends Error {\n type: string\n code: string\n\n constructor (message?: string, code?: string) {\n super(message ?? 'The operation was aborted')\n this.type = 'aborted'\n this.code = code ?? 'ABORT_ERR'\n }\n}\n\nexport interface AbortOptions {\n signal?: AbortSignal\n}\n\ninterface BasePushable<T> {\n /**\n * End the iterable after all values in the buffer (if any) have been yielded. If an\n * error is passed the buffer is cleared immediately and the next iteration will\n * throw the passed error\n */\n end(err?: Error): this\n\n /**\n * Push a value into the iterable. Values are yielded from the iterable in the order\n * they are pushed. Values not yet consumed from the iterable are buffered.\n */\n push(value: T): this\n\n /**\n * Returns a promise that resolves when the underlying queue becomes empty (e.g.\n * this.readableLength === 0).\n *\n * If an AbortSignal is passed as an option and that signal aborts, it only\n * causes the returned promise to reject - it does not end the pushable.\n */\n onEmpty(options?: AbortOptions): Promise<void>\n\n /**\n * This property contains the number of bytes (or objects) in the queue ready to be read.\n *\n * If `objectMode` is true, this is the number of objects in the queue, if false it's the\n * total number of bytes in the queue.\n */\n readableLength: number\n}\n\n/**\n * An iterable that you can push values into.\n */\nexport interface Pushable<T, R = void, N = unknown> extends AsyncGenerator<T, R, N>, BasePushable<T> {}\n\n/**\n * Similar to `pushable`, except it yields multiple buffered chunks at a time. All values yielded from the iterable will be arrays.\n */\nexport interface PushableV<T, R = void, N = unknown> extends AsyncGenerator<T[], R, N>, BasePushable<T> {}\n\nexport interface Options {\n /**\n * A boolean value that means non-`Uint8Array`s will be passed to `.push`, default: `false`\n */\n objectMode?: boolean\n\n /**\n * A function called after *all* values have been yielded from the iterator (including\n * buffered values). In the case when the iterator is ended with an error it will be\n * passed the error as a parameter.\n */\n onEnd?(err?: Error): void\n}\n\nexport interface DoneResult { done: true }\nexport interface ValueResult<T> { done: false, value: T }\nexport type NextResult<T> = ValueResult<T> | DoneResult\n\ninterface getNext<T, V = T> { (buffer: FIFO<T>): NextResult<V> }\n\nexport interface ObjectPushableOptions extends Options {\n objectMode: true\n}\n\nexport interface BytePushableOptions extends Options {\n objectMode?: false\n}\n\n/**\n * Create a new async iterable. The values yielded from calls to `.next()`\n * or when used in a `for await of`loop are \"pushed\" into the iterable.\n * Returns an async iterable object with additional methods.\n */\nexport function pushable<T extends { byteLength: number } = Uint8Array> (options?: BytePushableOptions): Pushable<T>\nexport function pushable<T> (options: ObjectPushableOptions): Pushable<T>\nexport function pushable<T> (options: Options = {}): Pushable<T> {\n const getNext = (buffer: FIFO<T>): NextResult<T> => {\n const next: Next<T> | undefined = buffer.shift()\n\n if (next == null) {\n return { done: true }\n }\n\n if (next.error != null) {\n throw next.error\n }\n\n return {\n done: next.done === true,\n // @ts-expect-error if done is false, value will be present\n value: next.value\n }\n }\n\n return _pushable<T, T, Pushable<T>>(getNext, options)\n}\n\nexport function pushableV<T extends { byteLength: number } = Uint8Array> (options?: BytePushableOptions): PushableV<T>\nexport function pushableV<T> (options: ObjectPushableOptions): PushableV<T>\nexport function pushableV<T> (options: Options = {}): PushableV<T> {\n const getNext = (buffer: FIFO<T>): NextResult<T[]> => {\n let next: Next<T> | undefined\n const values: T[] = []\n\n while (!buffer.isEmpty()) {\n next = buffer.shift()\n\n if (next == null) {\n break\n }\n\n if (next.error != null) {\n throw next.error\n }\n\n if (next.done === false) {\n // @ts-expect-error if done is false value should be pushed\n values.push(next.value)\n }\n }\n\n if (next == null) {\n return { done: true }\n }\n\n return {\n done: next.done === true,\n value: values\n }\n }\n\n return _pushable<T, T[], PushableV<T>>(getNext, options)\n}\n\nfunction _pushable<PushType, ValueType, ReturnType> (getNext: getNext<PushType, ValueType>, options?: Options): ReturnType {\n options = options ?? {}\n let onEnd = options.onEnd\n let buffer = new FIFO<PushType>()\n let pushable: any\n let onNext: ((next: Next<PushType>) => ReturnType) | null\n let ended: boolean\n let drain = deferred()\n\n const waitNext = async (): Promise<NextResult<ValueType>> => {\n try {\n if (!buffer.isEmpty()) {\n return getNext(buffer)\n }\n\n if (ended) {\n return { done: true }\n }\n\n return await new Promise<NextResult<ValueType>>((resolve, reject) => {\n onNext = (next: Next<PushType>) => {\n onNext = null\n buffer.push(next)\n\n try {\n resolve(getNext(buffer))\n } catch (err) {\n reject(err)\n }\n\n return pushable\n }\n })\n } finally {\n if (buffer.isEmpty()) {\n // settle promise in the microtask queue to give consumers a chance to\n // await after calling .push\n queueMicrotask(() => {\n drain.resolve()\n drain = deferred()\n })\n }\n }\n }\n\n const bufferNext = (next: Next<PushType>): ReturnType => {\n if (onNext != null) {\n return onNext(next)\n }\n\n buffer.push(next)\n return pushable\n }\n\n const bufferError = (err: Error): ReturnType => {\n buffer = new FIFO()\n\n if (onNext != null) {\n return onNext({ error: err })\n }\n\n buffer.push({ error: err })\n return pushable\n }\n\n const push = (value: PushType): ReturnType => {\n if (ended) {\n return pushable\n }\n\n // @ts-expect-error `byteLength` is not declared on PushType\n if (options?.objectMode !== true && value?.byteLength == null) {\n throw new Error('objectMode was not true but tried to push non-Uint8Array value')\n }\n\n return bufferNext({ done: false, value })\n }\n const end = (err?: Error): ReturnType => {\n if (ended) return pushable\n ended = true\n\n return (err != null) ? bufferError(err) : bufferNext({ done: true })\n }\n const _return = (): DoneResult => {\n buffer = new FIFO()\n end()\n\n return { done: true }\n }\n const _throw = (err: Error): DoneResult => {\n end(err)\n\n return { done: true }\n }\n\n pushable = {\n [Symbol.asyncIterator] () { return this },\n next: waitNext,\n return: _return,\n throw: _throw,\n push,\n end,\n get readableLength (): number {\n return buffer.size\n },\n onEmpty: async (options?: AbortOptions) => {\n const signal = options?.signal\n signal?.throwIfAborted()\n\n if (buffer.isEmpty()) {\n return\n }\n\n let cancel: Promise<void> | undefined\n let listener: (() => void) | undefined\n\n if (signal != null) {\n cancel = new Promise((resolve, reject) => {\n listener = () => {\n reject(new AbortError())\n }\n\n signal.addEventListener('abort', listener)\n })\n }\n\n try {\n await Promise.race([\n drain.promise,\n cancel\n ])\n } finally {\n if (listener != null && signal != null) {\n signal?.removeEventListener('abort', listener)\n }\n }\n }\n }\n\n if (onEnd == null) {\n return pushable\n }\n\n const _pushable = pushable\n\n pushable = {\n [Symbol.asyncIterator] () { return this },\n next () {\n return _pushable.next()\n },\n throw (err: Error) {\n _pushable.throw(err)\n\n if (onEnd != null) {\n onEnd(err)\n onEnd = undefined\n }\n\n return { done: true }\n },\n return () {\n _pushable.return()\n\n if (onEnd != null) {\n onEnd()\n onEnd = undefined\n }\n\n return { done: true }\n },\n push,\n end (err: Error) {\n _pushable.end(err)\n\n if (onEnd != null) {\n onEnd(err)\n onEnd = undefined\n }\n\n return pushable\n },\n get readableLength () {\n return _pushable.readableLength\n },\n onEmpty: (opts?: AbortOptions) => {\n return _pushable.onEmpty(opts)\n }\n }\n\n return pushable\n}\n","import { ClientTransport } from '../transport/transport';\nimport {\n AnyService,\n ProcErrors,\n ProcHasInit,\n ProcInit,\n ProcInput,\n ProcOutput,\n ProcType,\n AnyServiceSchemaMap,\n InstantiatedServiceSchemaMap,\n} from './services';\nimport { pushable } from 'it-pushable';\nimport type { Pushable } from 'it-pushable';\nimport {\n OpaqueTransportMessage,\n ControlFlags,\n TransportClientId,\n isStreamClose,\n PartialTransportMessage,\n} from '../transport/message';\nimport { Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { Err, Result, UNEXPECTED_DISCONNECT } from './result';\nimport { EventMap } from '../transport/events';\nimport { Connection } from '../transport/session';\nimport { createProcTelemetryInfo, getPropagationContext } from '../tracing';\nimport { ClientHandshakeOptions } from './handshake';\n\n// helper to make next, yield, and return all the same type\nexport type AsyncIter<T> = AsyncGenerator<T, T>;\n\n/**\n * A helper type to transform an actual service type into a type\n * we can case to in the proxy.\n * @template Router - The type of the Router.\n */\ntype ServiceClient<Router extends AnyService> = {\n [ProcName in keyof Router['procedures']]: ProcType<\n Router,\n ProcName\n > extends 'rpc'\n ? {\n rpc: (\n input: Static<ProcInput<Router, ProcName>>,\n ) => Promise<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >;\n }\n : ProcType<Router, ProcName> extends 'upload'\n ? ProcHasInit<Router, ProcName> extends true\n ? {\n upload: (init: Static<ProcInit<Router, ProcName>>) => Promise<\n [\n Pushable<Static<ProcInput<Router, ProcName>>>, // input\n Promise<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >, // output\n ]\n >;\n }\n : {\n upload: () => Promise<\n [\n Pushable<Static<ProcInput<Router, ProcName>>>, // input\n Promise<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >, // output\n ]\n >;\n }\n : ProcType<Router, ProcName> extends 'stream'\n ? ProcHasInit<Router, ProcName> extends true\n ? {\n stream: (init: Static<ProcInit<Router, ProcName>>) => Promise<\n [\n Pushable<Static<ProcInput<Router, ProcName>>>, // input\n AsyncIter<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >, // output\n () => void, // close handle\n ]\n >;\n }\n : {\n stream: () => Promise<\n [\n Pushable<Static<ProcInput<Router, ProcName>>>, // input\n AsyncIter<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >, // output\n () => void, // close handle\n ]\n >;\n }\n : ProcType<Router, ProcName> extends 'subscription'\n ? {\n subscribe: (input: Static<ProcInput<Router, ProcName>>) => Promise<\n [\n AsyncIter<\n Result<\n Static<ProcOutput<Router, ProcName>>,\n Static<ProcErrors<Router, ProcName>>\n >\n >, // output\n () => void, // close handle\n ]\n >;\n }\n : never;\n};\n\n/**\n * Defines a type that represents a client for a server with a set of services.\n * @template Srv - The type of the server.\n */\nexport type Client<\n Services extends AnyServiceSchemaMap,\n IS extends\n InstantiatedServiceSchemaMap<Services> = InstantiatedServiceSchemaMap<Services>,\n> = {\n [SvcName in keyof IS]: ServiceClient<IS[SvcName]>;\n};\n\ninterface ProxyCallbackOptions {\n path: Array<string>;\n args: Array<unknown>;\n}\n\ntype ProxyCallback = (opts: ProxyCallbackOptions) => unknown;\n/* eslint-disable-next-line @typescript-eslint/no-empty-function */\nconst noop = () => {};\n\nfunction _createRecursiveProxy(\n callback: ProxyCallback,\n path: Array<string>,\n): unknown {\n const proxy: unknown = new Proxy(noop, {\n // property access, recurse and add field to path\n get(_obj, key) {\n if (typeof key !== 'string') return undefined;\n return _createRecursiveProxy(callback, [...path, key]);\n },\n // hit the end, let's invoke the handler\n apply(_target, _this, args) {\n return callback({\n path,\n args,\n });\n },\n });\n\n return proxy;\n}\n\nexport interface ClientOptions {\n connectOnInvoke: boolean;\n eagerlyConnect: boolean;\n}\n\nconst defaultClientOptions: ClientOptions = {\n connectOnInvoke: true,\n eagerlyConnect: true,\n};\n\n/**\n * Creates a client for a given server using the provided transport.\n * Note that the client only needs the type of the server, not the actual\n * server definition itself.\n *\n * This relies on a proxy to dynamically create the client, so the client\n * will be typed as if it were the actual server with the appropriate services\n * and procedures.\n *\n * @template Srv - The type of the server.\n * @param {Transport} transport - The transport to use for communication.\n * @param {TransportClientId} serverId - The ID of the server to connect to.\n * @param {Partial<ClientOptions>} providedClientOptions - The options for the client.\n * @returns The client for the server.\n */\nexport function createClient<ServiceSchemaMap extends AnyServiceSchemaMap>(\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n providedClientOptions: Partial<\n ClientOptions & {\n handshakeOptions: ClientHandshakeOptions;\n }\n > = {},\n): Client<ServiceSchemaMap> {\n if (providedClientOptions.handshakeOptions) {\n transport.extendHandshake(providedClientOptions.handshakeOptions);\n }\n\n const options = { ...defaultClientOptions, ...providedClientOptions };\n if (options.eagerlyConnect) {\n void transport.connect(serverId);\n }\n\n return _createRecursiveProxy(async (opts) => {\n const [serviceName, procName, procType] = [...opts.path];\n if (!(serviceName && procName && procType)) {\n throw new Error(\n 'invalid river call, ensure the service and procedure you are calling exists',\n );\n }\n\n const [input] = opts.args;\n if (options.connectOnInvoke && !transport.connections.has(serverId)) {\n void transport.connect(serverId);\n }\n\n if (procType === 'rpc') {\n return handleRpc(transport, serverId, input, serviceName, procName);\n } else if (procType === 'stream') {\n return handleStream(transport, serverId, input, serviceName, procName);\n } else if (procType === 'subscribe') {\n return handleSubscribe(transport, serverId, input, serviceName, procName);\n } else if (procType === 'upload') {\n return handleUpload(transport, serverId, input, serviceName, procName);\n } else {\n throw new Error(`invalid river call, unknown procedure type ${procType}`);\n }\n }, []) as Client<ServiceSchemaMap>;\n}\n\nfunction createSessionDisconnectHandler(\n from: TransportClientId,\n cb: () => void,\n) {\n return (evt: EventMap['sessionStatus']) => {\n if (evt.status === 'disconnect' && evt.session.to === from) {\n cb();\n }\n };\n}\n\nfunction handleRpc(\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n input: unknown,\n serviceName: string,\n procedureName: string,\n) {\n const streamId = nanoid();\n const { span, ctx } = createProcTelemetryInfo(\n transport,\n 'rpc',\n serviceName,\n procedureName,\n streamId,\n );\n transport.send(serverId, {\n streamId,\n serviceName,\n procedureName,\n payload: input,\n tracing: getPropagationContext(ctx),\n controlFlags: ControlFlags.StreamOpenBit | ControlFlags.StreamClosedBit,\n });\n\n const responsePromise = new Promise((resolve) => {\n // on disconnect, set a timer to return an error\n // on (re)connect, clear the timer\n const onSessionStatus = createSessionDisconnectHandler(serverId, () => {\n cleanup();\n resolve(\n Err({\n code: UNEXPECTED_DISCONNECT,\n message: `${serverId} unexpectedly disconnected`,\n }),\n );\n });\n\n function cleanup() {\n transport.removeEventListener('message', onMessage);\n transport.removeEventListener('sessionStatus', onSessionStatus);\n span.end();\n }\n\n function onMessage(msg: OpaqueTransportMessage) {\n if (msg.streamId !== streamId) return;\n if (msg.to !== transport.clientId) return;\n\n // cleanup and resolve as soon as we get a message\n cleanup();\n resolve(msg.payload);\n }\n\n transport.addEventListener('message', onMessage);\n transport.addEventListener('sessionStatus', onSessionStatus);\n });\n return responsePromise;\n}\n\nfunction handleStream(\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n init: unknown,\n serviceName: string,\n procedureName: string,\n) {\n const streamId = nanoid();\n const { span, ctx } = createProcTelemetryInfo(\n transport,\n 'stream',\n serviceName,\n procedureName,\n streamId,\n );\n const inputStream = pushable({ objectMode: true });\n const outputStream = pushable({ objectMode: true });\n let firstMessage = true;\n let healthyClose = true;\n\n if (init) {\n transport.send(serverId, {\n streamId,\n serviceName,\n procedureName,\n payload: init,\n tracing: getPropagationContext(ctx),\n controlFlags: ControlFlags.StreamOpenBit,\n });\n\n firstMessage = false;\n }\n\n // input -> transport\n // this gets cleaned up on inputStream.end() which is called by closeHandler\n const pipeInputToTransport = async () => {\n for await (const rawIn of inputStream) {\n const m: PartialTransportMessage = {\n streamId,\n payload: rawIn,\n controlFlags: 0,\n };\n\n if (firstMessage) {\n m.serviceName = serviceName;\n m.procedureName = procedureName;\n m.tracing = getPropagationContext(ctx);\n m.controlFlags |= ControlFlags.StreamOpenBit;\n firstMessage = false;\n }\n\n transport.send(serverId, m);\n }\n\n // after ending input stream, send a close message to the server\n if (!healthyClose) return;\n transport.sendCloseStream(serverId, streamId);\n };\n\n void pipeInputToTransport();\n\n // transport -> output\n function onMessage(msg: OpaqueTransportMessage) {\n if (msg.streamId !== streamId) return;\n if (msg.to !== transport.clientId) return;\n\n if (isStreamClose(msg.controlFlags)) {\n cleanup();\n } else {\n outputStream.push(msg.payload);\n }\n }\n\n function cleanup() {\n inputStream.end();\n outputStream.end();\n transport.removeEventListener('message', onMessage);\n transport.removeEventListener('sessionStatus', onSessionStatus);\n span.end();\n }\n\n // close stream after disconnect + grace period elapses\n const onSessionStatus = createSessionDisconnectHandler(serverId, () => {\n outputStream.push(\n Err({\n code: UNEXPECTED_DISCONNECT,\n message: `${serverId} unexpectedly disconnected`,\n }),\n );\n healthyClose = false;\n cleanup();\n });\n\n transport.addEventListener('message', onMessage);\n transport.addEventListener('sessionStatus', onSessionStatus);\n return [inputStream, outputStream, cleanup];\n}\n\nfunction handleSubscribe(\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n input: unknown,\n serviceName: string,\n procedureName: string,\n) {\n const streamId = nanoid();\n const { span, ctx } = createProcTelemetryInfo(\n transport,\n 'subscription',\n serviceName,\n procedureName,\n streamId,\n );\n\n transport.send(serverId, {\n streamId,\n serviceName,\n procedureName,\n payload: input,\n tracing: getPropagationContext(ctx),\n controlFlags: ControlFlags.StreamOpenBit,\n });\n\n let healthyClose = true;\n\n // transport -> output\n const outputStream = pushable({ objectMode: true });\n function onMessage(msg: OpaqueTransportMessage) {\n if (msg.streamId !== streamId) return;\n if (msg.to !== transport.clientId) return;\n\n if (isStreamClose(msg.controlFlags)) {\n cleanup();\n } else {\n outputStream.push(msg.payload);\n }\n }\n\n function cleanup() {\n outputStream.end();\n transport.removeEventListener('message', onMessage);\n transport.removeEventListener('sessionStatus', onSessionStatus);\n span.end();\n }\n\n const closeHandler = () => {\n cleanup();\n if (!healthyClose) return;\n transport.sendCloseStream(serverId, streamId);\n };\n\n // close stream after disconnect + grace period elapses\n const onSessionStatus = createSessionDisconnectHandler(serverId, () => {\n outputStream.push(\n Err({\n code: UNEXPECTED_DISCONNECT,\n message: `${serverId} unexpectedly disconnected`,\n }),\n );\n healthyClose = false;\n cleanup();\n });\n\n transport.addEventListener('message', onMessage);\n transport.addEventListener('sessionStatus', onSessionStatus);\n return [outputStream, closeHandler];\n}\n\nfunction handleUpload(\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n init: unknown,\n serviceName: string,\n procedureName: string,\n) {\n const streamId = nanoid();\n const { span, ctx } = createProcTelemetryInfo(\n transport,\n 'upload',\n serviceName,\n procedureName,\n streamId,\n );\n const inputStream = pushable({ objectMode: true });\n let firstMessage = true;\n let healthyClose = true;\n\n if (init) {\n transport.send(serverId, {\n streamId,\n serviceName,\n procedureName,\n payload: init,\n tracing: getPropagationContext(ctx),\n controlFlags: ControlFlags.StreamOpenBit,\n });\n\n firstMessage = false;\n }\n\n // input -> transport\n // this gets cleaned up on inputStream.end(), which the caller should call.\n const pipeInputToTransport = async () => {\n for await (const rawIn of inputStream) {\n const m: PartialTransportMessage = {\n streamId,\n payload: rawIn,\n controlFlags: 0,\n };\n\n if (firstMessage) {\n m.serviceName = serviceName;\n m.procedureName = procedureName;\n m.tracing = getPropagationContext(ctx);\n m.controlFlags |= ControlFlags.StreamOpenBit;\n firstMessage = false;\n }\n\n transport.send(serverId, m);\n }\n\n // after ending input stream, send a close message to the server\n if (!healthyClose) return;\n transport.sendCloseStream(serverId, streamId);\n };\n\n void pipeInputToTransport();\n\n const responsePromise = new Promise((resolve) => {\n // on disconnect, set a timer to return an error\n // on (re)connect, clear the timer\n const onSessionStatus = createSessionDisconnectHandler(serverId, () => {\n healthyClose = false;\n cleanup();\n resolve(\n Err({\n code: UNEXPECTED_DISCONNECT,\n message: `${serverId} unexpectedly disconnected`,\n }),\n );\n });\n\n function cleanup() {\n inputStream.end();\n transport.removeEventListener('message', onMessage);\n transport.removeEventListener('sessionStatus', onSessionStatus);\n span.end();\n }\n\n function onMessage(msg: OpaqueTransportMessage) {\n if (msg.streamId !== streamId) return;\n if (msg.to !== transport.clientId) return;\n\n // cleanup and resolve as soon as we get a message\n cleanup();\n resolve(msg.payload);\n }\n\n transport.addEventListener('message', onMessage);\n transport.addEventListener('sessionStatus', onSessionStatus);\n });\n return [inputStream, responsePromise];\n}\n","import {\n TLiteral,\n TNever,\n TObject,\n TSchema,\n TString,\n TUnion,\n Type,\n} from '@sinclair/typebox';\nimport { Client } from './client';\n\ntype TLiteralString = TLiteral<string>;\n\nexport type RiverErrorSchema =\n | TObject<{\n code: TLiteralString | TUnion<Array<TLiteralString>>;\n message: TLiteralString | TString;\n }>\n | TObject<{\n code: TLiteralString | TUnion<Array<TLiteralString>>;\n message: TLiteralString | TString;\n extras: TSchema;\n }>;\n\nexport type RiverError =\n | TUnion<Array<RiverErrorSchema>>\n | RiverErrorSchema\n | TNever;\n\nexport const UNCAUGHT_ERROR = 'UNCAUGHT_ERROR';\nexport const UNEXPECTED_DISCONNECT = 'UNEXPECTED_DISCONNECT';\nexport const RiverUncaughtSchema = Type.Object({\n code: Type.Union([\n Type.Literal(UNCAUGHT_ERROR),\n Type.Literal(UNEXPECTED_DISCONNECT),\n ]),\n message: Type.String(),\n});\n\nexport type Result<T, E> =\n | {\n ok: true;\n payload: T;\n }\n | {\n ok: false;\n payload: E;\n };\n\nexport function Ok<const T extends Array<unknown>, const E>(p: T): Result<T, E>;\nexport function Ok<const T extends ReadonlyArray<unknown>, const E>(\n p: T,\n): Result<T, E>;\nexport function Ok<const T, const E>(payload: T): Result<T, E>;\nexport function Ok<const T, const E>(payload: T): Result<T, E> {\n return {\n ok: true,\n payload,\n };\n}\n\nexport function Err<const T, const E>(error: E): Result<T, E> {\n return {\n ok: false,\n payload: error,\n };\n}\n\n/**\n * Refine a {@link Result} type to its returned payload.\n */\nexport type ResultUnwrapOk<R> = R extends Result<infer T, infer __E>\n ? T\n : never;\n\n/**\n * Refine a {@link Result} type to its error payload.\n */\nexport type ResultUnwrapErr<R> = R extends Result<infer __T, infer E>\n ? E\n : never;\n\n/**\n * Retrieve the output type for a procedure, represented as a {@link Result}\n * type.\n * Example:\n * ```\n * type Message = Output<typeof client, 'serviceName', 'procedureName'>\n * ```\n */\nexport type Output<\n RiverClient,\n ServiceName extends keyof RiverClient,\n ProcedureName extends keyof RiverClient[ServiceName],\n Procedure = RiverClient[ServiceName][ProcedureName],\n Fn extends (...args: never) => unknown = (...args: never) => unknown,\n> = RiverClient extends Client<infer __ServiceSchemaMap>\n ? Procedure extends object\n ? Procedure extends object & { rpc: infer RpcHandler extends Fn }\n ? Awaited<ReturnType<RpcHandler>>\n : Procedure extends object & { upload: infer UploadHandler extends Fn }\n ? Awaited<ReturnType<UploadHandler>> extends [\n infer __UploadInputMessage,\n infer UploadOutputMessage,\n ]\n ? Awaited<UploadOutputMessage>\n : never\n : Procedure extends object & { stream: infer StreamHandler extends Fn }\n ? Awaited<ReturnType<StreamHandler>> extends [\n infer __StreamInputMessage,\n AsyncGenerator<infer StreamOutputMessage>,\n infer __StreamCloseHandle,\n ]\n ? StreamOutputMessage\n : never\n : Procedure extends object & {\n subscribe: infer SubscriptionHandler extends Fn;\n }\n ? Awaited<ReturnType<SubscriptionHandler>> extends [\n AsyncGenerator<infer SubscriptionOutputMessage>,\n infer __SubscriptionCloseHandle,\n ]\n ? SubscriptionOutputMessage\n : never\n : never\n : never\n : never;\n","import { Static } from '@sinclair/typebox';\nimport { ServerTransport } from '../transport/transport';\nimport { AnyProcedure, PayloadType } from './procedures';\nimport {\n AnyService,\n InstantiatedServiceSchemaMap,\n AnyServiceSchemaMap,\n} from './services';\nimport { pushable } from 'it-pushable';\nimport type { Pushable } from 'it-pushable';\nimport {\n ControlMessagePayloadSchema,\n OpaqueTransportMessage,\n isStreamClose,\n isStreamOpen,\n TransportClientId,\n ControlFlags,\n} from '../transport/message';\nimport {\n ServiceContext,\n ServiceContextWithState,\n ServiceContextWithTransportInfo,\n} from './context';\nimport { Logger } from '../logging/log';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n Err,\n Result,\n RiverError,\n RiverUncaughtSchema,\n UNCAUGHT_ERROR,\n} from './result';\nimport { EventMap } from '../transport/events';\nimport { Connection } from '../transport/session';\nimport { coerceErrorString } from '../util/stringify';\nimport { Span, SpanStatusCode } from '@opentelemetry/api';\nimport { createHandlerSpan } from '../tracing';\nimport { ServerHandshakeOptions } from './handshake';\n\n/**\n * Represents a server with a set of services. Use {@link createServer} to create it.\n * @template Services - The type of services provided by the server.\n */\nexport interface Server<Services extends AnyServiceSchemaMap> {\n services: InstantiatedServiceSchemaMap<Services>;\n streams: Map<string, ProcStream>;\n close(): Promise<void>;\n}\n\ninterface ProcStream {\n id: string;\n serviceName: string;\n procedureName: string;\n incoming: Pushable<PayloadType>;\n outgoing: Pushable<Result<Static<PayloadType>, Static<RiverError>>>;\n promises: {\n outputHandler: Promise<unknown>;\n inputHandler: Promise<unknown>;\n };\n}\n\nclass RiverServer<Services extends AnyServiceSchemaMap> {\n transport: ServerTransport<Connection>;\n services: InstantiatedServiceSchemaMap<Services>;\n contextMap: Map<AnyService, ServiceContextWithState<object>>;\n // map of streamId to ProcStream\n streamMap: Map<string, ProcStream>;\n // map of client to their open streams by streamId\n clientStreams: Map<TransportClientId, Set<string>>;\n disconnectedSessions: Set<TransportClientId>;\n\n private log?: Logger;\n constructor(\n transport: ServerTransport<Connection>,\n services: Services,\n handshakeOptions?: ServerHandshakeOptions,\n extendedContext?: Omit<ServiceContext, 'state'>,\n ) {\n const instances: Record<string, AnyService> = {};\n\n this.services = instances as InstantiatedServiceSchemaMap<Services>;\n this.contextMap = new Map();\n\n for (const [name, service] of Object.entries(services)) {\n const instance = service.instantiate();\n instances[name] = instance;\n\n this.contextMap.set(instance, {\n ...extendedContext,\n state: instance.state,\n });\n }\n\n if (handshakeOptions) {\n transport.extendHandshake(handshakeOptions);\n }\n\n this.transport = transport;\n this.disconnectedSessions = new Set();\n this.streamMap = new Map();\n this.clientStreams = new Map();\n this.transport.addEventListener('message', this.onMessage);\n this.transport.addEventListener('sessionStatus', this.onSessionStatus);\n this.log = transport.log;\n }\n\n get streams() {\n return this.streamMap;\n }\n\n onMessage = async (message: OpaqueTransportMessage) => {\n if (message.to !== this.transport.clientId) {\n this.log?.info(\n `got msg with destination that isn't this server, ignoring`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n },\n );\n return;\n }\n\n let procStream = this.streamMap.get(message.streamId);\n const isInitMessage = !procStream;\n\n // create a proc stream if it doesnt exist\n procStream ||= this.createNewProcStream(message);\n if (!procStream) {\n // if we fail to create a proc stream, just abort\n return;\n }\n\n await this.pushToStream(procStream, message, isInitMessage);\n };\n\n // cleanup streams on session close\n onSessionStatus = async (evt: EventMap['sessionStatus']) => {\n if (evt.status !== 'disconnect') return;\n\n const disconnectedClientId = evt.session.to;\n this.log?.info(\n `got session disconnect from ${disconnectedClientId}, cleaning up streams`,\n evt.session.loggingMetadata,\n );\n\n const streamsFromThisClient = this.clientStreams.get(disconnectedClientId);\n if (!streamsFromThisClient) return;\n\n this.disconnectedSessions.add(disconnectedClientId);\n await Promise.all(\n Array.from(streamsFromThisClient).map(this.cleanupStream),\n );\n this.disconnectedSessions.delete(disconnectedClientId);\n this.clientStreams.delete(disconnectedClientId);\n };\n\n async close() {\n this.transport.removeEventListener('message', this.onMessage);\n this.transport.removeEventListener('sessionStatus', this.onSessionStatus);\n await Promise.all([...this.streamMap.keys()].map(this.cleanupStream));\n\n for (const context of this.contextMap.values()) {\n if (Symbol.dispose in context.state) {\n const dispose = context.state[Symbol.dispose];\n if (typeof dispose === 'function') {\n dispose();\n }\n }\n }\n }\n\n createNewProcStream(message: OpaqueTransportMessage) {\n if (!isStreamOpen(message.controlFlags)) {\n this.log?.error(\n `can't create a new procedure stream from a message that doesn't have the stream open bit set`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n tags: ['invariant-violation'],\n },\n );\n return;\n }\n\n if (!message.procedureName || !message.serviceName) {\n this.log?.warn(\n `missing procedure or service name in stream open message`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n },\n );\n return;\n }\n\n if (!(message.serviceName in this.services)) {\n this.log?.warn(`couldn't find service ${message.serviceName}`, {\n clientId: this.transport.clientId,\n transportMessage: message,\n });\n return;\n }\n\n const service = this.services[message.serviceName];\n const serviceContext = this.getContext(service, message.serviceName);\n if (!(message.procedureName in service.procedures)) {\n this.log?.warn(\n `couldn't find a matching procedure for ${message.serviceName}.${message.procedureName}`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n },\n );\n return;\n }\n\n const session = this.transport.sessions.get(message.from);\n if (!session) {\n this.log?.warn(`couldn't find session for ${message.from}`, {\n clientId: this.transport.clientId,\n transportMessage: message,\n });\n return;\n }\n\n const procedure = service.procedures[message.procedureName];\n const incoming: ProcStream['incoming'] = pushable({ objectMode: true });\n const outgoing: ProcStream['outgoing'] = pushable({ objectMode: true });\n const needsClose =\n procedure.type === 'subscription' || procedure.type === 'stream';\n const disposables: Array<() => void> = [];\n\n const outputHandler: Promise<unknown> =\n // sending outgoing messages back to client\n needsClose\n ? // subscription and stream case, we need to send a close bit after the response stream\n (async () => {\n for await (const response of outgoing) {\n this.transport.send(session.to, {\n streamId: message.streamId,\n controlFlags: 0,\n payload: response,\n });\n }\n\n // we ended, send a close bit back to the client\n // also, if the client has disconnected, we don't need to send a close\n if (!this.disconnectedSessions.has(message.from)) {\n this.transport.sendCloseStream(session.to, message.streamId);\n }\n\n // call disposables returned from handlers\n disposables.forEach((d) => d());\n })()\n : // rpc and upload case, we just send the response back with close bit\n (async () => {\n const response = await outgoing.next().then((res) => res.value);\n if (response) {\n this.transport.send(session.to, {\n streamId: message.streamId,\n controlFlags: ControlFlags.StreamClosedBit,\n payload: response,\n });\n\n // call disposables returned from handlers\n disposables.forEach((d) => d());\n }\n })();\n\n const errorHandler = (err: unknown, span: Span) => {\n const errorMsg = coerceErrorString(err);\n this.log?.error(\n `procedure ${message.serviceName}.${message.procedureName} threw an uncaught error: ${errorMsg}`,\n session.loggingMetadata,\n );\n\n span.recordException(err instanceof Error ? err : new Error(errorMsg));\n span.setStatus({ code: SpanStatusCode.ERROR });\n outgoing.push(\n Err({\n code: UNCAUGHT_ERROR,\n message: errorMsg,\n } satisfies Static<typeof RiverUncaughtSchema>),\n );\n };\n\n const sessionMeta = this.transport.sessionHandshakeMetadata.get(session);\n if (!sessionMeta) {\n this.log?.error(`session doesn't have handshake metadata`, {\n ...session.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n // pump incoming message stream -> handler -> outgoing message stream\n let inputHandler: Promise<unknown>;\n const procHasInitMessage = 'init' in procedure;\n const serviceContextWithTransportInfo: ServiceContextWithTransportInfo<object> =\n {\n ...serviceContext,\n to: message.to,\n from: message.from,\n streamId: message.streamId,\n session,\n metadata: sessionMeta,\n };\n\n switch (procedure.type) {\n case 'rpc':\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n const inputMessage = await incoming.next();\n if (inputMessage.done) {\n return;\n }\n try {\n const outputMessage = await procedure.handler(\n serviceContextWithTransportInfo,\n inputMessage.value,\n );\n outgoing.push(outputMessage);\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n break;\n case 'stream':\n if (procHasInitMessage) {\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n const initMessage = await incoming.next();\n if (initMessage.done) {\n return;\n }\n\n try {\n const dispose = await procedure.handler(\n serviceContextWithTransportInfo,\n initMessage.value,\n incoming,\n outgoing,\n );\n\n if (dispose) {\n disposables.push(dispose);\n }\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n } else {\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n try {\n const dispose = await procedure.handler(\n serviceContextWithTransportInfo,\n incoming,\n outgoing,\n );\n if (dispose) {\n disposables.push(dispose);\n }\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n }\n break;\n case 'subscription':\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n const inputMessage = await incoming.next();\n if (inputMessage.done) {\n return;\n }\n\n try {\n const dispose = await procedure.handler(\n serviceContextWithTransportInfo,\n inputMessage.value,\n outgoing,\n );\n\n if (dispose) {\n disposables.push(dispose);\n }\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n break;\n case 'upload':\n if (procHasInitMessage) {\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n const initMessage = await incoming.next();\n if (initMessage.done) {\n return;\n }\n try {\n const outputMessage = await procedure.handler(\n serviceContextWithTransportInfo,\n initMessage.value,\n incoming,\n );\n\n if (!this.disconnectedSessions.has(message.from)) {\n outgoing.push(outputMessage);\n }\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n } else {\n inputHandler = createHandlerSpan(\n procedure.type,\n message,\n async (span) => {\n try {\n const outputMessage = await procedure.handler(\n serviceContextWithTransportInfo,\n incoming,\n );\n\n if (!this.disconnectedSessions.has(message.from)) {\n outgoing.push(outputMessage);\n }\n } catch (err) {\n errorHandler(err, span);\n } finally {\n span.end();\n }\n },\n );\n }\n\n break;\n default:\n // procedure is inferred to be never here as this is not a valid procedure type\n // we cast just to this.log\n this.log?.warn(\n `got request for invalid procedure type ${\n (procedure as AnyProcedure).type\n } at ${message.serviceName}.${message.procedureName}`,\n { ...session.loggingMetadata, transportMessage: message },\n );\n return;\n }\n\n const procStream: ProcStream = {\n id: message.streamId,\n incoming,\n outgoing,\n serviceName: message.serviceName,\n procedureName: message.procedureName,\n promises: { inputHandler, outputHandler },\n };\n\n this.streamMap.set(message.streamId, procStream);\n\n // add this stream to ones from that client so we can clean it up in the case of a disconnect without close\n const streamsFromThisClient =\n this.clientStreams.get(message.from) ?? new Set();\n streamsFromThisClient.add(message.streamId);\n this.clientStreams.set(message.from, streamsFromThisClient);\n\n return procStream;\n }\n\n async pushToStream(\n procStream: ProcStream,\n message: OpaqueTransportMessage,\n isInit?: boolean,\n ) {\n const { serviceName, procedureName } = procStream;\n const procedure = this.services[serviceName].procedures[procedureName];\n const procHasInitMessage = 'init' in procedure;\n\n if (isInit && procHasInitMessage) {\n if (Value.Check(procedure.init, message.payload)) {\n procStream.incoming.push(message.payload as PayloadType);\n } else {\n this.log?.error(\n `procedure ${serviceName}.${procedureName} received invalid init payload`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n validationErrors: [\n ...Value.Errors(procedure.init, message.payload),\n ],\n },\n );\n }\n } else if (Value.Check(procedure.input, message.payload)) {\n procStream.incoming.push(message.payload as PayloadType);\n } else if (!Value.Check(ControlMessagePayloadSchema, message.payload)) {\n // whelp we got a message that isn't a control message and doesn't match the procedure input\n // so definitely not a valid payload\n this.log?.error(\n `procedure ${serviceName}.${procedureName} received invalid payload`,\n {\n clientId: this.transport.clientId,\n transportMessage: message,\n validationErrors: [...Value.Errors(procedure.input, message.payload)],\n },\n );\n }\n\n if (isStreamClose(message.controlFlags)) {\n await this.cleanupStream(message.streamId);\n\n const streamsFromThisClient = this.clientStreams.get(message.from);\n if (streamsFromThisClient) {\n streamsFromThisClient.delete(message.streamId);\n if (streamsFromThisClient.size === 0) {\n this.clientStreams.delete(message.from);\n }\n }\n }\n }\n\n private getContext(service: AnyService, serviceName: string) {\n const context = this.contextMap.get(service);\n if (!context) {\n const err = `no context found for ${serviceName}`;\n this.log?.error(err, {\n clientId: this.transport.clientId,\n tags: ['invariant-violation'],\n });\n throw new Error(err);\n }\n\n return context;\n }\n\n cleanupStream = async (id: string) => {\n const stream = this.streamMap.get(id);\n if (!stream) {\n return;\n }\n\n // end the streams and wait for the handlers to finish\n stream.incoming.end();\n await stream.promises.inputHandler;\n stream.outgoing.end();\n await stream.promises.outputHandler;\n this.streamMap.delete(id);\n };\n}\n\n/**\n * Creates a server instance that listens for incoming messages from a transport and routes them to the appropriate service and procedure.\n * The server tracks the state of each service along with open streams and the extended context object.\n * @param transport - The transport to listen to.\n * @param services - An object containing all the services to be registered on the server.\n * @param handshakeOptions - An optional object containing additional handshake options to be passed to the transport.\n * @param extendedContext - An optional object containing additional context to be passed to all services.\n * @returns A promise that resolves to a server instance with the registered services.\n */\nexport function createServer<Services extends AnyServiceSchemaMap>(\n transport: ServerTransport<Connection>,\n services: Services,\n providedServerOptions?: Partial<{\n handshakeOptions?: ServerHandshakeOptions;\n extendedContext?: Omit<ServiceContext, 'state'>;\n }>,\n): Server<Services> {\n return new RiverServer(\n transport,\n services,\n providedServerOptions?.handshakeOptions,\n providedServerOptions?.extendedContext,\n );\n}\n","import { Static, TSchema } from '@sinclair/typebox';\nimport { ParsedMetadata } from './context';\n\ntype ConstructHandshake<T extends TSchema> = () =>\n | Static<T>\n | Promise<Static<T>>;\n\ntype ValidateHandshake<T extends TSchema> = (\n metadata: Static<T>,\n previousParsedMetadata?: ParsedMetadata,\n) => false | ParsedMetadata | Promise<false | ParsedMetadata>;\n\nexport interface ClientHandshakeOptions<\n MetadataSchema extends TSchema = TSchema,\n> {\n /**\n * Schema for the metadata that the client sends to the server\n * during the handshake.\n */\n schema: MetadataSchema;\n\n /**\n * Gets the {@link HandshakeRequestMetadata} to send to the server.\n */\n construct: ConstructHandshake<MetadataSchema>;\n}\n\nexport interface ServerHandshakeOptions<\n MetadataSchema extends TSchema = TSchema,\n> {\n /**\n * Schema for the metadata that the server receives from the client\n * during the handshake.\n */\n schema: MetadataSchema;\n\n /**\n * Parses the {@link HandshakeRequestMetadata} sent by the client, transforming\n * it into {@link ParsedHandshakeMetadata}.\n *\n * May return `false` if the client should be rejected.\n *\n * @param metadata - The metadata sent by the client.\n * @param session - The session that the client would be associated with.\n * @param isReconnect - Whether the client is reconnecting to the session,\n * or if this is a new session.\n */\n validate: ValidateHandshake<MetadataSchema>;\n}\n\nexport function createClientHandshakeOptions<\n MetadataSchema extends TSchema = TSchema,\n>(\n schema: MetadataSchema,\n construct: ConstructHandshake<MetadataSchema>,\n): ClientHandshakeOptions {\n return { schema, construct };\n}\n\nexport function createServerHandshakeOptions<\n MetadataSchema extends TSchema = TSchema,\n>(\n schema: MetadataSchema,\n validate: ValidateHandshake<MetadataSchema>,\n): ServerHandshakeOptions {\n return { schema, validate: validate as ValidateHandshake<TSchema> };\n}\n"],"mappings":";;;;;;;;;;;AAAA,SAAS,YAA6B;AA6J/B,SAAS,gBACd,UACA,iBACwB;AACxB,QAAM,0BAA0B,OAAO,QAAQ,QAAQ,EAAE,OAEvD,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM;AACxB,QAAI,IAAI,IAAI,MAAM,UAAU;AAC5B,WAAO;AAAA,EACT,GAAG,CAAC,CAAC;AAEL,QAAM,SAAiC;AAAA,IACrC,UAAU;AAAA,EACZ;AAEA,MAAI,iBAAiB;AACnB,WAAO,kBAAkB,KAAK,OAAO,eAAe;AAAA,EACtD;AAEA,SAAO;AACT;AAoBO,IAAM,gBAAN,MAAM,eAGX;AAAA;AAAA;AAAA;AAAA,EAImB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA;AAAA,EAMC,YACR,QACA,YACA;AACA,SAAK,kBAAkB,OAAO;AAC9B,SAAK,aAAa;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAwDA,OAAO,SAA+B,QAAqC;AACzE,WAAO,IAAI,gBAAgB,MAAM;AAAA,EACnC;AAAA;AAAA,EAqEA,OAAO,OACL,oBAGA,iBACqC;AACrC,QAAI;AACJ,QAAI;AAEJ,QACE,qBAAqB,sBACrB,OAAO,mBAAmB,oBAAoB,YAC9C;AACA,UAAI,CAAC,iBAAiB;AACpB,cAAM,IAAI,MAAM,mCAAmC;AAAA,MACrD;AAEA,eAAS;AACT,mBAAa;AAAA,IACf,OAAO;AACL,eAAS,EAAE,iBAAiB,OAAO,CAAC,GAAG;AACvC,mBAAa;AAAA,IACf;AAEA,WAAO,IAAI,eAAc,QAAQ,UAAU;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,YAAqC;AACnC,WAAO;AAAA,MACL,YAAY,OAAO;AAAA,QACjB,OAAO,QAAQ,KAAK,UAAU,EAAE,IAAI,CAAC,CAAC,UAAU,OAAO,MAAM;AAAA,UAC3D;AAAA,UACA;AAAA,YACE,OAAO,KAAK,OAAO,QAAQ,KAAK;AAAA,YAChC,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA;AAAA,YAElC,GAAI,iBAAiB,UACjB,EAAE,aAAa,QAAQ,YAAY,IACnC,CAAC;AAAA;AAAA,YAEL,GAAI,YAAY,UACZ;AAAA,cACE,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA,YACpC,IACA,CAAC;AAAA,YACL,MAAM,QAAQ;AAAA;AAAA,YAEd,GAAI,UAAU,UACV;AAAA,cACE,MAAM,KAAK,OAAO,QAAQ,IAAI;AAAA,YAChC,IACA,CAAC;AAAA,UACP;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,cAA0C;AACxC,WAAO,OAAO,OAAO;AAAA,MACnB,OAAO,KAAK,gBAAgB;AAAA,MAC5B,YAAY,KAAK;AAAA,IACnB,CAAC;AAAA,EACH;AACF;AAQA,IAAM,kBAAN,MAA4C;AAAA;AAAA;AAAA;AAAA,EAIvB;AAAA;AAAA;AAAA;AAAA,EAKnB,YAAY,QAAqC;AAC/C,SAAK,SAAS;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAsBA,WAAiD,YAAkB;AACjE,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAoBA,SACE,YAC2D;AAC3D,WAAO,cAAc,OAAO,KAAK,QAAQ,UAAU;AAAA,EACrD;AACF;;;ACzeA,SAAyB,QAAAA,aAAqB;AA8Q9C,SAAS,IAAI;AAAA,EACX;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAWG;AACD,SAAO;AAAA,IACL,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AA8DA,SAAS,OAAO;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAaG;AACD,SAAO,SAAS,UAAa,SAAS,OAClC;AAAA,IACE,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IACA;AAAA,IACE,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACN;AAiCA,SAAS,aAAa;AAAA,EACpB;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAWG;AACD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AA8DA,SAAS,OAAO;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAaG;AACD,SAAO,SAAS,UAAa,SAAS,OAClC;AAAA,IACE,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IACA;AAAA,IACE,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACN;AAMO,IAAM,YAAY;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;;;ACxjBe,SAAR,SAA0B;AAChC,QAAM,WAAW,CAAC;AAElB,WAAS,UAAU,IAAI,QAAQ,CAAC,SAAS,WAAW;AACnD,aAAS,UAAU;AACnB,aAAS,SAAS;AAAA,EACnB,CAAC;AAED,SAAO;AACR;;;ACDA,IAAM,YAAN,MAAe;EACN;EACU;EACT;EACA;EACD;EAEP,YAAa,KAAW;AACtB,QAAI,EAAE,MAAM,OAAQ,MAAM,IAAK,SAAS,GAAG;AACzC,YAAM,IAAI,MAAM,mDAAmD;;AAGrE,SAAK,SAAS,IAAI,MAAM,GAAG;AAC3B,SAAK,OAAO,MAAM;AAClB,SAAK,MAAM;AACX,SAAK,MAAM;AACX,SAAK,OAAO;EACd;EAEA,KAAM,MAAa;AACjB,QAAI,KAAK,OAAO,KAAK,GAAG,MAAM,QAAW;AACvC,aAAO;;AAGT,SAAK,OAAO,KAAK,GAAG,IAAI;AACxB,SAAK,MAAO,KAAK,MAAM,IAAK,KAAK;AAEjC,WAAO;EACT;EAEA,QAAK;AACH,UAAM,OAAO,KAAK,OAAO,KAAK,GAAG;AAEjC,QAAI,SAAS,QAAW;AACtB,aAAO;;AAGT,SAAK,OAAO,KAAK,GAAG,IAAI;AACxB,SAAK,MAAO,KAAK,MAAM,IAAK,KAAK;AACjC,WAAO;EACT;EAEA,UAAO;AACL,WAAO,KAAK,OAAO,KAAK,GAAG,MAAM;EACnC;;AAUI,IAAO,OAAP,MAAW;EACR;EACU;EACT;EACA;EAER,YAAa,UAAuB,CAAA,GAAE;AACpC,SAAK,MAAM,QAAQ,cAAc;AACjC,SAAK,OAAO,IAAI,UAAa,KAAK,GAAG;AACrC,SAAK,OAAO,KAAK;AACjB,SAAK,OAAO;EACd;EAEA,cAAe,KAAQ;AACrB,QAAI,KAAK,cAAc,MAAM;AAC3B,aAAO,IAAI;;AAGb,WAAO;EACT;EAEA,KAAM,KAAY;AAChB,QAAI,KAAK,SAAS,MAAM;AACtB,WAAK,QAAQ,KAAK,cAAc,IAAI,KAAK;;AAG3C,QAAI,CAAC,KAAK,KAAK,KAAK,GAAG,GAAG;AACxB,YAAM,OAAO,KAAK;AAClB,WAAK,OAAO,KAAK,OAAO,IAAI,UAAa,IAAI,KAAK,KAAK,OAAO,MAAM;AACpE,WAAK,KAAK,KAAK,GAAG;;EAEtB;EAEA,QAAK;AACH,QAAI,MAAM,KAAK,KAAK,MAAK;AAEzB,QAAI,QAAQ,UAAc,KAAK,KAAK,QAAQ,MAAO;AACjD,YAAM,OAAO,KAAK,KAAK;AACvB,WAAK,KAAK,OAAO;AACjB,WAAK,OAAO;AACZ,YAAM,KAAK,KAAK,MAAK;;AAGvB,QAAI,KAAK,SAAS,MAAM;AACtB,WAAK,QAAQ,KAAK,cAAc,IAAI,KAAK;;AAG3C,WAAO;EACT;EAEA,UAAO;AACL,WAAO,KAAK,KAAK,QAAO;EAC1B;;;;AC9DI,IAAO,aAAP,cAA0B,MAAK;EACnC;EACA;EAEA,YAAa,SAAkB,MAAa;AAC1C,UAAM,WAAW,2BAA2B;AAC5C,SAAK,OAAO;AACZ,SAAK,OAAO,QAAQ;EACtB;;AAoFI,SAAU,SAAa,UAAmB,CAAA,GAAE;AAChD,QAAM,UAAU,CAAC,WAAkC;AACjD,UAAM,OAA4B,OAAO,MAAK;AAE9C,QAAI,QAAQ,MAAM;AAChB,aAAO,EAAE,MAAM,KAAI;;AAGrB,QAAI,KAAK,SAAS,MAAM;AACtB,YAAM,KAAK;;AAGb,WAAO;MACL,MAAM,KAAK,SAAS;;MAEpB,OAAO,KAAK;;EAEhB;AAEA,SAAO,UAA6B,SAAS,OAAO;AACtD;AAuCA,SAAS,UAA4C,SAAuC,SAAiB;AAC3G,YAAU,WAAW,CAAA;AACrB,MAAI,QAAQ,QAAQ;AACpB,MAAI,SAAS,IAAI,KAAI;AACrB,MAAIC;AACJ,MAAI;AACJ,MAAI;AACJ,MAAI,QAAQ,OAAQ;AAEpB,QAAM,WAAW,YAA2C;AAC1D,QAAI;AACF,UAAI,CAAC,OAAO,QAAO,GAAI;AACrB,eAAO,QAAQ,MAAM;;AAGvB,UAAI,OAAO;AACT,eAAO,EAAE,MAAM,KAAI;;AAGrB,aAAO,MAAM,IAAI,QAA+B,CAAC,SAAS,WAAU;AAClE,iBAAS,CAAC,SAAwB;AAChC,mBAAS;AACT,iBAAO,KAAK,IAAI;AAEhB,cAAI;AACF,oBAAQ,QAAQ,MAAM,CAAC;mBAChB,KAAK;AACZ,mBAAO,GAAG;;AAGZ,iBAAOA;QACT;MACF,CAAC;;AAED,UAAI,OAAO,QAAO,GAAI;AAGpB,uBAAe,MAAK;AAClB,gBAAM,QAAO;AACb,kBAAQ,OAAQ;QAClB,CAAC;;;EAGP;AAEA,QAAM,aAAa,CAAC,SAAoC;AACtD,QAAI,UAAU,MAAM;AAClB,aAAO,OAAO,IAAI;;AAGpB,WAAO,KAAK,IAAI;AAChB,WAAOA;EACT;AAEA,QAAM,cAAc,CAAC,QAA0B;AAC7C,aAAS,IAAI,KAAI;AAEjB,QAAI,UAAU,MAAM;AAClB,aAAO,OAAO,EAAE,OAAO,IAAG,CAAE;;AAG9B,WAAO,KAAK,EAAE,OAAO,IAAG,CAAE;AAC1B,WAAOA;EACT;AAEA,QAAM,OAAO,CAAC,UAA+B;AAC3C,QAAI,OAAO;AACT,aAAOA;;AAIT,QAAI,SAAS,eAAe,QAAQ,OAAO,cAAc,MAAM;AAC7D,YAAM,IAAI,MAAM,gEAAgE;;AAGlF,WAAO,WAAW,EAAE,MAAM,OAAO,MAAK,CAAE;EAC1C;AACA,QAAM,MAAM,CAAC,QAA2B;AACtC,QAAI;AAAO,aAAOA;AAClB,YAAQ;AAER,WAAQ,OAAO,OAAQ,YAAY,GAAG,IAAI,WAAW,EAAE,MAAM,KAAI,CAAE;EACrE;AACA,QAAM,UAAU,MAAiB;AAC/B,aAAS,IAAI,KAAI;AACjB,QAAG;AAEH,WAAO,EAAE,MAAM,KAAI;EACrB;AACA,QAAM,SAAS,CAAC,QAA0B;AACxC,QAAI,GAAG;AAEP,WAAO,EAAE,MAAM,KAAI;EACrB;AAEA,EAAAA,YAAW;IACT,CAAC,OAAO,aAAa,IAAC;AAAM,aAAO;IAAK;IACxC,MAAM;IACN,QAAQ;IACR,OAAO;IACP;IACA;IACA,IAAI,iBAAc;AAChB,aAAO,OAAO;IAChB;IACA,SAAS,OAAOC,aAA0B;AACxC,YAAM,SAASA,UAAS;AACxB,cAAQ,eAAc;AAEtB,UAAI,OAAO,QAAO,GAAI;AACpB;;AAGF,UAAI;AACJ,UAAI;AAEJ,UAAI,UAAU,MAAM;AAClB,iBAAS,IAAI,QAAQ,CAAC,SAAS,WAAU;AACvC,qBAAW,MAAK;AACd,mBAAO,IAAI,WAAU,CAAE;UACzB;AAEA,iBAAO,iBAAiB,SAAS,QAAQ;QAC3C,CAAC;;AAGH,UAAI;AACF,cAAM,QAAQ,KAAK;UACjB,MAAM;UACN;SACD;;AAED,YAAI,YAAY,QAAQ,UAAU,MAAM;AACtC,kBAAQ,oBAAoB,SAAS,QAAQ;;;IAGnD;;AAGF,MAAI,SAAS,MAAM;AACjB,WAAOD;;AAGT,QAAME,aAAYF;AAElB,EAAAA,YAAW;IACT,CAAC,OAAO,aAAa,IAAC;AAAM,aAAO;IAAK;IACxC,OAAI;AACF,aAAOE,WAAU,KAAI;IACvB;IACA,MAAO,KAAU;AACf,MAAAA,WAAU,MAAM,GAAG;AAEnB,UAAI,SAAS,MAAM;AACjB,cAAM,GAAG;AACT,gBAAQ;;AAGV,aAAO,EAAE,MAAM,KAAI;IACrB;IACA,SAAM;AACJ,MAAAA,WAAU,OAAM;AAEhB,UAAI,SAAS,MAAM;AACjB,cAAK;AACL,gBAAQ;;AAGV,aAAO,EAAE,MAAM,KAAI;IACrB;IACA;IACA,IAAK,KAAU;AACb,MAAAA,WAAU,IAAI,GAAG;AAEjB,UAAI,SAAS,MAAM;AACjB,cAAM,GAAG;AACT,gBAAQ;;AAGV,aAAOF;IACT;IACA,IAAI,iBAAc;AAChB,aAAOE,WAAU;IACnB;IACA,SAAS,CAAC,SAAuB;AAC/B,aAAOA,WAAU,QAAQ,IAAI;IAC/B;;AAGF,SAAOF;AACT;;;ACnXA,SAAS,cAAc;;;ACtBvB;AAAA,EAOE,QAAAG;AAAA,OACK;AAqBA,IAAM,iBAAiB;AACvB,IAAM,wBAAwB;AAC9B,IAAM,sBAAsBA,MAAK,OAAO;AAAA,EAC7C,MAAMA,MAAK,MAAM;AAAA,IACfA,MAAK,QAAQ,cAAc;AAAA,IAC3BA,MAAK,QAAQ,qBAAqB;AAAA,EACpC,CAAC;AAAA,EACD,SAASA,MAAK,OAAO;AACvB,CAAC;AAiBM,SAAS,GAAqB,SAA0B;AAC7D,SAAO;AAAA,IACL,IAAI;AAAA,IACJ;AAAA,EACF;AACF;AAEO,SAAS,IAAsB,OAAwB;AAC5D,SAAO;AAAA,IACL,IAAI;AAAA,IACJ,SAAS;AAAA,EACX;AACF;;;ADgFA,IAAM,OAAO,MAAM;AAAC;AAEpB,SAAS,sBACP,UACA,MACS;AACT,QAAM,QAAiB,IAAI,MAAM,MAAM;AAAA;AAAA,IAErC,IAAI,MAAM,KAAK;AACb,UAAI,OAAO,QAAQ;AAAU,eAAO;AACpC,aAAO,sBAAsB,UAAU,CAAC,GAAG,MAAM,GAAG,CAAC;AAAA,IACvD;AAAA;AAAA,IAEA,MAAM,SAAS,OAAO,MAAM;AAC1B,aAAO,SAAS;AAAA,QACd;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAOA,IAAM,uBAAsC;AAAA,EAC1C,iBAAiB;AAAA,EACjB,gBAAgB;AAClB;AAiBO,SAAS,aACd,WACA,UACA,wBAII,CAAC,GACqB;AAC1B,MAAI,sBAAsB,kBAAkB;AAC1C,cAAU,gBAAgB,sBAAsB,gBAAgB;AAAA,EAClE;AAEA,QAAM,UAAU,EAAE,GAAG,sBAAsB,GAAG,sBAAsB;AACpE,MAAI,QAAQ,gBAAgB;AAC1B,SAAK,UAAU,QAAQ,QAAQ;AAAA,EACjC;AAEA,SAAO,sBAAsB,OAAO,SAAS;AAC3C,UAAM,CAAC,aAAa,UAAU,QAAQ,IAAI,CAAC,GAAG,KAAK,IAAI;AACvD,QAAI,EAAE,eAAe,YAAY,WAAW;AAC1C,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,CAAC,KAAK,IAAI,KAAK;AACrB,QAAI,QAAQ,mBAAmB,CAAC,UAAU,YAAY,IAAI,QAAQ,GAAG;AACnE,WAAK,UAAU,QAAQ,QAAQ;AAAA,IACjC;AAEA,QAAI,aAAa,OAAO;AACtB,aAAO,UAAU,WAAW,UAAU,OAAO,aAAa,QAAQ;AAAA,IACpE,WAAW,aAAa,UAAU;AAChC,aAAO,aAAa,WAAW,UAAU,OAAO,aAAa,QAAQ;AAAA,IACvE,WAAW,aAAa,aAAa;AACnC,aAAO,gBAAgB,WAAW,UAAU,OAAO,aAAa,QAAQ;AAAA,IAC1E,WAAW,aAAa,UAAU;AAChC,aAAO,aAAa,WAAW,UAAU,OAAO,aAAa,QAAQ;AAAA,IACvE,OAAO;AACL,YAAM,IAAI,MAAM,8CAA8C,QAAQ,EAAE;AAAA,IAC1E;AAAA,EACF,GAAG,CAAC,CAAC;AACP;AAEA,SAAS,+BACP,MACA,IACA;AACA,SAAO,CAAC,QAAmC;AACzC,QAAI,IAAI,WAAW,gBAAgB,IAAI,QAAQ,OAAO,MAAM;AAC1D,SAAG;AAAA,IACL;AAAA,EACF;AACF;AAEA,SAAS,UACP,WACA,UACA,OACA,aACA,eACA;AACA,QAAM,WAAW,OAAO;AACxB,QAAM,EAAE,MAAM,IAAI,IAAI;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,YAAU,KAAK,UAAU;AAAA,IACvB;AAAA,IACA;AAAA,IACA;AAAA,IACA,SAAS;AAAA,IACT,SAAS,sBAAsB,GAAG;AAAA,IAClC;AAAA,EACF,CAAC;AAED,QAAM,kBAAkB,IAAI,QAAQ,CAAC,YAAY;AAG/C,UAAM,kBAAkB,+BAA+B,UAAU,MAAM;AACrE,cAAQ;AACR;AAAA,QACE,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS,GAAG,QAAQ;AAAA,QACtB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,aAAS,UAAU;AACjB,gBAAU,oBAAoB,WAAW,SAAS;AAClD,gBAAU,oBAAoB,iBAAiB,eAAe;AAC9D,WAAK,IAAI;AAAA,IACX;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,IAAI,aAAa;AAAU;AAC/B,UAAI,IAAI,OAAO,UAAU;AAAU;AAGnC,cAAQ;AACR,cAAQ,IAAI,OAAO;AAAA,IACrB;AAEA,cAAU,iBAAiB,WAAW,SAAS;AAC/C,cAAU,iBAAiB,iBAAiB,eAAe;AAAA,EAC7D,CAAC;AACD,SAAO;AACT;AAEA,SAAS,aACP,WACA,UACA,MACA,aACA,eACA;AACA,QAAM,WAAW,OAAO;AACxB,QAAM,EAAE,MAAM,IAAI,IAAI;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,QAAM,cAAc,SAAS,EAAE,YAAY,KAAK,CAAC;AACjD,QAAM,eAAe,SAAS,EAAE,YAAY,KAAK,CAAC;AAClD,MAAI,eAAe;AACnB,MAAI,eAAe;AAEnB,MAAI,MAAM;AACR,cAAU,KAAK,UAAU;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,MACT,SAAS,sBAAsB,GAAG;AAAA,MAClC;AAAA,IACF,CAAC;AAED,mBAAe;AAAA,EACjB;AAIA,QAAM,uBAAuB,YAAY;AACvC,qBAAiB,SAAS,aAAa;AACrC,YAAM,IAA6B;AAAA,QACjC;AAAA,QACA,SAAS;AAAA,QACT,cAAc;AAAA,MAChB;AAEA,UAAI,cAAc;AAChB,UAAE,cAAc;AAChB,UAAE,gBAAgB;AAClB,UAAE,UAAU,sBAAsB,GAAG;AACrC,UAAE;AACF,uBAAe;AAAA,MACjB;AAEA,gBAAU,KAAK,UAAU,CAAC;AAAA,IAC5B;AAGA,QAAI,CAAC;AAAc;AACnB,cAAU,gBAAgB,UAAU,QAAQ;AAAA,EAC9C;AAEA,OAAK,qBAAqB;AAG1B,WAAS,UAAU,KAA6B;AAC9C,QAAI,IAAI,aAAa;AAAU;AAC/B,QAAI,IAAI,OAAO,UAAU;AAAU;AAEnC,QAAI,cAAc,IAAI,YAAY,GAAG;AACnC,cAAQ;AAAA,IACV,OAAO;AACL,mBAAa,KAAK,IAAI,OAAO;AAAA,IAC/B;AAAA,EACF;AAEA,WAAS,UAAU;AACjB,gBAAY,IAAI;AAChB,iBAAa,IAAI;AACjB,cAAU,oBAAoB,WAAW,SAAS;AAClD,cAAU,oBAAoB,iBAAiB,eAAe;AAC9D,SAAK,IAAI;AAAA,EACX;AAGA,QAAM,kBAAkB,+BAA+B,UAAU,MAAM;AACrE,iBAAa;AAAA,MACX,IAAI;AAAA,QACF,MAAM;AAAA,QACN,SAAS,GAAG,QAAQ;AAAA,MACtB,CAAC;AAAA,IACH;AACA,mBAAe;AACf,YAAQ;AAAA,EACV,CAAC;AAED,YAAU,iBAAiB,WAAW,SAAS;AAC/C,YAAU,iBAAiB,iBAAiB,eAAe;AAC3D,SAAO,CAAC,aAAa,cAAc,OAAO;AAC5C;AAEA,SAAS,gBACP,WACA,UACA,OACA,aACA,eACA;AACA,QAAM,WAAW,OAAO;AACxB,QAAM,EAAE,MAAM,IAAI,IAAI;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,YAAU,KAAK,UAAU;AAAA,IACvB;AAAA,IACA;AAAA,IACA;AAAA,IACA,SAAS;AAAA,IACT,SAAS,sBAAsB,GAAG;AAAA,IAClC;AAAA,EACF,CAAC;AAED,MAAI,eAAe;AAGnB,QAAM,eAAe,SAAS,EAAE,YAAY,KAAK,CAAC;AAClD,WAAS,UAAU,KAA6B;AAC9C,QAAI,IAAI,aAAa;AAAU;AAC/B,QAAI,IAAI,OAAO,UAAU;AAAU;AAEnC,QAAI,cAAc,IAAI,YAAY,GAAG;AACnC,cAAQ;AAAA,IACV,OAAO;AACL,mBAAa,KAAK,IAAI,OAAO;AAAA,IAC/B;AAAA,EACF;AAEA,WAAS,UAAU;AACjB,iBAAa,IAAI;AACjB,cAAU,oBAAoB,WAAW,SAAS;AAClD,cAAU,oBAAoB,iBAAiB,eAAe;AAC9D,SAAK,IAAI;AAAA,EACX;AAEA,QAAM,eAAe,MAAM;AACzB,YAAQ;AACR,QAAI,CAAC;AAAc;AACnB,cAAU,gBAAgB,UAAU,QAAQ;AAAA,EAC9C;AAGA,QAAM,kBAAkB,+BAA+B,UAAU,MAAM;AACrE,iBAAa;AAAA,MACX,IAAI;AAAA,QACF,MAAM;AAAA,QACN,SAAS,GAAG,QAAQ;AAAA,MACtB,CAAC;AAAA,IACH;AACA,mBAAe;AACf,YAAQ;AAAA,EACV,CAAC;AAED,YAAU,iBAAiB,WAAW,SAAS;AAC/C,YAAU,iBAAiB,iBAAiB,eAAe;AAC3D,SAAO,CAAC,cAAc,YAAY;AACpC;AAEA,SAAS,aACP,WACA,UACA,MACA,aACA,eACA;AACA,QAAM,WAAW,OAAO;AACxB,QAAM,EAAE,MAAM,IAAI,IAAI;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,QAAM,cAAc,SAAS,EAAE,YAAY,KAAK,CAAC;AACjD,MAAI,eAAe;AACnB,MAAI,eAAe;AAEnB,MAAI,MAAM;AACR,cAAU,KAAK,UAAU;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,MACT,SAAS,sBAAsB,GAAG;AAAA,MAClC;AAAA,IACF,CAAC;AAED,mBAAe;AAAA,EACjB;AAIA,QAAM,uBAAuB,YAAY;AACvC,qBAAiB,SAAS,aAAa;AACrC,YAAM,IAA6B;AAAA,QACjC;AAAA,QACA,SAAS;AAAA,QACT,cAAc;AAAA,MAChB;AAEA,UAAI,cAAc;AAChB,UAAE,cAAc;AAChB,UAAE,gBAAgB;AAClB,UAAE,UAAU,sBAAsB,GAAG;AACrC,UAAE;AACF,uBAAe;AAAA,MACjB;AAEA,gBAAU,KAAK,UAAU,CAAC;AAAA,IAC5B;AAGA,QAAI,CAAC;AAAc;AACnB,cAAU,gBAAgB,UAAU,QAAQ;AAAA,EAC9C;AAEA,OAAK,qBAAqB;AAE1B,QAAM,kBAAkB,IAAI,QAAQ,CAAC,YAAY;AAG/C,UAAM,kBAAkB,+BAA+B,UAAU,MAAM;AACrE,qBAAe;AACf,cAAQ;AACR;AAAA,QACE,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS,GAAG,QAAQ;AAAA,QACtB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,aAAS,UAAU;AACjB,kBAAY,IAAI;AAChB,gBAAU,oBAAoB,WAAW,SAAS;AAClD,gBAAU,oBAAoB,iBAAiB,eAAe;AAC9D,WAAK,IAAI;AAAA,IACX;AAEA,aAAS,UAAU,KAA6B;AAC9C,UAAI,IAAI,aAAa;AAAU;AAC/B,UAAI,IAAI,OAAO,UAAU;AAAU;AAGnC,cAAQ;AACR,cAAQ,IAAI,OAAO;AAAA,IACrB;AAEA,cAAU,iBAAiB,WAAW,SAAS;AAC/C,cAAU,iBAAiB,iBAAiB,eAAe;AAAA,EAC7D,CAAC;AACD,SAAO,CAAC,aAAa,eAAe;AACtC;;;AEniBA,SAAS,aAAa;AAWtB,SAAe,sBAAsB;AA0BrC,IAAM,cAAN,MAAwD;AAAA,EACtD;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAEA;AAAA;AAAA,EAEA;AAAA,EACA;AAAA,EAEQ;AAAA,EACR,YACE,WACA,UACA,kBACA,iBACA;AACA,UAAM,YAAwC,CAAC;AAE/C,SAAK,WAAW;AAChB,SAAK,aAAa,oBAAI,IAAI;AAE1B,eAAW,CAAC,MAAM,OAAO,KAAK,OAAO,QAAQ,QAAQ,GAAG;AACtD,YAAM,WAAW,QAAQ,YAAY;AACrC,gBAAU,IAAI,IAAI;AAElB,WAAK,WAAW,IAAI,UAAU;AAAA,QAC5B,GAAG;AAAA,QACH,OAAO,SAAS;AAAA,MAClB,CAAC;AAAA,IACH;AAEA,QAAI,kBAAkB;AACpB,gBAAU,gBAAgB,gBAAgB;AAAA,IAC5C;AAEA,SAAK,YAAY;AACjB,SAAK,uBAAuB,oBAAI,IAAI;AACpC,SAAK,YAAY,oBAAI,IAAI;AACzB,SAAK,gBAAgB,oBAAI,IAAI;AAC7B,SAAK,UAAU,iBAAiB,WAAW,KAAK,SAAS;AACzD,SAAK,UAAU,iBAAiB,iBAAiB,KAAK,eAAe;AACrE,SAAK,MAAM,UAAU;AAAA,EACvB;AAAA,EAEA,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAY,OAAO,YAAoC;AACrD,QAAI,QAAQ,OAAO,KAAK,UAAU,UAAU;AAC1C,WAAK,KAAK;AAAA,QACR;AAAA,QACA;AAAA,UACE,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,QACpB;AAAA,MACF;AACA;AAAA,IACF;AAEA,QAAI,aAAa,KAAK,UAAU,IAAI,QAAQ,QAAQ;AACpD,UAAM,gBAAgB,CAAC;AAGvB,mBAAe,KAAK,oBAAoB,OAAO;AAC/C,QAAI,CAAC,YAAY;AAEf;AAAA,IACF;AAEA,UAAM,KAAK,aAAa,YAAY,SAAS,aAAa;AAAA,EAC5D;AAAA;AAAA,EAGA,kBAAkB,OAAO,QAAmC;AAC1D,QAAI,IAAI,WAAW;AAAc;AAEjC,UAAM,uBAAuB,IAAI,QAAQ;AACzC,SAAK,KAAK;AAAA,MACR,+BAA+B,oBAAoB;AAAA,MACnD,IAAI,QAAQ;AAAA,IACd;AAEA,UAAM,wBAAwB,KAAK,cAAc,IAAI,oBAAoB;AACzE,QAAI,CAAC;AAAuB;AAE5B,SAAK,qBAAqB,IAAI,oBAAoB;AAClD,UAAM,QAAQ;AAAA,MACZ,MAAM,KAAK,qBAAqB,EAAE,IAAI,KAAK,aAAa;AAAA,IAC1D;AACA,SAAK,qBAAqB,OAAO,oBAAoB;AACrD,SAAK,cAAc,OAAO,oBAAoB;AAAA,EAChD;AAAA,EAEA,MAAM,QAAQ;AACZ,SAAK,UAAU,oBAAoB,WAAW,KAAK,SAAS;AAC5D,SAAK,UAAU,oBAAoB,iBAAiB,KAAK,eAAe;AACxE,UAAM,QAAQ,IAAI,CAAC,GAAG,KAAK,UAAU,KAAK,CAAC,EAAE,IAAI,KAAK,aAAa,CAAC;AAEpE,eAAW,WAAW,KAAK,WAAW,OAAO,GAAG;AAC9C,UAAI,OAAO,WAAW,QAAQ,OAAO;AACnC,cAAM,UAAU,QAAQ,MAAM,OAAO,OAAO;AAC5C,YAAI,OAAO,YAAY,YAAY;AACjC,kBAAQ;AAAA,QACV;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,oBAAoB,SAAiC;AACnD,QAAI,CAAC,aAAa,QAAQ,YAAY,GAAG;AACvC,WAAK,KAAK;AAAA,QACR;AAAA,QACA;AAAA,UACE,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B;AAAA,MACF;AACA;AAAA,IACF;AAEA,QAAI,CAAC,QAAQ,iBAAiB,CAAC,QAAQ,aAAa;AAClD,WAAK,KAAK;AAAA,QACR;AAAA,QACA;AAAA,UACE,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,QACpB;AAAA,MACF;AACA;AAAA,IACF;AAEA,QAAI,EAAE,QAAQ,eAAe,KAAK,WAAW;AAC3C,WAAK,KAAK,KAAK,yBAAyB,QAAQ,WAAW,IAAI;AAAA,QAC7D,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,MACpB,CAAC;AACD;AAAA,IACF;AAEA,UAAM,UAAU,KAAK,SAAS,QAAQ,WAAW;AACjD,UAAM,iBAAiB,KAAK,WAAW,SAAS,QAAQ,WAAW;AACnE,QAAI,EAAE,QAAQ,iBAAiB,QAAQ,aAAa;AAClD,WAAK,KAAK;AAAA,QACR,0CAA0C,QAAQ,WAAW,IAAI,QAAQ,aAAa;AAAA,QACtF;AAAA,UACE,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,QACpB;AAAA,MACF;AACA;AAAA,IACF;AAEA,UAAM,UAAU,KAAK,UAAU,SAAS,IAAI,QAAQ,IAAI;AACxD,QAAI,CAAC,SAAS;AACZ,WAAK,KAAK,KAAK,6BAA6B,QAAQ,IAAI,IAAI;AAAA,QAC1D,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,MACpB,CAAC;AACD;AAAA,IACF;AAEA,UAAM,YAAY,QAAQ,WAAW,QAAQ,aAAa;AAC1D,UAAM,WAAmC,SAAS,EAAE,YAAY,KAAK,CAAC;AACtE,UAAM,WAAmC,SAAS,EAAE,YAAY,KAAK,CAAC;AACtE,UAAM,aACJ,UAAU,SAAS,kBAAkB,UAAU,SAAS;AAC1D,UAAM,cAAiC,CAAC;AAExC,UAAM;AAAA;AAAA,MAEJ;AAAA;AAAA,SAEK,YAAY;AACX,2BAAiB,YAAY,UAAU;AACrC,iBAAK,UAAU,KAAK,QAAQ,IAAI;AAAA,cAC9B,UAAU,QAAQ;AAAA,cAClB,cAAc;AAAA,cACd,SAAS;AAAA,YACX,CAAC;AAAA,UACH;AAIA,cAAI,CAAC,KAAK,qBAAqB,IAAI,QAAQ,IAAI,GAAG;AAChD,iBAAK,UAAU,gBAAgB,QAAQ,IAAI,QAAQ,QAAQ;AAAA,UAC7D;AAGA,sBAAY,QAAQ,CAAC,MAAM,EAAE,CAAC;AAAA,QAChC,GAAG;AAAA;AAAA;AAAA,SAEF,YAAY;AACX,gBAAM,WAAW,MAAM,SAAS,KAAK,EAAE,KAAK,CAAC,QAAQ,IAAI,KAAK;AAC9D,cAAI,UAAU;AACZ,iBAAK,UAAU,KAAK,QAAQ,IAAI;AAAA,cAC9B,UAAU,QAAQ;AAAA,cAClB;AAAA,cACA,SAAS;AAAA,YACX,CAAC;AAGD,wBAAY,QAAQ,CAAC,MAAM,EAAE,CAAC;AAAA,UAChC;AAAA,QACF,GAAG;AAAA;AAAA;AAET,UAAM,eAAe,CAAC,KAAc,SAAe;AACjD,YAAM,WAAW,kBAAkB,GAAG;AACtC,WAAK,KAAK;AAAA,QACR,aAAa,QAAQ,WAAW,IAAI,QAAQ,aAAa,6BAA6B,QAAQ;AAAA,QAC9F,QAAQ;AAAA,MACV;AAEA,WAAK,gBAAgB,eAAe,QAAQ,MAAM,IAAI,MAAM,QAAQ,CAAC;AACrE,WAAK,UAAU,EAAE,MAAM,eAAe,MAAM,CAAC;AAC7C,eAAS;AAAA,QACP,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAA8C;AAAA,MAChD;AAAA,IACF;AAEA,UAAM,cAAc,KAAK,UAAU,yBAAyB,IAAI,OAAO;AACvE,QAAI,CAAC,aAAa;AAChB,WAAK,KAAK,MAAM,2CAA2C;AAAA,QACzD,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAGA,QAAI;AACJ,UAAM,qBAAqB,UAAU;AACrC,UAAM,kCACJ;AAAA,MACE,GAAG;AAAA,MACH,IAAI,QAAQ;AAAA,MACZ,MAAM,QAAQ;AAAA,MACd,UAAU,QAAQ;AAAA,MAClB;AAAA,MACA,UAAU;AAAA,IACZ;AAEF,YAAQ,UAAU,MAAM;AAAA,MACtB,KAAK;AACH,uBAAe;AAAA,UACb,UAAU;AAAA,UACV;AAAA,UACA,OAAO,SAAS;AACd,kBAAM,eAAe,MAAM,SAAS,KAAK;AACzC,gBAAI,aAAa,MAAM;AACrB;AAAA,YACF;AACA,gBAAI;AACF,oBAAM,gBAAgB,MAAM,UAAU;AAAA,gBACpC;AAAA,gBACA,aAAa;AAAA,cACf;AACA,uBAAS,KAAK,aAAa;AAAA,YAC7B,SAAS,KAAK;AACZ,2BAAa,KAAK,IAAI;AAAA,YACxB,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF,KAAK;AACH,YAAI,oBAAoB;AACtB,yBAAe;AAAA,YACb,UAAU;AAAA,YACV;AAAA,YACA,OAAO,SAAS;AACd,oBAAM,cAAc,MAAM,SAAS,KAAK;AACxC,kBAAI,YAAY,MAAM;AACpB;AAAA,cACF;AAEA,kBAAI;AACF,sBAAM,UAAU,MAAM,UAAU;AAAA,kBAC9B;AAAA,kBACA,YAAY;AAAA,kBACZ;AAAA,kBACA;AAAA,gBACF;AAEA,oBAAI,SAAS;AACX,8BAAY,KAAK,OAAO;AAAA,gBAC1B;AAAA,cACF,SAAS,KAAK;AACZ,6BAAa,KAAK,IAAI;AAAA,cACxB,UAAE;AACA,qBAAK,IAAI;AAAA,cACX;AAAA,YACF;AAAA,UACF;AAAA,QACF,OAAO;AACL,yBAAe;AAAA,YACb,UAAU;AAAA,YACV;AAAA,YACA,OAAO,SAAS;AACd,kBAAI;AACF,sBAAM,UAAU,MAAM,UAAU;AAAA,kBAC9B;AAAA,kBACA;AAAA,kBACA;AAAA,gBACF;AACA,oBAAI,SAAS;AACX,8BAAY,KAAK,OAAO;AAAA,gBAC1B;AAAA,cACF,SAAS,KAAK;AACZ,6BAAa,KAAK,IAAI;AAAA,cACxB,UAAE;AACA,qBAAK,IAAI;AAAA,cACX;AAAA,YACF;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF,KAAK;AACH,uBAAe;AAAA,UACb,UAAU;AAAA,UACV;AAAA,UACA,OAAO,SAAS;AACd,kBAAM,eAAe,MAAM,SAAS,KAAK;AACzC,gBAAI,aAAa,MAAM;AACrB;AAAA,YACF;AAEA,gBAAI;AACF,oBAAM,UAAU,MAAM,UAAU;AAAA,gBAC9B;AAAA,gBACA,aAAa;AAAA,gBACb;AAAA,cACF;AAEA,kBAAI,SAAS;AACX,4BAAY,KAAK,OAAO;AAAA,cAC1B;AAAA,YACF,SAAS,KAAK;AACZ,2BAAa,KAAK,IAAI;AAAA,YACxB,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF,KAAK;AACH,YAAI,oBAAoB;AACtB,yBAAe;AAAA,YACb,UAAU;AAAA,YACV;AAAA,YACA,OAAO,SAAS;AACd,oBAAM,cAAc,MAAM,SAAS,KAAK;AACxC,kBAAI,YAAY,MAAM;AACpB;AAAA,cACF;AACA,kBAAI;AACF,sBAAM,gBAAgB,MAAM,UAAU;AAAA,kBACpC;AAAA,kBACA,YAAY;AAAA,kBACZ;AAAA,gBACF;AAEA,oBAAI,CAAC,KAAK,qBAAqB,IAAI,QAAQ,IAAI,GAAG;AAChD,2BAAS,KAAK,aAAa;AAAA,gBAC7B;AAAA,cACF,SAAS,KAAK;AACZ,6BAAa,KAAK,IAAI;AAAA,cACxB,UAAE;AACA,qBAAK,IAAI;AAAA,cACX;AAAA,YACF;AAAA,UACF;AAAA,QACF,OAAO;AACL,yBAAe;AAAA,YACb,UAAU;AAAA,YACV;AAAA,YACA,OAAO,SAAS;AACd,kBAAI;AACF,sBAAM,gBAAgB,MAAM,UAAU;AAAA,kBACpC;AAAA,kBACA;AAAA,gBACF;AAEA,oBAAI,CAAC,KAAK,qBAAqB,IAAI,QAAQ,IAAI,GAAG;AAChD,2BAAS,KAAK,aAAa;AAAA,gBAC7B;AAAA,cACF,SAAS,KAAK;AACZ,6BAAa,KAAK,IAAI;AAAA,cACxB,UAAE;AACA,qBAAK,IAAI;AAAA,cACX;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAEA;AAAA,MACF;AAGE,aAAK,KAAK;AAAA,UACR,0CACG,UAA2B,IAC9B,OAAO,QAAQ,WAAW,IAAI,QAAQ,aAAa;AAAA,UACnD,EAAE,GAAG,QAAQ,iBAAiB,kBAAkB,QAAQ;AAAA,QAC1D;AACA;AAAA,IACJ;AAEA,UAAM,aAAyB;AAAA,MAC7B,IAAI,QAAQ;AAAA,MACZ;AAAA,MACA;AAAA,MACA,aAAa,QAAQ;AAAA,MACrB,eAAe,QAAQ;AAAA,MACvB,UAAU,EAAE,cAAc,cAAc;AAAA,IAC1C;AAEA,SAAK,UAAU,IAAI,QAAQ,UAAU,UAAU;AAG/C,UAAM,wBACJ,KAAK,cAAc,IAAI,QAAQ,IAAI,KAAK,oBAAI,IAAI;AAClD,0BAAsB,IAAI,QAAQ,QAAQ;AAC1C,SAAK,cAAc,IAAI,QAAQ,MAAM,qBAAqB;AAE1D,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,aACJ,YACA,SACA,QACA;AACA,UAAM,EAAE,aAAa,cAAc,IAAI;AACvC,UAAM,YAAY,KAAK,SAAS,WAAW,EAAE,WAAW,aAAa;AACrE,UAAM,qBAAqB,UAAU;AAErC,QAAI,UAAU,oBAAoB;AAChC,UAAI,MAAM,MAAM,UAAU,MAAM,QAAQ,OAAO,GAAG;AAChD,mBAAW,SAAS,KAAK,QAAQ,OAAsB;AAAA,MACzD,OAAO;AACL,aAAK,KAAK;AAAA,UACR,aAAa,WAAW,IAAI,aAAa;AAAA,UACzC;AAAA,YACE,UAAU,KAAK,UAAU;AAAA,YACzB,kBAAkB;AAAA,YAClB,kBAAkB;AAAA,cAChB,GAAG,MAAM,OAAO,UAAU,MAAM,QAAQ,OAAO;AAAA,YACjD;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,WAAW,MAAM,MAAM,UAAU,OAAO,QAAQ,OAAO,GAAG;AACxD,iBAAW,SAAS,KAAK,QAAQ,OAAsB;AAAA,IACzD,WAAW,CAAC,MAAM,MAAM,6BAA6B,QAAQ,OAAO,GAAG;AAGrE,WAAK,KAAK;AAAA,QACR,aAAa,WAAW,IAAI,aAAa;AAAA,QACzC;AAAA,UACE,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,UAClB,kBAAkB,CAAC,GAAG,MAAM,OAAO,UAAU,OAAO,QAAQ,OAAO,CAAC;AAAA,QACtE;AAAA,MACF;AAAA,IACF;AAEA,QAAI,cAAc,QAAQ,YAAY,GAAG;AACvC,YAAM,KAAK,cAAc,QAAQ,QAAQ;AAEzC,YAAM,wBAAwB,KAAK,cAAc,IAAI,QAAQ,IAAI;AACjE,UAAI,uBAAuB;AACzB,8BAAsB,OAAO,QAAQ,QAAQ;AAC7C,YAAI,sBAAsB,SAAS,GAAG;AACpC,eAAK,cAAc,OAAO,QAAQ,IAAI;AAAA,QACxC;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,WAAW,SAAqB,aAAqB;AAC3D,UAAM,UAAU,KAAK,WAAW,IAAI,OAAO;AAC3C,QAAI,CAAC,SAAS;AACZ,YAAM,MAAM,wBAAwB,WAAW;AAC/C,WAAK,KAAK,MAAM,KAAK;AAAA,QACnB,UAAU,KAAK,UAAU;AAAA,QACzB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,gBAAgB,OAAO,OAAe;AACpC,UAAMC,UAAS,KAAK,UAAU,IAAI,EAAE;AACpC,QAAI,CAACA,SAAQ;AACX;AAAA,IACF;AAGA,IAAAA,QAAO,SAAS,IAAI;AACpB,UAAMA,QAAO,SAAS;AACtB,IAAAA,QAAO,SAAS,IAAI;AACpB,UAAMA,QAAO,SAAS;AACtB,SAAK,UAAU,OAAO,EAAE;AAAA,EAC1B;AACF;AAWO,SAAS,aACd,WACA,UACA,uBAIkB;AAClB,SAAO,IAAI;AAAA,IACT;AAAA,IACA;AAAA,IACA,uBAAuB;AAAA,IACvB,uBAAuB;AAAA,EACzB;AACF;;;ACriBO,SAAS,6BAGd,QACA,WACwB;AACxB,SAAO,EAAE,QAAQ,UAAU;AAC7B;AAEO,SAAS,6BAGd,QACA,UACwB;AACxB,SAAO,EAAE,QAAQ,SAAiD;AACpE;","names":["Type","pushable","options","_pushable","Type","stream"]}
|
|
File without changes
|
|
File without changes
|