@milaboratories/pl-client 2.17.7 → 2.17.9

package/dist/core/websocket_stream.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"websocket_stream.cjs","sources":["../../src/core/websocket_stream.ts"],"sourcesContent":["import { WebSocket, type WebSocketInit, type Dispatcher, ErrorEvent } from \"undici\";\nimport type { BiDiStream } from \"./abstract_stream\";\nimport Denque from \"denque\";\nimport type { RetryConfig } from \"../helpers/retry_strategy\";\nimport { RetryStrategy } from \"../helpers/retry_strategy\";\nimport { DisconnectedError } from \"./errors\";\n\ninterface QueuedMessage<InType extends object> {\n  message: InType;\n  resolve: () => void;\n  reject: (error: Error) => void;\n}\n\ninterface ResponseResolver<OutType extends object> {\n  resolve: (value: IteratorResult<OutType>) => void;\n  reject: (error: Error) => void;\n}\n\nenum ConnectionState {\n  NEW = 0,\n  CONNECTING = 1,\n  CONNECTED = 2,\n  CLOSING = 3,\n  CLOSED = 4,\n}\n\nexport type WSStreamOptions<ClientMsg extends object, ServerMsg extends object> = {\n  abortSignal?: AbortSignal;\n\n  dispatcher?: Dispatcher;\n  jwtToken?: string;\n  retryConfig?: Partial<RetryConfig>;\n\n  onComplete?: (stream: WebSocketBiDiStream<ClientMsg, ServerMsg>) => void | Promise<void>;\n};\n\n/**\n * WebSocket-based bidirectional stream implementation for LLTransaction.\n * Implements BiDiStream interface which is compatible with DuplexStreamingCall.\n */\nexport class WebSocketBiDiStream<\n  ClientMsg extends object,\n  ServerMsg extends object,\n> implements BiDiStream<ClientMsg, ServerMsg> {\n  // Connection\n  private ws: WebSocket | null = null;\n  private connectionState: ConnectionState = ConnectionState.NEW;\n  private readonly reconnection: RetryStrategy;\n\n  // Send management\n  private readonly sendQueue = new Denque<QueuedMessage<ClientMsg>>();\n  private sendCompleted = false;\n  private readonly onComplete: (\n    stream: WebSocketBiDiStream<ClientMsg, ServerMsg>,\n  ) => void | Promise<void>;\n\n  // Response management\n  private readonly responseQueue = new Denque<ServerMsg>();\n  private responseResolvers: ResponseResolver<ServerMsg>[] = [];\n\n  // Error tracking\n  private lastError?: Error;\n\n  // === Public API ===\n\n  public readonly requests = {\n    send: async (message: ClientMsg): Promise<void> => {\n      return await this.enqueueSend(message);\n    },\n\n    complete: async (): Promise<void> => {\n      if (this.sendCompleted) return;\n\n      await this.drainSendQueue(); // ensure we sent all already queued messages before closing the stream\n      try {\n        await this.onComplete(this); // custom onComplete may send additional messages\n      } catch {\n        // When 'complete' gets called concurrently with connection break or over a broken\n        // transaction stream (server decided it should drop transaction), server would close\n        // connection anyway on its end. We can safely ignore error here and just continue working.\n      }\n      this.sendCompleted = true;\n    },\n  };\n\n  public readonly responses: AsyncIterable<ServerMsg> = {\n    [Symbol.asyncIterator]: () => this.createResponseIterator(),\n  };\n\n  public close(): void {\n    this.reconnection.cancel();\n\n    if (this.connectionState < ConnectionState.CONNECTED) {\n      // Never reached CONNECTED state. ws.close() will never trigger 'close' event.\n      this.ws?.close();\n      this.onClose();\n      return;\n    }\n\n    if (!this.progressConnectionState(ConnectionState.CLOSING)) return;\n    this.ws!.close();\n  }\n\n  constructor(\n    private readonly url: string,\n    private readonly serializeClientMessage: (message: ClientMsg) => Uint8Array,\n    private readonly parseServerMessage: (data: Uint8Array) => ServerMsg,\n    private readonly options: WSStreamOptions<ClientMsg, ServerMsg> = {},\n  ) {\n    this.onComplete = this.options.onComplete ?? ((stream) => stream.close());\n\n    const retryConfig = this.options.retryConfig ?? {};\n    this.reconnection = new RetryStrategy(retryConfig, {\n      onRetry: () => {\n        void this.connect();\n      },\n      onMaxAttemptsReached: (error) => this.handleError(error),\n    });\n\n    if (this.options.abortSignal?.aborted) {\n      this.progressConnectionState(ConnectionState.CLOSED);\n      return;\n    }\n\n    this.options.abortSignal?.addEventListener(\"abort\", () => this.close());\n    this.connect();\n  }\n\n  // === Connection Lifecycle ===\n\n  private connect(): void {\n    if (this.options.abortSignal?.aborted) return;\n\n    // Prevent reconnecting after first successful connection.\n    if (!this.progressConnectionState(ConnectionState.CONNECTING)) return;\n\n    try {\n      this.ws = this.createWebSocket();\n\n      this.ws.addEventListener(\"open\", () => this.onOpen());\n      this.ws.addEventListener(\"message\", (event) => this.onMessage(event.data));\n      this.ws.addEventListener(\"error\", (error) => this.onError(error));\n      this.ws.addEventListener(\"close\", () => this.onClose());\n    } catch (error) {\n      this.lastError = this.toError(error);\n      this.reconnection.schedule();\n    }\n  }\n\n  private createWebSocket(): WebSocket {\n    const options: WebSocketInit = {};\n\n    if (this.options.jwtToken)\n      options.headers = { authorization: `Bearer ${this.options.jwtToken}` };\n    if (this.options.dispatcher) options.dispatcher = this.options.dispatcher;\n\n    const ws = new WebSocket(this.url, options);\n    ws.binaryType = \"arraybuffer\";\n    return ws;\n  }\n\n  private onOpen(): void {\n    this.progressConnectionState(ConnectionState.CONNECTED);\n    this.processSendQueue();\n  }\n\n  private onMessage(data: unknown): void {\n    if (!(data instanceof ArrayBuffer)) {\n      this.handleError(new Error(`Unexpected WS message format: ${typeof data}`));\n      return;\n    }\n\n    try {\n      const message = this.parseServerMessage(new Uint8Array(data));\n      this.deliverResponse(message);\n    } catch (error) {\n      this.handleError(this.toError(error));\n    }\n  }\n\n  private onError(error: unknown): void {\n    if (this.connectionState < ConnectionState.CONNECTED) {\n      // Try to connect several times until we succeed or run out of attempts.\n      this.lastError = this.toError(error);\n      this.reconnection.schedule();\n      return;\n    }\n\n    this.handleError(this.toError(error));\n  }\n\n  private onClose(): void {\n    this.progressConnectionState(ConnectionState.CLOSED);\n\n    // If abort signal was triggered, use that as the error source\n    if (this.options.abortSignal?.aborted && !this.lastError) {\n      const reason = this.options.abortSignal.reason;\n      if (reason instanceof Error) {\n        this.lastError = reason;\n      } else if (reason !== undefined) {\n        this.lastError = new Error(String(reason), { cause: reason });\n      } else {\n        this.lastError = this.createStreamClosedError();\n      }\n    }\n\n    if (!this.lastError) {\n      this.rejectAllSendOperations(this.createStreamClosedError());\n      this.resolveAllPendingResponses(); // unblock active async iterator\n    } else {\n      this.rejectAllPendingOperations(this.lastError);\n    }\n  }\n\n  // === Send Queue Management ===\n\n  private enqueueSend(message: ClientMsg): Promise<void> {\n    if (this.sendCompleted) {\n      throw new Error(\"Cannot send: stream already completed\");\n    }\n\n    if (this.options.abortSignal?.aborted) {\n      throw new Error(\"Cannot send: stream aborted\");\n    }\n\n    return new Promise<void>((resolve, reject) => {\n      this.sendQueue.push({ message, resolve, reject });\n      this.processSendQueue();\n    });\n  }\n\n  private processSendQueue(): void {\n    if (!this.canSendMessages()) return;\n\n    while (this.sendQueue.length > 0) {\n      const queued = this.sendQueue.shift()!;\n      this.sendQueuedMessage(queued);\n    }\n  }\n\n  private canSendMessages(): boolean {\n    return this.connectionState === ConnectionState.CONNECTED;\n  }\n\n  private sendQueuedMessage(queued: QueuedMessage<ClientMsg>): void {\n    try {\n      const ws = this.ws;\n      if (!ws) {\n        throw new Error(\"WebSocket is not connected\");\n      }\n\n      // Check if WebSocket is in a valid state for sending\n      if (ws.readyState !== WebSocket.OPEN) {\n        throw new Error(`WebSocket is not open (readyState: ${ws.readyState})`);\n      }\n\n      const binary = this.serializeClientMessage(queued.message);\n      ws.send(binary);\n      queued.resolve();\n    } catch (error) {\n      queued.reject(this.toError(error));\n    }\n  }\n\n  private async drainSendQueue(): Promise<void> {\n    const POLL_INTERVAL_MS = 5;\n\n    while (this.sendQueue.length > 0) {\n      await this.waitForCondition(() => this.sendQueue.length === 0, POLL_INTERVAL_MS);\n    }\n  }\n\n  private waitForCondition(condition: () => boolean, intervalMs: number): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      if (this.options.abortSignal?.aborted) {\n        return reject(this.toError(this.options.abortSignal.reason) ?? new Error(\"Stream aborted\"));\n      }\n\n      let timeoutId: ReturnType<typeof setTimeout>;\n      const onAbort = () => {\n        clearTimeout(timeoutId);\n        reject(this.toError(this.options.abortSignal?.reason) ?? new Error(\"Stream aborted\"));\n      };\n\n      this.options.abortSignal?.addEventListener(\"abort\", onAbort, { once: true });\n\n      const check = () => {\n        if (condition() || this.isStreamEnded()) {\n          this.options.abortSignal?.removeEventListener(\"abort\", onAbort);\n          resolve();\n        } else {\n          timeoutId = setTimeout(check, intervalMs);\n        }\n      };\n\n      check();\n    });\n  }\n\n  // === Response Delivery ===\n\n  private deliverResponse(message: ServerMsg): void {\n    if (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.resolve({ value: message, done: false });\n    } else {\n      this.responseQueue.push(message);\n    }\n  }\n\n  private async *createResponseIterator(): AsyncIterator<ServerMsg> {\n    while (true) {\n      const result = await this.nextResponse();\n\n      if (result.done) break;\n\n      yield result.value;\n    }\n  }\n\n  private nextResponse(): Promise<IteratorResult<ServerMsg>> {\n    return new Promise<IteratorResult<ServerMsg>>((resolve, reject) => {\n      // Fast path: message already available\n      if (this.responseQueue.length > 0) {\n        const message = this.responseQueue.shift()!;\n        resolve({ value: message, done: false });\n        return;\n      }\n\n      // Stream ended\n      if (this.isStreamEnded()) {\n        if (this.lastError) {\n          reject(this.lastError);\n        } else {\n          resolve({ value: undefined as any, done: true });\n        }\n        return;\n      }\n\n      // Wait for next message\n      this.responseResolvers.push({ resolve, reject });\n    });\n  }\n\n  private resolveAllPendingResponses(): void {\n    while (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.resolve({ value: undefined as any, done: true });\n    }\n  }\n\n  // === Error Handling ===\n\n  private handleError(error: Error): void {\n    this.lastError = error;\n    this.close();\n  }\n\n  private rejectAllPendingOperations(error: Error): void {\n    this.rejectAllSendOperations(error);\n    this.rejectAllResponseResolvers(error);\n  }\n\n  private rejectAllSendOperations(error: Error): void {\n    while (this.sendQueue.length > 0) {\n      const queued = this.sendQueue.shift()!;\n      queued.reject(error);\n    }\n  }\n\n  private rejectAllResponseResolvers(error: Error): void {\n    while (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.reject(error);\n    }\n  }\n\n  private createStreamClosedError(): Error {\n    if (this.options.abortSignal?.aborted) {\n      const reason = this.options.abortSignal.reason;\n      if (reason instanceof Error) {\n        return reason;\n      }\n      return new Error(\"Stream aborted\", { cause: reason });\n    }\n\n    return new Error(\"Stream closed\");\n  }\n\n  // === Helpers ===\n\n  private isStreamEnded(): boolean {\n    return (\n      this.connectionState === ConnectionState.CLOSED || this.options.abortSignal?.aborted || false\n    );\n  }\n\n  private toError(error: unknown): Error {\n    if (error instanceof Error) return error;\n    if (error instanceof ErrorEvent) {\n      const err = error.error;\n      // undici WebSocket throws TypeError with empty message on socket close\n      // (e.g., when connection is lost or server disconnects)\n      if (err instanceof TypeError && !err.message) {\n        return new DisconnectedError(\"WebSocket connection closed unexpectedly\");\n      }\n      return err instanceof Error ? err : new Error(\"WebSocket error\", { cause: error });\n    }\n    return new Error(String(error));\n  }\n\n  /**\n   * Connection state progresses linearly from NEW to CLOSED and never goes back.\n   * This internal contract dramatically simplifies the internal stream state management.\n   *\n   * If you ever feel the need to make this contract less strict, think twice.\n   */\n  private progressConnectionState(newState: ConnectionState): boolean {\n    if (newState < this.connectionState) {\n      return false;\n    }\n    this.connectionState = newState;\n    return true;\n  }\n}\n"],"names":["RetryStrategy","WebSocket","ErrorEvent","DisconnectedError"],"mappings":";;;;;;;AAkBA,IAAK,eAMJ;AAND,CAAA,UAAK,eAAe,EAAA;AAClB,IAAA,eAAA,CAAA,eAAA,CAAA,KAAA,CAAA,GAAA,CAAA,CAAA,GAAA,KAAO;AACP,IAAA,eAAA,CAAA,eAAA,CAAA,YAAA,CAAA,GAAA,CAAA,CAAA,GAAA,YAAc;AACd,IAAA,eAAA,CAAA,eAAA,CAAA,WAAA,CAAA,GAAA,CAAA,CAAA,GAAA,WAAa;AACb,IAAA,eAAA,CAAA,eAAA,CAAA,SAAA,CAAA,GAAA,CAAA,CAAA,GAAA,SAAW;AACX,IAAA,eAAA,CAAA,eAAA,CAAA,QAAA,CAAA,GAAA,CAAA,CAAA,GAAA,QAAU;AACZ,CAAC,EANI,eAAe,KAAf,eAAe,GAAA,EAAA,CAAA,CAAA;AAkBpB;;;AAGG;MACU,mBAAmB,CAAA;AAgEX,IAAA,GAAA;AACA,IAAA,sBAAA;AACA,IAAA,kBAAA;AACA,IAAA,OAAA;;IA9DX,EAAE,GAAqB,IAAI;AAC3B,IAAA,eAAe,GAAoB,eAAe,CAAC,GAAG;AAC7C,IAAA,YAAY;;AAGZ,IAAA,SAAS,GAAG,IAAI,MAAM,EAA4B;IAC3D,aAAa,GAAG,KAAK;AACZ,IAAA,UAAU;;AAKV,IAAA,aAAa,GAAG,IAAI,MAAM,EAAa;IAChD,iBAAiB,GAAkC,EAAE;;AAGrD,IAAA,SAAS;;AAID,IAAA,QAAQ,GAAG;AACzB,QAAA,IAAI,EAAE,OAAO,OAAkB,KAAmB;AAChD,YAAA,OAAO,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QACxC,CAAC;QAED,QAAQ,EAAE,YAA0B;YAClC,IAAI,IAAI,CAAC,aAAa;gBAAE;AAExB,YAAA,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;AAC5B,YAAA,IAAI;gBACF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC9B;AAAE,YAAA,MAAM;;;;YAIR;AACA,YAAA,IAAI,CAAC,aAAa,GAAG,IAAI;QAC3B,CAAC;KACF;AAEe,IAAA,SAAS,GAA6B;QACpD,CAAC,MAAM,CAAC,aAAa,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE;KAC5D;IAEM,KAAK,GAAA;AACV,QAAA,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QAE1B,IAAI,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC,SAAS,EAAE;;AAEpD,YAAA,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE;YAChB,IAAI,CAAC,OAAO,EAAE;YACd;QACF;QAEA,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC,OAAO,CAAC;YAAE;AAC5D,QAAA,IAAI,CAAC,EAAG,CAAC,KAAK,EAAE;IAClB;AAEA,IAAA,WAAA,CACmB,GAAW,EACX,sBAA0D,EAC1D,kBAAmD,EACnD,UAAiD,EAAE,EAAA;QAHnD,IAAA,CAAA,GAAG,GAAH,GAAG;QACH,IAAA,CAAA,sBAAsB,GAAtB,sBAAsB;QACtB,IAAA,CAAA,kBAAkB,GAAlB,kBAAkB;QAClB,IAAA,CAAA,OAAO,GAAP,OAAO;QAExB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC;QAEzE,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE;AAClD,QAAA,IAAI,CAAC,YAAY,GAAG,IAAIA,4BAAa,CAAC,WAAW,EAAE;YACjD,OAAO,EAAE,MAAK;AACZ,gBAAA,KAAK,IAAI,CAAC,OAAO,EAAE;YACrB,CAAC;YACD,oBAAoB,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC;AACzD,SAAA,CAAC;QAEF,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE;AACrC,YAAA,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC,MAAM,CAAC;YACpD;QACF;AAEA,QAAA,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACvE,IAAI,CAAC,OAAO,EAAE;IAChB;;IAIQ,OAAO,GAAA;AACb,QAAA,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO;YAAE;;QAGvC,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC,UAAU,CAAC;YAAE;AAE/D,QAAA,IAAI;AACF,YAAA,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,eAAe,EAAE;AAEhC,YAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;YACrD,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC1E,YAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AACjE,YAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACzD;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;AACpC,YAAA,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;QAC9B;IACF;IAEQ,eAAe,GAAA;QACrB,MAAM,OAAO,GAAkB,EAAE;AAEjC,QAAA,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ;AACvB,YAAA,OAAO,CAAC,OAAO,GAAG,EAAE,aAAa,EAAE,CAAA,OAAA,EAAU,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAA,CAAE,EAAE;AACxE,QAAA,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU;QAEzE,MAAM,EAAE,GAAG,IAAIC,gBAAS,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC;AAC3C,QAAA,EAAE,CAAC,UAAU,GAAG,aAAa;AAC7B,QAAA,OAAO,EAAE;IACX;IAEQ,MAAM,GAAA;AACZ,QAAA,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC,SAAS,CAAC;QACvD,IAAI,CAAC,gBAAgB,EAAE;IACzB;AAEQ,IAAA,SAAS,CAAC,IAAa,EAAA;AAC7B,QAAA,IAAI,EAAE,IAAI,YAAY,WAAW,CAAC,EAAE;AAClC,YAAA,IAAI,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,CAAA,8BAAA,EAAiC,OAAO,IAAI,CAAA,CAAE,CAAC,CAAC;YAC3E;QACF;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;AAC7D,YAAA,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;QAC/B;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACvC;IACF;AAEQ,IAAA,OAAO,CAAC,KAAc,EAAA;QAC5B,IAAI,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC,SAAS,EAAE;;YAEpD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;AACpC,YAAA,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YAC5B;QACF;QAEA,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC;IAEQ,OAAO,GAAA;AACb,QAAA,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC,MAAM,CAAC;;AAGpD,QAAA,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACxD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM;AAC9C,YAAA,IAAI,MAAM,YAAY,KAAK,EAAE;AAC3B,gBAAA,IAAI,CAAC,SAAS,GAAG,MAAM;YACzB;AAAO,iBAAA,IAAI,MAAM,KAAK,SAAS,EAAE;AAC/B,gBAAA,IAAI,CAAC,SAAS,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;YAC/D;iBAAO;AACL,gBAAA,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,uBAAuB,EAAE;YACjD;QACF;AAEA,QAAA,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YACnB,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC;AAC5D,YAAA,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC;aAAO;AACL,YAAA,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,SAAS,CAAC;QACjD;IACF;;AAIQ,IAAA,WAAW,CAAC,OAAkB,EAAA;AACpC,QAAA,IAAI,IAAI,CAAC,aAAa,EAAE;AACtB,YAAA,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC;QAC1D;QAEA,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE;AACrC,YAAA,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC;QAChD;QAEA,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,KAAI;AAC3C,YAAA,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YACjD,IAAI,CAAC,gBAAgB,EAAE;AACzB,QAAA,CAAC,CAAC;IACJ;IAEQ,gBAAgB,GAAA;AACtB,QAAA,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;YAAE;QAE7B,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAG;AACtC,YAAA,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC;QAChC;IACF;IAEQ,eAAe,GAAA;AACrB,QAAA,OAAO,IAAI,CAAC,eAAe,KAAK,eAAe,CAAC,SAAS;IAC3D;AAEQ,IAAA,iBAAiB,CAAC,MAAgC,EAAA;AACxD,QAAA,IAAI;AACF,YAAA,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE;YAClB,IAAI,CAAC,EAAE,EAAE;AACP,gBAAA,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC;YAC/C;;YAGA,IAAI,EAAE,CAAC,UAAU,KAAKA,gBAAS,CAAC,IAAI,EAAE;gBACpC,MAAM,IAAI,KAAK,CAAC,CAAA,mCAAA,EAAsC,EAAE,CAAC,UAAU,CAAA,CAAA,CAAG,CAAC;YACzE;YAEA,MAAM,MAAM,GAAG,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC;AAC1D,YAAA,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACf,MAAM,CAAC,OAAO,EAAE;QAClB;QAAE,OAAO,KAAK,EAAE;YACd,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC;IACF;AAEQ,IAAA,MAAM,cAAc,GAAA;QAC1B,MAAM,gBAAgB,GAAG,CAAC;QAE1B,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AAChC,YAAA,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,gBAAgB,CAAC;QAClF;IACF;IAEQ,gBAAgB,CAAC,SAAwB,EAAE,UAAkB,EAAA;QACnE,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,KAAI;YAC3C,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE;gBACrC,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAC7F;AAEA,YAAA,IAAI,SAAwC;YAC5C,MAAM,OAAO,GAAG,MAAK;gBACnB,YAAY,CAAC,SAAS,CAAC;gBACvB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;AACvF,YAAA,CAAC;AAED,YAAA,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAE5E,MAAM,KAAK,GAAG,MAAK;gBACjB,IAAI,SAAS,EAAE,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;oBACvC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC;AAC/D,oBAAA,OAAO,EAAE;gBACX;qBAAO;AACL,oBAAA,SAAS,GAAG,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC;gBAC3C;AACF,YAAA,CAAC;AAED,YAAA,KAAK,EAAE;AACT,QAAA,CAAC,CAAC;IACJ;;AAIQ,IAAA,eAAe,CAAC,OAAkB,EAAA;QACxC,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACnD;aAAO;AACL,YAAA,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC;QAClC;IACF;IAEQ,OAAO,sBAAsB,GAAA;QACnC,OAAO,IAAI,EAAE;AACX,YAAA,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE;YAExC,IAAI,MAAM,CAAC,IAAI;gBAAE;YAEjB,MAAM,MAAM,CAAC,KAAK;QACpB;IACF;IAEQ,YAAY,GAAA;QAClB,OAAO,IAAI,OAAO,CAA4B,CAAC,OAAO,EAAE,MAAM,KAAI;;YAEhE,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;gBACjC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAG;gBAC3C,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACxC;YACF;;AAGA,YAAA,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;AACxB,gBAAA,IAAI,IAAI,CAAC,SAAS,EAAE;AAClB,oBAAA,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC;gBACxB;qBAAO;oBACL,OAAO,CAAC,EAAE,KAAK,EAAE,SAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAClD;gBACA;YACF;;YAGA,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAClD,QAAA,CAAC,CAAC;IACJ;IAEQ,0BAA0B,GAAA;QAChC,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC3D;IACF;;AAIQ,IAAA,WAAW,CAAC,KAAY,EAAA;AAC9B,QAAA,IAAI,CAAC,SAAS,GAAG,KAAK;QACtB,IAAI,CAAC,KAAK,EAAE;IACd;AAEQ,IAAA,0BAA0B,CAAC,KAAY,EAAA;AAC7C,QAAA,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;AACnC,QAAA,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC;IACxC;AAEQ,IAAA,uBAAuB,CAAC,KAAY,EAAA;QAC1C,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAG;AACtC,YAAA,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;QACtB;IACF;AAEQ,IAAA,0BAA0B,CAAC,KAAY,EAAA;QAC7C,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC;QACxB;IACF;IAEQ,uBAAuB,GAAA;QAC7B,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE;YACrC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM;AAC9C,YAAA,IAAI,MAAM,YAAY,KAAK,EAAE;AAC3B,gBAAA,OAAO,MAAM;YACf;YACA,OAAO,IAAI,KAAK,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QACvD;AAEA,QAAA,OAAO,IAAI,KAAK,CAAC,eAAe,CAAC;IACnC;;IAIQ,aAAa,GAAA;AACnB,QAAA,QACE,IAAI,CAAC,eAAe,KAAK,eAAe,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,IAAI,KAAK;IAEjG;AAEQ,IAAA,OAAO,CAAC,KAAc,EAAA;QAC5B,IAAI,KAAK,YAAY,KAAK;AAAE,YAAA,OAAO,KAAK;AACxC,QAAA,IAAI,KAAK,YAAYC,iBAAU,EAAE;AAC/B,YAAA,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK;;;YAGvB,IAAI,GAAG,YAAY,SAAS,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;AAC5C,gBAAA,OAAO,IAAIC,wBAAiB,CAAC,0CAA0C,CAAC;YAC1E;YACA,OAAO,GAAG,YAAY,KAAK,GAAG,GAAG,GAAG,IAAI,KAAK,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACpF;QACA,OAAO,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjC;AAEA;;;;;AAKG;AACK,IAAA,uBAAuB,CAAC,QAAyB,EAAA;AACvD,QAAA,IAAI,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE;AACnC,YAAA,OAAO,KAAK;QACd;AACA,QAAA,IAAI,CAAC,eAAe,GAAG,QAAQ;AAC/B,QAAA,OAAO,IAAI;IACb;AACD;;;;"}
+{"version":3,"file":"websocket_stream.cjs","names":["Denque","RetryStrategy","WebSocket","ErrorEvent","DisconnectedError"],"sources":["../../src/core/websocket_stream.ts"],"sourcesContent":["import { WebSocket, type WebSocketInit, type Dispatcher, ErrorEvent } from \"undici\";\nimport type { BiDiStream } from \"./abstract_stream\";\nimport Denque from \"denque\";\nimport type { RetryConfig } from \"../helpers/retry_strategy\";\nimport { RetryStrategy } from \"../helpers/retry_strategy\";\nimport { DisconnectedError } from \"./errors\";\n\ninterface QueuedMessage<InType extends object> {\n  message: InType;\n  resolve: () => void;\n  reject: (error: Error) => void;\n}\n\ninterface ResponseResolver<OutType extends object> {\n  resolve: (value: IteratorResult<OutType>) => void;\n  reject: (error: Error) => void;\n}\n\nenum ConnectionState {\n  NEW = 0,\n  CONNECTING = 1,\n  CONNECTED = 2,\n  CLOSING = 3,\n  CLOSED = 4,\n}\n\nexport type WSStreamOptions<ClientMsg extends object, ServerMsg extends object> = {\n  abortSignal?: AbortSignal;\n\n  dispatcher?: Dispatcher;\n  jwtToken?: string;\n  retryConfig?: Partial<RetryConfig>;\n\n  onComplete?: (stream: WebSocketBiDiStream<ClientMsg, ServerMsg>) => void | Promise<void>;\n};\n\n/**\n * WebSocket-based bidirectional stream implementation for LLTransaction.\n * Implements BiDiStream interface which is compatible with DuplexStreamingCall.\n */\nexport class WebSocketBiDiStream<\n  ClientMsg extends object,\n  ServerMsg extends object,\n> implements BiDiStream<ClientMsg, ServerMsg> {\n  // Connection\n  private ws: WebSocket | null = null;\n  private connectionState: ConnectionState = ConnectionState.NEW;\n  private readonly reconnection: RetryStrategy;\n\n  // Send management\n  private readonly sendQueue = new Denque<QueuedMessage<ClientMsg>>();\n  private sendCompleted = false;\n  private readonly onComplete: (\n    stream: WebSocketBiDiStream<ClientMsg, ServerMsg>,\n  ) => void | Promise<void>;\n\n  // Response management\n  private readonly responseQueue = new Denque<ServerMsg>();\n  private responseResolvers: ResponseResolver<ServerMsg>[] = [];\n\n  // Error tracking\n  private lastError?: Error;\n\n  // Abort listener reference for cleanup\n  private readonly abortHandler?: () => void;\n\n  // === Public API ===\n\n  public readonly requests = {\n    send: async (message: ClientMsg): Promise<void> => {\n      return await this.enqueueSend(message);\n    },\n\n    complete: async (): Promise<void> => {\n      if (this.sendCompleted) return;\n\n      await this.drainSendQueue(); // ensure we sent all already queued messages before closing the stream\n      try {\n        await this.onComplete(this); // custom onComplete may send additional messages\n      } catch {\n        // When 'complete' gets called concurrently with connection break or over a broken\n        // transaction stream (server decided it should drop transaction), server would close\n        // connection anyway on its end. We can safely ignore error here and just continue working.\n      }\n      this.sendCompleted = true;\n    },\n  };\n\n  public readonly responses: AsyncIterable<ServerMsg> = {\n    [Symbol.asyncIterator]: () => this.createResponseIterator(),\n  };\n\n  public close(): void {\n    this.reconnection.cancel();\n\n    if (this.connectionState < ConnectionState.CONNECTED) {\n      // Never reached CONNECTED state. ws.close() will never trigger 'close' event.\n      this.ws?.close();\n      this.onClose();\n      return;\n    }\n\n    if (!this.progressConnectionState(ConnectionState.CLOSING)) return;\n    this.ws!.close();\n  }\n\n  constructor(\n    private readonly url: string,\n    private readonly serializeClientMessage: (message: ClientMsg) => Uint8Array,\n    private readonly parseServerMessage: (data: Uint8Array) => ServerMsg,\n    private readonly options: WSStreamOptions<ClientMsg, ServerMsg> = {},\n  ) {\n    this.onComplete = this.options.onComplete ?? ((stream) => stream.close());\n\n    const retryConfig = this.options.retryConfig ?? {};\n    this.reconnection = new RetryStrategy(retryConfig, {\n      onRetry: () => {\n        void this.connect();\n      },\n      onMaxAttemptsReached: (error) => this.handleError(error),\n    });\n\n    if (this.options.abortSignal?.aborted) {\n      this.progressConnectionState(ConnectionState.CLOSED);\n      return;\n    }\n\n    this.abortHandler = () => this.close();\n    this.options.abortSignal?.addEventListener(\"abort\", this.abortHandler, { once: true });\n    this.connect();\n  }\n\n  // === Connection Lifecycle ===\n\n  private connect(): void {\n    if (this.options.abortSignal?.aborted) return;\n\n    // Prevent reconnecting after first successful connection.\n    if (!this.progressConnectionState(ConnectionState.CONNECTING)) return;\n\n    try {\n      this.ws = this.createWebSocket();\n\n      this.ws.addEventListener(\"open\", () => this.onOpen());\n      this.ws.addEventListener(\"message\", (event) => this.onMessage(event.data));\n      this.ws.addEventListener(\"error\", (error) => this.onError(error));\n      this.ws.addEventListener(\"close\", () => this.onClose());\n    } catch (error) {\n      this.lastError = this.toError(error);\n      this.reconnection.schedule();\n    }\n  }\n\n  private createWebSocket(): WebSocket {\n    const options: WebSocketInit = {};\n\n    if (this.options.jwtToken)\n      options.headers = { authorization: `Bearer ${this.options.jwtToken}` };\n    if (this.options.dispatcher) options.dispatcher = this.options.dispatcher;\n\n    const ws = new WebSocket(this.url, options);\n    ws.binaryType = \"arraybuffer\";\n    return ws;\n  }\n\n  private onOpen(): void {\n    this.progressConnectionState(ConnectionState.CONNECTED);\n    this.processSendQueue();\n  }\n\n  private onMessage(data: unknown): void {\n    if (!(data instanceof ArrayBuffer)) {\n      this.handleError(new Error(`Unexpected WS message format: ${typeof data}`));\n      return;\n    }\n\n    try {\n      const message = this.parseServerMessage(new Uint8Array(data));\n      this.deliverResponse(message);\n    } catch (error) {\n      this.handleError(this.toError(error));\n    }\n  }\n\n  private onError(error: unknown): void {\n    if (this.connectionState < ConnectionState.CONNECTED) {\n      // Try to connect several times until we succeed or run out of attempts.\n      this.lastError = this.toError(error);\n      this.reconnection.schedule();\n      return;\n    }\n\n    this.handleError(this.toError(error));\n  }\n\n  private onClose(): void {\n    this.progressConnectionState(ConnectionState.CLOSED);\n\n    // Clean up abort listener to prevent memory leaks\n    if (this.abortHandler) {\n      this.options.abortSignal?.removeEventListener(\"abort\", this.abortHandler);\n    }\n\n    // If abort signal was triggered, use that as the error source\n    if (this.options.abortSignal?.aborted && !this.lastError) {\n      const reason = this.options.abortSignal.reason;\n      if (reason instanceof Error) {\n        this.lastError = reason;\n      } else if (reason !== undefined) {\n        this.lastError = new Error(String(reason), { cause: reason });\n      } else {\n        this.lastError = this.createStreamClosedError();\n      }\n    }\n\n    if (!this.lastError) {\n      this.rejectAllSendOperations(this.createStreamClosedError());\n      this.resolveAllPendingResponses(); // unblock active async iterator\n    } else {\n      this.rejectAllPendingOperations(this.lastError);\n    }\n  }\n\n  // === Send Queue Management ===\n\n  private enqueueSend(message: ClientMsg): Promise<void> {\n    if (this.sendCompleted) {\n      throw new Error(\"Cannot send: stream already completed\");\n    }\n\n    if (this.options.abortSignal?.aborted) {\n      throw new Error(\"Cannot send: stream aborted\");\n    }\n\n    return new Promise<void>((resolve, reject) => {\n      this.sendQueue.push({ message, resolve, reject });\n      this.processSendQueue();\n    });\n  }\n\n  private processSendQueue(): void {\n    if (!this.canSendMessages()) return;\n\n    while (this.sendQueue.length > 0) {\n      const queued = this.sendQueue.shift()!;\n      this.sendQueuedMessage(queued);\n    }\n  }\n\n  private canSendMessages(): boolean {\n    return this.connectionState === ConnectionState.CONNECTED;\n  }\n\n  private sendQueuedMessage(queued: QueuedMessage<ClientMsg>): void {\n    try {\n      const ws = this.ws;\n      if (!ws) {\n        throw new Error(\"WebSocket is not connected\");\n      }\n\n      // Check if WebSocket is in a valid state for sending\n      if (ws.readyState !== WebSocket.OPEN) {\n        throw new Error(`WebSocket is not open (readyState: ${ws.readyState})`);\n      }\n\n      const binary = this.serializeClientMessage(queued.message);\n      ws.send(binary);\n      queued.resolve();\n    } catch (error) {\n      queued.reject(this.toError(error));\n    }\n  }\n\n  private async drainSendQueue(): Promise<void> {\n    const POLL_INTERVAL_MS = 5;\n\n    while (this.sendQueue.length > 0) {\n      await this.waitForCondition(() => this.sendQueue.length === 0, POLL_INTERVAL_MS);\n    }\n  }\n\n  private waitForCondition(condition: () => boolean, intervalMs: number): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      if (this.options.abortSignal?.aborted) {\n        return reject(this.toError(this.options.abortSignal.reason) ?? new Error(\"Stream aborted\"));\n      }\n\n      let timeoutId: ReturnType<typeof setTimeout>;\n      const onAbort = () => {\n        clearTimeout(timeoutId);\n        reject(this.toError(this.options.abortSignal?.reason) ?? new Error(\"Stream aborted\"));\n      };\n\n      this.options.abortSignal?.addEventListener(\"abort\", onAbort, { once: true });\n\n      const check = () => {\n        if (condition() || this.isStreamEnded()) {\n          this.options.abortSignal?.removeEventListener(\"abort\", onAbort);\n          resolve();\n        } else {\n          timeoutId = setTimeout(check, intervalMs);\n        }\n      };\n\n      check();\n    });\n  }\n\n  // === Response Delivery ===\n\n  private deliverResponse(message: ServerMsg): void {\n    if (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.resolve({ value: message, done: false });\n    } else {\n      this.responseQueue.push(message);\n    }\n  }\n\n  private async *createResponseIterator(): AsyncIterator<ServerMsg> {\n    while (true) {\n      const result = await this.nextResponse();\n\n      if (result.done) break;\n\n      yield result.value;\n    }\n  }\n\n  private nextResponse(): Promise<IteratorResult<ServerMsg>> {\n    return new Promise<IteratorResult<ServerMsg>>((resolve, reject) => {\n      // Fast path: message already available\n      if (this.responseQueue.length > 0) {\n        const message = this.responseQueue.shift()!;\n        resolve({ value: message, done: false });\n        return;\n      }\n\n      // Stream ended\n      if (this.isStreamEnded()) {\n        if (this.lastError) {\n          reject(this.lastError);\n        } else {\n          resolve({ value: undefined as any, done: true });\n        }\n        return;\n      }\n\n      // Wait for next message\n      this.responseResolvers.push({ resolve, reject });\n    });\n  }\n\n  private resolveAllPendingResponses(): void {\n    while (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.resolve({ value: undefined as any, done: true });\n    }\n  }\n\n  // === Error Handling ===\n\n  private handleError(error: Error): void {\n    this.lastError = error;\n    this.close();\n  }\n\n  private rejectAllPendingOperations(error: Error): void {\n    this.rejectAllSendOperations(error);\n    this.rejectAllResponseResolvers(error);\n  }\n\n  private rejectAllSendOperations(error: Error): void {\n    while (this.sendQueue.length > 0) {\n      const queued = this.sendQueue.shift()!;\n      queued.reject(error);\n    }\n  }\n\n  private rejectAllResponseResolvers(error: Error): void {\n    while (this.responseResolvers.length > 0) {\n      const resolver = this.responseResolvers.shift()!;\n      resolver.reject(error);\n    }\n  }\n\n  private createStreamClosedError(): Error {\n    if (this.options.abortSignal?.aborted) {\n      const reason = this.options.abortSignal.reason;\n      if (reason instanceof Error) {\n        return reason;\n      }\n      return new Error(\"Stream aborted\", { cause: reason });\n    }\n\n    return new Error(\"Stream closed\");\n  }\n\n  // === Helpers ===\n\n  private isStreamEnded(): boolean {\n    return (\n      this.connectionState === ConnectionState.CLOSED || this.options.abortSignal?.aborted || false\n    );\n  }\n\n  private toError(error: unknown): Error {\n    if (error instanceof Error) return error;\n    if (error instanceof ErrorEvent) {\n      const err = error.error;\n      // undici WebSocket throws TypeError with empty message on socket close\n      // (e.g., when connection is lost or server disconnects)\n      if (err instanceof TypeError && !err.message) {\n        return new DisconnectedError(\"WebSocket connection closed unexpectedly\");\n      }\n      return err instanceof Error ? err : new Error(\"WebSocket error\", { cause: error });\n    }\n    return new Error(String(error));\n  }\n\n  /**\n   * Connection state progresses linearly from NEW to CLOSED and never goes back.\n   * This internal contract dramatically simplifies the internal stream state management.\n   *\n   * If you ever feel the need to make this contract less strict, think twice.\n   */\n  private progressConnectionState(newState: ConnectionState): boolean {\n    if (newState < this.connectionState) {\n      return false;\n    }\n    this.connectionState = newState;\n    return true;\n  }\n}\n"],"mappings":";;;;;;;;AAkBA,IAAK,kBAAL;AACE;AACA;AACA;AACA;AACA;;EALG,sBAMJ;;;;;AAgBD,IAAa,sBAAb,MAG8C;CAE5C,AAAQ,KAAuB;CAC/B,AAAQ,kBAAmC,gBAAgB;CAC3D,AAAiB;CAGjB,AAAiB,YAAY,IAAIA,gBAAkC;CACnE,AAAQ,gBAAgB;CACxB,AAAiB;CAKjB,AAAiB,gBAAgB,IAAIA,gBAAmB;CACxD,AAAQ,oBAAmD,EAAE;CAG7D,AAAQ;CAGR,AAAiB;CAIjB,AAAgB,WAAW;EACzB,MAAM,OAAO,YAAsC;AACjD,UAAO,MAAM,KAAK,YAAY,QAAQ;;EAGxC,UAAU,YAA2B;AACnC,OAAI,KAAK,cAAe;AAExB,SAAM,KAAK,gBAAgB;AAC3B,OAAI;AACF,UAAM,KAAK,WAAW,KAAK;WACrB;AAKR,QAAK,gBAAgB;;EAExB;CAED,AAAgB,YAAsC,GACnD,OAAO,sBAAsB,KAAK,wBAAwB,EAC5D;CAED,AAAO,QAAc;AACnB,OAAK,aAAa,QAAQ;AAE1B,MAAI,KAAK,kBAAkB,gBAAgB,WAAW;AAEpD,QAAK,IAAI,OAAO;AAChB,QAAK,SAAS;AACd;;AAGF,MAAI,CAAC,KAAK,wBAAwB,gBAAgB,QAAQ,CAAE;AAC5D,OAAK,GAAI,OAAO;;CAGlB,YACE,AAAiB,KACjB,AAAiB,wBACjB,AAAiB,oBACjB,AAAiB,UAAiD,EAAE,EACpE;EAJiB;EACA;EACA;EACA;AAEjB,OAAK,aAAa,KAAK,QAAQ,gBAAgB,WAAW,OAAO,OAAO;AAGxE,OAAK,eAAe,IAAIC,qCADJ,KAAK,QAAQ,eAAe,EAAE,EACC;GACjD,eAAe;AACb,IAAK,KAAK,SAAS;;GAErB,uBAAuB,UAAU,KAAK,YAAY,MAAM;GACzD,CAAC;AAEF,MAAI,KAAK,QAAQ,aAAa,SAAS;AACrC,QAAK,wBAAwB,gBAAgB,OAAO;AACpD;;AAGF,OAAK,qBAAqB,KAAK,OAAO;AACtC,OAAK,QAAQ,aAAa,iBAAiB,SAAS,KAAK,cAAc,EAAE,MAAM,MAAM,CAAC;AACtF,OAAK,SAAS;;CAKhB,AAAQ,UAAgB;AACtB,MAAI,KAAK,QAAQ,aAAa,QAAS;AAGvC,MAAI,CAAC,KAAK,wBAAwB,gBAAgB,WAAW,CAAE;AAE/D,MAAI;AACF,QAAK,KAAK,KAAK,iBAAiB;AAEhC,QAAK,GAAG,iBAAiB,cAAc,KAAK,QAAQ,CAAC;AACrD,QAAK,GAAG,iBAAiB,YAAY,UAAU,KAAK,UAAU,MAAM,KAAK,CAAC;AAC1E,QAAK,GAAG,iBAAiB,UAAU,UAAU,KAAK,QAAQ,MAAM,CAAC;AACjE,QAAK,GAAG,iBAAiB,eAAe,KAAK,SAAS,CAAC;WAChD,OAAO;AACd,QAAK,YAAY,KAAK,QAAQ,MAAM;AACpC,QAAK,aAAa,UAAU;;;CAIhC,AAAQ,kBAA6B;EACnC,MAAM,UAAyB,EAAE;AAEjC,MAAI,KAAK,QAAQ,SACf,SAAQ,UAAU,EAAE,eAAe,UAAU,KAAK,QAAQ,YAAY;AACxE,MAAI,KAAK,QAAQ,WAAY,SAAQ,aAAa,KAAK,QAAQ;EAE/D,MAAM,KAAK,IAAIC,iBAAU,KAAK,KAAK,QAAQ;AAC3C,KAAG,aAAa;AAChB,SAAO;;CAGT,AAAQ,SAAe;AACrB,OAAK,wBAAwB,gBAAgB,UAAU;AACvD,OAAK,kBAAkB;;CAGzB,AAAQ,UAAU,MAAqB;AACrC,MAAI,EAAE,gBAAgB,cAAc;AAClC,QAAK,4BAAY,IAAI,MAAM,iCAAiC,OAAO,OAAO,CAAC;AAC3E;;AAGF,MAAI;GACF,MAAM,UAAU,KAAK,mBAAmB,IAAI,WAAW,KAAK,CAAC;AAC7D,QAAK,gBAAgB,QAAQ;WACtB,OAAO;AACd,QAAK,YAAY,KAAK,QAAQ,MAAM,CAAC;;;CAIzC,AAAQ,QAAQ,OAAsB;AACpC,MAAI,KAAK,kBAAkB,gBAAgB,WAAW;AAEpD,QAAK,YAAY,KAAK,QAAQ,MAAM;AACpC,QAAK,aAAa,UAAU;AAC5B;;AAGF,OAAK,YAAY,KAAK,QAAQ,MAAM,CAAC;;CAGvC,AAAQ,UAAgB;AACtB,OAAK,wBAAwB,gBAAgB,OAAO;AAGpD,MAAI,KAAK,aACP,MAAK,QAAQ,aAAa,oBAAoB,SAAS,KAAK,aAAa;AAI3E,MAAI,KAAK,QAAQ,aAAa,WAAW,CAAC,KAAK,WAAW;GACxD,MAAM,SAAS,KAAK,QAAQ,YAAY;AACxC,OAAI,kBAAkB,MACpB,MAAK,YAAY;YACR,WAAW,OACpB,MAAK,YAAY,IAAI,MAAM,OAAO,OAAO,EAAE,EAAE,OAAO,QAAQ,CAAC;OAE7D,MAAK,YAAY,KAAK,yBAAyB;;AAInD,MAAI,CAAC,KAAK,WAAW;AACnB,QAAK,wBAAwB,KAAK,yBAAyB,CAAC;AAC5D,QAAK,4BAA4B;QAEjC,MAAK,2BAA2B,KAAK,UAAU;;CAMnD,AAAQ,YAAY,SAAmC;AACrD,MAAI,KAAK,cACP,OAAM,IAAI,MAAM,wCAAwC;AAG1D,MAAI,KAAK,QAAQ,aAAa,QAC5B,OAAM,IAAI,MAAM,8BAA8B;AAGhD,SAAO,IAAI,SAAe,SAAS,WAAW;AAC5C,QAAK,UAAU,KAAK;IAAE;IAAS;IAAS;IAAQ,CAAC;AACjD,QAAK,kBAAkB;IACvB;;CAGJ,AAAQ,mBAAyB;AAC/B,MAAI,CAAC,KAAK,iBAAiB,CAAE;AAE7B,SAAO,KAAK,UAAU,SAAS,GAAG;GAChC,MAAM,SAAS,KAAK,UAAU,OAAO;AACrC,QAAK,kBAAkB,OAAO;;;CAIlC,AAAQ,kBAA2B;AACjC,SAAO,KAAK,oBAAoB,gBAAgB;;CAGlD,AAAQ,kBAAkB,QAAwC;AAChE,MAAI;GACF,MAAM,KAAK,KAAK;AAChB,OAAI,CAAC,GACH,OAAM,IAAI,MAAM,6BAA6B;AAI/C,OAAI,GAAG,eAAeA,iBAAU,KAC9B,OAAM,IAAI,MAAM,sCAAsC,GAAG,WAAW,GAAG;GAGzE,MAAM,SAAS,KAAK,uBAAuB,OAAO,QAAQ;AAC1D,MAAG,KAAK,OAAO;AACf,UAAO,SAAS;WACT,OAAO;AACd,UAAO,OAAO,KAAK,QAAQ,MAAM,CAAC;;;CAItC,MAAc,iBAAgC;EAC5C,MAAM,mBAAmB;AAEzB,SAAO,KAAK,UAAU,SAAS,EAC7B,OAAM,KAAK,uBAAuB,KAAK,UAAU,WAAW,GAAG,iBAAiB;;CAIpF,AAAQ,iBAAiB,WAA0B,YAAmC;AACpF,SAAO,IAAI,SAAe,SAAS,WAAW;AAC5C,OAAI,KAAK,QAAQ,aAAa,QAC5B,QAAO,OAAO,KAAK,QAAQ,KAAK,QAAQ,YAAY,OAAO,oBAAI,IAAI,MAAM,iBAAiB,CAAC;GAG7F,IAAI;GACJ,MAAM,gBAAgB;AACpB,iBAAa,UAAU;AACvB,WAAO,KAAK,QAAQ,KAAK,QAAQ,aAAa,OAAO,oBAAI,IAAI,MAAM,iBAAiB,CAAC;;AAGvF,QAAK,QAAQ,aAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;GAE5E,MAAM,cAAc;AAClB,QAAI,WAAW,IAAI,KAAK,eAAe,EAAE;AACvC,UAAK,QAAQ,aAAa,oBAAoB,SAAS,QAAQ;AAC/D,cAAS;UAET,aAAY,WAAW,OAAO,WAAW;;AAI7C,UAAO;IACP;;CAKJ,AAAQ,gBAAgB,SAA0B;AAChD,MAAI,KAAK,kBAAkB,SAAS,EAElC,CADiB,KAAK,kBAAkB,OAAO,CACtC,QAAQ;GAAE,OAAO;GAAS,MAAM;GAAO,CAAC;MAEjD,MAAK,cAAc,KAAK,QAAQ;;CAIpC,OAAe,yBAAmD;AAChE,SAAO,MAAM;GACX,MAAM,SAAS,MAAM,KAAK,cAAc;AAExC,OAAI,OAAO,KAAM;AAEjB,SAAM,OAAO;;;CAIjB,AAAQ,eAAmD;AACzD,SAAO,IAAI,SAAoC,SAAS,WAAW;AAEjE,OAAI,KAAK,cAAc,SAAS,GAAG;AAEjC,YAAQ;KAAE,OADM,KAAK,cAAc,OAAO;KAChB,MAAM;KAAO,CAAC;AACxC;;AAIF,OAAI,KAAK,eAAe,EAAE;AACxB,QAAI,KAAK,UACP,QAAO,KAAK,UAAU;QAEtB,SAAQ;KAAE,OAAO;KAAkB,MAAM;KAAM,CAAC;AAElD;;AAIF,QAAK,kBAAkB,KAAK;IAAE;IAAS;IAAQ,CAAC;IAChD;;CAGJ,AAAQ,6BAAmC;AACzC,SAAO,KAAK,kBAAkB,SAAS,EAErC,CADiB,KAAK,kBAAkB,OAAO,CACtC,QAAQ;GAAE,OAAO;GAAkB,MAAM;GAAM,CAAC;;CAM7D,AAAQ,YAAY,OAAoB;AACtC,OAAK,YAAY;AACjB,OAAK,OAAO;;CAGd,AAAQ,2BAA2B,OAAoB;AACrD,OAAK,wBAAwB,MAAM;AACnC,OAAK,2BAA2B,MAAM;;CAGxC,AAAQ,wBAAwB,OAAoB;AAClD,SAAO,KAAK,UAAU,SAAS,EAE7B,CADe,KAAK,UAAU,OAAO,CAC9B,OAAO,MAAM;;CAIxB,AAAQ,2BAA2B,OAAoB;AACrD,SAAO,KAAK,kBAAkB,SAAS,EAErC,CADiB,KAAK,kBAAkB,OAAO,CACtC,OAAO,MAAM;;CAI1B,AAAQ,0BAAiC;AACvC,MAAI,KAAK,QAAQ,aAAa,SAAS;GACrC,MAAM,SAAS,KAAK,QAAQ,YAAY;AACxC,OAAI,kBAAkB,MACpB,QAAO;AAET,UAAO,IAAI,MAAM,kBAAkB,EAAE,OAAO,QAAQ,CAAC;;AAGvD,yBAAO,IAAI,MAAM,gBAAgB;;CAKnC,AAAQ,gBAAyB;AAC/B,SACE,KAAK,oBAAoB,gBAAgB,UAAU,KAAK,QAAQ,aAAa,WAAW;;CAI5F,AAAQ,QAAQ,OAAuB;AACrC,MAAI,iBAAiB,MAAO,QAAO;AACnC,MAAI,iBAAiBC,mBAAY;GAC/B,MAAM,MAAM,MAAM;AAGlB,OAAI,eAAe,aAAa,CAAC,IAAI,QACnC,QAAO,IAAIC,iCAAkB,2CAA2C;AAE1E,UAAO,eAAe,QAAQ,MAAM,IAAI,MAAM,mBAAmB,EAAE,OAAO,OAAO,CAAC;;AAEpF,SAAO,IAAI,MAAM,OAAO,MAAM,CAAC;;;;;;;;CASjC,AAAQ,wBAAwB,UAAoC;AAClE,MAAI,WAAW,KAAK,gBAClB,QAAO;AAET,OAAK,kBAAkB;AACvB,SAAO"}