comfyui-node 1.3.0 → 1.4.0

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 (61) hide show
  1. package/README.md +39 -33
  2. package/dist/.tsbuildinfo +1 -1
  3. package/dist/call-wrapper.d.ts +1 -0
  4. package/dist/call-wrapper.d.ts.map +1 -1
  5. package/dist/call-wrapper.js +136 -41
  6. package/dist/call-wrapper.js.map +1 -1
  7. package/dist/index.d.ts +1 -0
  8. package/dist/index.d.ts.map +1 -1
  9. package/dist/index.js +1 -0
  10. package/dist/index.js.map +1 -1
  11. package/dist/pool/WorkflowPool.d.ts +37 -0
  12. package/dist/pool/WorkflowPool.d.ts.map +1 -0
  13. package/dist/pool/WorkflowPool.js +435 -0
  14. package/dist/pool/WorkflowPool.js.map +1 -0
  15. package/dist/pool/client/ClientManager.d.ts +36 -0
  16. package/dist/pool/client/ClientManager.d.ts.map +1 -0
  17. package/dist/pool/client/ClientManager.js +104 -0
  18. package/dist/pool/client/ClientManager.js.map +1 -0
  19. package/dist/pool/failover/SmartFailoverStrategy.d.ts +18 -0
  20. package/dist/pool/failover/SmartFailoverStrategy.d.ts.map +1 -0
  21. package/dist/pool/failover/SmartFailoverStrategy.js +65 -0
  22. package/dist/pool/failover/SmartFailoverStrategy.js.map +1 -0
  23. package/dist/pool/failover/Strategy.d.ts +10 -0
  24. package/dist/pool/failover/Strategy.d.ts.map +1 -0
  25. package/dist/pool/failover/Strategy.js +2 -0
  26. package/dist/pool/failover/Strategy.js.map +1 -0
  27. package/dist/pool/index.d.ts +8 -0
  28. package/dist/pool/index.d.ts.map +1 -0
  29. package/dist/pool/index.js +4 -0
  30. package/dist/pool/index.js.map +1 -0
  31. package/dist/pool/queue/QueueAdapter.d.ts +31 -0
  32. package/dist/pool/queue/QueueAdapter.d.ts.map +1 -0
  33. package/dist/pool/queue/QueueAdapter.js +2 -0
  34. package/dist/pool/queue/QueueAdapter.js.map +1 -0
  35. package/dist/pool/queue/adapters/memory.d.ts +21 -0
  36. package/dist/pool/queue/adapters/memory.d.ts.map +1 -0
  37. package/dist/pool/queue/adapters/memory.js +96 -0
  38. package/dist/pool/queue/adapters/memory.js.map +1 -0
  39. package/dist/pool/types/events.d.ts +75 -0
  40. package/dist/pool/types/events.d.ts.map +1 -0
  41. package/dist/pool/types/events.js +2 -0
  42. package/dist/pool/types/events.js.map +1 -0
  43. package/dist/pool/types/job.d.ts +56 -0
  44. package/dist/pool/types/job.d.ts.map +1 -0
  45. package/dist/pool/types/job.js +2 -0
  46. package/dist/pool/types/job.js.map +1 -0
  47. package/dist/pool/utils/clone.d.ts +2 -0
  48. package/dist/pool/utils/clone.d.ts.map +1 -0
  49. package/dist/pool/utils/clone.js +7 -0
  50. package/dist/pool/utils/clone.js.map +1 -0
  51. package/dist/pool/utils/hash.d.ts +5 -0
  52. package/dist/pool/utils/hash.d.ts.map +1 -0
  53. package/dist/pool/utils/hash.js +19 -0
  54. package/dist/pool/utils/hash.js.map +1 -0
  55. package/dist/types/event.d.ts +3 -2
  56. package/dist/types/event.d.ts.map +1 -1
  57. package/dist/workflow.d.ts +30 -0
  58. package/dist/workflow.d.ts.map +1 -1
  59. package/dist/workflow.js +30 -1
  60. package/dist/workflow.js.map +1 -1
  61. package/package.json +6 -6
@@ -0,0 +1,435 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { TypedEventTarget } from "../typed-event-target.js";
3
+ import { Workflow } from "../workflow.js";
4
+ import { PromptBuilder } from "../prompt-builder.js";
5
+ import { CallWrapper } from "../call-wrapper.js";
6
+ import { MemoryQueueAdapter } from "./queue/adapters/memory.js";
7
+ import { SmartFailoverStrategy } from "./failover/SmartFailoverStrategy.js";
8
+ import { ClientManager } from "./client/ClientManager.js";
9
+ import { hashWorkflow } from "./utils/hash.js";
10
+ import { cloneDeep } from "./utils/clone.js";
11
+ const DEFAULT_MAX_ATTEMPTS = 3;
12
+ const DEFAULT_RETRY_DELAY = 1000;
13
+ export class WorkflowPool extends TypedEventTarget {
14
+ queue;
15
+ strategy;
16
+ clientManager;
17
+ opts;
18
+ jobStore = new Map();
19
+ initPromise;
20
+ processing = false;
21
+ activeJobs = new Map();
22
+ constructor(clients, opts) {
23
+ super();
24
+ this.strategy = opts?.failoverStrategy ?? new SmartFailoverStrategy();
25
+ this.queue = opts?.queueAdapter ?? new MemoryQueueAdapter();
26
+ this.clientManager = new ClientManager(this.strategy);
27
+ this.opts = opts ?? {};
28
+ this.clientManager.on("client:state", (ev) => {
29
+ this.dispatchEvent(new CustomEvent("client:state", { detail: ev.detail }));
30
+ });
31
+ this.clientManager.on("client:blocked_workflow", (ev) => {
32
+ this.dispatchEvent(new CustomEvent("client:blocked_workflow", { detail: ev.detail }));
33
+ });
34
+ this.clientManager.on("client:unblocked_workflow", (ev) => {
35
+ this.dispatchEvent(new CustomEvent("client:unblocked_workflow", { detail: ev.detail }));
36
+ });
37
+ this.initPromise = this.clientManager
38
+ .initialize(clients)
39
+ .then(() => {
40
+ this.dispatchEvent(new CustomEvent("pool:ready", {
41
+ detail: { clientIds: this.clientManager.list().map((c) => c.id) }
42
+ }));
43
+ })
44
+ .catch((error) => {
45
+ this.dispatchEvent(new CustomEvent("pool:error", { detail: { error } }));
46
+ });
47
+ }
48
+ async ready() {
49
+ await this.initPromise;
50
+ }
51
+ async enqueue(workflowInput, options) {
52
+ await this.ready();
53
+ const workflowJson = this.normalizeWorkflow(workflowInput);
54
+ const workflowHash = hashWorkflow(workflowJson);
55
+ const jobId = options?.jobId ?? this.generateJobId();
56
+ // Extract workflow metadata (outputAliases, outputNodeIds, etc.) if input is a Workflow instance
57
+ let workflowMeta;
58
+ if (workflowInput instanceof Workflow) {
59
+ workflowMeta = {
60
+ outputNodeIds: workflowInput.outputNodeIds ?? [],
61
+ outputAliases: workflowInput.outputAliases ?? {}
62
+ };
63
+ }
64
+ const payload = {
65
+ jobId,
66
+ workflow: workflowJson,
67
+ workflowHash,
68
+ attempts: 0,
69
+ enqueuedAt: Date.now(),
70
+ workflowMeta,
71
+ options: {
72
+ maxAttempts: options?.maxAttempts ?? DEFAULT_MAX_ATTEMPTS,
73
+ retryDelayMs: options?.retryDelayMs ?? DEFAULT_RETRY_DELAY,
74
+ priority: options?.priority ?? 0,
75
+ preferredClientIds: options?.preferredClientIds ?? [],
76
+ excludeClientIds: options?.excludeClientIds ?? [],
77
+ metadata: options?.metadata ?? {},
78
+ includeOutputs: options?.includeOutputs ?? []
79
+ }
80
+ };
81
+ const record = {
82
+ ...payload,
83
+ attachments: options?.attachments,
84
+ status: "queued"
85
+ };
86
+ this.jobStore.set(jobId, record);
87
+ await this.queue.enqueue(payload, { priority: payload.options.priority });
88
+ this.dispatchEvent(new CustomEvent("job:queued", { detail: { job: record } }));
89
+ void this.processQueue();
90
+ return jobId;
91
+ }
92
+ getJob(jobId) {
93
+ return this.jobStore.get(jobId);
94
+ }
95
+ async cancel(jobId) {
96
+ const record = this.jobStore.get(jobId);
97
+ if (!record) {
98
+ return false;
99
+ }
100
+ if (record.status === "queued") {
101
+ const removed = await this.queue.remove(jobId);
102
+ if (removed) {
103
+ record.status = "cancelled";
104
+ record.completedAt = Date.now();
105
+ this.dispatchEvent(new CustomEvent("job:cancelled", { detail: { job: record } }));
106
+ return true;
107
+ }
108
+ }
109
+ const active = this.activeJobs.get(jobId);
110
+ if (active?.cancel) {
111
+ await active.cancel();
112
+ record.status = "cancelled";
113
+ record.completedAt = Date.now();
114
+ this.dispatchEvent(new CustomEvent("job:cancelled", { detail: { job: record } }));
115
+ return true;
116
+ }
117
+ return false;
118
+ }
119
+ async shutdown() {
120
+ await this.queue.shutdown();
121
+ for (const [, ctx] of this.activeJobs) {
122
+ ctx.release({ success: false });
123
+ }
124
+ this.activeJobs.clear();
125
+ }
126
+ async getQueueStats() {
127
+ return this.queue.stats();
128
+ }
129
+ normalizeWorkflow(input) {
130
+ if (typeof input === "string") {
131
+ return JSON.parse(input);
132
+ }
133
+ if (input instanceof Workflow) {
134
+ return cloneDeep(input.json ?? {});
135
+ }
136
+ if (typeof input?.toJSON === "function") {
137
+ return cloneDeep(input.toJSON());
138
+ }
139
+ return cloneDeep(input);
140
+ }
141
+ generateJobId() {
142
+ try {
143
+ return randomUUID();
144
+ }
145
+ catch {
146
+ return WorkflowPool.fallbackId();
147
+ }
148
+ }
149
+ static fallbackId() {
150
+ return (globalThis.crypto && "randomUUID" in globalThis.crypto)
151
+ ? globalThis.crypto.randomUUID()
152
+ : `job_${Math.random().toString(36).slice(2, 10)}`;
153
+ }
154
+ scheduleProcess(delayMs) {
155
+ const wait = Math.max(delayMs, 10);
156
+ setTimeout(() => {
157
+ void this.processQueue();
158
+ }, wait);
159
+ }
160
+ applyAutoSeed(workflow) {
161
+ const autoSeeds = {};
162
+ for (const [nodeId, nodeValue] of Object.entries(workflow)) {
163
+ if (!nodeValue || typeof nodeValue !== "object")
164
+ continue;
165
+ const inputs = nodeValue.inputs;
166
+ if (!inputs || typeof inputs !== "object")
167
+ continue;
168
+ if (typeof inputs.seed === "number" && inputs.seed === -1) {
169
+ const val = Math.floor(Math.random() * 2_147_483_647);
170
+ inputs.seed = val;
171
+ autoSeeds[nodeId] = val;
172
+ }
173
+ }
174
+ return autoSeeds;
175
+ }
176
+ async processQueue() {
177
+ if (this.processing) {
178
+ return;
179
+ }
180
+ this.processing = true;
181
+ try {
182
+ while (true) {
183
+ const reservation = await this.queue.reserve();
184
+ if (!reservation) {
185
+ break;
186
+ }
187
+ const job = this.jobStore.get(reservation.payload.jobId);
188
+ if (!job) {
189
+ await this.queue.commit(reservation.reservationId);
190
+ continue;
191
+ }
192
+ const lease = this.clientManager.claim(job);
193
+ if (!lease) {
194
+ await this.queue.retry(reservation.reservationId, { delayMs: job.options.retryDelayMs });
195
+ this.scheduleProcess(job.options.retryDelayMs);
196
+ break;
197
+ }
198
+ this.runJob({ reservation, job, clientId: lease.clientId, release: lease.release }).catch((error) => {
199
+ console.error("[WorkflowPool] Unhandled job error", error);
200
+ });
201
+ }
202
+ }
203
+ finally {
204
+ this.processing = false;
205
+ }
206
+ }
207
+ async runJob(ctx) {
208
+ const { reservation, job, clientId, release } = ctx;
209
+ const managed = this.clientManager.getClient(clientId);
210
+ const client = managed?.client;
211
+ if (!client) {
212
+ await this.queue.retry(reservation.reservationId, { delayMs: job.options.retryDelayMs });
213
+ release({ success: false });
214
+ return;
215
+ }
216
+ job.status = "running";
217
+ job.clientId = clientId;
218
+ job.attempts += 1;
219
+ reservation.payload.attempts = job.attempts;
220
+ job.startedAt = Date.now();
221
+ this.dispatchEvent(new CustomEvent("job:started", { detail: { job } }));
222
+ const workflowPayload = cloneDeep(reservation.payload.workflow);
223
+ if (job.attachments?.length) {
224
+ for (const attachment of job.attachments) {
225
+ const filename = attachment.filename ?? `${job.jobId}-${attachment.nodeId}-${attachment.inputName}.bin`;
226
+ const blob = attachment.file instanceof Buffer ? new Blob([new Uint8Array(attachment.file)]) : attachment.file;
227
+ await client.ext.file.uploadImage(blob, filename, { override: true });
228
+ const node = workflowPayload[attachment.nodeId];
229
+ if (node?.inputs) {
230
+ node.inputs[attachment.inputName] = filename;
231
+ }
232
+ }
233
+ }
234
+ const autoSeeds = this.applyAutoSeed(workflowPayload);
235
+ let wfInstance = Workflow.from(workflowPayload);
236
+ if (job.options.includeOutputs?.length) {
237
+ for (const nodeId of job.options.includeOutputs) {
238
+ if (nodeId) {
239
+ wfInstance = wfInstance.output(nodeId);
240
+ }
241
+ }
242
+ }
243
+ wfInstance.inferDefaultOutputs?.();
244
+ // Use stored metadata if available (from Workflow instance), otherwise extract from recreated instance
245
+ const outputNodeIds = reservation.payload.workflowMeta?.outputNodeIds ??
246
+ wfInstance.outputNodeIds ??
247
+ job.options.includeOutputs ?? [];
248
+ const outputAliases = reservation.payload.workflowMeta?.outputAliases ??
249
+ wfInstance.outputAliases ?? {};
250
+ let promptBuilder = new PromptBuilder(wfInstance.json, wfInstance.inputPaths ?? [], outputNodeIds);
251
+ for (const nodeId of outputNodeIds) {
252
+ const alias = outputAliases[nodeId] ?? nodeId;
253
+ promptBuilder = promptBuilder.setOutputNode(alias, nodeId);
254
+ }
255
+ const wrapper = new CallWrapper(client, promptBuilder);
256
+ let pendingSettled = false;
257
+ let resolvePending;
258
+ let rejectPending;
259
+ const pendingPromise = new Promise((resolve, reject) => {
260
+ resolvePending = () => {
261
+ if (!pendingSettled) {
262
+ pendingSettled = true;
263
+ resolve();
264
+ }
265
+ };
266
+ rejectPending = (err) => {
267
+ if (!pendingSettled) {
268
+ pendingSettled = true;
269
+ reject(err);
270
+ }
271
+ };
272
+ });
273
+ let resolveCompletion;
274
+ let rejectCompletion;
275
+ const completionPromise = new Promise((resolve, reject) => {
276
+ resolveCompletion = resolve;
277
+ rejectCompletion = reject;
278
+ });
279
+ wrapper.onProgress((progress, promptId) => {
280
+ if (!job.promptId && promptId) {
281
+ job.promptId = promptId;
282
+ }
283
+ this.dispatchEvent(new CustomEvent("job:progress", {
284
+ detail: { jobId: job.jobId, clientId, progress }
285
+ }));
286
+ });
287
+ wrapper.onPreview((blob, promptId) => {
288
+ if (!job.promptId && promptId) {
289
+ job.promptId = promptId;
290
+ }
291
+ this.dispatchEvent(new CustomEvent("job:preview", {
292
+ detail: { jobId: job.jobId, clientId, blob }
293
+ }));
294
+ });
295
+ wrapper.onPreviewMeta((payload, promptId) => {
296
+ if (!job.promptId && promptId) {
297
+ job.promptId = promptId;
298
+ }
299
+ this.dispatchEvent(new CustomEvent("job:preview_meta", {
300
+ detail: { jobId: job.jobId, clientId, payload }
301
+ }));
302
+ });
303
+ wrapper.onOutput((key, data, promptId) => {
304
+ if (!job.promptId && promptId) {
305
+ job.promptId = promptId;
306
+ }
307
+ this.dispatchEvent(new CustomEvent("job:output", {
308
+ detail: { jobId: job.jobId, clientId, key: String(key), data }
309
+ }));
310
+ });
311
+ wrapper.onPending((promptId) => {
312
+ if (!job.promptId && promptId) {
313
+ job.promptId = promptId;
314
+ }
315
+ this.dispatchEvent(new CustomEvent("job:accepted", { detail: { job } }));
316
+ resolvePending?.();
317
+ });
318
+ wrapper.onStart((promptId) => {
319
+ if (!job.promptId && promptId) {
320
+ job.promptId = promptId;
321
+ }
322
+ });
323
+ wrapper.onFinished((data, promptId) => {
324
+ if (!job.promptId && promptId) {
325
+ job.promptId = promptId;
326
+ }
327
+ job.status = "completed";
328
+ job.lastError = undefined;
329
+ const resultPayload = {};
330
+ for (const nodeId of outputNodeIds) {
331
+ const alias = outputAliases[nodeId] ?? nodeId;
332
+ // CallWrapper uses alias keys when mapOutputKeys is configured, fallback to nodeId
333
+ const nodeResult = data[alias];
334
+ const fallbackResult = data[nodeId];
335
+ const finalResult = nodeResult !== undefined ? nodeResult : fallbackResult;
336
+ resultPayload[alias] = finalResult;
337
+ }
338
+ resultPayload._nodes = [...outputNodeIds];
339
+ resultPayload._aliases = { ...outputAliases };
340
+ if (job.promptId) {
341
+ resultPayload._promptId = job.promptId;
342
+ }
343
+ if (Object.keys(autoSeeds).length) {
344
+ resultPayload._autoSeeds = { ...autoSeeds };
345
+ }
346
+ job.result = resultPayload;
347
+ job.completedAt = Date.now();
348
+ this.dispatchEvent(new CustomEvent("job:completed", { detail: { job } }));
349
+ resolveCompletion?.();
350
+ });
351
+ wrapper.onFailed((error, promptId) => {
352
+ if (!job.promptId && promptId) {
353
+ job.promptId = promptId;
354
+ }
355
+ job.lastError = error;
356
+ rejectPending?.(error);
357
+ rejectCompletion?.(error);
358
+ });
359
+ try {
360
+ const exec = wrapper.run();
361
+ await pendingPromise;
362
+ this.activeJobs.set(job.jobId, {
363
+ reservation,
364
+ job,
365
+ clientId,
366
+ release,
367
+ cancel: async () => {
368
+ try {
369
+ if (job.promptId) {
370
+ await client.ext.queue.interrupt(job.promptId);
371
+ }
372
+ }
373
+ finally {
374
+ this.activeJobs.delete(job.jobId);
375
+ await this.queue.discard(reservation.reservationId, new Error("cancelled"));
376
+ release({ success: false });
377
+ }
378
+ }
379
+ });
380
+ const result = await exec;
381
+ if (result === false) {
382
+ // Execution failed - try to get the error from completionPromise rejection
383
+ try {
384
+ await completionPromise;
385
+ }
386
+ catch (err) {
387
+ throw err;
388
+ }
389
+ throw job.lastError ?? new Error("Execution failed");
390
+ }
391
+ await completionPromise;
392
+ await this.queue.commit(reservation.reservationId);
393
+ release({ success: true });
394
+ }
395
+ catch (error) {
396
+ const latestStatus = this.jobStore.get(job.jobId)?.status;
397
+ if (latestStatus === "cancelled") {
398
+ release({ success: false });
399
+ return;
400
+ }
401
+ job.lastError = error;
402
+ job.status = "failed";
403
+ this.clientManager.recordFailure(clientId, job, error);
404
+ const remainingAttempts = job.options.maxAttempts - job.attempts;
405
+ const willRetry = remainingAttempts > 0;
406
+ this.dispatchEvent(new CustomEvent("job:failed", {
407
+ detail: { job, willRetry }
408
+ }));
409
+ if (willRetry) {
410
+ const delay = this.opts.retryBackoffMs ?? job.options.retryDelayMs;
411
+ this.dispatchEvent(new CustomEvent("job:retrying", { detail: { job, delayMs: delay } }));
412
+ job.status = "queued";
413
+ job.clientId = undefined;
414
+ job.promptId = undefined;
415
+ job.startedAt = undefined;
416
+ job.completedAt = undefined;
417
+ job.result = undefined;
418
+ await this.queue.retry(reservation.reservationId, { delayMs: delay });
419
+ this.dispatchEvent(new CustomEvent("job:queued", { detail: { job } }));
420
+ this.scheduleProcess(delay);
421
+ release({ success: false });
422
+ }
423
+ else {
424
+ job.completedAt = Date.now();
425
+ await this.queue.discard(reservation.reservationId, error);
426
+ release({ success: false });
427
+ }
428
+ }
429
+ finally {
430
+ this.activeJobs.delete(job.jobId);
431
+ void this.processQueue();
432
+ }
433
+ }
434
+ }
435
+ //# sourceMappingURL=WorkflowPool.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"WorkflowPool.js","sourceRoot":"","sources":["../../src/pool/WorkflowPool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAE5D,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAGhE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAkB7C,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B,MAAM,mBAAmB,GAAG,IAAI,CAAC;AAEjC,MAAM,OAAO,YAAa,SAAQ,gBAAsC;IAC9D,KAAK,CAAe;IACpB,QAAQ,CAAmB;IAC3B,aAAa,CAAgB;IAC7B,IAAI,CAAmB;IACvB,QAAQ,GAA0B,IAAI,GAAG,EAAE,CAAC;IAC5C,WAAW,CAAgB;IAC3B,UAAU,GAAG,KAAK,CAAC;IACnB,UAAU,GAAiC,IAAI,GAAG,EAAE,CAAC;IAE7D,YAAY,OAAmB,EAAE,IAAuB;QACtD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,IAAI,EAAE,gBAAgB,IAAI,IAAI,qBAAqB,EAAE,CAAC;QACtE,IAAI,CAAC,KAAK,GAAG,IAAI,EAAE,YAAY,IAAI,IAAI,kBAAkB,EAAE,CAAC;QAC5D,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,EAAE,EAAE,EAAE;YAC3C,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,yBAAyB,EAAE,CAAC,EAAE,EAAE,EAAE;YACtD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,2BAA2B,EAAE,CAAC,EAAE,EAAE,EAAE;YACxD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,aAAa;aAClC,UAAU,CAAC,OAAO,CAAC;aACnB,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,YAAY,EAAE;gBAC/C,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE;aAClE,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,WAAW,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,aAA4B,EAAE,OAA4B;QACtE,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC3D,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;QAErD,iGAAiG;QACjG,IAAI,YAA8F,CAAC;QACnG,IAAI,aAAa,YAAY,QAAQ,EAAE,CAAC;YACtC,YAAY,GAAG;gBACb,aAAa,EAAG,aAAqB,CAAC,aAAa,IAAI,EAAE;gBACzD,aAAa,EAAG,aAAqB,CAAC,aAAa,IAAI,EAAE;aAC1D,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAuB;YAClC,KAAK;YACL,QAAQ,EAAE,YAAY;YACtB,YAAY;YACZ,QAAQ,EAAE,CAAC;YACX,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;YACtB,YAAY;YACZ,OAAO,EAAE;gBACP,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,oBAAoB;gBACzD,YAAY,EAAE,OAAO,EAAE,YAAY,IAAI,mBAAmB;gBAC1D,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,CAAC;gBAChC,kBAAkB,EAAE,OAAO,EAAE,kBAAkB,IAAI,EAAE;gBACrD,gBAAgB,EAAE,OAAO,EAAE,gBAAgB,IAAI,EAAE;gBACjD,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,EAAE;gBACjC,cAAc,EAAE,OAAO,EAAE,cAAc,IAAI,EAAE;aAC9C;SACF,CAAC;QAEF,MAAM,MAAM,GAAc;YACxB,GAAG,OAAO;YACV,WAAW,EAAE,OAAO,EAAE,WAAW;YACjC,MAAM,EAAE,QAAQ;SACjB,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QAC/E,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,CAAC,KAAa;QAClB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa;QACxB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC/C,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC5B,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAChC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;gBAClF,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,MAAM,EAAE,MAAM,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC;YAC5B,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;YAClF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC5B,KAAK,MAAM,CAAC,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACtC,GAAG,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAEO,iBAAiB,CAAC,KAAoB;QAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,OAAO,SAAS,CAAE,KAAa,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,OAAQ,KAAa,EAAE,MAAM,KAAK,UAAU,EAAE,CAAC;YACjD,OAAO,SAAS,CAAE,KAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,SAAS,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC;YACH,OAAO,UAAU,EAAE,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,YAAY,CAAC,UAAU,EAAE,CAAC;QACnC,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,UAAU;QACvB,OAAO,CAAC,UAAU,CAAC,MAAM,IAAI,YAAY,IAAI,UAAU,CAAC,MAAM,CAAC;YAC7D,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE;YAChC,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;IACvD,CAAC;IAEO,eAAe,CAAC,OAAe;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACnC,UAAU,CAAC,GAAG,EAAE;YACd,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3B,CAAC,EAAE,IAAI,CAAC,CAAC;IACX,CAAC;IAEO,aAAa,CAAC,QAA6B;QACjD,MAAM,SAAS,GAA2B,EAAE,CAAC;QAC7C,KAAK,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ;gBAAE,SAAS;YAC1D,MAAM,MAAM,GAAI,SAAiB,CAAC,MAAM,CAAC;YACzC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;gBAAE,SAAS;YACpD,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;gBAC1D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC;gBACtD,MAAM,CAAC,IAAI,GAAG,GAAG,CAAC;gBAClB,SAAS,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC/C,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,MAAM;gBACR,CAAC;gBACD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzD,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;oBACnD,SAAS;gBACX,CAAC;gBACD,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC5C,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;oBACzF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;oBAC/C,MAAM;gBACR,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAClG,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;gBAC7D,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,GAAqB;QACxC,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,GAAG,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;YACzF,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC;QACvB,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACxB,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC;QAClB,WAAW,CAAC,OAAO,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC5C,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QAExE,MAAM,eAAe,GAAG,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAwB,CAAC;QAEvF,IAAI,GAAG,CAAC,WAAW,EAAE,MAAM,EAAE,CAAC;YAC5B,KAAK,MAAM,UAAU,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;gBACzC,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,IAAI,GAAG,GAAG,CAAC,KAAK,IAAI,UAAU,CAAC,MAAM,IAAI,UAAU,CAAC,SAAS,MAAM,CAAC;gBACxG,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,YAAY,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC/G,MAAM,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAEtE,MAAM,IAAI,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBAChD,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;oBACjB,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;QACtD,IAAI,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAChD,IAAI,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,EAAE,CAAC;YACvC,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAChD,IAAI,MAAM,EAAE,CAAC;oBACX,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,MAAa,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;QACH,CAAC;QACA,UAAkB,CAAC,mBAAmB,EAAE,EAAE,CAAC;QAE5C,uGAAuG;QACvG,MAAM,aAAa,GAAa,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,aAAa;YAC5E,UAAkB,CAAC,aAAa;YACjC,GAAG,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC;QACnC,MAAM,aAAa,GAA2B,WAAW,CAAC,OAAO,CAAC,YAAY,EAAE,aAAa;YAC1F,UAAkB,CAAC,aAAa,IAAI,EAAE,CAAC;QAE1C,IAAI,aAAa,GAAG,IAAI,aAAa,CAClC,UAAkB,CAAC,IAAI,EACvB,UAAkB,CAAC,UAAU,IAAI,EAAE,EACpC,aAAoB,CACrB,CAAC;QACF,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC;YAC9C,aAAa,GAAG,aAAa,CAAC,aAAa,CAAC,KAAY,EAAE,MAAa,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAEvD,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IAAI,cAAwC,CAAC;QAC7C,IAAI,aAAqD,CAAC;QAC1D,MAAM,cAAc,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3D,cAAc,GAAG,GAAG,EAAE;gBACpB,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,cAAc,GAAG,IAAI,CAAC;oBACtB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;YACF,aAAa,GAAG,CAAC,GAAG,EAAE,EAAE;gBACtB,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,cAAc,GAAG,IAAI,CAAC;oBACtB,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;YACH,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,iBAA2C,CAAC;QAChD,IAAI,gBAAwD,CAAC;QAC7D,MAAM,iBAAiB,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC9D,iBAAiB,GAAG,OAAO,CAAC;YAC5B,gBAAgB,GAAG,MAAM,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE;YACxC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,cAAc,EAAE;gBAC9B,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE;aACjD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE;YACnC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;gBAC7B,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE;aAC7C,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,kBAAkB,EAAE;gBAClC,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE;aAChD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YACvC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,YAAY,EAAE;gBAC5B,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE;aAC/D,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;YACzE,cAAc,EAAE,EAAE,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC3B,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE;YACpC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,GAAG,CAAC,MAAM,GAAG,WAAW,CAAC;YACzB,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;YAE1B,MAAM,aAAa,GAA4B,EAAE,CAAC;YAClD,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;gBACnC,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC;gBAC9C,mFAAmF;gBACnF,MAAM,UAAU,GAAI,IAAY,CAAC,KAAK,CAAC,CAAC;gBACxC,MAAM,cAAc,GAAI,IAAY,CAAC,MAAM,CAAC,CAAC;gBAC7C,MAAM,WAAW,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC;gBAE3E,aAAa,CAAC,KAAK,CAAC,GAAG,WAAW,CAAC;YACrC,CAAC;YACD,aAAa,CAAC,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,CAAC;YAC1C,aAAa,CAAC,QAAQ,GAAG,EAAE,GAAG,aAAa,EAAE,CAAC;YAC9C,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACjB,aAAa,CAAC,SAAS,GAAG,GAAG,CAAC,QAAQ,CAAC;YACzC,CAAC;YACD,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC;gBAClC,aAAa,CAAC,UAAU,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC;YAC9C,CAAC;YACD,GAAG,CAAC,MAAM,GAAG,aAAa,CAAC;YAC3B,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1E,iBAAiB,EAAE,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;YACnC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC9B,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAC1B,CAAC;YACD,GAAG,CAAC,SAAS,GAAG,KAAK,CAAC;YACtB,aAAa,EAAE,CAAC,KAAK,CAAC,CAAC;YACvB,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,cAAc,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE;gBAC7B,WAAW;gBACX,GAAG;gBACH,QAAQ;gBACR,OAAO;gBACP,MAAM,EAAE,KAAK,IAAI,EAAE;oBACjB,IAAI,CAAC;wBACH,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;4BACjB,MAAM,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;wBACjD,CAAC;oBACH,CAAC;4BAAS,CAAC;wBACT,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;wBAClC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,aAAa,EAAE,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;wBAC5E,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC;YAC1B,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;gBACrB,2EAA2E;gBAC3E,IAAI,CAAC;oBACH,MAAM,iBAAiB,CAAC;gBAC1B,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,GAAG,CAAC;gBACZ,CAAC;gBACD,MAAM,GAAG,CAAC,SAAS,IAAI,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACvD,CAAC;YACD,MAAM,iBAAiB,CAAC;YACxB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;YACnD,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;YAC1D,IAAI,YAAY,KAAK,WAAW,EAAE,CAAC;gBACjC,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC5B,OAAO;YACT,CAAC;YACD,GAAG,CAAC,SAAS,GAAG,KAAK,CAAC;YACtB,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC;YACtB,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,iBAAiB,GAAG,GAAG,CAAC,OAAO,CAAC,WAAW,GAAG,GAAG,CAAC,QAAQ,CAAC;YACjE,MAAM,SAAS,GAAG,iBAAiB,GAAG,CAAC,CAAC;YACxC,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,YAAY,EAAE;gBAC5B,MAAM,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE;aAC3B,CAAC,CACH,CAAC;YACF,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC;gBACnE,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;gBACzF,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC;gBACtB,GAAG,CAAC,QAAQ,GAAG,SAAS,CAAC;gBACzB,GAAG,CAAC,QAAQ,GAAG,SAAS,CAAC;gBACzB,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;gBAC1B,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC;gBAC5B,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC;gBACvB,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtE,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;gBACvE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC5B,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9B,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;gBAC3D,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClC,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,36 @@
1
+ import type { ComfyApi } from "../../client.js";
2
+ import { TypedEventTarget } from "../../typed-event-target.js";
3
+ import type { JobRecord } from "../types/job.js";
4
+ import type { FailoverStrategy } from "../failover/Strategy.js";
5
+ import type { WorkflowPoolEventMap } from "../types/events.js";
6
+ export interface ManagedClient {
7
+ client: ComfyApi;
8
+ id: string;
9
+ online: boolean;
10
+ busy: boolean;
11
+ lastError?: unknown;
12
+ lastSeenAt: number;
13
+ supportedWorkflows: Set<string>;
14
+ }
15
+ interface ClientLease {
16
+ client: ComfyApi;
17
+ clientId: string;
18
+ release: (opts?: {
19
+ success?: boolean;
20
+ }) => void;
21
+ }
22
+ export declare class ClientManager extends TypedEventTarget<WorkflowPoolEventMap> {
23
+ private clients;
24
+ private strategy;
25
+ constructor(strategy: FailoverStrategy);
26
+ private emitBlocked;
27
+ private emitUnblocked;
28
+ initialize(clients: ComfyApi[]): Promise<void>;
29
+ addClient(client: ComfyApi): Promise<void>;
30
+ list(): ManagedClient[];
31
+ getClient(clientId: string): ManagedClient | undefined;
32
+ claim(job: JobRecord): ClientLease | null;
33
+ recordFailure(clientId: string, job: JobRecord, error: unknown): void;
34
+ }
35
+ export {};
36
+ //# sourceMappingURL=ClientManager.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ClientManager.d.ts","sourceRoot":"","sources":["../../../src/pool/client/ClientManager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE/D,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,QAAQ,CAAC;IACjB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;CACjC;AAED,UAAU,WAAW;IACnB,MAAM,EAAE,QAAQ,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,KAAK,IAAI,CAAC;CACjD;AAED,qBAAa,aAAc,SAAQ,gBAAgB,CAAC,oBAAoB,CAAC;IACvE,OAAO,CAAC,OAAO,CAAuB;IACtC,OAAO,CAAC,QAAQ,CAAmB;gBAEvB,QAAQ,EAAE,gBAAgB;IAKtC,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,aAAa;IAIf,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAM9C,SAAS,CAAC,MAAM,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BhD,IAAI,IAAI,aAAa,EAAE;IAIvB,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa,GAAG,SAAS;IAItD,KAAK,CAAC,GAAG,EAAE,SAAS,GAAG,WAAW,GAAG,IAAI;IAkCzC,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO;CAiB/D"}
@@ -0,0 +1,104 @@
1
+ import { TypedEventTarget } from "../../typed-event-target.js";
2
+ export class ClientManager extends TypedEventTarget {
3
+ clients = [];
4
+ strategy;
5
+ constructor(strategy) {
6
+ super();
7
+ this.strategy = strategy;
8
+ }
9
+ emitBlocked(clientId, workflowHash) {
10
+ this.dispatchEvent(new CustomEvent("client:blocked_workflow", { detail: { clientId, workflowHash } }));
11
+ }
12
+ emitUnblocked(clientId, workflowHash) {
13
+ this.dispatchEvent(new CustomEvent("client:unblocked_workflow", { detail: { clientId, workflowHash } }));
14
+ }
15
+ async initialize(clients) {
16
+ for (const client of clients) {
17
+ await this.addClient(client);
18
+ }
19
+ }
20
+ async addClient(client) {
21
+ await client.init();
22
+ const managed = {
23
+ client,
24
+ id: client.id,
25
+ online: true,
26
+ busy: false,
27
+ lastSeenAt: Date.now(),
28
+ supportedWorkflows: new Set()
29
+ };
30
+ this.clients.push(managed);
31
+ client.on("disconnected", () => {
32
+ managed.online = false;
33
+ managed.busy = false;
34
+ managed.lastSeenAt = Date.now();
35
+ this.dispatchEvent(new CustomEvent("client:state", {
36
+ detail: { clientId: managed.id, online: false, busy: false, lastError: managed.lastError }
37
+ }));
38
+ });
39
+ client.on("reconnected", () => {
40
+ managed.online = true;
41
+ managed.lastSeenAt = Date.now();
42
+ this.dispatchEvent(new CustomEvent("client:state", {
43
+ detail: { clientId: managed.id, online: true, busy: managed.busy }
44
+ }));
45
+ });
46
+ }
47
+ list() {
48
+ return [...this.clients];
49
+ }
50
+ getClient(clientId) {
51
+ return this.clients.find((c) => c.id === clientId);
52
+ }
53
+ claim(job) {
54
+ const candidates = this.clients.filter((c) => c.online && !c.busy);
55
+ const preferred = job.options.preferredClientIds?.length
56
+ ? candidates.filter((c) => job.options.preferredClientIds?.includes(c.id))
57
+ : candidates;
58
+ const filtered = preferred
59
+ .filter((client) => !job.options.excludeClientIds?.includes(client.id))
60
+ .filter((client) => !this.strategy.shouldSkipClient(client, job));
61
+ const chosen = filtered[0];
62
+ if (!chosen) {
63
+ return null;
64
+ }
65
+ chosen.busy = true;
66
+ chosen.lastSeenAt = Date.now();
67
+ return {
68
+ client: chosen.client,
69
+ clientId: chosen.id,
70
+ release: (opts) => {
71
+ chosen.busy = false;
72
+ if (opts?.success) {
73
+ const wasBlocked = this.strategy.isWorkflowBlocked?.(chosen, job.workflowHash) ?? false;
74
+ this.strategy.recordSuccess(chosen, job);
75
+ const stillBlocked = this.strategy.isWorkflowBlocked?.(chosen, job.workflowHash) ?? false;
76
+ if (wasBlocked && !stillBlocked) {
77
+ this.emitUnblocked(chosen.id, job.workflowHash);
78
+ }
79
+ }
80
+ this.dispatchEvent(new CustomEvent("client:state", {
81
+ detail: { clientId: chosen.id, online: chosen.online, busy: chosen.busy }
82
+ }));
83
+ }
84
+ };
85
+ }
86
+ recordFailure(clientId, job, error) {
87
+ const client = this.clients.find((c) => c.id === clientId);
88
+ if (!client) {
89
+ return;
90
+ }
91
+ client.lastError = error;
92
+ client.busy = false;
93
+ const wasBlocked = this.strategy.isWorkflowBlocked?.(client, job.workflowHash) ?? false;
94
+ this.strategy.recordFailure(client, job, error);
95
+ const isBlocked = this.strategy.isWorkflowBlocked?.(client, job.workflowHash) ?? false;
96
+ if (!wasBlocked && isBlocked) {
97
+ this.emitBlocked(client.id, job.workflowHash);
98
+ }
99
+ this.dispatchEvent(new CustomEvent("client:state", {
100
+ detail: { clientId: client.id, online: client.online, busy: client.busy, lastError: error }
101
+ }));
102
+ }
103
+ }
104
+ //# sourceMappingURL=ClientManager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ClientManager.js","sourceRoot":"","sources":["../../../src/pool/client/ClientManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAqB/D,MAAM,OAAO,aAAc,SAAQ,gBAAsC;IAC/D,OAAO,GAAoB,EAAE,CAAC;IAC9B,QAAQ,CAAmB;IAEnC,YAAY,QAA0B;QACpC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,WAAW,CAAC,QAAgB,EAAE,YAAoB;QACxD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC;IACzG,CAAC;IAEO,aAAa,CAAC,QAAgB,EAAE,YAAoB;QAC1D,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3G,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAmB;QAClC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAgB;QAC9B,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,MAAM,OAAO,GAAkB;YAC7B,MAAM;YACN,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,KAAK;YACX,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;YACtB,kBAAkB,EAAE,IAAI,GAAG,EAAE;SAC9B,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE3B,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;YAC7B,OAAO,CAAC,MAAM,GAAG,KAAK,CAAC;YACvB,OAAO,CAAC,IAAI,GAAG,KAAK,CAAC;YACrB,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE;gBACjD,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;aAC3F,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;YAC5B,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;YACtB,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE;gBACjD,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE;aACnE,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,SAAS,CAAC,QAAgB;QACxB,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,GAAc;QAClB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,kBAAkB,EAAE,MAAM;YACtD,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC1E,CAAC,CAAC,UAAU,CAAC;QACf,MAAM,QAAQ,GAAG,SAAS;aACvB,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;aACtE,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,MAAM,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,OAAO,EAAE,CAAC,IAA4B,EAAE,EAAE;gBACxC,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC;gBACpB,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;oBAClB,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC;oBACxF,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;oBACzC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC;oBAC1F,IAAI,UAAU,IAAI,CAAC,YAAY,EAAE,CAAC;wBAChC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE;oBACjD,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;iBAC1E,CAAC,CAAC,CAAC;YACN,CAAC;SACF,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,QAAgB,EAAE,GAAc,EAAE,KAAc;QAC5D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC;QACpB,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC;QACxF,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC;QACvF,IAAI,CAAC,UAAU,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,cAAc,EAAE;YACjD,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE;SAC5F,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}
@@ -0,0 +1,18 @@
1
+ import type { ManagedClient } from "../client/ClientManager.js";
2
+ import type { JobRecord } from "../types/job.js";
3
+ import type { FailoverStrategy } from "./Strategy.js";
4
+ export declare class SmartFailoverStrategy implements FailoverStrategy {
5
+ private workflowFailures;
6
+ private cooldownMs;
7
+ private maxFailuresBeforeBlock;
8
+ constructor(opts?: {
9
+ cooldownMs?: number;
10
+ maxFailuresBeforeBlock?: number;
11
+ });
12
+ shouldSkipClient(client: ManagedClient, job: JobRecord): boolean;
13
+ recordFailure(client: ManagedClient, job: JobRecord, error: unknown): void;
14
+ recordSuccess(client: ManagedClient, job: JobRecord): void;
15
+ resetForWorkflow(workflowHash: string): void;
16
+ isWorkflowBlocked(client: ManagedClient, workflowHash: string): boolean;
17
+ }
18
+ //# sourceMappingURL=SmartFailoverStrategy.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"SmartFailoverStrategy.d.ts","sourceRoot":"","sources":["../../../src/pool/failover/SmartFailoverStrategy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAOtD,qBAAa,qBAAsB,YAAW,gBAAgB;IAC5D,OAAO,CAAC,gBAAgB,CAA6D;IACrF,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,sBAAsB,CAAS;gBAE3B,IAAI,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,sBAAsB,CAAC,EAAE,MAAM,CAAA;KAAE;IAK3E,gBAAgB,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO;IAgBhE,aAAa,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI;IAe1E,aAAa,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,SAAS,GAAG,IAAI;IAW1D,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAM5C,iBAAiB,CAAC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO;CAWxE"}