@replit/river 0.23.0 → 0.23.2

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 (63) hide show
  1. package/dist/{chunk-PCKHBAVP.js → chunk-3ZS3RHZF.js} +2 -2
  2. package/dist/chunk-3ZS3RHZF.js.map +1 -0
  3. package/dist/{chunk-DZOATC6M.js → chunk-6Z4YTBOM.js} +2 -2
  4. package/dist/{chunk-ZF2UFTNN.js → chunk-KPLLLP3F.js} +2 -2
  5. package/dist/{chunk-MJUFKPBT.js → chunk-LHLQY5FR.js} +45 -25
  6. package/dist/chunk-LHLQY5FR.js.map +1 -0
  7. package/dist/{chunk-OTQNCLFH.js → chunk-SPP7ACQN.js} +29 -5
  8. package/dist/chunk-SPP7ACQN.js.map +1 -0
  9. package/dist/{chunk-VOJVLWVX.js → chunk-ZPUDMABQ.js} +32 -16
  10. package/dist/chunk-ZPUDMABQ.js.map +1 -0
  11. package/dist/{connection-7582fb92.d.ts → connection-6eb86e22.d.ts} +1 -1
  12. package/dist/{connection-5685d817.d.ts → connection-a25109b5.d.ts} +1 -1
  13. package/dist/{index-a6fe0edd.d.ts → index-72b688b7.d.ts} +16 -16
  14. package/dist/logging/index.cjs +28 -4
  15. package/dist/logging/index.cjs.map +1 -1
  16. package/dist/logging/index.d.cts +3 -2
  17. package/dist/logging/index.d.ts +3 -2
  18. package/dist/logging/index.js +1 -1
  19. package/dist/router/index.cjs +30 -14
  20. package/dist/router/index.cjs.map +1 -1
  21. package/dist/router/index.d.cts +6 -5
  22. package/dist/router/index.d.ts +6 -5
  23. package/dist/router/index.js +3 -3
  24. package/dist/{services-be91b485.d.ts → services-0af830a4.d.ts} +3 -1
  25. package/dist/{services-eb9326a1.d.ts → services-eeb6e56f.d.ts} +3 -1
  26. package/dist/transport/impls/uds/client.cjs +31 -19
  27. package/dist/transport/impls/uds/client.cjs.map +1 -1
  28. package/dist/transport/impls/uds/client.d.cts +3 -2
  29. package/dist/transport/impls/uds/client.d.ts +3 -2
  30. package/dist/transport/impls/uds/client.js +4 -4
  31. package/dist/transport/impls/uds/server.cjs +30 -19
  32. package/dist/transport/impls/uds/server.cjs.map +1 -1
  33. package/dist/transport/impls/uds/server.d.cts +3 -2
  34. package/dist/transport/impls/uds/server.d.ts +3 -2
  35. package/dist/transport/impls/uds/server.js +4 -4
  36. package/dist/transport/impls/ws/client.cjs +31 -19
  37. package/dist/transport/impls/ws/client.cjs.map +1 -1
  38. package/dist/transport/impls/ws/client.d.cts +3 -2
  39. package/dist/transport/impls/ws/client.d.ts +3 -2
  40. package/dist/transport/impls/ws/client.js +4 -4
  41. package/dist/transport/impls/ws/server.cjs +30 -19
  42. package/dist/transport/impls/ws/server.cjs.map +1 -1
  43. package/dist/transport/impls/ws/server.d.cts +4 -3
  44. package/dist/transport/impls/ws/server.d.ts +4 -3
  45. package/dist/transport/impls/ws/server.js +4 -4
  46. package/dist/transport/index.cjs +43 -23
  47. package/dist/transport/index.cjs.map +1 -1
  48. package/dist/transport/index.d.cts +3 -2
  49. package/dist/transport/index.d.ts +3 -2
  50. package/dist/transport/index.js +3 -3
  51. package/dist/util/testHelpers.cjs +7 -7
  52. package/dist/util/testHelpers.cjs.map +1 -1
  53. package/dist/util/testHelpers.d.cts +3 -2
  54. package/dist/util/testHelpers.d.ts +3 -2
  55. package/dist/util/testHelpers.js +4 -4
  56. package/dist/util/testHelpers.js.map +1 -1
  57. package/package.json +4 -4
  58. package/dist/chunk-MJUFKPBT.js.map +0 -1
  59. package/dist/chunk-OTQNCLFH.js.map +0 -1
  60. package/dist/chunk-PCKHBAVP.js.map +0 -1
  61. package/dist/chunk-VOJVLWVX.js.map +0 -1
  62. /package/dist/{chunk-DZOATC6M.js.map → chunk-6Z4YTBOM.js.map} +0 -0
  63. /package/dist/{chunk-ZF2UFTNN.js.map → chunk-KPLLLP3F.js.map} +0 -0
@@ -146,7 +146,7 @@ var import_nanoid2 = require("nanoid");
146
146
  var import_api = require("@opentelemetry/api");
147
147
 
148
148
  // package.json
149
- var version = "0.23.0";
149
+ var version = "0.23.2";
150
150
 
151
151
  // tracing/index.ts
152
152
  function createSessionTelemetryInfo(session, propagationCtx) {
@@ -271,7 +271,7 @@ var Session = class {
271
271
  const fullMsg = this.constructMsg(msg);
272
272
  log?.debug(`sending msg`, {
273
273
  ...this.loggingMetadata,
274
- fullTransportMessage: fullMsg
274
+ transportMessage: fullMsg
275
275
  });
276
276
  if (this.connection) {
277
277
  const ok = this.connection.send(this.codec.toBuffer(fullMsg));
@@ -281,13 +281,13 @@ var Session = class {
281
281
  `failed to send msg to ${fullMsg.to}, connection is probably dead`,
282
282
  {
283
283
  ...this.loggingMetadata,
284
- fullTransportMessage: fullMsg
284
+ transportMessage: fullMsg
285
285
  }
286
286
  );
287
287
  } else {
288
288
  log?.info(
289
289
  `failed to send msg to ${fullMsg.to}, connection not ready yet`,
290
- { ...this.loggingMetadata, fullTransportMessage: fullMsg }
290
+ { ...this.loggingMetadata, transportMessage: fullMsg }
291
291
  );
292
292
  }
293
293
  return fullMsg.id;
@@ -328,7 +328,7 @@ var Session = class {
328
328
  for (const msg of this.sendBuffer) {
329
329
  log?.debug(`resending msg`, {
330
330
  ...this.loggingMetadata,
331
- fullTransportMessage: msg,
331
+ transportMessage: msg,
332
332
  connId: conn.id
333
333
  });
334
334
  const ok = conn.send(this.codec.toBuffer(msg));
@@ -340,7 +340,7 @@ var Session = class {
340
340
  });
341
341
  log?.error(errMsg, {
342
342
  ...this.loggingMetadata,
343
- fullTransportMessage: msg,
343
+ transportMessage: msg,
344
344
  connId: conn.id,
345
345
  tags: ["invariant-violation"]
346
346
  });
@@ -373,8 +373,8 @@ var Session = class {
373
373
  replaceWithNewConnection(newConn) {
374
374
  this.closeStaleConnection(newConn);
375
375
  this.cancelGrace();
376
- this.connection = newConn;
377
376
  this.sendBufferedMessages(newConn);
377
+ this.connection = newConn;
378
378
  }
379
379
  beginGrace(cb) {
380
380
  log?.info(
@@ -658,7 +658,10 @@ var Transport = class {
658
658
  }
659
659
  if (!import_value.Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {
660
660
  log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {
661
- clientId: this.clientId
661
+ clientId: this.clientId,
662
+ validationErrors: [
663
+ ...import_value.Value.Errors(OpaqueTransportMessageSchema, parsedMsg)
664
+ ]
662
665
  });
663
666
  return null;
664
667
  }
@@ -676,7 +679,7 @@ var Transport = class {
676
679
  if (!session) {
677
680
  log?.error(`no existing session for ${msg.from}`, {
678
681
  clientId: this.clientId,
679
- fullTransportMessage: msg,
682
+ transportMessage: msg,
680
683
  tags: ["invariant-violation"]
681
684
  });
682
685
  return;
@@ -684,19 +687,19 @@ var Transport = class {
684
687
  session.cancelGrace();
685
688
  log?.debug(`received msg`, {
686
689
  clientId: this.clientId,
687
- fullTransportMessage: msg
690
+ transportMessage: msg
688
691
  });
689
692
  if (msg.seq !== session.nextExpectedSeq) {
690
693
  if (msg.seq < session.nextExpectedSeq) {
691
694
  log?.debug(
692
695
  `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,
693
- { clientId: this.clientId, fullTransportMessage: msg }
696
+ { clientId: this.clientId, transportMessage: msg }
694
697
  );
695
698
  } else {
696
699
  const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;
697
700
  log?.error(`${errMsg}, marking connection as dead`, {
698
701
  clientId: this.clientId,
699
- fullTransportMessage: msg,
702
+ transportMessage: msg,
700
703
  tags: ["invariant-violation"]
701
704
  });
702
705
  this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);
@@ -714,7 +717,7 @@ var Transport = class {
714
717
  } else {
715
718
  log?.debug(`discarding msg (ack bit set)`, {
716
719
  clientId: this.clientId,
717
- fullTransportMessage: msg
720
+ transportMessage: msg
718
721
  });
719
722
  }
720
723
  }
@@ -745,7 +748,7 @@ var Transport = class {
745
748
  const err = "transport is destroyed, cant send";
746
749
  log?.error(err, {
747
750
  clientId: this.clientId,
748
- partialTransportMessage: msg,
751
+ transportMessage: msg,
749
752
  tags: ["invariant-violation"]
750
753
  });
751
754
  this.protocolError(ProtocolError.UseAfterDestroy, err);
@@ -753,7 +756,7 @@ var Transport = class {
753
756
  } else if (this.state === "closed") {
754
757
  log?.info(`transport closed when sending, discarding`, {
755
758
  clientId: this.clientId,
756
- partialTransportMessage: msg
759
+ transportMessage: msg
757
760
  });
758
761
  return void 0;
759
762
  }
@@ -923,14 +926,18 @@ var ServerTransport = class extends Transport {
923
926
  conn.send(this.codec.toBuffer(responseMsg));
924
927
  log?.warn(`received malformed handshake metadata from ${from}`, {
925
928
  clientId: this.clientId,
926
- connId: conn.id
929
+ connId: conn.id,
930
+ validationErrors: [
931
+ ...import_value.Value.Errors(this.handshakeExtensions.schema, rawMetadata)
932
+ ]
927
933
  });
928
934
  this.protocolError(ProtocolError.HandshakeFailed, reason);
929
935
  return false;
930
936
  }
937
+ const previousParsedMetadata = session ? this.sessionHandshakeMetadata.get(session) : void 0;
931
938
  parsedMetadata = await this.handshakeExtensions.validate(
932
939
  rawMetadata,
933
- session
940
+ previousParsedMetadata
934
941
  );
935
942
  if (parsedMetadata === false) {
936
943
  const reason = "rejected by handshake handler";
@@ -977,11 +984,15 @@ var ServerTransport = class extends Transport {
977
984
  reason
978
985
  });
979
986
  conn.send(this.codec.toBuffer(responseMsg2));
980
- const logData = { ...parsed.payload ?? {}, metadata: "redacted" };
981
987
  log?.warn(reason, {
982
988
  clientId: this.clientId,
983
989
  connId: conn.id,
984
- partialTransportMessage: { ...parsed, payload: logData }
990
+ // safe to log metadata here as we remove the payload
991
+ // before passing it to user-land
992
+ transportMessage: parsed,
993
+ validationErrors: [
994
+ ...import_value.Value.Errors(ControlMessageHandshakeRequestSchema, parsed.payload)
995
+ ]
985
996
  });
986
997
  this.protocolError(
987
998
  ProtocolError.HandshakeFailed,
@@ -1 +1 @@
1
- {"version":3,"sources":["../../../../transport/impls/ws/server.ts","../../../../transport/transport.ts","../../../../transport/message.ts","../../../../logging/log.ts","../../../../transport/events.ts","../../../../transport/session.ts","../../../../tracing/index.ts","../../../../package.json","../../../../util/stringify.ts","../../../../codec/json.ts","../../../../transport/impls/ws/connection.ts"],"sourcesContent":["import { TransportClientId } from '../../message';\nimport {\n ServerTransport,\n ProvidedServerTransportOptions,\n} from '../../transport';\nimport { WebSocketServer } from 'ws';\nimport { WebSocketConnection } from './connection';\nimport { WsLike } from './wslike';\n\nexport class WebSocketServerTransport extends ServerTransport<WebSocketConnection> {\n wss: WebSocketServer;\n\n constructor(\n wss: WebSocketServer,\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.wss = wss;\n this.wss.on('connection', this.connectionHandler);\n }\n\n connectionHandler = (ws: WsLike) => {\n const conn = new WebSocketConnection(ws);\n this.handleConnection(conn);\n };\n\n close() {\n super.close();\n this.wss.off('connection', this.connectionHandler);\n }\n}\n","import { Codec } from '../codec/types';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n OpaqueTransportMessage,\n OpaqueTransportMessageSchema,\n TransportClientId,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n handshakeRequestMessage,\n handshakeResponseMessage,\n PartialTransportMessage,\n ControlFlags,\n ControlMessagePayloadSchema,\n isAck,\n PROTOCOL_VERSION,\n ClientHandshakeOptions,\n ServerHandshakeOptions,\n} from './message';\nimport { log } from '../logging/log';\nimport {\n EventDispatcher,\n EventHandler,\n EventTypes,\n ProtocolError,\n ProtocolErrorType,\n} from './events';\nimport { Connection, Session, SessionOptions } from './session';\nimport { Static, TSchema } from '@sinclair/typebox';\nimport { coerceErrorString } from '../util/stringify';\nimport { ConnectionRetryOptions, LeakyBucketRateLimit } from './rateLimit';\nimport { NaiveJsonCodec } from '../codec';\nimport tracer, {\n PropagationContext,\n createConnectionTelemetryInfo,\n getPropagationContext,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Represents the possible states of a transport.\n * @property {'open'} open - The transport is open and operational (note that this doesn't mean it is actively connected)\n * @property {'closed'} closed - The transport is closed and not operational, but can be reopened.\n * @property {'destroyed'} destroyed - The transport is permanently destroyed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed' | 'destroyed';\n\ntype TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n heartbeatIntervalMs: 1_000,\n heartbeatsUntilDead: 2,\n sessionDisconnectGraceMs: 5_000,\n codec: NaiveJsonCodec,\n};\n\ntype ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n baseIntervalMs: 250,\n maxJitterMs: 200,\n maxBackoffMs: 32_000,\n attemptBudgetCapacity: 5,\n budgetRestoreIntervalMs: 200,\n};\n\nconst defaultClientTransportOptions: ClientTransportOptions = {\n ...defaultTransportOptions,\n ...defaultConnectionRetryOptions,\n};\n\ntype ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nconst defaultServerTransportOptions: ServerTransportOptions = {\n ...defaultTransportOptions,\n};\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions and connections. Its responsibilities include:\n *\n * 1) Constructing a new {@link Session} and {@link Connection} on {@link TransportMessage}s from new clients.\n * After constructing the {@link Connection}, {@link onConnect} is called which adds it to the connection map.\n * 2) Delegating message listening of the connection to the newly created {@link Connection}.\n * From this point on, the {@link Connection} is responsible for *reading* and *writing*\n * messages from the connection.\n * 3) When a connection is closed, the {@link Transport} calls {@link onDisconnect} which closes the\n * connection via {@link Connection.close} and removes it from the {@link connections} map.\n\n *\n * ```plaintext\n * ▲\n * incoming │\n * messages │\n * ▼\n * ┌─────────────┐ 1:N ┌───────────┐ 1:1* ┌────────────┐\n * │ Transport │ ◄─────► │ Session │ ◄─────► │ Connection │\n * └─────────────┘ └───────────┘ └────────────┘\n * ▲ * (may or may not be initialized yet)\n * │\n * ▼\n * ┌───────────┐\n * │ Message │\n * │ Listeners │\n * └───────────┘\n * ```\n * @abstract\n */\nexport abstract class Transport<ConnType extends Connection> {\n /**\n * A flag indicating whether the transport has been destroyed.\n * A destroyed transport will not attempt to reconnect and cannot be used again.\n */\n state: TransportStatus;\n\n /**\n * The {@link Codec} used to encode and decode messages.\n */\n codec: Codec;\n\n /**\n * The client ID of this transport.\n */\n clientId: TransportClientId;\n\n /**\n * The map of {@link Session}s managed by this transport.\n */\n sessions: Map<TransportClientId, Session<ConnType>>;\n\n /**\n * The map of {@link Connection}s managed by this transport.\n */\n get connections() {\n return new Map(\n [...this.sessions]\n .map(([client, session]) => [client, session.connection])\n .filter((entry): entry is [string, ConnType] => entry[1] !== undefined),\n );\n }\n\n /**\n * The event dispatcher for handling events of type EventTypes.\n */\n eventDispatcher: EventDispatcher<EventTypes>;\n\n /**\n * The options for this transport.\n */\n protected options: TransportOptions;\n\n /**\n * Creates a new Transport instance.\n * This should also set up {@link onConnect}, and {@link onDisconnect} listeners.\n * @param codec The codec used to encode and decode messages.\n * @param clientId The client ID of this transport.\n */\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedTransportOptions,\n ) {\n this.options = { ...defaultTransportOptions, ...providedOptions };\n this.eventDispatcher = new EventDispatcher();\n this.sessions = new Map();\n this.codec = this.options.codec;\n this.clientId = clientId;\n this.state = 'open';\n }\n\n /**\n * This is called immediately after a new connection is established and we\n * may or may not know the identity of the connected client.\n * It should attach all the necessary listeners to the connection for lifecycle\n * events (i.e. data, close, error)\n *\n * This method is implemented by {@link ClientTransport} and {@link ServerTransport}.\n */\n protected abstract handleConnection(\n conn: ConnType,\n to: TransportClientId,\n ): void;\n\n /**\n * Called when a new connection is established\n * and we know the identity of the connected client.\n * @param conn The connection object.\n */\n protected onConnect(\n conn: ConnType,\n connectedTo: TransportClientId,\n session: Session<ConnType>,\n isReconnect: boolean,\n ) {\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'connect',\n conn,\n });\n\n conn.telemetry = createConnectionTelemetryInfo(\n conn,\n session.telemetry.span,\n );\n\n if (isReconnect) {\n session.replaceWithNewConnection(conn);\n log?.info(`reconnected to ${connectedTo}`, session.loggingMetadata);\n }\n }\n\n protected createSession(\n to: TransportClientId,\n conn?: ConnType,\n propagationCtx?: PropagationContext,\n ) {\n const session = new Session<ConnType>(\n conn,\n this.clientId,\n to,\n this.options,\n propagationCtx,\n );\n this.sessions.set(session.to, session);\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'connect',\n session,\n });\n return session;\n }\n\n protected getOrCreateSession(\n to: TransportClientId,\n conn?: ConnType,\n sessionId?: string,\n propagationCtx?: PropagationContext,\n ) {\n let session = this.sessions.get(to);\n let isReconnect = session !== undefined;\n\n if (\n session?.advertisedSessionId !== undefined &&\n sessionId !== undefined &&\n session.advertisedSessionId !== sessionId\n ) {\n log?.info(\n `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,\n session.loggingMetadata,\n );\n this.deleteSession(session);\n isReconnect = false;\n session = undefined;\n }\n\n if (!session) {\n session = this.createSession(to, conn, propagationCtx);\n log?.info(\n `no session for ${to}, created a new one`,\n session.loggingMetadata,\n );\n }\n\n if (sessionId !== undefined) {\n session.advertisedSessionId = sessionId;\n }\n\n return { session, isReconnect };\n }\n\n protected deleteSession(session: Session<ConnType>) {\n session.close();\n session.telemetry.span.end();\n this.sessions.delete(session.to);\n log?.info(\n `session ${session.id} disconnect from ${session.to}`,\n session.loggingMetadata,\n );\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'disconnect',\n session,\n });\n }\n\n /**\n * The downstream implementation needs to call this when a connection is closed.\n * @param conn The connection object.\n * @param connectedTo The peer we are connected to.\n */\n protected onDisconnect(conn: ConnType, session: Session<ConnType>) {\n conn.telemetry?.span.end();\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'disconnect',\n conn,\n });\n\n session.connection = undefined;\n session.beginGrace(() => {\n session.telemetry.span.addEvent('session grace period expired');\n this.deleteSession(session);\n });\n }\n\n /**\n * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.\n * @param msg The message to parse.\n * @returns The parsed message, or null if the message is malformed or invalid.\n */\n protected parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n const parsedMsg = this.codec.fromBuffer(msg);\n\n if (parsedMsg === null) {\n const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n log?.error(`received malformed msg, killing conn: ${decodedBuffer}`, {\n clientId: this.clientId,\n });\n return null;\n }\n\n if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n clientId: this.clientId,\n });\n return null;\n }\n\n return parsedMsg;\n }\n\n /**\n * Called when a message is received by this transport.\n * You generally shouldn't need to override this in downstream transport implementations.\n * @param msg The received message.\n */\n protected handleMsg(msg: OpaqueTransportMessage) {\n if (this.state !== 'open') return;\n const session = this.sessions.get(msg.from);\n if (!session) {\n log?.error(`no existing session for ${msg.from}`, {\n clientId: this.clientId,\n fullTransportMessage: msg,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n // got a msg so we know the other end is alive, reset the grace period\n session.cancelGrace();\n\n log?.debug(`received msg`, {\n clientId: this.clientId,\n fullTransportMessage: msg,\n });\n if (msg.seq !== session.nextExpectedSeq) {\n if (msg.seq < session.nextExpectedSeq) {\n log?.debug(\n `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,\n { clientId: this.clientId, fullTransportMessage: msg },\n );\n } else {\n const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;\n log?.error(`${errMsg}, marking connection as dead`, {\n clientId: this.clientId,\n fullTransportMessage: msg,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);\n session.telemetry.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message order violated',\n });\n session.close();\n }\n\n return;\n }\n\n session.updateBookkeeping(msg.ack, msg.seq);\n\n // don't dispatch explicit acks\n if (!isAck(msg.controlFlags)) {\n this.eventDispatcher.dispatchEvent('message', msg);\n } else {\n log?.debug(`discarding msg (ack bit set)`, {\n clientId: this.clientId,\n fullTransportMessage: msg,\n });\n }\n }\n\n /**\n * Adds a listener to this transport.\n * @param the type of event to listen for\n * @param handler The message handler to add.\n */\n addEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.addEventListener(type, handler);\n }\n\n /**\n * Removes a listener from this transport.\n * @param the type of event to un-listen on\n * @param handler The message handler to remove.\n */\n removeEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.removeEventListener(type, handler);\n }\n\n /**\n * Sends a message over this transport, delegating to the appropriate connection to actually\n * send the message.\n * @param msg The message to send.\n * @returns The ID of the sent message or undefined if it wasn't sent\n */\n send(\n to: TransportClientId,\n msg: PartialTransportMessage,\n ): string | undefined {\n if (this.state === 'destroyed') {\n const err = 'transport is destroyed, cant send';\n log?.error(err, {\n clientId: this.clientId,\n partialTransportMessage: msg,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.UseAfterDestroy, err);\n return undefined;\n } else if (this.state === 'closed') {\n log?.info(`transport closed when sending, discarding`, {\n clientId: this.clientId,\n partialTransportMessage: msg,\n });\n return undefined;\n }\n\n return this.getOrCreateSession(to).session.send(msg);\n }\n\n // control helpers\n sendCloseStream(to: TransportClientId, streamId: string) {\n return this.send(to, {\n streamId: streamId,\n controlFlags: ControlFlags.StreamClosedBit,\n payload: {\n type: 'CLOSE' as const,\n } satisfies Static<typeof ControlMessagePayloadSchema>,\n });\n }\n\n protected protocolError(type: ProtocolErrorType, message: string) {\n this.eventDispatcher.dispatchEvent('protocolError', { type, message });\n }\n\n /**\n * Default close implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.close() at the end.\n * Closes the transport. Any messages sent while the transport is closed will be silently discarded.\n */\n close() {\n this.state = 'closed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n log?.info(`manually closed transport`, { clientId: this.clientId });\n }\n\n /**\n * Default destroy implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.destroy() at the end.\n * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.\n */\n destroy() {\n this.state = 'destroyed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n log?.info(`manually destroyed transport`, { clientId: this.clientId });\n }\n}\n\nexport abstract class ClientTransport<\n ConnType extends Connection,\n HandshakeMetadata extends TSchema = TSchema,\n> extends Transport<ConnType> {\n /**\n * The options for this transport.\n */\n protected options: ClientTransportOptions;\n\n /**\n * The map of reconnect promises for each client ID.\n */\n inflightConnectionPromises: Map<TransportClientId, Promise<ConnType>>;\n retryBudget: LeakyBucketRateLimit;\n\n /**\n * A flag indicating whether the transport should automatically reconnect\n * when a connection is dropped.\n * Realistically, this should always be true for clients unless you are writing\n * tests or a special case where you don't want to reconnect.\n */\n reconnectOnConnectionDrop = true;\n\n /**\n * Optional handshake options for this client.\n */\n handshakeExtensions?: ClientHandshakeOptions<HandshakeMetadata>;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedClientTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultClientTransportOptions,\n ...providedOptions,\n };\n this.inflightConnectionPromises = new Map();\n this.retryBudget = new LeakyBucketRateLimit(this.options);\n }\n\n extendHandshake(options: ClientHandshakeOptions<HandshakeMetadata>) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType, to: TransportClientId): void {\n if (this.state !== 'open') return;\n let session: Session<ConnType> | undefined = undefined;\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n log?.warn(\n `connection to ${to} timed out waiting for handshake, closing`,\n { clientId: this.clientId, connectedTo: to, connId: conn.id },\n );\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const handshakeHandler = (data: Uint8Array) => {\n const maybeSession = this.receiveHandshakeResponseMessage(data, conn);\n if (!maybeSession) {\n conn.close();\n return;\n } else {\n session = maybeSession;\n clearTimeout(handshakeTimeout);\n }\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener((data) => {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message parse failure',\n });\n conn.close();\n return;\n }\n\n this.handleMsg(parsed);\n });\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (session) {\n this.onDisconnect(conn, session);\n }\n log?.info(`connection to ${to} disconnected`, {\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n });\n this.inflightConnectionPromises.delete(to);\n if (this.reconnectOnConnectionDrop) {\n void this.connect(to);\n }\n });\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n log?.warn(`error in connection to ${to}: ${coerceErrorString(err)}`, {\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n });\n });\n }\n\n receiveHandshakeResponseMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Session<ConnType> | false {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake response',\n });\n log?.warn(`received invalid handshake resp`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n fullTransportMessage: parsed,\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake resp',\n );\n return false;\n }\n\n if (!parsed.payload.status.ok) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake rejected',\n });\n log?.warn(`received handshake rejection`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n fullTransportMessage: parsed,\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n parsed.payload.status.reason,\n );\n return false;\n }\n\n log?.debug(`handshake from ${parsed.from} ok`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n fullTransportMessage: parsed,\n });\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.status.sessionId,\n );\n\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n // After a successful connection, we start restoring the budget\n // so that the next time we try to connect, we don't hit the client\n // with backoff forever.\n this.retryBudget.startRestoringBudget(parsed.from);\n return session;\n }\n\n /**\n * Abstract method that creates a new {@link Connection} object.\n * This should call {@link handleConnection} when the connection is created.\n * The downstream client implementation needs to implement this.\n *\n * @param to The client ID of the node to connect to.\n * @returns The new connection object.\n */\n protected abstract createNewOutgoingConnection(\n to: TransportClientId,\n ): Promise<ConnType>;\n\n /**\n * Manually attempts to connect to a client.\n * @param to The client ID of the node to connect to.\n */\n async connect(to: TransportClientId): Promise<void> {\n const canProceedWithConnection = () => this.state === 'open';\n if (!canProceedWithConnection()) {\n log?.info(\n `transport state is no longer open, cancelling attempt to connect to ${to}`,\n { clientId: this.clientId, connectedTo: to },\n );\n return;\n }\n\n let reconnectPromise = this.inflightConnectionPromises.get(to);\n if (!reconnectPromise) {\n // check budget\n const budgetConsumed = this.retryBudget.getBudgetConsumed(to);\n if (!this.retryBudget.hasBudget(to)) {\n const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;\n log?.warn(errMsg, { clientId: this.clientId, connectedTo: to });\n this.protocolError(ProtocolError.RetriesExceeded, errMsg);\n return;\n }\n\n let sleep = Promise.resolve();\n const backoffMs = this.retryBudget.getBackoffMs(to);\n if (backoffMs > 0) {\n sleep = new Promise((resolve) => setTimeout(resolve, backoffMs));\n }\n\n log?.info(`attempting connection to ${to} (${backoffMs}ms backoff)`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n this.retryBudget.consumeBudget(to);\n reconnectPromise = tracer.startActiveSpan('connect', async (span) => {\n try {\n span.addEvent('backoff', { backoffMs });\n await sleep;\n if (!canProceedWithConnection()) {\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('connecting');\n const conn = await this.createNewOutgoingConnection(to);\n if (!canProceedWithConnection()) {\n log?.info(\n `transport state is no longer open, closing pre-handshake connection to ${to}`,\n {\n clientId: this.clientId,\n connectedTo: to,\n connId: conn.id,\n },\n );\n conn.close();\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('sending handshake');\n const ok = await this.sendHandshake(to, conn);\n if (!ok) {\n conn.close();\n throw new Error('failed to send handshake');\n }\n\n return conn;\n } catch (err) {\n // rethrow the error so that the promise is rejected\n // as it was before we wrapped it in a span\n const errStr = coerceErrorString(err);\n span.recordException(errStr);\n span.setStatus({ code: SpanStatusCode.ERROR });\n throw err;\n } finally {\n span.end();\n }\n });\n\n this.inflightConnectionPromises.set(to, reconnectPromise);\n } else {\n log?.info(`attempting connection to ${to} (reusing previous attempt)`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n }\n\n try {\n await reconnectPromise;\n } catch (error: unknown) {\n this.inflightConnectionPromises.delete(to);\n const errStr = coerceErrorString(error);\n\n if (!this.reconnectOnConnectionDrop || !canProceedWithConnection()) {\n log?.warn(`connection to ${to} failed (${errStr})`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n } else {\n log?.warn(`connection to ${to} failed (${errStr}), retrying`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n return this.connect(to);\n }\n }\n }\n\n protected deleteSession(session: Session<ConnType>) {\n this.inflightConnectionPromises.delete(session.to);\n super.deleteSession(session);\n }\n\n protected async sendHandshake(to: TransportClientId, conn: ConnType) {\n let metadata: Static<HandshakeMetadata> | undefined;\n\n if (this.handshakeExtensions) {\n metadata = await this.handshakeExtensions.construct();\n if (!Value.Check(this.handshakeExtensions.schema, metadata)) {\n log?.error(`constructed handshake metadata did not match schema`, {\n clientId: this.clientId,\n connectedTo: to,\n tags: ['invariant-violation'],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'handshake metadata did not match schema',\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake meta mismatch',\n });\n return false;\n }\n }\n\n const { session } = this.getOrCreateSession(to, conn);\n const requestMsg = handshakeRequestMessage(\n this.clientId,\n to,\n session.id,\n metadata,\n getPropagationContext(session.telemetry.ctx),\n );\n log?.debug(`sending handshake request to ${to}`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n conn.send(this.codec.toBuffer(requestMsg));\n return true;\n }\n\n close() {\n this.retryBudget.close();\n super.close();\n }\n}\n\nexport abstract class ServerTransport<\n ConnType extends Connection,\n CustomMetadataSchema extends TSchema = TSchema,\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<CustomMetadataSchema>;\n\n /**\n * A map of session handshake data for each session.\n */\n sessionHandshakeMetadata: WeakMap<Session<ConnType>, ParsedMetadata>;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultServerTransportOptions,\n ...providedOptions,\n };\n this.sessionHandshakeMetadata = new WeakMap();\n log?.info(`initiated server transport`, {\n clientId: this.clientId,\n protocolVersion: PROTOCOL_VERSION,\n });\n }\n\n extendHandshake(options: ServerHandshakeOptions<CustomMetadataSchema>) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType) {\n if (this.state !== 'open') return;\n\n log?.info(`new incoming connection`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n\n let session: Session<ConnType> | undefined = undefined;\n const client = () => session?.to ?? 'unknown';\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n log?.warn(\n `connection to ${client()} timed out waiting for handshake, closing`,\n {\n clientId: this.clientId,\n connectedTo: client(),\n connId: conn.id,\n },\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake timeout',\n });\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const buffer: Array<Uint8Array> = [];\n let receivedHandshakeMessage = false;\n\n const handshakeHandler = (data: Uint8Array) => {\n // if we've already received, just buffer the data\n if (receivedHandshakeMessage) {\n buffer.push(data);\n return;\n }\n\n receivedHandshakeMessage = true;\n clearTimeout(handshakeTimeout);\n\n void this.receiveHandshakeRequestMessage(data, conn).then(\n (maybeSession) => {\n if (!maybeSession) {\n conn.close();\n return;\n }\n\n session = maybeSession;\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n const dataHandler = (data: Uint8Array) => {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.close();\n return;\n }\n\n this.handleMsg(parsed);\n };\n\n // process any data we missed\n for (const data of buffer) {\n dataHandler(data);\n }\n\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener(dataHandler);\n buffer.length = 0;\n },\n );\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (!session) return;\n log?.info(`connection to ${client()} disconnected`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n this.onDisconnect(conn, session);\n });\n\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n if (!session) return;\n log?.warn(\n `connection to ${client()} got an error: ${coerceErrorString(err)}`,\n { clientId: this.clientId, connId: conn.id },\n );\n });\n }\n\n private async validateHandshakeMetadata(\n conn: ConnType,\n session: Session<ConnType> | undefined,\n rawMetadata: Static<\n typeof ControlMessageHandshakeRequestSchema\n >['metadata'],\n from: TransportClientId,\n ): Promise<ParsedMetadata | false> {\n let parsedMetadata: ParsedMetadata = {};\n if (this.handshakeExtensions) {\n // check that the metadata that was sent is the correct shape\n if (!Value.Check(this.handshakeExtensions.schema, rawMetadata)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'malformed handshake meta',\n });\n const reason = 'received malformed handshake metadata';\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n // note: do _not_ log the metadata, it may contain secrets\n log?.warn(`received malformed handshake metadata from ${from}`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n parsedMetadata = await this.handshakeExtensions.validate(\n rawMetadata,\n session,\n );\n\n // handler rejected the connection\n if (parsedMetadata === false) {\n const reason = 'rejected by handshake handler';\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n log?.warn(`rejected handshake from ${from}`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n }\n\n return parsedMetadata;\n }\n\n async receiveHandshakeRequestMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Promise<Session<ConnType> | false> {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake request',\n });\n const reason = 'received invalid handshake msg';\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n // note: do _not_ log the payload, it may contain secrets\n const logData = { ...(parsed.payload ?? {}), metadata: 'redacted' };\n log?.warn(reason, {\n clientId: this.clientId,\n connId: conn.id,\n partialTransportMessage: { ...parsed, payload: logData },\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake request',\n );\n return false;\n }\n\n // double check protocol version here\n const gotVersion = parsed.payload.protocolVersion;\n if (gotVersion !== PROTOCOL_VERSION) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'incorrect protocol version',\n });\n\n const reason = `incorrect version (got: ${gotVersion} wanted ${PROTOCOL_VERSION})`;\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n log?.warn(\n `received handshake msg with incompatible protocol version (got: ${gotVersion}, expected: ${PROTOCOL_VERSION})`,\n { clientId: this.clientId, connId: conn.id },\n );\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n const oldSession = this.sessions.get(parsed.from);\n const parsedMetadata = await this.validateHandshakeMetadata(\n conn,\n oldSession,\n parsed.payload.metadata,\n parsed.from,\n );\n\n if (parsedMetadata === false) {\n return false;\n }\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.sessionId,\n parsed.tracing,\n );\n\n this.sessionHandshakeMetadata.set(session, parsedMetadata);\n\n log?.debug(\n `handshake from ${parsed.from} ok, responding with handshake success`,\n { clientId: this.clientId, connId: conn.id },\n );\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: true,\n sessionId: session.id,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n return session;\n }\n}\n","import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { Connection, Session } from './session';\nimport { PropagationContext } from '../tracing';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Control flags for transport messages.\n * An RPC message is coded with StreamOpenBit | StreamClosedBit.\n * Streams are expected to start with StreamOpenBit sent and the client SHOULD send an empty\n * message with StreamClosedBit to close the stream handler on the server, indicating that\n * it will not be using the stream anymore.\n */\nexport const enum ControlFlags {\n AckBit = 0b0001,\n StreamOpenBit = 0b0010,\n StreamClosedBit = 0b0100,\n}\n\nexport interface ClientHandshakeOptions<MetadataSchema extends TSchema> {\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: () => Static<MetadataSchema> | Promise<Static<MetadataSchema>>;\n}\n\nexport interface ServerHandshakeOptions<MetadataSchema extends TSchema> {\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: (\n metadata: Static<MetadataSchema>,\n session?: Session<Connection>,\n ) => false | ParsedMetadata | Promise<false | ParsedMetadata>;\n}\n\n/**\n * Generic Typebox schema for a transport message.\n * @template T The type of the payload.\n * @param {T} t The payload schema.\n * @returns The transport message schema.\n */\nexport const TransportMessageSchema = <T extends TSchema>(t: T) =>\n Type.Object({\n id: Type.String(),\n from: Type.String(),\n to: Type.String(),\n seq: Type.Integer(),\n ack: Type.Integer(),\n serviceName: Type.Optional(Type.String()),\n procedureName: Type.Optional(Type.String()),\n streamId: Type.String(),\n controlFlags: Type.Integer(),\n tracing: Type.Optional(\n Type.Object({\n traceparent: Type.String(),\n tracestate: Type.String(),\n }),\n ),\n payload: t,\n });\n\n/**\n * Defines the schema for a transport acknowledgement message. This is never constructed manually\n * and is only used internally by the library for tracking inflight messages.\n * @returns The transport message schema.\n */\nexport const ControlMessageAckSchema = Type.Object({\n type: Type.Literal('ACK'),\n});\n\n/**\n * Defines the schema for a transport close message. This is never constructed manually and is only\n * used internally by the library for closing and cleaning up streams.\n */\nexport const ControlMessageCloseSchema = Type.Object({\n type: Type.Literal('CLOSE'),\n});\n\nexport const PROTOCOL_VERSION = 'v1.1';\nexport const ControlMessageHandshakeRequestSchema = Type.Object({\n type: Type.Literal('HANDSHAKE_REQ'),\n protocolVersion: Type.String(),\n sessionId: Type.String(),\n metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const ControlMessageHandshakeResponseSchema = Type.Object({\n type: Type.Literal('HANDSHAKE_RESP'),\n status: Type.Union([\n Type.Object({\n ok: Type.Literal(true),\n sessionId: Type.String(),\n }),\n Type.Object({\n ok: Type.Literal(false),\n reason: Type.String(),\n }),\n ]),\n});\n\nexport const ControlMessagePayloadSchema = Type.Union([\n ControlMessageCloseSchema,\n ControlMessageAckSchema,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n]);\n\n/**\n * Defines the schema for an opaque transport message that is agnostic to any\n * procedure/service.\n * @returns The transport message schema.\n */\nexport const OpaqueTransportMessageSchema = TransportMessageSchema(\n Type.Unknown(),\n);\n\n/**\n * Represents a transport message. This is the same type as {@link TransportMessageSchema} but\n * we can't statically infer generics from generic Typebox schemas so we have to define it again here.\n *\n * TypeScript can't enforce types when a bitmask is involved, so these are the semantics of\n * `controlFlags`:\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `streamId` must be set to a unique value\n * (suggestion: use `nanoid`).\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `serviceName` and `procedureName` must be set.\n * * If `controlFlags & StreamClosedBit == StreamClosedBit` and the kind is `stream` or `subscription`,\n * `payload` should be discarded (usually contains a control message).\n * * If `controlFlags & AckBit == AckBit`, the message is an explicit acknowledgement message and doesn't\n * contain any payload that is relevant to the application so should not be delivered.\n * @template Payload The type of the payload.\n */\nexport interface TransportMessage<Payload = unknown> {\n id: string;\n from: string;\n to: string;\n seq: number;\n ack: number;\n serviceName?: string;\n procedureName?: string;\n streamId: string;\n controlFlags: number;\n tracing?: PropagationContext;\n payload: Payload;\n}\n\nexport type PartialTransportMessage<Payload = unknown> = Omit<\n TransportMessage<Payload>,\n 'id' | 'from' | 'to' | 'seq' | 'ack'\n>;\n\nexport function handshakeRequestMessage(\n from: TransportClientId,\n to: TransportClientId,\n sessionId: string,\n metadata?: unknown,\n tracing?: PropagationContext,\n): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n tracing,\n payload: {\n type: 'HANDSHAKE_REQ',\n protocolVersion: PROTOCOL_VERSION,\n sessionId,\n metadata,\n } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n };\n}\n\nexport function handshakeResponseMessage(\n from: TransportClientId,\n to: TransportClientId,\n status: Static<typeof ControlMessageHandshakeResponseSchema>['status'],\n): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n payload: {\n type: 'HANDSHAKE_RESP',\n status,\n } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n };\n}\n\n/**\n * A type alias for a transport message with an opaque payload.\n * @template T - The type of the opaque payload.\n */\nexport type OpaqueTransportMessage = TransportMessage;\nexport type TransportClientId = string;\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an ack message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the AckBit, false otherwise.\n */\nexport function isAck(controlFlag: number): boolean {\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n return (controlFlag & ControlFlags.AckBit) === ControlFlags.AckBit;\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream open message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamOpenBit, false otherwise.\n */\nexport function isStreamOpen(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamOpenBit) === ControlFlags.StreamOpenBit\n );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream close message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamCloseBit, false otherwise.\n */\nexport function isStreamClose(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamClosedBit) ===\n ControlFlags.StreamClosedBit\n );\n}\n","import { OpaqueTransportMessage } from '../transport';\nimport { PartialTransportMessage } from '../transport/message';\n\nconst LoggingLevels = {\n debug: -1,\n info: 0,\n warn: 1,\n error: 2,\n} as const;\ntype LoggingLevel = keyof typeof LoggingLevels;\n\nexport type LogFn = (\n msg: string,\n ctx?: MessageMetadata,\n level?: LoggingLevel,\n) => void;\nexport type Logger = {\n [key in LoggingLevel]: (msg: string, metadata?: MessageMetadata) => void;\n};\n\nexport type Tags = 'invariant-violation';\n\nexport type MessageMetadata = Record<string, unknown> &\n Partial<{\n protocolVersion: string;\n clientId: string;\n connectedTo: string;\n sessionId: string;\n connId: string;\n fullTransportMessage: OpaqueTransportMessage;\n partialTransportMessage: Partial<PartialTransportMessage>;\n tags: Array<Tags>;\n }>;\n\nclass BaseLogger implements Logger {\n minLevel: LoggingLevel;\n private output: LogFn;\n\n constructor(output: LogFn, minLevel: LoggingLevel = 'info') {\n this.minLevel = minLevel;\n this.output = output;\n }\n\n debug(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.debug) {\n this.output(msg, metadata ?? {}, 'debug');\n }\n }\n\n info(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.info) {\n this.output(msg, metadata ?? {}, 'info');\n }\n }\n\n warn(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.warn) {\n this.output(msg, metadata ?? {}, 'warn');\n }\n }\n\n error(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.error) {\n this.output(msg, metadata ?? {}, 'error');\n }\n }\n}\n\nexport const stringLogger: LogFn = (msg, _ctx, level = 'info') => {\n console.log(`[river:${level}] ${msg}`);\n};\n\nconst colorMap = {\n debug: '\\u001b[34m',\n info: '\\u001b[32m',\n warn: '\\u001b[33m',\n error: '\\u001b[31m',\n};\n\nexport const coloredStringLogger: LogFn = (msg, _ctx, level = 'info') => {\n const color = colorMap[level];\n console.log(`[river:${color}${level}\\u001b[0m] ${msg}`);\n};\n\nexport const jsonLogger: LogFn = (msg, ctx, level) => {\n console.log(JSON.stringify({ msg, ctx, level }));\n};\n\nexport let log: Logger | undefined = undefined;\n\nexport function bindLogger(fn: undefined, level?: LoggingLevel): undefined;\nexport function bindLogger(fn: LogFn | Logger, level?: LoggingLevel): Logger;\nexport function bindLogger(\n fn: LogFn | Logger | undefined,\n level?: LoggingLevel,\n): Logger | undefined {\n if (typeof fn === 'function') {\n log = new BaseLogger(fn, level);\n return log;\n }\n\n log = fn;\n return fn;\n}\n","import { OpaqueTransportMessage } from './message';\nimport { Connection, Session } from './session';\n\ntype ConnectionStatus = 'connect' | 'disconnect';\nexport const ProtocolError = {\n RetriesExceeded: 'conn_retry_exceeded',\n HandshakeFailed: 'handshake_failed',\n UseAfterDestroy: 'use_after_destroy',\n MessageOrderingViolated: 'message_ordering_violated',\n} as const;\n\nexport type ProtocolErrorType =\n (typeof ProtocolError)[keyof typeof ProtocolError];\n\nexport interface EventMap {\n message: OpaqueTransportMessage;\n connectionStatus: {\n status: ConnectionStatus;\n conn: Connection;\n };\n sessionStatus: {\n status: ConnectionStatus;\n session: Session<Connection>;\n };\n protocolError: {\n type: ProtocolErrorType;\n message: string;\n };\n}\n\nexport type EventTypes = keyof EventMap;\nexport type EventHandler<K extends EventTypes> = (\n event: EventMap[K],\n) => unknown;\n\nexport class EventDispatcher<T extends EventTypes> {\n private eventListeners: { [K in T]?: Set<EventHandler<K>> } = {};\n\n numberOfListeners<K extends T>(eventType: K) {\n return this.eventListeners[eventType]?.size ?? 0;\n }\n\n addEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n if (!this.eventListeners[eventType]) {\n this.eventListeners[eventType] = new Set();\n }\n\n this.eventListeners[eventType]?.add(handler);\n }\n\n removeEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n this.eventListeners[eventType]?.delete(handler);\n }\n }\n\n dispatchEvent<K extends T>(eventType: K, event: EventMap[K]) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n for (const handler of handlers) {\n handler(event);\n }\n }\n }\n}\n","import { customAlphabet } from 'nanoid';\nimport {\n ControlFlags,\n ControlMessageAckSchema,\n OpaqueTransportMessage,\n PartialTransportMessage,\n TransportClientId,\n TransportMessage,\n} from './message';\nimport { Codec } from '../codec';\nimport { MessageMetadata, log } from '../logging/log';\nimport { Static } from '@sinclair/typebox';\nimport {\n PropagationContext,\n TelemetryInfo,\n createSessionTelemetryInfo,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nconst nanoid = customAlphabet('1234567890abcdefghijklmnopqrstuvxyz', 6);\nexport const unsafeId = () => nanoid();\n\ntype SequenceNumber = number;\n\n/**\n * A connection is the actual raw underlying transport connection.\n * It’s responsible for dispatching to/from the actual connection itself\n * This should be instantiated as soon as the client/server has a connection\n * It’s tied to the lifecycle of the underlying transport connection (i.e. if the WS drops, this connection should be deleted)\n */\nexport abstract class Connection {\n id: string;\n telemetry?: TelemetryInfo;\n constructor() {\n this.id = `conn-${nanoid(12)}`; // for debugging, no collision safety needed\n }\n\n /**\n * Handle adding a callback for when a message is received.\n * @param msg The message that was received.\n */\n abstract addDataListener(cb: (msg: Uint8Array) => void): void;\n abstract removeDataListener(cb: (msg: Uint8Array) => void): void;\n\n /**\n * Handle adding a callback for when the connection is closed.\n * This should also be called if an error happens.\n * @param cb The callback to call when the connection is closed.\n */\n abstract addCloseListener(cb: () => void): void;\n\n /**\n * Handle adding a callback for when an error is received.\n * This should only be used for logging errors, all cleanup\n * should be delegated to addCloseListener.\n *\n * The implementer should take care such that the implemented\n * connection will call both the close and error callbacks\n * on an error.\n *\n * @param cb The callback to call when an error is received.\n */\n abstract addErrorListener(cb: (err: Error) => void): void;\n\n /**\n * Sends a message over the connection.\n * @param msg The message to send.\n * @returns true if the message was sent, false otherwise.\n */\n abstract send(msg: Uint8Array): boolean;\n\n /**\n * Closes the connection.\n */\n abstract close(): void;\n}\n\nexport interface SessionOptions {\n /**\n * Frequency at which to send heartbeat acknowledgements\n */\n heartbeatIntervalMs: number;\n /**\n * Number of elapsed heartbeats without a response message before we consider\n * the connection dead.\n */\n heartbeatsUntilDead: number;\n /**\n * Duration to wait between connection disconnect and actual session disconnect\n */\n sessionDisconnectGraceMs: number;\n /**\n * The codec to use for encoding/decoding messages over the wire\n */\n codec: Codec;\n}\n\n/**\n * A session is a higher-level abstraction that operates over the span of potentially multiple transport-level connections\n * - It’s responsible for tracking any metadata for a particular client that might need to be persisted across connections (i.e. the sendBuffer, ack, seq)\n * - This will only be considered disconnected if\n * - the server tells the client that we’ve reconnected but it doesn’t recognize us anymore (server definitely died) or\n * - we hit a grace period after a connection disconnect\n */\nexport class Session<ConnType extends Connection> {\n private codec: Codec;\n private options: SessionOptions;\n telemetry: TelemetryInfo;\n\n /**\n * The buffer of messages that have been sent but not yet acknowledged.\n */\n private sendBuffer: Array<OpaqueTransportMessage> = [];\n\n /**\n * The active connection associated with this session\n */\n connection?: ConnType;\n readonly from: TransportClientId;\n readonly to: TransportClientId;\n\n /**\n * The unique ID of this session.\n */\n id: string;\n\n /**\n * What the other side advertised as their session ID\n * for this session.\n */\n advertisedSessionId?: string;\n\n /**\n * Number of messages we've sent along this session (excluding handshake and acks)\n */\n private seq: SequenceNumber = 0;\n\n /**\n * Number of unique messages we've received this session (excluding handshake and acks)\n */\n private ack: SequenceNumber = 0;\n\n /**\n * The grace period between when the inner connection is disconnected\n * and when we should consider the entire session disconnected.\n */\n private disconnectionGrace?: ReturnType<typeof setTimeout>;\n\n /**\n * Number of heartbeats we've sent without a response.\n */\n private heartbeatMisses: number;\n\n /**\n * The interval for sending heartbeats.\n */\n private heartbeat: ReturnType<typeof setInterval>;\n\n constructor(\n conn: ConnType | undefined,\n from: TransportClientId,\n to: TransportClientId,\n options: SessionOptions,\n propagationCtx?: PropagationContext,\n ) {\n this.id = `session-${nanoid(12)}`;\n this.options = options;\n this.from = from;\n this.to = to;\n this.connection = conn;\n this.codec = options.codec;\n\n // setup heartbeat\n this.heartbeatMisses = 0;\n this.heartbeat = setInterval(\n () => this.sendHeartbeat(),\n options.heartbeatIntervalMs,\n );\n this.telemetry = createSessionTelemetryInfo(this, propagationCtx);\n }\n\n get loggingMetadata(): Omit<MessageMetadata, 'parsedMsg'> {\n return {\n clientId: this.from,\n connectedTo: this.to,\n sessionId: this.id,\n connId: this.connection?.id,\n };\n }\n\n /**\n * Sends a message over the session's connection.\n * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.\n *\n * @param msg The partial message to be sent, which will be constructed into a full message.\n * @param addToSendBuff Whether to add the message to the send buffer for retry.\n * @returns The full transport ID of the message that was attempted to be sent.\n */\n send(msg: PartialTransportMessage): string {\n const fullMsg: TransportMessage = this.constructMsg(msg);\n log?.debug(`sending msg`, {\n ...this.loggingMetadata,\n fullTransportMessage: fullMsg,\n });\n\n if (this.connection) {\n const ok = this.connection.send(this.codec.toBuffer(fullMsg));\n if (ok) return fullMsg.id;\n log?.info(\n `failed to send msg to ${fullMsg.to}, connection is probably dead`,\n {\n ...this.loggingMetadata,\n fullTransportMessage: fullMsg,\n },\n );\n } else {\n log?.info(\n `failed to send msg to ${fullMsg.to}, connection not ready yet`,\n { ...this.loggingMetadata, fullTransportMessage: fullMsg },\n );\n }\n\n return fullMsg.id;\n }\n\n sendHeartbeat() {\n const misses = this.heartbeatMisses;\n const missDuration = misses * this.options.heartbeatIntervalMs;\n if (misses > this.options.heartbeatsUntilDead) {\n if (this.connection) {\n log?.info(\n `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n this.loggingMetadata,\n );\n this.telemetry.span.addEvent('closing connection due to inactivity');\n this.closeStaleConnection();\n }\n return;\n }\n\n this.send({\n streamId: 'heartbeat',\n controlFlags: ControlFlags.AckBit,\n payload: {\n type: 'ACK',\n } satisfies Static<typeof ControlMessageAckSchema>,\n });\n this.heartbeatMisses++;\n }\n\n resetBufferedMessages() {\n this.sendBuffer = [];\n this.seq = 0;\n this.ack = 0;\n }\n\n sendBufferedMessages(conn: ConnType) {\n log?.info(`resending ${this.sendBuffer.length} buffered messages`, {\n ...this.loggingMetadata,\n connId: conn.id,\n });\n for (const msg of this.sendBuffer) {\n log?.debug(`resending msg`, {\n ...this.loggingMetadata,\n fullTransportMessage: msg,\n connId: conn.id,\n });\n const ok = conn.send(this.codec.toBuffer(msg));\n if (!ok) {\n // this should never happen unless the transport has an\n // incorrect implementation of `createNewOutgoingConnection`\n const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: errMsg,\n });\n\n log?.error(errMsg, {\n ...this.loggingMetadata,\n fullTransportMessage: msg,\n connId: conn.id,\n tags: ['invariant-violation'],\n });\n conn.close();\n return;\n }\n }\n }\n\n updateBookkeeping(ack: number, seq: number) {\n if (seq + 1 < this.ack) {\n log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {\n ...this.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n this.ack = seq + 1;\n }\n\n closeStaleConnection(conn?: ConnType) {\n if (this.connection === undefined || this.connection === conn) return;\n log?.info(\n `closing old inner connection from session to ${this.to}`,\n this.loggingMetadata,\n );\n this.connection.close();\n this.connection = undefined;\n }\n\n replaceWithNewConnection(newConn: ConnType) {\n this.closeStaleConnection(newConn);\n this.cancelGrace();\n this.connection = newConn;\n this.sendBufferedMessages(newConn);\n }\n\n beginGrace(cb: () => void) {\n log?.info(\n `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,\n this.loggingMetadata,\n );\n this.disconnectionGrace = setTimeout(() => {\n this.close();\n cb();\n }, this.options.sessionDisconnectGraceMs);\n }\n\n // called on reconnect of the underlying session\n cancelGrace() {\n this.heartbeatMisses = 0;\n clearTimeout(this.disconnectionGrace);\n this.disconnectionGrace = undefined;\n }\n\n // closed when we want to discard the whole session\n // (i.e. shutdown or session disconnect)\n close() {\n this.closeStaleConnection();\n this.cancelGrace();\n this.resetBufferedMessages();\n clearInterval(this.heartbeat);\n }\n\n get connected() {\n return this.connection !== undefined;\n }\n\n get nextExpectedSeq() {\n return this.ack;\n }\n\n constructMsg<Payload>(\n partialMsg: PartialTransportMessage<Payload>,\n ): TransportMessage<Payload> {\n const msg = {\n ...partialMsg,\n id: unsafeId(),\n to: this.to,\n from: this.from,\n seq: this.seq,\n ack: this.ack,\n };\n\n this.seq++;\n this.sendBuffer.push(msg);\n return msg;\n }\n\n inspectSendBuffer(): ReadonlyArray<OpaqueTransportMessage> {\n return this.sendBuffer;\n }\n}\n","import {\n Context,\n Span,\n SpanKind,\n context,\n propagation,\n trace,\n} from '@opentelemetry/api';\nimport { version as RIVER_VERSION } from '../package.json';\nimport { ValidProcType } from '../router';\nimport { Connection, OpaqueTransportMessage, Session } from '../transport';\n\nexport interface PropagationContext {\n traceparent: string;\n tracestate: string;\n}\n\nexport interface TelemetryInfo {\n span: Span;\n ctx: Context;\n}\n\nexport function getPropagationContext(\n ctx: Context,\n): PropagationContext | undefined {\n const tracing = {\n traceparent: '',\n tracestate: '',\n };\n propagation.inject(ctx, tracing);\n return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n session: Session<Connection>,\n propagationCtx?: PropagationContext,\n): TelemetryInfo {\n const ctx = propagationCtx\n ? propagation.extract(context.active(), propagationCtx)\n : context.active();\n\n const span = tracer.startSpan(\n `session ${session.id}`,\n {\n attributes: {\n component: 'river',\n 'river.session.id': session.id,\n 'river.session.to': session.to,\n 'river.session.from': session.from,\n },\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n connection: Connection,\n sessionSpan: Span,\n): TelemetryInfo {\n const ctx = trace.setSpan(context.active(), sessionSpan);\n const span = tracer.startSpan(\n `connection ${connection.id}`,\n {\n attributes: {\n component: 'river',\n 'river.connection.id': connection.id,\n },\n links: [{ context: sessionSpan.spanContext() }],\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n kind: ValidProcType,\n serviceName: string,\n procedureName: string,\n streamId: string,\n): TelemetryInfo {\n const ctx = context.active();\n const span = tracer.startSpan(\n `procedure call ${serviceName}.${procedureName}`,\n {\n attributes: {\n component: 'river',\n 'river.method.kind': kind,\n 'river.method.service': serviceName,\n 'river.method.name': procedureName,\n 'river.streamId': streamId,\n 'span.kind': 'client',\n },\n kind: SpanKind.CLIENT,\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createHandlerSpan(\n kind: ValidProcType,\n message: OpaqueTransportMessage,\n fn: (span: Span) => Promise<unknown>,\n) {\n const ctx = message.tracing\n ? propagation.extract(context.active(), message.tracing)\n : context.active();\n\n return tracer.startActiveSpan(\n `procedure handler ${message.serviceName}.${message.procedureName}`,\n {\n attributes: {\n component: 'river',\n 'river.method.kind': kind,\n 'river.method.service': message.serviceName,\n 'river.method.name': message.procedureName,\n 'river.streamId': message.streamId,\n 'span.kind': 'server',\n },\n kind: SpanKind.SERVER,\n },\n ctx,\n fn,\n );\n}\n\nconst tracer = trace.getTracer('river', RIVER_VERSION);\nexport default tracer;\n","{\n \"name\": \"@replit/river\",\n \"description\": \"It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!\",\n \"version\": \"0.23.0\",\n \"type\": \"module\",\n \"exports\": {\n \".\": {\n \"import\": \"./dist/router/index.js\",\n \"require\": \"./dist/router/index.cjs\"\n },\n \"./logging\": {\n \"import\": \"./dist/logging/index.js\",\n \"require\": \"./dist/logging/index.cjs\"\n },\n \"./codec\": {\n \"import\": \"./dist/codec/index.js\",\n \"require\": \"./dist/codec/index.cjs\"\n },\n \"./transport\": {\n \"import\": \"./dist/transport/index.js\",\n \"require\": \"./dist/transport/index.cjs\"\n },\n \"./transport/ws/client\": {\n \"import\": \"./dist/transport/impls/ws/client.js\",\n \"require\": \"./dist/transport/impls/ws/client.cjs\"\n },\n \"./transport/ws/server\": {\n \"import\": \"./dist/transport/impls/ws/server.js\",\n \"require\": \"./dist/transport/impls/ws/server.cjs\"\n },\n \"./transport/uds/client\": {\n \"import\": \"./dist/transport/impls/uds/client.js\",\n \"require\": \"./dist/transport/impls/uds/client.cjs\"\n },\n \"./transport/uds/server\": {\n \"import\": \"./dist/transport/impls/uds/server.js\",\n \"require\": \"./dist/transport/impls/uds/server.cjs\"\n },\n \"./test-util\": {\n \"import\": \"./dist/util/testHelpers.js\",\n \"require\": \"./dist/util/testHelpers.cjs\"\n }\n },\n \"sideEffects\": [\n \"./dist/logging/index.js\"\n ],\n \"files\": [\n \"dist\"\n ],\n \"dependencies\": {\n \"@msgpack/msgpack\": \"^3.0.0-beta2\",\n \"it-pushable\": \"^3.2.3\",\n \"nanoid\": \"^4.0.2\",\n \"ws\": \"^8.17.0\"\n },\n \"peerDependencies\": {\n \"@opentelemetry/api\": \"^1.7.0\",\n \"@sinclair/typebox\": \"~0.32.8\"\n },\n \"devDependencies\": {\n \"@types/ws\": \"^8.5.5\",\n \"@typescript-eslint/eslint-plugin\": \"^7.1.0\",\n \"@typescript-eslint/parser\": \"^7.1.0\",\n \"@vitest/ui\": \"^1.3.1\",\n \"eslint\": \"^8.57.0\",\n \"eslint-config-prettier\": \"^9.1.0\",\n \"eslint-plugin-prettier\": \"^5.1.3\",\n \"prettier\": \"^3.0.0\",\n \"tsup\": \"^7.2.0\",\n \"typescript\": \"^5.2.2\",\n \"vitest\": \"^1.3.1\",\n \"vscode-langservers-extracted\": \"4.8.0\"\n },\n \"scripts\": {\n \"check\": \"tsc --noEmit && npm run format && npm run lint\",\n \"format\": \"npx prettier . --check\",\n \"format:fix\": \"npx prettier . --write\",\n \"lint\": \"eslint .\",\n \"lint:fix\": \"eslint . --fix\",\n \"fix\": \"npm run format:fix && npm run lint:fix\",\n \"build\": \"rm -rf dist && tsup && du -sh dist\",\n \"prepack\": \"npm run build\",\n \"release\": \"npm publish --access public\",\n \"test:ui\": \"echo \\\"remember to go to /__vitest__ in the webview\\\" && vitest --ui --api.host 0.0.0.0 --api.port 3000\",\n \"test\": \"vitest --test-timeout=500\",\n \"test:single\": \"vitest run --test-timeout=500 --reporter=dot\",\n \"test:flake\": \"./flake.sh\",\n \"bench\": \"vitest bench\"\n },\n \"engines\": {\n \"node\": \">=16\"\n },\n \"keywords\": [\n \"rpc\",\n \"websockets\",\n \"jsonschema\"\n ],\n \"author\": \"Jacky Zhao\",\n \"license\": \"MIT\"\n}\n","export function coerceErrorString(err: unknown): string {\n if (err instanceof Error) {\n return err.message || 'unknown reason';\n }\n\n return `[coerced to error] ${String(err)}`;\n}\n","import { Codec } from './types';\n\nconst encoder = new TextEncoder();\nconst decoder = new TextDecoder();\n\n// Convert Uint8Array to base64\nfunction uint8ArrayToBase64(uint8Array: Uint8Array) {\n let binary = '';\n uint8Array.forEach((byte) => {\n binary += String.fromCharCode(byte);\n });\n return btoa(binary);\n}\n\n// Convert base64 to Uint8Array\nfunction base64ToUint8Array(base64: string) {\n const binaryString = atob(base64);\n const uint8Array = new Uint8Array(binaryString.length);\n for (let i = 0; i < binaryString.length; i++) {\n uint8Array[i] = binaryString.charCodeAt(i);\n }\n return uint8Array;\n}\n\ninterface Base64EncodedValue {\n $t: string;\n}\n\n/**\n * Naive JSON codec implementation using JSON.stringify and JSON.parse.\n * @type {Codec}\n */\nexport const NaiveJsonCodec: Codec = {\n toBuffer: (obj: object) => {\n return encoder.encode(\n JSON.stringify(obj, function replacer<\n T extends object,\n >(this: T, key: keyof T) {\n const val = this[key];\n if (val instanceof Uint8Array) {\n return { $t: uint8ArrayToBase64(val) } satisfies Base64EncodedValue;\n } else {\n return val;\n }\n }),\n );\n },\n fromBuffer: (buff: Uint8Array) => {\n try {\n const parsed = JSON.parse(\n decoder.decode(buff),\n function reviver(_key, val: unknown) {\n if ((val as Base64EncodedValue | undefined)?.$t) {\n return base64ToUint8Array((val as Base64EncodedValue).$t);\n } else {\n return val;\n }\n },\n ) as unknown;\n\n if (typeof parsed === 'object') return parsed;\n return null;\n } catch {\n return null;\n }\n },\n};\n","import { Connection } from '../../session';\nimport { WsLike } from './wslike';\n\nexport class WebSocketConnection extends Connection {\n errorCb: null | ((err: Error) => void) = null;\n closeCb: null | (() => void) = null;\n\n ws: WsLike;\n\n constructor(ws: WsLike) {\n super();\n this.ws = ws;\n this.ws.binaryType = 'arraybuffer';\n\n // Websockets are kinda shitty, they emit error events with no\n // information other than it errored, so we have to do some extra\n // work to figure out what happened.\n let didError = false;\n this.ws.onerror = () => {\n didError = true;\n };\n this.ws.onclose = ({ code, reason }) => {\n if (didError && this.errorCb) {\n this.errorCb(\n new Error(\n `websocket closed with code and reason: ${code} - ${reason}`,\n ),\n );\n\n return;\n }\n\n if (this.closeCb) {\n this.closeCb();\n }\n };\n }\n\n addDataListener(cb: (msg: Uint8Array) => void) {\n this.ws.onmessage = (msg) => cb(msg.data as Uint8Array);\n }\n\n removeDataListener(): void {\n this.ws.onmessage = null;\n }\n\n addCloseListener(cb: () => void): void {\n this.closeCb = cb;\n }\n\n addErrorListener(cb: (err: Error) => void): void {\n this.errorCb = cb;\n }\n\n send(payload: Uint8Array) {\n if (this.ws.readyState === this.ws.OPEN) {\n this.ws.send(payload);\n return true;\n } else {\n return false;\n }\n }\n\n close() {\n this.ws.close();\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,mBAAsB;;;ACDtB,qBAAsC;AACtC,oBAAuB;AA6DhB,IAAM,yBAAyB,CAAoB,MACxD,oBAAK,OAAO;AAAA,EACV,IAAI,oBAAK,OAAO;AAAA,EAChB,MAAM,oBAAK,OAAO;AAAA,EAClB,IAAI,oBAAK,OAAO;AAAA,EAChB,KAAK,oBAAK,QAAQ;AAAA,EAClB,KAAK,oBAAK,QAAQ;AAAA,EAClB,aAAa,oBAAK,SAAS,oBAAK,OAAO,CAAC;AAAA,EACxC,eAAe,oBAAK,SAAS,oBAAK,OAAO,CAAC;AAAA,EAC1C,UAAU,oBAAK,OAAO;AAAA,EACtB,cAAc,oBAAK,QAAQ;AAAA,EAC3B,SAAS,oBAAK;AAAA,IACZ,oBAAK,OAAO;AAAA,MACV,aAAa,oBAAK,OAAO;AAAA,MACzB,YAAY,oBAAK,OAAO;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EACA,SAAS;AACX,CAAC;AAOI,IAAM,0BAA0B,oBAAK,OAAO;AAAA,EACjD,MAAM,oBAAK,QAAQ,KAAK;AAC1B,CAAC;AAMM,IAAM,4BAA4B,oBAAK,OAAO;AAAA,EACnD,MAAM,oBAAK,QAAQ,OAAO;AAC5B,CAAC;AAEM,IAAM,mBAAmB;AACzB,IAAM,uCAAuC,oBAAK,OAAO;AAAA,EAC9D,MAAM,oBAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,oBAAK,OAAO;AAAA,EAC7B,WAAW,oBAAK,OAAO;AAAA,EACvB,UAAU,oBAAK,SAAS,oBAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,wCAAwC,oBAAK,OAAO;AAAA,EAC/D,MAAM,oBAAK,QAAQ,gBAAgB;AAAA,EACnC,QAAQ,oBAAK,MAAM;AAAA,IACjB,oBAAK,OAAO;AAAA,MACV,IAAI,oBAAK,QAAQ,IAAI;AAAA,MACrB,WAAW,oBAAK,OAAO;AAAA,IACzB,CAAC;AAAA,IACD,oBAAK,OAAO;AAAA,MACV,IAAI,oBAAK,QAAQ,KAAK;AAAA,MACtB,QAAQ,oBAAK,OAAO;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AACH,CAAC;AAEM,IAAM,8BAA8B,oBAAK,MAAM;AAAA,EACpD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,+BAA+B;AAAA,EAC1C,oBAAK,QAAQ;AACf;AA6DO,SAAS,yBACd,MACA,IACA,QACwE;AACxE,SAAO;AAAA,IACL,QAAI,sBAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,cAAU,sBAAO;AAAA,IACjB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAcO,SAAS,MAAM,aAA8B;AAElD,UAAQ,cAAc,oBAAyB;AACjD;;;AC/IO,IAAI,MAA0B;;;ACpF9B,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AA0BO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,kBAA+B,WAAc;AAC3C,WAAO,KAAK,eAAe,SAAS,GAAG,QAAQ;AAAA,EACjD;AAAA,EAEA,iBAA8B,WAAc,SAA0B;AACpE,QAAI,CAAC,KAAK,eAAe,SAAS,GAAG;AACnC,WAAK,eAAe,SAAS,IAAI,oBAAI,IAAI;AAAA,IAC3C;AAEA,SAAK,eAAe,SAAS,GAAG,IAAI,OAAO;AAAA,EAC7C;AAAA,EAEA,oBAAiC,WAAc,SAA0B;AACvE,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,WAAK,eAAe,SAAS,GAAG,OAAO,OAAO;AAAA,IAChD;AAAA,EACF;AAAA,EAEA,cAA2B,WAAc,OAAoB;AAC3D,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,iBAAW,WAAW,UAAU;AAC9B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;ACjEA,IAAAA,iBAA+B;;;ACA/B,iBAOO;;;ACJL,cAAW;;;AD8BN,SAAS,2BACd,SACA,gBACe;AACf,QAAM,MAAM,iBACR,uBAAY,QAAQ,mBAAQ,OAAO,GAAG,cAAc,IACpD,mBAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,QAAQ,EAAE;AAAA,IACrB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB,QAAQ;AAAA,QAC5B,oBAAoB,QAAQ;AAAA,QAC5B,sBAAsB,QAAQ;AAAA,MAChC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,8BACd,YACA,aACe;AACf,QAAM,MAAM,iBAAM,QAAQ,mBAAQ,OAAO,GAAG,WAAW;AACvD,QAAM,OAAO,OAAO;AAAA,IAClB,cAAc,WAAW,EAAE;AAAA,IAC3B;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,uBAAuB,WAAW;AAAA,MACpC;AAAA,MACA,OAAO,CAAC,EAAE,SAAS,YAAY,YAAY,EAAE,CAAC;AAAA,IAChD;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAuDA,IAAM,SAAS,iBAAM,UAAU,SAAS,OAAa;;;ADjHrD,IAAAC,cAA+B;AAE/B,IAAMC,cAAS,+BAAe,uCAAuC,CAAC;AAC/D,IAAM,WAAW,MAAMA,QAAO;AAU9B,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQA,QAAO,EAAE,CAAC;AAAA,EAC9B;AAwCF;AA6BO,IAAM,UAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKQ,aAA4C,CAAC;AAAA;AAAA;AAAA;AAAA,EAKrD;AAAA,EACS;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA;AAAA;AAAA;AAAA;AAAA,EAKQ,MAAsB;AAAA;AAAA;AAAA;AAAA,EAKtB,MAAsB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMtB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAER,YACE,MACA,MACA,IACA,SACA,gBACA;AACA,SAAK,KAAK,WAAWA,QAAO,EAAE,CAAC;AAC/B,SAAK,UAAU;AACf,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,aAAa;AAClB,SAAK,QAAQ,QAAQ;AAGrB,SAAK,kBAAkB;AACvB,SAAK,YAAY;AAAA,MACf,MAAM,KAAK,cAAc;AAAA,MACzB,QAAQ;AAAA,IACV;AACA,SAAK,YAAY,2BAA2B,MAAM,cAAc;AAAA,EAClE;AAAA,EAEA,IAAI,kBAAsD;AACxD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,QAAQ,KAAK,YAAY;AAAA,IAC3B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,KAAK,KAAsC;AACzC,UAAM,UAA4B,KAAK,aAAa,GAAG;AACvD,SAAK,MAAM,eAAe;AAAA,MACxB,GAAG,KAAK;AAAA,MACR,sBAAsB;AAAA,IACxB,CAAC;AAED,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,KAAK,WAAW,KAAK,KAAK,MAAM,SAAS,OAAO,CAAC;AAC5D,UAAI;AAAI,eAAO,QAAQ;AACvB,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC;AAAA,UACE,GAAG,KAAK;AAAA,UACR,sBAAsB;AAAA,QACxB;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC,EAAE,GAAG,KAAK,iBAAiB,sBAAsB,QAAQ;AAAA,MAC3D;AAAA,IACF;AAEA,WAAO,QAAQ;AAAA,EACjB;AAAA,EAEA,gBAAgB;AACd,UAAM,SAAS,KAAK;AACpB,UAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,QAAI,SAAS,KAAK,QAAQ,qBAAqB;AAC7C,UAAI,KAAK,YAAY;AACnB,aAAK;AAAA,UACH,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,qBAAqB;AAAA,MAC5B;AACA;AAAA,IACF;AAEA,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AACD,SAAK;AAAA,EACP;AAAA,EAEA,wBAAwB;AACtB,SAAK,aAAa,CAAC;AACnB,SAAK,MAAM;AACX,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,qBAAqB,MAAgB;AACnC,SAAK,KAAK,aAAa,KAAK,WAAW,MAAM,sBAAsB;AAAA,MACjE,GAAG,KAAK;AAAA,MACR,QAAQ,KAAK;AAAA,IACf,CAAC;AACD,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,MAAM,iBAAiB;AAAA,QAC1B,GAAG,KAAK;AAAA,QACR,sBAAsB;AAAA,QACtB,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,YAAM,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG,CAAC;AAC7C,UAAI,CAAC,IAAI;AAGP,cAAM,SAAS,sCAAsC,KAAK,EAAE;AAC5D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,MAAM,QAAQ;AAAA,UACjB,GAAG,KAAK;AAAA,UACR,sBAAsB;AAAA,UACtB,QAAQ,KAAK;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM;AACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,QAAI,MAAM,IAAI,KAAK,KAAK;AACtB,WAAK,MAAM,sBAAsB,GAAG,UAAU,KAAK,GAAG,IAAI;AAAA,QACxD,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAEA,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AAAA,EACnB;AAAA,EAEA,qBAAqB,MAAiB;AACpC,QAAI,KAAK,eAAe,UAAa,KAAK,eAAe;AAAM;AAC/D,SAAK;AAAA,MACH,gDAAgD,KAAK,EAAE;AAAA,MACvD,KAAK;AAAA,IACP;AACA,SAAK,WAAW,MAAM;AACtB,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,yBAAyB,SAAmB;AAC1C,SAAK,qBAAqB,OAAO;AACjC,SAAK,YAAY;AACjB,SAAK,aAAa;AAClB,SAAK,qBAAqB,OAAO;AAAA,EACnC;AAAA,EAEA,WAAW,IAAgB;AACzB,SAAK;AAAA,MACH,YAAY,KAAK,QAAQ,wBAAwB,oCAAoC,KAAK,EAAE;AAAA,MAC5F,KAAK;AAAA,IACP;AACA,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,MAAM;AACX,SAAG;AAAA,IACL,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA;AAAA,EAGA,cAAc;AACZ,SAAK,kBAAkB;AACvB,iBAAa,KAAK,kBAAkB;AACpC,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA,EAIA,QAAQ;AACN,SAAK,qBAAqB;AAC1B,SAAK,YAAY;AACjB,SAAK,sBAAsB;AAC3B,kBAAc,KAAK,SAAS;AAAA,EAC9B;AAAA,EAEA,IAAI,YAAY;AACd,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,SAAS;AAAA,MACb,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,SAAK,WAAW,KAAK,GAAG;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,oBAA2D;AACzD,WAAO,KAAK;AAAA,EACd;AACF;;;AGtXO,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACJA,IAAM,UAAU,IAAI,YAAY;AAChC,IAAM,UAAU,IAAI,YAAY;AAGhC,SAAS,mBAAmB,YAAwB;AAClD,MAAI,SAAS;AACb,aAAW,QAAQ,CAAC,SAAS;AAC3B,cAAU,OAAO,aAAa,IAAI;AAAA,EACpC,CAAC;AACD,SAAO,KAAK,MAAM;AACpB;AAGA,SAAS,mBAAmB,QAAgB;AAC1C,QAAM,eAAe,KAAK,MAAM;AAChC,QAAM,aAAa,IAAI,WAAW,aAAa,MAAM;AACrD,WAAS,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;AAC5C,eAAW,CAAC,IAAI,aAAa,WAAW,CAAC;AAAA,EAC3C;AACA,SAAO;AACT;AAUO,IAAM,iBAAwB;AAAA,EACnC,UAAU,CAAC,QAAgB;AACzB,WAAO,QAAQ;AAAA,MACb,KAAK,UAAU,KAAK,SAAS,SAElB,KAAc;AACvB,cAAM,MAAM,KAAK,GAAG;AACpB,YAAI,eAAe,YAAY;AAC7B,iBAAO,EAAE,IAAI,mBAAmB,GAAG,EAAE;AAAA,QACvC,OAAO;AACL,iBAAO;AAAA,QACT;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EACA,YAAY,CAAC,SAAqB;AAChC,QAAI;AACF,YAAM,SAAS,KAAK;AAAA,QAClB,QAAQ,OAAO,IAAI;AAAA,QACnB,SAAS,QAAQ,MAAM,KAAc;AACnC,cAAK,KAAwC,IAAI;AAC/C,mBAAO,mBAAoB,IAA2B,EAAE;AAAA,UAC1D,OAAO;AACL,mBAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF;AAEA,UAAI,OAAO,WAAW;AAAU,eAAO;AACvC,aAAO;AAAA,IACT,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;;;AR9BA,IAAAC,cAA+B;AAexB,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AAAA,EACH,GAAG;AACL;AAMA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AACL;AAgCO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA;AAAA,EAK3D;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,cAAc;AAChB,WAAO,IAAI;AAAA,MACT,CAAC,GAAG,KAAK,QAAQ,EACd,IAAI,CAAC,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,QAAQ,UAAU,CAAC,EACvD,OAAO,CAAC,UAAuC,MAAM,CAAC,MAAM,MAAS;AAAA,IAC1E;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQV,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW,oBAAI,IAAI;AACxB,SAAK,QAAQ,KAAK,QAAQ;AAC1B,SAAK,WAAW;AAChB,SAAK,QAAQ;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAoBU,UACR,MACA,aACA,SACA,aACA;AACA,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,YAAY;AAAA,MACf;AAAA,MACA,QAAQ,UAAU;AAAA,IACpB;AAEA,QAAI,aAAa;AACf,cAAQ,yBAAyB,IAAI;AACrC,WAAK,KAAK,kBAAkB,WAAW,IAAI,QAAQ,eAAe;AAAA,IACpE;AAAA,EACF;AAAA,EAEU,cACR,IACA,MACA,gBACA;AACA,UAAM,UAAU,IAAI;AAAA,MAClB;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACF;AACA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AACrC,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEU,mBACR,IACA,MACA,WACA,gBACA;AACA,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,QAAI,cAAc,YAAY;AAE9B,QACE,SAAS,wBAAwB,UACjC,cAAc,UACd,QAAQ,wBAAwB,WAChC;AACA,WAAK;AAAA,QACH,eAAe,EAAE,6DAA6D,QAAQ,mBAAmB,UAAU,SAAS;AAAA,QAC5H,QAAQ;AAAA,MACV;AACA,WAAK,cAAc,OAAO;AAC1B,oBAAc;AACd,gBAAU;AAAA,IACZ;AAEA,QAAI,CAAC,SAAS;AACZ,gBAAU,KAAK,cAAc,IAAI,MAAM,cAAc;AACrD,WAAK;AAAA,QACH,kBAAkB,EAAE;AAAA,QACpB,QAAQ;AAAA,MACV;AAAA,IACF;AAEA,QAAI,cAAc,QAAW;AAC3B,cAAQ,sBAAsB;AAAA,IAChC;AAEA,WAAO,EAAE,SAAS,YAAY;AAAA,EAChC;AAAA,EAEU,cAAc,SAA4B;AAClD,YAAQ,MAAM;AACd,YAAQ,UAAU,KAAK,IAAI;AAC3B,SAAK,SAAS,OAAO,QAAQ,EAAE;AAC/B,SAAK;AAAA,MACH,WAAW,QAAQ,EAAE,oBAAoB,QAAQ,EAAE;AAAA,MACnD,QAAQ;AAAA,IACV;AACA,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,aAAa,MAAgB,SAA4B;AACjE,SAAK,WAAW,KAAK,IAAI;AACzB,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,aAAa;AACrB,YAAQ,WAAW,MAAM;AACvB,cAAQ,UAAU,KAAK,SAAS,8BAA8B;AAC9D,WAAK,cAAc,OAAO;AAAA,IAC5B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,SAAS,KAAgD;AACjE,UAAM,YAAY,KAAK,MAAM,WAAW,GAAG;AAE3C,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,MAAM,yCAAyC,aAAa,IAAI;AAAA,QACnE,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,mBAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QAC/D,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU;AAAQ;AAC3B,UAAM,UAAU,KAAK,SAAS,IAAI,IAAI,IAAI;AAC1C,QAAI,CAAC,SAAS;AACZ,WAAK,MAAM,2BAA2B,IAAI,IAAI,IAAI;AAAA,QAChD,UAAU,KAAK;AAAA,QACf,sBAAsB;AAAA,QACtB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAGA,YAAQ,YAAY;AAEpB,SAAK,MAAM,gBAAgB;AAAA,MACzB,UAAU,KAAK;AAAA,MACf,sBAAsB;AAAA,IACxB,CAAC;AACD,QAAI,IAAI,QAAQ,QAAQ,iBAAiB;AACvC,UAAI,IAAI,MAAM,QAAQ,iBAAiB;AACrC,aAAK;AAAA,UACH,oCAAoC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AAAA,UACnF,EAAE,UAAU,KAAK,UAAU,sBAAsB,IAAI;AAAA,QACvD;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AACrG,aAAK,MAAM,GAAG,MAAM,gCAAgC;AAAA,UAClD,UAAU,KAAK;AAAA,UACf,sBAAsB;AAAA,UACtB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,gBAAQ,UAAU,KAAK,UAAU;AAAA,UAC/B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,gBAAQ,MAAM;AAAA,MAChB;AAEA;AAAA,IACF;AAEA,YAAQ,kBAAkB,IAAI,KAAK,IAAI,GAAG;AAG1C,QAAI,CAAC,MAAM,IAAI,YAAY,GAAG;AAC5B,WAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,IACnD,OAAO;AACL,WAAK,MAAM,gCAAgC;AAAA,QACzC,UAAU,KAAK;AAAA,QACf,sBAAsB;AAAA,MACxB,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBACE,MACA,SACM;AACN,SAAK,gBAAgB,iBAAiB,MAAM,OAAO;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,oBACE,MACA,SACM;AACN,SAAK,gBAAgB,oBAAoB,MAAM,OAAO;AAAA,EACxD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,KACE,IACA,KACoB;AACpB,QAAI,KAAK,UAAU,aAAa;AAC9B,YAAM,MAAM;AACZ,WAAK,MAAM,KAAK;AAAA,QACd,UAAU,KAAK;AAAA,QACf,yBAAyB;AAAA,QACzB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD,WAAK,cAAc,cAAc,iBAAiB,GAAG;AACrD,aAAO;AAAA,IACT,WAAW,KAAK,UAAU,UAAU;AAClC,WAAK,KAAK,6CAA6C;AAAA,QACrD,UAAU,KAAK;AAAA,QACf,yBAAyB;AAAA,MAC3B,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO,KAAK,mBAAmB,EAAE,EAAE,QAAQ,KAAK,GAAG;AAAA,EACrD;AAAA;AAAA,EAGA,gBAAgB,IAAuB,UAAkB;AACvD,WAAO,KAAK,KAAK,IAAI;AAAA,MACnB;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEU,cAAc,MAAyB,SAAiB;AAChE,SAAK,gBAAgB,cAAc,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAAA,EACvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAU;AACR,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,gCAAgC,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACvE;AACF;AAwWO,IAAe,kBAAf,cAGG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,2BAA2B,oBAAI,QAAQ;AAC5C,SAAK,KAAK,8BAA8B;AAAA,MACtC,UAAU,KAAK;AAAA,MACf,iBAAiB;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,SAAuD;AACrE,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU;AAAQ;AAE3B,SAAK,KAAK,2BAA2B;AAAA,MACnC,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,QAAI,UAAyC;AAC7C,UAAM,SAAS,MAAM,SAAS,MAAM;AAGpC,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK;AAAA,UACH,iBAAiB,OAAO,CAAC;AAAA,UACzB;AAAA,YACE,UAAU,KAAK;AAAA,YACf,aAAa,OAAO;AAAA,YACpB,QAAQ,KAAK;AAAA,UACf;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,SAA4B,CAAC;AACnC,QAAI,2BAA2B;AAE/B,UAAM,mBAAmB,CAAC,SAAqB;AAE7C,UAAI,0BAA0B;AAC5B,eAAO,KAAK,IAAI;AAChB;AAAA,MACF;AAEA,iCAA2B;AAC3B,mBAAa,gBAAgB;AAE7B,WAAK,KAAK,+BAA+B,MAAM,IAAI,EAAE;AAAA,QACnD,CAAC,iBAAiB;AAChB,cAAI,CAAC,cAAc;AACjB,iBAAK,MAAM;AACX;AAAA,UACF;AAEA,oBAAU;AAIV,gBAAM,cAAc,CAACC,UAAqB;AACxC,kBAAM,SAAS,KAAK,SAASA,KAAI;AACjC,gBAAI,CAAC,QAAQ;AACX,mBAAK,MAAM;AACX;AAAA,YACF;AAEA,iBAAK,UAAU,MAAM;AAAA,UACvB;AAGA,qBAAWA,SAAQ,QAAQ;AACzB,wBAAYA,KAAI;AAAA,UAClB;AAEA,eAAK,mBAAmB,gBAAgB;AACxC,eAAK,gBAAgB,WAAW;AAChC,iBAAO,SAAS;AAAA,QAClB;AAAA,MACF;AAAA,IACF;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,CAAC;AAAS;AACd,WAAK,KAAK,iBAAiB,OAAO,CAAC,iBAAiB;AAAA,QAClD,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,WAAK,aAAa,MAAM,OAAO;AAAA,IACjC,CAAC;AAED,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,UAAI,CAAC;AAAS;AACd,WAAK;AAAA,QACH,iBAAiB,OAAO,CAAC,kBAAkB,kBAAkB,GAAG,CAAC;AAAA,QACjE,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,0BACZ,MACA,SACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,mBAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,SAAS;AACf,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAE1C,aAAK,KAAK,8CAA8C,IAAI,IAAI;AAAA,UAC9D,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,QACf,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAEA,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,cAAM,SAAS;AACf,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,aAAK,KAAK,2BAA2B,IAAI,IAAI;AAAA,UAC3C,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,QACf,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,+BACJ,MACA,MACoC;AACpC,UAAM,SAAS,KAAK,SAAS,IAAI;AACjC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,mBAAM,MAAM,sCAAsC,OAAO,OAAO,GAAG;AACtE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,YAAM,SAAS;AACf,YAAMC,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAE1C,YAAM,UAAU,EAAE,GAAI,OAAO,WAAW,CAAC,GAAI,UAAU,WAAW;AAClE,WAAK,KAAK,QAAQ;AAAA,QAChB,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA,QACb,yBAAyB,EAAE,GAAG,QAAQ,SAAS,QAAQ;AAAA,MACzD,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAGA,UAAM,aAAa,OAAO,QAAQ;AAClC,QAAI,eAAe,kBAAkB;AACnC,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AAED,YAAM,SAAS,2BAA2B,UAAU,WAAW,gBAAgB;AAC/E,YAAMA,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK;AAAA,QACH,mEAAmE,UAAU,eAAe,gBAAgB;AAAA,QAC5G,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AACA,WAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,aAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,SAAS,IAAI,OAAO,IAAI;AAChD,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,QAAI,mBAAmB,OAAO;AAC5B,aAAO;AAAA,IACT;AAEA,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,SAAK,yBAAyB,IAAI,SAAS,cAAc;AAEzD,SAAK;AAAA,MACH,kBAAkB,OAAO,IAAI;AAAA,MAC7B,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,IAC7C;AACA,UAAM,cAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,MACvE,IAAI;AAAA,MACJ,WAAW,QAAQ;AAAA,IACrB,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAEtD,WAAO;AAAA,EACT;AACF;;;ASrnCO,IAAM,sBAAN,cAAkC,WAAW;AAAA,EAClD,UAAyC;AAAA,EACzC,UAA+B;AAAA,EAE/B;AAAA,EAEA,YAAY,IAAY;AACtB,UAAM;AACN,SAAK,KAAK;AACV,SAAK,GAAG,aAAa;AAKrB,QAAI,WAAW;AACf,SAAK,GAAG,UAAU,MAAM;AACtB,iBAAW;AAAA,IACb;AACA,SAAK,GAAG,UAAU,CAAC,EAAE,MAAM,OAAO,MAAM;AACtC,UAAI,YAAY,KAAK,SAAS;AAC5B,aAAK;AAAA,UACH,IAAI;AAAA,YACF,0CAA0C,IAAI,MAAM,MAAM;AAAA,UAC5D;AAAA,QACF;AAEA;AAAA,MACF;AAEA,UAAI,KAAK,SAAS;AAChB,aAAK,QAAQ;AAAA,MACf;AAAA,IACF;AAAA,EACF;AAAA,EAEA,gBAAgB,IAA+B;AAC7C,SAAK,GAAG,YAAY,CAAC,QAAQ,GAAG,IAAI,IAAkB;AAAA,EACxD;AAAA,EAEA,qBAA2B;AACzB,SAAK,GAAG,YAAY;AAAA,EACtB;AAAA,EAEA,iBAAiB,IAAsB;AACrC,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,iBAAiB,IAAgC;AAC/C,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,KAAK,SAAqB;AACxB,QAAI,KAAK,GAAG,eAAe,KAAK,GAAG,MAAM;AACvC,WAAK,GAAG,KAAK,OAAO;AACpB,aAAO;AAAA,IACT,OAAO;AACL,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,SAAK,GAAG,MAAM;AAAA,EAChB;AACF;;;AVzDO,IAAM,2BAAN,cAAuC,gBAAqC;AAAA,EACjF;AAAA,EAEA,YACE,KACA,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,MAAM;AACX,SAAK,IAAI,GAAG,cAAc,KAAK,iBAAiB;AAAA,EAClD;AAAA,EAEA,oBAAoB,CAAC,OAAe;AAClC,UAAM,OAAO,IAAI,oBAAoB,EAAE;AACvC,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,QAAQ;AACN,UAAM,MAAM;AACZ,SAAK,IAAI,IAAI,cAAc,KAAK,iBAAiB;AAAA,EACnD;AACF;","names":["import_nanoid","import_api","nanoid","import_api","data","responseMsg"]}
1
+ {"version":3,"sources":["../../../../transport/impls/ws/server.ts","../../../../transport/transport.ts","../../../../transport/message.ts","../../../../logging/log.ts","../../../../transport/events.ts","../../../../transport/session.ts","../../../../tracing/index.ts","../../../../package.json","../../../../util/stringify.ts","../../../../codec/json.ts","../../../../transport/impls/ws/connection.ts"],"sourcesContent":["import { TransportClientId } from '../../message';\nimport {\n ServerTransport,\n ProvidedServerTransportOptions,\n} from '../../transport';\nimport { WebSocketServer } from 'ws';\nimport { WebSocketConnection } from './connection';\nimport { WsLike } from './wslike';\n\nexport class WebSocketServerTransport extends ServerTransport<WebSocketConnection> {\n wss: WebSocketServer;\n\n constructor(\n wss: WebSocketServer,\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.wss = wss;\n this.wss.on('connection', this.connectionHandler);\n }\n\n connectionHandler = (ws: WsLike) => {\n const conn = new WebSocketConnection(ws);\n this.handleConnection(conn);\n };\n\n close() {\n super.close();\n this.wss.off('connection', this.connectionHandler);\n }\n}\n","import { Codec } from '../codec/types';\nimport { Value } from '@sinclair/typebox/value';\nimport {\n OpaqueTransportMessage,\n OpaqueTransportMessageSchema,\n TransportClientId,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n handshakeRequestMessage,\n handshakeResponseMessage,\n PartialTransportMessage,\n ControlFlags,\n ControlMessagePayloadSchema,\n isAck,\n PROTOCOL_VERSION,\n ClientHandshakeOptions,\n ServerHandshakeOptions,\n} from './message';\nimport { log } from '../logging/log';\nimport {\n EventDispatcher,\n EventHandler,\n EventTypes,\n ProtocolError,\n ProtocolErrorType,\n} from './events';\nimport { Connection, Session, SessionOptions } from './session';\nimport { Static, TSchema } from '@sinclair/typebox';\nimport { coerceErrorString } from '../util/stringify';\nimport { ConnectionRetryOptions, LeakyBucketRateLimit } from './rateLimit';\nimport { NaiveJsonCodec } from '../codec';\nimport tracer, {\n PropagationContext,\n createConnectionTelemetryInfo,\n getPropagationContext,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Represents the possible states of a transport.\n * @property {'open'} open - The transport is open and operational (note that this doesn't mean it is actively connected)\n * @property {'closed'} closed - The transport is closed and not operational, but can be reopened.\n * @property {'destroyed'} destroyed - The transport is permanently destroyed and cannot be reopened.\n */\nexport type TransportStatus = 'open' | 'closed' | 'destroyed';\n\ntype TransportOptions = SessionOptions;\n\nexport type ProvidedTransportOptions = Partial<TransportOptions>;\n\nexport const defaultTransportOptions: TransportOptions = {\n heartbeatIntervalMs: 1_000,\n heartbeatsUntilDead: 2,\n sessionDisconnectGraceMs: 5_000,\n codec: NaiveJsonCodec,\n};\n\ntype ClientTransportOptions = TransportOptions & ConnectionRetryOptions;\n\nexport type ProvidedClientTransportOptions = Partial<ClientTransportOptions>;\n\nconst defaultConnectionRetryOptions: ConnectionRetryOptions = {\n baseIntervalMs: 250,\n maxJitterMs: 200,\n maxBackoffMs: 32_000,\n attemptBudgetCapacity: 5,\n budgetRestoreIntervalMs: 200,\n};\n\nconst defaultClientTransportOptions: ClientTransportOptions = {\n ...defaultTransportOptions,\n ...defaultConnectionRetryOptions,\n};\n\ntype ServerTransportOptions = TransportOptions;\n\nexport type ProvidedServerTransportOptions = Partial<ServerTransportOptions>;\n\nconst defaultServerTransportOptions: ServerTransportOptions = {\n ...defaultTransportOptions,\n};\n\n/**\n * Transports manage the lifecycle (creation/deletion) of sessions and connections. Its responsibilities include:\n *\n * 1) Constructing a new {@link Session} and {@link Connection} on {@link TransportMessage}s from new clients.\n * After constructing the {@link Connection}, {@link onConnect} is called which adds it to the connection map.\n * 2) Delegating message listening of the connection to the newly created {@link Connection}.\n * From this point on, the {@link Connection} is responsible for *reading* and *writing*\n * messages from the connection.\n * 3) When a connection is closed, the {@link Transport} calls {@link onDisconnect} which closes the\n * connection via {@link Connection.close} and removes it from the {@link connections} map.\n\n *\n * ```plaintext\n * ▲\n * incoming │\n * messages │\n * ▼\n * ┌─────────────┐ 1:N ┌───────────┐ 1:1* ┌────────────┐\n * │ Transport │ ◄─────► │ Session │ ◄─────► │ Connection │\n * └─────────────┘ └───────────┘ └────────────┘\n * ▲ * (may or may not be initialized yet)\n * │\n * ▼\n * ┌───────────┐\n * │ Message │\n * │ Listeners │\n * └───────────┘\n * ```\n * @abstract\n */\nexport abstract class Transport<ConnType extends Connection> {\n /**\n * A flag indicating whether the transport has been destroyed.\n * A destroyed transport will not attempt to reconnect and cannot be used again.\n */\n state: TransportStatus;\n\n /**\n * The {@link Codec} used to encode and decode messages.\n */\n codec: Codec;\n\n /**\n * The client ID of this transport.\n */\n clientId: TransportClientId;\n\n /**\n * The map of {@link Session}s managed by this transport.\n */\n sessions: Map<TransportClientId, Session<ConnType>>;\n\n /**\n * The map of {@link Connection}s managed by this transport.\n */\n get connections() {\n return new Map(\n [...this.sessions]\n .map(([client, session]) => [client, session.connection])\n .filter((entry): entry is [string, ConnType] => entry[1] !== undefined),\n );\n }\n\n /**\n * The event dispatcher for handling events of type EventTypes.\n */\n eventDispatcher: EventDispatcher<EventTypes>;\n\n /**\n * The options for this transport.\n */\n protected options: TransportOptions;\n\n /**\n * Creates a new Transport instance.\n * This should also set up {@link onConnect}, and {@link onDisconnect} listeners.\n * @param codec The codec used to encode and decode messages.\n * @param clientId The client ID of this transport.\n */\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedTransportOptions,\n ) {\n this.options = { ...defaultTransportOptions, ...providedOptions };\n this.eventDispatcher = new EventDispatcher();\n this.sessions = new Map();\n this.codec = this.options.codec;\n this.clientId = clientId;\n this.state = 'open';\n }\n\n /**\n * This is called immediately after a new connection is established and we\n * may or may not know the identity of the connected client.\n * It should attach all the necessary listeners to the connection for lifecycle\n * events (i.e. data, close, error)\n *\n * This method is implemented by {@link ClientTransport} and {@link ServerTransport}.\n */\n protected abstract handleConnection(\n conn: ConnType,\n to: TransportClientId,\n ): void;\n\n /**\n * Called when a new connection is established\n * and we know the identity of the connected client.\n * @param conn The connection object.\n */\n protected onConnect(\n conn: ConnType,\n connectedTo: TransportClientId,\n session: Session<ConnType>,\n isReconnect: boolean,\n ) {\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'connect',\n conn,\n });\n\n conn.telemetry = createConnectionTelemetryInfo(\n conn,\n session.telemetry.span,\n );\n\n if (isReconnect) {\n session.replaceWithNewConnection(conn);\n log?.info(`reconnected to ${connectedTo}`, session.loggingMetadata);\n }\n }\n\n protected createSession(\n to: TransportClientId,\n conn?: ConnType,\n propagationCtx?: PropagationContext,\n ) {\n const session = new Session<ConnType>(\n conn,\n this.clientId,\n to,\n this.options,\n propagationCtx,\n );\n this.sessions.set(session.to, session);\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'connect',\n session,\n });\n return session;\n }\n\n protected getOrCreateSession(\n to: TransportClientId,\n conn?: ConnType,\n sessionId?: string,\n propagationCtx?: PropagationContext,\n ) {\n let session = this.sessions.get(to);\n let isReconnect = session !== undefined;\n\n if (\n session?.advertisedSessionId !== undefined &&\n sessionId !== undefined &&\n session.advertisedSessionId !== sessionId\n ) {\n log?.info(\n `session for ${to} already exists but has a different session id (expected: ${session.advertisedSessionId}, got: ${sessionId}), creating a new one`,\n session.loggingMetadata,\n );\n this.deleteSession(session);\n isReconnect = false;\n session = undefined;\n }\n\n if (!session) {\n session = this.createSession(to, conn, propagationCtx);\n log?.info(\n `no session for ${to}, created a new one`,\n session.loggingMetadata,\n );\n }\n\n if (sessionId !== undefined) {\n session.advertisedSessionId = sessionId;\n }\n\n return { session, isReconnect };\n }\n\n protected deleteSession(session: Session<ConnType>) {\n session.close();\n session.telemetry.span.end();\n this.sessions.delete(session.to);\n log?.info(\n `session ${session.id} disconnect from ${session.to}`,\n session.loggingMetadata,\n );\n this.eventDispatcher.dispatchEvent('sessionStatus', {\n status: 'disconnect',\n session,\n });\n }\n\n /**\n * The downstream implementation needs to call this when a connection is closed.\n * @param conn The connection object.\n * @param connectedTo The peer we are connected to.\n */\n protected onDisconnect(conn: ConnType, session: Session<ConnType>) {\n conn.telemetry?.span.end();\n this.eventDispatcher.dispatchEvent('connectionStatus', {\n status: 'disconnect',\n conn,\n });\n\n session.connection = undefined;\n session.beginGrace(() => {\n session.telemetry.span.addEvent('session grace period expired');\n this.deleteSession(session);\n });\n }\n\n /**\n * Parses a message from a Uint8Array into a {@link OpaqueTransportMessage}.\n * @param msg The message to parse.\n * @returns The parsed message, or null if the message is malformed or invalid.\n */\n protected parseMsg(msg: Uint8Array): OpaqueTransportMessage | null {\n const parsedMsg = this.codec.fromBuffer(msg);\n\n if (parsedMsg === null) {\n const decodedBuffer = new TextDecoder().decode(Buffer.from(msg));\n log?.error(`received malformed msg, killing conn: ${decodedBuffer}`, {\n clientId: this.clientId,\n });\n return null;\n }\n\n if (!Value.Check(OpaqueTransportMessageSchema, parsedMsg)) {\n log?.error(`received invalid msg: ${JSON.stringify(parsedMsg)}`, {\n clientId: this.clientId,\n validationErrors: [\n ...Value.Errors(OpaqueTransportMessageSchema, parsedMsg),\n ],\n });\n return null;\n }\n\n return parsedMsg;\n }\n\n /**\n * Called when a message is received by this transport.\n * You generally shouldn't need to override this in downstream transport implementations.\n * @param msg The received message.\n */\n protected handleMsg(msg: OpaqueTransportMessage) {\n if (this.state !== 'open') return;\n const session = this.sessions.get(msg.from);\n if (!session) {\n log?.error(`no existing session for ${msg.from}`, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n // got a msg so we know the other end is alive, reset the grace period\n session.cancelGrace();\n\n log?.debug(`received msg`, {\n clientId: this.clientId,\n transportMessage: msg,\n });\n if (msg.seq !== session.nextExpectedSeq) {\n if (msg.seq < session.nextExpectedSeq) {\n log?.debug(\n `received duplicate msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq}), discarding`,\n { clientId: this.clientId, transportMessage: msg },\n );\n } else {\n const errMsg = `received out-of-order msg (got seq: ${msg.seq}, wanted seq: ${session.nextExpectedSeq})`;\n log?.error(`${errMsg}, marking connection as dead`, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.MessageOrderingViolated, errMsg);\n session.telemetry.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message order violated',\n });\n session.close();\n }\n\n return;\n }\n\n session.updateBookkeeping(msg.ack, msg.seq);\n\n // don't dispatch explicit acks\n if (!isAck(msg.controlFlags)) {\n this.eventDispatcher.dispatchEvent('message', msg);\n } else {\n log?.debug(`discarding msg (ack bit set)`, {\n clientId: this.clientId,\n transportMessage: msg,\n });\n }\n }\n\n /**\n * Adds a listener to this transport.\n * @param the type of event to listen for\n * @param handler The message handler to add.\n */\n addEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.addEventListener(type, handler);\n }\n\n /**\n * Removes a listener from this transport.\n * @param the type of event to un-listen on\n * @param handler The message handler to remove.\n */\n removeEventListener<K extends EventTypes, T extends EventHandler<K>>(\n type: K,\n handler: T,\n ): void {\n this.eventDispatcher.removeEventListener(type, handler);\n }\n\n /**\n * Sends a message over this transport, delegating to the appropriate connection to actually\n * send the message.\n * @param msg The message to send.\n * @returns The ID of the sent message or undefined if it wasn't sent\n */\n send(\n to: TransportClientId,\n msg: PartialTransportMessage,\n ): string | undefined {\n if (this.state === 'destroyed') {\n const err = 'transport is destroyed, cant send';\n log?.error(err, {\n clientId: this.clientId,\n transportMessage: msg,\n tags: ['invariant-violation'],\n });\n this.protocolError(ProtocolError.UseAfterDestroy, err);\n return undefined;\n } else if (this.state === 'closed') {\n log?.info(`transport closed when sending, discarding`, {\n clientId: this.clientId,\n transportMessage: msg,\n });\n return undefined;\n }\n\n return this.getOrCreateSession(to).session.send(msg);\n }\n\n // control helpers\n sendCloseStream(to: TransportClientId, streamId: string) {\n return this.send(to, {\n streamId: streamId,\n controlFlags: ControlFlags.StreamClosedBit,\n payload: {\n type: 'CLOSE' as const,\n } satisfies Static<typeof ControlMessagePayloadSchema>,\n });\n }\n\n protected protocolError(type: ProtocolErrorType, message: string) {\n this.eventDispatcher.dispatchEvent('protocolError', { type, message });\n }\n\n /**\n * Default close implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.close() at the end.\n * Closes the transport. Any messages sent while the transport is closed will be silently discarded.\n */\n close() {\n this.state = 'closed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n log?.info(`manually closed transport`, { clientId: this.clientId });\n }\n\n /**\n * Default destroy implementation for transports. You should override this in the downstream\n * implementation if you need to do any additional cleanup and call super.destroy() at the end.\n * Destroys the transport. Any messages sent while the transport is destroyed will throw an error.\n */\n destroy() {\n this.state = 'destroyed';\n for (const session of this.sessions.values()) {\n this.deleteSession(session);\n }\n\n log?.info(`manually destroyed transport`, { clientId: this.clientId });\n }\n}\n\nexport abstract class ClientTransport<\n ConnType extends Connection,\n HandshakeMetadata extends TSchema = TSchema,\n> extends Transport<ConnType> {\n /**\n * The options for this transport.\n */\n protected options: ClientTransportOptions;\n\n /**\n * The map of reconnect promises for each client ID.\n */\n inflightConnectionPromises: Map<TransportClientId, Promise<ConnType>>;\n retryBudget: LeakyBucketRateLimit;\n\n /**\n * A flag indicating whether the transport should automatically reconnect\n * when a connection is dropped.\n * Realistically, this should always be true for clients unless you are writing\n * tests or a special case where you don't want to reconnect.\n */\n reconnectOnConnectionDrop = true;\n\n /**\n * Optional handshake options for this client.\n */\n handshakeExtensions?: ClientHandshakeOptions<HandshakeMetadata>;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedClientTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultClientTransportOptions,\n ...providedOptions,\n };\n this.inflightConnectionPromises = new Map();\n this.retryBudget = new LeakyBucketRateLimit(this.options);\n }\n\n extendHandshake(options: ClientHandshakeOptions<HandshakeMetadata>) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType, to: TransportClientId): void {\n if (this.state !== 'open') return;\n let session: Session<ConnType> | undefined = undefined;\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n log?.warn(\n `connection to ${to} timed out waiting for handshake, closing`,\n { clientId: this.clientId, connectedTo: to, connId: conn.id },\n );\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const handshakeHandler = (data: Uint8Array) => {\n const maybeSession = this.receiveHandshakeResponseMessage(data, conn);\n if (!maybeSession) {\n conn.close();\n return;\n } else {\n session = maybeSession;\n clearTimeout(handshakeTimeout);\n }\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener((data) => {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'message parse failure',\n });\n conn.close();\n return;\n }\n\n this.handleMsg(parsed);\n });\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (session) {\n this.onDisconnect(conn, session);\n }\n log?.info(`connection to ${to} disconnected`, {\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n });\n this.inflightConnectionPromises.delete(to);\n if (this.reconnectOnConnectionDrop) {\n void this.connect(to);\n }\n });\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n log?.warn(`error in connection to ${to}: ${coerceErrorString(err)}`, {\n ...session?.loggingMetadata,\n clientId: this.clientId,\n connectedTo: to,\n });\n });\n }\n\n receiveHandshakeResponseMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Session<ConnType> | false {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeResponseSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake response',\n });\n log?.warn(`received invalid handshake resp`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n validationErrors: [\n ...Value.Errors(\n ControlMessageHandshakeResponseSchema,\n parsed.payload,\n ),\n ],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake resp',\n );\n return false;\n }\n\n if (!parsed.payload.status.ok) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake rejected',\n });\n log?.warn(`received handshake rejection`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n parsed.payload.status.reason,\n );\n return false;\n }\n\n log?.debug(`handshake from ${parsed.from} ok`, {\n clientId: this.clientId,\n connectedTo: parsed.from,\n transportMessage: parsed,\n });\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.status.sessionId,\n );\n\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n // After a successful connection, we start restoring the budget\n // so that the next time we try to connect, we don't hit the client\n // with backoff forever.\n this.retryBudget.startRestoringBudget(parsed.from);\n return session;\n }\n\n /**\n * Abstract method that creates a new {@link Connection} object.\n * This should call {@link handleConnection} when the connection is created.\n * The downstream client implementation needs to implement this.\n *\n * @param to The client ID of the node to connect to.\n * @returns The new connection object.\n */\n protected abstract createNewOutgoingConnection(\n to: TransportClientId,\n ): Promise<ConnType>;\n\n /**\n * Manually attempts to connect to a client.\n * @param to The client ID of the node to connect to.\n */\n async connect(to: TransportClientId): Promise<void> {\n const canProceedWithConnection = () => this.state === 'open';\n if (!canProceedWithConnection()) {\n log?.info(\n `transport state is no longer open, cancelling attempt to connect to ${to}`,\n { clientId: this.clientId, connectedTo: to },\n );\n return;\n }\n\n let reconnectPromise = this.inflightConnectionPromises.get(to);\n if (!reconnectPromise) {\n // check budget\n const budgetConsumed = this.retryBudget.getBudgetConsumed(to);\n if (!this.retryBudget.hasBudget(to)) {\n const errMsg = `tried to connect to ${to} but retry budget exceeded (more than ${budgetConsumed} attempts in the last ${this.retryBudget.totalBudgetRestoreTime}ms)`;\n log?.warn(errMsg, { clientId: this.clientId, connectedTo: to });\n this.protocolError(ProtocolError.RetriesExceeded, errMsg);\n return;\n }\n\n let sleep = Promise.resolve();\n const backoffMs = this.retryBudget.getBackoffMs(to);\n if (backoffMs > 0) {\n sleep = new Promise((resolve) => setTimeout(resolve, backoffMs));\n }\n\n log?.info(`attempting connection to ${to} (${backoffMs}ms backoff)`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n this.retryBudget.consumeBudget(to);\n reconnectPromise = tracer.startActiveSpan('connect', async (span) => {\n try {\n span.addEvent('backoff', { backoffMs });\n await sleep;\n if (!canProceedWithConnection()) {\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('connecting');\n const conn = await this.createNewOutgoingConnection(to);\n if (!canProceedWithConnection()) {\n log?.info(\n `transport state is no longer open, closing pre-handshake connection to ${to}`,\n {\n clientId: this.clientId,\n connectedTo: to,\n connId: conn.id,\n },\n );\n conn.close();\n throw new Error('transport state is no longer open');\n }\n\n span.addEvent('sending handshake');\n const ok = await this.sendHandshake(to, conn);\n if (!ok) {\n conn.close();\n throw new Error('failed to send handshake');\n }\n\n return conn;\n } catch (err) {\n // rethrow the error so that the promise is rejected\n // as it was before we wrapped it in a span\n const errStr = coerceErrorString(err);\n span.recordException(errStr);\n span.setStatus({ code: SpanStatusCode.ERROR });\n throw err;\n } finally {\n span.end();\n }\n });\n\n this.inflightConnectionPromises.set(to, reconnectPromise);\n } else {\n log?.info(`attempting connection to ${to} (reusing previous attempt)`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n }\n\n try {\n await reconnectPromise;\n } catch (error: unknown) {\n this.inflightConnectionPromises.delete(to);\n const errStr = coerceErrorString(error);\n\n if (!this.reconnectOnConnectionDrop || !canProceedWithConnection()) {\n log?.warn(`connection to ${to} failed (${errStr})`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n } else {\n log?.warn(`connection to ${to} failed (${errStr}), retrying`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n return this.connect(to);\n }\n }\n }\n\n protected deleteSession(session: Session<ConnType>) {\n this.inflightConnectionPromises.delete(session.to);\n super.deleteSession(session);\n }\n\n protected async sendHandshake(to: TransportClientId, conn: ConnType) {\n let metadata: Static<HandshakeMetadata> | undefined;\n\n if (this.handshakeExtensions) {\n metadata = await this.handshakeExtensions.construct();\n if (!Value.Check(this.handshakeExtensions.schema, metadata)) {\n log?.error(`constructed handshake metadata did not match schema`, {\n clientId: this.clientId,\n connectedTo: to,\n validationErrors: [\n ...Value.Errors(this.handshakeExtensions.schema, metadata),\n ],\n tags: ['invariant-violation'],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'handshake metadata did not match schema',\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake meta mismatch',\n });\n return false;\n }\n }\n\n // dont pass conn here as we dont want the session to start using the conn\n // until we have finished the handshake\n const { session } = this.getOrCreateSession(to);\n const requestMsg = handshakeRequestMessage(\n this.clientId,\n to,\n session.id,\n metadata,\n getPropagationContext(session.telemetry.ctx),\n );\n log?.debug(`sending handshake request to ${to}`, {\n clientId: this.clientId,\n connectedTo: to,\n });\n conn.send(this.codec.toBuffer(requestMsg));\n return true;\n }\n\n close() {\n this.retryBudget.close();\n super.close();\n }\n}\n\nexport abstract class ServerTransport<\n ConnType extends Connection,\n CustomMetadataSchema extends TSchema = TSchema,\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<CustomMetadataSchema>;\n\n /**\n * A map of session handshake data for each session.\n */\n sessionHandshakeMetadata: WeakMap<Session<ConnType>, ParsedMetadata>;\n\n constructor(\n clientId: TransportClientId,\n providedOptions?: ProvidedServerTransportOptions,\n ) {\n super(clientId, providedOptions);\n this.options = {\n ...defaultServerTransportOptions,\n ...providedOptions,\n };\n this.sessionHandshakeMetadata = new WeakMap();\n log?.info(`initiated server transport`, {\n clientId: this.clientId,\n protocolVersion: PROTOCOL_VERSION,\n });\n }\n\n extendHandshake(options: ServerHandshakeOptions<CustomMetadataSchema>) {\n this.handshakeExtensions = options;\n }\n\n protected handleConnection(conn: ConnType) {\n if (this.state !== 'open') return;\n\n log?.info(`new incoming connection`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n\n let session: Session<ConnType> | undefined = undefined;\n const client = () => session?.to ?? 'unknown';\n\n // kill the conn after the grace period if we haven't received a handshake\n const handshakeTimeout = setTimeout(() => {\n if (!session) {\n log?.warn(\n `connection to ${client()} timed out waiting for handshake, closing`,\n {\n clientId: this.clientId,\n connectedTo: client(),\n connId: conn.id,\n },\n );\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'handshake timeout',\n });\n conn.close();\n }\n }, this.options.sessionDisconnectGraceMs);\n\n const buffer: Array<Uint8Array> = [];\n let receivedHandshakeMessage = false;\n\n const handshakeHandler = (data: Uint8Array) => {\n // if we've already received, just buffer the data\n if (receivedHandshakeMessage) {\n buffer.push(data);\n return;\n }\n\n receivedHandshakeMessage = true;\n clearTimeout(handshakeTimeout);\n\n void this.receiveHandshakeRequestMessage(data, conn).then(\n (maybeSession) => {\n if (!maybeSession) {\n conn.close();\n return;\n }\n\n session = maybeSession;\n\n // when we are done handshake sequence,\n // remove handshake listener and use the normal message listener\n const dataHandler = (data: Uint8Array) => {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.close();\n return;\n }\n\n this.handleMsg(parsed);\n };\n\n // process any data we missed\n for (const data of buffer) {\n dataHandler(data);\n }\n\n conn.removeDataListener(handshakeHandler);\n conn.addDataListener(dataHandler);\n buffer.length = 0;\n },\n );\n };\n\n conn.addDataListener(handshakeHandler);\n conn.addCloseListener(() => {\n if (!session) return;\n log?.info(`connection to ${client()} disconnected`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n this.onDisconnect(conn, session);\n });\n\n conn.addErrorListener((err) => {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'connection error',\n });\n if (!session) return;\n log?.warn(\n `connection to ${client()} got an error: ${coerceErrorString(err)}`,\n { clientId: this.clientId, connId: conn.id },\n );\n });\n }\n\n private async validateHandshakeMetadata(\n conn: ConnType,\n session: Session<ConnType> | undefined,\n rawMetadata: Static<\n typeof ControlMessageHandshakeRequestSchema\n >['metadata'],\n from: TransportClientId,\n ): Promise<ParsedMetadata | false> {\n let parsedMetadata: ParsedMetadata = {};\n if (this.handshakeExtensions) {\n // check that the metadata that was sent is the correct shape\n if (!Value.Check(this.handshakeExtensions.schema, rawMetadata)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'malformed handshake meta',\n });\n const reason = 'received malformed handshake metadata';\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n // note: do _not_ log the metadata, it may contain secrets\n log?.warn(`received malformed handshake metadata from ${from}`, {\n clientId: this.clientId,\n connId: conn.id,\n validationErrors: [\n ...Value.Errors(this.handshakeExtensions.schema, rawMetadata),\n ],\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n const previousParsedMetadata = session\n ? this.sessionHandshakeMetadata.get(session)\n : undefined;\n\n parsedMetadata = await this.handshakeExtensions.validate(\n rawMetadata,\n previousParsedMetadata,\n );\n\n // handler rejected the connection\n if (parsedMetadata === false) {\n const reason = 'rejected by handshake handler';\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: reason,\n });\n const responseMsg = handshakeResponseMessage(this.clientId, from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n log?.warn(`rejected handshake from ${from}`, {\n clientId: this.clientId,\n connId: conn.id,\n });\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n }\n\n return parsedMetadata;\n }\n\n async receiveHandshakeRequestMessage(\n data: Uint8Array,\n conn: ConnType,\n ): Promise<Session<ConnType> | false> {\n const parsed = this.parseMsg(data);\n if (!parsed) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'non-transport message',\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'received non-transport message',\n );\n return false;\n }\n\n if (!Value.Check(ControlMessageHandshakeRequestSchema, parsed.payload)) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'invalid handshake request',\n });\n const reason = 'received invalid handshake msg';\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n log?.warn(reason, {\n clientId: this.clientId,\n connId: conn.id,\n // safe to log metadata here as we remove the payload\n // before passing it to user-land\n transportMessage: parsed,\n validationErrors: [\n ...Value.Errors(ControlMessageHandshakeRequestSchema, parsed.payload),\n ],\n });\n this.protocolError(\n ProtocolError.HandshakeFailed,\n 'invalid handshake request',\n );\n return false;\n }\n\n // double check protocol version here\n const gotVersion = parsed.payload.protocolVersion;\n if (gotVersion !== PROTOCOL_VERSION) {\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: 'incorrect protocol version',\n });\n\n const reason = `incorrect version (got: ${gotVersion} wanted ${PROTOCOL_VERSION})`;\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: false,\n reason,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n log?.warn(\n `received handshake msg with incompatible protocol version (got: ${gotVersion}, expected: ${PROTOCOL_VERSION})`,\n { clientId: this.clientId, connId: conn.id },\n );\n this.protocolError(ProtocolError.HandshakeFailed, reason);\n return false;\n }\n\n const oldSession = this.sessions.get(parsed.from);\n const parsedMetadata = await this.validateHandshakeMetadata(\n conn,\n oldSession,\n parsed.payload.metadata,\n parsed.from,\n );\n\n if (parsedMetadata === false) {\n return false;\n }\n\n const { session, isReconnect } = this.getOrCreateSession(\n parsed.from,\n conn,\n parsed.payload.sessionId,\n parsed.tracing,\n );\n\n this.sessionHandshakeMetadata.set(session, parsedMetadata);\n\n log?.debug(\n `handshake from ${parsed.from} ok, responding with handshake success`,\n { clientId: this.clientId, connId: conn.id },\n );\n const responseMsg = handshakeResponseMessage(this.clientId, parsed.from, {\n ok: true,\n sessionId: session.id,\n });\n conn.send(this.codec.toBuffer(responseMsg));\n this.onConnect(conn, parsed.from, session, isReconnect);\n\n return session;\n }\n}\n","import { Type, TSchema, Static } from '@sinclair/typebox';\nimport { nanoid } from 'nanoid';\nimport { PropagationContext } from '../tracing';\nimport { ParsedMetadata } from '../router/context';\n\n/**\n * Control flags for transport messages.\n * An RPC message is coded with StreamOpenBit | StreamClosedBit.\n * Streams are expected to start with StreamOpenBit sent and the client SHOULD send an empty\n * message with StreamClosedBit to close the stream handler on the server, indicating that\n * it will not be using the stream anymore.\n */\nexport const enum ControlFlags {\n AckBit = 0b0001,\n StreamOpenBit = 0b0010,\n StreamClosedBit = 0b0100,\n}\n\nexport interface ClientHandshakeOptions<MetadataSchema extends TSchema> {\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: () => Static<MetadataSchema> | Promise<Static<MetadataSchema>>;\n}\n\nexport interface ServerHandshakeOptions<MetadataSchema extends TSchema> {\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: (\n metadata: Static<MetadataSchema>,\n previousParsedMetadata?: ParsedMetadata,\n ) => false | ParsedMetadata | Promise<false | ParsedMetadata>;\n}\n\n/**\n * Generic Typebox schema for a transport message.\n * @template T The type of the payload.\n * @param {T} t The payload schema.\n * @returns The transport message schema.\n */\nexport const TransportMessageSchema = <T extends TSchema>(t: T) =>\n Type.Object({\n id: Type.String(),\n from: Type.String(),\n to: Type.String(),\n seq: Type.Integer(),\n ack: Type.Integer(),\n serviceName: Type.Optional(Type.String()),\n procedureName: Type.Optional(Type.String()),\n streamId: Type.String(),\n controlFlags: Type.Integer(),\n tracing: Type.Optional(\n Type.Object({\n traceparent: Type.String(),\n tracestate: Type.String(),\n }),\n ),\n payload: t,\n });\n\n/**\n * Defines the schema for a transport acknowledgement message. This is never constructed manually\n * and is only used internally by the library for tracking inflight messages.\n * @returns The transport message schema.\n */\nexport const ControlMessageAckSchema = Type.Object({\n type: Type.Literal('ACK'),\n});\n\n/**\n * Defines the schema for a transport close message. This is never constructed manually and is only\n * used internally by the library for closing and cleaning up streams.\n */\nexport const ControlMessageCloseSchema = Type.Object({\n type: Type.Literal('CLOSE'),\n});\n\nexport const PROTOCOL_VERSION = 'v1.1';\nexport const ControlMessageHandshakeRequestSchema = Type.Object({\n type: Type.Literal('HANDSHAKE_REQ'),\n protocolVersion: Type.String(),\n sessionId: Type.String(),\n metadata: Type.Optional(Type.Unknown()),\n});\n\nexport const ControlMessageHandshakeResponseSchema = Type.Object({\n type: Type.Literal('HANDSHAKE_RESP'),\n status: Type.Union([\n Type.Object({\n ok: Type.Literal(true),\n sessionId: Type.String(),\n }),\n Type.Object({\n ok: Type.Literal(false),\n reason: Type.String(),\n }),\n ]),\n});\n\nexport const ControlMessagePayloadSchema = Type.Union([\n ControlMessageCloseSchema,\n ControlMessageAckSchema,\n ControlMessageHandshakeRequestSchema,\n ControlMessageHandshakeResponseSchema,\n]);\n\n/**\n * Defines the schema for an opaque transport message that is agnostic to any\n * procedure/service.\n * @returns The transport message schema.\n */\nexport const OpaqueTransportMessageSchema = TransportMessageSchema(\n Type.Unknown(),\n);\n\n/**\n * Represents a transport message. This is the same type as {@link TransportMessageSchema} but\n * we can't statically infer generics from generic Typebox schemas so we have to define it again here.\n *\n * TypeScript can't enforce types when a bitmask is involved, so these are the semantics of\n * `controlFlags`:\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `streamId` must be set to a unique value\n * (suggestion: use `nanoid`).\n * * If `controlFlags & StreamOpenBit == StreamOpenBit`, `serviceName` and `procedureName` must be set.\n * * If `controlFlags & StreamClosedBit == StreamClosedBit` and the kind is `stream` or `subscription`,\n * `payload` should be discarded (usually contains a control message).\n * * If `controlFlags & AckBit == AckBit`, the message is an explicit acknowledgement message and doesn't\n * contain any payload that is relevant to the application so should not be delivered.\n * @template Payload The type of the payload.\n */\nexport interface TransportMessage<Payload = unknown> {\n id: string;\n from: string;\n to: string;\n seq: number;\n ack: number;\n serviceName?: string;\n procedureName?: string;\n streamId: string;\n controlFlags: number;\n tracing?: PropagationContext;\n payload: Payload;\n}\n\nexport type PartialTransportMessage<Payload = unknown> = Omit<\n TransportMessage<Payload>,\n 'id' | 'from' | 'to' | 'seq' | 'ack'\n>;\n\nexport function handshakeRequestMessage(\n from: TransportClientId,\n to: TransportClientId,\n sessionId: string,\n metadata?: unknown,\n tracing?: PropagationContext,\n): TransportMessage<Static<typeof ControlMessageHandshakeRequestSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n tracing,\n payload: {\n type: 'HANDSHAKE_REQ',\n protocolVersion: PROTOCOL_VERSION,\n sessionId,\n metadata,\n } satisfies Static<typeof ControlMessageHandshakeRequestSchema>,\n };\n}\n\nexport function handshakeResponseMessage(\n from: TransportClientId,\n to: TransportClientId,\n status: Static<typeof ControlMessageHandshakeResponseSchema>['status'],\n): TransportMessage<Static<typeof ControlMessageHandshakeResponseSchema>> {\n return {\n id: nanoid(),\n from,\n to,\n seq: 0,\n ack: 0,\n streamId: nanoid(),\n controlFlags: 0,\n payload: {\n type: 'HANDSHAKE_RESP',\n status,\n } satisfies Static<typeof ControlMessageHandshakeResponseSchema>,\n };\n}\n\n/**\n * A type alias for a transport message with an opaque payload.\n * @template T - The type of the opaque payload.\n */\nexport type OpaqueTransportMessage = TransportMessage;\nexport type TransportClientId = string;\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is an ack message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the AckBit, false otherwise.\n */\nexport function isAck(controlFlag: number): boolean {\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n return (controlFlag & ControlFlags.AckBit) === ControlFlags.AckBit;\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream open message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamOpenBit, false otherwise.\n */\nexport function isStreamOpen(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamOpenBit) === ControlFlags.StreamOpenBit\n );\n}\n\n/**\n * Checks if the given control flag (usually found in msg.controlFlag) is a stream close message.\n * @param controlFlag - The control flag to check.\n * @returns True if the control flag contains the StreamCloseBit, false otherwise.\n */\nexport function isStreamClose(controlFlag: number): boolean {\n return (\n /* eslint-disable-next-line @typescript-eslint/no-unsafe-enum-comparison */\n (controlFlag & ControlFlags.StreamClosedBit) ===\n ControlFlags.StreamClosedBit\n );\n}\n","import { ValueError } from '@sinclair/typebox/value';\nimport { OpaqueTransportMessage } from '../transport';\n\nconst LoggingLevels = {\n debug: -1,\n info: 0,\n warn: 1,\n error: 2,\n} as const;\ntype LoggingLevel = keyof typeof LoggingLevels;\n\nexport type LogFn = (\n msg: string,\n ctx?: MessageMetadata,\n level?: LoggingLevel,\n) => void;\nexport type Logger = {\n [key in LoggingLevel]: (msg: string, metadata?: MessageMetadata) => void;\n};\n\nexport type Tags = 'invariant-violation';\n\nconst cleanedLogFn = (log: LogFn) => {\n return (msg: string, metadata?: MessageMetadata) => {\n // skip cloning object if metadata has no transportMessage\n if (!metadata?.transportMessage) {\n log(msg, metadata);\n return;\n }\n\n // clone metadata and clean transportMessage\n const { payload, tracing, ...rest } = metadata.transportMessage;\n metadata.transportMessage = rest;\n log(msg, metadata);\n };\n};\n\nexport type MessageMetadata = Partial<{\n protocolVersion: string;\n clientId: string;\n connectedTo: string;\n sessionId: string;\n connId: string;\n transportMessage: Partial<OpaqueTransportMessage>;\n validationErrors: Array<ValueError>;\n tags: Array<Tags>;\n}>;\n\nclass BaseLogger implements Logger {\n minLevel: LoggingLevel;\n private output: LogFn;\n\n constructor(output: LogFn, minLevel: LoggingLevel = 'info') {\n this.minLevel = minLevel;\n this.output = output;\n }\n\n debug(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.debug) {\n this.output(msg, metadata ?? {}, 'debug');\n }\n }\n\n info(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.info) {\n this.output(msg, metadata ?? {}, 'info');\n }\n }\n\n warn(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.warn) {\n this.output(msg, metadata ?? {}, 'warn');\n }\n }\n\n error(msg: string, metadata?: MessageMetadata) {\n if (LoggingLevels[this.minLevel] <= LoggingLevels.error) {\n this.output(msg, metadata ?? {}, 'error');\n }\n }\n}\n\nexport const stringLogger: LogFn = (msg, _ctx, level = 'info') => {\n console.log(`[river:${level}] ${msg}`);\n};\n\nconst colorMap = {\n debug: '\\u001b[34m',\n info: '\\u001b[32m',\n warn: '\\u001b[33m',\n error: '\\u001b[31m',\n};\n\nexport const coloredStringLogger: LogFn = (msg, _ctx, level = 'info') => {\n const color = colorMap[level];\n console.log(`[river:${color}${level}\\u001b[0m] ${msg}`);\n};\n\nexport const jsonLogger: LogFn = (msg, ctx, level) => {\n console.log(JSON.stringify({ msg, ctx, level }));\n};\n\nlet _log: Logger | undefined = undefined;\n\n// log proxy to clean transport payloads\nexport let log: Logger | undefined = undefined;\nconst createLogProxy = (log: Logger) => ({\n debug: cleanedLogFn(log.debug.bind(log)),\n info: cleanedLogFn(log.info.bind(log)),\n warn: cleanedLogFn(log.warn.bind(log)),\n error: cleanedLogFn(log.error.bind(log)),\n});\n\nexport function bindLogger(\n fn: LogFn | Logger | undefined,\n level?: LoggingLevel,\n): void {\n // clear logger\n if (!fn) {\n _log = undefined;\n log = undefined;\n return;\n }\n\n // construct logger from fn\n if (typeof fn === 'function') {\n _log = new BaseLogger(fn, level);\n log = createLogProxy(_log);\n return;\n }\n\n // just assign\n _log = fn;\n log = createLogProxy(_log);\n}\n","import { OpaqueTransportMessage } from './message';\nimport { Connection, Session } from './session';\n\ntype ConnectionStatus = 'connect' | 'disconnect';\nexport const ProtocolError = {\n RetriesExceeded: 'conn_retry_exceeded',\n HandshakeFailed: 'handshake_failed',\n UseAfterDestroy: 'use_after_destroy',\n MessageOrderingViolated: 'message_ordering_violated',\n} as const;\n\nexport type ProtocolErrorType =\n (typeof ProtocolError)[keyof typeof ProtocolError];\n\nexport interface EventMap {\n message: OpaqueTransportMessage;\n connectionStatus: {\n status: ConnectionStatus;\n conn: Connection;\n };\n sessionStatus: {\n status: ConnectionStatus;\n session: Session<Connection>;\n };\n protocolError: {\n type: ProtocolErrorType;\n message: string;\n };\n}\n\nexport type EventTypes = keyof EventMap;\nexport type EventHandler<K extends EventTypes> = (\n event: EventMap[K],\n) => unknown;\n\nexport class EventDispatcher<T extends EventTypes> {\n private eventListeners: { [K in T]?: Set<EventHandler<K>> } = {};\n\n numberOfListeners<K extends T>(eventType: K) {\n return this.eventListeners[eventType]?.size ?? 0;\n }\n\n addEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n if (!this.eventListeners[eventType]) {\n this.eventListeners[eventType] = new Set();\n }\n\n this.eventListeners[eventType]?.add(handler);\n }\n\n removeEventListener<K extends T>(eventType: K, handler: EventHandler<K>) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n this.eventListeners[eventType]?.delete(handler);\n }\n }\n\n dispatchEvent<K extends T>(eventType: K, event: EventMap[K]) {\n const handlers = this.eventListeners[eventType];\n if (handlers) {\n for (const handler of handlers) {\n handler(event);\n }\n }\n }\n}\n","import { customAlphabet } from 'nanoid';\nimport {\n ControlFlags,\n ControlMessageAckSchema,\n OpaqueTransportMessage,\n PartialTransportMessage,\n TransportClientId,\n TransportMessage,\n} from './message';\nimport { Codec } from '../codec';\nimport { MessageMetadata, log } from '../logging/log';\nimport { Static } from '@sinclair/typebox';\nimport {\n PropagationContext,\n TelemetryInfo,\n createSessionTelemetryInfo,\n} from '../tracing';\nimport { SpanStatusCode } from '@opentelemetry/api';\n\nconst nanoid = customAlphabet('1234567890abcdefghijklmnopqrstuvxyz', 6);\nexport const unsafeId = () => nanoid();\n\ntype SequenceNumber = number;\n\n/**\n * A connection is the actual raw underlying transport connection.\n * It’s responsible for dispatching to/from the actual connection itself\n * This should be instantiated as soon as the client/server has a connection\n * It’s tied to the lifecycle of the underlying transport connection (i.e. if the WS drops, this connection should be deleted)\n */\nexport abstract class Connection {\n id: string;\n telemetry?: TelemetryInfo;\n constructor() {\n this.id = `conn-${nanoid(12)}`; // for debugging, no collision safety needed\n }\n\n /**\n * Handle adding a callback for when a message is received.\n * @param msg The message that was received.\n */\n abstract addDataListener(cb: (msg: Uint8Array) => void): void;\n abstract removeDataListener(cb: (msg: Uint8Array) => void): void;\n\n /**\n * Handle adding a callback for when the connection is closed.\n * This should also be called if an error happens.\n * @param cb The callback to call when the connection is closed.\n */\n abstract addCloseListener(cb: () => void): void;\n\n /**\n * Handle adding a callback for when an error is received.\n * This should only be used for logging errors, all cleanup\n * should be delegated to addCloseListener.\n *\n * The implementer should take care such that the implemented\n * connection will call both the close and error callbacks\n * on an error.\n *\n * @param cb The callback to call when an error is received.\n */\n abstract addErrorListener(cb: (err: Error) => void): void;\n\n /**\n * Sends a message over the connection.\n * @param msg The message to send.\n * @returns true if the message was sent, false otherwise.\n */\n abstract send(msg: Uint8Array): boolean;\n\n /**\n * Closes the connection.\n */\n abstract close(): void;\n}\n\nexport interface SessionOptions {\n /**\n * Frequency at which to send heartbeat acknowledgements\n */\n heartbeatIntervalMs: number;\n /**\n * Number of elapsed heartbeats without a response message before we consider\n * the connection dead.\n */\n heartbeatsUntilDead: number;\n /**\n * Duration to wait between connection disconnect and actual session disconnect\n */\n sessionDisconnectGraceMs: number;\n /**\n * The codec to use for encoding/decoding messages over the wire\n */\n codec: Codec;\n}\n\n/**\n * A session is a higher-level abstraction that operates over the span of potentially multiple transport-level connections\n * - It’s responsible for tracking any metadata for a particular client that might need to be persisted across connections (i.e. the sendBuffer, ack, seq)\n * - This will only be considered disconnected if\n * - the server tells the client that we’ve reconnected but it doesn’t recognize us anymore (server definitely died) or\n * - we hit a grace period after a connection disconnect\n */\nexport class Session<ConnType extends Connection> {\n private codec: Codec;\n private options: SessionOptions;\n telemetry: TelemetryInfo;\n\n /**\n * The buffer of messages that have been sent but not yet acknowledged.\n */\n private sendBuffer: Array<OpaqueTransportMessage> = [];\n\n /**\n * The active connection associated with this session\n */\n connection?: ConnType;\n readonly from: TransportClientId;\n readonly to: TransportClientId;\n\n /**\n * The unique ID of this session.\n */\n id: string;\n\n /**\n * What the other side advertised as their session ID\n * for this session.\n */\n advertisedSessionId?: string;\n\n /**\n * Number of messages we've sent along this session (excluding handshake and acks)\n */\n private seq: SequenceNumber = 0;\n\n /**\n * Number of unique messages we've received this session (excluding handshake and acks)\n */\n private ack: SequenceNumber = 0;\n\n /**\n * The grace period between when the inner connection is disconnected\n * and when we should consider the entire session disconnected.\n */\n private disconnectionGrace?: ReturnType<typeof setTimeout>;\n\n /**\n * Number of heartbeats we've sent without a response.\n */\n private heartbeatMisses: number;\n\n /**\n * The interval for sending heartbeats.\n */\n private heartbeat: ReturnType<typeof setInterval>;\n\n constructor(\n conn: ConnType | undefined,\n from: TransportClientId,\n to: TransportClientId,\n options: SessionOptions,\n propagationCtx?: PropagationContext,\n ) {\n this.id = `session-${nanoid(12)}`;\n this.options = options;\n this.from = from;\n this.to = to;\n this.connection = conn;\n this.codec = options.codec;\n\n // setup heartbeat\n this.heartbeatMisses = 0;\n this.heartbeat = setInterval(\n () => this.sendHeartbeat(),\n options.heartbeatIntervalMs,\n );\n this.telemetry = createSessionTelemetryInfo(this, propagationCtx);\n }\n\n get loggingMetadata(): Omit<MessageMetadata, 'parsedMsg'> {\n return {\n clientId: this.from,\n connectedTo: this.to,\n sessionId: this.id,\n connId: this.connection?.id,\n };\n }\n\n /**\n * Sends a message over the session's connection.\n * If the connection is not ready or the message fails to send, the message can be buffered for retry unless skipped.\n *\n * @param msg The partial message to be sent, which will be constructed into a full message.\n * @param addToSendBuff Whether to add the message to the send buffer for retry.\n * @returns The full transport ID of the message that was attempted to be sent.\n */\n send(msg: PartialTransportMessage): string {\n const fullMsg: TransportMessage = this.constructMsg(msg);\n log?.debug(`sending msg`, {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n });\n\n if (this.connection) {\n const ok = this.connection.send(this.codec.toBuffer(fullMsg));\n if (ok) return fullMsg.id;\n log?.info(\n `failed to send msg to ${fullMsg.to}, connection is probably dead`,\n {\n ...this.loggingMetadata,\n transportMessage: fullMsg,\n },\n );\n } else {\n log?.info(\n `failed to send msg to ${fullMsg.to}, connection not ready yet`,\n { ...this.loggingMetadata, transportMessage: fullMsg },\n );\n }\n\n return fullMsg.id;\n }\n\n sendHeartbeat() {\n const misses = this.heartbeatMisses;\n const missDuration = misses * this.options.heartbeatIntervalMs;\n if (misses > this.options.heartbeatsUntilDead) {\n if (this.connection) {\n log?.info(\n `closing connection to ${this.to} due to inactivity (missed ${misses} heartbeats which is ${missDuration}ms)`,\n this.loggingMetadata,\n );\n this.telemetry.span.addEvent('closing connection due to inactivity');\n this.closeStaleConnection();\n }\n return;\n }\n\n this.send({\n streamId: 'heartbeat',\n controlFlags: ControlFlags.AckBit,\n payload: {\n type: 'ACK',\n } satisfies Static<typeof ControlMessageAckSchema>,\n });\n this.heartbeatMisses++;\n }\n\n resetBufferedMessages() {\n this.sendBuffer = [];\n this.seq = 0;\n this.ack = 0;\n }\n\n sendBufferedMessages(conn: ConnType) {\n log?.info(`resending ${this.sendBuffer.length} buffered messages`, {\n ...this.loggingMetadata,\n connId: conn.id,\n });\n for (const msg of this.sendBuffer) {\n log?.debug(`resending msg`, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n });\n const ok = conn.send(this.codec.toBuffer(msg));\n if (!ok) {\n // this should never happen unless the transport has an\n // incorrect implementation of `createNewOutgoingConnection`\n const errMsg = `failed to send buffered message to ${this.to} (sus, this is a fresh connection)`;\n conn.telemetry?.span.setStatus({\n code: SpanStatusCode.ERROR,\n message: errMsg,\n });\n\n log?.error(errMsg, {\n ...this.loggingMetadata,\n transportMessage: msg,\n connId: conn.id,\n tags: ['invariant-violation'],\n });\n conn.close();\n return;\n }\n }\n }\n\n updateBookkeeping(ack: number, seq: number) {\n if (seq + 1 < this.ack) {\n log?.error(`received stale seq ${seq} + 1 < ${this.ack}`, {\n ...this.loggingMetadata,\n tags: ['invariant-violation'],\n });\n return;\n }\n\n this.sendBuffer = this.sendBuffer.filter((unacked) => unacked.seq >= ack);\n this.ack = seq + 1;\n }\n\n closeStaleConnection(conn?: ConnType) {\n if (this.connection === undefined || this.connection === conn) return;\n log?.info(\n `closing old inner connection from session to ${this.to}`,\n this.loggingMetadata,\n );\n this.connection.close();\n this.connection = undefined;\n }\n\n replaceWithNewConnection(newConn: ConnType) {\n this.closeStaleConnection(newConn);\n this.cancelGrace();\n this.sendBufferedMessages(newConn);\n this.connection = newConn;\n }\n\n beginGrace(cb: () => void) {\n log?.info(\n `starting ${this.options.sessionDisconnectGraceMs}ms grace period until session to ${this.to} is closed`,\n this.loggingMetadata,\n );\n this.disconnectionGrace = setTimeout(() => {\n this.close();\n cb();\n }, this.options.sessionDisconnectGraceMs);\n }\n\n // called on reconnect of the underlying session\n cancelGrace() {\n this.heartbeatMisses = 0;\n clearTimeout(this.disconnectionGrace);\n this.disconnectionGrace = undefined;\n }\n\n // closed when we want to discard the whole session\n // (i.e. shutdown or session disconnect)\n close() {\n this.closeStaleConnection();\n this.cancelGrace();\n this.resetBufferedMessages();\n clearInterval(this.heartbeat);\n }\n\n get connected() {\n return this.connection !== undefined;\n }\n\n get nextExpectedSeq() {\n return this.ack;\n }\n\n constructMsg<Payload>(\n partialMsg: PartialTransportMessage<Payload>,\n ): TransportMessage<Payload> {\n const msg = {\n ...partialMsg,\n id: unsafeId(),\n to: this.to,\n from: this.from,\n seq: this.seq,\n ack: this.ack,\n };\n\n this.seq++;\n this.sendBuffer.push(msg);\n return msg;\n }\n\n inspectSendBuffer(): ReadonlyArray<OpaqueTransportMessage> {\n return this.sendBuffer;\n }\n}\n","import {\n Context,\n Span,\n SpanKind,\n context,\n propagation,\n trace,\n} from '@opentelemetry/api';\nimport { version as RIVER_VERSION } from '../package.json';\nimport { ValidProcType } from '../router';\nimport { Connection, OpaqueTransportMessage, Session } from '../transport';\n\nexport interface PropagationContext {\n traceparent: string;\n tracestate: string;\n}\n\nexport interface TelemetryInfo {\n span: Span;\n ctx: Context;\n}\n\nexport function getPropagationContext(\n ctx: Context,\n): PropagationContext | undefined {\n const tracing = {\n traceparent: '',\n tracestate: '',\n };\n propagation.inject(ctx, tracing);\n return tracing;\n}\n\nexport function createSessionTelemetryInfo(\n session: Session<Connection>,\n propagationCtx?: PropagationContext,\n): TelemetryInfo {\n const ctx = propagationCtx\n ? propagation.extract(context.active(), propagationCtx)\n : context.active();\n\n const span = tracer.startSpan(\n `session ${session.id}`,\n {\n attributes: {\n component: 'river',\n 'river.session.id': session.id,\n 'river.session.to': session.to,\n 'river.session.from': session.from,\n },\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createConnectionTelemetryInfo(\n connection: Connection,\n sessionSpan: Span,\n): TelemetryInfo {\n const ctx = trace.setSpan(context.active(), sessionSpan);\n const span = tracer.startSpan(\n `connection ${connection.id}`,\n {\n attributes: {\n component: 'river',\n 'river.connection.id': connection.id,\n },\n links: [{ context: sessionSpan.spanContext() }],\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createProcTelemetryInfo(\n kind: ValidProcType,\n serviceName: string,\n procedureName: string,\n streamId: string,\n): TelemetryInfo {\n const ctx = context.active();\n const span = tracer.startSpan(\n `procedure call ${serviceName}.${procedureName}`,\n {\n attributes: {\n component: 'river',\n 'river.method.kind': kind,\n 'river.method.service': serviceName,\n 'river.method.name': procedureName,\n 'river.streamId': streamId,\n 'span.kind': 'client',\n },\n kind: SpanKind.CLIENT,\n },\n ctx,\n );\n\n return { span, ctx };\n}\n\nexport function createHandlerSpan(\n kind: ValidProcType,\n message: OpaqueTransportMessage,\n fn: (span: Span) => Promise<unknown>,\n) {\n const ctx = message.tracing\n ? propagation.extract(context.active(), message.tracing)\n : context.active();\n\n return tracer.startActiveSpan(\n `procedure handler ${message.serviceName}.${message.procedureName}`,\n {\n attributes: {\n component: 'river',\n 'river.method.kind': kind,\n 'river.method.service': message.serviceName,\n 'river.method.name': message.procedureName,\n 'river.streamId': message.streamId,\n 'span.kind': 'server',\n },\n kind: SpanKind.SERVER,\n },\n ctx,\n fn,\n );\n}\n\nconst tracer = trace.getTracer('river', RIVER_VERSION);\nexport default tracer;\n","{\n \"name\": \"@replit/river\",\n \"description\": \"It's like tRPC but... with JSON Schema Support, duplex streaming and support for service multiplexing. Transport agnostic!\",\n \"version\": \"0.23.2\",\n \"type\": \"module\",\n \"exports\": {\n \".\": {\n \"import\": \"./dist/router/index.js\",\n \"require\": \"./dist/router/index.cjs\"\n },\n \"./logging\": {\n \"import\": \"./dist/logging/index.js\",\n \"require\": \"./dist/logging/index.cjs\"\n },\n \"./codec\": {\n \"import\": \"./dist/codec/index.js\",\n \"require\": \"./dist/codec/index.cjs\"\n },\n \"./transport\": {\n \"import\": \"./dist/transport/index.js\",\n \"require\": \"./dist/transport/index.cjs\"\n },\n \"./transport/ws/client\": {\n \"import\": \"./dist/transport/impls/ws/client.js\",\n \"require\": \"./dist/transport/impls/ws/client.cjs\"\n },\n \"./transport/ws/server\": {\n \"import\": \"./dist/transport/impls/ws/server.js\",\n \"require\": \"./dist/transport/impls/ws/server.cjs\"\n },\n \"./transport/uds/client\": {\n \"import\": \"./dist/transport/impls/uds/client.js\",\n \"require\": \"./dist/transport/impls/uds/client.cjs\"\n },\n \"./transport/uds/server\": {\n \"import\": \"./dist/transport/impls/uds/server.js\",\n \"require\": \"./dist/transport/impls/uds/server.cjs\"\n },\n \"./test-util\": {\n \"import\": \"./dist/util/testHelpers.js\",\n \"require\": \"./dist/util/testHelpers.cjs\"\n }\n },\n \"sideEffects\": [\n \"./dist/logging/index.js\"\n ],\n \"files\": [\n \"dist\"\n ],\n \"dependencies\": {\n \"@msgpack/msgpack\": \"^3.0.0-beta2\",\n \"it-pushable\": \"^3.2.3\",\n \"nanoid\": \"^4.0.2\",\n \"ws\": \"^8.17.0\"\n },\n \"peerDependencies\": {\n \"@opentelemetry/api\": \"^1.7.0\",\n \"@sinclair/typebox\": \"~0.32.8\"\n },\n \"devDependencies\": {\n \"@types/ws\": \"^8.5.5\",\n \"@typescript-eslint/eslint-plugin\": \"^7.8.0\",\n \"@typescript-eslint/parser\": \"^7.8.0\",\n \"@vitest/ui\": \"^1.3.1\",\n \"eslint\": \"^8.57.0\",\n \"eslint-config-prettier\": \"^9.1.0\",\n \"eslint-plugin-prettier\": \"^5.1.3\",\n \"prettier\": \"^3.0.0\",\n \"tsup\": \"^7.2.0\",\n \"typescript\": \"^5.4.5\",\n \"vitest\": \"^1.3.1\",\n \"vscode-langservers-extracted\": \"4.8.0\"\n },\n \"scripts\": {\n \"check\": \"tsc --noEmit && npm run format && npm run lint\",\n \"format\": \"npx prettier . --check\",\n \"format:fix\": \"npx prettier . --write\",\n \"lint\": \"eslint .\",\n \"lint:fix\": \"eslint . --fix\",\n \"fix\": \"npm run format:fix && npm run lint:fix\",\n \"build\": \"rm -rf dist && tsup && du -sh dist\",\n \"prepack\": \"npm run build\",\n \"release\": \"npm publish --access public\",\n \"test:ui\": \"echo \\\"remember to go to /__vitest__ in the webview\\\" && vitest --ui --api.host 0.0.0.0 --api.port 3000\",\n \"test\": \"vitest --test-timeout=500\",\n \"test:single\": \"vitest run --test-timeout=500 --reporter=dot\",\n \"test:flake\": \"./flake.sh\",\n \"bench\": \"vitest bench\"\n },\n \"engines\": {\n \"node\": \">=16\"\n },\n \"keywords\": [\n \"rpc\",\n \"websockets\",\n \"jsonschema\"\n ],\n \"author\": \"Jacky Zhao\",\n \"license\": \"MIT\"\n}\n","export function coerceErrorString(err: unknown): string {\n if (err instanceof Error) {\n return err.message || 'unknown reason';\n }\n\n return `[coerced to error] ${String(err)}`;\n}\n","import { Codec } from './types';\n\nconst encoder = new TextEncoder();\nconst decoder = new TextDecoder();\n\n// Convert Uint8Array to base64\nfunction uint8ArrayToBase64(uint8Array: Uint8Array) {\n let binary = '';\n uint8Array.forEach((byte) => {\n binary += String.fromCharCode(byte);\n });\n return btoa(binary);\n}\n\n// Convert base64 to Uint8Array\nfunction base64ToUint8Array(base64: string) {\n const binaryString = atob(base64);\n const uint8Array = new Uint8Array(binaryString.length);\n for (let i = 0; i < binaryString.length; i++) {\n uint8Array[i] = binaryString.charCodeAt(i);\n }\n return uint8Array;\n}\n\ninterface Base64EncodedValue {\n $t: string;\n}\n\n/**\n * Naive JSON codec implementation using JSON.stringify and JSON.parse.\n * @type {Codec}\n */\nexport const NaiveJsonCodec: Codec = {\n toBuffer: (obj: object) => {\n return encoder.encode(\n JSON.stringify(obj, function replacer<\n T extends object,\n >(this: T, key: keyof T) {\n const val = this[key];\n if (val instanceof Uint8Array) {\n return { $t: uint8ArrayToBase64(val) } satisfies Base64EncodedValue;\n } else {\n return val;\n }\n }),\n );\n },\n fromBuffer: (buff: Uint8Array) => {\n try {\n const parsed = JSON.parse(\n decoder.decode(buff),\n function reviver(_key, val: unknown) {\n if ((val as Base64EncodedValue | undefined)?.$t) {\n return base64ToUint8Array((val as Base64EncodedValue).$t);\n } else {\n return val;\n }\n },\n ) as unknown;\n\n if (typeof parsed === 'object') return parsed;\n return null;\n } catch {\n return null;\n }\n },\n};\n","import { Connection } from '../../session';\nimport { WsLike } from './wslike';\n\nexport class WebSocketConnection extends Connection {\n errorCb: null | ((err: Error) => void) = null;\n closeCb: null | (() => void) = null;\n\n ws: WsLike;\n\n constructor(ws: WsLike) {\n super();\n this.ws = ws;\n this.ws.binaryType = 'arraybuffer';\n\n // Websockets are kinda shitty, they emit error events with no\n // information other than it errored, so we have to do some extra\n // work to figure out what happened.\n let didError = false;\n this.ws.onerror = () => {\n didError = true;\n };\n this.ws.onclose = ({ code, reason }) => {\n if (didError && this.errorCb) {\n this.errorCb(\n new Error(\n `websocket closed with code and reason: ${code} - ${reason}`,\n ),\n );\n\n return;\n }\n\n if (this.closeCb) {\n this.closeCb();\n }\n };\n }\n\n addDataListener(cb: (msg: Uint8Array) => void) {\n this.ws.onmessage = (msg) => cb(msg.data as Uint8Array);\n }\n\n removeDataListener(): void {\n this.ws.onmessage = null;\n }\n\n addCloseListener(cb: () => void): void {\n this.closeCb = cb;\n }\n\n addErrorListener(cb: (err: Error) => void): void {\n this.errorCb = cb;\n }\n\n send(payload: Uint8Array) {\n if (this.ws.readyState === this.ws.OPEN) {\n this.ws.send(payload);\n return true;\n } else {\n return false;\n }\n }\n\n close() {\n this.ws.close();\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,mBAAsB;;;ACDtB,qBAAsC;AACtC,oBAAuB;AA4DhB,IAAM,yBAAyB,CAAoB,MACxD,oBAAK,OAAO;AAAA,EACV,IAAI,oBAAK,OAAO;AAAA,EAChB,MAAM,oBAAK,OAAO;AAAA,EAClB,IAAI,oBAAK,OAAO;AAAA,EAChB,KAAK,oBAAK,QAAQ;AAAA,EAClB,KAAK,oBAAK,QAAQ;AAAA,EAClB,aAAa,oBAAK,SAAS,oBAAK,OAAO,CAAC;AAAA,EACxC,eAAe,oBAAK,SAAS,oBAAK,OAAO,CAAC;AAAA,EAC1C,UAAU,oBAAK,OAAO;AAAA,EACtB,cAAc,oBAAK,QAAQ;AAAA,EAC3B,SAAS,oBAAK;AAAA,IACZ,oBAAK,OAAO;AAAA,MACV,aAAa,oBAAK,OAAO;AAAA,MACzB,YAAY,oBAAK,OAAO;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EACA,SAAS;AACX,CAAC;AAOI,IAAM,0BAA0B,oBAAK,OAAO;AAAA,EACjD,MAAM,oBAAK,QAAQ,KAAK;AAC1B,CAAC;AAMM,IAAM,4BAA4B,oBAAK,OAAO;AAAA,EACnD,MAAM,oBAAK,QAAQ,OAAO;AAC5B,CAAC;AAEM,IAAM,mBAAmB;AACzB,IAAM,uCAAuC,oBAAK,OAAO;AAAA,EAC9D,MAAM,oBAAK,QAAQ,eAAe;AAAA,EAClC,iBAAiB,oBAAK,OAAO;AAAA,EAC7B,WAAW,oBAAK,OAAO;AAAA,EACvB,UAAU,oBAAK,SAAS,oBAAK,QAAQ,CAAC;AACxC,CAAC;AAEM,IAAM,wCAAwC,oBAAK,OAAO;AAAA,EAC/D,MAAM,oBAAK,QAAQ,gBAAgB;AAAA,EACnC,QAAQ,oBAAK,MAAM;AAAA,IACjB,oBAAK,OAAO;AAAA,MACV,IAAI,oBAAK,QAAQ,IAAI;AAAA,MACrB,WAAW,oBAAK,OAAO;AAAA,IACzB,CAAC;AAAA,IACD,oBAAK,OAAO;AAAA,MACV,IAAI,oBAAK,QAAQ,KAAK;AAAA,MACtB,QAAQ,oBAAK,OAAO;AAAA,IACtB,CAAC;AAAA,EACH,CAAC;AACH,CAAC;AAEM,IAAM,8BAA8B,oBAAK,MAAM;AAAA,EACpD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,+BAA+B;AAAA,EAC1C,oBAAK,QAAQ;AACf;AA6DO,SAAS,yBACd,MACA,IACA,QACwE;AACxE,SAAO;AAAA,IACL,QAAI,sBAAO;AAAA,IACX;AAAA,IACA;AAAA,IACA,KAAK;AAAA,IACL,KAAK;AAAA,IACL,cAAU,sBAAO;AAAA,IACjB,cAAc;AAAA,IACd,SAAS;AAAA,MACP,MAAM;AAAA,MACN;AAAA,IACF;AAAA,EACF;AACF;AAcO,SAAS,MAAM,aAA8B;AAElD,UAAQ,cAAc,oBAAyB;AACjD;;;AC7HO,IAAI,MAA0B;;;ACrG9B,IAAM,gBAAgB;AAAA,EAC3B,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,iBAAiB;AAAA,EACjB,yBAAyB;AAC3B;AA0BO,IAAM,kBAAN,MAA4C;AAAA,EACzC,iBAAsD,CAAC;AAAA,EAE/D,kBAA+B,WAAc;AAC3C,WAAO,KAAK,eAAe,SAAS,GAAG,QAAQ;AAAA,EACjD;AAAA,EAEA,iBAA8B,WAAc,SAA0B;AACpE,QAAI,CAAC,KAAK,eAAe,SAAS,GAAG;AACnC,WAAK,eAAe,SAAS,IAAI,oBAAI,IAAI;AAAA,IAC3C;AAEA,SAAK,eAAe,SAAS,GAAG,IAAI,OAAO;AAAA,EAC7C;AAAA,EAEA,oBAAiC,WAAc,SAA0B;AACvE,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,WAAK,eAAe,SAAS,GAAG,OAAO,OAAO;AAAA,IAChD;AAAA,EACF;AAAA,EAEA,cAA2B,WAAc,OAAoB;AAC3D,UAAM,WAAW,KAAK,eAAe,SAAS;AAC9C,QAAI,UAAU;AACZ,iBAAW,WAAW,UAAU;AAC9B,gBAAQ,KAAK;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF;;;ACjEA,IAAAA,iBAA+B;;;ACA/B,iBAOO;;;ACJL,cAAW;;;AD8BN,SAAS,2BACd,SACA,gBACe;AACf,QAAM,MAAM,iBACR,uBAAY,QAAQ,mBAAQ,OAAO,GAAG,cAAc,IACpD,mBAAQ,OAAO;AAEnB,QAAM,OAAO,OAAO;AAAA,IAClB,WAAW,QAAQ,EAAE;AAAA,IACrB;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,oBAAoB,QAAQ;AAAA,QAC5B,oBAAoB,QAAQ;AAAA,QAC5B,sBAAsB,QAAQ;AAAA,MAChC;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAEO,SAAS,8BACd,YACA,aACe;AACf,QAAM,MAAM,iBAAM,QAAQ,mBAAQ,OAAO,GAAG,WAAW;AACvD,QAAM,OAAO,OAAO;AAAA,IAClB,cAAc,WAAW,EAAE;AAAA,IAC3B;AAAA,MACE,YAAY;AAAA,QACV,WAAW;AAAA,QACX,uBAAuB,WAAW;AAAA,MACpC;AAAA,MACA,OAAO,CAAC,EAAE,SAAS,YAAY,YAAY,EAAE,CAAC;AAAA,IAChD;AAAA,IACA;AAAA,EACF;AAEA,SAAO,EAAE,MAAM,IAAI;AACrB;AAuDA,IAAM,SAAS,iBAAM,UAAU,SAAS,OAAa;;;ADjHrD,IAAAC,cAA+B;AAE/B,IAAMC,cAAS,+BAAe,uCAAuC,CAAC;AAC/D,IAAM,WAAW,MAAMA,QAAO;AAU9B,IAAe,aAAf,MAA0B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA,cAAc;AACZ,SAAK,KAAK,QAAQA,QAAO,EAAE,CAAC;AAAA,EAC9B;AAwCF;AA6BO,IAAM,UAAN,MAA2C;AAAA,EACxC;AAAA,EACA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKQ,aAA4C,CAAC;AAAA;AAAA;AAAA;AAAA,EAKrD;AAAA,EACS;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA;AAAA;AAAA;AAAA;AAAA,EAKQ,MAAsB;AAAA;AAAA;AAAA;AAAA,EAKtB,MAAsB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMtB;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAER,YACE,MACA,MACA,IACA,SACA,gBACA;AACA,SAAK,KAAK,WAAWA,QAAO,EAAE,CAAC;AAC/B,SAAK,UAAU;AACf,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,aAAa;AAClB,SAAK,QAAQ,QAAQ;AAGrB,SAAK,kBAAkB;AACvB,SAAK,YAAY;AAAA,MACf,MAAM,KAAK,cAAc;AAAA,MACzB,QAAQ;AAAA,IACV;AACA,SAAK,YAAY,2BAA2B,MAAM,cAAc;AAAA,EAClE;AAAA,EAEA,IAAI,kBAAsD;AACxD,WAAO;AAAA,MACL,UAAU,KAAK;AAAA,MACf,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,QAAQ,KAAK,YAAY;AAAA,IAC3B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,KAAK,KAAsC;AACzC,UAAM,UAA4B,KAAK,aAAa,GAAG;AACvD,SAAK,MAAM,eAAe;AAAA,MACxB,GAAG,KAAK;AAAA,MACR,kBAAkB;AAAA,IACpB,CAAC;AAED,QAAI,KAAK,YAAY;AACnB,YAAM,KAAK,KAAK,WAAW,KAAK,KAAK,MAAM,SAAS,OAAO,CAAC;AAC5D,UAAI;AAAI,eAAO,QAAQ;AACvB,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC;AAAA,UACE,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,QACpB;AAAA,MACF;AAAA,IACF,OAAO;AACL,WAAK;AAAA,QACH,yBAAyB,QAAQ,EAAE;AAAA,QACnC,EAAE,GAAG,KAAK,iBAAiB,kBAAkB,QAAQ;AAAA,MACvD;AAAA,IACF;AAEA,WAAO,QAAQ;AAAA,EACjB;AAAA,EAEA,gBAAgB;AACd,UAAM,SAAS,KAAK;AACpB,UAAM,eAAe,SAAS,KAAK,QAAQ;AAC3C,QAAI,SAAS,KAAK,QAAQ,qBAAqB;AAC7C,UAAI,KAAK,YAAY;AACnB,aAAK;AAAA,UACH,yBAAyB,KAAK,EAAE,8BAA8B,MAAM,wBAAwB,YAAY;AAAA,UACxG,KAAK;AAAA,QACP;AACA,aAAK,UAAU,KAAK,SAAS,sCAAsC;AACnE,aAAK,qBAAqB;AAAA,MAC5B;AACA;AAAA,IACF;AAEA,SAAK,KAAK;AAAA,MACR,UAAU;AAAA,MACV;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AACD,SAAK;AAAA,EACP;AAAA,EAEA,wBAAwB;AACtB,SAAK,aAAa,CAAC;AACnB,SAAK,MAAM;AACX,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,qBAAqB,MAAgB;AACnC,SAAK,KAAK,aAAa,KAAK,WAAW,MAAM,sBAAsB;AAAA,MACjE,GAAG,KAAK;AAAA,MACR,QAAQ,KAAK;AAAA,IACf,CAAC;AACD,eAAW,OAAO,KAAK,YAAY;AACjC,WAAK,MAAM,iBAAiB;AAAA,QAC1B,GAAG,KAAK;AAAA,QACR,kBAAkB;AAAA,QAClB,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,YAAM,KAAK,KAAK,KAAK,KAAK,MAAM,SAAS,GAAG,CAAC;AAC7C,UAAI,CAAC,IAAI;AAGP,cAAM,SAAS,sCAAsC,KAAK,EAAE;AAC5D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AAED,aAAK,MAAM,QAAQ;AAAA,UACjB,GAAG,KAAK;AAAA,UACR,kBAAkB;AAAA,UAClB,QAAQ,KAAK;AAAA,UACb,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM;AACX;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAa,KAAa;AAC1C,QAAI,MAAM,IAAI,KAAK,KAAK;AACtB,WAAK,MAAM,sBAAsB,GAAG,UAAU,KAAK,GAAG,IAAI;AAAA,QACxD,GAAG,KAAK;AAAA,QACR,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAEA,SAAK,aAAa,KAAK,WAAW,OAAO,CAAC,YAAY,QAAQ,OAAO,GAAG;AACxE,SAAK,MAAM,MAAM;AAAA,EACnB;AAAA,EAEA,qBAAqB,MAAiB;AACpC,QAAI,KAAK,eAAe,UAAa,KAAK,eAAe;AAAM;AAC/D,SAAK;AAAA,MACH,gDAAgD,KAAK,EAAE;AAAA,MACvD,KAAK;AAAA,IACP;AACA,SAAK,WAAW,MAAM;AACtB,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,yBAAyB,SAAmB;AAC1C,SAAK,qBAAqB,OAAO;AACjC,SAAK,YAAY;AACjB,SAAK,qBAAqB,OAAO;AACjC,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,WAAW,IAAgB;AACzB,SAAK;AAAA,MACH,YAAY,KAAK,QAAQ,wBAAwB,oCAAoC,KAAK,EAAE;AAAA,MAC5F,KAAK;AAAA,IACP;AACA,SAAK,qBAAqB,WAAW,MAAM;AACzC,WAAK,MAAM;AACX,SAAG;AAAA,IACL,GAAG,KAAK,QAAQ,wBAAwB;AAAA,EAC1C;AAAA;AAAA,EAGA,cAAc;AACZ,SAAK,kBAAkB;AACvB,iBAAa,KAAK,kBAAkB;AACpC,SAAK,qBAAqB;AAAA,EAC5B;AAAA;AAAA;AAAA,EAIA,QAAQ;AACN,SAAK,qBAAqB;AAC1B,SAAK,YAAY;AACjB,SAAK,sBAAsB;AAC3B,kBAAc,KAAK,SAAS;AAAA,EAC9B;AAAA,EAEA,IAAI,YAAY;AACd,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,IAAI,kBAAkB;AACpB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,aACE,YAC2B;AAC3B,UAAM,MAAM;AAAA,MACV,GAAG;AAAA,MACH,IAAI,SAAS;AAAA,MACb,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,IACZ;AAEA,SAAK;AACL,SAAK,WAAW,KAAK,GAAG;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,oBAA2D;AACzD,WAAO,KAAK;AAAA,EACd;AACF;;;AGtXO,SAAS,kBAAkB,KAAsB;AACtD,MAAI,eAAe,OAAO;AACxB,WAAO,IAAI,WAAW;AAAA,EACxB;AAEA,SAAO,sBAAsB,OAAO,GAAG,CAAC;AAC1C;;;ACJA,IAAM,UAAU,IAAI,YAAY;AAChC,IAAM,UAAU,IAAI,YAAY;AAGhC,SAAS,mBAAmB,YAAwB;AAClD,MAAI,SAAS;AACb,aAAW,QAAQ,CAAC,SAAS;AAC3B,cAAU,OAAO,aAAa,IAAI;AAAA,EACpC,CAAC;AACD,SAAO,KAAK,MAAM;AACpB;AAGA,SAAS,mBAAmB,QAAgB;AAC1C,QAAM,eAAe,KAAK,MAAM;AAChC,QAAM,aAAa,IAAI,WAAW,aAAa,MAAM;AACrD,WAAS,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;AAC5C,eAAW,CAAC,IAAI,aAAa,WAAW,CAAC;AAAA,EAC3C;AACA,SAAO;AACT;AAUO,IAAM,iBAAwB;AAAA,EACnC,UAAU,CAAC,QAAgB;AACzB,WAAO,QAAQ;AAAA,MACb,KAAK,UAAU,KAAK,SAAS,SAElB,KAAc;AACvB,cAAM,MAAM,KAAK,GAAG;AACpB,YAAI,eAAe,YAAY;AAC7B,iBAAO,EAAE,IAAI,mBAAmB,GAAG,EAAE;AAAA,QACvC,OAAO;AACL,iBAAO;AAAA,QACT;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EACA,YAAY,CAAC,SAAqB;AAChC,QAAI;AACF,YAAM,SAAS,KAAK;AAAA,QAClB,QAAQ,OAAO,IAAI;AAAA,QACnB,SAAS,QAAQ,MAAM,KAAc;AACnC,cAAK,KAAwC,IAAI;AAC/C,mBAAO,mBAAoB,IAA2B,EAAE;AAAA,UAC1D,OAAO;AACL,mBAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF;AAEA,UAAI,OAAO,WAAW;AAAU,eAAO;AACvC,aAAO;AAAA,IACT,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;;;AR9BA,IAAAC,cAA+B;AAexB,IAAM,0BAA4C;AAAA,EACvD,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,OAAO;AACT;AAMA,IAAM,gCAAwD;AAAA,EAC5D,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,cAAc;AAAA,EACd,uBAAuB;AAAA,EACvB,yBAAyB;AAC3B;AAEA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AAAA,EACH,GAAG;AACL;AAMA,IAAM,gCAAwD;AAAA,EAC5D,GAAG;AACL;AAgCO,IAAe,YAAf,MAAsD;AAAA;AAAA;AAAA;AAAA;AAAA,EAK3D;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,cAAc;AAChB,WAAO,IAAI;AAAA,MACT,CAAC,GAAG,KAAK,QAAQ,EACd,IAAI,CAAC,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,QAAQ,UAAU,CAAC,EACvD,OAAO,CAAC,UAAuC,MAAM,CAAC,MAAM,MAAS;AAAA,IAC1E;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA;AAAA;AAAA;AAAA,EAKU;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQV,YACE,UACA,iBACA;AACA,SAAK,UAAU,EAAE,GAAG,yBAAyB,GAAG,gBAAgB;AAChE,SAAK,kBAAkB,IAAI,gBAAgB;AAC3C,SAAK,WAAW,oBAAI,IAAI;AACxB,SAAK,QAAQ,KAAK,QAAQ;AAC1B,SAAK,WAAW;AAChB,SAAK,QAAQ;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAoBU,UACR,MACA,aACA,SACA,aACA;AACA,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,SAAK,YAAY;AAAA,MACf;AAAA,MACA,QAAQ,UAAU;AAAA,IACpB;AAEA,QAAI,aAAa;AACf,cAAQ,yBAAyB,IAAI;AACrC,WAAK,KAAK,kBAAkB,WAAW,IAAI,QAAQ,eAAe;AAAA,IACpE;AAAA,EACF;AAAA,EAEU,cACR,IACA,MACA,gBACA;AACA,UAAM,UAAU,IAAI;AAAA,MAClB;AAAA,MACA,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACF;AACA,SAAK,SAAS,IAAI,QAAQ,IAAI,OAAO;AACrC,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEU,mBACR,IACA,MACA,WACA,gBACA;AACA,QAAI,UAAU,KAAK,SAAS,IAAI,EAAE;AAClC,QAAI,cAAc,YAAY;AAE9B,QACE,SAAS,wBAAwB,UACjC,cAAc,UACd,QAAQ,wBAAwB,WAChC;AACA,WAAK;AAAA,QACH,eAAe,EAAE,6DAA6D,QAAQ,mBAAmB,UAAU,SAAS;AAAA,QAC5H,QAAQ;AAAA,MACV;AACA,WAAK,cAAc,OAAO;AAC1B,oBAAc;AACd,gBAAU;AAAA,IACZ;AAEA,QAAI,CAAC,SAAS;AACZ,gBAAU,KAAK,cAAc,IAAI,MAAM,cAAc;AACrD,WAAK;AAAA,QACH,kBAAkB,EAAE;AAAA,QACpB,QAAQ;AAAA,MACV;AAAA,IACF;AAEA,QAAI,cAAc,QAAW;AAC3B,cAAQ,sBAAsB;AAAA,IAChC;AAEA,WAAO,EAAE,SAAS,YAAY;AAAA,EAChC;AAAA,EAEU,cAAc,SAA4B;AAClD,YAAQ,MAAM;AACd,YAAQ,UAAU,KAAK,IAAI;AAC3B,SAAK,SAAS,OAAO,QAAQ,EAAE;AAC/B,SAAK;AAAA,MACH,WAAW,QAAQ,EAAE,oBAAoB,QAAQ,EAAE;AAAA,MACnD,QAAQ;AAAA,IACV;AACA,SAAK,gBAAgB,cAAc,iBAAiB;AAAA,MAClD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,aAAa,MAAgB,SAA4B;AACjE,SAAK,WAAW,KAAK,IAAI;AACzB,SAAK,gBAAgB,cAAc,oBAAoB;AAAA,MACrD,QAAQ;AAAA,MACR;AAAA,IACF,CAAC;AAED,YAAQ,aAAa;AACrB,YAAQ,WAAW,MAAM;AACvB,cAAQ,UAAU,KAAK,SAAS,8BAA8B;AAC9D,WAAK,cAAc,OAAO;AAAA,IAC5B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,SAAS,KAAgD;AACjE,UAAM,YAAY,KAAK,MAAM,WAAW,GAAG;AAE3C,QAAI,cAAc,MAAM;AACtB,YAAM,gBAAgB,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK,GAAG,CAAC;AAC/D,WAAK,MAAM,yCAAyC,aAAa,IAAI;AAAA,QACnE,UAAU,KAAK;AAAA,MACjB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,mBAAM,MAAM,8BAA8B,SAAS,GAAG;AACzD,WAAK,MAAM,yBAAyB,KAAK,UAAU,SAAS,CAAC,IAAI;AAAA,QAC/D,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,UAChB,GAAG,mBAAM,OAAO,8BAA8B,SAAS;AAAA,QACzD;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAU,KAA6B;AAC/C,QAAI,KAAK,UAAU;AAAQ;AAC3B,UAAM,UAAU,KAAK,SAAS,IAAI,IAAI,IAAI;AAC1C,QAAI,CAAC,SAAS;AACZ,WAAK,MAAM,2BAA2B,IAAI,IAAI,IAAI;AAAA,QAChD,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD;AAAA,IACF;AAGA,YAAQ,YAAY;AAEpB,SAAK,MAAM,gBAAgB;AAAA,MACzB,UAAU,KAAK;AAAA,MACf,kBAAkB;AAAA,IACpB,CAAC;AACD,QAAI,IAAI,QAAQ,QAAQ,iBAAiB;AACvC,UAAI,IAAI,MAAM,QAAQ,iBAAiB;AACrC,aAAK;AAAA,UACH,oCAAoC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AAAA,UACnF,EAAE,UAAU,KAAK,UAAU,kBAAkB,IAAI;AAAA,QACnD;AAAA,MACF,OAAO;AACL,cAAM,SAAS,uCAAuC,IAAI,GAAG,iBAAiB,QAAQ,eAAe;AACrG,aAAK,MAAM,GAAG,MAAM,gCAAgC;AAAA,UAClD,UAAU,KAAK;AAAA,UACf,kBAAkB;AAAA,UAClB,MAAM,CAAC,qBAAqB;AAAA,QAC9B,CAAC;AACD,aAAK,cAAc,cAAc,yBAAyB,MAAM;AAChE,gBAAQ,UAAU,KAAK,UAAU;AAAA,UAC/B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,gBAAQ,MAAM;AAAA,MAChB;AAEA;AAAA,IACF;AAEA,YAAQ,kBAAkB,IAAI,KAAK,IAAI,GAAG;AAG1C,QAAI,CAAC,MAAM,IAAI,YAAY,GAAG;AAC5B,WAAK,gBAAgB,cAAc,WAAW,GAAG;AAAA,IACnD,OAAO;AACL,WAAK,MAAM,gCAAgC;AAAA,QACzC,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,MACpB,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBACE,MACA,SACM;AACN,SAAK,gBAAgB,iBAAiB,MAAM,OAAO;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,oBACE,MACA,SACM;AACN,SAAK,gBAAgB,oBAAoB,MAAM,OAAO;AAAA,EACxD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,KACE,IACA,KACoB;AACpB,QAAI,KAAK,UAAU,aAAa;AAC9B,YAAM,MAAM;AACZ,WAAK,MAAM,KAAK;AAAA,QACd,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,QAClB,MAAM,CAAC,qBAAqB;AAAA,MAC9B,CAAC;AACD,WAAK,cAAc,cAAc,iBAAiB,GAAG;AACrD,aAAO;AAAA,IACT,WAAW,KAAK,UAAU,UAAU;AAClC,WAAK,KAAK,6CAA6C;AAAA,QACrD,UAAU,KAAK;AAAA,QACf,kBAAkB;AAAA,MACpB,CAAC;AACD,aAAO;AAAA,IACT;AAEA,WAAO,KAAK,mBAAmB,EAAE,EAAE,QAAQ,KAAK,GAAG;AAAA,EACrD;AAAA;AAAA,EAGA,gBAAgB,IAAuB,UAAkB;AACvD,WAAO,KAAK,KAAK,IAAI;AAAA,MACnB;AAAA,MACA;AAAA,MACA,SAAS;AAAA,QACP,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEU,cAAc,MAAyB,SAAiB;AAChE,SAAK,gBAAgB,cAAc,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAAA,EACvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ;AACN,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,6BAA6B,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAU;AACR,SAAK,QAAQ;AACb,eAAW,WAAW,KAAK,SAAS,OAAO,GAAG;AAC5C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,KAAK,gCAAgC,EAAE,UAAU,KAAK,SAAS,CAAC;AAAA,EACvE;AACF;AAmXO,IAAe,kBAAf,cAGG,UAAoB;AAAA;AAAA;AAAA;AAAA,EAIlB;AAAA;AAAA;AAAA;AAAA,EAKV;AAAA;AAAA;AAAA;AAAA,EAKA;AAAA,EAEA,YACE,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,UAAU;AAAA,MACb,GAAG;AAAA,MACH,GAAG;AAAA,IACL;AACA,SAAK,2BAA2B,oBAAI,QAAQ;AAC5C,SAAK,KAAK,8BAA8B;AAAA,MACtC,UAAU,KAAK;AAAA,MACf,iBAAiB;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEA,gBAAgB,SAAuD;AACrE,SAAK,sBAAsB;AAAA,EAC7B;AAAA,EAEU,iBAAiB,MAAgB;AACzC,QAAI,KAAK,UAAU;AAAQ;AAE3B,SAAK,KAAK,2BAA2B;AAAA,MACnC,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,QAAI,UAAyC;AAC7C,UAAM,SAAS,MAAM,SAAS,MAAM;AAGpC,UAAM,mBAAmB,WAAW,MAAM;AACxC,UAAI,CAAC,SAAS;AACZ,aAAK;AAAA,UACH,iBAAiB,OAAO,CAAC;AAAA,UACzB;AAAA,YACE,UAAU,KAAK;AAAA,YACf,aAAa,OAAO;AAAA,YACpB,QAAQ,KAAK;AAAA,UACf;AAAA,QACF;AACA,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,aAAK,MAAM;AAAA,MACb;AAAA,IACF,GAAG,KAAK,QAAQ,wBAAwB;AAExC,UAAM,SAA4B,CAAC;AACnC,QAAI,2BAA2B;AAE/B,UAAM,mBAAmB,CAAC,SAAqB;AAE7C,UAAI,0BAA0B;AAC5B,eAAO,KAAK,IAAI;AAChB;AAAA,MACF;AAEA,iCAA2B;AAC3B,mBAAa,gBAAgB;AAE7B,WAAK,KAAK,+BAA+B,MAAM,IAAI,EAAE;AAAA,QACnD,CAAC,iBAAiB;AAChB,cAAI,CAAC,cAAc;AACjB,iBAAK,MAAM;AACX;AAAA,UACF;AAEA,oBAAU;AAIV,gBAAM,cAAc,CAACC,UAAqB;AACxC,kBAAM,SAAS,KAAK,SAASA,KAAI;AACjC,gBAAI,CAAC,QAAQ;AACX,mBAAK,MAAM;AACX;AAAA,YACF;AAEA,iBAAK,UAAU,MAAM;AAAA,UACvB;AAGA,qBAAWA,SAAQ,QAAQ;AACzB,wBAAYA,KAAI;AAAA,UAClB;AAEA,eAAK,mBAAmB,gBAAgB;AACxC,eAAK,gBAAgB,WAAW;AAChC,iBAAO,SAAS;AAAA,QAClB;AAAA,MACF;AAAA,IACF;AAEA,SAAK,gBAAgB,gBAAgB;AACrC,SAAK,iBAAiB,MAAM;AAC1B,UAAI,CAAC;AAAS;AACd,WAAK,KAAK,iBAAiB,OAAO,CAAC,iBAAiB;AAAA,QAClD,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA,MACf,CAAC;AACD,WAAK,aAAa,MAAM,OAAO;AAAA,IACjC,CAAC;AAED,SAAK,iBAAiB,CAAC,QAAQ;AAC7B,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,UAAI,CAAC;AAAS;AACd,WAAK;AAAA,QACH,iBAAiB,OAAO,CAAC,kBAAkB,kBAAkB,GAAG,CAAC;AAAA,QACjE,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,0BACZ,MACA,SACA,aAGA,MACiC;AACjC,QAAI,iBAAiC,CAAC;AACtC,QAAI,KAAK,qBAAqB;AAE5B,UAAI,CAAC,mBAAM,MAAM,KAAK,oBAAoB,QAAQ,WAAW,GAAG;AAC9D,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,SAAS;AACf,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAE1C,aAAK,KAAK,8CAA8C,IAAI,IAAI;AAAA,UAC9D,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,UACb,kBAAkB;AAAA,YAChB,GAAG,mBAAM,OAAO,KAAK,oBAAoB,QAAQ,WAAW;AAAA,UAC9D;AAAA,QACF,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAEA,YAAM,yBAAyB,UAC3B,KAAK,yBAAyB,IAAI,OAAO,IACzC;AAEJ,uBAAiB,MAAM,KAAK,oBAAoB;AAAA,QAC9C;AAAA,QACA;AAAA,MACF;AAGA,UAAI,mBAAmB,OAAO;AAC5B,cAAM,SAAS;AACf,aAAK,WAAW,KAAK,UAAU;AAAA,UAC7B,MAAM,2BAAe;AAAA,UACrB,SAAS;AAAA,QACX,CAAC;AACD,cAAM,cAAc,yBAAyB,KAAK,UAAU,MAAM;AAAA,UAChE,IAAI;AAAA,UACJ;AAAA,QACF,CAAC;AACD,aAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,aAAK,KAAK,2BAA2B,IAAI,IAAI;AAAA,UAC3C,UAAU,KAAK;AAAA,UACf,QAAQ,KAAK;AAAA,QACf,CAAC;AACD,aAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,+BACJ,MACA,MACoC;AACpC,UAAM,SAAS,KAAK,SAAS,IAAI;AACjC,QAAI,CAAC,QAAQ;AACX,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,mBAAM,MAAM,sCAAsC,OAAO,OAAO,GAAG;AACtE,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AACD,YAAM,SAAS;AACf,YAAMC,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK,KAAK,QAAQ;AAAA,QAChB,UAAU,KAAK;AAAA,QACf,QAAQ,KAAK;AAAA;AAAA;AAAA,QAGb,kBAAkB;AAAA,QAClB,kBAAkB;AAAA,UAChB,GAAG,mBAAM,OAAO,sCAAsC,OAAO,OAAO;AAAA,QACtE;AAAA,MACF,CAAC;AACD,WAAK;AAAA,QACH,cAAc;AAAA,QACd;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAGA,UAAM,aAAa,OAAO,QAAQ;AAClC,QAAI,eAAe,kBAAkB;AACnC,WAAK,WAAW,KAAK,UAAU;AAAA,QAC7B,MAAM,2BAAe;AAAA,QACrB,SAAS;AAAA,MACX,CAAC;AAED,YAAM,SAAS,2BAA2B,UAAU,WAAW,gBAAgB;AAC/E,YAAMA,eAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,QACvE,IAAI;AAAA,QACJ;AAAA,MACF,CAAC;AACD,WAAK,KAAK,KAAK,MAAM,SAASA,YAAW,CAAC;AAC1C,WAAK;AAAA,QACH,mEAAmE,UAAU,eAAe,gBAAgB;AAAA,QAC5G,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,MAC7C;AACA,WAAK,cAAc,cAAc,iBAAiB,MAAM;AACxD,aAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,SAAS,IAAI,OAAO,IAAI;AAChD,UAAM,iBAAiB,MAAM,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,QAAI,mBAAmB,OAAO;AAC5B,aAAO;AAAA,IACT;AAEA,UAAM,EAAE,SAAS,YAAY,IAAI,KAAK;AAAA,MACpC,OAAO;AAAA,MACP;AAAA,MACA,OAAO,QAAQ;AAAA,MACf,OAAO;AAAA,IACT;AAEA,SAAK,yBAAyB,IAAI,SAAS,cAAc;AAEzD,SAAK;AAAA,MACH,kBAAkB,OAAO,IAAI;AAAA,MAC7B,EAAE,UAAU,KAAK,UAAU,QAAQ,KAAK,GAAG;AAAA,IAC7C;AACA,UAAM,cAAc,yBAAyB,KAAK,UAAU,OAAO,MAAM;AAAA,MACvE,IAAI;AAAA,MACJ,WAAW,QAAQ;AAAA,IACrB,CAAC;AACD,SAAK,KAAK,KAAK,MAAM,SAAS,WAAW,CAAC;AAC1C,SAAK,UAAU,MAAM,OAAO,MAAM,SAAS,WAAW;AAEtD,WAAO;AAAA,EACT;AACF;;;AS7oCO,IAAM,sBAAN,cAAkC,WAAW;AAAA,EAClD,UAAyC;AAAA,EACzC,UAA+B;AAAA,EAE/B;AAAA,EAEA,YAAY,IAAY;AACtB,UAAM;AACN,SAAK,KAAK;AACV,SAAK,GAAG,aAAa;AAKrB,QAAI,WAAW;AACf,SAAK,GAAG,UAAU,MAAM;AACtB,iBAAW;AAAA,IACb;AACA,SAAK,GAAG,UAAU,CAAC,EAAE,MAAM,OAAO,MAAM;AACtC,UAAI,YAAY,KAAK,SAAS;AAC5B,aAAK;AAAA,UACH,IAAI;AAAA,YACF,0CAA0C,IAAI,MAAM,MAAM;AAAA,UAC5D;AAAA,QACF;AAEA;AAAA,MACF;AAEA,UAAI,KAAK,SAAS;AAChB,aAAK,QAAQ;AAAA,MACf;AAAA,IACF;AAAA,EACF;AAAA,EAEA,gBAAgB,IAA+B;AAC7C,SAAK,GAAG,YAAY,CAAC,QAAQ,GAAG,IAAI,IAAkB;AAAA,EACxD;AAAA,EAEA,qBAA2B;AACzB,SAAK,GAAG,YAAY;AAAA,EACtB;AAAA,EAEA,iBAAiB,IAAsB;AACrC,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,iBAAiB,IAAgC;AAC/C,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,KAAK,SAAqB;AACxB,QAAI,KAAK,GAAG,eAAe,KAAK,GAAG,MAAM;AACvC,WAAK,GAAG,KAAK,OAAO;AACpB,aAAO;AAAA,IACT,OAAO;AACL,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,QAAQ;AACN,SAAK,GAAG,MAAM;AAAA,EAChB;AACF;;;AVzDO,IAAM,2BAAN,cAAuC,gBAAqC;AAAA,EACjF;AAAA,EAEA,YACE,KACA,UACA,iBACA;AACA,UAAM,UAAU,eAAe;AAC/B,SAAK,MAAM;AACX,SAAK,IAAI,GAAG,cAAc,KAAK,iBAAiB;AAAA,EAClD;AAAA,EAEA,oBAAoB,CAAC,OAAe;AAClC,UAAM,OAAO,IAAI,oBAAoB,EAAE;AACvC,SAAK,iBAAiB,IAAI;AAAA,EAC5B;AAAA,EAEA,QAAQ;AACN,UAAM,MAAM;AACZ,SAAK,IAAI,IAAI,cAAc,KAAK,iBAAiB;AAAA,EACnD;AACF;","names":["import_nanoid","import_api","nanoid","import_api","data","responseMsg"]}