agents 0.23.0 → 0.25.0

package/dist/websockets-DUfRHPRq.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"websockets-DUfRHPRq.js","names":[],"sources":["../src/websockets/connection.ts","../src/websockets/callables-target.ts","../src/websockets/protocol.ts","../src/websockets/websockets.ts"],"sourcesContent":["// Polyfill WebSocket status code constants for environments that don't have them\n// in order to support libraries that expect standards-compatible WebSocket\n// implementations (e.g. PartySocket)\n\nimport type {\n  Connection,\n  ConnectionSetStateFn,\n  ConnectionState,\n  LifecycleSockets\n} from \"../lifecycle\";\n\nif (!(\"OPEN\" in WebSocket)) {\n  const WebSocketStatus = {\n    // @ts-expect-error\n    CONNECTING: WebSocket.READY_STATE_CONNECTING,\n    // @ts-expect-error\n    OPEN: WebSocket.READY_STATE_OPEN,\n    // @ts-expect-error\n    CLOSING: WebSocket.READY_STATE_CLOSING,\n    // @ts-expect-error\n    CLOSED: WebSocket.READY_STATE_CLOSED\n  };\n\n  Object.assign(WebSocket, WebSocketStatus);\n  // @ts-expect-error\n  Object.assign(WebSocket.prototype, WebSocketStatus);\n}\n\n/**\n * Store both platform attachments and user attachments in different namespaces\n */\ntype ConnectionAttachments = {\n  __pk: {\n    id: string;\n    tags: string[];\n    uri?: string;\n  };\n  __user?: unknown;\n};\n\nfunction tryGetManagedWebSocketMeta(\n  ws: WebSocket\n): ConnectionAttachments[\"__pk\"] | null {\n  try {\n    // Avoid AttachmentCache.get() here: externally accepted sockets\n    // can have an attachment without the managed __pk namespace.\n    const attachment = WebSocket.prototype.deserializeAttachment.call(\n      ws\n    ) as unknown;\n    if (!attachment || typeof attachment !== \"object\") {\n      return null;\n    }\n    if (!(\"__pk\" in attachment)) {\n      return null;\n    }\n    const pk = (attachment as ConnectionAttachments).__pk as unknown;\n    if (!pk || typeof pk !== \"object\") {\n      return null;\n    }\n    const { id, tags } = pk as {\n      id?: unknown;\n      tags?: unknown;\n    };\n    if (typeof id !== \"string\") {\n      return null;\n    }\n    const { uri } = pk as { uri?: unknown };\n    return {\n      id,\n      tags: Array.isArray(tags) ? tags : [],\n      uri: typeof uri === \"string\" ? uri : undefined\n    } satisfies ConnectionAttachments[\"__pk\"];\n  } catch {\n    return null;\n  }\n}\n\nexport function isManagedWebSocket(ws: WebSocket): boolean {\n  return tryGetManagedWebSocketMeta(ws) !== null;\n}\n\n/**\n * Cache websocket attachments to avoid having to rehydrate them on every property access.\n */\nclass AttachmentCache {\n  #cache = new WeakMap<WebSocket, ConnectionAttachments>();\n\n  get(ws: WebSocket): ConnectionAttachments {\n    let attachment = this.#cache.get(ws);\n    if (!attachment) {\n      attachment = WebSocket.prototype.deserializeAttachment.call(\n        ws\n      ) as ConnectionAttachments;\n      if (attachment !== undefined) {\n        this.#cache.set(ws, attachment);\n      } else {\n        throw new Error(\"Missing managed WebSocket lifecycle attachment\");\n      }\n    }\n\n    return attachment;\n  }\n\n  set(ws: WebSocket, attachment: ConnectionAttachments) {\n    this.#cache.set(ws, attachment);\n    WebSocket.prototype.serializeAttachment.call(ws, attachment);\n  }\n}\n\nconst attachments = new AttachmentCache();\nconst connections = new WeakSet<Connection>();\nconst isWrapped = (ws: WebSocket): ws is Connection => {\n  return connections.has(ws as Connection);\n};\n\n/**\n * Wraps a WebSocket with Connection fields that rehydrate the\n * socket attachments lazily only when requested.\n */\nexport const createConnection = (ws: WebSocket | Connection): Connection => {\n  if (isWrapped(ws)) {\n    return ws;\n  }\n\n  // if state was set on the socket before initializing the connection,\n  // capture it here so we can persist it again\n  let initialState;\n  if (\"state\" in ws) {\n    initialState = ws.state;\n    delete ws.state;\n  }\n\n  const connection = Object.defineProperties(ws, {\n    id: {\n      configurable: true,\n      get() {\n        return attachments.get(ws).__pk.id;\n      }\n    },\n    uri: {\n      configurable: true,\n      get() {\n        return attachments.get(ws).__pk.uri ?? null;\n      }\n    },\n    tags: {\n      configurable: true,\n      get() {\n        // Default to [] for connections accepted before tags were stored\n        return attachments.get(ws).__pk.tags ?? [];\n      }\n    },\n    state: {\n      configurable: true,\n      get() {\n        const attachment = attachments.get(ws);\n        return (attachment.__user ?? null) as ConnectionState<unknown>;\n      }\n    },\n    setState: {\n      configurable: true,\n      value: function setState<T>(setState: T | ConnectionSetStateFn<T>) {\n        const state =\n          setState instanceof Function\n            ? setState((this as Connection<T>).state)\n            : setState;\n        attachments.set(ws, {\n          ...attachments.get(ws),\n          __user: state ?? null\n        });\n        return state as ConnectionState<T>;\n      }\n    }\n  }) as Connection;\n\n  if (initialState) {\n    connection.setState(initialState);\n  }\n\n  connections.add(connection);\n  return connection;\n};\n\nclass ConnectionIterator<T> implements IterableIterator<Connection<T>> {\n  private index = 0;\n  #sockets: WebSocket[] | undefined;\n  constructor(\n    private sockets: LifecycleSockets,\n    private tag?: string\n  ) {}\n\n  [Symbol.iterator](): IterableIterator<Connection<T>> {\n    return this;\n  }\n\n  next(): IteratorResult<Connection<T>, number | undefined> {\n    const sockets =\n      this.#sockets ?? (this.#sockets = this.sockets.get(this.tag));\n\n    let socket: WebSocket;\n    while ((socket = sockets[this.index++])) {\n      // Only yield open sockets.\n      if (socket.readyState === WebSocket.OPEN) {\n        // Durable Objects hibernation APIs allow storing arbitrary sockets via\n        // `state.acceptWebSocket()`. Ignore sockets without our attachment.\n        if (!isManagedWebSocket(socket)) {\n          continue;\n        }\n        const value = createConnection(socket) as Connection<T>;\n        return { done: false, value };\n      }\n    }\n\n    // reached the end of the iteratee\n    return { done: true, value: undefined };\n  }\n}\n\n/**\n * Deduplicate and validate connection tags.\n * Returns the final tag array (always includes the connection id as the first tag).\n */\nfunction prepareTags(connectionId: string, userTags: string[]): string[] {\n  const tags = [connectionId, ...userTags.filter((t) => t !== connectionId)];\n\n  // validate tags against documented restrictions\n  // https://developers.cloudflare.com/durable-objects/api/hibernatable-websockets-api/#state-methods-for-websockets\n  if (tags.length > 10) {\n    throw new Error(\n      \"A connection can only have 10 tags, including the default id tag.\"\n    );\n  }\n\n  for (const tag of tags) {\n    if (typeof tag !== \"string\") {\n      throw new Error(`A connection tag must be a string. Received: ${tag}`);\n    }\n    if (tag === \"\") {\n      throw new Error(\"A connection tag must not be an empty string.\");\n    }\n    if (tag.length > 256) {\n      throw new Error(\"A connection tag must not exceed 256 characters\");\n    }\n  }\n\n  return tags;\n}\n\n/** The platform-backed manager for hibernating WebSockets. */\nexport class ConnectionManager<TState = unknown> {\n  constructor(private controller: LifecycleSockets) {}\n\n  getConnection<T = TState>(id: string) {\n    // TODO: Should we cache the connections?\n    const sockets = this.controller.get(id);\n    const matching = sockets.filter((ws) => {\n      return tryGetManagedWebSocketMeta(ws)?.id === id;\n    });\n\n    if (matching.length === 0) return undefined;\n    if (matching.length === 1)\n      return createConnection(matching[0]) as Connection<T>;\n\n    throw new Error(\n      `More than one connection found for id ${id}. Did you mean to use getConnections(tag) instead?`\n    );\n  }\n\n  getConnections<T = TState>(tag?: string | undefined) {\n    return new ConnectionIterator<T>(this.controller, tag);\n  }\n\n  accept(connection: Connection, options: { tags: string[] }) {\n    const tags = prepareTags(connection.id, options.tags);\n\n    this.controller.accept(connection, tags);\n    attachments.set(connection, {\n      __pk: {\n        id: connection.id,\n        tags,\n        uri: connection.uri ?? undefined\n      },\n      __user: null\n    });\n\n    return createConnection(connection);\n  }\n}\n","import { RpcTarget } from \"cloudflare:workers\";\nimport { decoratedMethods } from \"../callable-decorator\";\n\n/** A named remote method ready to be exposed on a callables root. */\nexport type CallableInvoker = (...args: unknown[]) => unknown;\n\n/**\n * The single exposure policy for callable names. `Object.prototype` and\n * `RpcTarget.prototype` members are unreachable over Cap'n Web anyway\n * and are silently excluded; `then` is rejected loudly because exposing\n * it would make the remote stub thenable.\n */\nfunction assertExposable(name: string): boolean {\n  if (name === \"then\") {\n    throw new Error(\n      'A callables target cannot expose a method named \"then\" — it would make the remote stub thenable'\n    );\n  }\n  return !(\n    name === \"constructor\" ||\n    name in Object.prototype ||\n    name in RpcTarget.prototype\n  );\n}\n\n/**\n * The exposable prototype methods of a callables target, each bound to\n * invoke on the real instance (so private fields and `this` behave).\n *\n * Cap'n Web resolves methods on the prototype chain and rejects own\n * instance properties, so only prototype methods participate. The\n * nearest declaration wins for overridden names.\n *\n * @param target - The callables target to enumerate.\n * @returns Method names mapped to invokers on the target.\n */\nexport function exposableMethods(\n  target: RpcTarget\n): ReadonlyMap<string, CallableInvoker> {\n  const methods = new Map<string, CallableInvoker>();\n  const seen = new Set<string>();\n  let prototype: object | null = Object.getPrototypeOf(target);\n  while (\n    prototype &&\n    prototype !== RpcTarget.prototype &&\n    prototype !== Object.prototype\n  ) {\n    for (const name of Object.getOwnPropertyNames(prototype)) {\n      if (seen.has(name)) continue;\n      seen.add(name);\n      if (!assertExposable(name)) continue;\n      const descriptor = Object.getOwnPropertyDescriptor(prototype, name);\n      if (!descriptor || typeof descriptor.value !== \"function\") continue;\n      // SAFETY: `typeof descriptor.value === \"function\"` was checked\n      // above; TypeScript cannot narrow a descriptor's `value` field.\n      const method = descriptor.value as CallableInvoker;\n      methods.set(name, (...args) => Reflect.apply(method, target, args));\n    }\n    prototype = Object.getPrototypeOf(prototype);\n  }\n  return methods;\n}\n\n/**\n * Build a Cap'n Web session root exposing exactly the given methods.\n *\n * Cap'n Web resolves methods on the prototype chain, rejects own\n * instance properties, and breaks on Proxy-wrapped roots — so the root\n * is a private `RpcTarget` subclass whose prototype carries the\n * methods and nothing else.\n *\n * @param methods - Method names mapped to their invokers.\n * @returns A root suitable as a Cap'n Web session's local main.\n */\nexport function buildCallablesRoot(\n  methods: ReadonlyMap<string, CallableInvoker>\n): RpcTarget {\n  class CallablesRoot extends RpcTarget {}\n  for (const [name, invoke] of methods) {\n    Object.defineProperty(CallablesRoot.prototype, name, {\n      value: invoke,\n      writable: true,\n      configurable: true,\n      enumerable: false\n    });\n  }\n  return new CallablesRoot();\n}\n\n/**\n * Build a callables target from a host's `@callable()`-decorated\n * methods — the fallback interface source when no `RpcTarget` is\n * configured (an explicit target is preferred and wins).\n *\n * Methods are resolved on the host at call time, so framework wrapping\n * applied after construction (e.g. Agent's context auto-wrapping) is\n * honored. Methods registered with `streaming: true` expect the legacy\n * Agent RPC protocol's injected `StreamingResponse` and are not exposed\n * here — return a `ReadableStream` from an `RpcTarget` method instead.\n *\n * @param host - The object whose decorated methods form the interface.\n * @returns A target exposing the decorated methods, or `undefined`\n * when the host has none.\n */\nexport function callablesFromDecorated(host: object): RpcTarget | undefined {\n  const methods = new Map<string, CallableInvoker>();\n  for (const [name, metadata] of decoratedMethods(host)) {\n    if (metadata.streaming || !assertExposable(name)) continue;\n    methods.set(name, (...args) => {\n      const method = Reflect.get(host, name) as unknown;\n      if (typeof method !== \"function\") {\n        throw new Error(`Method ${name} is not callable`);\n      }\n      return Reflect.apply(method, host, args);\n    });\n  }\n  if (methods.size === 0) return undefined;\n  return buildCallablesRoot(methods);\n}\n","/**\n * Isomorphic pieces of the callables RPC wire contract.\n *\n * Imported from browser bundles (via the client) and from the Worker\n * runtime, so it must not import `cloudflare:workers`.\n */\n\n/** Query parameter selecting the callables Cap'n Web RPC endpoint. */\nexport const CALLABLES_RPC_QUERY = \"__agents_rpc\";\nexport const CALLABLES_RPC_VALUE = \"capnweb\";\n\n/** Whether a request is a WebSocket upgrade addressed to callables RPC. */\nexport function isCallablesRpcUpgrade(request: Request): boolean {\n  return (\n    request.headers.get(\"Upgrade\")?.toLowerCase() === \"websocket\" &&\n    new URL(request.url).searchParams.get(CALLABLES_RPC_QUERY) ===\n      CALLABLES_RPC_VALUE\n  );\n}\n\n/**\n * Derive the callables RPC socket URL from a host route. Accepts http(s)\n * or ws(s) URLs and returns a ws(s) URL with the RPC query parameter set.\n */\nexport function callablesRpcUrl(url: string | URL): string {\n  const resolved = new URL(url);\n  if (resolved.protocol === \"http:\") resolved.protocol = \"ws:\";\n  if (resolved.protocol === \"https:\") resolved.protocol = \"wss:\";\n  resolved.searchParams.set(CALLABLES_RPC_QUERY, CALLABLES_RPC_VALUE);\n  return resolved.toString();\n}\n","import { RpcTarget } from \"cloudflare:workers\";\nimport { newWorkersWebSocketRpcResponse } from \"capnweb\";\nimport { nanoid } from \"nanoid\";\nimport {\n  LifecycleCapability,\n  type CapabilityWebSocketUpgradeContext,\n  type Connection,\n  type ConnectionSetStateFn,\n  type ConnectionState\n} from \"../lifecycle\";\nimport {\n  ConnectionManager,\n  createConnection,\n  isManagedWebSocket\n} from \"./connection\";\nimport {\n  buildCallablesRoot,\n  exposableMethods,\n  type CallableInvoker\n} from \"./callables-target\";\nimport type {\n  WebSocketHandlers,\n  WebSocketMessage,\n  WebSocketsOptions\n} from \"./options\";\nimport { isCallablesRpcUpgrade } from \"./protocol\";\n\n/**\n * Reserved close codes the runtime synthesizes when there was no real\n * Close frame from the peer (1005 NoStatusReceived, 1006 AbnormalClosure,\n * 1015 TLSHandshake). They cannot appear in an outgoing Close frame, and\n * there is no peer left to receive a reciprocation.\n */\nfunction isReservedCloseCode(code: number): boolean {\n  return code === 1005 || code === 1006 || code === 1015;\n}\n\n/**\n * Reciprocate a peer-initiated Close frame to complete the handshake, as\n * the Hibernation API contract requires. Best-effort: swallows errors\n * from already-closed sockets or invalid codes/reasons, and skips\n * reciprocation entirely for reserved codes (dead transport).\n */\nfunction reciprocateClose(ws: WebSocket, code: number, reason: string): void {\n  if (isReservedCloseCode(code)) return;\n  try {\n    ws.close(code, reason);\n  } catch {\n    // Already closed, oversize reason, or another unrecoverable\n    // invariant — the handshake is either done or out of our control.\n  }\n}\n\n/**\n * Opt-in WebSocket support for Lifecycle Objects.\n *\n * Lifecycle itself does not model WebSockets — hosts that want them\n * install this capability, which owns the connection subsystem end to\n * end: it claims upgrades, accepts hibernating sockets, dispatches\n * `onConnect`/`onMessage`/`onClose` inside the host invocation\n * boundary, reciprocates close handshakes, and answers\n * `getConnections()`/`getConnection()`.\n *\n * ```ts\n * class Room extends DurableObject<Env> {\n *   readonly webSockets = new WebSockets({\n *     handlers: {\n *       onConnect: (connection) => connection.send(\"welcome\"),\n *       onMessage: (connection, message) => { ... },\n *       onClose: (connection, code) => { ... }\n *     },\n *     callables: new RoomCallables()\n *   });\n *   readonly lifecycle = Lifecycle.install(this).use(this.webSockets);\n * }\n * ```\n *\n * `callables` exposes an `RpcTarget`'s prototype methods to remote\n * callers over a Cap'n Web session claimed from `?__agents_rpc=capnweb`\n * upgrades. Methods run through the host invocation boundary, may\n * return a `ReadableStream` to stream results, and emit\n * `rpc`/`rpc:error` capability events. Callable sessions are\n * non-hibernating: while a client holds one open, the Durable Object\n * stays pinned in memory.\n *\n * There is no separate browser client: against an `Agent`, the\n * `useAgent` hook's `stub`/`call` reach the same interface over the\n * protocol socket. A plain host's endpoint is reached with capnweb\n * directly — `newWebSocketRpcSession(new WebSocket(callablesRpcUrl(url)))`.\n *\n * @experimental The API surface may change before stabilizing.\n */\nexport class WebSockets extends LifecycleCapability {\n  readonly #handlers: WebSocketHandlers | undefined;\n  readonly #getConnectionTags: WebSocketsOptions[\"getConnectionTags\"];\n  readonly #callablesTarget: RpcTarget | undefined;\n  #manager: ConnectionManager | undefined;\n\n  constructor(options: WebSocketsOptions = {}) {\n    super(\"websockets\");\n    this.#handlers = options.handlers;\n    this.#getConnectionTags = options.getConnectionTags;\n    this.#callablesTarget = options.callables\n      ? this.#buildCallablesTarget(options.callables)\n      : undefined;\n  }\n\n  // ── Lifecycle capability hooks ─────────────────────────────────────────\n\n  /** Claim callables RPC upgrades and, with handlers, plain upgrades. */\n  onWebSocketUpgrade({\n    request\n  }: CapabilityWebSocketUpgradeContext):\n    | Promise<Response>\n    | Response\n    | undefined {\n    if (isCallablesRpcUpgrade(request)) {\n      if (!this.#callablesTarget) return undefined;\n      return newWorkersWebSocketRpcResponse(request, this.#callablesTarget);\n    }\n    if (!this.#handlers) return undefined;\n    return this.#acceptConnection(request);\n  }\n\n  /** Dispatch a platform message wake for a capability-owned socket. */\n  async onWebSocketMessage(\n    ws: WebSocket,\n    message: WebSocketMessage\n  ): Promise<boolean> {\n    if (!isManagedWebSocket(ws)) return false;\n    const connection = createConnection(ws);\n    await this.lifecycle.runInHostContext(\n      () => this.#handlers?.onMessage?.(connection, message),\n      { connection }\n    );\n    return true;\n  }\n\n  /** Dispatch and reciprocate a close wake for an owned socket. */\n  async onWebSocketClose(\n    ws: WebSocket,\n    code: number,\n    reason: string,\n    wasClean: boolean\n  ): Promise<boolean> {\n    if (!isManagedWebSocket(ws)) return false;\n    const connection = createConnection(ws);\n    try {\n      await this.lifecycle.runInHostContext(\n        () => this.#handlers?.onClose?.(connection, code, reason, wasClean),\n        { connection }\n      );\n    } finally {\n      reciprocateClose(ws, code, reason);\n    }\n    return true;\n  }\n\n  /** Dispatch an error wake for an owned socket. */\n  async onWebSocketError(ws: WebSocket, error: unknown): Promise<boolean> {\n    if (!isManagedWebSocket(ws)) return false;\n    const connection = createConnection(ws);\n    await this.lifecycle.runInHostContext(\n      () => this.#handlers?.onError?.(connection, error),\n      { connection }\n    );\n    return true;\n  }\n\n  /**\n   * Close every owned connection during explicit host destruction. The\n   * capability owns its sockets' lifetimes, so it also owns tearing\n   * them down.\n   */\n  dispose(): void {\n    for (const connection of this.getConnections()) {\n      try {\n        connection.close(1001, \"Durable Object destroyed\");\n      } catch {\n        // Already closed or mid-handshake — nothing left to tear down.\n      }\n    }\n  }\n\n  // ── Connections ────────────────────────────────────────────────────────\n\n  /** Open connections accepted by this capability, optionally by tag. */\n  getConnections<TState = unknown>(\n    tag?: string\n  ): IterableIterator<Connection<TState>> {\n    return this.#connectionManager.getConnections<TState>(tag);\n  }\n\n  /** One connection accepted by this capability, by id. */\n  getConnection<TState = unknown>(id: string): Connection<TState> | undefined {\n    return this.#connectionManager.getConnection<TState>(id);\n  }\n\n  get #connectionManager(): ConnectionManager {\n    this.#manager ??= new ConnectionManager(this.lifecycle.sockets);\n    return this.#manager;\n  }\n\n  async #acceptConnection(request: Request): Promise<Response> {\n    const { 0: clientWebSocket, 1: serverWebSocket } = new WebSocketPair();\n    const url = new URL(request.url);\n    // `||`, not `??`: an empty `?_pk=` value must fall back to a\n    // generated id — an empty connection id would later throw in tag\n    // validation and reject the upgrade.\n    const connectionId = url.searchParams.get(\"_pk\") || nanoid();\n\n    let connection: Connection = Object.assign(serverWebSocket, {\n      id: connectionId,\n      uri: request.url,\n      tags: [] as string[],\n      state: null as unknown as ConnectionState<unknown>,\n      setState<T = unknown>(setState: T | ConnectionSetStateFn<T>) {\n        // Pre-accept shim: hold state on the socket until accept()\n        // persists it into the hibernation attachment.\n        const state =\n          setState instanceof Function\n            ? setState(this.state as ConnectionState<T>)\n            : setState;\n        this.state = state as ConnectionState<T>;\n        return this.state as ConnectionState<T>;\n      }\n    });\n\n    const ctx = { request };\n    const tags = this.#getConnectionTags\n      ? await this.#getConnectionTags(connection, ctx)\n      : [];\n\n    // Hibernating WebSockets remain connected while the object is evicted.\n    connection = this.#connectionManager.accept(connection, { tags });\n    await this.lifecycle.runInHostContext(\n      () => this.#handlers?.onConnect?.(connection, ctx),\n      { connection, request }\n    );\n\n    return new Response(null, { status: 101, webSocket: clientWebSocket });\n  }\n\n  // ── Callables ──────────────────────────────────────────────────────────\n\n  /**\n   * Wrap a callables target for serving: every exposable method\n   * dispatches through the host invocation boundary and emits\n   * `rpc`/`rpc:error` events.\n   */\n  #buildCallablesTarget(target: RpcTarget): RpcTarget {\n    const dispatching = new Map<string, CallableInvoker>();\n    for (const [name, invoke] of exposableMethods(target)) {\n      dispatching.set(name, (...args) =>\n        this.#dispatchCallable(name, () => invoke(...args))\n      );\n    }\n    return buildCallablesRoot(dispatching);\n  }\n\n  async #dispatchCallable(\n    name: string,\n    invoke: () => unknown\n  ): Promise<unknown> {\n    // Throws with installation guidance when the capability was never\n    // installed with Lifecycle.use().\n    const services = this.lifecycle;\n    try {\n      const result = await services.runInHostContext(invoke);\n      services.events.emit(\"rpc\", {\n        method: name,\n        streaming: result instanceof ReadableStream\n      });\n      return result;\n    } catch (error) {\n      services.events.emit(\"rpc:error\", {\n        method: name,\n        error: error instanceof Error ? error.message : String(error)\n      });\n      throw error;\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AAWA,IAAI,EAAE,UAAU,YAAY;CAC1B,MAAM,kBAAkB;EAEtB,YAAY,UAAU;EAEtB,MAAM,UAAU;EAEhB,SAAS,UAAU;EAEnB,QAAQ,UAAU;CACpB;CAEA,OAAO,OAAO,WAAW,eAAe;CAExC,OAAO,OAAO,UAAU,WAAW,eAAe;AACpD;AAcA,SAAS,2BACP,IACsC;CACtC,IAAI;EAGF,MAAM,aAAa,UAAU,UAAU,sBAAsB,KAC3D,EACF;EACA,IAAI,CAAC,cAAc,OAAO,eAAe,UACvC,OAAO;EAET,IAAI,EAAE,UAAU,aACd,OAAO;EAET,MAAM,KAAM,WAAqC;EACjD,IAAI,CAAC,MAAM,OAAO,OAAO,UACvB,OAAO;EAET,MAAM,EAAE,IAAI,SAAS;EAIrB,IAAI,OAAO,OAAO,UAChB,OAAO;EAET,MAAM,EAAE,QAAQ;EAChB,OAAO;GACL;GACA,MAAM,MAAM,QAAQ,IAAI,IAAI,OAAO,CAAC;GACpC,KAAK,OAAO,QAAQ,WAAW,MAAM,KAAA;EACvC;CACF,QAAQ;EACN,OAAO;CACT;AACF;AAEA,SAAgB,mBAAmB,IAAwB;CACzD,OAAO,2BAA2B,EAAE,MAAM;AAC5C;;;;;AAKA,IAAM,kBAAN,MAAsB;;2DACX,IAAI,QAA0C,CAAA;;CAEvD,IAAI,IAAsC;EACxC,IAAI,aAAA,uBAAA,QAAa,IAAA,CAAA,CAAY,IAAI,EAAE;EACnC,IAAI,CAAC,YAAY;GACf,aAAa,UAAU,UAAU,sBAAsB,KACrD,EACF;GACA,IAAI,eAAe,KAAA,GACjB,uBAAA,QAAA,IAAA,CAAA,CAAY,IAAI,IAAI,UAAU;QAE9B,MAAM,IAAI,MAAM,gDAAgD;EAEpE;EAEA,OAAO;CACT;CAEA,IAAI,IAAe,YAAmC;EACpD,uBAAA,QAAA,IAAA,CAAA,CAAY,IAAI,IAAI,UAAU;EAC9B,UAAU,UAAU,oBAAoB,KAAK,IAAI,UAAU;CAC7D;AACF;AAEA,MAAM,cAAc,IAAI,gBAAgB;AACxC,MAAM,8BAAc,IAAI,QAAoB;AAC5C,MAAM,aAAa,OAAoC;CACrD,OAAO,YAAY,IAAI,EAAgB;AACzC;;;;;AAMA,MAAa,oBAAoB,OAA2C;CAC1E,IAAI,UAAU,EAAE,GACd,OAAO;CAKT,IAAI;CACJ,IAAI,WAAW,IAAI;EACjB,eAAe,GAAG;EAClB,OAAO,GAAG;CACZ;CAEA,MAAM,aAAa,OAAO,iBAAiB,IAAI;EAC7C,IAAI;GACF,cAAc;GACd,MAAM;IACJ,OAAO,YAAY,IAAI,EAAE,CAAC,CAAC,KAAK;GAClC;EACF;EACA,KAAK;GACH,cAAc;GACd,MAAM;IACJ,OAAO,YAAY,IAAI,EAAE,CAAC,CAAC,KAAK,OAAO;GACzC;EACF;EACA,MAAM;GACJ,cAAc;GACd,MAAM;IAEJ,OAAO,YAAY,IAAI,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC;GAC3C;EACF;EACA,OAAO;GACL,cAAc;GACd,MAAM;IAEJ,OADmB,YAAY,IAAI,EAClB,CAAC,CAAC,UAAU;GAC/B;EACF;EACA,UAAU;GACR,cAAc;GACd,OAAO,SAAS,SAAY,UAAuC;IACjE,MAAM,QACJ,oBAAoB,WAChB,SAAU,KAAuB,KAAK,IACtC;IACN,YAAY,IAAI,IAAI;KAClB,GAAG,YAAY,IAAI,EAAE;KACrB,QAAQ,SAAS;IACnB,CAAC;IACD,OAAO;GACT;EACF;CACF,CAAC;CAED,IAAI,cACF,WAAW,SAAS,YAAY;CAGlC,YAAY,IAAI,UAAU;CAC1B,OAAO;AACT;;mBAUG,OAAO;AARV,IAAM,qBAAN,MAAuE;CAGrE,YACE,SACA,KACA;EAFQ,KAAA,UAAA;EACA,KAAA,MAAA;EAJV,KAAQ,QAAQ;;CAKb;CAEH,CAAA,oBAAqD;EACnD,OAAO;CACT;CAEA,OAA0D;EACxD,MAAM,UAAA,uBAAA,UACJ,IAAA,KAAA,uBAAA,UAAkB,MAAgB,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAA;EAE7D,IAAI;EACJ,OAAQ,SAAS,QAAQ,KAAK,UAE5B,IAAI,OAAO,eAAe,UAAU,MAAM;GAGxC,IAAI,CAAC,mBAAmB,MAAM,GAC5B;GAGF,OAAO;IAAE,MAAM;IAAO,OADR,iBAAiB,MACL;GAAE;EAC9B;EAIF,OAAO;GAAE,MAAM;GAAM,OAAO,KAAA;EAAU;CACxC;AACF;;;;;AAMA,SAAS,YAAY,cAAsB,UAA8B;CACvE,MAAM,OAAO,CAAC,cAAc,GAAG,SAAS,QAAQ,MAAM,MAAM,YAAY,CAAC;CAIzE,IAAI,KAAK,SAAS,IAChB,MAAM,IAAI,MACR,mEACF;CAGF,KAAK,MAAM,OAAO,MAAM;EACtB,IAAI,OAAO,QAAQ,UACjB,MAAM,IAAI,MAAM,gDAAgD,KAAK;EAEvE,IAAI,QAAQ,IACV,MAAM,IAAI,MAAM,+CAA+C;EAEjE,IAAI,IAAI,SAAS,KACf,MAAM,IAAI,MAAM,iDAAiD;CAErE;CAEA,OAAO;AACT;;AAGA,IAAa,oBAAb,MAAiD;CAC/C,YAAY,YAAsC;EAA9B,KAAA,aAAA;CAA+B;CAEnD,cAA0B,IAAY;EAGpC,MAAM,WADU,KAAK,WAAW,IAAI,EACb,CAAC,CAAC,QAAQ,OAAO;GACtC,OAAO,2BAA2B,EAAE,CAAC,EAAE,OAAO;EAChD,CAAC;EAED,IAAI,SAAS,WAAW,GAAG,OAAO,KAAA;EAClC,IAAI,SAAS,WAAW,GACtB,OAAO,iBAAiB,SAAS,EAAE;EAErC,MAAM,IAAI,MACR,yCAAyC,GAAG,mDAC9C;CACF;CAEA,eAA2B,KAA0B;EACnD,OAAO,IAAI,mBAAsB,KAAK,YAAY,GAAG;CACvD;CAEA,OAAO,YAAwB,SAA6B;EAC1D,MAAM,OAAO,YAAY,WAAW,IAAI,QAAQ,IAAI;EAEpD,KAAK,WAAW,OAAO,YAAY,IAAI;EACvC,YAAY,IAAI,YAAY;GAC1B,MAAM;IACJ,IAAI,WAAW;IACf;IACA,KAAK,WAAW,OAAO,KAAA;GACzB;GACA,QAAQ;EACV,CAAC;EAED,OAAO,iBAAiB,UAAU;CACpC;AACF;;;;;;;;;ACnRA,SAAS,gBAAgB,MAAuB;CAC9C,IAAI,SAAS,QACX,MAAM,IAAI,MACR,mGACF;CAEF,OAAO,EACL,SAAS,iBACT,QAAQ,OAAO,aACf,QAAQ,UAAU;AAEtB;;;;;;;;;;;;AAaA,SAAgB,iBACd,QACsC;CACtC,MAAM,0BAAU,IAAI,IAA6B;CACjD,MAAM,uBAAO,IAAI,IAAY;CAC7B,IAAI,YAA2B,OAAO,eAAe,MAAM;CAC3D,OACE,aACA,cAAc,UAAU,aACxB,cAAc,OAAO,WACrB;EACA,KAAK,MAAM,QAAQ,OAAO,oBAAoB,SAAS,GAAG;GACxD,IAAI,KAAK,IAAI,IAAI,GAAG;GACpB,KAAK,IAAI,IAAI;GACb,IAAI,CAAC,gBAAgB,IAAI,GAAG;GAC5B,MAAM,aAAa,OAAO,yBAAyB,WAAW,IAAI;GAClE,IAAI,CAAC,cAAc,OAAO,WAAW,UAAU,YAAY;GAG3D,MAAM,SAAS,WAAW;GAC1B,QAAQ,IAAI,OAAO,GAAG,SAAS,QAAQ,MAAM,QAAQ,QAAQ,IAAI,CAAC;EACpE;EACA,YAAY,OAAO,eAAe,SAAS;CAC7C;CACA,OAAO;AACT;;;;;;;;;;;;AAaA,SAAgB,mBACd,SACW;CACX,MAAM,sBAAsB,UAAU,CAAC;CACvC,KAAK,MAAM,CAAC,MAAM,WAAW,SAC3B,OAAO,eAAe,cAAc,WAAW,MAAM;EACnD,OAAO;EACP,UAAU;EACV,cAAc;EACd,YAAY;CACd,CAAC;CAEH,OAAO,IAAI,cAAc;AAC3B;;;;;;;;;;;;;;;;AAiBA,SAAgB,uBAAuB,MAAqC;CAC1E,MAAM,0BAAU,IAAI,IAA6B;CACjD,KAAK,MAAM,CAAC,MAAM,aAAa,iBAAiB,IAAI,GAAG;EACrD,IAAI,SAAS,aAAa,CAAC,gBAAgB,IAAI,GAAG;EAClD,QAAQ,IAAI,OAAO,GAAG,SAAS;GAC7B,MAAM,SAAS,QAAQ,IAAI,MAAM,IAAI;GACrC,IAAI,OAAO,WAAW,YACpB,MAAM,IAAI,MAAM,UAAU,KAAK,iBAAiB;GAElD,OAAO,QAAQ,MAAM,QAAQ,MAAM,IAAI;EACzC,CAAC;CACH;CACA,IAAI,QAAQ,SAAS,GAAG,OAAO,KAAA;CAC/B,OAAO,mBAAmB,OAAO;AACnC;;;;;;;;;;AC9GA,MAAa,sBAAsB;AACnC,MAAa,sBAAsB;;AAGnC,SAAgB,sBAAsB,SAA2B;CAC/D,OACE,QAAQ,QAAQ,IAAI,SAAS,CAAC,EAAE,YAAY,MAAM,eAClD,IAAI,IAAI,QAAQ,GAAG,CAAC,CAAC,aAAa,IAAA,cAAuB,MAAA;AAG7D;;;;;AAMA,SAAgB,gBAAgB,KAA2B;CACzD,MAAM,WAAW,IAAI,IAAI,GAAG;CAC5B,IAAI,SAAS,aAAa,SAAS,SAAS,WAAW;CACvD,IAAI,SAAS,aAAa,UAAU,SAAS,WAAW;CACxD,SAAS,aAAa,IAAI,qBAAqB,mBAAmB;CAClE,OAAO,SAAS,SAAS;AAC3B;;;;;;;;;ACGA,SAAS,oBAAoB,MAAuB;CAClD,OAAO,SAAS,QAAQ,SAAS,QAAQ,SAAS;AACpD;;;;;;;AAQA,SAAS,iBAAiB,IAAe,MAAc,QAAsB;CAC3E,IAAI,oBAAoB,IAAI,GAAG;CAC/B,IAAI;EACF,GAAG,MAAM,MAAM,MAAM;CACvB,QAAQ,CAGR;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCA,IAAa,aAAb,cAAgC,oBAAoB;CAMlD,YAAY,UAA6B,CAAC,GAAG;EAC3C,MAAM,YAAY;;;;;;EAClB,uBAAA,WAAA,MAAiB,QAAQ,QAAA;EACzB,uBAAA,oBAAA,MAA0B,QAAQ,iBAAA;EAClC,uBAAA,kBAAA,MAAwB,QAAQ,YAAA,kBAAA,mBAC5B,MAAA,qBAAA,CAAA,CAAA,KAAA,MAA2B,QAAQ,SAAS,IAC5C,KAAA,CAAA;CACN;;CAKA,mBAAmB,EACjB,WAIY;EACZ,IAAI,sBAAsB,OAAO,GAAG;GAClC,IAAI,CAAA,uBAAA,kBAAC,IAAA,GAAuB,OAAO,KAAA;GACnC,OAAO,+BAA+B,SAAA,uBAAA,kBAAS,IAAA,CAAqB;EACtE;EACA,IAAI,CAAA,uBAAA,WAAC,IAAA,GAAgB,OAAO,KAAA;EAC5B,OAAA,kBAAA,mBAAO,MAAA,iBAAA,CAAA,CAAA,KAAA,MAAuB,OAAO;CACvC;;CAGA,MAAM,mBACJ,IACA,SACkB;EAClB,IAAI,CAAC,mBAAmB,EAAE,GAAG,OAAO;EACpC,MAAM,aAAa,iBAAiB,EAAE;EACtC,MAAM,KAAK,UAAU,uBAAA,uBAAA,WACb,IAAA,CAAA,EAAgB,YAAY,YAAY,OAAO,GACrD,EAAE,WAAW,CACf;EACA,OAAO;CACT;;CAGA,MAAM,iBACJ,IACA,MACA,QACA,UACkB;EAClB,IAAI,CAAC,mBAAmB,EAAE,GAAG,OAAO;EACpC,MAAM,aAAa,iBAAiB,EAAE;EACtC,IAAI;GACF,MAAM,KAAK,UAAU,uBAAA,uBAAA,WACb,IAAA,CAAA,EAAgB,UAAU,YAAY,MAAM,QAAQ,QAAQ,GAClE,EAAE,WAAW,CACf;EACF,UAAU;GACR,iBAAiB,IAAI,MAAM,MAAM;EACnC;EACA,OAAO;CACT;;CAGA,MAAM,iBAAiB,IAAe,OAAkC;EACtE,IAAI,CAAC,mBAAmB,EAAE,GAAG,OAAO;EACpC,MAAM,aAAa,iBAAiB,EAAE;EACtC,MAAM,KAAK,UAAU,uBAAA,uBAAA,WACb,IAAA,CAAA,EAAgB,UAAU,YAAY,KAAK,GACjD,EAAE,WAAW,CACf;EACA,OAAO;CACT;;;;;;CAOA,UAAgB;EACd,KAAK,MAAM,cAAc,KAAK,eAAe,GAC3C,IAAI;GACF,WAAW,MAAM,MAAM,0BAA0B;EACnD,QAAQ,CAER;CAEJ;;CAKA,eACE,KACsC;EACtC,OAAA,uBAAA,KAAA,kBAAA,mBAAO,IAAA,CAAsB,CAAC,CAAC,eAAuB,GAAG;CAC3D;;CAGA,cAAgC,IAA4C;EAC1E,OAAA,uBAAA,KAAA,kBAAA,mBAAO,IAAA,CAAsB,CAAC,CAAC,cAAsB,EAAE;CACzD;AAsFF;AApFE,SAAA,yBAA4C;CAC1C,uBAAA,UAAA,IAAA,KAAA,uBAAA,UAAA,MAAkB,IAAI,kBAAkB,KAAK,UAAU,OAAO,CAAA;CAC9D,OAAA,uBAAA,UAAO,IAAA;AACT;AAEA,eAAA,kBAAwB,SAAqC;CAC3D,MAAM,EAAE,GAAG,iBAAiB,GAAG,oBAAoB,IAAI,cAAc;CAKrE,MAAM,eAAe,IAJL,IAAI,QAAQ,GAIL,CAAC,CAAC,aAAa,IAAI,KAAK,KAAK,OAAO;CAE3D,IAAI,aAAyB,OAAO,OAAO,iBAAiB;EAC1D,IAAI;EACJ,KAAK,QAAQ;EACb,MAAM,CAAC;EACP,OAAO;EACP,SAAsB,UAAuC;GAG3D,MAAM,QACJ,oBAAoB,WAChB,SAAS,KAAK,KAA2B,IACzC;GACN,KAAK,QAAQ;GACb,OAAO,KAAK;EACd;CACF,CAAC;CAED,MAAM,MAAM,EAAE,QAAQ;CACtB,MAAM,OAAA,uBAAA,oBAAO,IAAA,IACT,MAAA,uBAAA,oBAAM,IAAA,CAAA,CAAA,KAAA,MAAwB,YAAY,GAAG,IAC7C,CAAC;CAGL,aAAA,uBAAA,KAAA,kBAAA,mBAAa,IAAA,CAAsB,CAAC,CAAC,OAAO,YAAY,EAAE,KAAK,CAAC;CAChE,MAAM,KAAK,UAAU,uBAAA,uBAAA,WACb,IAAA,CAAA,EAAgB,YAAY,YAAY,GAAG,GACjD;EAAE;EAAY;CAAQ,CACxB;CAEA,OAAO,IAAI,SAAS,MAAM;EAAE,QAAQ;EAAK,WAAW;CAAgB,CAAC;AACvE;;;;;;AASA,SAAA,sBAAsB,QAA8B;CAClD,MAAM,8BAAc,IAAI,IAA6B;CACrD,KAAK,MAAM,CAAC,MAAM,WAAW,iBAAiB,MAAM,GAClD,YAAY,IAAI,OAAO,GAAG,SAAA,kBAAA,mBACxB,MAAA,iBAAA,CAAA,CAAA,KAAA,MAAuB,YAAY,OAAO,GAAG,IAAI,CAAC,CACpD;CAEF,OAAO,mBAAmB,WAAW;AACvC;AAEA,eAAA,kBACE,MACA,QACkB;CAGlB,MAAM,WAAW,KAAK;CACtB,IAAI;EACF,MAAM,SAAS,MAAM,SAAS,iBAAiB,MAAM;EACrD,SAAS,OAAO,KAAK,OAAO;GAC1B,QAAQ;GACR,WAAW,kBAAkB;EAC/B,CAAC;EACD,OAAO;CACT,SAAS,OAAO;EACd,SAAS,OAAO,KAAK,aAAa;GAChC,QAAQ;GACR,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;EAC9D,CAAC;EACD,MAAM;CACR;AACF"}

package/dist/ws-chat-transport-rWwta645.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ws-chat-transport-rWwta645.js","names":[],"sources":["../src/chat/replay-batch.ts","../src/chat/ws-chat-transport.ts"],"sourcesContent":["import type { UIMessageChunk } from \"ai\";\n\n/**\n * Batches chunks replayed during stream resume.\n *\n * Replaying every chunk separately can exceed React's update limit. Adjacent\n * deltas for the same message part are merged until the replay pauses or ends.\n */\nexport class ReplayChunkBatch {\n  private chunks: UIMessageChunk[] = [];\n  private tailMergeKey: string | null = null;\n  private cancelWindow: (() => void) | undefined;\n  private startMessageId: string | undefined;\n\n  constructor(\n    private readonly controller: ReadableStreamDefaultController<UIMessageChunk>,\n    private readonly closeWindow: (flush: () => void) => () => void = (\n      flush\n    ) => {\n      const timer = setTimeout(flush, 0);\n      return () => clearTimeout(timer);\n    }\n  ) {}\n\n  get bufferedStartMessageId(): string | undefined {\n    return this.startMessageId;\n  }\n\n  get isEmpty(): boolean {\n    return this.chunks.length === 0;\n  }\n\n  push(chunk: UIMessageChunk): void {\n    const key = mergeKeyOf(chunk);\n    const tail = this.chunks[this.chunks.length - 1];\n\n    if (tail !== undefined && key !== null && key === this.tailMergeKey) {\n      appendDeltaText(tail, chunk);\n      return;\n    }\n\n    if (this.chunks.length === 0) {\n      this.cancelWindow = this.closeWindow(() => {\n        this.cancelWindow = undefined;\n        this.flush();\n      });\n    }\n\n    if (chunk.type === \"start\") {\n      this.startMessageId = chunk.messageId;\n    }\n    this.chunks.push(chunk);\n    this.tailMergeKey = key;\n  }\n\n  discard(): void {\n    this.cancelWindow?.();\n    this.cancelWindow = undefined;\n    this.chunks = [];\n    this.tailMergeKey = null;\n    this.startMessageId = undefined;\n  }\n\n  flush(): void {\n    this.cancelWindow?.();\n    this.cancelWindow = undefined;\n\n    if (this.chunks.length === 0) return;\n    const batch = this.chunks;\n    this.chunks = [];\n    this.tailMergeKey = null;\n    this.startMessageId = undefined;\n    try {\n      for (const chunk of batch) {\n        this.controller.enqueue(chunk);\n      }\n    } catch {\n      // The stream is already closed; callers still need to finish cleanup.\n    }\n  }\n\n  enqueueNow(chunk: UIMessageChunk): void {\n    this.controller.enqueue(chunk);\n  }\n\n  closeNow(): void {\n    this.controller.close();\n  }\n\n  errorNow(error: Error): void {\n    this.controller.error(error);\n  }\n}\n\ntype ChatResponseFrame = {\n  body?: string;\n  continuation?: boolean;\n  done?: boolean;\n  replay?: boolean;\n  replayComplete?: boolean;\n};\n\n/** Buffers replay chunks and flushes them before live or boundary frames. */\nexport function applyChatResponseFrame(\n  batch: ReplayChunkBatch,\n  frame: ChatResponseFrame\n): void {\n  const endsReplayBurst =\n    frame.replay !== true ||\n    frame.done === true ||\n    frame.replayComplete === true;\n\n  const body = frame.body?.trim();\n  if (!body) {\n    if (endsReplayBurst) batch.flush();\n    return;\n  }\n\n  let chunk: UIMessageChunk;\n  try {\n    chunk = JSON.parse(body) as UIMessageChunk;\n  } catch {\n    // Do not leave buffered chunks behind a malformed boundary frame.\n    if (endsReplayBurst) batch.flush();\n    return;\n  }\n\n  if (endsReplayBurst) {\n    batch.flush();\n    batch.enqueueNow(chunk);\n    return;\n  }\n\n  if (supersedesBufferedPass(batch, chunk, frame)) {\n    batch.discard();\n  }\n\n  batch.push(chunk);\n}\n\nfunction supersedesBufferedPass(\n  batch: ReplayChunkBatch,\n  chunk: UIMessageChunk,\n  frame: ChatResponseFrame\n): boolean {\n  return (\n    chunk.type === \"start\" &&\n    frame.continuation !== true &&\n    chunk.messageId !== undefined &&\n    chunk.messageId === batch.bufferedStartMessageId\n  );\n}\n\n/**\n * Reports an error after buffered content. `controller.error()` would discard\n * chunks that have not reached the consumer.\n */\nexport function failChatStream(batch: ReplayChunkBatch, message: string): void {\n  if (batch.isEmpty) {\n    batch.errorNow(new Error(message));\n    return;\n  }\n\n  batch.flush();\n  batch.enqueueNow({ errorText: message, type: \"error\" });\n  batch.closeNow();\n}\n\nfunction mergeKeyOf(chunk: UIMessageChunk): string | null {\n  switch (chunk.type) {\n    case \"text-delta\":\n    case \"reasoning-delta\":\n      return chunk.providerMetadata === undefined\n        ? `${chunk.type}\\u0000${chunk.id}`\n        : null;\n    case \"tool-input-delta\":\n      return `tool-input-delta\\u0000${chunk.toolCallId}`;\n    default:\n      return null;\n  }\n}\n\nfunction appendDeltaText(target: UIMessageChunk, source: UIMessageChunk): void {\n  if (\n    target.type === \"tool-input-delta\" &&\n    source.type === \"tool-input-delta\"\n  ) {\n    target.inputTextDelta += source.inputTextDelta;\n    return;\n  }\n  if (\n    (target.type === \"text-delta\" || target.type === \"reasoning-delta\") &&\n    (source.type === \"text-delta\" || source.type === \"reasoning-delta\")\n  ) {\n    target.delta += source.delta;\n  }\n}\n","/**\n * Framework-neutral WebSocket transport for AI SDK chat clients.\n *\n * Speaks the CF_AGENT protocol directly and returns AI SDK UI message chunks\n * without reconstructing an HTTP response or parsing an SSE stream.\n */\n\nimport type { ChatTransport, UIMessage, UIMessageChunk } from \"ai\";\nimport { nanoid } from \"nanoid\";\nimport {\n  applyChatResponseFrame,\n  failChatStream,\n  ReplayChunkBatch\n} from \"./replay-batch\";\nimport { MessageType, type OutgoingMessage } from \"./wire-types\";\n\n/**\n * Short safety-net timeout for a resume probe when the server has said nothing.\n * Under normal operation the server answers a `STREAM_RESUME_REQUEST` with\n * `STREAM_RESUMING`, `STREAM_RESUME_NONE`, or `STREAM_PENDING` well before this.\n */\nconst RESUME_PROBE_TIMEOUT_MS = 5000;\n\n/**\n * Extended backstop applied once the server says a turn is pending\n * (`STREAM_PENDING`, #1784). The pre-stream window (queueing, MCP setup,\n * debounce, model latency) can exceed the short probe timeout, and the server\n * guarantees a follow-up `STREAM_RESUMING` or `STREAM_RESUME_NONE` — so we wait\n * much longer (refreshed on every keep-waiting frame) but still cap it so a\n * dropped follow-up degrades to a null resolve instead of hanging forever.\n */\nconst RESUME_PENDING_TIMEOUT_MS = 60000;\n\n/**\n * Agent-like interface for sending/receiving WebSocket messages.\n * Matches `AgentClient` from `agents/client` and the connection returned by\n * `useAgent` from `agents/react`.\n */\nexport interface AgentConnection {\n  send: (data: string) => void;\n  addEventListener: (\n    type: string,\n    listener: (event: MessageEvent) => void,\n    options?: { signal?: AbortSignal }\n  ) => void;\n  removeEventListener: (\n    type: string,\n    listener: (event: MessageEvent) => void\n  ) => void;\n}\n\nexport type WebSocketChatTransportOptions<\n  ChatMessage extends UIMessage = UIMessage\n> = {\n  /** The framework-neutral AgentClient or useAgent connection. */\n  agent: AgentConnection;\n  /**\n   * Callback to add custom fields to the request body before sending.\n   */\n  prepareBody?: (options: {\n    messages: ChatMessage[];\n    trigger: \"submit-message\" | \"regenerate-message\";\n    messageId?: string;\n  }) => Promise<Record<string, unknown>> | Record<string, unknown>;\n  /**\n   * Optional set to track active request IDs.\n   * IDs are added when a request starts and removed when it completes.\n   * Used by the onAgentMessage handler to skip messages already handled by the transport.\n   */\n  activeRequestIds?: Set<string>;\n  /**\n   * Whether generic client-side abort/cancel lifecycle should cancel the\n   * server turn. Explicit cancellation via cancelActiveServerTurn() always\n   * sends CF_AGENT_CHAT_REQUEST_CANCEL.\n   * @default false\n   */\n  cancelOnClientAbort?: boolean;\n};\n\n/**\n * ChatTransport that sends messages over WebSocket and returns a\n * ReadableStream<UIMessageChunk> that the AI SDK's useChat consumes directly.\n *\n * This low-level transport handles new and regenerated request streams plus\n * cancellation. Higher-level protocol coordination such as automatic reconnect\n * resume, cross-tab transcript synchronization, and client-tool continuations\n * is provided by integrations such as `useAgentChat`.\n */\nexport class WebSocketChatTransport<\n  ChatMessage extends UIMessage = UIMessage\n> implements ChatTransport<ChatMessage> {\n  agent: AgentConnection;\n  private prepareBody?: WebSocketChatTransportOptions<ChatMessage>[\"prepareBody\"];\n  private activeRequestIds?: Set<string>;\n  private cancelOnClientAbort: boolean;\n\n  // Pending resume resolver — set by reconnectToStream, called by\n  // handleStreamResuming when onAgentMessage sees CF_AGENT_STREAM_RESUMING.\n  private _resumeResolver: ((data: { id: string }) => void) | null = null;\n  // Pending \"no stream\" resolver — called by handleStreamResumeNone\n  // when onAgentMessage sees CF_AGENT_STREAM_RESUME_NONE.\n  private _resumeNoneResolver:\n    | ((data: { probeId?: string }) => boolean)\n    | null = null;\n  // Keep-waiting hook (#1784) — set by whichever resume path is currently\n  // awaiting, called by handleStreamPending when onAgentMessage sees\n  // CF_AGENT_STREAM_PENDING. Extends the path's probe timeout so a slow\n  // pre-stream window (queue / MCP / model latency) does not resolve early.\n  private _onStreamPending: (() => void) | null = null;\n  // Retransmits the current handshake on a replacement socket without starting\n  // a second AI SDK resume request. Set only while a resolver is active.\n  private _retryResumeProbe: (() => void) | null = null;\n  // Set when a client-side tool result/approval is expected to trigger\n  // a new continuation stream. In this mode reconnectToStream() returns\n  // a deferred ReadableStream immediately so AI SDK status can transition\n  // to \"submitted\" before the server starts streaming.\n  private _expectToolContinuation = false;\n  private _abortToolContinuation: (() => boolean) | null = null;\n  private _activeServerTurnId: string | null = null;\n  private _cancelAttachedStream: (() => boolean) | null = null;\n  // Local-only detach for a resume stream owned by an obsolete Chat/agent\n  // generation. Unlike explicit cancellation, this never cancels server work.\n  private _detachResumeStream: (() => boolean) | null = null;\n\n  constructor(options: WebSocketChatTransportOptions<ChatMessage>) {\n    this.agent = options.agent;\n    this.prepareBody = options.prepareBody;\n    this.activeRequestIds = options.activeRequestIds;\n    this.cancelOnClientAbort = options.cancelOnClientAbort ?? false;\n  }\n\n  /**\n   * Point the singleton transport at a new Agent connection. A pending resolver\n   * belongs to the old Chat/socket generation and must settle before messages\n   * from the replacement connection can be consumed (#1914 review).\n   */\n  setAgent(agent: AgentConnection) {\n    if (this.agent === agent) return;\n    this.resetResumeState();\n    this.agent = agent;\n  }\n\n  setCancelOnClientAbort(cancelOnClientAbort: boolean) {\n    this.cancelOnClientAbort = cancelOnClientAbort;\n  }\n\n  /**\n   * Explicitly cancel the active server turn, if any.\n   * This is separate from generic client-side abort/cancel lifecycle so\n   * clients can detach locally without stopping server work.\n   */\n  cancelActiveServerTurn(): boolean {\n    const requestId = this._activeServerTurnId;\n    let cancelledRequest = false;\n\n    if (requestId) {\n      this.sendCancelFrame(requestId);\n      this._cancelAttachedStream?.();\n      this.clearActiveServerTurn(requestId);\n      cancelledRequest = true;\n    }\n\n    const cancelledToolContinuation = this.abortActiveToolContinuation();\n    return cancelledRequest || cancelledToolContinuation;\n  }\n\n  private sendCancelFrame(requestId: string) {\n    try {\n      this.agent.send(\n        JSON.stringify({\n          id: requestId,\n          type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL\n        })\n      );\n    } catch {\n      // Ignore failures (e.g. agent already disconnected)\n    }\n  }\n\n  private setActiveServerTurn(\n    requestId: string,\n    cancelAttachedStream: (() => boolean) | null\n  ) {\n    this._activeServerTurnId = requestId;\n    this._cancelAttachedStream = cancelAttachedStream;\n  }\n\n  private clearActiveServerTurn(requestId: string) {\n    if (this._activeServerTurnId === requestId) {\n      this._activeServerTurnId = null;\n      this._cancelAttachedStream = null;\n    }\n  }\n\n  /**\n   * Mark that the next reconnectToStream() call should attach to a\n   * server-initiated tool continuation rather than a page-load resume.\n   */\n  expectToolContinuation() {\n    this._expectToolContinuation = true;\n  }\n\n  /**\n   * Abort the active client-side tool continuation stream, if one is attached\n   * to a server request id.\n   */\n  abortActiveToolContinuation(): boolean {\n    return this._abortToolContinuation?.() ?? false;\n  }\n\n  /**\n   * True when the transport is waiting for a resume handshake.\n   */\n  isAwaitingResume(): boolean {\n    return this._resumeResolver !== null || this._resumeNoneResolver !== null;\n  }\n\n  /**\n   * Settle and detach the current handshake without interpreting it as a\n   * server-idle response. Used when the owning hook/agent generation changes.\n   */\n  cancelPendingResume(): boolean {\n    const resolveNone = this._resumeNoneResolver;\n    if (!resolveNone) return false;\n    return resolveNone({});\n  }\n\n  /**\n   * Invalidate all client-side resume state for an obsolete hook/agent\n   * generation without cancelling its durable server turn.\n   */\n  resetResumeState(): void {\n    this._expectToolContinuation = false;\n    this.cancelPendingResume();\n    this._detachResumeStream?.();\n  }\n\n  /**\n   * Re-send the active handshake request on the latest socket generation. This\n   * preserves one AI SDK resume operation while recovering a request/reply lost\n   * with the previous WebSocket.\n   */\n  retryPendingResume(): boolean {\n    const retry = this._retryResumeProbe;\n    if (!retry) return false;\n    retry();\n    return true;\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUMING.\n   * If reconnectToStream is waiting, this handles the resume handshake\n   * (ACK + stream creation) and returns true. Otherwise returns false\n   * so the caller can use its own fallback path.\n   */\n  handleStreamResuming(data: { id: string }): boolean {\n    if (!this._resumeResolver) return false;\n    this._resumeResolver(data);\n    return true;\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUME_NONE.\n   * If reconnectToStream is waiting, resolves the promise with null\n   * immediately (no 5-second timeout). Returns true if handled.\n   */\n  handleStreamResumeNone(data: { probeId?: string } = {}): boolean {\n    if (!this._resumeNoneResolver) return false;\n    return this._resumeNoneResolver(data);\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_PENDING (#1784):\n   * the server accepted a turn but its stream has not started yet. If a resume\n   * path is awaiting, extend its probe timeout (so it keeps waiting for the\n   * eventual STREAM_RESUMING / STREAM_RESUME_NONE instead of resolving null\n   * after the short window). Returns true if a waiting path consumed it.\n   */\n  handleStreamPending(): boolean {\n    if (!this._onStreamPending) return false;\n    this._onStreamPending();\n    return true;\n  }\n\n  /**\n   * Called by the hook's shared message handler when a server turn finishes\n   * outside the currently attached transport stream, such as after local-only\n   * client cleanup.\n   */\n  handleServerTurnCompleted(requestId: string) {\n    this.clearActiveServerTurn(requestId);\n  }\n\n  /**\n   * Register a server turn that is being rendered outside a transport-owned\n   * stream, such as the hook's fallback cross-tab/resume observer path.\n   */\n  observeServerTurn(requestId: string) {\n    this.setActiveServerTurn(requestId, null);\n  }\n\n  async sendMessages(options: {\n    chatId: string;\n    messages: ChatMessage[];\n    abortSignal: AbortSignal | undefined;\n    trigger: \"submit-message\" | \"regenerate-message\";\n    messageId?: string;\n    body?: object;\n    headers?: Record<string, string> | Headers;\n    metadata?: unknown;\n  }): Promise<ReadableStream<UIMessageChunk>> {\n    const requestId = nanoid(8);\n    const abortController = new AbortController();\n    let completed = false;\n    let requestSent = false;\n\n    // Build the request body\n    let extraBody: Record<string, unknown> = {};\n    if (this.prepareBody) {\n      extraBody = await this.prepareBody({\n        messages: options.messages,\n        trigger: options.trigger,\n        messageId: options.messageId\n      });\n    }\n    if (options.body) {\n      extraBody = {\n        ...extraBody,\n        ...(options.body as Record<string, unknown>)\n      };\n    }\n\n    const bodyPayload = JSON.stringify({\n      messages: options.messages,\n      trigger: options.trigger,\n      ...extraBody\n    });\n\n    // Track this request so the onAgentMessage handler skips it\n    this.activeRequestIds?.add(requestId);\n\n    // Create a ReadableStream<UIMessageChunk> that emits parsed chunks\n    // as they arrive over the WebSocket\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n\n    // Single cleanup helper — every terminal path (done, error, abort)\n    // goes through here exactly once.\n    // keepId: when true, do NOT remove requestId from activeIds. Used by\n    // explicit cancellation so onAgentMessage skips in-flight chunks\n    // and the server's final done:true signal until cleanup happens there.\n    const finish = (\n      action: () => void,\n      keepId = false,\n      clearServerTurn = true\n    ) => {\n      if (completed) return;\n      completed = true;\n      if (clearServerTurn) {\n        this.clearActiveServerTurn(requestId);\n      }\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (!keepId) {\n        activeIds?.delete(requestId);\n      }\n      abortController.abort();\n    };\n\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n\n    const cancelActiveRequest = () => {\n      if (completed) return false;\n      finish(() => streamController.error(abortError), true);\n      return true;\n    };\n    this.setActiveServerTurn(requestId, cancelActiveRequest);\n\n    // Abort handler: terminate the local stream. By default, generic AI SDK\n    // abort/cancel lifecycle is local-only so durable server turns can continue\n    // and be resumed. Use cancelActiveServerTurn() for explicit user/app\n    // cancellation, or cancelOnClientAbort for request-lifetime semantics.\n    const onAbort = () => {\n      if (completed) return;\n      if (this.cancelOnClientAbort) {\n        if (requestSent) {\n          this.sendCancelFrame(requestId);\n        }\n        finish(() => streamController.error(abortError), requestSent);\n      } else {\n        finish(() => streamController.error(abortError), false, !requestSent);\n      }\n    };\n\n    // streamController is assigned synchronously by start(), so it is\n    // always available by the time onAbort or onMessage can fire.\n    let streamController!: ReadableStreamDefaultController<UIMessageChunk>;\n\n    const stream = new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        streamController = controller;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<ChatMessage>;\n\n            if (data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE) return;\n            if (data.id !== requestId) return;\n\n            if (data.error) {\n              finish(() =>\n                controller.error(new Error(data.body || \"Stream error\"))\n              );\n              return;\n            }\n\n            // Parse the body as UIMessageChunk and enqueue\n            if (data.body?.trim()) {\n              try {\n                const chunk = JSON.parse(data.body) as UIMessageChunk;\n                controller.enqueue(chunk);\n              } catch {\n                // Skip malformed chunk bodies\n              }\n            }\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () => {\n          finish(() => controller.close(), false, false);\n        };\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: abortController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: abortController.signal\n        });\n      },\n      cancel() {\n        onAbort();\n      }\n    });\n\n    // Handle abort from the caller\n    if (options.abortSignal) {\n      options.abortSignal.addEventListener(\"abort\", onAbort, { once: true });\n      if (options.abortSignal.aborted) onAbort();\n    }\n\n    if (completed) {\n      return stream;\n    }\n\n    // Send the request over WebSocket\n    requestSent = true;\n    agent.send(\n      JSON.stringify({\n        id: requestId,\n        init: {\n          method: \"POST\",\n          body: bodyPayload\n        },\n        type: MessageType.CF_AGENT_USE_CHAT_REQUEST\n      })\n    );\n\n    return stream;\n  }\n\n  async reconnectToStream(_options: {\n    chatId: string;\n  }): Promise<ReadableStream<UIMessageChunk> | null> {\n    // A transport has one handshake slot. Returning null for an unexpected\n    // concurrent caller is safer than overwriting callbacks owned by the first\n    // AI SDK request (StrictMode/manual overlap); the hook serializes its normal\n    // mount, tool, public, and reconnect entry points.\n    if (this.isAwaitingResume()) return null;\n\n    if (this._expectToolContinuation) {\n      this._expectToolContinuation = false;\n      return this._createToolContinuationStream();\n    }\n\n    // Detect whether the server has an active stream for this chat.\n    // Instead of registering another message listener (which races with the\n    // hook), expose identity-owned callbacks consumed synchronously by the\n    // shared handler.\n    const activeIds = this.activeRequestIds;\n    const probeId = nanoid(8);\n\n    return new Promise<ReadableStream<UIMessageChunk> | null>((resolve) => {\n      let resolved = false;\n      let timeout: ReturnType<typeof setTimeout> | undefined;\n      let resumeResolver: ((data: { id: string }) => void) | null = null;\n      let resumeNoneResolver: ((data: { probeId?: string }) => boolean) | null =\n        null;\n      let onStreamPending: (() => void) | null = null;\n      let retryResumeProbe: (() => void) | null = null;\n\n      const clearOwnedCallbacks = () => {\n        if (resumeResolver && this._resumeResolver === resumeResolver) {\n          this._resumeResolver = null;\n        }\n        if (\n          resumeNoneResolver &&\n          this._resumeNoneResolver === resumeNoneResolver\n        ) {\n          this._resumeNoneResolver = null;\n        }\n        if (onStreamPending && this._onStreamPending === onStreamPending) {\n          this._onStreamPending = null;\n        }\n        if (retryResumeProbe && this._retryResumeProbe === retryResumeProbe) {\n          this._retryResumeProbe = null;\n        }\n      };\n\n      const done = (value: ReadableStream<UIMessageChunk> | null) => {\n        if (resolved) return;\n        resolved = true;\n        clearOwnedCallbacks();\n        if (timeout) clearTimeout(timeout);\n        resolve(value);\n      };\n\n      const armTimeout = (delay: number) => {\n        if (timeout) clearTimeout(timeout);\n        timeout = setTimeout(() => done(null), delay);\n      };\n\n      // Keep-waiting (#1784): the server says a turn is accepted but its stream\n      // has not started. Extend this operation's own timeout only.\n      onStreamPending = () => {\n        if (resolved) return;\n        armTimeout(RESUME_PENDING_TIMEOUT_MS);\n      };\n\n      resumeNoneResolver = (data) => {\n        if (data.probeId && data.probeId !== probeId) return false;\n        done(null);\n        return true;\n      };\n\n      resumeResolver = (data: { id: string }) => {\n        const requestId = data.id;\n        activeIds?.add(requestId);\n        const stream = this._createResumeStream(requestId);\n\n        this.agent.send(\n          JSON.stringify({\n            type: MessageType.CF_AGENT_STREAM_RESUME_ACK,\n            id: requestId\n          })\n        );\n\n        done(stream);\n      };\n\n      // Re-arm both the request and timeout on each replacement socket. This is\n      // a retransmission of this resolver's handshake, not a second Chat resume.\n      retryResumeProbe = () => {\n        if (resolved) return;\n        armTimeout(RESUME_PROBE_TIMEOUT_MS);\n        try {\n          this.agent.send(\n            JSON.stringify({\n              type: MessageType.CF_AGENT_STREAM_RESUME_REQUEST,\n              probeId\n            })\n          );\n        } catch {\n          // The next socket open retries again; the timeout remains a backstop.\n        }\n      };\n\n      this._onStreamPending = onStreamPending;\n      this._resumeNoneResolver = resumeNoneResolver;\n      this._resumeResolver = resumeResolver;\n      this._retryResumeProbe = retryResumeProbe;\n      retryResumeProbe();\n    });\n  }\n\n  /**\n   * Creates a deferred ReadableStream for client-side tool continuations.\n   * The stream is returned immediately so AI SDK status becomes \"submitted\"\n   * right after addToolOutput()/addToolApprovalResponse(), then it waits for\n   * the server to announce the continuation via STREAM_RESUMING.\n   */\n  private _createToolContinuationStream(): ReadableStream<UIMessageChunk> {\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n    const streamController = new AbortController();\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n    let completed = false;\n    let requestId: string | null = null;\n    let readerController: ReadableStreamDefaultController<UIMessageChunk> | null =\n      null;\n    // Held outside start() so the close and detach paths can flush replayed\n    // content before they close the stream.\n    let replayBatch: ReplayChunkBatch | null = null;\n    const probeId = nanoid(8);\n    let onResumeRef: ((data: { id: string }) => void) | null = null;\n    let onResumeNoneRef: ((data: { probeId?: string }) => boolean) | null =\n      null;\n    let onStreamPendingRef: (() => void) | null = null;\n    let retryResumeProbeRef: (() => void) | null = null;\n    let abortToolContinuationRef: (() => boolean) | null = null;\n    let detachResumeStreamRef: (() => boolean) | null = null;\n\n    const clearOwnedHandshake = () => {\n      if (onResumeRef && this._resumeResolver === onResumeRef) {\n        this._resumeResolver = null;\n      }\n      if (onResumeNoneRef && this._resumeNoneResolver === onResumeNoneRef) {\n        this._resumeNoneResolver = null;\n      }\n      if (onStreamPendingRef && this._onStreamPending === onStreamPendingRef) {\n        this._onStreamPending = null;\n      }\n      if (\n        retryResumeProbeRef &&\n        this._retryResumeProbe === retryResumeProbeRef\n      ) {\n        this._retryResumeProbe = null;\n      }\n    };\n\n    const finish = (action: () => void, keepRequestId = false) => {\n      if (completed) return;\n      completed = true;\n      if (this._abortToolContinuation === abortToolContinuationRef) {\n        this._abortToolContinuation = null;\n      }\n      if (this._detachResumeStream === detachResumeStreamRef) {\n        this._detachResumeStream = null;\n      }\n      clearOwnedHandshake();\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (requestId && !keepRequestId) {\n        activeIds?.delete(requestId);\n      }\n      streamController.abort();\n    };\n\n    const transport = this;\n\n    abortToolContinuationRef = () => {\n      if (completed) return false;\n\n      if (requestId === null) {\n        finish(() => readerController?.error(abortError));\n        return true;\n      }\n\n      try {\n        agent.send(\n          JSON.stringify({\n            type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL,\n            id: requestId\n          })\n        );\n      } catch {\n        // Ignore failures (e.g. agent already disconnected)\n      }\n\n      // Keep the ID so the shared message handler ignores in-flight chunks\n      // until the server's terminal frame performs its normal cleanup.\n      finish(() => readerController?.error(abortError), true);\n      return true;\n    };\n    this._abortToolContinuation = abortToolContinuationRef;\n    detachResumeStreamRef = () => {\n      if (completed) return false;\n      finish(() => {\n        replayBatch?.flush();\n        readerController?.close();\n      });\n      return true;\n    };\n    this._detachResumeStream = detachResumeStreamRef;\n\n    return new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        readerController = controller;\n        const batch = new ReplayChunkBatch(controller);\n        replayBatch = batch;\n        let timeout: ReturnType<typeof setTimeout> | undefined;\n\n        const armTimeout = (delay: number) => {\n          if (timeout) clearTimeout(timeout);\n          timeout = setTimeout(() => finish(() => controller.close()), delay);\n        };\n\n        const onResumeNone = (data: { probeId?: string }) => {\n          if (data.probeId && data.probeId !== probeId) return false;\n          finish(() => controller.close());\n          return true;\n        };\n\n        const onResume = (data: { id: string }) => {\n          if (requestId) return;\n\n          requestId = data.id;\n          activeIds?.add(requestId);\n          clearOwnedHandshake();\n          if (timeout) clearTimeout(timeout);\n\n          agent.send(\n            JSON.stringify({\n              type: MessageType.CF_AGENT_STREAM_RESUME_ACK,\n              id: requestId\n            })\n          );\n        };\n\n        const onStreamPending = () => {\n          if (completed) return;\n          armTimeout(RESUME_PENDING_TIMEOUT_MS);\n        };\n\n        const retryResumeProbe = () => {\n          if (completed || requestId !== null) return;\n          armTimeout(RESUME_PROBE_TIMEOUT_MS);\n          try {\n            transport.agent.send(\n              JSON.stringify({\n                type: MessageType.CF_AGENT_STREAM_RESUME_REQUEST,\n                probeId\n              })\n            );\n          } catch {\n            // A later socket open can retry this same deferred handshake.\n          }\n        };\n\n        onResumeRef = onResume;\n        onResumeNoneRef = onResumeNone;\n        onStreamPendingRef = onStreamPending;\n        retryResumeProbeRef = retryResumeProbe;\n        transport._resumeResolver = onResume;\n        transport._resumeNoneResolver = onResumeNone;\n        transport._onStreamPending = onStreamPending;\n        transport._retryResumeProbe = retryResumeProbe;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<UIMessage>;\n\n            if (\n              data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE ||\n              requestId == null ||\n              data.id !== requestId\n            ) {\n              return;\n            }\n\n            if (data.error) {\n              finish(() => failChatStream(batch, data.body || \"Stream error\"));\n              return;\n            }\n\n            applyChatResponseFrame(batch, data);\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () =>\n          finish(() => {\n            batch.flush();\n            controller.close();\n          });\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: streamController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: streamController.signal\n        });\n\n        retryResumeProbe();\n      },\n      cancel() {\n        if (requestId && transport.cancelOnClientAbort) {\n          transport.sendCancelFrame(requestId);\n          finish(() => {}, true);\n        } else {\n          finish(() => {});\n        }\n      }\n    });\n  }\n\n  /**\n   * Creates a ReadableStream that receives resumed stream chunks\n   * and forwards them to useChat as UIMessageChunk objects.\n   */\n  private _createResumeStream(\n    requestId: string\n  ): ReadableStream<UIMessageChunk> {\n    // Read agent at resolve time (not when reconnectToStream was called)\n    // so chunk listener attaches to the latest socket after _pk changes.\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n    const chunkController = new AbortController();\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n    let completed = false;\n    let detachResumeStream: (() => boolean) | null = null;\n\n    const finish = (\n      action: () => void,\n      keepId = false,\n      clearServerTurn = true\n    ) => {\n      if (completed) return;\n      completed = true;\n      if (clearServerTurn) {\n        this.clearActiveServerTurn(requestId);\n      }\n      if (this._detachResumeStream === detachResumeStream) {\n        this._detachResumeStream = null;\n      }\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (!keepId) {\n        activeIds?.delete(requestId);\n      }\n      chunkController.abort();\n    };\n\n    let streamController: ReadableStreamDefaultController<UIMessageChunk> | null =\n      null;\n    // Held outside start() so the close and detach paths can flush replayed\n    // content before they close the stream.\n    let replayBatch: ReplayChunkBatch | null = null;\n    const cancelActiveRequest = () => {\n      if (completed) return false;\n      finish(() => streamController?.error(abortError), true);\n      return true;\n    };\n    this.setActiveServerTurn(requestId, cancelActiveRequest);\n\n    const transport = this;\n    detachResumeStream = () => {\n      if (completed) return false;\n      finish(() => {\n        replayBatch?.flush();\n        streamController?.close();\n      });\n      return true;\n    };\n    this._detachResumeStream = detachResumeStream;\n\n    return new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        streamController = controller;\n        const batch = new ReplayChunkBatch(controller);\n        replayBatch = batch;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<UIMessage>;\n\n            if (data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE) return;\n            if (data.id !== requestId) return;\n\n            if (data.error) {\n              finish(() => failChatStream(batch, data.body || \"Stream error\"));\n              return;\n            }\n\n            applyChatResponseFrame(batch, data);\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () => {\n          finish(\n            () => {\n              batch.flush();\n              controller.close();\n            },\n            false,\n            false\n          );\n        };\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: chunkController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: chunkController.signal\n        });\n      },\n      cancel() {\n        if (transport.cancelOnClientAbort) {\n          transport.sendCancelFrame(requestId);\n          finish(() => {}, true);\n        } else {\n          finish(() => {}, false, false);\n        }\n      }\n    });\n  }\n}\n"],"mappings":";;;;;;;;;AAQA,IAAa,mBAAb,MAA8B;CAM5B,YACE,YACA,eACE,UACG;EACH,MAAM,QAAQ,WAAW,OAAO,CAAC;EACjC,aAAa,aAAa,KAAK;CACjC,GACA;EAPiB,KAAA,aAAA;EACA,KAAA,cAAA;EAPnB,KAAQ,SAA2B,CAAC;EACpC,KAAQ,eAA8B;CAYnC;CAEH,IAAI,yBAA6C;EAC/C,OAAO,KAAK;CACd;CAEA,IAAI,UAAmB;EACrB,OAAO,KAAK,OAAO,WAAW;CAChC;CAEA,KAAK,OAA6B;EAChC,MAAM,MAAM,WAAW,KAAK;EAC5B,MAAM,OAAO,KAAK,OAAO,KAAK,OAAO,SAAS;EAE9C,IAAI,SAAS,KAAA,KAAa,QAAQ,QAAQ,QAAQ,KAAK,cAAc;GACnE,gBAAgB,MAAM,KAAK;GAC3B;EACF;EAEA,IAAI,KAAK,OAAO,WAAW,GACzB,KAAK,eAAe,KAAK,kBAAkB;GACzC,KAAK,eAAe,KAAA;GACpB,KAAK,MAAM;EACb,CAAC;EAGH,IAAI,MAAM,SAAS,SACjB,KAAK,iBAAiB,MAAM;EAE9B,KAAK,OAAO,KAAK,KAAK;EACtB,KAAK,eAAe;CACtB;CAEA,UAAgB;EACd,KAAK,eAAe;EACpB,KAAK,eAAe,KAAA;EACpB,KAAK,SAAS,CAAC;EACf,KAAK,eAAe;EACpB,KAAK,iBAAiB,KAAA;CACxB;CAEA,QAAc;EACZ,KAAK,eAAe;EACpB,KAAK,eAAe,KAAA;EAEpB,IAAI,KAAK,OAAO,WAAW,GAAG;EAC9B,MAAM,QAAQ,KAAK;EACnB,KAAK,SAAS,CAAC;EACf,KAAK,eAAe;EACpB,KAAK,iBAAiB,KAAA;EACtB,IAAI;GACF,KAAK,MAAM,SAAS,OAClB,KAAK,WAAW,QAAQ,KAAK;EAEjC,QAAQ,CAER;CACF;CAEA,WAAW,OAA6B;EACtC,KAAK,WAAW,QAAQ,KAAK;CAC/B;CAEA,WAAiB;EACf,KAAK,WAAW,MAAM;CACxB;CAEA,SAAS,OAAoB;EAC3B,KAAK,WAAW,MAAM,KAAK;CAC7B;AACF;;AAWA,SAAgB,uBACd,OACA,OACM;CACN,MAAM,kBACJ,MAAM,WAAW,QACjB,MAAM,SAAS,QACf,MAAM,mBAAmB;CAE3B,MAAM,OAAO,MAAM,MAAM,KAAK;CAC9B,IAAI,CAAC,MAAM;EACT,IAAI,iBAAiB,MAAM,MAAM;EACjC;CACF;CAEA,IAAI;CACJ,IAAI;EACF,QAAQ,KAAK,MAAM,IAAI;CACzB,QAAQ;EAEN,IAAI,iBAAiB,MAAM,MAAM;EACjC;CACF;CAEA,IAAI,iBAAiB;EACnB,MAAM,MAAM;EACZ,MAAM,WAAW,KAAK;EACtB;CACF;CAEA,IAAI,uBAAuB,OAAO,OAAO,KAAK,GAC5C,MAAM,QAAQ;CAGhB,MAAM,KAAK,KAAK;AAClB;AAEA,SAAS,uBACP,OACA,OACA,OACS;CACT,OACE,MAAM,SAAS,WACf,MAAM,iBAAiB,QACvB,MAAM,cAAc,KAAA,KACpB,MAAM,cAAc,MAAM;AAE9B;;;;;AAMA,SAAgB,eAAe,OAAyB,SAAuB;CAC7E,IAAI,MAAM,SAAS;EACjB,MAAM,SAAS,IAAI,MAAM,OAAO,CAAC;EACjC;CACF;CAEA,MAAM,MAAM;CACZ,MAAM,WAAW;EAAE,WAAW;EAAS,MAAM;CAAQ,CAAC;CACtD,MAAM,SAAS;AACjB;AAEA,SAAS,WAAW,OAAsC;CACxD,QAAQ,MAAM,MAAd;EACE,KAAK;EACL,KAAK,mBACH,OAAO,MAAM,qBAAqB,KAAA,IAC9B,GAAG,MAAM,KAAK,QAAQ,MAAM,OAC5B;EACN,KAAK,oBACH,OAAO,yBAAyB,MAAM;EACxC,SACE,OAAO;CACX;AACF;AAEA,SAAS,gBAAgB,QAAwB,QAA8B;CAC7E,IACE,OAAO,SAAS,sBAChB,OAAO,SAAS,oBAChB;EACA,OAAO,kBAAkB,OAAO;EAChC;CACF;CACA,KACG,OAAO,SAAS,gBAAgB,OAAO,SAAS,uBAChD,OAAO,SAAS,gBAAgB,OAAO,SAAS,oBAEjD,OAAO,SAAS,OAAO;AAE3B;;;;;;;;AC/KA,MAAM,0BAA0B;;;;;;;;;AAUhC,MAAM,4BAA4B;;;;;;;;;;AAyDlC,IAAa,yBAAb,MAEwC;CAkCtC,YAAY,SAAqD;EA1BjE,KAAQ,kBAA2D;EAGnE,KAAQ,sBAEG;EAKX,KAAQ,mBAAwC;EAGhD,KAAQ,oBAAyC;EAKjD,KAAQ,0BAA0B;EAClC,KAAQ,yBAAiD;EACzD,KAAQ,sBAAqC;EAC7C,KAAQ,wBAAgD;EAGxD,KAAQ,sBAA8C;EAGpD,KAAK,QAAQ,QAAQ;EACrB,KAAK,cAAc,QAAQ;EAC3B,KAAK,mBAAmB,QAAQ;EAChC,KAAK,sBAAsB,QAAQ,uBAAuB;CAC5D;;;;;;CAOA,SAAS,OAAwB;EAC/B,IAAI,KAAK,UAAU,OAAO;EAC1B,KAAK,iBAAiB;EACtB,KAAK,QAAQ;CACf;CAEA,uBAAuB,qBAA8B;EACnD,KAAK,sBAAsB;CAC7B;;;;;;CAOA,yBAAkC;EAChC,MAAM,YAAY,KAAK;EACvB,IAAI,mBAAmB;EAEvB,IAAI,WAAW;GACb,KAAK,gBAAgB,SAAS;GAC9B,KAAK,wBAAwB;GAC7B,KAAK,sBAAsB,SAAS;GACpC,mBAAmB;EACrB;EAEA,MAAM,4BAA4B,KAAK,4BAA4B;EACnE,OAAO,oBAAoB;CAC7B;CAEA,gBAAwB,WAAmB;EACzC,IAAI;GACF,KAAK,MAAM,KACT,KAAK,UAAU;IACb,IAAI;IACJ,MAAA;GACF,CAAC,CACH;EACF,QAAQ,CAER;CACF;CAEA,oBACE,WACA,sBACA;EACA,KAAK,sBAAsB;EAC3B,KAAK,wBAAwB;CAC/B;CAEA,sBAA8B,WAAmB;EAC/C,IAAI,KAAK,wBAAwB,WAAW;GAC1C,KAAK,sBAAsB;GAC3B,KAAK,wBAAwB;EAC/B;CACF;;;;;CAMA,yBAAyB;EACvB,KAAK,0BAA0B;CACjC;;;;;CAMA,8BAAuC;EACrC,OAAO,KAAK,yBAAyB,KAAK;CAC5C;;;;CAKA,mBAA4B;EAC1B,OAAO,KAAK,oBAAoB,QAAQ,KAAK,wBAAwB;CACvE;;;;;CAMA,sBAA+B;EAC7B,MAAM,cAAc,KAAK;EACzB,IAAI,CAAC,aAAa,OAAO;EACzB,OAAO,YAAY,CAAC,CAAC;CACvB;;;;;CAMA,mBAAyB;EACvB,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB;EACzB,KAAK,sBAAsB;CAC7B;;;;;;CAOA,qBAA8B;EAC5B,MAAM,QAAQ,KAAK;EACnB,IAAI,CAAC,OAAO,OAAO;EACnB,MAAM;EACN,OAAO;CACT;;;;;;;CAQA,qBAAqB,MAA+B;EAClD,IAAI,CAAC,KAAK,iBAAiB,OAAO;EAClC,KAAK,gBAAgB,IAAI;EACzB,OAAO;CACT;;;;;;CAOA,uBAAuB,OAA6B,CAAC,GAAY;EAC/D,IAAI,CAAC,KAAK,qBAAqB,OAAO;EACtC,OAAO,KAAK,oBAAoB,IAAI;CACtC;;;;;;;;CASA,sBAA+B;EAC7B,IAAI,CAAC,KAAK,kBAAkB,OAAO;EACnC,KAAK,iBAAiB;EACtB,OAAO;CACT;;;;;;CAOA,0BAA0B,WAAmB;EAC3C,KAAK,sBAAsB,SAAS;CACtC;;;;;CAMA,kBAAkB,WAAmB;EACnC,KAAK,oBAAoB,WAAW,IAAI;CAC1C;CAEA,MAAM,aAAa,SASyB;EAC1C,MAAM,YAAY,OAAO,CAAC;EAC1B,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,IAAI,YAAY;EAChB,IAAI,cAAc;EAGlB,IAAI,YAAqC,CAAC;EAC1C,IAAI,KAAK,aACP,YAAY,MAAM,KAAK,YAAY;GACjC,UAAU,QAAQ;GAClB,SAAS,QAAQ;GACjB,WAAW,QAAQ;EACrB,CAAC;EAEH,IAAI,QAAQ,MACV,YAAY;GACV,GAAG;GACH,GAAI,QAAQ;EACd;EAGF,MAAM,cAAc,KAAK,UAAU;GACjC,UAAU,QAAQ;GAClB,SAAS,QAAQ;GACjB,GAAG;EACL,CAAC;EAGD,KAAK,kBAAkB,IAAI,SAAS;EAIpC,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EAOvB,MAAM,UACJ,QACA,SAAS,OACT,kBAAkB,SACf;GACH,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,iBACF,KAAK,sBAAsB,SAAS;GAEtC,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,CAAC,QACH,WAAW,OAAO,SAAS;GAE7B,gBAAgB,MAAM;EACxB;EAEA,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAElB,MAAM,4BAA4B;GAChC,IAAI,WAAW,OAAO;GACtB,aAAa,iBAAiB,MAAM,UAAU,GAAG,IAAI;GACrD,OAAO;EACT;EACA,KAAK,oBAAoB,WAAW,mBAAmB;EAMvD,MAAM,gBAAgB;GACpB,IAAI,WAAW;GACf,IAAI,KAAK,qBAAqB;IAC5B,IAAI,aACF,KAAK,gBAAgB,SAAS;IAEhC,aAAa,iBAAiB,MAAM,UAAU,GAAG,WAAW;GAC9D,OACE,aAAa,iBAAiB,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW;EAExE;EAIA,IAAI;EAEJ,MAAM,SAAS,IAAI,eAA+B;GAChD,MAAM,YAAY;IAChB,mBAAmB;IAEnB,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IAAI,KAAK,SAAA,8BAAiD;MAC1D,IAAI,KAAK,OAAO,WAAW;MAE3B,IAAI,KAAK,OAAO;OACd,aACE,WAAW,MAAM,IAAI,MAAM,KAAK,QAAQ,cAAc,CAAC,CACzD;OACA;MACF;MAGA,IAAI,KAAK,MAAM,KAAK,GAClB,IAAI;OACF,MAAM,QAAQ,KAAK,MAAM,KAAK,IAAI;OAClC,WAAW,QAAQ,KAAK;MAC1B,QAAQ,CAER;MAGF,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBAAgB;KACpB,aAAa,WAAW,MAAM,GAAG,OAAO,KAAK;IAC/C;IAEA,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,gBAAgB,OAC1B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,gBAAgB,OAC1B,CAAC;GACH;GACA,SAAS;IACP,QAAQ;GACV;EACF,CAAC;EAGD,IAAI,QAAQ,aAAa;GACvB,QAAQ,YAAY,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;GACrE,IAAI,QAAQ,YAAY,SAAS,QAAQ;EAC3C;EAEA,IAAI,WACF,OAAO;EAIT,cAAc;EACd,MAAM,KACJ,KAAK,UAAU;GACb,IAAI;GACJ,MAAM;IACJ,QAAQ;IACR,MAAM;GACR;GACA,MAAA;EACF,CAAC,CACH;EAEA,OAAO;CACT;CAEA,MAAM,kBAAkB,UAE2B;EAKjD,IAAI,KAAK,iBAAiB,GAAG,OAAO;EAEpC,IAAI,KAAK,yBAAyB;GAChC,KAAK,0BAA0B;GAC/B,OAAO,KAAK,8BAA8B;EAC5C;EAMA,MAAM,YAAY,KAAK;EACvB,MAAM,UAAU,OAAO,CAAC;EAExB,OAAO,IAAI,SAAgD,YAAY;GACrE,IAAI,WAAW;GACf,IAAI;GACJ,IAAI,iBAA0D;GAC9D,IAAI,qBACF;GACF,IAAI,kBAAuC;GAC3C,IAAI,mBAAwC;GAE5C,MAAM,4BAA4B;IAChC,IAAI,kBAAkB,KAAK,oBAAoB,gBAC7C,KAAK,kBAAkB;IAEzB,IACE,sBACA,KAAK,wBAAwB,oBAE7B,KAAK,sBAAsB;IAE7B,IAAI,mBAAmB,KAAK,qBAAqB,iBAC/C,KAAK,mBAAmB;IAE1B,IAAI,oBAAoB,KAAK,sBAAsB,kBACjD,KAAK,oBAAoB;GAE7B;GAEA,MAAM,QAAQ,UAAiD;IAC7D,IAAI,UAAU;IACd,WAAW;IACX,oBAAoB;IACpB,IAAI,SAAS,aAAa,OAAO;IACjC,QAAQ,KAAK;GACf;GAEA,MAAM,cAAc,UAAkB;IACpC,IAAI,SAAS,aAAa,OAAO;IACjC,UAAU,iBAAiB,KAAK,IAAI,GAAG,KAAK;GAC9C;GAIA,wBAAwB;IACtB,IAAI,UAAU;IACd,WAAW,yBAAyB;GACtC;GAEA,sBAAsB,SAAS;IAC7B,IAAI,KAAK,WAAW,KAAK,YAAY,SAAS,OAAO;IACrD,KAAK,IAAI;IACT,OAAO;GACT;GAEA,kBAAkB,SAAyB;IACzC,MAAM,YAAY,KAAK;IACvB,WAAW,IAAI,SAAS;IACxB,MAAM,SAAS,KAAK,oBAAoB,SAAS;IAEjD,KAAK,MAAM,KACT,KAAK,UAAU;KACb,MAAA;KACA,IAAI;IACN,CAAC,CACH;IAEA,KAAK,MAAM;GACb;GAIA,yBAAyB;IACvB,IAAI,UAAU;IACd,WAAW,uBAAuB;IAClC,IAAI;KACF,KAAK,MAAM,KACT,KAAK,UAAU;MACb,MAAA;MACA;KACF,CAAC,CACH;IACF,QAAQ,CAER;GACF;GAEA,KAAK,mBAAmB;GACxB,KAAK,sBAAsB;GAC3B,KAAK,kBAAkB;GACvB,KAAK,oBAAoB;GACzB,iBAAiB;EACnB,CAAC;CACH;;;;;;;CAQA,gCAAwE;EACtE,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EACvB,MAAM,mBAAmB,IAAI,gBAAgB;EAC7C,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAClB,IAAI,YAAY;EAChB,IAAI,YAA2B;EAC/B,IAAI,mBACF;EAGF,IAAI,cAAuC;EAC3C,MAAM,UAAU,OAAO,CAAC;EACxB,IAAI,cAAuD;EAC3D,IAAI,kBACF;EACF,IAAI,qBAA0C;EAC9C,IAAI,sBAA2C;EAC/C,IAAI,2BAAmD;EACvD,IAAI,wBAAgD;EAEpD,MAAM,4BAA4B;GAChC,IAAI,eAAe,KAAK,oBAAoB,aAC1C,KAAK,kBAAkB;GAEzB,IAAI,mBAAmB,KAAK,wBAAwB,iBAClD,KAAK,sBAAsB;GAE7B,IAAI,sBAAsB,KAAK,qBAAqB,oBAClD,KAAK,mBAAmB;GAE1B,IACE,uBACA,KAAK,sBAAsB,qBAE3B,KAAK,oBAAoB;EAE7B;EAEA,MAAM,UAAU,QAAoB,gBAAgB,UAAU;GAC5D,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,KAAK,2BAA2B,0BAClC,KAAK,yBAAyB;GAEhC,IAAI,KAAK,wBAAwB,uBAC/B,KAAK,sBAAsB;GAE7B,oBAAoB;GACpB,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,aAAa,CAAC,eAChB,WAAW,OAAO,SAAS;GAE7B,iBAAiB,MAAM;EACzB;EAEA,MAAM,YAAY;EAElB,iCAAiC;GAC/B,IAAI,WAAW,OAAO;GAEtB,IAAI,cAAc,MAAM;IACtB,aAAa,kBAAkB,MAAM,UAAU,CAAC;IAChD,OAAO;GACT;GAEA,IAAI;IACF,MAAM,KACJ,KAAK,UAAU;KACb,MAAA;KACA,IAAI;IACN,CAAC,CACH;GACF,QAAQ,CAER;GAIA,aAAa,kBAAkB,MAAM,UAAU,GAAG,IAAI;GACtD,OAAO;EACT;EACA,KAAK,yBAAyB;EAC9B,8BAA8B;GAC5B,IAAI,WAAW,OAAO;GACtB,aAAa;IACX,aAAa,MAAM;IACnB,kBAAkB,MAAM;GAC1B,CAAC;GACD,OAAO;EACT;EACA,KAAK,sBAAsB;EAE3B,OAAO,IAAI,eAA+B;GACxC,MAAM,YAAY;IAChB,mBAAmB;IACnB,MAAM,QAAQ,IAAI,iBAAiB,UAAU;IAC7C,cAAc;IACd,IAAI;IAEJ,MAAM,cAAc,UAAkB;KACpC,IAAI,SAAS,aAAa,OAAO;KACjC,UAAU,iBAAiB,aAAa,WAAW,MAAM,CAAC,GAAG,KAAK;IACpE;IAEA,MAAM,gBAAgB,SAA+B;KACnD,IAAI,KAAK,WAAW,KAAK,YAAY,SAAS,OAAO;KACrD,aAAa,WAAW,MAAM,CAAC;KAC/B,OAAO;IACT;IAEA,MAAM,YAAY,SAAyB;KACzC,IAAI,WAAW;KAEf,YAAY,KAAK;KACjB,WAAW,IAAI,SAAS;KACxB,oBAAoB;KACpB,IAAI,SAAS,aAAa,OAAO;KAEjC,MAAM,KACJ,KAAK,UAAU;MACb,MAAA;MACA,IAAI;KACN,CAAC,CACH;IACF;IAEA,MAAM,wBAAwB;KAC5B,IAAI,WAAW;KACf,WAAW,yBAAyB;IACtC;IAEA,MAAM,yBAAyB;KAC7B,IAAI,aAAa,cAAc,MAAM;KACrC,WAAW,uBAAuB;KAClC,IAAI;MACF,UAAU,MAAM,KACd,KAAK,UAAU;OACb,MAAA;OACA;MACF,CAAC,CACH;KACF,QAAQ,CAER;IACF;IAEA,cAAc;IACd,kBAAkB;IAClB,qBAAqB;IACrB,sBAAsB;IACtB,UAAU,kBAAkB;IAC5B,UAAU,sBAAsB;IAChC,UAAU,mBAAmB;IAC7B,UAAU,oBAAoB;IAE9B,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IACE,KAAK,SAAA,gCACL,aAAa,QACb,KAAK,OAAO,WAEZ;MAGF,IAAI,KAAK,OAAO;OACd,aAAa,eAAe,OAAO,KAAK,QAAQ,cAAc,CAAC;OAC/D;MACF;MAEA,uBAAuB,OAAO,IAAI;MAElC,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBACJ,aAAa;KACX,MAAM,MAAM;KACZ,WAAW,MAAM;IACnB,CAAC;IAEH,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,iBAAiB,OAC3B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,iBAAiB,OAC3B,CAAC;IAED,iBAAiB;GACnB;GACA,SAAS;IACP,IAAI,aAAa,UAAU,qBAAqB;KAC9C,UAAU,gBAAgB,SAAS;KACnC,aAAa,CAAC,GAAG,IAAI;IACvB,OACE,aAAa,CAAC,CAAC;GAEnB;EACF,CAAC;CACH;;;;;CAMA,oBACE,WACgC;EAGhC,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EACvB,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAClB,IAAI,YAAY;EAChB,IAAI,qBAA6C;EAEjD,MAAM,UACJ,QACA,SAAS,OACT,kBAAkB,SACf;GACH,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,iBACF,KAAK,sBAAsB,SAAS;GAEtC,IAAI,KAAK,wBAAwB,oBAC/B,KAAK,sBAAsB;GAE7B,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,CAAC,QACH,WAAW,OAAO,SAAS;GAE7B,gBAAgB,MAAM;EACxB;EAEA,IAAI,mBACF;EAGF,IAAI,cAAuC;EAC3C,MAAM,4BAA4B;GAChC,IAAI,WAAW,OAAO;GACtB,aAAa,kBAAkB,MAAM,UAAU,GAAG,IAAI;GACtD,OAAO;EACT;EACA,KAAK,oBAAoB,WAAW,mBAAmB;EAEvD,MAAM,YAAY;EAClB,2BAA2B;GACzB,IAAI,WAAW,OAAO;GACtB,aAAa;IACX,aAAa,MAAM;IACnB,kBAAkB,MAAM;GAC1B,CAAC;GACD,OAAO;EACT;EACA,KAAK,sBAAsB;EAE3B,OAAO,IAAI,eAA+B;GACxC,MAAM,YAAY;IAChB,mBAAmB;IACnB,MAAM,QAAQ,IAAI,iBAAiB,UAAU;IAC7C,cAAc;IAEd,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IAAI,KAAK,SAAA,8BAAiD;MAC1D,IAAI,KAAK,OAAO,WAAW;MAE3B,IAAI,KAAK,OAAO;OACd,aAAa,eAAe,OAAO,KAAK,QAAQ,cAAc,CAAC;OAC/D;MACF;MAEA,uBAAuB,OAAO,IAAI;MAElC,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBAAgB;KACpB,aACQ;MACJ,MAAM,MAAM;MACZ,WAAW,MAAM;KACnB,GACA,OACA,KACF;IACF;IAEA,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,gBAAgB,OAC1B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,gBAAgB,OAC1B,CAAC;GACH;GACA,SAAS;IACP,IAAI,UAAU,qBAAqB;KACjC,UAAU,gBAAgB,SAAS;KACnC,aAAa,CAAC,GAAG,IAAI;IACvB,OACE,aAAa,CAAC,GAAG,OAAO,KAAK;GAEjC;EACF,CAAC;CACH;AACF"}