@replit/river 0.200.0-rc.6 → 0.200.0-rc.7

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.
Files changed (68) hide show
  1. package/README.md +2 -2
  2. package/dist/{chunk-TMNVRR5R.js → chunk-3XKJOFZA.js} +193 -90
  3. package/dist/chunk-3XKJOFZA.js.map +1 -0
  4. package/dist/{chunk-IGNFE3QW.js → chunk-BZBEE2VR.js} +2 -2
  5. package/dist/chunk-BZBEE2VR.js.map +1 -0
  6. package/dist/{chunk-WE2PAPNC.js → chunk-CXPZSAU4.js} +102 -95
  7. package/dist/chunk-CXPZSAU4.js.map +1 -0
  8. package/dist/{chunk-WF3AW2CB.js → chunk-F2E7ILHW.js} +41 -45
  9. package/dist/chunk-F2E7ILHW.js.map +1 -0
  10. package/dist/{chunk-NOLXEIRP.js → chunk-U4W75CMT.js} +3 -3
  11. package/dist/chunk-U4W75CMT.js.map +1 -0
  12. package/dist/{chunk-GXRIBO3N.js → chunk-VLW5OKZG.js} +2 -2
  13. package/dist/{chunk-MXU6DJVG.js → chunk-XIJVDPYY.js} +161 -159
  14. package/dist/chunk-XIJVDPYY.js.map +1 -0
  15. package/dist/{client-f56a6da3.d.ts → client-829bf1f9.d.ts} +3 -1
  16. package/dist/{connection-6031a354.d.ts → connection-5e67d641.d.ts} +1 -1
  17. package/dist/{context-73df8978.d.ts → context-9eabf54f.d.ts} +132 -81
  18. package/dist/router/index.cjs +162 -160
  19. package/dist/router/index.cjs.map +1 -1
  20. package/dist/router/index.d.cts +7 -7
  21. package/dist/router/index.d.ts +7 -7
  22. package/dist/router/index.js +6 -6
  23. package/dist/{server-9f31d98f.d.ts → server-d82a2d9b.d.ts} +8 -3
  24. package/dist/{services-69d72cd5.d.ts → services-6a446f04.d.ts} +157 -139
  25. package/dist/transport/impls/ws/client.cjs +285 -178
  26. package/dist/transport/impls/ws/client.cjs.map +1 -1
  27. package/dist/transport/impls/ws/client.d.cts +3 -3
  28. package/dist/transport/impls/ws/client.d.ts +3 -3
  29. package/dist/transport/impls/ws/client.js +7 -7
  30. package/dist/transport/impls/ws/server.cjs +224 -128
  31. package/dist/transport/impls/ws/server.cjs.map +1 -1
  32. package/dist/transport/impls/ws/server.d.cts +3 -3
  33. package/dist/transport/impls/ws/server.d.ts +3 -3
  34. package/dist/transport/impls/ws/server.js +5 -5
  35. package/dist/transport/index.cjs +322 -218
  36. package/dist/transport/index.cjs.map +1 -1
  37. package/dist/transport/index.d.cts +3 -3
  38. package/dist/transport/index.d.ts +3 -3
  39. package/dist/transport/index.js +7 -7
  40. package/dist/util/testHelpers.cjs +227 -133
  41. package/dist/util/testHelpers.cjs.map +1 -1
  42. package/dist/util/testHelpers.d.cts +15 -12
  43. package/dist/util/testHelpers.d.ts +15 -12
  44. package/dist/util/testHelpers.js +41 -43
  45. package/dist/util/testHelpers.js.map +1 -1
  46. package/package.json +1 -1
  47. package/dist/chunk-IGNFE3QW.js.map +0 -1
  48. package/dist/chunk-MU25KVV7.js +0 -112
  49. package/dist/chunk-MU25KVV7.js.map +0 -1
  50. package/dist/chunk-MXU6DJVG.js.map +0 -1
  51. package/dist/chunk-NOLXEIRP.js.map +0 -1
  52. package/dist/chunk-TMNVRR5R.js.map +0 -1
  53. package/dist/chunk-WE2PAPNC.js.map +0 -1
  54. package/dist/chunk-WF3AW2CB.js.map +0 -1
  55. package/dist/connection-11991b13.d.ts +0 -31
  56. package/dist/transport/impls/uds/client.cjs +0 -1696
  57. package/dist/transport/impls/uds/client.cjs.map +0 -1
  58. package/dist/transport/impls/uds/client.d.cts +0 -18
  59. package/dist/transport/impls/uds/client.d.ts +0 -18
  60. package/dist/transport/impls/uds/client.js +0 -38
  61. package/dist/transport/impls/uds/client.js.map +0 -1
  62. package/dist/transport/impls/uds/server.cjs +0 -1663
  63. package/dist/transport/impls/uds/server.cjs.map +0 -1
  64. package/dist/transport/impls/uds/server.d.cts +0 -19
  65. package/dist/transport/impls/uds/server.d.ts +0 -19
  66. package/dist/transport/impls/uds/server.js +0 -33
  67. package/dist/transport/impls/uds/server.js.map +0 -1
  68. /package/dist/{chunk-GXRIBO3N.js.map → chunk-VLW5OKZG.js.map} +0 -0
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../router/services.ts","../router/result.ts","../router/streams.ts","../router/procedures.ts","../router/client.ts","../router/server.ts","../router/handshake.ts"],"sourcesContent":["import { Type, TSchema, Static } from '@sinclair/typebox';\nimport {\n Branded,\n ProcedureMap,\n Unbranded,\n AnyProcedure,\n PayloadType,\n ProcedureErrorSchemaType,\n OutputReaderErrorSchema,\n} from './procedures';\nimport { ServiceContext } from './context';\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 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> = Static<S['procedures'][ProcName]['init']>;\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] extends { input: PayloadType }\n ? Static<S['procedures'][ProcName]['input']>\n : never;\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> = Static<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> =\n | Static<S['procedures'][ProcName]['errors']>\n | Static<typeof OutputReaderErrorSchema>;\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: (extendedContext: ServiceContext) => State;\n}\n\n// TODO remove once clients migrate to v2\nexport interface SerializedProcedureSchemaProtocolv1 {\n init?: PayloadType;\n input: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n type: 'rpc' | 'subscription' | 'upload' | 'stream';\n}\n\n// TODO remove once clients migrate to v2\nexport interface SerializedServiceSchemaProtocolv1 {\n procedures: Record<string, SerializedProcedureSchemaProtocolv1>;\n}\n\n// TODO remove once clients migrate to v2\nexport interface SerializedServerSchemaProtocolv1 {\n handshakeSchema?: TSchema;\n services: Record<string, SerializedServiceSchemaProtocolv1>;\n}\n\n// TODO remove once clients migrate to v2\n/**\n * Same as {@link serializeSchema} but with a format that is compatible with\n * protocolv1. This is useful to be able to continue to generate schemas for older\n * clients as they are still supported.\n */\nexport function serializeSchemaBackwardsCompatible(\n services: AnyServiceSchemaMap,\n handshakeSchema?: TSchema,\n): SerializedServerSchemaProtocolv1 {\n const serializedServiceObject = Object.entries(services).reduce<\n Record<string, SerializedServiceSchemaProtocolv1>\n >((acc, [name, value]) => {\n acc[name] = value.serializeBackwardsCompatible();\n return acc;\n }, {});\n\n const schema: SerializedServerSchemaProtocolv1 = {\n services: serializedServiceObject,\n };\n\n if (handshakeSchema) {\n schema.handshakeSchema = Type.Strict(handshakeSchema);\n }\n\n return schema;\n}\n\nexport interface SerializedProcedureSchema {\n init: PayloadType;\n input?: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n type: 'rpc' | 'subscription' | 'upload' | 'stream';\n}\n\nexport interface SerializedServiceSchema {\n procedures: Record<string, SerializedProcedureSchema>;\n}\n\nexport interface SerializedServerSchema {\n handshakeSchema?: TSchema;\n services: Record<string, SerializedServiceSchema>;\n}\n\n/**\n * Serializes a server schema into a plain object that is JSON compatible.\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: (\n extendedContext: ServiceContext,\n ) => 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 * init: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, init) {\n * ctx.state.count += init.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 * init: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, init) {\n * ctx.state.count += init.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 * init: Type.Object({ amount: Type.Number() }),\n * output: Type.Object({ current: Type.Number() }),\n * async handler(ctx, init) {\n * ctx.state.count += init.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 * init: Type.Object({ a: Type.Number(), b: Type.Number() }),\n * output: Type.Object({ result: Type.Number() }),\n * async handler(ctx, init) {\n * return Ok({ result: init.a + init.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 init: Type.Strict(procDef.init),\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 `input` field if the type declares it.\n ...('input' in procDef\n ? {\n input: Type.Strict(procDef.input),\n }\n : {}),\n },\n ]),\n ),\n };\n }\n\n // TODO remove once clients migrate to v2\n /**\n * Same as {@link ServiceSchema.serialize}, but with a format that is compatible with\n * protocol v1. This is useful to be able to continue to generate schemas for older\n * clients as they are still supported.\n */\n serializeBackwardsCompatible(): SerializedServiceSchemaProtocolv1 {\n return {\n procedures: Object.fromEntries(\n Object.entries(this.procedures).map(\n ([procName, procDef]): [\n string,\n SerializedProcedureSchemaProtocolv1,\n ] => {\n if (procDef.type === 'rpc' || procDef.type === 'subscription') {\n return [\n procName,\n {\n // BACKWARDS COMPAT: map init to input for protocolv1\n // this is the only change needed to make it compatible.\n input: Type.Strict(procDef.init),\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 },\n ];\n }\n\n // No backwards compatibility needed for upload and stream types, as having an `init`\n // all the time is compatible with protocol v1.\n return [\n procName,\n {\n init: Type.Strict(procDef.init),\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 input: Type.Strict(procDef.input),\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(extendedContext: ServiceContext): Service<State, Procedures> {\n return Object.freeze({\n state: this.initializeState(extendedContext),\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 {\n Static,\n TLiteral,\n TObject,\n TSchema,\n TString,\n TUnion,\n Type,\n} from '@sinclair/typebox';\nimport { Client } from './client';\nimport { ReadStream } from './streams';\n\ntype TLiteralString = TLiteral<string>;\n\nexport type BaseErrorSchemaType =\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\n/**\n * Takes in a specific error schema and returns a result schema the error\n */\nexport const ErrResultSchema = <T extends BaseErrorSchemaType>(t: T) =>\n Type.Object({\n ok: Type.Literal(false),\n payload: t,\n });\n\n/**\n * AnyResultSchema is a schema to validate any result.\n */\nexport const AnyResultSchema = Type.Union([\n Type.Object({\n ok: Type.Literal(false),\n payload: Type.Object({\n code: Type.String(),\n message: Type.String(),\n extras: Type.Optional(Type.Unknown()),\n }),\n }),\n\n Type.Object({\n ok: Type.Literal(true),\n payload: Type.Unknown(),\n }),\n]);\n\nexport interface OkResult<T> {\n ok: true;\n payload: T;\n}\nexport interface ErrResult<Err extends Static<BaseErrorSchemaType>> {\n ok: false;\n payload: Err;\n}\nexport type Result<T, Err extends Static<BaseErrorSchemaType>> =\n | OkResult<T>\n | ErrResult<Err>;\n\nexport function Ok<const T extends Array<unknown>>(p: T): OkResult<T>;\nexport function Ok<const T extends ReadonlyArray<unknown>>(p: T): OkResult<T>;\nexport function Ok<const T>(payload: T): OkResult<T>;\nexport function Ok<const T>(payload: T): OkResult<T> {\n return {\n ok: true,\n payload,\n };\n}\n\nexport function Err<const Err extends Static<BaseErrorSchemaType>>(\n error: Err,\n): ErrResult<Err> {\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 * Unwrap a {@link Result} type and return the payload if successful,\n * otherwise throws an error.\n * @param result - The result to unwrap.\n * @throws Will throw an error if the result is not ok.\n */\nexport function unwrap<T, Err extends Static<BaseErrorSchemaType>>(\n result: Result<T, Err>,\n): T {\n if (result.ok) {\n return result.payload;\n }\n\n throw new Error(\n `Cannot non-ok result, got: ${result.payload.code} - ${result.payload.message}`,\n );\n}\n\n/**\n * Refine a {@link Result} type to its error payload.\n */\nexport type ResultUnwrapErr<R> = R extends Result<infer __T, infer Err>\n ? Err\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 ? ReturnType<UploadHandler> extends [\n infer __UploadInputMessage,\n (...args: never) => Promise<infer UploadOutputMessage>,\n ]\n ? UploadOutputMessage\n : never\n : Procedure extends object & { stream: infer StreamHandler extends Fn }\n ? ReturnType<StreamHandler> extends [\n infer __StreamInputMessage,\n ReadStream<infer StreamOutputMessage, Static<BaseErrorSchemaType>>,\n ]\n ? StreamOutputMessage\n : never\n : Procedure extends object & {\n subscribe: infer SubscriptionHandler extends Fn;\n }\n ? Awaited<ReturnType<SubscriptionHandler>> extends ReadStream<\n infer SubscriptionOutputMessage,\n Static<BaseErrorSchemaType>\n >\n ? SubscriptionOutputMessage\n : never\n : never\n : never\n : never;\n","import { Static } from '@sinclair/typebox';\nimport { BaseErrorSchemaType, Err, Result } from './result';\n\nexport const StreamDrainedError = {\n code: 'STREAM_DRAINED',\n message: 'Stream was drained',\n} as const;\n\ntype ReadStreamResult<T, E extends Static<BaseErrorSchemaType>> = Result<\n T,\n E | typeof StreamDrainedError\n>;\n\n/**\n * A `ReadStream` represents a stream of data.\n *\n * This stream is not closable by the reader, the reader must wait for\n * the writer to close it.\n *\n * The stream can only be locked (aka consumed) once and will remain\n * locked, trying to lock the stream again will throw an TypeError.\n */\nexport interface ReadStream<T, E extends Static<BaseErrorSchemaType>> {\n /**\n * Stream implements AsyncIterator API and can be consumed via\n * for-await-of loops.\n *\n */\n [Symbol.asyncIterator](): {\n next(): Promise<\n | {\n done: false;\n value: ReadStreamResult<T, E>;\n }\n | {\n done: true;\n value: undefined;\n }\n >;\n };\n /**\n * `asArray` locks the stream and returns a promise that resolves\n * with an array of the stream's content when the stream is closed.\n *\n */\n asArray(): Promise<Array<ReadStreamResult<T, E>>>;\n /**\n * `drain` locks the stream and discards any existing or future data.\n *\n * If there is an existing Promise waiting for the next value,\n * `drain` causes it to resolve with a {@link StreamDrainedError} error.\n */\n drain(): undefined;\n /**\n * `isLocked` returns true if the stream is locked.\n */\n isLocked(): boolean;\n /**\n * `onClose` registers a callback that will be called when the stream\n * is closed. Returns a function that can be used to unregister the\n * listener.\n */\n onClose(cb: () => void): () => void;\n /**\n * `isClosed` returns true if the stream was closed by the writer.\n *\n * Note that the stream can still have queued data and can still be\n * consumed.\n */\n isClosed(): boolean;\n /**\n * `requestClose` sends a request to the writer to close the stream,\n * and resolves the writer closes its end of the stream..\n *\n * The stream can still receive more data after the request is sent. If you\n * no longer wish to use the stream, make sure to call `drain`.\n */\n requestClose(): Promise<undefined>;\n /**\n * `isCloseRequested` checks if the reader has requested to close the stream.\n */\n isCloseRequested(): boolean;\n}\n\n/**\n * A `WriteStream` is a streams that can be written to.\n */\nexport interface WriteStream<T> {\n /**\n * `write` writes a value to the stream. An error is thrown if writing to a closed stream.\n */\n write(value: T): undefined;\n /**\n * `close` signals the closure of the write stream, informing the reader that the stream is complete.\n * Calling close multiple times has no effect.\n */\n close(): undefined;\n /**\n * `isCloseRequested` checks if the reader has requested to close the stream.\n */\n isCloseRequested(): boolean;\n /**\n * `onCloseRequest` registers a callback that will be called when the stream's\n * reader requests a close. Returns a function that can be used to unregister the\n * listener.\n */\n onCloseRequest(cb: () => void): () => void;\n /**\n * `isClosed` returns true if the stream was closed by the writer.\n */\n isClosed(): boolean;\n /**\n * `onClose` registers a callback that will be called when the stream\n * is closed. Returns a function that can be used to unregister the\n * listener.\n */\n onClose(cb: () => void): () => void;\n}\n\n/**\n * Internal implementation of a `ReadStream`.\n * This won't be exposed as an interface to river\n * consumers directly, it has internal river methods\n * to pushed data to the stream and close it.\n */\nexport class ReadStreamImpl<T, E extends Static<BaseErrorSchemaType>>\n implements ReadStream<T, E>\n{\n /**\n * Whether the stream is closed.\n */\n private closed = false;\n /**\n * A list of listeners that will be called when the stream is closed.\n */\n private onCloseListeners: Set<() => void>;\n /**\n * Whether the user has requested to close the stream.\n */\n private closeRequested = false;\n /**\n * Used to signal to the outside world that the user has requested to close the stream.\n */\n private closeRequestCallback: () => void;\n /**\n * Whether the stream is locked.\n */\n private locked = false;\n /**\n * Whether drain was called.\n */\n private drained = false;\n /**\n * This flag allows us to avoid cases where drain was called,\n * but the stream is fully consumed and closed. We don't need\n * to signal that drain was closed.\n */\n private didDrainDisposeValues = false;\n /**\n * A list of values that have been pushed to the stream but not yet emitted to the user.\n */\n private queue: Array<ReadStreamResult<T, E>> = [];\n /**\n * Used by methods in the class to signal to the iterator that it\n * should check for the next value.\n */\n private nextPromise: Promise<void> | null = null;\n /**\n * Resolves nextPromise\n */\n private resolveNextPromise: null | (() => void) = null;\n\n constructor(closeRequestCallback: () => void) {\n this.closeRequestCallback = closeRequestCallback;\n this.onCloseListeners = new Set();\n }\n\n public [Symbol.asyncIterator]() {\n if (this.isLocked()) {\n throw new TypeError('ReadStream is already locked');\n }\n\n // first iteration with drain signals an error, the following one signals end of iteration.\n let didSignalDrain = false;\n this.locked = true;\n\n return {\n next: async () => {\n if (this.drained && didSignalDrain) {\n return {\n done: true,\n value: undefined,\n } as const;\n }\n\n // Wait until we have something in the queue\n while (this.queue.length === 0) {\n if (this.isClosed() && !this.didDrainDisposeValues) {\n return {\n done: true,\n value: undefined,\n } as const;\n }\n\n if (this.drained) {\n didSignalDrain = true;\n\n return {\n done: false,\n value: Err(StreamDrainedError),\n } as const;\n }\n\n if (!this.nextPromise) {\n this.nextPromise = new Promise<void>((resolve) => {\n this.resolveNextPromise = resolve;\n });\n }\n\n await this.nextPromise;\n this.nextPromise = null;\n this.resolveNextPromise = null;\n }\n\n // Unfortunately we have to use non-null assertion here, because T can be undefined\n // we already check for array length above anyway\n // eslint-disable-next-line @typescript-eslint/no-non-null-assertion\n const value = this.queue.shift()!;\n\n return { done: false, value } as const;\n },\n return: async () => {\n this.drain();\n return { done: true, value: undefined } as const;\n },\n };\n }\n\n public async asArray(): Promise<Array<ReadStreamResult<T, E>>> {\n const array: Array<ReadStreamResult<T, E>> = [];\n for await (const value of this) {\n array.push(value);\n }\n\n return array;\n }\n\n public drain(): undefined {\n if (this.drained) {\n return;\n }\n\n this.locked = true;\n this.drained = true;\n this.didDrainDisposeValues = this.queue.length > 0;\n this.queue.length = 0;\n\n this.resolveNextPromise?.();\n }\n\n public isClosed(): boolean {\n return this.closed;\n }\n\n public isLocked(): boolean {\n return this.locked;\n }\n\n public onClose(cb: () => void): () => void {\n if (this.isClosed()) {\n throw new Error('Stream is already closed');\n }\n\n this.onCloseListeners.add(cb);\n\n return () => {\n this.onCloseListeners.delete(cb);\n };\n }\n\n public requestClose(): Promise<undefined> {\n if (this.isClosed()) {\n throw new Error('Cannot request close after stream already closed');\n }\n\n if (!this.closeRequested) {\n this.closeRequested = true;\n this.closeRequestCallback();\n }\n\n return new Promise<undefined>((resolve) => {\n this.onClose(() => {\n resolve(undefined);\n });\n });\n }\n\n public isCloseRequested(): boolean {\n return this.closeRequested;\n }\n\n /**\n * @internal meant for use within river, not exposed as a public API\n *\n * Pushes a value to the stream.\n */\n public pushValue(value: Result<T, E>): undefined {\n if (this.drained) {\n return;\n }\n\n if (this.closed) {\n throw new Error('Cannot push to closed stream');\n }\n\n this.queue.push(value);\n this.resolveNextPromise?.();\n }\n\n /**\n * @internal meant for use within river, not exposed as a public API\n *\n * Triggers the close of the stream. Make sure to push all remaining\n * values before calling this method.\n */\n public triggerClose(): undefined {\n if (this.isClosed()) {\n throw new Error('Unexpected closing multiple times');\n }\n\n this.closed = true;\n this.resolveNextPromise?.();\n this.onCloseListeners.forEach((cb) => cb());\n this.onCloseListeners.clear();\n\n // TODO maybe log a warn if after a certain amount of time\n // the queue was not fully consumed\n }\n\n /**\n * @internal meant for use within river, not exposed as a public API\n */\n public hasValuesInQueue(): boolean {\n return this.queue.length > 0;\n }\n}\n\n/**\n * Internal implementation of a `WriteStream`.\n * This won't be exposed as an interface to river\n * consumers directly, it has internal river methods\n * to trigger a close request, a way to pass on close\n * signals, and a way to push data to the stream.\n */\nexport class WriteStreamImpl<T> implements WriteStream<T> {\n /**\n * Passed via constructor to pass on write requests\n */\n private writeCb: (value: T) => void;\n /**\n * Whether the stream is closed.\n */\n private closed = false;\n /**\n * A list of listeners that will be called when the stream is closed.\n */\n private onCloseListeners: Set<() => void>;\n /**\n * Whether the reader has requested to close the stream.\n */\n private closeRequested = false;\n /**\n * A list of listeners that will be called when a close request is triggered.\n */\n private onCloseRequestListeners: Set<() => void>;\n\n constructor(writeCb: (value: T) => void) {\n this.writeCb = writeCb;\n this.onCloseListeners = new Set();\n this.onCloseRequestListeners = new Set();\n }\n\n public write(value: T): undefined {\n if (this.isClosed()) {\n throw new Error('Cannot write to closed stream');\n }\n\n this.writeCb(value);\n }\n\n public isClosed(): boolean {\n return this.closed;\n }\n\n onClose(cb: () => void): () => void {\n if (this.isClosed()) {\n cb();\n\n return () => undefined;\n }\n\n this.onCloseListeners.add(cb);\n\n return () => this.onCloseListeners.delete(cb);\n }\n\n public close(): undefined {\n if (this.isClosed()) {\n return;\n }\n\n this.closed = true;\n this.onCloseListeners.forEach((cb) => cb());\n\n // cleanup\n this.onCloseListeners.clear();\n this.onCloseRequestListeners.clear();\n this.writeCb = () => undefined;\n }\n\n public isCloseRequested(): boolean {\n return this.closeRequested;\n }\n\n public onCloseRequest(cb: () => void): () => void {\n if (this.isClosed()) {\n throw new Error('Stream is already closed');\n }\n\n if (this.isCloseRequested()) {\n cb();\n\n return () => undefined;\n }\n\n this.onCloseRequestListeners.add(cb);\n\n return () => this.onCloseRequestListeners.delete(cb);\n }\n\n /**\n * @internal meant for use within river, not exposed as a public API\n *\n * Triggers a close request.\n */\n public triggerCloseRequest(): undefined {\n if (this.isCloseRequested()) {\n throw new Error('Cannot trigger close request multiple times');\n }\n\n if (this.isClosed()) {\n throw new Error('Cannot trigger close request on closed stream');\n }\n\n this.closeRequested = true;\n this.onCloseRequestListeners.forEach((cb) => cb());\n this.onCloseRequestListeners.clear();\n }\n}\n","/* eslint-disable @typescript-eslint/no-unnecessary-type-assertion */\nimport { Static, TNever, TSchema, TUnion, Type } from '@sinclair/typebox';\nimport { ProcedureHandlerContext } from './context';\nimport { BaseErrorSchemaType, Result } from './result';\nimport { ReadStream, WriteStream } from './streams';\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 single message from server (n:1).\n | 'upload'\n // Single message from client, stream from server (1:n).\n | 'subscription'\n // Bidirectional stream (n:n).\n | 'stream';\n\n/**\n * Represents the payload type for {@link Procedure}s.\n */\nexport type PayloadType = TSchema;\n\n/**\n * INTERNAL_RIVER_ERROR_CODE is the code that is used when an internal error occurs,\n * this means that some invariants expected by the river server implementation have\n * been violated. When encountering this error please report this to river maintainers.\n */\nexport const INTERNAL_RIVER_ERROR_CODE = 'INTERNAL_RIVER_ERROR';\n/**\n * UNCAUGHT_ERROR_CODE is the code that is used when an error is thrown\n * inside a procedure handler that's not required.\n */\nexport const UNCAUGHT_ERROR_CODE = 'UNCAUGHT_ERROR' as const;\n/**\n * UNEXPECTED_DISCONNECT_CODE is the code used the stream's session\n * disconnect unexpetedly.\n */\nexport const UNEXPECTED_DISCONNECT_CODE = 'UNEXPECTED_DISCONNECT' as const;\n/**\n * INVALID_REQUEST_CODE is the code used when a client's request is invalid.\n */\nexport const INVALID_REQUEST_CODE = 'INVALID_REQUEST' as const;\n/**\n * ABORT_CODE is the code used when either server or client aborts the stream.\n */\nexport const ABORT_CODE = 'ABORT' as const;\n\n/**\n * OutputReaderErrorSchema is the schema for all the errors that can be\n * emitted in the Output ReadStream on the client.\n */\nexport const OutputReaderErrorSchema = Type.Object({\n code: Type.Union([\n Type.Literal(INTERNAL_RIVER_ERROR_CODE),\n Type.Literal(UNCAUGHT_ERROR_CODE),\n Type.Literal(UNEXPECTED_DISCONNECT_CODE),\n Type.Literal(INVALID_REQUEST_CODE),\n Type.Literal(ABORT_CODE),\n ]),\n message: Type.String(),\n});\n\n/**\n * InputReaderErrorSchema is the schema for all the errors that can be\n * emitted in the Input ReadStream on the server.\n */\nexport const InputReaderErrorSchema = Type.Object({\n code: Type.Union([\n Type.Literal(UNCAUGHT_ERROR_CODE),\n Type.Literal(UNEXPECTED_DISCONNECT_CODE),\n Type.Literal(INVALID_REQUEST_CODE),\n Type.Literal(ABORT_CODE),\n ]),\n message: Type.String(),\n});\n\n/**\n * Represents an acceptable schema to pass to a procedure.\n * Just a type of a schema, not an actual schema.\n */\nexport type ProcedureErrorSchemaType =\n | TUnion<Array<BaseErrorSchemaType>>\n | BaseErrorSchemaType\n | TNever;\n\n/**\n * Procedure for a single message in both directions (1:1).\n *\n * @template State - The context state object.\n * @template Init - The TypeBox schema of the initialization object.\n * @template Output - The TypeBox schema of the output object.\n * @template Err - The TypeBox schema of the error object.\n */\nexport interface RpcProcedure<\n State,\n Init extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n> {\n type: 'rpc';\n init: Init;\n output: Output;\n errors: Err;\n description?: string;\n handler(\n context: ProcedureHandlerContext<State>,\n init: Static<Init>,\n ): Promise<Result<Static<Output>, Static<Err>>>;\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 Init - The TypeBox schema of the initialization object.\n * @template Input - The TypeBox schema of the input object.\n * @template Output - The TypeBox schema of the output object.\n * @template Err - The TypeBox schema of the error object.\n */\nexport interface UploadProcedure<\n State,\n Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n> {\n type: 'upload';\n init: Init;\n input: Input;\n output: Output;\n errors: Err;\n description?: string;\n handler(\n context: ProcedureHandlerContext<State>,\n init: Static<Init>,\n input: ReadStream<Static<Input>, Static<typeof InputReaderErrorSchema>>,\n ): Promise<Result<Static<Output>, Static<Err>>>;\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 Init - The TypeBox schema of the initialization object.\n * @template Output - The TypeBox schema of the output object.\n * @template Err - The TypeBox schema of the error object.\n */\nexport interface SubscriptionProcedure<\n State,\n Init extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n> {\n type: 'subscription';\n init: Init;\n output: Output;\n errors: Err;\n description?: string;\n handler(\n context: ProcedureHandlerContext<State>,\n init: Static<Init>,\n output: WriteStream<Result<Static<Output>, Static<Err>>>,\n ): Promise<void | undefined>;\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 Init - The TypeBox schema of the initialization object.\n * @template Input - The TypeBox schema of the input object.\n * @template Output - The TypeBox schema of the output object.\n * @template Err - The TypeBox schema of the error object.\n */\nexport interface StreamProcedure<\n State,\n Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n> {\n type: 'stream';\n init: Init;\n input: Input;\n output: Output;\n errors: Err;\n description?: string;\n handler(\n context: ProcedureHandlerContext<State>,\n init: Static<Init>,\n input: ReadStream<Static<Input>, Static<typeof InputReaderErrorSchema>>,\n output: WriteStream<Result<Static<Output>, Static<Err>>>,\n ): Promise<void | undefined>;\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 Input - The TypeBox schema of the input object.\n * @template Init - The TypeBox schema of the input initialization object, if any.\n * @template Output - The TypeBox schema of the output object.\n */\nexport type Procedure<\n State,\n Ty extends ValidProcType,\n Init extends PayloadType,\n Input extends PayloadType | null,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n> = { type: Ty } & (Input extends PayloadType\n ? Ty extends 'upload'\n ? UploadProcedure<State, Init, Input, Output, Err>\n : Ty extends 'stream'\n ? StreamProcedure<State, Init, Input, Output, Err>\n : never\n : Ty extends 'rpc'\n ? RpcProcedure<State, Init, Output, Err>\n : Ty extends 'subscription'\n ? SubscriptionProcedure<State, Init, Output, Err>\n : never);\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 | null,\n PayloadType,\n ProcedureErrorSchemaType\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, Init extends PayloadType, Output extends PayloadType>(def: {\n init: Init;\n output: Output;\n errors?: never;\n description?: string;\n handler: RpcProcedure<State, Init, Output, TNever>['handler'];\n}): Branded<RpcProcedure<State, Init, Output, TNever>>;\n\n// signature: explicit errors\nfunction rpc<\n State,\n Init extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n>(def: {\n init: Init;\n output: Output;\n errors: Err;\n description?: string;\n handler: RpcProcedure<State, Init, Output, Err>['handler'];\n}): Branded<RpcProcedure<State, Init, Output, Err>>;\n\n// implementation\nfunction rpc({\n init,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n description?: string;\n handler: RpcProcedure<\n object,\n PayloadType,\n PayloadType,\n ProcedureErrorSchemaType\n >['handler'];\n}) {\n return {\n ...(description ? { description } : {}),\n type: 'rpc',\n init,\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 Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n>(def: {\n init: Init;\n input: Input;\n output: Output;\n errors?: never;\n description?: string;\n handler: UploadProcedure<State, Init, Input, Output, TNever>['handler'];\n}): Branded<UploadProcedure<State, Init, Input, Output, TNever>>;\n\n// signature: init with explicit errors\nfunction upload<\n State,\n Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n>(def: {\n init: Init;\n input: Input;\n output: Output;\n errors: Err;\n description?: string;\n handler: UploadProcedure<State, Init, Input, Output, Err>['handler'];\n}): Branded<UploadProcedure<State, Init, Input, Output, Err>>;\n\n// implementation\nfunction upload({\n init,\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init: PayloadType;\n input: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n description?: string;\n handler: UploadProcedure<\n object,\n PayloadType,\n PayloadType,\n PayloadType,\n ProcedureErrorSchemaType\n >['handler'];\n}) {\n return {\n type: 'upload',\n ...(description ? { description } : {}),\n init,\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 Init extends PayloadType,\n Output extends PayloadType,\n>(def: {\n init: Init;\n output: Output;\n errors?: never;\n description?: string;\n handler: SubscriptionProcedure<State, Init, Output, TNever>['handler'];\n}): Branded<SubscriptionProcedure<State, Init, Output, TNever>>;\n\n// signature: explicit errors\nfunction subscription<\n State,\n Init extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n>(def: {\n init: Init;\n output: Output;\n errors: Err;\n description?: string;\n handler: SubscriptionProcedure<State, Init, Output, Err>['handler'];\n}): Branded<SubscriptionProcedure<State, Init, Output, Err>>;\n\n// implementation\nfunction subscription({\n init,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n description?: string;\n handler: SubscriptionProcedure<\n object,\n PayloadType,\n PayloadType,\n ProcedureErrorSchemaType\n >['handler'];\n}) {\n return {\n type: 'subscription',\n ...(description ? { description } : {}),\n init,\n output,\n errors,\n handler,\n };\n}\n\n/**\n * Creates a {@link StreamProcedure}, optionally with an initialization message.\n */\n// signature: with default errors\nfunction stream<\n State,\n Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n>(def: {\n init: Init;\n input: Input;\n output: Output;\n errors?: never;\n description?: string;\n handler: StreamProcedure<State, Init, Input, Output, TNever>['handler'];\n}): Branded<StreamProcedure<State, Init, Input, Output, TNever>>;\n\n// signature: explicit errors\nfunction stream<\n State,\n Init extends PayloadType,\n Input extends PayloadType,\n Output extends PayloadType,\n Err extends ProcedureErrorSchemaType,\n>(def: {\n init: Init;\n input: Input;\n output: Output;\n errors: Err;\n description?: string;\n handler: StreamProcedure<State, Init, Input, Output, Err>['handler'];\n}): Branded<StreamProcedure<State, Init, Input, Output, Err>>;\n\n// implementation\nfunction stream({\n init,\n input,\n output,\n errors = Type.Never(),\n description,\n handler,\n}: {\n init: PayloadType;\n input: PayloadType;\n output: PayloadType;\n errors?: ProcedureErrorSchemaType;\n description?: string;\n handler: StreamProcedure<\n object,\n PayloadType,\n PayloadType,\n PayloadType,\n ProcedureErrorSchemaType\n >['handler'];\n}) {\n return {\n type: 'stream',\n ...(description ? { description } : {}),\n init,\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","import {\n AnyService,\n ProcErrors,\n ProcInit,\n ProcInput,\n ProcOutput,\n ProcType,\n AnyServiceSchemaMap,\n InstantiatedServiceSchemaMap,\n} from './services';\nimport {\n OpaqueTransportMessage,\n ControlFlags,\n TransportClientId,\n isStreamClose,\n ControlMessageCloseSchema,\n isStreamCloseRequest,\n isStreamAbort,\n closeStreamMessage,\n requestCloseStreamMessage,\n abortMessage,\n} from '../transport/message';\nimport { Static } from '@sinclair/typebox';\nimport {\n BaseErrorSchemaType,\n Err,\n Result,\n AnyResultSchema,\n ErrResultSchema,\n} from './result';\nimport { EventMap } from '../transport/events';\nimport { Connection } from '../transport/connection';\nimport { Logger } from '../logging';\nimport { createProcTelemetryInfo, getPropagationContext } from '../tracing';\nimport { ClientHandshakeOptions } from './handshake';\nimport { ClientTransport } from '../transport/client';\nimport { generateId } from '../transport/id';\nimport {\n ReadStream,\n ReadStreamImpl,\n WriteStream,\n WriteStreamImpl,\n} from './streams';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n ABORT_CODE,\n OutputReaderErrorSchema,\n PayloadType,\n UNEXPECTED_DISCONNECT_CODE,\n ValidProcType,\n} from './procedures';\n\nconst OutputErrResultSchema = ErrResultSchema(OutputReaderErrorSchema);\n\ninterface CallOptions {\n signal?: AbortSignal;\n}\n\ntype RpcFn<\n Router extends AnyService,\n ProcName extends keyof Router['procedures'],\n> = (\n init: ProcInit<Router, ProcName>,\n options?: CallOptions,\n) => Promise<\n Result<ProcOutput<Router, ProcName>, ProcErrors<Router, ProcName>>\n>;\n\ntype UploadFn<\n Router extends AnyService,\n ProcName extends keyof Router['procedures'],\n> = (\n init: ProcInit<Router, ProcName>,\n options?: CallOptions,\n) => [\n WriteStream<ProcInput<Router, ProcName>>,\n () => Promise<\n Result<ProcOutput<Router, ProcName>, ProcErrors<Router, ProcName>>\n >,\n];\n\ntype StreamFn<\n Router extends AnyService,\n ProcName extends keyof Router['procedures'],\n> = (\n init: ProcInit<Router, ProcName>,\n options?: CallOptions,\n) => [\n WriteStream<ProcInput<Router, ProcName>>,\n ReadStream<ProcOutput<Router, ProcName>, ProcErrors<Router, ProcName>>,\n];\n\ntype SubscriptionFn<\n Router extends AnyService,\n ProcName extends keyof Router['procedures'],\n> = (\n init: ProcInit<Router, ProcName>,\n options?: CallOptions,\n) => ReadStream<ProcOutput<Router, ProcName>, ProcErrors<Router, ProcName>>;\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 // If your go-to-definition ended up here, you probably meant to\n // go to the procedure name. For example:\n // riverClient.myService.someprocedure.rpc({})\n // click here ^^^^^^^^^^^^^\n rpc: RpcFn<Router, ProcName>;\n }\n : ProcType<Router, ProcName> extends 'upload'\n ? {\n // If your go-to-definition ended up here, you probably meant to\n // go to the procedure name. For example:\n // riverClient.myService.someprocedure.upload({})\n // click here ^^^^^^^^^^^^^\n upload: UploadFn<Router, ProcName>;\n }\n : ProcType<Router, ProcName> extends 'stream'\n ? {\n // If your go-to-definition ended up here, you probably meant to\n // go to the procedure name. For example:\n // riverClient.myService.someprocedure.stream({})\n // click here ^^^^^^^^^^^^^\n stream: StreamFn<Router, ProcName>;\n }\n : ProcType<Router, ProcName> extends 'subscription'\n ? {\n // If your go-to-definition ended up here, you probably meant to\n // go to the procedure name. For example:\n // riverClient.myService.subscribe.stream({})\n // click here ^^^^^^^^^^^^^\n subscribe: SubscriptionFn<Router, ProcName>;\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 clientOptions = { ...defaultClientOptions, ...providedClientOptions };\n if (clientOptions.eagerlyConnect) {\n transport.connect(serverId);\n }\n\n return _createRecursiveProxy((opts) => {\n const [serviceName, procName, procMethod] = [...opts.path];\n if (!(serviceName && procName && procMethod)) {\n throw new Error(\n 'invalid river call, ensure the service and procedure you are calling exists',\n );\n }\n\n const [init, callOptions] = opts.args;\n\n if (clientOptions.connectOnInvoke && !transport.sessions.has(serverId)) {\n transport.connect(serverId);\n }\n\n if (\n procMethod !== 'rpc' &&\n procMethod !== 'subscribe' &&\n procMethod !== 'stream' &&\n procMethod !== 'upload'\n ) {\n throw new Error(\n `invalid river call, unknown procedure type ${procMethod}`,\n );\n }\n\n return handleProc(\n procMethod === 'subscribe' ? 'subscription' : procMethod,\n transport,\n serverId,\n init,\n serviceName,\n procName,\n callOptions ? (callOptions as CallOptions).signal : undefined,\n );\n }, []) as Client<ServiceSchemaMap>;\n}\n\ntype ClientProcReturn<ProcType extends ValidProcType> = ReturnType<\n ProcType extends 'rpc'\n ? RpcFn<AnyService, string>\n : ProcType extends 'upload'\n ? UploadFn<AnyService, string>\n : ProcType extends 'stream'\n ? StreamFn<AnyService, string>\n : ProcType extends 'subscription'\n ? SubscriptionFn<AnyService, string>\n : never\n>;\n\nfunction handleProc(\n procType: ValidProcType,\n transport: ClientTransport<Connection>,\n serverId: TransportClientId,\n init: Static<PayloadType>,\n serviceName: string,\n procedureName: string,\n abortSignal?: AbortSignal,\n): ClientProcReturn<ValidProcType> {\n const procClosesWithInit = procType === 'rpc' || procType === 'subscription';\n\n const streamId = generateId();\n const { span, ctx } = createProcTelemetryInfo(\n transport,\n procType,\n serviceName,\n procedureName,\n streamId,\n );\n let cleanClose = true;\n const inputWriter = new WriteStreamImpl<Static<PayloadType>>((rawIn) => {\n transport.send(serverId, {\n streamId,\n payload: rawIn,\n controlFlags: 0,\n tracing: getPropagationContext(ctx),\n });\n });\n inputWriter.onClose(() => {\n span.addEvent('inputWriter closed');\n\n if (!procClosesWithInit && cleanClose) {\n transport.send(serverId, closeStreamMessage(streamId));\n }\n\n if (outputReader.isClosed()) {\n cleanup();\n }\n });\n\n const outputReader = new ReadStreamImpl<\n Static<PayloadType>,\n Static<BaseErrorSchemaType>\n >(() => {\n transport.send(serverId, requestCloseStreamMessage(streamId));\n });\n outputReader.onClose(() => {\n span.addEvent('outputReader closed');\n\n if (inputWriter.isClosed()) {\n cleanup();\n }\n });\n\n function cleanup() {\n transport.removeEventListener('message', onMessage);\n transport.removeEventListener('sessionStatus', onSessionStatus);\n abortSignal?.removeEventListener('abort', onClientAbort);\n span.end();\n }\n\n function onClientAbort() {\n if (outputReader.isClosed() && inputWriter.isClosed()) {\n return;\n }\n\n span.addEvent('sending abort');\n\n cleanClose = false;\n\n if (!outputReader.isClosed()) {\n outputReader.pushValue(\n Err({\n code: ABORT_CODE,\n message: 'Aborted by client',\n }),\n );\n outputReader.triggerClose();\n }\n\n inputWriter.close();\n transport.send(\n serverId,\n abortMessage(\n streamId,\n Err({\n code: ABORT_CODE,\n message: 'Aborted by client',\n }),\n ),\n );\n }\n\n function onMessage(msg: OpaqueTransportMessage) {\n if (msg.streamId !== streamId) return;\n if (msg.to !== transport.clientId) {\n transport.log?.error('Got stream message from unexpected client', {\n clientId: transport.clientId,\n transportMessage: msg,\n });\n\n return;\n }\n\n if (isStreamCloseRequest(msg.controlFlags)) {\n inputWriter.triggerCloseRequest();\n }\n\n if (isStreamAbort(msg.controlFlags)) {\n cleanClose = false;\n\n span.addEvent('received abort');\n let abortResult: Static<typeof OutputErrResultSchema>;\n\n if (Value.Check(OutputErrResultSchema, msg.payload)) {\n abortResult = msg.payload;\n } else {\n abortResult = Err({\n code: ABORT_CODE,\n message: 'Stream aborted with invalid payload',\n });\n transport.log?.error(\n 'Got stream abort without a valid protocol error',\n {\n clientId: transport.clientId,\n transportMessage: msg,\n validationErrors: [\n ...Value.Errors(OutputErrResultSchema, msg.payload),\n ],\n },\n );\n }\n\n if (!outputReader.isClosed()) {\n outputReader.pushValue(abortResult);\n outputReader.triggerClose();\n }\n\n inputWriter.close();\n\n return;\n }\n\n if (outputReader.isClosed()) {\n span.recordException('Received message after output stream is closed');\n\n transport.log?.error('Received message after output stream is closed', {\n clientId: transport.clientId,\n transportMessage: msg,\n });\n\n return;\n }\n\n if (!Value.Check(ControlMessageCloseSchema, msg.payload)) {\n if (Value.Check(AnyResultSchema, msg.payload)) {\n outputReader.pushValue(msg.payload);\n } else {\n transport.log?.error(\n 'Got non-control payload, but was not a valid result',\n {\n clientId: transport.clientId,\n transportMessage: msg,\n validationErrors: [...Value.Errors(AnyResultSchema, msg.payload)],\n },\n );\n }\n }\n\n if (isStreamClose(msg.controlFlags)) {\n span.addEvent('received output close');\n\n outputReader.triggerClose();\n }\n }\n\n function onSessionStatus(evt: EventMap['sessionStatus']) {\n if (evt.status !== 'disconnect') {\n return;\n }\n\n if (evt.session.to !== serverId) {\n return;\n }\n\n cleanClose = false;\n if (!outputReader.isClosed()) {\n outputReader.pushValue(\n Err({\n code: UNEXPECTED_DISCONNECT_CODE,\n message: `${serverId} unexpectedly disconnected`,\n }),\n );\n }\n inputWriter.close();\n outputReader.triggerClose();\n }\n\n abortSignal?.addEventListener('abort', onClientAbort);\n transport.addEventListener('message', onMessage);\n transport.addEventListener('sessionStatus', onSessionStatus);\n\n transport.send(serverId, {\n streamId,\n serviceName,\n procedureName,\n tracing: getPropagationContext(ctx),\n payload: init,\n controlFlags: procClosesWithInit\n ? ControlFlags.StreamOpenBit | ControlFlags.StreamClosedBit\n : ControlFlags.StreamOpenBit,\n });\n\n if (procClosesWithInit) {\n inputWriter.close();\n }\n\n if (procType === 'subscription') {\n return outputReader;\n }\n\n if (procType === 'rpc') {\n return getSingleMessage(outputReader, transport.log);\n }\n\n if (procType === 'upload') {\n let didFinalize = false;\n return [\n inputWriter,\n async () => {\n if (didFinalize) {\n throw new Error('upload stream already finalized');\n }\n\n didFinalize = true;\n\n if (!inputWriter.isClosed()) {\n inputWriter.close();\n }\n\n return getSingleMessage(outputReader, transport.log);\n },\n ];\n }\n\n // good ol' `stream` procType\n return [inputWriter, outputReader];\n}\n\nasync function getSingleMessage(\n outputReader: ReadStream<unknown, Static<BaseErrorSchemaType>>,\n log?: Logger,\n): Promise<Result<unknown, Static<BaseErrorSchemaType>>> {\n const ret = await outputReader.asArray();\n\n if (ret.length > 1) {\n log?.error('Expected single message from server, got multiple');\n }\n\n return ret[0];\n}\n","import { Static } from '@sinclair/typebox';\nimport {\n PayloadType,\n ProcedureErrorSchemaType,\n InputReaderErrorSchema,\n OutputReaderErrorSchema,\n UNCAUGHT_ERROR_CODE,\n UNEXPECTED_DISCONNECT_CODE,\n AnyProcedure,\n ABORT_CODE,\n INVALID_REQUEST_CODE,\n INTERNAL_RIVER_ERROR_CODE,\n} from './procedures';\nimport {\n AnyService,\n InstantiatedServiceSchemaMap,\n AnyServiceSchemaMap,\n} from './services';\nimport {\n ControlMessagePayloadSchema,\n OpaqueTransportMessage,\n isStreamClose,\n isStreamOpen,\n ControlFlags,\n isStreamCloseRequest,\n isStreamAbort,\n requestCloseStreamMessage,\n closeStreamMessage,\n abortMessage,\n} from '../transport/message';\nimport {\n ServiceContext,\n ProcedureHandlerContext,\n ParsedMetadata,\n} from './context';\nimport { Logger, MessageMetadata } from '../logging/log';\nimport { Value } from '@sinclair/typebox/value';\nimport { Err, Result, Ok, ErrResultSchema, ErrResult } from './result';\nimport { EventMap } from '../transport/events';\nimport { coerceErrorString } from '../util/stringify';\nimport { Span, SpanStatusCode } from '@opentelemetry/api';\nimport { PropagationContext, createHandlerSpan } from '../tracing';\nimport { ServerHandshakeOptions } from './handshake';\nimport { Connection } from '../transport/connection';\nimport { ServerTransport } from '../transport/server';\nimport { ReadStreamImpl, WriteStreamImpl } from './streams';\n\ntype StreamId = string;\n\n/**\n * A result schema for errors that can be passed to input's readstream\n */\nconst InputErrResultSchema = ErrResultSchema(InputReaderErrorSchema);\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 /**\n * Services defined for this server.\n */\n services: InstantiatedServiceSchemaMap<Services>;\n /**\n * A set of stream ids that are currently open.\n */\n openStreams: Set<StreamId>;\n}\n\ntype InputHandlerReturn = Promise<(() => void) | void>;\n\ninterface NewProcStreamInput {\n procedure: AnyProcedure;\n procedureName: string;\n service: AnyService;\n serviceName: string;\n sessionMetadata: ParsedMetadata;\n loggingMetadata: MessageMetadata;\n streamId: string;\n controlFlags: number;\n tracingCtx?: PropagationContext;\n initPayload: Static<PayloadType>;\n from: string;\n sessionId: string;\n protocolVersion: string;\n passInitAsInputForBackwardsCompat: boolean;\n}\n\nclass RiverServer<Services extends AnyServiceSchemaMap>\n implements Server<Services>\n{\n private transport: ServerTransport<Connection>;\n private contextMap: Map<AnyService, ServiceContext & { state: object }>;\n private log?: Logger;\n /**\n * We create a tombstones for streams aborted by the server\n * so that we don't hit errors when the client has inflight\n * requests it sent before it saw the abort.\n * We track aborted streams for every session separately, so\n * that bad clients don't affect good clients.\n */\n private serverAbortedStreams: Map<string, LRUSet>;\n private maxAbortedStreamTombstonesPerSession: number;\n\n public openStreams: Set<string>;\n public services: InstantiatedServiceSchemaMap<Services>;\n\n constructor(\n transport: ServerTransport<Connection>,\n services: Services,\n handshakeOptions?: ServerHandshakeOptions,\n extendedContext?: Omit<ServiceContext, 'state'>,\n maxAbortedStreamTombstonesPerSession = 200,\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(extendedContext ?? {});\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.openStreams = new Set();\n this.serverAbortedStreams = new Map();\n this.maxAbortedStreamTombstonesPerSession =\n maxAbortedStreamTombstonesPerSession;\n this.log = transport.log;\n\n const handleMessage = (msg: EventMap['message']) => {\n if (msg.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: msg,\n },\n );\n return;\n }\n\n if (this.openStreams.has(msg.streamId)) {\n // has its own message handler\n return;\n }\n\n if (this.serverAbortedStreams.get(msg.from)?.has(msg.streamId)) {\n return;\n }\n\n const validated = this.validateNewProcStream(msg);\n\n if (!validated) {\n return;\n }\n\n this.createNewProcStream(validated);\n };\n this.transport.addEventListener('message', handleMessage);\n\n const handleSessionStatus = (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 this.serverAbortedStreams.delete(disconnectedClientId);\n };\n this.transport.addEventListener('sessionStatus', handleSessionStatus);\n\n this.transport.addEventListener('transportStatus', (evt) => {\n if (evt.status !== 'closed') return;\n\n this.transport.removeEventListener('message', handleMessage);\n this.transport.removeEventListener('sessionStatus', handleSessionStatus);\n });\n }\n\n private createNewProcStream({\n procedure,\n procedureName,\n service,\n serviceName,\n sessionMetadata,\n loggingMetadata,\n streamId,\n controlFlags,\n initPayload,\n from,\n sessionId,\n tracingCtx,\n protocolVersion,\n passInitAsInputForBackwardsCompat,\n }: NewProcStreamInput) {\n this.openStreams.add(streamId);\n\n let cleanClose = true;\n\n const onServerAbort = (errResult: Static<typeof InputErrResultSchema>) => {\n if (inputReader.isClosed() && outputWriter.isClosed()) {\n // Everything already closed, no-op.\n return;\n }\n\n cleanClose = false;\n\n if (!inputReader.isClosed()) {\n inputReader.pushValue(errResult);\n inputReader.triggerClose();\n }\n\n outputWriter.close();\n this.abortStream(from, streamId, errResult);\n };\n\n const onHandlerAbort = () => {\n onServerAbort(\n Err({\n code: ABORT_CODE,\n message: 'Aborted by server procedure handler',\n }),\n );\n };\n const handlerAbortController = new AbortController();\n handlerAbortController.signal.addEventListener('abort', onHandlerAbort);\n\n const clientAbortController = new AbortController();\n\n const onSessionStatus = (evt: EventMap['sessionStatus']) => {\n if (evt.status !== 'disconnect') {\n return;\n }\n\n if (evt.session.to !== from) {\n return;\n }\n\n cleanClose = false;\n\n const errPayload = {\n code: UNEXPECTED_DISCONNECT_CODE,\n message: `client unexpectedly disconnected`,\n } as const;\n if (!inputReader.isClosed()) {\n inputReader.pushValue(Err(errPayload));\n inputReader.triggerClose();\n }\n\n clientAbortController.abort(errPayload);\n\n outputWriter.close();\n };\n this.transport.addEventListener('sessionStatus', onSessionStatus);\n\n const onMessage = (msg: OpaqueTransportMessage) => {\n if (streamId !== msg.streamId) {\n return;\n }\n\n if (msg.from !== from) {\n this.log?.error('Got stream message from unexpected client', {\n ...loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n\n return;\n }\n\n if (isStreamCloseRequest(msg.controlFlags)) {\n outputWriter.triggerCloseRequest();\n }\n\n if (isStreamAbortBackwardsCompat(msg.controlFlags, protocolVersion)) {\n let abortResult: Static<typeof InputErrResultSchema>;\n if (Value.Check(InputErrResultSchema, msg.payload)) {\n abortResult = msg.payload;\n } else {\n abortResult = Err({\n code: ABORT_CODE,\n message: 'Stream aborted, client sent invalid payload',\n });\n this.log?.warn('Got stream abort without a valid protocol error', {\n ...loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: msg,\n validationErrors: [\n ...Value.Errors(InputErrResultSchema, msg.payload),\n ],\n tags: ['invalid-request'],\n });\n }\n\n if (!inputReader.isClosed()) {\n inputReader.pushValue(abortResult);\n inputReader.triggerClose();\n }\n\n outputWriter.close();\n\n clientAbortController.abort(abortResult.payload);\n\n return;\n }\n\n if (inputReader.isClosed()) {\n this.log?.warn('Received message after input stream is closed', {\n ...loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: msg,\n tags: ['invalid-request'],\n });\n\n onServerAbort(\n Err({\n code: INVALID_REQUEST_CODE,\n message: 'Received message after input stream is closed',\n }),\n );\n\n return;\n }\n\n if ('input' in procedure && Value.Check(procedure.input, msg.payload)) {\n inputReader.pushValue(Ok(msg.payload));\n } else if (!Value.Check(ControlMessagePayloadSchema, msg.payload)) {\n const validationErrors = [\n ...Value.Errors(ControlMessagePayloadSchema, msg.payload),\n ];\n let errMessage = 'Expected control payload for procedure with no input';\n if ('input' in procedure) {\n errMessage =\n 'Expected either control or input payload, validation failed for both';\n validationErrors.push(...Value.Errors(procedure.input, msg.payload));\n }\n\n this.log?.warn(errMessage, {\n ...loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: msg,\n validationErrors,\n tags: ['invalid-request'],\n });\n\n onServerAbort(\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n }\n\n if (isStreamCloseBackwardsCompat(msg.controlFlags, protocolVersion)) {\n inputReader.triggerClose();\n }\n };\n this.transport.addEventListener('message', onMessage);\n\n const onFinishedCallbacks: Array<() => void> = [];\n const cleanup = () => {\n this.transport.removeEventListener('message', onMessage);\n this.transport.removeEventListener('sessionStatus', onSessionStatus);\n handlerAbortController.signal.addEventListener('abort', onHandlerAbort);\n\n this.openStreams.delete(streamId);\n onFinishedCallbacks.forEach((cb) => {\n try {\n cb();\n } catch {\n // ignore user errors\n }\n });\n onFinishedCallbacks.length = 0;\n };\n\n const procClosesWithResponse =\n procedure.type === 'rpc' || procedure.type === 'upload';\n\n const inputReader = new ReadStreamImpl<\n Static<PayloadType>,\n Static<typeof InputReaderErrorSchema>\n >(() => {\n this.transport.send(from, requestCloseStreamMessage(streamId));\n });\n inputReader.onClose(() => {\n // TODO remove once clients migrate to v2\n if (protocolVersion === 'v1.1') {\n // in v1.1 a close in either direction should close everything\n // for upload/rpc it will handle the close after it responds\n if (!procClosesWithResponse && !outputWriter.isClosed()) {\n outputWriter.close();\n }\n }\n\n if (outputWriter.isClosed()) {\n cleanup();\n }\n });\n\n if (passInitAsInputForBackwardsCompat) {\n inputReader.pushValue(Ok(initPayload));\n }\n\n const outputWriter = new WriteStreamImpl<\n Result<Static<PayloadType>, Static<ProcedureErrorSchemaType>>\n >((response) => {\n this.transport.send(from, {\n streamId,\n controlFlags: procClosesWithResponse\n ? getStreamCloseBackwardsCompat(protocolVersion)\n : 0,\n payload: response,\n });\n });\n outputWriter.onClose(() => {\n if (!procClosesWithResponse && cleanClose) {\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\n const message = closeStreamMessage(streamId);\n // TODO remove once clients migrate to v2\n message.controlFlags = getStreamCloseBackwardsCompat(protocolVersion);\n\n this.transport.send(from, closeStreamMessage(streamId));\n }\n\n // TODO remove once clients migrate to v2\n if (protocolVersion === 'v1.1') {\n // in v1.1 a close in either direction should close everything\n if (!inputReader.isClosed()) {\n inputReader.triggerClose();\n }\n }\n\n if (inputReader.isClosed()) {\n cleanup();\n }\n });\n\n const onHandlerError = (err: unknown, span: Span) => {\n const errorMsg = coerceErrorString(err);\n\n span.recordException(err instanceof Error ? err : new Error(errorMsg));\n span.setStatus({ code: SpanStatusCode.ERROR });\n\n onServerAbort(\n Err({\n code: UNCAUGHT_ERROR_CODE,\n message: errorMsg,\n }),\n );\n };\n\n if (isStreamCloseBackwardsCompat(controlFlags, protocolVersion)) {\n inputReader.triggerClose();\n } else if (procedure.type === 'rpc' || procedure.type === 'subscription') {\n // Though things can work just fine if they eventually follow up with a stream\n // control message with a close bit set, it's an unusual client implementation!\n this.log?.warn(`${procedure.type} sent an init without a stream close`, {\n ...loggingMetadata,\n clientId: this.transport.clientId,\n });\n }\n\n const serviceContextWithTransportInfo: ProcedureHandlerContext<object> = {\n ...this.getContext(service, serviceName),\n from,\n sessionId,\n metadata: sessionMetadata,\n abortController: handlerAbortController,\n clientAbortSignal: clientAbortController.signal,\n onRequestFinished: (cb) => {\n if (inputReader.isClosed() && outputWriter.isClosed()) {\n // Everything already closed, call cleanup immediately.\n try {\n cb();\n } catch {\n // ignore user errors\n }\n\n return;\n }\n\n onFinishedCallbacks.push(cb);\n },\n };\n\n switch (procedure.type) {\n case 'rpc':\n void createHandlerSpan(\n procedure.type,\n serviceName,\n procedureName,\n streamId,\n tracingCtx,\n async (span): InputHandlerReturn => {\n try {\n // TODO handle never resolving after cleanup/full close\n // which would lead to us holding on to the closure forever\n const outputMessage = await procedure.handler(\n serviceContextWithTransportInfo,\n initPayload,\n );\n\n if (outputWriter.isClosed()) {\n // A disconnect happened\n return;\n }\n\n outputWriter.write(outputMessage);\n outputWriter.close();\n } catch (err) {\n onHandlerError(err, span);\n } finally {\n span.end();\n }\n },\n );\n break;\n case 'stream':\n void createHandlerSpan(\n procedure.type,\n serviceName,\n procedureName,\n streamId,\n tracingCtx,\n async (span): InputHandlerReturn => {\n try {\n // TODO handle never resolving after cleanup/full close\n // which would lead to us holding on to the closure forever\n await procedure.handler(\n serviceContextWithTransportInfo,\n initPayload,\n inputReader,\n outputWriter,\n );\n } catch (err) {\n onHandlerError(err, span);\n } finally {\n span.end();\n }\n },\n );\n\n break;\n case 'subscription':\n void createHandlerSpan(\n procedure.type,\n serviceName,\n procedureName,\n streamId,\n tracingCtx,\n async (span): InputHandlerReturn => {\n try {\n // TODO handle never resolving after cleanup/full close\n // which would lead to us holding on to the closure forever\n await procedure.handler(\n serviceContextWithTransportInfo,\n passInitAsInputForBackwardsCompat ? null : initPayload,\n outputWriter,\n );\n } catch (err) {\n onHandlerError(err, span);\n } finally {\n span.end();\n }\n },\n );\n break;\n case 'upload':\n void createHandlerSpan(\n procedure.type,\n serviceName,\n procedureName,\n streamId,\n tracingCtx,\n async (span): InputHandlerReturn => {\n try {\n // TODO handle never resolving after cleanup/full close\n // which would lead to us holding on to the closure forever\n const outputMessage = await procedure.handler(\n serviceContextWithTransportInfo,\n passInitAsInputForBackwardsCompat ? null : initPayload,\n inputReader,\n );\n\n if (outputWriter.isClosed()) {\n // A disconnect happened\n return;\n }\n outputWriter.write(outputMessage);\n outputWriter.close();\n } catch (err) {\n onHandlerError(err, span);\n } finally {\n span.end();\n }\n },\n );\n\n break;\n default:\n this.log?.error(\n `got request for invalid procedure type ${\n (procedure as AnyProcedure).type\n } at ${serviceName}.${procedureName}`,\n {\n ...loggingMetadata,\n tags: ['invariant-violation'],\n },\n );\n\n return;\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 private validateNewProcStream(\n initMessage: OpaqueTransportMessage,\n ): null | NewProcStreamInput {\n const session = this.transport.sessions.get(initMessage.from);\n if (!session) {\n const errMessage = `couldn't find a session for ${initMessage.from}`;\n this.log?.error(`couldn't find session for ${initMessage.from}`, {\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invariant-violation'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INTERNAL_RIVER_ERROR_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n const sessionMetadata = this.transport.sessionHandshakeMetadata.get(\n session.to,\n );\n if (!sessionMetadata) {\n const errMessage = `session doesn't have handshake metadata`;\n this.log?.error(errMessage, {\n ...session.loggingMetadata,\n tags: ['invariant-violation'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INTERNAL_RIVER_ERROR_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n if (!isStreamOpen(initMessage.controlFlags)) {\n const errMessage = `can't create a new procedure stream from a message that doesn't have the stream open bit set`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n if (!initMessage.serviceName) {\n const errMessage = `missing service name in stream open message`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n if (!initMessage.procedureName) {\n const errMessage = `missing procedure name in stream open message`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n if (!(initMessage.serviceName in this.services)) {\n const errMessage = `couldn't find service ${initMessage.serviceName}`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n const service = this.services[initMessage.serviceName];\n if (!(initMessage.procedureName in service.procedures)) {\n const errMessage = `couldn't find a matching procedure for ${initMessage.serviceName}.${initMessage.procedureName}`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n const procedure = service.procedures[initMessage.procedureName];\n\n let passInitAsInputForBackwardsCompat = false;\n if (\n session.protocolVersion === 'v1.1' &&\n (procedure.type === 'upload' || procedure.type === 'stream') &&\n Value.Check(procedure.input, initMessage.payload) &&\n Value.Check(procedure.init, {})\n ) {\n // TODO remove once clients migrate to v2\n // In v1.1 sometimes the first message is not `init`, but instead it's the `input`\n // this backwards compatibility path requires procedures to define their `init` as\n // an empty-object-compatible-schema (i.e. either actually empty or optional values)\n // The reason we don't check if `init` is satisified here is because false positives\n // are easy to hit, we'll err on the side of caution and treat it as an input, servers\n // that expect v1.1 clients should handle this case themselves.\n passInitAsInputForBackwardsCompat = true;\n } else if (!Value.Check(procedure.init, initMessage.payload)) {\n const errMessage = `procedure init failed validation`;\n this.log?.warn(errMessage, {\n ...session.loggingMetadata,\n clientId: this.transport.clientId,\n transportMessage: initMessage,\n tags: ['invalid-request'],\n });\n\n this.abortStream(\n initMessage.from,\n initMessage.streamId,\n Err({\n code: INVALID_REQUEST_CODE,\n message: errMessage,\n }),\n );\n\n return null;\n }\n\n return {\n sessionMetadata,\n procedure,\n procedureName: initMessage.procedureName,\n service,\n serviceName: initMessage.serviceName,\n loggingMetadata: {\n ...session.loggingMetadata,\n transportMessage: initMessage,\n },\n streamId: initMessage.streamId,\n controlFlags: initMessage.controlFlags,\n tracingCtx: initMessage.tracing,\n initPayload: initMessage.payload,\n from: initMessage.from,\n sessionId: session.id,\n protocolVersion: session.protocolVersion,\n passInitAsInputForBackwardsCompat,\n };\n }\n\n abortStream(\n to: string,\n streamId: string,\n payload: ErrResult<Static<typeof OutputReaderErrorSchema>>,\n ) {\n let abortedForSession = this.serverAbortedStreams.get(to);\n\n if (!abortedForSession) {\n abortedForSession = new LRUSet(this.maxAbortedStreamTombstonesPerSession);\n\n this.serverAbortedStreams.set(to, abortedForSession);\n }\n\n abortedForSession.add(streamId);\n\n this.transport.send(\n to,\n // TODO remove once clients migrate to v2\n this.transport.sessions.get(to)?.protocolVersion === 'v1.1'\n ? closeStreamMessage(streamId)\n : abortMessage(streamId, payload),\n );\n }\n}\n\nclass LRUSet {\n private items: Set<StreamId>;\n private maxItems: number;\n\n constructor(maxItems: number) {\n this.items = new Set();\n this.maxItems = maxItems;\n }\n\n add(item: string) {\n if (this.items.has(item)) {\n this.items.delete(item);\n } else if (this.items.size >= this.maxItems) {\n const first = this.items.values().next();\n if (!first.done) {\n this.items.delete(first.value);\n }\n }\n this.items.add(item);\n }\n\n has(item: string) {\n return this.items.has(item);\n }\n}\n\n// TODO remove once clients migrate to v2\nfunction isStreamAbortBackwardsCompat(\n controlFlags: ControlFlags,\n protocolVersion: string,\n) {\n if (protocolVersion === 'v1.1') {\n // in 1.1 we don't have abort\n return false;\n }\n\n return isStreamAbort(controlFlags);\n}\n\n// TODO remove once clients migrate to v2\nfunction isStreamCloseBackwardsCompat(\n controlFlags: ControlFlags,\n protocolVersion: string,\n) {\n if (protocolVersion === 'v1.1') {\n // in v1.1 the bits for close is what we use for abort now\n return isStreamAbort(controlFlags);\n }\n\n return isStreamClose(controlFlags);\n}\n\n// TODO remove once clients migrate to v2\nfunction getStreamCloseBackwardsCompat(protocolVersion: string) {\n if (protocolVersion === 'v1.1') {\n // in v1.1 the bits for close is what we use for abort now\n return ControlFlags.StreamAbortBit;\n }\n\n return ControlFlags.StreamClosedBit;\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 * Maximum number of aborted streams to keep track of to avoid\n * cascading stream errors.\n */\n maxAbortedStreamTombstonesPerSession?: number;\n }>,\n): Server<Services> {\n return new RiverServer(\n transport,\n services,\n providedServerOptions?.handshakeOptions,\n providedServerOptions?.extendedContext,\n providedServerOptions?.maxAbortedStreamTombstonesPerSession,\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;AAyM/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,EAOV;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,MAAM,KAAK,OAAO,QAAQ,IAAI;AAAA,YAC9B,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,WAAW,UACX;AAAA,cACE,OAAO,KAAK,OAAO,QAAQ,KAAK;AAAA,YAClC,IACA,CAAC;AAAA,UACP;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,+BAAkE;AAChE,WAAO;AAAA,MACL,YAAY,OAAO;AAAA,QACjB,OAAO,QAAQ,KAAK,UAAU,EAAE;AAAA,UAC9B,CAAC,CAAC,UAAU,OAAO,MAGd;AACH,gBAAI,QAAQ,SAAS,SAAS,QAAQ,SAAS,gBAAgB;AAC7D,qBAAO;AAAA,gBACL;AAAA,gBACA;AAAA;AAAA;AAAA,kBAGE,OAAO,KAAK,OAAO,QAAQ,IAAI;AAAA,kBAC/B,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA;AAAA,kBAElC,GAAI,iBAAiB,UACjB,EAAE,aAAa,QAAQ,YAAY,IACnC,CAAC;AAAA;AAAA,kBAEL,GAAI,YAAY,UACZ;AAAA,oBACE,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA,kBACpC,IACA,CAAC;AAAA,kBACL,MAAM,QAAQ;AAAA,gBAChB;AAAA,cACF;AAAA,YACF;AAIA,mBAAO;AAAA,cACL;AAAA,cACA;AAAA,gBACE,MAAM,KAAK,OAAO,QAAQ,IAAI;AAAA,gBAC9B,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA;AAAA,gBAElC,GAAI,iBAAiB,UACjB,EAAE,aAAa,QAAQ,YAAY,IACnC,CAAC;AAAA;AAAA,gBAEL,GAAI,YAAY,UACZ;AAAA,kBACE,QAAQ,KAAK,OAAO,QAAQ,MAAM;AAAA,gBACpC,IACA,CAAC;AAAA,gBACL,MAAM,QAAQ;AAAA,gBACd,OAAO,KAAK,OAAO,QAAQ,KAAK;AAAA,cAClC;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,YAAY,iBAA6D;AACvE,WAAO,OAAO,OAAO;AAAA,MACnB,OAAO,KAAK,gBAAgB,eAAe;AAAA,MAC3C,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;;;ACvlBA;AAAA,EAOE,QAAAA;AAAA,OACK;AAoBA,IAAM,kBAAkB,CAAgC,MAC7DA,MAAK,OAAO;AAAA,EACV,IAAIA,MAAK,QAAQ,KAAK;AAAA,EACtB,SAAS;AACX,CAAC;AAKI,IAAM,kBAAkBA,MAAK,MAAM;AAAA,EACxCA,MAAK,OAAO;AAAA,IACV,IAAIA,MAAK,QAAQ,KAAK;AAAA,IACtB,SAASA,MAAK,OAAO;AAAA,MACnB,MAAMA,MAAK,OAAO;AAAA,MAClB,SAASA,MAAK,OAAO;AAAA,MACrB,QAAQA,MAAK,SAASA,MAAK,QAAQ,CAAC;AAAA,IACtC,CAAC;AAAA,EACH,CAAC;AAAA,EAEDA,MAAK,OAAO;AAAA,IACV,IAAIA,MAAK,QAAQ,IAAI;AAAA,IACrB,SAASA,MAAK,QAAQ;AAAA,EACxB,CAAC;AACH,CAAC;AAiBM,SAAS,GAAY,SAAyB;AACnD,SAAO;AAAA,IACL,IAAI;AAAA,IACJ;AAAA,EACF;AACF;AAEO,SAAS,IACd,OACgB;AAChB,SAAO;AAAA,IACL,IAAI;AAAA,IACJ,SAAS;AAAA,EACX;AACF;;;AC/EO,IAAM,qBAAqB;AAAA,EAChC,MAAM;AAAA,EACN,SAAS;AACX;AAuHO,IAAM,iBAAN,MAEP;AAAA;AAAA;AAAA;AAAA,EAIU,SAAS;AAAA;AAAA;AAAA;AAAA,EAIT;AAAA;AAAA;AAAA;AAAA,EAIA,iBAAiB;AAAA;AAAA;AAAA;AAAA,EAIjB;AAAA;AAAA;AAAA;AAAA,EAIA,SAAS;AAAA;AAAA;AAAA;AAAA,EAIT,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMV,wBAAwB;AAAA;AAAA;AAAA;AAAA,EAIxB,QAAuC,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA,EAKxC,cAAoC;AAAA;AAAA;AAAA;AAAA,EAIpC,qBAA0C;AAAA,EAElD,YAAY,sBAAkC;AAC5C,SAAK,uBAAuB;AAC5B,SAAK,mBAAmB,oBAAI,IAAI;AAAA,EAClC;AAAA,EAEA,CAAQ,OAAO,aAAa,IAAI;AAC9B,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,UAAU,8BAA8B;AAAA,IACpD;AAGA,QAAI,iBAAiB;AACrB,SAAK,SAAS;AAEd,WAAO;AAAA,MACL,MAAM,YAAY;AAChB,YAAI,KAAK,WAAW,gBAAgB;AAClC,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO;AAAA,UACT;AAAA,QACF;AAGA,eAAO,KAAK,MAAM,WAAW,GAAG;AAC9B,cAAI,KAAK,SAAS,KAAK,CAAC,KAAK,uBAAuB;AAClD,mBAAO;AAAA,cACL,MAAM;AAAA,cACN,OAAO;AAAA,YACT;AAAA,UACF;AAEA,cAAI,KAAK,SAAS;AAChB,6BAAiB;AAEjB,mBAAO;AAAA,cACL,MAAM;AAAA,cACN,OAAO,IAAI,kBAAkB;AAAA,YAC/B;AAAA,UACF;AAEA,cAAI,CAAC,KAAK,aAAa;AACrB,iBAAK,cAAc,IAAI,QAAc,CAAC,YAAY;AAChD,mBAAK,qBAAqB;AAAA,YAC5B,CAAC;AAAA,UACH;AAEA,gBAAM,KAAK;AACX,eAAK,cAAc;AACnB,eAAK,qBAAqB;AAAA,QAC5B;AAKA,cAAM,QAAQ,KAAK,MAAM,MAAM;AAE/B,eAAO,EAAE,MAAM,OAAO,MAAM;AAAA,MAC9B;AAAA,MACA,QAAQ,YAAY;AAClB,aAAK,MAAM;AACX,eAAO,EAAE,MAAM,MAAM,OAAO,OAAU;AAAA,MACxC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,UAAkD;AAC7D,UAAM,QAAuC,CAAC;AAC9C,qBAAiB,SAAS,MAAM;AAC9B,YAAM,KAAK,KAAK;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA,EAEO,QAAmB;AACxB,QAAI,KAAK,SAAS;AAChB;AAAA,IACF;AAEA,SAAK,SAAS;AACd,SAAK,UAAU;AACf,SAAK,wBAAwB,KAAK,MAAM,SAAS;AACjD,SAAK,MAAM,SAAS;AAEpB,SAAK,qBAAqB;AAAA,EAC5B;AAAA,EAEO,WAAoB;AACzB,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,WAAoB;AACzB,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,QAAQ,IAA4B;AACzC,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,0BAA0B;AAAA,IAC5C;AAEA,SAAK,iBAAiB,IAAI,EAAE;AAE5B,WAAO,MAAM;AACX,WAAK,iBAAiB,OAAO,EAAE;AAAA,IACjC;AAAA,EACF;AAAA,EAEO,eAAmC;AACxC,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,kDAAkD;AAAA,IACpE;AAEA,QAAI,CAAC,KAAK,gBAAgB;AACxB,WAAK,iBAAiB;AACtB,WAAK,qBAAqB;AAAA,IAC5B;AAEA,WAAO,IAAI,QAAmB,CAAC,YAAY;AACzC,WAAK,QAAQ,MAAM;AACjB,gBAAQ,MAAS;AAAA,MACnB,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAAA,EAEO,mBAA4B;AACjC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,UAAU,OAAgC;AAC/C,QAAI,KAAK,SAAS;AAChB;AAAA,IACF;AAEA,QAAI,KAAK,QAAQ;AACf,YAAM,IAAI,MAAM,8BAA8B;AAAA,IAChD;AAEA,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQO,eAA0B;AAC/B,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,mCAAmC;AAAA,IACrD;AAEA,SAAK,SAAS;AACd,SAAK,qBAAqB;AAC1B,SAAK,iBAAiB,QAAQ,CAAC,OAAO,GAAG,CAAC;AAC1C,SAAK,iBAAiB,MAAM;AAAA,EAI9B;AAAA;AAAA;AAAA;AAAA,EAKO,mBAA4B;AACjC,WAAO,KAAK,MAAM,SAAS;AAAA,EAC7B;AACF;AASO,IAAM,kBAAN,MAAmD;AAAA;AAAA;AAAA;AAAA,EAIhD;AAAA;AAAA;AAAA;AAAA,EAIA,SAAS;AAAA;AAAA;AAAA;AAAA,EAIT;AAAA;AAAA;AAAA;AAAA,EAIA,iBAAiB;AAAA;AAAA;AAAA;AAAA,EAIjB;AAAA,EAER,YAAY,SAA6B;AACvC,SAAK,UAAU;AACf,SAAK,mBAAmB,oBAAI,IAAI;AAChC,SAAK,0BAA0B,oBAAI,IAAI;AAAA,EACzC;AAAA,EAEO,MAAM,OAAqB;AAChC,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,+BAA+B;AAAA,IACjD;AAEA,SAAK,QAAQ,KAAK;AAAA,EACpB;AAAA,EAEO,WAAoB;AACzB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,QAAQ,IAA4B;AAClC,QAAI,KAAK,SAAS,GAAG;AACnB,SAAG;AAEH,aAAO,MAAM;AAAA,IACf;AAEA,SAAK,iBAAiB,IAAI,EAAE;AAE5B,WAAO,MAAM,KAAK,iBAAiB,OAAO,EAAE;AAAA,EAC9C;AAAA,EAEO,QAAmB;AACxB,QAAI,KAAK,SAAS,GAAG;AACnB;AAAA,IACF;AAEA,SAAK,SAAS;AACd,SAAK,iBAAiB,QAAQ,CAAC,OAAO,GAAG,CAAC;AAG1C,SAAK,iBAAiB,MAAM;AAC5B,SAAK,wBAAwB,MAAM;AACnC,SAAK,UAAU,MAAM;AAAA,EACvB;AAAA,EAEO,mBAA4B;AACjC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,eAAe,IAA4B;AAChD,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,0BAA0B;AAAA,IAC5C;AAEA,QAAI,KAAK,iBAAiB,GAAG;AAC3B,SAAG;AAEH,aAAO,MAAM;AAAA,IACf;AAEA,SAAK,wBAAwB,IAAI,EAAE;AAEnC,WAAO,MAAM,KAAK,wBAAwB,OAAO,EAAE;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,sBAAiC;AACtC,QAAI,KAAK,iBAAiB,GAAG;AAC3B,YAAM,IAAI,MAAM,6CAA6C;AAAA,IAC/D;AAEA,QAAI,KAAK,SAAS,GAAG;AACnB,YAAM,IAAI,MAAM,+CAA+C;AAAA,IACjE;AAEA,SAAK,iBAAiB;AACtB,SAAK,wBAAwB,QAAQ,CAAC,OAAO,GAAG,CAAC;AACjD,SAAK,wBAAwB,MAAM;AAAA,EACrC;AACF;;;ACzcA,SAA0C,QAAAC,aAAY;AAwC/C,IAAM,4BAA4B;AAKlC,IAAM,sBAAsB;AAK5B,IAAM,6BAA6B;AAInC,IAAM,uBAAuB;AAI7B,IAAM,aAAa;AAMnB,IAAM,0BAA0BA,MAAK,OAAO;AAAA,EACjD,MAAMA,MAAK,MAAM;AAAA,IACfA,MAAK,QAAQ,yBAAyB;AAAA,IACtCA,MAAK,QAAQ,mBAAmB;AAAA,IAChCA,MAAK,QAAQ,0BAA0B;AAAA,IACvCA,MAAK,QAAQ,oBAAoB;AAAA,IACjCA,MAAK,QAAQ,UAAU;AAAA,EACzB,CAAC;AAAA,EACD,SAASA,MAAK,OAAO;AACvB,CAAC;AAMM,IAAM,yBAAyBA,MAAK,OAAO;AAAA,EAChD,MAAMA,MAAK,MAAM;AAAA,IACfA,MAAK,QAAQ,mBAAmB;AAAA,IAChCA,MAAK,QAAQ,0BAA0B;AAAA,IACvCA,MAAK,QAAQ,oBAAoB;AAAA,IACjCA,MAAK,QAAQ,UAAU;AAAA,EACzB,CAAC;AAAA,EACD,SAASA,MAAK,OAAO;AACvB,CAAC;AAmND,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;AAqCA,SAAS,OAAO;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAaG;AACD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;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;AAqCA,SAAS,OAAO;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA,SAASA,MAAK,MAAM;AAAA,EACpB;AAAA,EACA;AACF,GAaG;AACD,SAAO;AAAA,IACL,MAAM;AAAA,IACN,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAMO,IAAM,YAAY;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;;;ACteA,SAAS,aAAa;AAStB,IAAM,wBAAwB,gBAAgB,uBAAuB;AA+GrE,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,gBAAgB,EAAE,GAAG,sBAAsB,GAAG,sBAAsB;AAC1E,MAAI,cAAc,gBAAgB;AAChC,cAAU,QAAQ,QAAQ;AAAA,EAC5B;AAEA,SAAO,sBAAsB,CAAC,SAAS;AACrC,UAAM,CAAC,aAAa,UAAU,UAAU,IAAI,CAAC,GAAG,KAAK,IAAI;AACzD,QAAI,EAAE,eAAe,YAAY,aAAa;AAC5C,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,CAAC,MAAM,WAAW,IAAI,KAAK;AAEjC,QAAI,cAAc,mBAAmB,CAAC,UAAU,SAAS,IAAI,QAAQ,GAAG;AACtE,gBAAU,QAAQ,QAAQ;AAAA,IAC5B;AAEA,QACE,eAAe,SACf,eAAe,eACf,eAAe,YACf,eAAe,UACf;AACA,YAAM,IAAI;AAAA,QACR,8CAA8C,UAAU;AAAA,MAC1D;AAAA,IACF;AAEA,WAAO;AAAA,MACL,eAAe,cAAc,iBAAiB;AAAA,MAC9C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,cAAe,YAA4B,SAAS;AAAA,IACtD;AAAA,EACF,GAAG,CAAC,CAAC;AACP;AAcA,SAAS,WACP,UACA,WACA,UACA,MACA,aACA,eACA,aACiC;AACjC,QAAM,qBAAqB,aAAa,SAAS,aAAa;AAE9D,QAAM,WAAW,WAAW;AAC5B,QAAM,EAAE,MAAM,IAAI,IAAI;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,MAAI,aAAa;AACjB,QAAM,cAAc,IAAI,gBAAqC,CAAC,UAAU;AACtE,cAAU,KAAK,UAAU;AAAA,MACvB;AAAA,MACA,SAAS;AAAA,MACT,cAAc;AAAA,MACd,SAAS,sBAAsB,GAAG;AAAA,IACpC,CAAC;AAAA,EACH,CAAC;AACD,cAAY,QAAQ,MAAM;AACxB,SAAK,SAAS,oBAAoB;AAElC,QAAI,CAAC,sBAAsB,YAAY;AACrC,gBAAU,KAAK,UAAU,mBAAmB,QAAQ,CAAC;AAAA,IACvD;AAEA,QAAI,aAAa,SAAS,GAAG;AAC3B,cAAQ;AAAA,IACV;AAAA,EACF,CAAC;AAED,QAAM,eAAe,IAAI,eAGvB,MAAM;AACN,cAAU,KAAK,UAAU,0BAA0B,QAAQ,CAAC;AAAA,EAC9D,CAAC;AACD,eAAa,QAAQ,MAAM;AACzB,SAAK,SAAS,qBAAqB;AAEnC,QAAI,YAAY,SAAS,GAAG;AAC1B,cAAQ;AAAA,IACV;AAAA,EACF,CAAC;AAED,WAAS,UAAU;AACjB,cAAU,oBAAoB,WAAW,SAAS;AAClD,cAAU,oBAAoB,iBAAiB,eAAe;AAC9D,iBAAa,oBAAoB,SAAS,aAAa;AACvD,SAAK,IAAI;AAAA,EACX;AAEA,WAAS,gBAAgB;AACvB,QAAI,aAAa,SAAS,KAAK,YAAY,SAAS,GAAG;AACrD;AAAA,IACF;AAEA,SAAK,SAAS,eAAe;AAE7B,iBAAa;AAEb,QAAI,CAAC,aAAa,SAAS,GAAG;AAC5B,mBAAa;AAAA,QACX,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AACA,mBAAa,aAAa;AAAA,IAC5B;AAEA,gBAAY,MAAM;AAClB,cAAU;AAAA,MACR;AAAA,MACA;AAAA,QACE;AAAA,QACA,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAEA,WAAS,UAAU,KAA6B;AAC9C,QAAI,IAAI,aAAa;AAAU;AAC/B,QAAI,IAAI,OAAO,UAAU,UAAU;AACjC,gBAAU,KAAK,MAAM,6CAA6C;AAAA,QAChE,UAAU,UAAU;AAAA,QACpB,kBAAkB;AAAA,MACpB,CAAC;AAED;AAAA,IACF;AAEA,QAAI,qBAAqB,IAAI,YAAY,GAAG;AAC1C,kBAAY,oBAAoB;AAAA,IAClC;AAEA,QAAI,cAAc,IAAI,YAAY,GAAG;AACnC,mBAAa;AAEb,WAAK,SAAS,gBAAgB;AAC9B,UAAI;AAEJ,UAAI,MAAM,MAAM,uBAAuB,IAAI,OAAO,GAAG;AACnD,sBAAc,IAAI;AAAA,MACpB,OAAO;AACL,sBAAc,IAAI;AAAA,UAChB,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AACD,kBAAU,KAAK;AAAA,UACb;AAAA,UACA;AAAA,YACE,UAAU,UAAU;AAAA,YACpB,kBAAkB;AAAA,YAClB,kBAAkB;AAAA,cAChB,GAAG,MAAM,OAAO,uBAAuB,IAAI,OAAO;AAAA,YACpD;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,UAAI,CAAC,aAAa,SAAS,GAAG;AAC5B,qBAAa,UAAU,WAAW;AAClC,qBAAa,aAAa;AAAA,MAC5B;AAEA,kBAAY,MAAM;AAElB;AAAA,IACF;AAEA,QAAI,aAAa,SAAS,GAAG;AAC3B,WAAK,gBAAgB,gDAAgD;AAErE,gBAAU,KAAK,MAAM,kDAAkD;AAAA,QACrE,UAAU,UAAU;AAAA,QACpB,kBAAkB;AAAA,MACpB,CAAC;AAED;AAAA,IACF;AAEA,QAAI,CAAC,MAAM,MAAM,2BAA2B,IAAI,OAAO,GAAG;AACxD,UAAI,MAAM,MAAM,iBAAiB,IAAI,OAAO,GAAG;AAC7C,qBAAa,UAAU,IAAI,OAAO;AAAA,MACpC,OAAO;AACL,kBAAU,KAAK;AAAA,UACb;AAAA,UACA;AAAA,YACE,UAAU,UAAU;AAAA,YACpB,kBAAkB;AAAA,YAClB,kBAAkB,CAAC,GAAG,MAAM,OAAO,iBAAiB,IAAI,OAAO,CAAC;AAAA,UAClE;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,QAAI,cAAc,IAAI,YAAY,GAAG;AACnC,WAAK,SAAS,uBAAuB;AAErC,mBAAa,aAAa;AAAA,IAC5B;AAAA,EACF;AAEA,WAAS,gBAAgB,KAAgC;AACvD,QAAI,IAAI,WAAW,cAAc;AAC/B;AAAA,IACF;AAEA,QAAI,IAAI,QAAQ,OAAO,UAAU;AAC/B;AAAA,IACF;AAEA,iBAAa;AACb,QAAI,CAAC,aAAa,SAAS,GAAG;AAC5B,mBAAa;AAAA,QACX,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS,GAAG,QAAQ;AAAA,QACtB,CAAC;AAAA,MACH;AAAA,IACF;AACA,gBAAY,MAAM;AAClB,iBAAa,aAAa;AAAA,EAC5B;AAEA,eAAa,iBAAiB,SAAS,aAAa;AACpD,YAAU,iBAAiB,WAAW,SAAS;AAC/C,YAAU,iBAAiB,iBAAiB,eAAe;AAE3D,YAAU,KAAK,UAAU;AAAA,IACvB;AAAA,IACA;AAAA,IACA;AAAA,IACA,SAAS,sBAAsB,GAAG;AAAA,IAClC,SAAS;AAAA,IACT,cAAc;AAAA,EAGhB,CAAC;AAED,MAAI,oBAAoB;AACtB,gBAAY,MAAM;AAAA,EACpB;AAEA,MAAI,aAAa,gBAAgB;AAC/B,WAAO;AAAA,EACT;AAEA,MAAI,aAAa,OAAO;AACtB,WAAO,iBAAiB,cAAc,UAAU,GAAG;AAAA,EACrD;AAEA,MAAI,aAAa,UAAU;AACzB,QAAI,cAAc;AAClB,WAAO;AAAA,MACL;AAAA,MACA,YAAY;AACV,YAAI,aAAa;AACf,gBAAM,IAAI,MAAM,iCAAiC;AAAA,QACnD;AAEA,sBAAc;AAEd,YAAI,CAAC,YAAY,SAAS,GAAG;AAC3B,sBAAY,MAAM;AAAA,QACpB;AAEA,eAAO,iBAAiB,cAAc,UAAU,GAAG;AAAA,MACrD;AAAA,IACF;AAAA,EACF;AAGA,SAAO,CAAC,aAAa,YAAY;AACnC;AAEA,eAAe,iBACb,cACA,KACuD;AACvD,QAAM,MAAM,MAAM,aAAa,QAAQ;AAEvC,MAAI,IAAI,SAAS,GAAG;AAClB,SAAK,MAAM,mDAAmD;AAAA,EAChE;AAEA,SAAO,IAAI,CAAC;AACd;;;ACvfA,SAAS,SAAAC,cAAa;AAItB,SAAe,sBAAsB;AAYrC,IAAM,uBAAuB,gBAAgB,sBAAsB;AAoCnE,IAAM,cAAN,MAEA;AAAA,EACU;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA;AAAA,EACA;AAAA,EAED;AAAA,EACA;AAAA,EAEP,YACE,WACA,UACA,kBACA,iBACA,uCAAuC,KACvC;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,mBAAmB,CAAC,CAAC;AAC1D,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,cAAc,oBAAI,IAAI;AAC3B,SAAK,uBAAuB,oBAAI,IAAI;AACpC,SAAK,uCACH;AACF,SAAK,MAAM,UAAU;AAErB,UAAM,gBAAgB,CAAC,QAA6B;AAClD,UAAI,IAAI,OAAO,KAAK,UAAU,UAAU;AACtC,aAAK,KAAK;AAAA,UACR;AAAA,UACA;AAAA,YACE,UAAU,KAAK,UAAU;AAAA,YACzB,kBAAkB;AAAA,UACpB;AAAA,QACF;AACA;AAAA,MACF;AAEA,UAAI,KAAK,YAAY,IAAI,IAAI,QAAQ,GAAG;AAEtC;AAAA,MACF;AAEA,UAAI,KAAK,qBAAqB,IAAI,IAAI,IAAI,GAAG,IAAI,IAAI,QAAQ,GAAG;AAC9D;AAAA,MACF;AAEA,YAAM,YAAY,KAAK,sBAAsB,GAAG;AAEhD,UAAI,CAAC,WAAW;AACd;AAAA,MACF;AAEA,WAAK,oBAAoB,SAAS;AAAA,IACpC;AACA,SAAK,UAAU,iBAAiB,WAAW,aAAa;AAExD,UAAM,sBAAsB,CAAC,QAAmC;AAC9D,UAAI,IAAI,WAAW;AAAc;AAEjC,YAAM,uBAAuB,IAAI,QAAQ;AACzC,WAAK,KAAK;AAAA,QACR,+BAA+B,oBAAoB;AAAA,QACnD,IAAI,QAAQ;AAAA,MACd;AAEA,WAAK,qBAAqB,OAAO,oBAAoB;AAAA,IACvD;AACA,SAAK,UAAU,iBAAiB,iBAAiB,mBAAmB;AAEpE,SAAK,UAAU,iBAAiB,mBAAmB,CAAC,QAAQ;AAC1D,UAAI,IAAI,WAAW;AAAU;AAE7B,WAAK,UAAU,oBAAoB,WAAW,aAAa;AAC3D,WAAK,UAAU,oBAAoB,iBAAiB,mBAAmB;AAAA,IACzE,CAAC;AAAA,EACH;AAAA,EAEQ,oBAAoB;AAAA,IAC1B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAAuB;AACrB,SAAK,YAAY,IAAI,QAAQ;AAE7B,QAAI,aAAa;AAEjB,UAAM,gBAAgB,CAAC,cAAmD;AACxE,UAAI,YAAY,SAAS,KAAK,aAAa,SAAS,GAAG;AAErD;AAAA,MACF;AAEA,mBAAa;AAEb,UAAI,CAAC,YAAY,SAAS,GAAG;AAC3B,oBAAY,UAAU,SAAS;AAC/B,oBAAY,aAAa;AAAA,MAC3B;AAEA,mBAAa,MAAM;AACnB,WAAK,YAAY,MAAM,UAAU,SAAS;AAAA,IAC5C;AAEA,UAAM,iBAAiB,MAAM;AAC3B;AAAA,QACE,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAAA,IACF;AACA,UAAM,yBAAyB,IAAI,gBAAgB;AACnD,2BAAuB,OAAO,iBAAiB,SAAS,cAAc;AAEtE,UAAM,wBAAwB,IAAI,gBAAgB;AAElD,UAAM,kBAAkB,CAAC,QAAmC;AAC1D,UAAI,IAAI,WAAW,cAAc;AAC/B;AAAA,MACF;AAEA,UAAI,IAAI,QAAQ,OAAO,MAAM;AAC3B;AAAA,MACF;AAEA,mBAAa;AAEb,YAAM,aAAa;AAAA,QACjB,MAAM;AAAA,QACN,SAAS;AAAA,MACX;AACA,UAAI,CAAC,YAAY,SAAS,GAAG;AAC3B,oBAAY,UAAU,IAAI,UAAU,CAAC;AACrC,oBAAY,aAAa;AAAA,MAC3B;AAEA,4BAAsB,MAAM,UAAU;AAEtC,mBAAa,MAAM;AAAA,IACrB;AACA,SAAK,UAAU,iBAAiB,iBAAiB,eAAe;AAEhE,UAAM,YAAY,CAAC,QAAgC;AACjD,UAAI,aAAa,IAAI,UAAU;AAC7B;AAAA,MACF;AAEA,UAAI,IAAI,SAAS,MAAM;AACrB,aAAK,KAAK,MAAM,6CAA6C;AAAA,UAC3D,GAAG;AAAA,UACH,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AAED;AAAA,MACF;AAEA,UAAI,qBAAqB,IAAI,YAAY,GAAG;AAC1C,qBAAa,oBAAoB;AAAA,MACnC;AAEA,UAAI,6BAA6B,IAAI,cAAc,eAAe,GAAG;AACnE,YAAI;AACJ,YAAIC,OAAM,MAAM,sBAAsB,IAAI,OAAO,GAAG;AAClD,wBAAc,IAAI;AAAA,QACpB,OAAO;AACL,wBAAc,IAAI;AAAA,YAChB,MAAM;AAAA,YACN,SAAS;AAAA,UACX,CAAC;AACD,eAAK,KAAK,KAAK,mDAAmD;AAAA,YAChE,GAAG;AAAA,YACH,UAAU,KAAK,UAAU;AAAA,YACzB,kBAAkB;AAAA,YAClB,kBAAkB;AAAA,cAChB,GAAGA,OAAM,OAAO,sBAAsB,IAAI,OAAO;AAAA,YACnD;AAAA,YACA,MAAM,CAAC,iBAAiB;AAAA,UAC1B,CAAC;AAAA,QACH;AAEA,YAAI,CAAC,YAAY,SAAS,GAAG;AAC3B,sBAAY,UAAU,WAAW;AACjC,sBAAY,aAAa;AAAA,QAC3B;AAEA,qBAAa,MAAM;AAEnB,8BAAsB,MAAM,YAAY,OAAO;AAE/C;AAAA,MACF;AAEA,UAAI,YAAY,SAAS,GAAG;AAC1B,aAAK,KAAK,KAAK,iDAAiD;AAAA,UAC9D,GAAG;AAAA,UACH,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,UAClB,MAAM,CAAC,iBAAiB;AAAA,QAC1B,CAAC;AAED;AAAA,UACE,IAAI;AAAA,YACF,MAAM;AAAA,YACN,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAEA;AAAA,MACF;AAEA,UAAI,WAAW,aAAaA,OAAM,MAAM,UAAU,OAAO,IAAI,OAAO,GAAG;AACrE,oBAAY,UAAU,GAAG,IAAI,OAAO,CAAC;AAAA,MACvC,WAAW,CAACA,OAAM,MAAM,6BAA6B,IAAI,OAAO,GAAG;AACjE,cAAM,mBAAmB;AAAA,UACvB,GAAGA,OAAM,OAAO,6BAA6B,IAAI,OAAO;AAAA,QAC1D;AACA,YAAI,aAAa;AACjB,YAAI,WAAW,WAAW;AACxB,uBACE;AACF,2BAAiB,KAAK,GAAGA,OAAM,OAAO,UAAU,OAAO,IAAI,OAAO,CAAC;AAAA,QACrE;AAEA,aAAK,KAAK,KAAK,YAAY;AAAA,UACzB,GAAG;AAAA,UACH,UAAU,KAAK,UAAU;AAAA,UACzB,kBAAkB;AAAA,UAClB;AAAA,UACA,MAAM,CAAC,iBAAiB;AAAA,QAC1B,CAAC;AAED;AAAA,UACE,IAAI;AAAA,YACF,MAAM;AAAA,YACN,SAAS;AAAA,UACX,CAAC;AAAA,QACH;AAAA,MACF;AAEA,UAAI,6BAA6B,IAAI,cAAc,eAAe,GAAG;AACnE,oBAAY,aAAa;AAAA,MAC3B;AAAA,IACF;AACA,SAAK,UAAU,iBAAiB,WAAW,SAAS;AAEpD,UAAM,sBAAyC,CAAC;AAChD,UAAM,UAAU,MAAM;AACpB,WAAK,UAAU,oBAAoB,WAAW,SAAS;AACvD,WAAK,UAAU,oBAAoB,iBAAiB,eAAe;AACnE,6BAAuB,OAAO,iBAAiB,SAAS,cAAc;AAEtE,WAAK,YAAY,OAAO,QAAQ;AAChC,0BAAoB,QAAQ,CAAC,OAAO;AAClC,YAAI;AACF,aAAG;AAAA,QACL,QAAQ;AAAA,QAER;AAAA,MACF,CAAC;AACD,0BAAoB,SAAS;AAAA,IAC/B;AAEA,UAAM,yBACJ,UAAU,SAAS,SAAS,UAAU,SAAS;AAEjD,UAAM,cAAc,IAAI,eAGtB,MAAM;AACN,WAAK,UAAU,KAAK,MAAM,0BAA0B,QAAQ,CAAC;AAAA,IAC/D,CAAC;AACD,gBAAY,QAAQ,MAAM;AAExB,UAAI,oBAAoB,QAAQ;AAG9B,YAAI,CAAC,0BAA0B,CAAC,aAAa,SAAS,GAAG;AACvD,uBAAa,MAAM;AAAA,QACrB;AAAA,MACF;AAEA,UAAI,aAAa,SAAS,GAAG;AAC3B,gBAAQ;AAAA,MACV;AAAA,IACF,CAAC;AAED,QAAI,mCAAmC;AACrC,kBAAY,UAAU,GAAG,WAAW,CAAC;AAAA,IACvC;AAEA,UAAM,eAAe,IAAI,gBAEvB,CAAC,aAAa;AACd,WAAK,UAAU,KAAK,MAAM;AAAA,QACxB;AAAA,QACA,cAAc,yBACV,8BAA8B,eAAe,IAC7C;AAAA,QACJ,SAAS;AAAA,MACX,CAAC;AAAA,IACH,CAAC;AACD,iBAAa,QAAQ,MAAM;AACzB,UAAI,CAAC,0BAA0B,YAAY;AAIzC,cAAM,UAAU,mBAAmB,QAAQ;AAE3C,gBAAQ,eAAe,8BAA8B,eAAe;AAEpE,aAAK,UAAU,KAAK,MAAM,mBAAmB,QAAQ,CAAC;AAAA,MACxD;AAGA,UAAI,oBAAoB,QAAQ;AAE9B,YAAI,CAAC,YAAY,SAAS,GAAG;AAC3B,sBAAY,aAAa;AAAA,QAC3B;AAAA,MACF;AAEA,UAAI,YAAY,SAAS,GAAG;AAC1B,gBAAQ;AAAA,MACV;AAAA,IACF,CAAC;AAED,UAAM,iBAAiB,CAAC,KAAc,SAAe;AACnD,YAAM,WAAW,kBAAkB,GAAG;AAEtC,WAAK,gBAAgB,eAAe,QAAQ,MAAM,IAAI,MAAM,QAAQ,CAAC;AACrE,WAAK,UAAU,EAAE,MAAM,eAAe,MAAM,CAAC;AAE7C;AAAA,QACE,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAAA,IACF;AAEA,QAAI,6BAA6B,cAAc,eAAe,GAAG;AAC/D,kBAAY,aAAa;AAAA,IAC3B,WAAW,UAAU,SAAS,SAAS,UAAU,SAAS,gBAAgB;AAGxE,WAAK,KAAK,KAAK,GAAG,UAAU,IAAI,wCAAwC;AAAA,QACtE,GAAG;AAAA,QACH,UAAU,KAAK,UAAU;AAAA,MAC3B,CAAC;AAAA,IACH;AAEA,UAAM,kCAAmE;AAAA,MACvE,GAAG,KAAK,WAAW,SAAS,WAAW;AAAA,MACvC;AAAA,MACA;AAAA,MACA,UAAU;AAAA,MACV,iBAAiB;AAAA,MACjB,mBAAmB,sBAAsB;AAAA,MACzC,mBAAmB,CAAC,OAAO;AACzB,YAAI,YAAY,SAAS,KAAK,aAAa,SAAS,GAAG;AAErD,cAAI;AACF,eAAG;AAAA,UACL,QAAQ;AAAA,UAER;AAEA;AAAA,QACF;AAEA,4BAAoB,KAAK,EAAE;AAAA,MAC7B;AAAA,IACF;AAEA,YAAQ,UAAU,MAAM;AAAA,MACtB,KAAK;AACH,aAAK;AAAA,UACH,UAAU;AAAA,UACV;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,OAAO,SAA6B;AAClC,gBAAI;AAGF,oBAAM,gBAAgB,MAAM,UAAU;AAAA,gBACpC;AAAA,gBACA;AAAA,cACF;AAEA,kBAAI,aAAa,SAAS,GAAG;AAE3B;AAAA,cACF;AAEA,2BAAa,MAAM,aAAa;AAChC,2BAAa,MAAM;AAAA,YACrB,SAAS,KAAK;AACZ,6BAAe,KAAK,IAAI;AAAA,YAC1B,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF,KAAK;AACH,aAAK;AAAA,UACH,UAAU;AAAA,UACV;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,OAAO,SAA6B;AAClC,gBAAI;AAGF,oBAAM,UAAU;AAAA,gBACd;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACF;AAAA,YACF,SAAS,KAAK;AACZ,6BAAe,KAAK,IAAI;AAAA,YAC1B,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AAEA;AAAA,MACF,KAAK;AACH,aAAK;AAAA,UACH,UAAU;AAAA,UACV;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,OAAO,SAA6B;AAClC,gBAAI;AAGF,oBAAM,UAAU;AAAA,gBACd;AAAA,gBACA,oCAAoC,OAAO;AAAA,gBAC3C;AAAA,cACF;AAAA,YACF,SAAS,KAAK;AACZ,6BAAe,KAAK,IAAI;AAAA,YAC1B,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AACA;AAAA,MACF,KAAK;AACH,aAAK;AAAA,UACH,UAAU;AAAA,UACV;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,OAAO,SAA6B;AAClC,gBAAI;AAGF,oBAAM,gBAAgB,MAAM,UAAU;AAAA,gBACpC;AAAA,gBACA,oCAAoC,OAAO;AAAA,gBAC3C;AAAA,cACF;AAEA,kBAAI,aAAa,SAAS,GAAG;AAE3B;AAAA,cACF;AACA,2BAAa,MAAM,aAAa;AAChC,2BAAa,MAAM;AAAA,YACrB,SAAS,KAAK;AACZ,6BAAe,KAAK,IAAI;AAAA,YAC1B,UAAE;AACA,mBAAK,IAAI;AAAA,YACX;AAAA,UACF;AAAA,QACF;AAEA;AAAA,MACF;AACE,aAAK,KAAK;AAAA,UACR,0CACG,UAA2B,IAC9B,OAAO,WAAW,IAAI,aAAa;AAAA,UACnC;AAAA,YACE,GAAG;AAAA,YACH,MAAM,CAAC,qBAAqB;AAAA,UAC9B;AAAA,QACF;AAEA;AAAA,IACJ;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,EAEQ,sBACN,aAC2B;AAC3B,UAAM,UAAU,KAAK,UAAU,SAAS,IAAI,YAAY,IAAI;AAC5D,QAAI,CAAC,SAAS;AACZ,YAAM,aAAa,+BAA+B,YAAY,IAAI;AAClE,WAAK,KAAK,MAAM,6BAA6B,YAAY,IAAI,IAAI;AAAA,QAC/D,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,UAAM,kBAAkB,KAAK,UAAU,yBAAyB;AAAA,MAC9D,QAAQ;AAAA,IACV;AACA,QAAI,CAAC,iBAAiB;AACpB,YAAM,aAAa;AACnB,WAAK,KAAK,MAAM,YAAY;AAAA,QAC1B,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,aAAa,YAAY,YAAY,GAAG;AAC3C,YAAM,aAAa;AACnB,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,YAAY,aAAa;AAC5B,YAAM,aAAa;AACnB,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,YAAY,eAAe;AAC9B,YAAM,aAAa;AACnB,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,QAAI,EAAE,YAAY,eAAe,KAAK,WAAW;AAC/C,YAAM,aAAa,yBAAyB,YAAY,WAAW;AACnE,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,UAAM,UAAU,KAAK,SAAS,YAAY,WAAW;AACrD,QAAI,EAAE,YAAY,iBAAiB,QAAQ,aAAa;AACtD,YAAM,aAAa,0CAA0C,YAAY,WAAW,IAAI,YAAY,aAAa;AACjH,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,UAAM,YAAY,QAAQ,WAAW,YAAY,aAAa;AAE9D,QAAI,oCAAoC;AACxC,QACE,QAAQ,oBAAoB,WAC3B,UAAU,SAAS,YAAY,UAAU,SAAS,aACnDA,OAAM,MAAM,UAAU,OAAO,YAAY,OAAO,KAChDA,OAAM,MAAM,UAAU,MAAM,CAAC,CAAC,GAC9B;AAQA,0CAAoC;AAAA,IACtC,WAAW,CAACA,OAAM,MAAM,UAAU,MAAM,YAAY,OAAO,GAAG;AAC5D,YAAM,aAAa;AACnB,WAAK,KAAK,KAAK,YAAY;AAAA,QACzB,GAAG,QAAQ;AAAA,QACX,UAAU,KAAK,UAAU;AAAA,QACzB,kBAAkB;AAAA,QAClB,MAAM,CAAC,iBAAiB;AAAA,MAC1B,CAAC;AAED,WAAK;AAAA,QACH,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,IAAI;AAAA,UACF,MAAM;AAAA,UACN,SAAS;AAAA,QACX,CAAC;AAAA,MACH;AAEA,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA,eAAe,YAAY;AAAA,MAC3B;AAAA,MACA,aAAa,YAAY;AAAA,MACzB,iBAAiB;AAAA,QACf,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,MACpB;AAAA,MACA,UAAU,YAAY;AAAA,MACtB,cAAc,YAAY;AAAA,MAC1B,YAAY,YAAY;AAAA,MACxB,aAAa,YAAY;AAAA,MACzB,MAAM,YAAY;AAAA,MAClB,WAAW,QAAQ;AAAA,MACnB,iBAAiB,QAAQ;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,YACE,IACA,UACA,SACA;AACA,QAAI,oBAAoB,KAAK,qBAAqB,IAAI,EAAE;AAExD,QAAI,CAAC,mBAAmB;AACtB,0BAAoB,IAAI,OAAO,KAAK,oCAAoC;AAExE,WAAK,qBAAqB,IAAI,IAAI,iBAAiB;AAAA,IACrD;AAEA,sBAAkB,IAAI,QAAQ;AAE9B,SAAK,UAAU;AAAA,MACb;AAAA;AAAA,MAEA,KAAK,UAAU,SAAS,IAAI,EAAE,GAAG,oBAAoB,SACjD,mBAAmB,QAAQ,IAC3B,aAAa,UAAU,OAAO;AAAA,IACpC;AAAA,EACF;AACF;AAEA,IAAM,SAAN,MAAa;AAAA,EACH;AAAA,EACA;AAAA,EAER,YAAY,UAAkB;AAC5B,SAAK,QAAQ,oBAAI,IAAI;AACrB,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,IAAI,MAAc;AAChB,QAAI,KAAK,MAAM,IAAI,IAAI,GAAG;AACxB,WAAK,MAAM,OAAO,IAAI;AAAA,IACxB,WAAW,KAAK,MAAM,QAAQ,KAAK,UAAU;AAC3C,YAAM,QAAQ,KAAK,MAAM,OAAO,EAAE,KAAK;AACvC,UAAI,CAAC,MAAM,MAAM;AACf,aAAK,MAAM,OAAO,MAAM,KAAK;AAAA,MAC/B;AAAA,IACF;AACA,SAAK,MAAM,IAAI,IAAI;AAAA,EACrB;AAAA,EAEA,IAAI,MAAc;AAChB,WAAO,KAAK,MAAM,IAAI,IAAI;AAAA,EAC5B;AACF;AAGA,SAAS,6BACP,cACA,iBACA;AACA,MAAI,oBAAoB,QAAQ;AAE9B,WAAO;AAAA,EACT;AAEA,SAAO,cAAc,YAAY;AACnC;AAGA,SAAS,6BACP,cACA,iBACA;AACA,MAAI,oBAAoB,QAAQ;AAE9B,WAAO,cAAc,YAAY;AAAA,EACnC;AAEA,SAAO,cAAc,YAAY;AACnC;AAGA,SAAS,8BAA8B,iBAAyB;AAC9D,MAAI,oBAAoB,QAAQ;AAE9B;AAAA,EACF;AAEA;AACF;AAWO,SAAS,aACd,WACA,UACA,uBASkB;AAClB,SAAO,IAAI;AAAA,IACT;AAAA,IACA;AAAA,IACA,uBAAuB;AAAA,IACvB,uBAAuB;AAAA,IACvB,uBAAuB;AAAA,EACzB;AACF;;;AC15BO,SAAS,6BAGd,QACA,WACwB;AACxB,SAAO,EAAE,QAAQ,UAAU;AAC7B;AAEO,SAAS,6BAGd,QACA,UACwB;AACxB,SAAO,EAAE,QAAQ,SAAiD;AACpE;","names":["Type","Type","Value","Value"]}
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../transport/events.ts","../transport/transport.ts","../transport/connection.ts"],"sourcesContent":["import { Connection } from './connection';\nimport { OpaqueTransportMessage } from './message';\nimport { Session, SessionState } from './sessionStateMachine';\nimport { TransportStatus } from './transport';\n\nexport const ProtocolError = {\n RetriesExceeded: 'conn_retry_exceeded',\n HandshakeFailed: 'handshake_failed',\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 sessionStatus: {\n status: 'connect' | 'disconnect';\n session: Session<Connection>;\n };\n sessionTransition:\n | { state: SessionState.Connected }\n | { state: SessionState.Handshaking }\n | { state: SessionState.Connecting }\n | { state: SessionState.NoConnection };\n protocolError: {\n type: ProtocolErrorType;\n message: string;\n };\n transportStatus: {\n status: TransportStatus;\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 removeAllListeners() {\n this.eventListeners = {};\n }\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 // copying ensures that adding more listeners in a handler doesn't\n // affect the current dispatch.\n const copy = [...handlers];\n for (const handler of copy) {\n handler(event);\n }\n }\n }\n}\n","import {\n OpaqueTransportMessage,\n TransportClientId,\n PartialTransportMessage,\n} from './message';\nimport {\n BaseLogger,\n LogFn,\n Logger,\n LoggingLevel,\n createLogProxy,\n} from '../logging/log';\nimport {\n EventDispatcher,\n EventHandler,\n EventMap,\n EventTypes,\n ProtocolErrorType,\n} from './events';\nimport {\n ProvidedTransportOptions,\n TransportOptions,\n defaultTransportOptions,\n} from './options';\nimport {\n Session,\n SessionConnected,\n SessionConnecting,\n SessionHandshaking,\n SessionNoConnection,\n SessionState,\n SessionStateGraph,\n} from './sessionStateMachine';\nimport { Connection } from './connection';\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 permanently closed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed';\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions\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 * The status of the transport.\n */\n private status: TransportStatus;\n\n /**\n * The client ID of this transport.\n */\n clientId: TransportClientId;\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 sessions: Map<TransportClientId, Session<ConnType>>;\n\n /**\n * Creates a new Transport instance.\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.clientId = clientId;\n this.status = 'open';\n this.sessions = new Map();\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 * 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) {\n if (this.getStatus() !== 'open') return;\n this.eventDispatcher.dispatchEvent('message', msg);\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 abstract send(to: TransportClientId, msg: PartialTransportMessage): string;\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.status = 'closed';\n\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n this.eventDispatcher.dispatchEvent('transportStatus', {\n status: this.status,\n });\n\n this.eventDispatcher.removeAllListeners();\n\n this.log?.info(`manually closed transport`, { clientId: this.clientId });\n }\n\n getStatus(): TransportStatus {\n return this.status;\n }\n\n protected updateSession<S extends Session<ConnType>>(session: S): S {\n const activeSession = this.sessions.get(session.to);\n if (activeSession && activeSession.id !== session.id) {\n const msg = `attempt to transition active session for ${session.to} but active session (${activeSession.id}) is different from handle (${session.id})`;\n throw new Error(msg);\n }\n\n this.sessions.set(session.to, session);\n\n if (!activeSession) {\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'connect',\n session: session,\n });\n }\n\n this.eventDispatcher.dispatchEvent('sessionTransition', {\n state: session.state,\n session: session,\n } as EventMap['sessionTransition']);\n\n return session;\n }\n\n // state transitions\n protected deleteSession(session: Session<ConnType>) {\n session.log?.info(`closing session ${session.id}`, session.loggingMetadata);\n\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'disconnect',\n session: session,\n });\n\n session.close();\n this.sessions.delete(session.to);\n }\n\n // common listeners\n protected onSessionGracePeriodElapsed(session: SessionNoConnection) {\n this.log?.warn(\n `session to ${session.to} grace period elapsed, closing`,\n session.loggingMetadata,\n );\n\n this.deleteSession(session);\n }\n\n protected onConnectingFailed(\n session: SessionConnecting<ConnType>,\n ): SessionNoConnection {\n // transition to no connection\n const noConnectionSession =\n SessionStateGraph.transition.ConnectingToNoConnection(session, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n\n return this.updateSession(noConnectionSession);\n }\n\n protected onConnClosed(\n session: SessionHandshaking<ConnType> | SessionConnected<ConnType>,\n ): SessionNoConnection {\n // transition to no connection\n let noConnectionSession: SessionNoConnection;\n if (session.state === SessionState.Handshaking) {\n noConnectionSession =\n SessionStateGraph.transition.HandshakingToNoConnection(session, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n } else {\n noConnectionSession =\n SessionStateGraph.transition.ConnectedToNoConnection(session, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n }\n\n return this.updateSession(noConnectionSession);\n }\n}\n","import { TelemetryInfo } from '../tracing';\nimport { MessageMetadata } from '../logging';\nimport { generateId } from './id';\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-${generateId()}`; // 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\n if (this.telemetry?.span.isRecording() && spanContext) {\n metadata.telemetry = {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n };\n }\n\n return metadata;\n }\n\n // can't use event emitter because we need this to work in both node + browser\n private _dataListeners = new Set<(msg: Uint8Array) => void>();\n private _closeListeners = new Set<() => void>();\n private _errorListeners = new Set<(err: Error) => void>();\n\n get dataListeners() {\n return [...this._dataListeners];\n }\n\n get closeListeners() {\n return [...this._closeListeners];\n }\n\n get errorListeners() {\n return [...this._errorListeners];\n }\n\n /**\n * Handle adding a callback for when a message is received.\n * @param msg The message that was received.\n */\n addDataListener(cb: (msg: Uint8Array) => void) {\n this._dataListeners.add(cb);\n }\n\n removeDataListener(cb: (msg: Uint8Array) => void): void {\n this._dataListeners.delete(cb);\n }\n\n /**\n * Handle adding a callback for when the connection is closed.\n * This should also be called if an error happens and after notifying all the error listeners.\n * @param cb The callback to call when the connection is closed.\n */\n addCloseListener(cb: () => void): void {\n this._closeListeners.add(cb);\n }\n\n removeCloseListener(cb: () => void): void {\n this._closeListeners.delete(cb);\n }\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 addErrorListener(cb: (err: Error) => void): void {\n this._errorListeners.add(cb);\n }\n\n removeErrorListener(cb: (err: Error) => void): void {\n this._errorListeners.delete(cb);\n }\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"],"mappings":";;;;;;;;;;;;;AAKO,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AA8BO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,qBAAqB;AACnB,SAAK,iBAAiB,CAAC;AAAA,EACzB;AAAA,EAEA,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;AAGZ,YAAM,OAAO,CAAC,GAAG,QAAQ;AACzB,iBAAW,WAAW,MAAM;AAC1B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;ACbO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA,EAInD;AAAA;AAAA;AAAA;AAAA,EAKR;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA,EACV;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW;AAChB,SAAK,SAAS;AACd,SAAK,WAAW,oBAAI,IAAI;AAAA,EAC1B;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,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU,MAAM;AAAQ;AACjC,SAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,EACnD;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,EAUU,cAAc,MAAyB,SAAiB;AAChE,SAAK,gBAAgB,cAAc,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAAA,EACvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,SAAS;AAEd,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,gBAAgB,cAAc,mBAAmB;AAAA,MACpD,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,SAAK,gBAAgB,mBAAmB;AAExC,SAAK,KAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACzE;AAAA,EAEA,YAA6B;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEU,cAA2C,SAAe;AAClE,UAAM,gBAAgB,KAAK,SAAS,IAAI,QAAQ,EAAE;AAClD,QAAI,iBAAiB,cAAc,OAAO,QAAQ,IAAI;AACpD,YAAM,MAAM,4CAA4C,QAAQ,EAAE,wBAAwB,cAAc,EAAE,+BAA+B,QAAQ,EAAE;AACnJ,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AAErC,QAAI,CAAC,eAAe;AAClB,WAAK,gBAAgB,cAAc,iBAAiB;AAAA,QAClD,QAAQ;AAAA,QACR;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,gBAAgB,cAAc,qBAAqB;AAAA,MACtD,OAAO,QAAQ;AAAA,MACf;AAAA,IACF,CAAkC;AAElC,WAAO;AAAA,EACT;AAAA;AAAA,EAGU,cAAc,SAA4B;AAClD,YAAQ,KAAK,KAAK,mBAAmB,QAAQ,EAAE,IAAI,QAAQ,eAAe;AAE1E,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,MAAM;AACd,SAAK,SAAS,OAAO,QAAQ,EAAE;AAAA,EACjC;AAAA;AAAA,EAGU,4BAA4B,SAA8B;AAClE,SAAK,KAAK;AAAA,MACR,cAAc,QAAQ,EAAE;AAAA,MACxB,QAAQ;AAAA,IACV;AAEA,SAAK,cAAc,OAAO;AAAA,EAC5B;AAAA,EAEU,mBACR,SACqB;AAErB,UAAM,sBACJ,kBAAkB,WAAW,yBAAyB,SAAS;AAAA,MAC7D,6BAA6B,MAAM;AACjC,aAAK,4BAA4B,mBAAmB;AAAA,MACtD;AAAA,IACF,CAAC;AAEH,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AAAA,EAEU,aACR,SACqB;AAErB,QAAI;AACJ,QAAI,QAAQ,2CAAoC;AAC9C,4BACE,kBAAkB,WAAW,0BAA0B,SAAS;AAAA,QAC9D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL,OAAO;AACL,4BACE,kBAAkB,WAAW,wBAAwB,SAAS;AAAA,QAC5D,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,mBAAmB;AAAA,QACtD;AAAA,MACF,CAAC;AAAA,IACL;AAEA,WAAO,KAAK,cAAc,mBAAmB;AAAA,EAC/C;AACF;;;ACnQO,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQ,WAAW,CAAC;AAAA,EAChC;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,WAA4B,EAAE,QAAQ,KAAK,GAAG;AACpD,UAAM,cAAc,KAAK,WAAW,KAAK,YAAY;AAErD,QAAI,KAAK,WAAW,KAAK,YAAY,KAAK,aAAa;AACrD,eAAS,YAAY;AAAA,QACnB,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGQ,iBAAiB,oBAAI,IAA+B;AAAA,EACpD,kBAAkB,oBAAI,IAAgB;AAAA,EACtC,kBAAkB,oBAAI,IAA0B;AAAA,EAExD,IAAI,gBAAgB;AAClB,WAAO,CAAC,GAAG,KAAK,cAAc;AAAA,EAChC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA,EAEA,IAAI,iBAAiB;AACnB,WAAO,CAAC,GAAG,KAAK,eAAe;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAgB,IAA+B;AAC7C,SAAK,eAAe,IAAI,EAAE;AAAA,EAC5B;AAAA,EAEA,mBAAmB,IAAqC;AACtD,SAAK,eAAe,OAAO,EAAE;AAAA,EAC/B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBAAiB,IAAsB;AACrC,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAsB;AACxC,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,iBAAiB,IAAgC;AAC/C,SAAK,gBAAgB,IAAI,EAAE;AAAA,EAC7B;AAAA,EAEA,oBAAoB,IAAgC;AAClD,SAAK,gBAAgB,OAAO,EAAE;AAAA,EAChC;AAaF;","names":[]}
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../transport/options.ts","../transport/sessionStateMachine/common.ts","../transport/sessionStateMachine/SessionConnecting.ts","../transport/sessionStateMachine/SessionNoConnection.ts","../transport/sessionStateMachine/SessionWaitingForHandshake.ts","../transport/sessionStateMachine/SessionHandshaking.ts","../transport/sessionStateMachine/SessionConnected.ts","../transport/sessionStateMachine/transitions.ts"],"sourcesContent":["import { NaiveJsonCodec } from '../codec/json';\nimport { ConnectionRetryOptions } from './rateLimit';\nimport { SessionOptions } from './sessionStateMachine/common';\n\nexport type TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n heartbeatIntervalMs: 1_000,\n heartbeatsUntilDead: 2,\n sessionDisconnectGraceMs: 5_000,\n connectionTimeoutMs: 2_000,\n handshakeTimeoutMs: 1_000,\n codec: NaiveJsonCodec,\n};\n\nexport type ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n baseIntervalMs: 250,\n maxJitterMs: 200,\n maxBackoffMs: 32_000,\n attemptBudgetCapacity: 5,\n budgetRestoreIntervalMs: 200,\n};\n\nexport const defaultClientTransportOptions: ClientTransportOptions = {\n ...defaultTransportOptions,\n ...defaultConnectionRetryOptions,\n};\n\nexport type ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nexport const defaultServerTransportOptions: ServerTransportOptions = {\n ...defaultTransportOptions,\n};\n","import { Logger, MessageMetadata } from '../../logging';\nimport { TelemetryInfo } from '../../tracing';\nimport {\n OpaqueTransportMessage,\n OpaqueTransportMessageSchema,\n PartialTransportMessage,\n TransportClientId,\n TransportMessage,\n} from '../message';\nimport { Value } from '@sinclair/typebox/value';\nimport { SessionNoConnection } from './SessionNoConnection';\nimport { SessionConnecting } from './SessionConnecting';\nimport { SessionHandshaking } from './SessionHandshaking';\nimport { SessionConnected } from './SessionConnected';\nimport { Codec } from '../../codec';\nimport { Connection } from '../connection';\nimport { generateId } from '../id';\n\nexport const enum SessionState {\n NoConnection = 'NoConnection',\n Connecting = 'Connecting',\n Handshaking = 'Handshaking',\n Connected = 'Connected',\n WaitingForHandshake = 'WaitingForHandshake',\n}\n\nexport type Session<ConnType extends Connection> =\n | SessionNoConnection\n | SessionConnecting<ConnType>\n | SessionHandshaking<ConnType>\n | SessionConnected<ConnType>;\n\nexport const ERR_CONSUMED = `session state has been consumed and is no longer valid`;\n\nabstract class StateMachineState {\n abstract readonly state: SessionState;\n\n /*\n * Whether this state has been consumed\n * and we've moved on to another state\n */\n _isConsumed: boolean;\n\n // called when we're transitioning to another state\n // note that this is internal and should not be called directly\n // by consumers, the proxy will call this when the state is consumed\n // and we're transitioning to another state\n abstract _handleStateExit(): void;\n\n // called when we exit the state machine entirely\n // note that this is internal and should not be called directly\n // by consumers, the proxy will call this when .close is closed\n abstract _handleClose(): void;\n\n close(): void {\n this._handleClose();\n }\n\n constructor() {\n this._isConsumed = false;\n\n // proxy helps us prevent access to properties after the state has been consumed\n // e.g. if we hold a reference to a state and try to access it after it's been consumed\n // we intercept the access and throw an error to help catch bugs\n return new Proxy(this, {\n get(target, prop) {\n // always allow access to _isConsumed, id, and state\n if (prop === '_isConsumed' || prop === 'id' || prop === 'state') {\n return Reflect.get(target, prop);\n }\n\n // modify _handleStateExit\n if (prop === '_handleStateExit') {\n return () => {\n target._isConsumed = true;\n target._handleStateExit();\n };\n }\n\n // modify _handleClose\n if (prop === '_handleClose') {\n return () => {\n target._handleStateExit();\n target._handleClose();\n };\n }\n\n if (target._isConsumed) {\n throw new Error(\n `${ERR_CONSUMED}: getting ${prop.toString()} on consumed state`,\n );\n }\n\n return Reflect.get(target, prop);\n },\n set(target, prop, value) {\n if (target._isConsumed) {\n throw new Error(\n `${ERR_CONSUMED}: setting ${prop.toString()} on consumed state`,\n );\n }\n\n return Reflect.set(target, prop, value);\n },\n });\n }\n}\n\nexport interface SessionOptions {\n /**\n * Frequency at which to send heartbeat acknowledgements\n */\n heartbeatIntervalMs: number;\n /**\n * Number of elapsed heartbeats without a response message before we consider\n * the connection dead.\n */\n heartbeatsUntilDead: number;\n /**\n * Duration to wait between connection disconnect and actual session disconnect\n */\n sessionDisconnectGraceMs: number;\n /**\n * Connection timeout in milliseconds\n */\n connectionTimeoutMs: number;\n /**\n * Handshake timeout in milliseconds\n */\n handshakeTimeoutMs: number;\n /**\n * The codec to use for encoding/decoding messages over the wire\n */\n codec: Codec;\n}\n\n// all session states have a from and options\nexport abstract class CommonSession extends StateMachineState {\n readonly from: TransportClientId;\n readonly options: SessionOptions;\n\n log?: Logger;\n abstract get loggingMetadata(): MessageMetadata;\n\n constructor(\n from: TransportClientId,\n options: SessionOptions,\n log: Logger | undefined,\n ) {\n super();\n this.from = from;\n this.options = options;\n this.log = log;\n }\n\n parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n const parsedMsg = this.options.codec.fromBuffer(msg);\n\n if (parsedMsg === null) {\n const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n this.log?.error(\n `received malformed msg: ${decodedBuffer}`,\n this.loggingMetadata,\n );\n return null;\n }\n\n if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n this.log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n ...this.loggingMetadata,\n validationErrors: [\n ...Value.Errors(OpaqueTransportMessageSchema, parsedMsg),\n ],\n });\n\n return null;\n }\n\n return parsedMsg;\n }\n}\n\nexport type InheritedProperties = Pick<\n IdentifiedSession,\n 'id' | 'from' | 'to' | 'seq' | 'ack' | 'sendBuffer' | 'telemetry' | 'options'\n>;\n\nexport type SessionId = string;\n\n// all sessions where we know the other side's client id\nexport abstract class IdentifiedSession extends CommonSession {\n readonly id: SessionId;\n readonly telemetry: TelemetryInfo;\n readonly to: TransportClientId;\n readonly protocolVersion: string;\n\n /**\n * Index of the message we will send next (excluding handshake)\n */\n seq: number;\n\n /**\n * Number of unique messages we've received this session (excluding handshake)\n */\n ack: number;\n sendBuffer: Array<OpaqueTransportMessage>;\n\n constructor(\n id: SessionId,\n from: TransportClientId,\n to: TransportClientId,\n seq: number,\n ack: number,\n sendBuffer: Array<OpaqueTransportMessage>,\n telemetry: TelemetryInfo,\n options: SessionOptions,\n protocolVersion: string,\n log: Logger | undefined,\n ) {\n super(from, options, log);\n this.id = id;\n this.to = to;\n this.seq = seq;\n this.ack = ack;\n this.sendBuffer = sendBuffer;\n this.telemetry = telemetry;\n this.log = log;\n this.protocolVersion = protocolVersion;\n }\n\n get loggingMetadata(): MessageMetadata {\n const spanContext = this.telemetry.span.spanContext();\n\n return {\n clientId: this.from,\n connectedTo: this.to,\n sessionId: this.id,\n telemetry: {\n traceId: spanContext.traceId,\n spanId: spanContext.spanId,\n },\n };\n }\n\n constructMsg<Payload>(\n partialMsg: PartialTransportMessage<Payload>,\n ): TransportMessage<Payload> {\n const msg = {\n ...partialMsg,\n id: generateId(),\n to: this.to,\n from: this.from,\n seq: this.seq,\n ack: this.ack,\n };\n\n this.seq++;\n return msg;\n }\n\n nextSeq(): number {\n return this.sendBuffer.length > 0 ? this.sendBuffer[0].seq : this.seq;\n }\n\n send(msg: PartialTransportMessage): string {\n const constructedMsg = this.constructMsg(msg);\n this.sendBuffer.push(constructedMsg);\n return constructedMsg.id;\n }\n\n _handleStateExit(): void {\n // noop\n }\n\n _handleClose(): void {\n // zero out the buffer\n this.sendBuffer.length = 0;\n this.telemetry.span.end();\n }\n}\n","import { Connection } from '../connection';\nimport { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionConnectingListeners {\n onConnectionEstablished: (conn: Connection) => void;\n onConnectionFailed: (err: unknown) => void;\n\n // timeout related\n onConnectionTimeout: () => void;\n}\n\n/*\n * A session that is connecting but we don't have access to the raw connection\n * yet.\n *\n * Valid transitions:\n * - Connecting -> NoConnection (timeout)\n * - Connecting -> Handshaking (on connection established)\n */\nexport class SessionConnecting<\n ConnType extends Connection,\n> extends IdentifiedSession {\n readonly state = SessionState.Connecting as const;\n connPromise: Promise<ConnType>;\n listeners: SessionConnectingListeners;\n\n connectionTimeout?: ReturnType<typeof setTimeout>;\n\n constructor(\n connPromise: Promise<ConnType>,\n listeners: SessionConnectingListeners,\n ...args: ConstructorParameters<typeof IdentifiedSession>\n ) {\n super(...args);\n this.connPromise = connPromise;\n this.listeners = listeners;\n\n this.connectionTimeout = setTimeout(() => {\n listeners.onConnectionTimeout();\n }, this.options.connectionTimeoutMs);\n\n connPromise.then(\n (conn) => {\n if (this._isConsumed) return;\n listeners.onConnectionEstablished(conn);\n },\n (err) => {\n if (this._isConsumed) return;\n listeners.onConnectionFailed(err);\n },\n );\n }\n\n // close a pending connection if it resolves, ignore errors if the promise\n // ends up rejected anyways\n bestEffortClose() {\n void this.connPromise\n .then((conn) => conn.close())\n .catch(() => {\n // ignore errors\n });\n }\n\n _handleStateExit(): void {\n super._handleStateExit();\n clearTimeout(this.connectionTimeout);\n this.connectionTimeout = undefined;\n }\n\n _handleClose(): void {\n // close the pending connection if it resolves\n this.bestEffortClose();\n super._handleClose();\n }\n}\n","import { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionNoConnectionListeners {\n // timeout related\n onSessionGracePeriodElapsed: () => void;\n}\n\n/*\n * A session that is not connected and cannot send or receive messages.\n *\n * Valid transitions:\n * - NoConnection -> Connecting (on connect)\n */\nexport class SessionNoConnection extends IdentifiedSession {\n readonly state = SessionState.NoConnection as const;\n listeners: SessionNoConnectionListeners;\n\n gracePeriodTimeout?: ReturnType<typeof setTimeout>;\n\n constructor(\n listeners: SessionNoConnectionListeners,\n ...args: ConstructorParameters<typeof IdentifiedSession>\n ) {\n super(...args);\n this.listeners = listeners;\n\n this.gracePeriodTimeout = setTimeout(() => {\n this.listeners.onSessionGracePeriodElapsed();\n }, this.options.sessionDisconnectGraceMs);\n }\n\n _handleClose(): void {\n super._handleClose();\n }\n\n _handleStateExit(): void {\n super._handleStateExit();\n\n if (this.gracePeriodTimeout) {\n clearTimeout(this.gracePeriodTimeout);\n this.gracePeriodTimeout = undefined;\n }\n }\n}\n","import { MessageMetadata } from '../../logging';\nimport { Connection } from '../connection';\nimport { TransportMessage } from '../message';\nimport { SessionHandshakingListeners } from './SessionHandshaking';\nimport { CommonSession, SessionState } from './common';\n\n/*\n * Server-side session that has a connection but is waiting for the client to identify itself.\n *\n * Valid transitions:\n * - WaitingForHandshake -> NoConnection (on close)\n * - WaitingForHandshake -> Connected (on handshake)\n */\nexport class SessionWaitingForHandshake<\n ConnType extends Connection,\n> extends CommonSession {\n readonly state = SessionState.WaitingForHandshake as const;\n conn: ConnType;\n listeners: SessionHandshakingListeners;\n\n handshakeTimeout?: ReturnType<typeof setTimeout>;\n\n constructor(\n conn: ConnType,\n listeners: SessionHandshakingListeners,\n ...args: ConstructorParameters<typeof CommonSession>\n ) {\n super(...args);\n this.conn = conn;\n this.listeners = listeners;\n\n this.handshakeTimeout = setTimeout(() => {\n listeners.onHandshakeTimeout();\n }, this.options.handshakeTimeoutMs);\n\n this.conn.addDataListener(this.onHandshakeData);\n this.conn.addErrorListener(listeners.onConnectionErrored);\n this.conn.addCloseListener(listeners.onConnectionClosed);\n }\n\n onHandshakeData = (msg: Uint8Array) => {\n const parsedMsg = this.parseMsg(msg);\n if (parsedMsg === null) {\n this.listeners.onInvalidHandshake('could not parse message');\n return;\n }\n\n // after this fires, the listener is responsible for transitioning the session\n // and thus removing the handshake timeout\n this.listeners.onHandshake(parsedMsg);\n };\n\n get loggingMetadata(): MessageMetadata {\n return {\n clientId: this.from,\n connId: this.conn.id,\n };\n }\n\n sendHandshake(msg: TransportMessage): boolean {\n return this.conn.send(this.options.codec.toBuffer(msg));\n }\n\n _handleStateExit(): void {\n this.conn.removeDataListener(this.onHandshakeData);\n this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n clearTimeout(this.handshakeTimeout);\n this.handshakeTimeout = undefined;\n }\n\n _handleClose(): void {\n this.conn.close();\n }\n}\n","import { Connection } from '../connection';\nimport { OpaqueTransportMessage, TransportMessage } from '../message';\nimport { IdentifiedSession, SessionState } from './common';\n\nexport interface SessionHandshakingListeners {\n onConnectionErrored: (err: unknown) => void;\n onConnectionClosed: () => void;\n onHandshake: (msg: OpaqueTransportMessage) => void;\n onInvalidHandshake: (reason: string) => void;\n\n // timeout related\n onHandshakeTimeout: () => void;\n}\n\n/*\n * A session that is handshaking and waiting for the other side to identify itself.\n *\n * Valid transitions:\n * - Handshaking -> NoConnection (on close)\n * - Handshaking -> Connected (on handshake)\n */\nexport class SessionHandshaking<\n ConnType extends Connection,\n> extends IdentifiedSession {\n readonly state = SessionState.Handshaking as const;\n conn: ConnType;\n listeners: SessionHandshakingListeners;\n\n handshakeTimeout: ReturnType<typeof setTimeout>;\n\n constructor(\n conn: ConnType,\n listeners: SessionHandshakingListeners,\n ...args: ConstructorParameters<typeof IdentifiedSession>\n ) {\n super(...args);\n this.conn = conn;\n this.listeners = listeners;\n\n this.handshakeTimeout = setTimeout(() => {\n listeners.onHandshakeTimeout();\n }, this.options.handshakeTimeoutMs);\n\n this.conn.addDataListener(this.onHandshakeData);\n this.conn.addErrorListener(listeners.onConnectionErrored);\n this.conn.addCloseListener(listeners.onConnectionClosed);\n }\n\n onHandshakeData = (msg: Uint8Array) => {\n const parsedMsg = this.parseMsg(msg);\n if (parsedMsg === null) {\n this.listeners.onInvalidHandshake('could not parse message');\n return;\n }\n\n this.listeners.onHandshake(parsedMsg);\n };\n\n sendHandshake(msg: TransportMessage): boolean {\n return this.conn.send(this.options.codec.toBuffer(msg));\n }\n\n _handleStateExit(): void {\n super._handleStateExit();\n this.conn.removeDataListener(this.onHandshakeData);\n this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n clearTimeout(this.handshakeTimeout);\n }\n\n _handleClose(): void {\n super._handleClose();\n this.conn.close();\n }\n}\n","import { Static } from '@sinclair/typebox';\nimport {\n ControlFlags,\n ControlMessageAckSchema,\n OpaqueTransportMessage,\n PartialTransportMessage,\n isAck,\n} from '../message';\nimport { IdentifiedSession, SessionState } from './common';\nimport { Connection } from '../connection';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nexport interface SessionConnectedListeners {\n onConnectionErrored: (err: unknown) => void;\n onConnectionClosed: () => void;\n onMessage: (msg: OpaqueTransportMessage) => void;\n onInvalidMessage: (reason: string) => void;\n}\n\n/*\n * A session that is connected and can send and receive messages.\n *\n * Valid transitions:\n * - Connected -> NoConnection (on close)\n */\nexport class SessionConnected<\n ConnType extends Connection,\n> extends IdentifiedSession {\n readonly state = SessionState.Connected as const;\n conn: ConnType;\n listeners: SessionConnectedListeners;\n\n heartbeatHandle?: ReturnType<typeof setInterval> | undefined;\n heartbeatMisses = 0;\n\n get isActivelyHeartbeating() {\n return this.heartbeatHandle !== undefined;\n }\n\n updateBookkeeping(ack: number, seq: number) {\n this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n this.ack = seq + 1;\n this.heartbeatMisses = 0;\n }\n\n send(msg: PartialTransportMessage): string {\n const constructedMsg = this.constructMsg(msg);\n this.sendBuffer.push(constructedMsg);\n this.conn.send(this.options.codec.toBuffer(constructedMsg));\n return constructedMsg.id;\n }\n\n constructor(\n conn: ConnType,\n listeners: SessionConnectedListeners,\n ...args: ConstructorParameters<typeof IdentifiedSession>\n ) {\n super(...args);\n this.conn = conn;\n this.listeners = listeners;\n\n this.conn.addDataListener(this.onMessageData);\n this.conn.addCloseListener(listeners.onConnectionClosed);\n this.conn.addErrorListener(listeners.onConnectionErrored);\n\n // send any buffered messages\n if (this.sendBuffer.length > 0) {\n this.log?.debug(\n `sending ${this.sendBuffer.length} buffered messages`,\n this.loggingMetadata,\n );\n }\n\n for (const msg of this.sendBuffer) {\n conn.send(this.options.codec.toBuffer(msg));\n }\n\n // dont explicity clear the buffer, we'll just filter out old messages\n // when we receive an ack\n }\n\n startActiveHeartbeat() {\n this.heartbeatHandle = setInterval(() => {\n const misses = this.heartbeatMisses;\n const missDuration = misses * this.options.heartbeatIntervalMs;\n if (misses >= this.options.heartbeatsUntilDead) {\n this.log?.info(\n `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n this.loggingMetadata,\n );\n this.telemetry.span.addEvent('closing connection due to inactivity');\n this.conn.close();\n clearInterval(this.heartbeatHandle);\n this.heartbeatHandle = undefined;\n return;\n }\n\n this.sendHeartbeat();\n this.heartbeatMisses++;\n }, this.options.heartbeatIntervalMs);\n }\n\n private sendHeartbeat() {\n this.send({\n streamId: 'heartbeat',\n controlFlags: ControlFlags.AckBit,\n payload: {\n type: 'ACK',\n } satisfies Static<typeof ControlMessageAckSchema>,\n });\n }\n\n onMessageData = (msg: Uint8Array) => {\n const parsedMsg = this.parseMsg(msg);\n if (parsedMsg === null) return;\n\n // check message ordering here\n if (parsedMsg.seq !== this.ack) {\n if (parsedMsg.seq < this.ack) {\n this.log?.debug(\n `received duplicate msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack}), discarding`,\n {\n ...this.loggingMetadata,\n transportMessage: parsedMsg,\n },\n );\n } else {\n const reason = `received out-of-order msg (got seq: ${parsedMsg.seq}, wanted seq: ${this.ack})`;\n this.log?.error(reason, {\n ...this.loggingMetadata,\n transportMessage: parsedMsg,\n tags: ['invariant-violation'],\n });\n this.telemetry.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n\n this.listeners.onInvalidMessage(reason);\n }\n\n return;\n }\n\n // message is ok to update bookkeeping with\n this.log?.debug(`received msg`, {\n ...this.loggingMetadata,\n transportMessage: parsedMsg,\n });\n\n this.updateBookkeeping(parsedMsg.ack, parsedMsg.seq);\n\n // dispatch directly if its not an explicit ack\n if (!isAck(parsedMsg.controlFlags)) {\n this.listeners.onMessage(parsedMsg);\n return;\n }\n\n // discard acks (unless we aren't heartbeating in which case just respond)\n this.log?.debug(`discarding msg (ack bit set)`, {\n ...this.loggingMetadata,\n transportMessage: parsedMsg,\n });\n\n // if we are not actively heartbeating, we are in passive\n // heartbeat mode and should send a response to the ack\n if (!this.isActivelyHeartbeating) {\n this.sendHeartbeat();\n }\n };\n\n _handleStateExit(): void {\n super._handleStateExit();\n this.conn.removeDataListener(this.onMessageData);\n this.conn.removeCloseListener(this.listeners.onConnectionClosed);\n this.conn.removeErrorListener(this.listeners.onConnectionErrored);\n clearInterval(this.heartbeatHandle);\n this.heartbeatHandle = undefined;\n }\n\n _handleClose(): void {\n super._handleClose();\n this.conn.close();\n }\n}\n","import { OpaqueTransportMessage, TransportClientId } from '..';\nimport {\n SessionConnecting,\n SessionConnectingListeners,\n} from './SessionConnecting';\nimport {\n SessionNoConnection,\n SessionNoConnectionListeners,\n} from './SessionNoConnection';\nimport { IdentifiedSession, SessionOptions } from './common';\nimport { PropagationContext, createSessionTelemetryInfo } from '../../tracing';\nimport { SessionWaitingForHandshake } from './SessionWaitingForHandshake';\nimport {\n SessionHandshaking,\n SessionHandshakingListeners,\n} from './SessionHandshaking';\nimport {\n SessionConnected,\n SessionConnectedListeners,\n} from './SessionConnected';\nimport { generateId } from '../id';\nimport { Connection } from '../connection';\nimport { Logger } from '../../logging';\n\nfunction inheritSharedSession(\n session: IdentifiedSession,\n): ConstructorParameters<typeof IdentifiedSession> {\n return [\n session.id,\n session.from,\n session.to,\n session.seq,\n session.ack,\n session.sendBuffer,\n session.telemetry,\n session.options,\n session.protocolVersion,\n session.log,\n ];\n}\n\n/*\n * Session state machine:\n * 1. SessionNoConnection is the client entrypoint as\n * we know who the other side is already, we just need to connect\n * 5. SessionWaitingForHandshake is the server entrypoint\n * as we have a connection but don't know who the other side is yet\n *\n * 1. SessionNoConnection ◄──┐\n * │ reconnect / connect attempt │\n * ▼ │\n * 2. SessionConnecting │\n * │ connect success ──────────────┤ connect failure\n * ▼ │\n * 3. SessionHandshaking │\n * │ handshake success ┌──────┤ connection drop\n * 5. WaitingForHandshake │ handshake failure ─────┤ │\n * │ handshake success ▼ │ │ connection drop\n * ├───────────────────────► 4. SessionConnected │ │ heartbeat misses\n * │ │ invalid message ───────┼──────┘\n * │ ▼ │\n * └───────────────────────► x. Destroy Session ◄─────┘\n * handshake failure\n */\nexport const SessionStateGraph = {\n entrypoints: {\n NoConnection(\n to: TransportClientId,\n from: TransportClientId,\n listeners: SessionNoConnectionListeners,\n options: SessionOptions,\n protocolVersion: string,\n log?: Logger,\n ) {\n const id = `session-${generateId()}`;\n const telemetry = createSessionTelemetryInfo(id, to, from);\n const sendBuffer: Array<OpaqueTransportMessage> = [];\n\n const session = new SessionNoConnection(\n listeners,\n id,\n from,\n to,\n 0,\n 0,\n sendBuffer,\n telemetry,\n options,\n protocolVersion,\n log,\n );\n\n session.log?.info(`session ${session.id} created in NoConnection state`, {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n });\n\n return session;\n },\n WaitingForHandshake<ConnType extends Connection>(\n from: TransportClientId,\n conn: ConnType,\n listeners: SessionHandshakingListeners,\n options: SessionOptions,\n log?: Logger,\n ): SessionWaitingForHandshake<ConnType> {\n const session = new SessionWaitingForHandshake(\n conn,\n listeners,\n from,\n options,\n log,\n );\n\n session.log?.info(`session created in WaitingForHandshake state`, {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n });\n\n return session;\n },\n },\n // All of the transitions 'move'/'consume' the old session and return a new one.\n // After a session is transitioned, any usage of the old session will throw.\n transition: {\n // happy path transitions\n NoConnectionToConnecting<ConnType extends Connection>(\n oldSession: SessionNoConnection,\n connPromise: Promise<ConnType>,\n listeners: SessionConnectingListeners,\n ): SessionConnecting<ConnType> {\n const carriedState = inheritSharedSession(oldSession);\n oldSession._handleStateExit();\n\n const session = new SessionConnecting(\n connPromise,\n listeners,\n ...carriedState,\n );\n session.log?.info(\n `session ${session.id} transition from NoConnection to Connecting`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n return session;\n },\n ConnectingToHandshaking<ConnType extends Connection>(\n oldSession: SessionConnecting<ConnType>,\n conn: ConnType,\n listeners: SessionHandshakingListeners,\n ): SessionHandshaking<ConnType> {\n const carriedState = inheritSharedSession(oldSession);\n oldSession._handleStateExit();\n\n const session = new SessionHandshaking(conn, listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from Connecting to Handshaking`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n HandshakingToConnected<ConnType extends Connection>(\n oldSession: SessionHandshaking<ConnType>,\n listeners: SessionConnectedListeners,\n ): SessionConnected<ConnType> {\n const carriedState = inheritSharedSession(oldSession);\n const conn = oldSession.conn;\n oldSession._handleStateExit();\n\n const session = new SessionConnected(conn, listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from Handshaking to Connected`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n WaitingForHandshakeToConnected<ConnType extends Connection>(\n pendingSession: SessionWaitingForHandshake<ConnType>,\n oldSession: SessionNoConnection | undefined,\n sessionId: string,\n to: TransportClientId,\n propagationCtx: PropagationContext | undefined,\n listeners: SessionConnectedListeners,\n protocolVersion: string,\n ): SessionConnected<ConnType> {\n const conn = pendingSession.conn;\n const { from, options } = pendingSession;\n const carriedState: ConstructorParameters<typeof IdentifiedSession> =\n oldSession\n ? // old session exists, inherit state\n inheritSharedSession(oldSession)\n : // old session does not exist, create new state\n [\n sessionId,\n from,\n to,\n 0,\n 0,\n [],\n createSessionTelemetryInfo(sessionId, to, from, propagationCtx),\n options,\n protocolVersion,\n pendingSession.log,\n ];\n\n pendingSession._handleStateExit();\n oldSession?._handleStateExit();\n\n const session = new SessionConnected(conn, listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from WaitingForHandshake to Connected`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n // disconnect paths\n ConnectingToNoConnection<ConnType extends Connection>(\n oldSession: SessionConnecting<ConnType>,\n listeners: SessionNoConnectionListeners,\n ): SessionNoConnection {\n const carriedState = inheritSharedSession(oldSession);\n oldSession.bestEffortClose();\n oldSession._handleStateExit();\n\n const session = new SessionNoConnection(listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from Connecting to NoConnection`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n HandshakingToNoConnection<ConnType extends Connection>(\n oldSession: SessionHandshaking<ConnType>,\n listeners: SessionNoConnectionListeners,\n ): SessionNoConnection {\n const carriedState = inheritSharedSession(oldSession);\n oldSession.conn.close();\n oldSession._handleStateExit();\n\n const session = new SessionNoConnection(listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from Handshaking to NoConnection`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n ConnectedToNoConnection<ConnType extends Connection>(\n oldSession: SessionConnected<ConnType>,\n listeners: SessionNoConnectionListeners,\n ): SessionNoConnection {\n const carriedState = inheritSharedSession(oldSession);\n oldSession.conn.close();\n oldSession._handleStateExit();\n\n const session = new SessionNoConnection(listeners, ...carriedState);\n session.log?.info(\n `session ${session.id} transition from Connected to NoConnection`,\n {\n ...session.loggingMetadata,\n tags: ['state-transition'],\n },\n );\n\n return session;\n },\n },\n} as const;\n"],"mappings":";;;;;;;;;;;AAQO,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,qBAAqB;AAAA,EACrB,oBAAoB;AAAA,EACpB,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AAAA,EACH,GAAG;AACL;AAMO,IAAM,gCAAwD;AAAA,EACnE,GAAG;AACL;;;AC/BA,SAAS,aAAa;AASf,IAAW,eAAX,kBAAWA,kBAAX;AACL,EAAAA,cAAA,kBAAe;AACf,EAAAA,cAAA,gBAAa;AACb,EAAAA,cAAA,iBAAc;AACd,EAAAA,cAAA,eAAY;AACZ,EAAAA,cAAA,yBAAsB;AALN,SAAAA;AAAA,GAAA;AAcX,IAAM,eAAe;AAE5B,IAAe,oBAAf,MAAiC;AAAA;AAAA;AAAA;AAAA;AAAA,EAO/B;AAAA,EAaA,QAAc;AACZ,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,cAAc;AACZ,SAAK,cAAc;AAKnB,WAAO,IAAI,MAAM,MAAM;AAAA,MACrB,IAAI,QAAQ,MAAM;AAEhB,YAAI,SAAS,iBAAiB,SAAS,QAAQ,SAAS,SAAS;AAC/D,iBAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,QACjC;AAGA,YAAI,SAAS,oBAAoB;AAC/B,iBAAO,MAAM;AACX,mBAAO,cAAc;AACrB,mBAAO,iBAAiB;AAAA,UAC1B;AAAA,QACF;AAGA,YAAI,SAAS,gBAAgB;AAC3B,iBAAO,MAAM;AACX,mBAAO,iBAAiB;AACxB,mBAAO,aAAa;AAAA,UACtB;AAAA,QACF;AAEA,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,IAAI;AAAA,MACjC;AAAA,MACA,IAAI,QAAQ,MAAM,OAAO;AACvB,YAAI,OAAO,aAAa;AACtB,gBAAM,IAAI;AAAA,YACR,GAAG,YAAY,aAAa,KAAK,SAAS,CAAC;AAAA,UAC7C;AAAA,QACF;AAEA,eAAO,QAAQ,IAAI,QAAQ,MAAM,KAAK;AAAA,MACxC;AAAA,IACF,CAAC;AAAA,EACH;AACF;AA+BO,IAAe,gBAAf,cAAqC,kBAAkB;AAAA,EACnD;AAAA,EACA;AAAA,EAET;AAAA,EAGA,YACE,MACA,SACA,KACA;AACA,UAAM;AACN,SAAK,OAAO;AACZ,SAAK,UAAU;AACf,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,SAAS,KAAgD;AACvD,UAAM,YAAY,KAAK,QAAQ,MAAM,WAAW,GAAG;AAEnD,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,KAAK;AAAA,QACR,2BAA2B,aAAa;AAAA,QACxC,KAAK;AAAA,MACP;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,MAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,KAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QACpE,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,8BAA8B,SAAS;AAAA,QACzD;AAAA,MACF,CAAC;AAED,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AACF;AAUO,IAAe,oBAAf,cAAyC,cAAc;AAAA,EACnD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EACA;AAAA,EAEA,YACE,IACA,MACA,IACA,KACA,KACA,YACA,WACA,SACA,iBACA,KACA;AACA,UAAM,MAAM,SAAS,GAAG;AACxB,SAAK,KAAK;AACV,SAAK,KAAK;AACV,SAAK,MAAM;AACX,SAAK,MAAM;AACX,SAAK,aAAa;AAClB,SAAK,YAAY;AACjB,SAAK,MAAM;AACX,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,IAAI,kBAAmC;AACrC,UAAM,cAAc,KAAK,UAAU,KAAK,YAAY;AAEpD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,WAAW;AAAA,QACT,SAAS,YAAY;AAAA,QACrB,QAAQ,YAAY;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,WAAW;AAAA,MACf,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,WAAO;AAAA,EACT;AAAA,EAEA,UAAkB;AAChB,WAAO,KAAK,WAAW,SAAS,IAAI,KAAK,WAAW,CAAC,EAAE,MAAM,KAAK;AAAA,EACpE;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AACnC,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,mBAAyB;AAAA,EAEzB;AAAA,EAEA,eAAqB;AAEnB,SAAK,WAAW,SAAS;AACzB,SAAK,UAAU,KAAK,IAAI;AAAA,EAC1B;AACF;;;ACpQO,IAAM,oBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,aACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,cAAc;AACnB,SAAK,YAAY;AAEjB,SAAK,oBAAoB,WAAW,MAAM;AACxC,gBAAU,oBAAoB;AAAA,IAChC,GAAG,KAAK,QAAQ,mBAAmB;AAEnC,gBAAY;AAAA,MACV,CAAC,SAAS;AACR,YAAI,KAAK;AAAa;AACtB,kBAAU,wBAAwB,IAAI;AAAA,MACxC;AAAA,MACA,CAAC,QAAQ;AACP,YAAI,KAAK;AAAa;AACtB,kBAAU,mBAAmB,GAAG;AAAA,MAClC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA,EAIA,kBAAkB;AAChB,SAAK,KAAK,YACP,KAAK,CAAC,SAAS,KAAK,MAAM,CAAC,EAC3B,MAAM,MAAM;AAAA,IAEb,CAAC;AAAA,EACL;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,iBAAa,KAAK,iBAAiB;AACnC,SAAK,oBAAoB;AAAA,EAC3B;AAAA,EAEA,eAAqB;AAEnB,SAAK,gBAAgB;AACrB,UAAM,aAAa;AAAA,EACrB;AACF;;;AC7DO,IAAM,sBAAN,cAAkC,kBAAkB;AAAA,EAChD;AAAA,EACT;AAAA,EAEA;AAAA,EAEA,YACE,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,YAAY;AAEjB,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,UAAU,4BAA4B;AAAA,IAC7C,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AAAA,EACrB;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AAEvB,QAAI,KAAK,oBAAoB;AAC3B,mBAAa,KAAK,kBAAkB;AACpC,WAAK,qBAAqB;AAAA,IAC5B;AAAA,EACF;AACF;;;AC9BO,IAAM,6BAAN,cAEG,cAAc;AAAA,EACb;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,mBAAmB,WAAW,MAAM;AACvC,gBAAU,mBAAmB;AAAA,IAC/B,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AACxD,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AAAA,EACzD;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU,mBAAmB,yBAAyB;AAC3D;AAAA,IACF;AAIA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,IAAI,kBAAmC;AACrC,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK,KAAK;AAAA,IACpB;AAAA,EACF;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,iBAAa,KAAK,gBAAgB;AAClC,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,eAAqB;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;ACrDO,IAAM,qBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,mBAAmB,WAAW,MAAM;AACvC,gBAAU,mBAAmB;AAAA,IAC/B,GAAG,KAAK,QAAQ,kBAAkB;AAElC,SAAK,KAAK,gBAAgB,KAAK,eAAe;AAC9C,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AACxD,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AAAA,EACzD;AAAA,EAEA,kBAAkB,CAAC,QAAoB;AACrC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc,MAAM;AACtB,WAAK,UAAU,mBAAmB,yBAAyB;AAC3D;AAAA,IACF;AAEA,SAAK,UAAU,YAAY,SAAS;AAAA,EACtC;AAAA,EAEA,cAAc,KAAgC;AAC5C,WAAO,KAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,EACxD;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,eAAe;AACjD,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,iBAAa,KAAK,gBAAgB;AAAA,EACpC;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;AChEA,SAAS,sBAAsB;AAexB,IAAM,mBAAN,cAEG,kBAAkB;AAAA,EACjB;AAAA,EACT;AAAA,EACA;AAAA,EAEA;AAAA,EACA,kBAAkB;AAAA,EAElB,IAAI,yBAAyB;AAC3B,WAAO,KAAK,oBAAoB;AAAA,EAClC;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AACjB,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,KAAK,KAAsC;AACzC,UAAM,iBAAiB,KAAK,aAAa,GAAG;AAC5C,SAAK,WAAW,KAAK,cAAc;AACnC,SAAK,KAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,cAAc,CAAC;AAC1D,WAAO,eAAe;AAAA,EACxB;AAAA,EAEA,YACE,MACA,cACG,MACH;AACA,UAAM,GAAG,IAAI;AACb,SAAK,OAAO;AACZ,SAAK,YAAY;AAEjB,SAAK,KAAK,gBAAgB,KAAK,aAAa;AAC5C,SAAK,KAAK,iBAAiB,UAAU,kBAAkB;AACvD,SAAK,KAAK,iBAAiB,UAAU,mBAAmB;AAGxD,QAAI,KAAK,WAAW,SAAS,GAAG;AAC9B,WAAK,KAAK;AAAA,QACR,WAAW,KAAK,WAAW,MAAM;AAAA,QACjC,KAAK;AAAA,MACP;AAAA,IACF;AAEA,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,KAAK,KAAK,QAAQ,MAAM,SAAS,GAAG,CAAC;AAAA,IAC5C;AAAA,EAIF;AAAA,EAEA,uBAAuB;AACrB,SAAK,kBAAkB,YAAY,MAAM;AACvC,YAAM,SAAS,KAAK;AACpB,YAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,UAAI,UAAU,KAAK,QAAQ,qBAAqB;AAC9C,aAAK,KAAK;AAAA,UACR,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,KAAK,MAAM;AAChB,sBAAc,KAAK,eAAe;AAClC,aAAK,kBAAkB;AACvB;AAAA,MACF;AAEA,WAAK,cAAc;AACnB,WAAK;AAAA,IACP,GAAG,KAAK,QAAQ,mBAAmB;AAAA,EACrC;AAAA,EAEQ,gBAAgB;AACtB,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,CAAC,QAAoB;AACnC,UAAM,YAAY,KAAK,SAAS,GAAG;AACnC,QAAI,cAAc;AAAM;AAGxB,QAAI,UAAU,QAAQ,KAAK,KAAK;AAC9B,UAAI,UAAU,MAAM,KAAK,KAAK;AAC5B,aAAK,KAAK;AAAA,UACR,oCAAoC,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAAA,UAC1E;AAAA,YACE,GAAG,KAAK;AAAA,YACR,kBAAkB;AAAA,UACpB;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,UAAU,GAAG,iBAAiB,KAAK,GAAG;AAC5F,aAAK,KAAK,MAAM,QAAQ;AAAA,UACtB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,UAAU,KAAK,UAAU;AAAA,UAC5B,MAAM,eAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,UAAU,iBAAiB,MAAM;AAAA,MACxC;AAEA;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gBAAgB;AAAA,MAC9B,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,SAAK,kBAAkB,UAAU,KAAK,UAAU,GAAG;AAGnD,QAAI,CAAC,MAAM,UAAU,YAAY,GAAG;AAClC,WAAK,UAAU,UAAU,SAAS;AAClC;AAAA,IACF;AAGA,SAAK,KAAK,MAAM,gCAAgC;AAAA,MAC9C,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAID,QAAI,CAAC,KAAK,wBAAwB;AAChC,WAAK,cAAc;AAAA,IACrB;AAAA,EACF;AAAA,EAEA,mBAAyB;AACvB,UAAM,iBAAiB;AACvB,SAAK,KAAK,mBAAmB,KAAK,aAAa;AAC/C,SAAK,KAAK,oBAAoB,KAAK,UAAU,kBAAkB;AAC/D,SAAK,KAAK,oBAAoB,KAAK,UAAU,mBAAmB;AAChE,kBAAc,KAAK,eAAe;AAClC,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEA,eAAqB;AACnB,UAAM,aAAa;AACnB,SAAK,KAAK,MAAM;AAAA,EAClB;AACF;;;AChKA,SAAS,qBACP,SACiD;AACjD,SAAO;AAAA,IACL,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,QAAQ;AAAA,EACV;AACF;AAyBO,IAAM,oBAAoB;AAAA,EAC/B,aAAa;AAAA,IACX,aACE,IACA,MACA,WACA,SACA,iBACA,KACA;AACA,YAAM,KAAK,WAAW,WAAW,CAAC;AAClC,YAAM,YAAY,2BAA2B,IAAI,IAAI,IAAI;AACzD,YAAM,aAA4C,CAAC;AAEnD,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAEA,cAAQ,KAAK,KAAK,WAAW,QAAQ,EAAE,kCAAkC;AAAA,QACvE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,oBACE,MACA,MACA,WACA,SACA,KACsC;AACtC,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAEA,cAAQ,KAAK,KAAK,gDAAgD;AAAA,QAChE,GAAG,QAAQ;AAAA,QACX,MAAM,CAAC,kBAAkB;AAAA,MAC3B,CAAC;AAED,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA,EAGA,YAAY;AAAA;AAAA,IAEV,yBACE,YACA,aACA,WAC6B;AAC7B,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG;AAAA,MACL;AACA,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,wBACE,YACA,MACA,WAC8B;AAC9B,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,mBAAmB,MAAM,WAAW,GAAG,YAAY;AACvE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,uBACE,YACA,WAC4B;AAC5B,YAAM,eAAe,qBAAqB,UAAU;AACpD,YAAM,OAAO,WAAW;AACxB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,iBAAiB,MAAM,WAAW,GAAG,YAAY;AACrE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,+BACE,gBACA,YACA,WACA,IACA,gBACA,WACA,iBAC4B;AAC5B,YAAM,OAAO,eAAe;AAC5B,YAAM,EAAE,MAAM,QAAQ,IAAI;AAC1B,YAAM,eACJ;AAAA;AAAA,QAEI,qBAAqB,UAAU;AAAA;AAAA;AAAA,QAE/B;AAAA,UACE;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA,CAAC;AAAA,UACD,2BAA2B,WAAW,IAAI,MAAM,cAAc;AAAA,UAC9D;AAAA,UACA;AAAA,UACA,eAAe;AAAA,QACjB;AAAA;AAEN,qBAAe,iBAAiB;AAChC,kBAAY,iBAAiB;AAE7B,YAAM,UAAU,IAAI,iBAAiB,MAAM,WAAW,GAAG,YAAY;AACrE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,IAEA,yBACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,gBAAgB;AAC3B,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,0BACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,IACA,wBACE,YACA,WACqB;AACrB,YAAM,eAAe,qBAAqB,UAAU;AACpD,iBAAW,KAAK,MAAM;AACtB,iBAAW,iBAAiB;AAE5B,YAAM,UAAU,IAAI,oBAAoB,WAAW,GAAG,YAAY;AAClE,cAAQ,KAAK;AAAA,QACX,WAAW,QAAQ,EAAE;AAAA,QACrB;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,MAAM,CAAC,kBAAkB;AAAA,QAC3B;AAAA,MACF;AAEA,aAAO;AAAA,IACT;AAAA,EACF;AACF;","names":["SessionState"]}
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../transport/client.ts","../transport/rateLimit.ts"],"sourcesContent":["import { SpanStatusCode } from '@opentelemetry/api';\nimport { ClientHandshakeOptions } from '../router/handshake';\nimport {\n ControlMessageHandshakeResponseSchema,\n HandshakeErrorRetriableResponseCodes,\n OpaqueTransportMessage,\n PartialTransportMessage,\n TransportClientId,\n currentProtocolVersion,\n handshakeRequestMessage,\n} from './message';\nimport {\n ClientTransportOptions,\n ProvidedClientTransportOptions,\n defaultClientTransportOptions,\n} from './options';\nimport { LeakyBucketRateLimit } from './rateLimit';\nimport { Transport } from './transport';\nimport { coerceErrorString } from '../util/stringify';\nimport { ProtocolError } from './events';\nimport { Value } from '@sinclair/typebox/value';\nimport tracer, { getPropagationContext } from '../tracing';\nimport { Connection } from './connection';\nimport { MessageMetadata } from '../logging';\nimport { SessionConnecting } from './sessionStateMachine/SessionConnecting';\nimport { SessionHandshaking } from './sessionStateMachine/SessionHandshaking';\nimport { SessionConnected } from './sessionStateMachine/SessionConnected';\nimport { SessionStateGraph } from './sessionStateMachine/transitions';\nimport { SessionState } from './sessionStateMachine/common';\nimport { SessionNoConnection } from './sessionStateMachine/SessionNoConnection';\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 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.retryBudget = new LeakyBucketRateLimit(this.options);\n }\n\n extendHandshake(options: ClientHandshakeOptions) {\n this.handshakeExtensions = options;\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 private tryReconnecting(to: string) {\n if (this.reconnectOnConnectionDrop && this.getStatus() === 'open') {\n this.connect(to);\n }\n }\n\n send(to: string, msg: PartialTransportMessage): string {\n if (this.getStatus() === 'closed') {\n const err = 'transport is closed, cant send';\n this.log?.error(err, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n\n throw new Error(err);\n }\n\n let session = this.sessions.get(to);\n if (!session) {\n session = this.createUnconnectedSession(to);\n }\n\n return session.send(msg);\n }\n\n private createUnconnectedSession(to: string): SessionNoConnection {\n const session = SessionStateGraph.entrypoints.NoConnection(\n to,\n this.clientId,\n {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(session);\n },\n },\n this.options,\n currentProtocolVersion,\n this.log,\n );\n\n this.updateSession(session);\n return session;\n }\n\n // listeners\n protected onConnectingFailed(session: SessionConnecting<ConnType>) {\n const noConnectionSession = super.onConnectingFailed(session);\n this.tryReconnecting(noConnectionSession.to);\n return noConnectionSession;\n }\n\n protected onConnClosed(\n session: SessionHandshaking<ConnType> | SessionConnected<ConnType>,\n ) {\n const noConnectionSession = super.onConnClosed(session);\n this.tryReconnecting(noConnectionSession.to);\n return noConnectionSession;\n }\n\n protected onConnectionEstablished(\n session: SessionConnecting<ConnType>,\n conn: ConnType,\n ): SessionHandshaking<ConnType> {\n // transition to handshaking\n const handshakingSession =\n SessionStateGraph.transition.ConnectingToHandshaking(session, conn, {\n onConnectionErrored: (err) => {\n // just log, when we error we also emit close\n const errStr = coerceErrorString(err);\n this.log?.error(\n `connection to ${handshakingSession.to} errored during handshake: ${errStr}`,\n handshakingSession.loggingMetadata,\n );\n },\n onConnectionClosed: () => {\n this.log?.warn(\n `connection to ${handshakingSession.to} closed during handshake`,\n handshakingSession.loggingMetadata,\n );\n this.onConnClosed(handshakingSession);\n },\n onHandshake: (msg) => {\n this.onHandshakeResponse(handshakingSession, msg);\n },\n onInvalidHandshake: (reason) => {\n this.log?.error(\n `invalid handshake: ${reason}`,\n handshakingSession.loggingMetadata,\n );\n this.deleteSession(session);\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n },\n onHandshakeTimeout: () => {\n this.log?.error(\n `connection to ${handshakingSession.to} timed out during handshake`,\n handshakingSession.loggingMetadata,\n );\n this.onConnClosed(handshakingSession);\n },\n });\n\n this.updateSession(handshakingSession);\n void this.sendHandshake(handshakingSession);\n return handshakingSession;\n }\n\n private rejectHandshakeResponse(\n session: SessionHandshaking<ConnType>,\n reason: string,\n metadata: MessageMetadata,\n ) {\n session.conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n\n this.log?.warn(reason, metadata);\n this.deleteSession(session);\n }\n\n protected onHandshakeResponse(\n session: SessionHandshaking<ConnType>,\n msg: OpaqueTransportMessage,\n ) {\n // invariant: msg is a handshake response\n if (!Value.Check(ControlMessageHandshakeResponseSchema, msg.payload)) {\n const reason = `received invalid handshake response`;\n this.rejectHandshakeResponse(session, reason, {\n ...session.loggingMetadata,\n transportMessage: msg,\n validationErrors: [\n ...Value.Errors(ControlMessageHandshakeResponseSchema, msg.payload),\n ],\n });\n return;\n }\n\n // invariant: handshake response should be ok\n if (!msg.payload.status.ok) {\n // TODO: remove conditional check after we know code is always present\n const retriable = msg.payload.status.code\n ? Value.Check(\n HandshakeErrorRetriableResponseCodes,\n msg.payload.status.code,\n )\n : false;\n\n const reason = `handshake failed: ${msg.payload.status.reason}`;\n this.rejectHandshakeResponse(session, reason, {\n ...session.loggingMetadata,\n transportMessage: msg,\n });\n\n if (retriable) {\n this.tryReconnecting(session.to);\n } else {\n this.deleteSession(session);\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n }\n\n return;\n }\n\n // invariant: session id should match between client + server\n if (msg.payload.status.sessionId !== session.id) {\n const reason = `session id mismatch: expected ${session.id}, got ${msg.payload.status.sessionId}`;\n this.rejectHandshakeResponse(session, reason, {\n ...session.loggingMetadata,\n transportMessage: msg,\n });\n return;\n }\n\n // transition to connected!\n this.log?.info(`handshake from ${msg.from} ok`, {\n ...session.loggingMetadata,\n transportMessage: msg,\n });\n\n const connectedSession =\n SessionStateGraph.transition.HandshakingToConnected(session, {\n onConnectionErrored: (err) => {\n // just log, when we error we also emit close\n const errStr = coerceErrorString(err);\n this.log?.warn(\n `connection to ${connectedSession.to} errored: ${errStr}`,\n connectedSession.loggingMetadata,\n );\n },\n onConnectionClosed: () => {\n this.log?.info(\n `connection to ${connectedSession.to} closed`,\n connectedSession.loggingMetadata,\n );\n this.onConnClosed(connectedSession);\n },\n onMessage: (msg) => this.handleMsg(msg),\n onInvalidMessage: (reason) => {\n this.deleteSession(connectedSession);\n this.protocolError(ProtocolError.MessageOrderingViolated, reason);\n },\n });\n\n this.updateSession(connectedSession);\n this.retryBudget.startRestoringBudget(connectedSession.to);\n }\n\n /**\n * Manually attempts to connect to a client.\n * @param to The client ID of the node to connect to.\n */\n connect(to: TransportClientId) {\n // create a new session if one does not exist\n let session = this.sessions.get(to);\n session ??= this.createUnconnectedSession(to);\n\n if (session.state !== SessionState.NoConnection) {\n // already trying to connect\n this.log?.debug(\n `session to ${to} has state ${session.state}, skipping connect attempt`,\n session.loggingMetadata,\n );\n return;\n }\n\n if (this.getStatus() !== 'open') {\n this.log?.info(\n `transport state is no longer open, cancelling attempt to connect to ${to}`,\n session.loggingMetadata,\n );\n return;\n }\n\n // check budget\n if (!this.retryBudget.hasBudget(to)) {\n const budgetConsumed = this.retryBudget.getBudgetConsumed(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, session.loggingMetadata);\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 session.loggingMetadata,\n );\n\n this.retryBudget.consumeBudget(to);\n const reconnectPromise = tracer.startActiveSpan('connect', async (span) => {\n try {\n span.addEvent('backoff', { backoffMs });\n await sleep;\n if (this.getStatus() !== 'open') {\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('connecting');\n return await this.createNewOutgoingConnection(to);\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 const connectingSession =\n SessionStateGraph.transition.NoConnectionToConnecting(\n session,\n reconnectPromise,\n {\n onConnectionEstablished: (conn) => {\n this.log?.debug(\n `connection to ${connectingSession.to} established`,\n connectingSession.loggingMetadata,\n );\n\n // cast here because conn can't be narrowed to ConnType\n // in the callback due to variance rules\n this.onConnectionEstablished(connectingSession, conn as ConnType);\n },\n onConnectionFailed: (error: unknown) => {\n const errStr = coerceErrorString(error);\n this.log?.error(\n `error connecting to ${connectingSession.to}: ${errStr}`,\n connectingSession.loggingMetadata,\n );\n this.onConnectingFailed(connectingSession);\n },\n onConnectionTimeout: () => {\n this.log?.error(\n `connection to ${connectingSession.to} timed out`,\n connectingSession.loggingMetadata,\n );\n this.onConnectingFailed(connectingSession);\n },\n },\n );\n\n this.updateSession(connectingSession);\n }\n\n private async sendHandshake(session: SessionHandshaking<ConnType>) {\n let metadata: unknown = undefined;\n\n if (this.handshakeExtensions) {\n metadata = await this.handshakeExtensions.construct();\n }\n\n const requestMsg = handshakeRequestMessage({\n from: this.clientId,\n to: session.to,\n sessionId: session.id,\n expectedSessionState: {\n nextExpectedSeq: session.ack,\n nextSentSeq: session.nextSeq(),\n },\n metadata,\n tracing: getPropagationContext(session.telemetry.ctx),\n });\n\n this.log?.debug(`sending handshake request to ${session.to}`, {\n ...session.loggingMetadata,\n transportMessage: requestMsg,\n });\n\n session.sendHandshake(requestMsg);\n }\n\n close() {\n this.retryBudget.close();\n super.close();\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;;;AC4CxB,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;;;ADlHA,SAAS,aAAa;AAWf,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA,EAEV;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,cAAc,IAAI,qBAAqB,KAAK,OAAO;AAAA,EAC1D;AAAA,EAEA,gBAAgB,SAAiC;AAC/C,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAcQ,gBAAgB,IAAY;AAClC,QAAI,KAAK,6BAA6B,KAAK,UAAU,MAAM,QAAQ;AACjE,WAAK,QAAQ,EAAE;AAAA,IACjB;AAAA,EACF;AAAA,EAEA,KAAK,IAAY,KAAsC;AACrD,QAAI,KAAK,UAAU,MAAM,UAAU;AACjC,YAAM,MAAM;AACZ,WAAK,KAAK,MAAM,KAAK;AAAA,QACnB,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AAED,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,QAAI,CAAC,SAAS;AACZ,gBAAU,KAAK,yBAAyB,EAAE;AAAA,IAC5C;AAEA,WAAO,QAAQ,KAAK,GAAG;AAAA,EACzB;AAAA,EAEQ,yBAAyB,IAAiC;AAChE,UAAM,UAAU,kBAAkB,YAAY;AAAA,MAC5C;AAAA,MACA,KAAK;AAAA,MACL;AAAA,QACE,6BAA6B,MAAM;AACjC,eAAK,4BAA4B,OAAO;AAAA,QAC1C;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,IACP;AAEA,SAAK,cAAc,OAAO;AAC1B,WAAO;AAAA,EACT;AAAA;AAAA,EAGU,mBAAmB,SAAsC;AACjE,UAAM,sBAAsB,MAAM,mBAAmB,OAAO;AAC5D,SAAK,gBAAgB,oBAAoB,EAAE;AAC3C,WAAO;AAAA,EACT;AAAA,EAEU,aACR,SACA;AACA,UAAM,sBAAsB,MAAM,aAAa,OAAO;AACtD,SAAK,gBAAgB,oBAAoB,EAAE;AAC3C,WAAO;AAAA,EACT;AAAA,EAEU,wBACR,SACA,MAC8B;AAE9B,UAAM,qBACJ,kBAAkB,WAAW,wBAAwB,SAAS,MAAM;AAAA,MAClE,qBAAqB,CAAC,QAAQ;AAE5B,cAAM,SAAS,kBAAkB,GAAG;AACpC,aAAK,KAAK;AAAA,UACR,iBAAiB,mBAAmB,EAAE,8BAA8B,MAAM;AAAA,UAC1E,mBAAmB;AAAA,QACrB;AAAA,MACF;AAAA,MACA,oBAAoB,MAAM;AACxB,aAAK,KAAK;AAAA,UACR,iBAAiB,mBAAmB,EAAE;AAAA,UACtC,mBAAmB;AAAA,QACrB;AACA,aAAK,aAAa,kBAAkB;AAAA,MACtC;AAAA,MACA,aAAa,CAAC,QAAQ;AACpB,aAAK,oBAAoB,oBAAoB,GAAG;AAAA,MAClD;AAAA,MACA,oBAAoB,CAAC,WAAW;AAC9B,aAAK,KAAK;AAAA,UACR,sBAAsB,MAAM;AAAA,UAC5B,mBAAmB;AAAA,QACrB;AACA,aAAK,cAAc,OAAO;AAC1B,aAAK,cAAc,cAAc,iBAAiB,MAAM;AAAA,MAC1D;AAAA,MACA,oBAAoB,MAAM;AACxB,aAAK,KAAK;AAAA,UACR,iBAAiB,mBAAmB,EAAE;AAAA,UACtC,mBAAmB;AAAA,QACrB;AACA,aAAK,aAAa,kBAAkB;AAAA,MACtC;AAAA,IACF,CAAC;AAEH,SAAK,cAAc,kBAAkB;AACrC,SAAK,KAAK,cAAc,kBAAkB;AAC1C,WAAO;AAAA,EACT;AAAA,EAEQ,wBACN,SACA,QACA,UACA;AACA,YAAQ,KAAK,WAAW,KAAK,UAAU;AAAA,MACrC,MAAM,eAAe;AAAA,MACrB,SAAS;AAAA,IACX,CAAC;AAED,SAAK,KAAK,KAAK,QAAQ,QAAQ;AAC/B,SAAK,cAAc,OAAO;AAAA,EAC5B;AAAA,EAEU,oBACR,SACA,KACA;AAEA,QAAI,CAAC,MAAM,MAAM,uCAAuC,IAAI,OAAO,GAAG;AACpE,YAAM,SAAS;AACf,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,QAClB,kBAAkB;AAAA,UAChB,GAAG,MAAM,OAAO,uCAAuC,IAAI,OAAO;AAAA,QACpE;AAAA,MACF,CAAC;AACD;AAAA,IACF;AAGA,QAAI,CAAC,IAAI,QAAQ,OAAO,IAAI;AAE1B,YAAM,YAAY,IAAI,QAAQ,OAAO,OACjC,MAAM;AAAA,QACJ;AAAA,QACA,IAAI,QAAQ,OAAO;AAAA,MACrB,IACA;AAEJ,YAAM,SAAS,qBAAqB,IAAI,QAAQ,OAAO,MAAM;AAC7D,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,MACpB,CAAC;AAED,UAAI,WAAW;AACb,aAAK,gBAAgB,QAAQ,EAAE;AAAA,MACjC,OAAO;AACL,aAAK,cAAc,OAAO;AAC1B,aAAK,cAAc,cAAc,iBAAiB,MAAM;AAAA,MAC1D;AAEA;AAAA,IACF;AAGA,QAAI,IAAI,QAAQ,OAAO,cAAc,QAAQ,IAAI;AAC/C,YAAM,SAAS,iCAAiC,QAAQ,EAAE,SAAS,IAAI,QAAQ,OAAO,SAAS;AAC/F,WAAK,wBAAwB,SAAS,QAAQ;AAAA,QAC5C,GAAG,QAAQ;AAAA,QACX,kBAAkB;AAAA,MACpB,CAAC;AACD;AAAA,IACF;AAGA,SAAK,KAAK,KAAK,kBAAkB,IAAI,IAAI,OAAO;AAAA,MAC9C,GAAG,QAAQ;AAAA,MACX,kBAAkB;AAAA,IACpB,CAAC;AAED,UAAM,mBACJ,kBAAkB,WAAW,uBAAuB,SAAS;AAAA,MAC3D,qBAAqB,CAAC,QAAQ;AAE5B,cAAM,SAAS,kBAAkB,GAAG;AACpC,aAAK,KAAK;AAAA,UACR,iBAAiB,iBAAiB,EAAE,aAAa,MAAM;AAAA,UACvD,iBAAiB;AAAA,QACnB;AAAA,MACF;AAAA,MACA,oBAAoB,MAAM;AACxB,aAAK,KAAK;AAAA,UACR,iBAAiB,iBAAiB,EAAE;AAAA,UACpC,iBAAiB;AAAA,QACnB;AACA,aAAK,aAAa,gBAAgB;AAAA,MACpC;AAAA,MACA,WAAW,CAACA,SAAQ,KAAK,UAAUA,IAAG;AAAA,MACtC,kBAAkB,CAAC,WAAW;AAC5B,aAAK,cAAc,gBAAgB;AACnC,aAAK,cAAc,cAAc,yBAAyB,MAAM;AAAA,MAClE;AAAA,IACF,CAAC;AAEH,SAAK,cAAc,gBAAgB;AACnC,SAAK,YAAY,qBAAqB,iBAAiB,EAAE;AAAA,EAC3D;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAAQ,IAAuB;AAE7B,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,gBAAY,KAAK,yBAAyB,EAAE;AAE5C,QAAI,QAAQ,6CAAqC;AAE/C,WAAK,KAAK;AAAA,QACR,cAAc,EAAE,cAAc,QAAQ,KAAK;AAAA,QAC3C,QAAQ;AAAA,MACV;AACA;AAAA,IACF;AAEA,QAAI,KAAK,UAAU,MAAM,QAAQ;AAC/B,WAAK,KAAK;AAAA,QACR,uEAAuE,EAAE;AAAA,QACzE,QAAQ;AAAA,MACV;AACA;AAAA,IACF;AAGA,QAAI,CAAC,KAAK,YAAY,UAAU,EAAE,GAAG;AACnC,YAAM,iBAAiB,KAAK,YAAY,kBAAkB,EAAE;AAC5D,YAAM,SAAS,uBAAuB,EAAE,yCAAyC,cAAc,yBAAyB,KAAK,YAAY,sBAAsB;AAC/J,WAAK,KAAK,MAAM,QAAQ,QAAQ,eAAe;AAC/C,WAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD;AAAA,IACF;AAEA,QAAI,QAAQ,QAAQ,QAAQ;AAC5B,UAAM,YAAY,KAAK,YAAY,aAAa,EAAE;AAClD,QAAI,YAAY,GAAG;AACjB,cAAQ,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,SAAS,CAAC;AAAA,IACjE;AAEA,SAAK,KAAK;AAAA,MACR,4BAA4B,EAAE,KAAK,SAAS;AAAA,MAC5C,QAAQ;AAAA,IACV;AAEA,SAAK,YAAY,cAAc,EAAE;AACjC,UAAM,mBAAmB,gBAAO,gBAAgB,WAAW,OAAO,SAAS;AACzE,UAAI;AACF,aAAK,SAAS,WAAW,EAAE,UAAU,CAAC;AACtC,cAAM;AACN,YAAI,KAAK,UAAU,MAAM,QAAQ;AAC/B,gBAAM,IAAI,MAAM,mCAAmC;AAAA,QACrD;AAEA,aAAK,SAAS,YAAY;AAC1B,eAAO,MAAM,KAAK,4BAA4B,EAAE;AAAA,MAClD,SAAS,KAAK;AAGZ,cAAM,SAAS,kBAAkB,GAAG;AACpC,aAAK,gBAAgB,MAAM;AAC3B,aAAK,UAAU,EAAE,MAAM,eAAe,MAAM,CAAC;AAC7C,cAAM;AAAA,MACR,UAAE;AACA,aAAK,IAAI;AAAA,MACX;AAAA,IACF,CAAC;AAED,UAAM,oBACJ,kBAAkB,WAAW;AAAA,MAC3B;AAAA,MACA;AAAA,MACA;AAAA,QACE,yBAAyB,CAAC,SAAS;AACjC,eAAK,KAAK;AAAA,YACR,iBAAiB,kBAAkB,EAAE;AAAA,YACrC,kBAAkB;AAAA,UACpB;AAIA,eAAK,wBAAwB,mBAAmB,IAAgB;AAAA,QAClE;AAAA,QACA,oBAAoB,CAAC,UAAmB;AACtC,gBAAM,SAAS,kBAAkB,KAAK;AACtC,eAAK,KAAK;AAAA,YACR,uBAAuB,kBAAkB,EAAE,KAAK,MAAM;AAAA,YACtD,kBAAkB;AAAA,UACpB;AACA,eAAK,mBAAmB,iBAAiB;AAAA,QAC3C;AAAA,QACA,qBAAqB,MAAM;AACzB,eAAK,KAAK;AAAA,YACR,iBAAiB,kBAAkB,EAAE;AAAA,YACrC,kBAAkB;AAAA,UACpB;AACA,eAAK,mBAAmB,iBAAiB;AAAA,QAC3C;AAAA,MACF;AAAA,IACF;AAEF,SAAK,cAAc,iBAAiB;AAAA,EACtC;AAAA,EAEA,MAAc,cAAc,SAAuC;AACjE,QAAI,WAAoB;AAExB,QAAI,KAAK,qBAAqB;AAC5B,iBAAW,MAAM,KAAK,oBAAoB,UAAU;AAAA,IACtD;AAEA,UAAM,aAAa,wBAAwB;AAAA,MACzC,MAAM,KAAK;AAAA,MACX,IAAI,QAAQ;AAAA,MACZ,WAAW,QAAQ;AAAA,MACnB,sBAAsB;AAAA,QACpB,iBAAiB,QAAQ;AAAA,QACzB,aAAa,QAAQ,QAAQ;AAAA,MAC/B;AAAA,MACA;AAAA,MACA,SAAS,sBAAsB,QAAQ,UAAU,GAAG;AAAA,IACtD,CAAC;AAED,SAAK,KAAK,MAAM,gCAAgC,QAAQ,EAAE,IAAI;AAAA,MAC5D,GAAG,QAAQ;AAAA,MACX,kBAAkB;AAAA,IACpB,CAAC;AAED,YAAQ,cAAc,UAAU;AAAA,EAClC;AAAA,EAEA,QAAQ;AACN,SAAK,YAAY,MAAM;AACvB,UAAM,MAAM;AAAA,EACd;AACF;","names":["msg"]}
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../transport/server.ts"],"sourcesContent":["import { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\nimport { ServerHandshakeOptions } from '../router/handshake';\nimport {\n ControlMessageHandshakeRequestSchema,\n HandshakeErrorResponseCodes,\n OpaqueTransportMessage,\n acceptedProtocolVersions,\n PartialTransportMessage,\n TransportClientId,\n handshakeResponseMessage,\n currentProtocolVersion,\n} from './message';\nimport {\n ProvidedServerTransportOptions,\n ServerTransportOptions,\n defaultServerTransportOptions,\n} from './options';\nimport { Transport } from './transport';\nimport { coerceErrorString } from '../util/stringify';\nimport { Static } from '@sinclair/typebox';\nimport { Value } from '@sinclair/typebox/value';\nimport { ProtocolError } from './events';\nimport { Connection } from './connection';\nimport { MessageMetadata } from '../logging';\nimport { SessionWaitingForHandshake } from './sessionStateMachine/SessionWaitingForHandshake';\nimport { Session, SessionState } from './sessionStateMachine/common';\nimport { SessionStateGraph } from './sessionStateMachine/transitions';\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 = new Map<TransportClientId, ParsedMetadata>();\n pendingSessions = new Set<SessionWaitingForHandshake<ConnType>>();\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultServerTransportOptions,\n ...providedOptions,\n };\n this.log?.info(`initiated server transport`, {\n clientId: this.clientId,\n protocolVersion: currentProtocolVersion,\n });\n }\n\n extendHandshake(options: ServerHandshakeOptions) {\n this.handshakeExtensions = options;\n }\n\n send(to: string, msg: PartialTransportMessage): string {\n if (this.getStatus() === 'closed') {\n const err = 'transport is closed, cant send';\n this.log?.error(err, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n\n throw new Error(err);\n }\n\n const session = this.sessions.get(to);\n if (!session) {\n const err = `session to ${to} does not exist`;\n this.log?.error(err, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n\n throw new Error(err);\n }\n\n return session.send(msg);\n }\n\n protected deletePendingSession(\n pendingSession: SessionWaitingForHandshake<ConnType>,\n ) {\n pendingSession.close();\n // we don't dispatch a session disconnect event\n // for a non-identified session, just delete directly\n\n this.pendingSessions.delete(pendingSession);\n }\n\n protected deleteSession(session: Session<ConnType>): void {\n this.sessionHandshakeMetadata.delete(session.to);\n super.deleteSession(session);\n }\n\n protected handleConnection(conn: ConnType) {\n if (this.getStatus() !== 'open') return;\n\n this.log?.info(`new incoming connection`, {\n ...conn.loggingMetadata,\n clientId: this.clientId,\n });\n\n let receivedHandshake = false;\n const pendingSession = SessionStateGraph.entrypoints.WaitingForHandshake(\n this.clientId,\n conn,\n {\n onConnectionClosed: () => {\n this.log?.warn(\n `connection from unknown closed before handshake finished`,\n pendingSession.loggingMetadata,\n );\n\n this.deletePendingSession(pendingSession);\n },\n onConnectionErrored: (err) => {\n const errorString = coerceErrorString(err);\n this.log?.warn(\n `connection from unknown errored before handshake finished: ${errorString}`,\n pendingSession.loggingMetadata,\n );\n\n this.deletePendingSession(pendingSession);\n },\n onHandshakeTimeout: () => {\n this.log?.warn(\n `connection from unknown timed out before handshake finished`,\n pendingSession.loggingMetadata,\n );\n\n this.deletePendingSession(pendingSession);\n },\n onHandshake: (msg) => {\n if (receivedHandshake) {\n this.log?.error(\n `received multiple handshake messages from pending session`,\n {\n ...pendingSession.loggingMetadata,\n connectedTo: msg.from,\n transportMessage: msg,\n },\n );\n\n this.deletePendingSession(pendingSession);\n return;\n }\n\n // let this resolve async, we just need to make sure its only\n // called once so we don't race while transitioning to connected\n // onHandshakeRequest is async as custom validation may be async\n receivedHandshake = true;\n void this.onHandshakeRequest(pendingSession, msg);\n },\n onInvalidHandshake: (reason) => {\n this.log?.error(\n `invalid handshake: ${reason}`,\n pendingSession.loggingMetadata,\n );\n this.deletePendingSession(pendingSession);\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n },\n },\n this.options,\n this.log,\n );\n\n this.pendingSessions.add(pendingSession);\n }\n\n private rejectHandshakeRequest(\n session: SessionWaitingForHandshake<ConnType>,\n to: TransportClientId,\n reason: string,\n code: Static<typeof HandshakeErrorResponseCodes>,\n metadata: MessageMetadata,\n ) {\n session.conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n\n this.log?.warn(reason, metadata);\n\n session.sendHandshake(\n handshakeResponseMessage({\n from: this.clientId,\n to,\n status: {\n ok: false,\n code,\n reason,\n },\n }),\n );\n\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n this.deletePendingSession(session);\n }\n\n protected async onHandshakeRequest(\n session: SessionWaitingForHandshake<ConnType>,\n msg: OpaqueTransportMessage,\n ) {\n // invariant: msg is a handshake request\n if (!Value.Check(ControlMessageHandshakeRequestSchema, msg.payload)) {\n this.rejectHandshakeRequest(\n session,\n msg.from,\n 'received invalid handshake request',\n 'MALFORMED_HANDSHAKE',\n {\n ...session.loggingMetadata,\n transportMessage: msg,\n connectedTo: msg.from,\n validationErrors: [\n ...Value.Errors(ControlMessageHandshakeRequestSchema, msg.payload),\n ],\n },\n );\n\n return;\n }\n\n // invariant: handshake request passes all the validation\n const gotVersion = msg.payload.protocolVersion;\n if (!acceptedProtocolVersions.includes(gotVersion)) {\n this.rejectHandshakeRequest(\n session,\n msg.from,\n `expected protocol version oneof [${acceptedProtocolVersions.toString()}], got ${gotVersion}`,\n 'PROTOCOL_VERSION_MISMATCH',\n {\n ...session.loggingMetadata,\n connectedTo: msg.from,\n transportMessage: msg,\n },\n );\n\n return;\n }\n\n let oldSession = this.sessions.get(msg.from);\n\n // invariant: must pass custom validation if defined\n const parsedMetadata = await this.validateHandshakeMetadata(\n session,\n oldSession,\n msg.payload.metadata,\n msg.from,\n );\n\n if (parsedMetadata === false) {\n return;\n }\n\n // 4 connect cases\n // 1. new session\n // we dont have a session and the client is requesting a new one\n // we can create the session as normal\n // 2. client is reconnecting to an existing session but we don't have it\n // reject this handshake, there's nothing we can do to salvage it\n // 3. transparent reconnect (old session exists and is the same as the client wants)\n // assign to old session\n // 4. hard reconnect (oldSession exists but but the client wants a new one)\n // we close the old session and create a new one\n let connectCase:\n | 'new session'\n | 'unknown session'\n | 'transparent reconnection'\n | 'hard reconnection' = 'new session';\n if (oldSession && oldSession.id === msg.payload.sessionId) {\n connectCase = 'transparent reconnection';\n\n // invariant: ordering must be correct\n const clientNextExpectedSeq =\n msg.payload.expectedSessionState.nextExpectedSeq;\n // TODO: remove nullish coalescing when we're sure this is always set\n const clientNextSentSeq =\n msg.payload.expectedSessionState.nextSentSeq ?? 0;\n const ourNextSeq = oldSession.nextSeq();\n const ourAck = oldSession.ack;\n\n // two incorrect cases where we cannot permit a reconnect:\n // - if the client is about to send a message in the future w.r.t to the server\n // - client.seq > server.ack => nextSentSeq > oldSession.ack\n if (clientNextSentSeq > ourAck) {\n this.rejectHandshakeRequest(\n session,\n msg.from,\n `client is in the future: server wanted next message to be ${ourAck} but client would have sent ${clientNextSentSeq}`,\n 'SESSION_STATE_MISMATCH',\n {\n ...session.loggingMetadata,\n connectedTo: msg.from,\n transportMessage: msg,\n },\n );\n\n return;\n }\n\n // - if the server is about to send a message in the future w.r.t to the client\n // - server.seq > client.ack => oldSession.nextSeq() > nextExpectedSeq\n if (ourNextSeq > clientNextExpectedSeq) {\n this.rejectHandshakeRequest(\n session,\n msg.from,\n `server is in the future: client wanted next message to be ${clientNextExpectedSeq} but server would have sent ${ourNextSeq}`,\n 'SESSION_STATE_MISMATCH',\n {\n ...session.loggingMetadata,\n connectedTo: msg.from,\n transportMessage: msg,\n },\n );\n\n return;\n }\n\n // transparent reconnect seems ok, proceed by transitioning old session\n // to not connected\n if (oldSession.state === SessionState.Connected) {\n const noConnectionSession =\n SessionStateGraph.transition.ConnectedToNoConnection(oldSession, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n\n oldSession = noConnectionSession;\n } else if (oldSession.state === SessionState.Handshaking) {\n const noConnectionSession =\n SessionStateGraph.transition.HandshakingToNoConnection(oldSession, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n\n oldSession = noConnectionSession;\n } else if (oldSession.state === SessionState.Connecting) {\n const noConnectionSession =\n SessionStateGraph.transition.ConnectingToNoConnection(oldSession, {\n onSessionGracePeriodElapsed: () => {\n this.onSessionGracePeriodElapsed(noConnectionSession);\n },\n });\n\n oldSession = noConnectionSession;\n }\n\n this.updateSession(oldSession);\n } else if (oldSession) {\n connectCase = 'hard reconnection';\n\n // just nuke the old session entirely and proceed as if this was new\n this.deleteSession(oldSession);\n oldSession = undefined;\n } else {\n connectCase = 'unknown session';\n\n const clientNextExpectedSeq =\n msg.payload.expectedSessionState.nextExpectedSeq;\n // TODO: remove nullish coalescing when we're sure this is always set\n const clientNextSentSeq =\n msg.payload.expectedSessionState.nextSentSeq ?? 0;\n\n if (clientNextSentSeq > 0 || clientNextExpectedSeq > 0) {\n // we don't have a session, but the client is trying to reconnect\n // to an old session. we can't do anything about this, so we reject\n this.rejectHandshakeRequest(\n session,\n msg.from,\n `client is trying to reconnect to a session the server don't know about: ${msg.payload.sessionId}`,\n 'SESSION_STATE_MISMATCH',\n {\n ...session.loggingMetadata,\n connectedTo: msg.from,\n transportMessage: msg,\n },\n );\n return;\n }\n }\n\n // from this point on, we're committed to connecting\n const sessionId = msg.payload.sessionId;\n this.log?.info(\n `handshake from ${msg.from} ok (${connectCase}), responding with handshake success`,\n {\n ...session.loggingMetadata,\n connectedTo: msg.from,\n },\n );\n const responseMsg = handshakeResponseMessage({\n from: this.clientId,\n to: msg.from,\n status: {\n ok: true,\n sessionId,\n },\n });\n session.sendHandshake(responseMsg);\n\n // transition\n const connectedSession =\n SessionStateGraph.transition.WaitingForHandshakeToConnected(\n session,\n // by this point oldSession is either no connection or we dont have an old session\n oldSession,\n sessionId,\n msg.from,\n msg.tracing,\n {\n onConnectionErrored: (err) => {\n // just log, when we error we also emit close\n const errStr = coerceErrorString(err);\n this.log?.warn(\n `connection to ${connectedSession.to} errored: ${errStr}`,\n connectedSession.loggingMetadata,\n );\n },\n onConnectionClosed: () => {\n this.log?.info(\n `connection to ${connectedSession.to} closed`,\n connectedSession.loggingMetadata,\n );\n this.onConnClosed(connectedSession);\n },\n onMessage: (msg) => this.handleMsg(msg),\n onInvalidMessage: (reason) => {\n this.protocolError(ProtocolError.MessageOrderingViolated, reason);\n this.deleteSession(connectedSession);\n },\n },\n gotVersion,\n );\n\n this.sessionHandshakeMetadata.set(connectedSession.to, parsedMetadata);\n this.updateSession(connectedSession);\n this.pendingSessions.delete(session);\n connectedSession.startActiveHeartbeat();\n }\n\n private async validateHandshakeMetadata(\n handshakingSession: SessionWaitingForHandshake<ConnType>,\n existingSession: 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 this.rejectHandshakeRequest(\n handshakingSession,\n from,\n 'received malformed handshake metadata',\n 'MALFORMED_HANDSHAKE_META',\n {\n ...handshakingSession.loggingMetadata,\n connectedTo: from,\n validationErrors: [\n ...Value.Errors(this.handshakeExtensions.schema, rawMetadata),\n ],\n },\n );\n\n return false;\n }\n\n const previousParsedMetadata = existingSession\n ? this.sessionHandshakeMetadata.get(existingSession.to)\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 this.rejectHandshakeRequest(\n handshakingSession,\n from,\n 'rejected by handshake handler',\n 'REJECTED_BY_CUSTOM_HANDLER',\n {\n ...handshakingSession.loggingMetadata,\n connectedTo: from,\n clientId: this.clientId,\n },\n );\n\n return false;\n }\n }\n\n return parsedMetadata;\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAAA,SAAS,sBAAsB;AAqB/B,SAAS,aAAa;AAQf,IAAe,kBAAf,cAEG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA,2BAA2B,oBAAI,IAAuC;AAAA,EACtE,kBAAkB,oBAAI,IAA0C;AAAA,EAEhE,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,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,EAEA,KAAK,IAAY,KAAsC;AACrD,QAAI,KAAK,UAAU,MAAM,UAAU;AACjC,YAAM,MAAM;AACZ,WAAK,KAAK,MAAM,KAAK;AAAA,QACnB,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AAED,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,UAAM,UAAU,KAAK,SAAS,IAAI,EAAE;AACpC,QAAI,CAAC,SAAS;AACZ,YAAM,MAAM,cAAc,EAAE;AAC5B,WAAK,KAAK,MAAM,KAAK;AAAA,QACnB,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AAED,YAAM,IAAI,MAAM,GAAG;AAAA,IACrB;AAEA,WAAO,QAAQ,KAAK,GAAG;AAAA,EACzB;AAAA,EAEU,qBACR,gBACA;AACA,mBAAe,MAAM;AAIrB,SAAK,gBAAgB,OAAO,cAAc;AAAA,EAC5C;AAAA,EAEU,cAAc,SAAkC;AACxD,SAAK,yBAAyB,OAAO,QAAQ,EAAE;AAC/C,UAAM,cAAc,OAAO;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU,MAAM;AAAQ;AAEjC,SAAK,KAAK,KAAK,2BAA2B;AAAA,MACxC,GAAG,KAAK;AAAA,MACR,UAAU,KAAK;AAAA,IACjB,CAAC;AAED,QAAI,oBAAoB;AACxB,UAAM,iBAAiB,kBAAkB,YAAY;AAAA,MACnD,KAAK;AAAA,MACL;AAAA,MACA;AAAA,QACE,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR;AAAA,YACA,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,qBAAqB,CAAC,QAAQ;AAC5B,gBAAM,cAAc,kBAAkB,GAAG;AACzC,eAAK,KAAK;AAAA,YACR,8DAA8D,WAAW;AAAA,YACzE,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR;AAAA,YACA,eAAe;AAAA,UACjB;AAEA,eAAK,qBAAqB,cAAc;AAAA,QAC1C;AAAA,QACA,aAAa,CAAC,QAAQ;AACpB,cAAI,mBAAmB;AACrB,iBAAK,KAAK;AAAA,cACR;AAAA,cACA;AAAA,gBACE,GAAG,eAAe;AAAA,gBAClB,aAAa,IAAI;AAAA,gBACjB,kBAAkB;AAAA,cACpB;AAAA,YACF;AAEA,iBAAK,qBAAqB,cAAc;AACxC;AAAA,UACF;AAKA,8BAAoB;AACpB,eAAK,KAAK,mBAAmB,gBAAgB,GAAG;AAAA,QAClD;AAAA,QACA,oBAAoB,CAAC,WAAW;AAC9B,eAAK,KAAK;AAAA,YACR,sBAAsB,MAAM;AAAA,YAC5B,eAAe;AAAA,UACjB;AACA,eAAK,qBAAqB,cAAc;AACxC,eAAK,cAAc,cAAc,iBAAiB,MAAM;AAAA,QAC1D;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL,KAAK;AAAA,IACP;AAEA,SAAK,gBAAgB,IAAI,cAAc;AAAA,EACzC;AAAA,EAEQ,uBACN,SACA,IACA,QACA,MACA,UACA;AACA,YAAQ,KAAK,WAAW,KAAK,UAAU;AAAA,MACrC,MAAM,eAAe;AAAA,MACrB,SAAS;AAAA,IACX,CAAC;AAED,SAAK,KAAK,KAAK,QAAQ,QAAQ;AAE/B,YAAQ;AAAA,MACN,yBAAyB;AAAA,QACvB,MAAM,KAAK;AAAA,QACX;AAAA,QACA,QAAQ;AAAA,UACN,IAAI;AAAA,UACJ;AAAA,UACA;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,SAAK,qBAAqB,OAAO;AAAA,EACnC;AAAA,EAEA,MAAgB,mBACd,SACA,KACA;AAEA,QAAI,CAAC,MAAM,MAAM,sCAAsC,IAAI,OAAO,GAAG;AACnE,WAAK;AAAA,QACH;AAAA,QACA,IAAI;AAAA,QACJ;AAAA,QACA;AAAA,QACA;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,kBAAkB;AAAA,UAClB,aAAa,IAAI;AAAA,UACjB,kBAAkB;AAAA,YAChB,GAAG,MAAM,OAAO,sCAAsC,IAAI,OAAO;AAAA,UACnE;AAAA,QACF;AAAA,MACF;AAEA;AAAA,IACF;AAGA,UAAM,aAAa,IAAI,QAAQ;AAC/B,QAAI,CAAC,yBAAyB,SAAS,UAAU,GAAG;AAClD,WAAK;AAAA,QACH;AAAA,QACA,IAAI;AAAA,QACJ,oCAAoC,yBAAyB,SAAS,CAAC,UAAU,UAAU;AAAA,QAC3F;AAAA,QACA;AAAA,UACE,GAAG,QAAQ;AAAA,UACX,aAAa,IAAI;AAAA,UACjB,kBAAkB;AAAA,QACpB;AAAA,MACF;AAEA;AAAA,IACF;AAEA,QAAI,aAAa,KAAK,SAAS,IAAI,IAAI,IAAI;AAG3C,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,IAAI,QAAQ;AAAA,MACZ,IAAI;AAAA,IACN;AAEA,QAAI,mBAAmB,OAAO;AAC5B;AAAA,IACF;AAYA,QAAI,cAIsB;AAC1B,QAAI,cAAc,WAAW,OAAO,IAAI,QAAQ,WAAW;AACzD,oBAAc;AAGd,YAAM,wBACJ,IAAI,QAAQ,qBAAqB;AAEnC,YAAM,oBACJ,IAAI,QAAQ,qBAAqB,eAAe;AAClD,YAAM,aAAa,WAAW,QAAQ;AACtC,YAAM,SAAS,WAAW;AAK1B,UAAI,oBAAoB,QAAQ;AAC9B,aAAK;AAAA,UACH;AAAA,UACA,IAAI;AAAA,UACJ,6DAA6D,MAAM,+BAA+B,iBAAiB;AAAA,UACnH;AAAA,UACA;AAAA,YACE,GAAG,QAAQ;AAAA,YACX,aAAa,IAAI;AAAA,YACjB,kBAAkB;AAAA,UACpB;AAAA,QACF;AAEA;AAAA,MACF;AAIA,UAAI,aAAa,uBAAuB;AACtC,aAAK;AAAA,UACH;AAAA,UACA,IAAI;AAAA,UACJ,6DAA6D,qBAAqB,+BAA+B,UAAU;AAAA,UAC3H;AAAA,UACA;AAAA,YACE,GAAG,QAAQ;AAAA,YACX,aAAa,IAAI;AAAA,YACjB,kBAAkB;AAAA,UACpB;AAAA,QACF;AAEA;AAAA,MACF;AAIA,UAAI,WAAW,uCAAkC;AAC/C,cAAM,sBACJ,kBAAkB,WAAW,wBAAwB,YAAY;AAAA,UAC/D,6BAA6B,MAAM;AACjC,iBAAK,4BAA4B,mBAAmB;AAAA,UACtD;AAAA,QACF,CAAC;AAEH,qBAAa;AAAA,MACf,WAAW,WAAW,2CAAoC;AACxD,cAAM,sBACJ,kBAAkB,WAAW,0BAA0B,YAAY;AAAA,UACjE,6BAA6B,MAAM;AACjC,iBAAK,4BAA4B,mBAAmB;AAAA,UACtD;AAAA,QACF,CAAC;AAEH,qBAAa;AAAA,MACf,WAAW,WAAW,yCAAmC;AACvD,cAAM,sBACJ,kBAAkB,WAAW,yBAAyB,YAAY;AAAA,UAChE,6BAA6B,MAAM;AACjC,iBAAK,4BAA4B,mBAAmB;AAAA,UACtD;AAAA,QACF,CAAC;AAEH,qBAAa;AAAA,MACf;AAEA,WAAK,cAAc,UAAU;AAAA,IAC/B,WAAW,YAAY;AACrB,oBAAc;AAGd,WAAK,cAAc,UAAU;AAC7B,mBAAa;AAAA,IACf,OAAO;AACL,oBAAc;AAEd,YAAM,wBACJ,IAAI,QAAQ,qBAAqB;AAEnC,YAAM,oBACJ,IAAI,QAAQ,qBAAqB,eAAe;AAElD,UAAI,oBAAoB,KAAK,wBAAwB,GAAG;AAGtD,aAAK;AAAA,UACH;AAAA,UACA,IAAI;AAAA,UACJ,2EAA2E,IAAI,QAAQ,SAAS;AAAA,UAChG;AAAA,UACA;AAAA,YACE,GAAG,QAAQ;AAAA,YACX,aAAa,IAAI;AAAA,YACjB,kBAAkB;AAAA,UACpB;AAAA,QACF;AACA;AAAA,MACF;AAAA,IACF;AAGA,UAAM,YAAY,IAAI,QAAQ;AAC9B,SAAK,KAAK;AAAA,MACR,kBAAkB,IAAI,IAAI,QAAQ,WAAW;AAAA,MAC7C;AAAA,QACE,GAAG,QAAQ;AAAA,QACX,aAAa,IAAI;AAAA,MACnB;AAAA,IACF;AACA,UAAM,cAAc,yBAAyB;AAAA,MAC3C,MAAM,KAAK;AAAA,MACX,IAAI,IAAI;AAAA,MACR,QAAQ;AAAA,QACN,IAAI;AAAA,QACJ;AAAA,MACF;AAAA,IACF,CAAC;AACD,YAAQ,cAAc,WAAW;AAGjC,UAAM,mBACJ,kBAAkB,WAAW;AAAA,MAC3B;AAAA;AAAA,MAEA;AAAA,MACA;AAAA,MACA,IAAI;AAAA,MACJ,IAAI;AAAA,MACJ;AAAA,QACE,qBAAqB,CAAC,QAAQ;AAE5B,gBAAM,SAAS,kBAAkB,GAAG;AACpC,eAAK,KAAK;AAAA,YACR,iBAAiB,iBAAiB,EAAE,aAAa,MAAM;AAAA,YACvD,iBAAiB;AAAA,UACnB;AAAA,QACF;AAAA,QACA,oBAAoB,MAAM;AACxB,eAAK,KAAK;AAAA,YACR,iBAAiB,iBAAiB,EAAE;AAAA,YACpC,iBAAiB;AAAA,UACnB;AACA,eAAK,aAAa,gBAAgB;AAAA,QACpC;AAAA,QACA,WAAW,CAACA,SAAQ,KAAK,UAAUA,IAAG;AAAA,QACtC,kBAAkB,CAAC,WAAW;AAC5B,eAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,eAAK,cAAc,gBAAgB;AAAA,QACrC;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAEF,SAAK,yBAAyB,IAAI,iBAAiB,IAAI,cAAc;AACrE,SAAK,cAAc,gBAAgB;AACnC,SAAK,gBAAgB,OAAO,OAAO;AACnC,qBAAiB,qBAAqB;AAAA,EACxC;AAAA,EAEA,MAAc,0BACZ,oBACA,iBACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,MAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,YACE,GAAG,mBAAmB;AAAA,YACtB,aAAa;AAAA,YACb,kBAAkB;AAAA,cAChB,GAAG,MAAM,OAAO,KAAK,oBAAoB,QAAQ,WAAW;AAAA,YAC9D;AAAA,UACF;AAAA,QACF;AAEA,eAAO;AAAA,MACT;AAEA,YAAM,yBAAyB,kBAC3B,KAAK,yBAAyB,IAAI,gBAAgB,EAAE,IACpD;AAEJ,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,YACE,GAAG,mBAAmB;AAAA,YACtB,aAAa;AAAA,YACb,UAAU,KAAK;AAAA,UACjB;AAAA,QACF;AAEA,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;","names":["msg"]}