@milaboratories/pl-client 2.16.11 → 2.16.13

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (54) hide show
  1. package/dist/core/driver.cjs +1 -1
  2. package/dist/core/driver.cjs.map +1 -1
  3. package/dist/core/driver.js +1 -1
  4. package/dist/core/driver.js.map +1 -1
  5. package/dist/core/errors.cjs +2 -0
  6. package/dist/core/errors.cjs.map +1 -1
  7. package/dist/core/errors.d.ts.map +1 -1
  8. package/dist/core/errors.js +2 -0
  9. package/dist/core/errors.js.map +1 -1
  10. package/dist/core/ll_client.cjs +32 -9
  11. package/dist/core/ll_client.cjs.map +1 -1
  12. package/dist/core/ll_client.d.ts.map +1 -1
  13. package/dist/core/ll_client.js +32 -9
  14. package/dist/core/ll_client.js.map +1 -1
  15. package/dist/core/ll_transaction.cjs +10 -0
  16. package/dist/core/ll_transaction.cjs.map +1 -1
  17. package/dist/core/ll_transaction.d.ts +1 -0
  18. package/dist/core/ll_transaction.d.ts.map +1 -1
  19. package/dist/core/ll_transaction.js +10 -0
  20. package/dist/core/ll_transaction.js.map +1 -1
  21. package/dist/core/websocket_stream.cjs +330 -0
  22. package/dist/core/websocket_stream.cjs.map +1 -0
  23. package/dist/core/websocket_stream.d.ts +67 -0
  24. package/dist/core/websocket_stream.d.ts.map +1 -0
  25. package/dist/core/websocket_stream.js +328 -0
  26. package/dist/core/websocket_stream.js.map +1 -0
  27. package/dist/helpers/retry_strategy.cjs +92 -0
  28. package/dist/helpers/retry_strategy.cjs.map +1 -0
  29. package/dist/helpers/retry_strategy.d.ts +24 -0
  30. package/dist/helpers/retry_strategy.d.ts.map +1 -0
  31. package/dist/helpers/retry_strategy.js +89 -0
  32. package/dist/helpers/retry_strategy.js.map +1 -0
  33. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.cjs +136 -0
  34. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.cjs.map +1 -1
  35. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.d.ts +75 -1
  36. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.d.ts.map +1 -1
  37. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.js +135 -1
  38. package/dist/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.js.map +1 -1
  39. package/dist/proto-rest/index.cjs +16 -2
  40. package/dist/proto-rest/index.cjs.map +1 -1
  41. package/dist/proto-rest/index.d.ts.map +1 -1
  42. package/dist/proto-rest/index.js +16 -2
  43. package/dist/proto-rest/index.js.map +1 -1
  44. package/package.json +6 -6
  45. package/src/core/driver.ts +1 -1
  46. package/src/core/errors.ts +1 -0
  47. package/src/core/ll_client.ts +42 -9
  48. package/src/core/ll_transaction.test.ts +18 -0
  49. package/src/core/ll_transaction.ts +12 -0
  50. package/src/core/websocket_stream.test.ts +423 -0
  51. package/src/core/websocket_stream.ts +400 -0
  52. package/src/helpers/retry_strategy.ts +123 -0
  53. package/src/proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.ts +179 -1
  54. package/src/proto-rest/index.ts +17 -2
@@ -0,0 +1,330 @@
1
+ 'use strict';
2
+
3
+ var undici = require('undici');
4
+ var Denque = require('denque');
5
+ var retry_strategy = require('../helpers/retry_strategy.cjs');
6
+
7
+ var ConnectionState;
8
+ (function (ConnectionState) {
9
+ ConnectionState[ConnectionState["NEW"] = 0] = "NEW";
10
+ ConnectionState[ConnectionState["CONNECTING"] = 1] = "CONNECTING";
11
+ ConnectionState[ConnectionState["CONNECTED"] = 2] = "CONNECTED";
12
+ ConnectionState[ConnectionState["CLOSING"] = 3] = "CLOSING";
13
+ ConnectionState[ConnectionState["CLOSED"] = 4] = "CLOSED";
14
+ })(ConnectionState || (ConnectionState = {}));
15
+ /**
16
+ * WebSocket-based bidirectional stream implementation for LLTransaction.
17
+ * Implements BiDiStream interface which is compatible with DuplexStreamingCall.
18
+ */
19
+ class WebSocketBiDiStream {
20
+ url;
21
+ serializeClientMessage;
22
+ parseServerMessage;
23
+ options;
24
+ // Connection
25
+ ws = null;
26
+ connectionState = ConnectionState.NEW;
27
+ reconnection;
28
+ // Send management
29
+ sendQueue = new Denque();
30
+ sendCompleted = false;
31
+ onComplete;
32
+ // Response management
33
+ responseQueue = new Denque();
34
+ responseResolvers = [];
35
+ // Error tracking
36
+ lastError;
37
+ // === Public API ===
38
+ requests = {
39
+ send: async (message) => {
40
+ return await this.enqueueSend(message);
41
+ },
42
+ complete: async () => {
43
+ if (this.sendCompleted)
44
+ return;
45
+ await this.drainSendQueue(); // ensure we sent all already queued messages before closing the stream
46
+ try {
47
+ await this.onComplete(this); // custom onComplete may send additional messages
48
+ }
49
+ catch (_) {
50
+ // When 'complete' gets called concurrently with connection break or over a broken
51
+ // transaction stream (server decided it should drop transaction), server would close
52
+ // connection anyway on its end. We can safely ignore error here and just continue working.
53
+ }
54
+ this.sendCompleted = true;
55
+ },
56
+ };
57
+ responses = {
58
+ [Symbol.asyncIterator]: () => this.createResponseIterator(),
59
+ };
60
+ close() {
61
+ this.reconnection.cancel();
62
+ if (this.connectionState < ConnectionState.CONNECTED) {
63
+ // Never reached CONNECTED state. ws.close() will never trigger 'close' event.
64
+ this.ws?.close();
65
+ this.onClose();
66
+ return;
67
+ }
68
+ if (!this.progressConnectionState(ConnectionState.CLOSING))
69
+ return;
70
+ this.ws.close();
71
+ }
72
+ constructor(url, serializeClientMessage, parseServerMessage, options = {}) {
73
+ this.url = url;
74
+ this.serializeClientMessage = serializeClientMessage;
75
+ this.parseServerMessage = parseServerMessage;
76
+ this.options = options;
77
+ this.onComplete = this.options.onComplete ?? ((stream) => stream.close());
78
+ const retryConfig = this.options.retryConfig ?? {};
79
+ this.reconnection = new retry_strategy.RetryStrategy(retryConfig, {
80
+ onRetry: () => { void this.connect(); },
81
+ onMaxAttemptsReached: (error) => this.handleError(error),
82
+ });
83
+ if (this.options.abortSignal?.aborted) {
84
+ this.progressConnectionState(ConnectionState.CLOSED);
85
+ return;
86
+ }
87
+ this.options.abortSignal?.addEventListener('abort', () => this.close());
88
+ this.connect();
89
+ }
90
+ // === Connection Lifecycle ===
91
+ connect() {
92
+ if (this.options.abortSignal?.aborted)
93
+ return;
94
+ // Prevent reconnecting after first successful connection.
95
+ if (!this.progressConnectionState(ConnectionState.CONNECTING))
96
+ return;
97
+ try {
98
+ this.ws = this.createWebSocket();
99
+ this.ws.addEventListener('open', () => this.onOpen());
100
+ this.ws.addEventListener('message', (event) => this.onMessage(event.data));
101
+ this.ws.addEventListener('error', (error) => this.onError(error));
102
+ this.ws.addEventListener('close', () => this.onClose());
103
+ }
104
+ catch (error) {
105
+ this.lastError = this.toError(error);
106
+ this.reconnection.schedule();
107
+ }
108
+ }
109
+ createWebSocket() {
110
+ const options = {};
111
+ if (this.options.jwtToken)
112
+ options.headers = { authorization: `Bearer ${this.options.jwtToken}` };
113
+ if (this.options.dispatcher)
114
+ options.dispatcher = this.options.dispatcher;
115
+ const ws = new undici.WebSocket(this.url, options);
116
+ ws.binaryType = 'arraybuffer';
117
+ return ws;
118
+ }
119
+ onOpen() {
120
+ this.progressConnectionState(ConnectionState.CONNECTED);
121
+ this.processSendQueue();
122
+ }
123
+ onMessage(data) {
124
+ if (!(data instanceof ArrayBuffer)) {
125
+ this.handleError(new Error(`Unexpected WS message format: ${typeof data}`));
126
+ return;
127
+ }
128
+ try {
129
+ const message = this.parseServerMessage(new Uint8Array(data));
130
+ this.deliverResponse(message);
131
+ }
132
+ catch (error) {
133
+ this.handleError(this.toError(error));
134
+ }
135
+ }
136
+ onError(error) {
137
+ if (this.connectionState < ConnectionState.CONNECTED) {
138
+ // Try to connect several times until we succeed or run out of attempts.
139
+ this.lastError = this.toError(error);
140
+ this.reconnection.schedule();
141
+ return;
142
+ }
143
+ this.handleError(this.toError(error));
144
+ }
145
+ onClose() {
146
+ this.progressConnectionState(ConnectionState.CLOSED);
147
+ if (!this.lastError) {
148
+ this.rejectAllSendOperations(this.createStreamClosedError());
149
+ this.resolveAllPendingResponses(); // unblock active async iterator
150
+ }
151
+ else {
152
+ this.rejectAllPendingOperations(this.lastError);
153
+ }
154
+ }
155
+ // === Send Queue Management ===
156
+ enqueueSend(message) {
157
+ if (this.sendCompleted) {
158
+ throw new Error('Cannot send: stream already completed');
159
+ }
160
+ if (this.options.abortSignal?.aborted) {
161
+ throw new Error('Cannot send: stream aborted');
162
+ }
163
+ return new Promise((resolve, reject) => {
164
+ this.sendQueue.push({ message, resolve, reject });
165
+ this.processSendQueue();
166
+ });
167
+ }
168
+ processSendQueue() {
169
+ if (!this.canSendMessages())
170
+ return;
171
+ while (this.sendQueue.length > 0) {
172
+ const queued = this.sendQueue.shift();
173
+ this.sendQueuedMessage(queued);
174
+ }
175
+ }
176
+ canSendMessages() {
177
+ return this.connectionState === ConnectionState.CONNECTED;
178
+ }
179
+ sendQueuedMessage(queued) {
180
+ try {
181
+ const ws = this.ws;
182
+ if (!ws) {
183
+ throw new Error('WebSocket is not connected');
184
+ }
185
+ // Check if WebSocket is in a valid state for sending
186
+ if (ws.readyState !== undici.WebSocket.OPEN) {
187
+ throw new Error(`WebSocket is not open (readyState: ${ws.readyState})`);
188
+ }
189
+ const binary = this.serializeClientMessage(queued.message);
190
+ ws.send(binary);
191
+ queued.resolve();
192
+ }
193
+ catch (error) {
194
+ queued.reject(this.toError(error));
195
+ }
196
+ }
197
+ async drainSendQueue() {
198
+ const POLL_INTERVAL_MS = 5;
199
+ while (this.sendQueue.length > 0) {
200
+ await this.waitForCondition(() => this.sendQueue.length === 0, POLL_INTERVAL_MS);
201
+ }
202
+ }
203
+ waitForCondition(condition, intervalMs) {
204
+ return new Promise((resolve, reject) => {
205
+ if (this.options.abortSignal?.aborted) {
206
+ return reject(this.toError(this.options.abortSignal.reason) ?? new Error('Stream aborted'));
207
+ }
208
+ let timeoutId;
209
+ const onAbort = () => {
210
+ clearTimeout(timeoutId);
211
+ reject(this.toError(this.options.abortSignal?.reason) ?? new Error('Stream aborted'));
212
+ };
213
+ this.options.abortSignal?.addEventListener('abort', onAbort, { once: true });
214
+ const check = () => {
215
+ if (condition() || this.isStreamEnded()) {
216
+ this.options.abortSignal?.removeEventListener('abort', onAbort);
217
+ resolve();
218
+ }
219
+ else {
220
+ timeoutId = setTimeout(check, intervalMs);
221
+ }
222
+ };
223
+ check();
224
+ });
225
+ }
226
+ // === Response Delivery ===
227
+ deliverResponse(message) {
228
+ if (this.responseResolvers.length > 0) {
229
+ const resolver = this.responseResolvers.shift();
230
+ resolver.resolve({ value: message, done: false });
231
+ }
232
+ else {
233
+ this.responseQueue.push(message);
234
+ }
235
+ }
236
+ async *createResponseIterator() {
237
+ while (true) {
238
+ const result = await this.nextResponse();
239
+ if (result.done)
240
+ break;
241
+ yield result.value;
242
+ }
243
+ }
244
+ nextResponse() {
245
+ return new Promise((resolve, reject) => {
246
+ // Fast path: message already available
247
+ if (this.responseQueue.length > 0) {
248
+ const message = this.responseQueue.shift();
249
+ resolve({ value: message, done: false });
250
+ return;
251
+ }
252
+ // Stream ended
253
+ if (this.isStreamEnded()) {
254
+ if (this.lastError) {
255
+ reject(this.lastError);
256
+ }
257
+ else {
258
+ resolve({ value: undefined, done: true });
259
+ }
260
+ return;
261
+ }
262
+ // Wait for next message
263
+ this.responseResolvers.push({ resolve, reject });
264
+ });
265
+ }
266
+ resolveAllPendingResponses() {
267
+ while (this.responseResolvers.length > 0) {
268
+ const resolver = this.responseResolvers.shift();
269
+ resolver.resolve({ value: undefined, done: true });
270
+ }
271
+ }
272
+ // === Error Handling ===
273
+ handleError(error) {
274
+ this.lastError = error;
275
+ this.close();
276
+ }
277
+ rejectAllPendingOperations(error) {
278
+ this.rejectAllSendOperations(error);
279
+ this.rejectAllResponseResolvers(error);
280
+ }
281
+ rejectAllSendOperations(error) {
282
+ while (this.sendQueue.length > 0) {
283
+ const queued = this.sendQueue.shift();
284
+ queued.reject(error);
285
+ }
286
+ }
287
+ rejectAllResponseResolvers(error) {
288
+ while (this.responseResolvers.length > 0) {
289
+ const resolver = this.responseResolvers.shift();
290
+ resolver.reject(error);
291
+ }
292
+ }
293
+ createStreamClosedError() {
294
+ if (this.options.abortSignal?.aborted) {
295
+ const reason = this.options.abortSignal.reason;
296
+ if (reason instanceof Error) {
297
+ return reason;
298
+ }
299
+ return new Error('Stream aborted', { cause: reason });
300
+ }
301
+ return new Error('Stream closed');
302
+ }
303
+ // === Helpers ===
304
+ isStreamEnded() {
305
+ return this.connectionState === ConnectionState.CLOSED || this.options.abortSignal?.aborted || false;
306
+ }
307
+ toError(error) {
308
+ if (error instanceof Error)
309
+ return error;
310
+ if (error instanceof undici.ErrorEvent)
311
+ return error.error;
312
+ return new Error(String(error));
313
+ }
314
+ /**
315
+ * Connection state progresses linearly from NEW to CLOSED and never goes back.
316
+ * This internal contract dramatically simplifies the internal stream state management.
317
+ *
318
+ * If you ever feel the need to make this contract less strict, think twice.
319
+ */
320
+ progressConnectionState(newState) {
321
+ if (newState < this.connectionState) {
322
+ return false;
323
+ }
324
+ this.connectionState = newState;
325
+ return true;
326
+ }
327
+ }
328
+
329
+ exports.WebSocketBiDiStream = WebSocketBiDiStream;
330
+ //# sourceMappingURL=websocket_stream.cjs.map
@@ -0,0 +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';\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<ClientMsg extends object, ServerMsg extends object> 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: (stream: WebSocketBiDiStream<ClientMsg, ServerMsg>) => 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 (_: unknown) {\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: () => { void this.connect(); },\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) 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 (!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(\n () => this.sendQueue.length === 0,\n POLL_INTERVAL_MS,\n );\n }\n }\n\n private waitForCondition(\n condition: () => boolean,\n intervalMs: number,\n ): 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 this.connectionState === ConnectionState.CLOSED || this.options.abortSignal?.aborted || false;\n }\n\n private toError(error: unknown): Error {\n if (error instanceof Error) return error;\n if (error instanceof ErrorEvent) return error.error;\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"],"mappings":";;;;;;AAiBA,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;AA2DX,IAAA,GAAA;AACA,IAAA,sBAAA;AACA,IAAA,kBAAA;AACA,IAAA,OAAA;;IA5DX,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;;AAGV,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;YAAE,OAAO,CAAU,EAAE;;;;YAIrB;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,EAAG,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACvC,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;AAAE,YAAA,OAAO,CAAC,OAAO,GAAG,EAAE,aAAa,EAAE,CAAA,OAAA,EAAU,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAA,CAAE,EAAE;AACjG,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;AAEpD,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,CACzB,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EACjC,gBAAgB,CACjB;QACH;IACF;IAEQ,gBAAgB,CACtB,SAAwB,EACxB,UAAkB,EAAA;QAElB,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,OAAO,IAAI,CAAC,eAAe,KAAK,eAAe,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,IAAI,KAAK;IACtG;AAEQ,IAAA,OAAO,CAAC,KAAc,EAAA;QAC5B,IAAI,KAAK,YAAY,KAAK;AAAE,YAAA,OAAO,KAAK;QACxC,IAAI,KAAK,YAAYC,iBAAU;YAAE,OAAO,KAAK,CAAC,KAAK;QACnD,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;;;;"}
@@ -0,0 +1,67 @@
1
+ import { type Dispatcher } from 'undici';
2
+ import type { BiDiStream } from './abstract_stream';
3
+ import type { RetryConfig } from '../helpers/retry_strategy';
4
+ export type WSStreamOptions<ClientMsg extends object, ServerMsg extends object> = {
5
+ abortSignal?: AbortSignal;
6
+ dispatcher?: Dispatcher;
7
+ jwtToken?: string;
8
+ retryConfig?: Partial<RetryConfig>;
9
+ onComplete?: (stream: WebSocketBiDiStream<ClientMsg, ServerMsg>) => void | Promise<void>;
10
+ };
11
+ /**
12
+ * WebSocket-based bidirectional stream implementation for LLTransaction.
13
+ * Implements BiDiStream interface which is compatible with DuplexStreamingCall.
14
+ */
15
+ export declare class WebSocketBiDiStream<ClientMsg extends object, ServerMsg extends object> implements BiDiStream<ClientMsg, ServerMsg> {
16
+ private readonly url;
17
+ private readonly serializeClientMessage;
18
+ private readonly parseServerMessage;
19
+ private readonly options;
20
+ private ws;
21
+ private connectionState;
22
+ private readonly reconnection;
23
+ private readonly sendQueue;
24
+ private sendCompleted;
25
+ private readonly onComplete;
26
+ private readonly responseQueue;
27
+ private responseResolvers;
28
+ private lastError?;
29
+ readonly requests: {
30
+ send: (message: ClientMsg) => Promise<void>;
31
+ complete: () => Promise<void>;
32
+ };
33
+ readonly responses: AsyncIterable<ServerMsg>;
34
+ close(): void;
35
+ constructor(url: string, serializeClientMessage: (message: ClientMsg) => Uint8Array, parseServerMessage: (data: Uint8Array) => ServerMsg, options?: WSStreamOptions<ClientMsg, ServerMsg>);
36
+ private connect;
37
+ private createWebSocket;
38
+ private onOpen;
39
+ private onMessage;
40
+ private onError;
41
+ private onClose;
42
+ private enqueueSend;
43
+ private processSendQueue;
44
+ private canSendMessages;
45
+ private sendQueuedMessage;
46
+ private drainSendQueue;
47
+ private waitForCondition;
48
+ private deliverResponse;
49
+ private createResponseIterator;
50
+ private nextResponse;
51
+ private resolveAllPendingResponses;
52
+ private handleError;
53
+ private rejectAllPendingOperations;
54
+ private rejectAllSendOperations;
55
+ private rejectAllResponseResolvers;
56
+ private createStreamClosedError;
57
+ private isStreamEnded;
58
+ private toError;
59
+ /**
60
+ * Connection state progresses linearly from NEW to CLOSED and never goes back.
61
+ * This internal contract dramatically simplifies the internal stream state management.
62
+ *
63
+ * If you ever feel the need to make this contract less strict, think twice.
64
+ */
65
+ private progressConnectionState;
66
+ }
67
+ //# sourceMappingURL=websocket_stream.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket_stream.d.ts","sourceRoot":"","sources":["../../src/core/websocket_stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiC,KAAK,UAAU,EAAc,MAAM,QAAQ,CAAC;AACpF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAsB7D,MAAM,MAAM,eAAe,CAAC,SAAS,SAAS,MAAM,EAAE,SAAS,SAAS,MAAM,IAAI;IAChF,WAAW,CAAC,EAAE,WAAW,CAAC;IAE1B,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IAEnC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,mBAAmB,CAAC,SAAS,EAAE,SAAS,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1F,CAAC;AAEF;;;GAGG;AACH,qBAAa,mBAAmB,CAAC,SAAS,SAAS,MAAM,EAAE,SAAS,SAAS,MAAM,CAAE,YAAW,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC;IA2D5H,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,sBAAsB;IACvC,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO;IA5D1B,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,eAAe,CAAwC;IAC/D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAgB;IAG7C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA0C;IACpE,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA8E;IAGzG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA2B;IACzD,OAAO,CAAC,iBAAiB,CAAqC;IAG9D,OAAO,CAAC,SAAS,CAAC,CAAQ;IAI1B,SAAgB,QAAQ;wBACA,SAAS,KAAG,OAAO,CAAC,IAAI,CAAC;wBAI3B,OAAO,CAAC,IAAI,CAAC;MAajC;IAEF,SAAgB,SAAS,EAAE,aAAa,CAAC,SAAS,CAAC,CAEjD;IAEK,KAAK,IAAI,IAAI;gBAeD,GAAG,EAAE,MAAM,EACX,sBAAsB,EAAE,CAAC,OAAO,EAAE,SAAS,KAAK,UAAU,EAC1D,kBAAkB,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,SAAS,EACnD,OAAO,GAAE,eAAe,CAAC,SAAS,EAAE,SAAS,CAAM;IAqBtE,OAAO,CAAC,OAAO;IAmBf,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,MAAM;IAKd,OAAO,CAAC,SAAS;IAcjB,OAAO,CAAC,OAAO;IAWf,OAAO,CAAC,OAAO;IAaf,OAAO,CAAC,WAAW;IAenB,OAAO,CAAC,gBAAgB;IASxB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,iBAAiB;YAoBX,cAAc;IAW5B,OAAO,CAAC,gBAAgB;IAgCxB,OAAO,CAAC,eAAe;YASR,sBAAsB;IAUrC,OAAO,CAAC,YAAY;IAwBpB,OAAO,CAAC,0BAA0B;IASlC,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,0BAA0B;IAKlC,OAAO,CAAC,uBAAuB;IAO/B,OAAO,CAAC,0BAA0B;IAOlC,OAAO,CAAC,uBAAuB;IAc/B,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,OAAO;IAMf;;;;;OAKG;IACH,OAAO,CAAC,uBAAuB;CAOhC"}