comfyui-node 1.3.1 → 1.4.1

package/dist/pool/WorkflowPool.js

@@ -0,0 +1,438 @@
+import { randomUUID } from "node:crypto";
+import { TypedEventTarget } from "../typed-event-target.js";
+import { Workflow } from "../workflow.js";
+import { PromptBuilder } from "../prompt-builder.js";
+import { CallWrapper } from "../call-wrapper.js";
+import { MemoryQueueAdapter } from "./queue/adapters/memory.js";
+import { SmartFailoverStrategy } from "./failover/SmartFailoverStrategy.js";
+import { ClientManager } from "./client/ClientManager.js";
+import { hashWorkflow } from "./utils/hash.js";
+import { cloneDeep } from "./utils/clone.js";
+const DEFAULT_MAX_ATTEMPTS = 3;
+const DEFAULT_RETRY_DELAY = 1000;
+export class WorkflowPool extends TypedEventTarget {
+    queue;
+    strategy;
+    clientManager;
+    opts;
+    jobStore = new Map();
+    initPromise;
+    processing = false;
+    activeJobs = new Map();
+    constructor(clients, opts) {
+        super();
+        this.strategy = opts?.failoverStrategy ?? new SmartFailoverStrategy();
+        this.queue = opts?.queueAdapter ?? new MemoryQueueAdapter();
+        this.clientManager = new ClientManager(this.strategy, {
+            healthCheckIntervalMs: opts?.healthCheckIntervalMs ?? 30000
+        });
+        this.opts = opts ?? {};
+        this.clientManager.on("client:state", (ev) => {
+            this.dispatchEvent(new CustomEvent("client:state", { detail: ev.detail }));
+        });
+        this.clientManager.on("client:blocked_workflow", (ev) => {
+            this.dispatchEvent(new CustomEvent("client:blocked_workflow", { detail: ev.detail }));
+        });
+        this.clientManager.on("client:unblocked_workflow", (ev) => {
+            this.dispatchEvent(new CustomEvent("client:unblocked_workflow", { detail: ev.detail }));
+        });
+        this.initPromise = this.clientManager
+            .initialize(clients)
+            .then(() => {
+            this.dispatchEvent(new CustomEvent("pool:ready", {
+                detail: { clientIds: this.clientManager.list().map((c) => c.id) }
+            }));
+        })
+            .catch((error) => {
+            this.dispatchEvent(new CustomEvent("pool:error", { detail: { error } }));
+        });
+    }
+    async ready() {
+        await this.initPromise;
+    }
+    async enqueue(workflowInput, options) {
+        await this.ready();
+        const workflowJson = this.normalizeWorkflow(workflowInput);
+        const workflowHash = hashWorkflow(workflowJson);
+        const jobId = options?.jobId ?? this.generateJobId();
+        // Extract workflow metadata (outputAliases, outputNodeIds, etc.) if input is a Workflow instance
+        let workflowMeta;
+        if (workflowInput instanceof Workflow) {
+            workflowMeta = {
+                outputNodeIds: workflowInput.outputNodeIds ?? [],
+                outputAliases: workflowInput.outputAliases ?? {}
+            };
+        }
+        const payload = {
+            jobId,
+            workflow: workflowJson,
+            workflowHash,
+            attempts: 0,
+            enqueuedAt: Date.now(),
+            workflowMeta,
+            options: {
+                maxAttempts: options?.maxAttempts ?? DEFAULT_MAX_ATTEMPTS,
+                retryDelayMs: options?.retryDelayMs ?? DEFAULT_RETRY_DELAY,
+                priority: options?.priority ?? 0,
+                preferredClientIds: options?.preferredClientIds ?? [],
+                excludeClientIds: options?.excludeClientIds ?? [],
+                metadata: options?.metadata ?? {},
+                includeOutputs: options?.includeOutputs ?? []
+            }
+        };
+        const record = {
+            ...payload,
+            attachments: options?.attachments,
+            status: "queued"
+        };
+        this.jobStore.set(jobId, record);
+        await this.queue.enqueue(payload, { priority: payload.options.priority });
+        this.dispatchEvent(new CustomEvent("job:queued", { detail: { job: record } }));
+        void this.processQueue();
+        return jobId;
+    }
+    getJob(jobId) {
+        return this.jobStore.get(jobId);
+    }
+    async cancel(jobId) {
+        const record = this.jobStore.get(jobId);
+        if (!record) {
+            return false;
+        }
+        if (record.status === "queued") {
+            const removed = await this.queue.remove(jobId);
+            if (removed) {
+                record.status = "cancelled";
+                record.completedAt = Date.now();
+                this.dispatchEvent(new CustomEvent("job:cancelled", { detail: { job: record } }));
+                return true;
+            }
+        }
+        const active = this.activeJobs.get(jobId);
+        if (active?.cancel) {
+            await active.cancel();
+            record.status = "cancelled";
+            record.completedAt = Date.now();
+            this.dispatchEvent(new CustomEvent("job:cancelled", { detail: { job: record } }));
+            return true;
+        }
+        return false;
+    }
+    async shutdown() {
+        this.clientManager.destroy();
+        await this.queue.shutdown();
+        for (const [, ctx] of this.activeJobs) {
+            ctx.release({ success: false });
+        }
+        this.activeJobs.clear();
+    }
+    async getQueueStats() {
+        return this.queue.stats();
+    }
+    normalizeWorkflow(input) {
+        if (typeof input === "string") {
+            return JSON.parse(input);
+        }
+        if (input instanceof Workflow) {
+            return cloneDeep(input.json ?? {});
+        }
+        if (typeof input?.toJSON === "function") {
+            return cloneDeep(input.toJSON());
+        }
+        return cloneDeep(input);
+    }
+    generateJobId() {
+        try {
+            return randomUUID();
+        }
+        catch {
+            return WorkflowPool.fallbackId();
+        }
+    }
+    static fallbackId() {
+        return (globalThis.crypto && "randomUUID" in globalThis.crypto)
+            ? globalThis.crypto.randomUUID()
+            : `job_${Math.random().toString(36).slice(2, 10)}`;
+    }
+    scheduleProcess(delayMs) {
+        const wait = Math.max(delayMs, 10);
+        setTimeout(() => {
+            void this.processQueue();
+        }, wait);
+    }
+    applyAutoSeed(workflow) {
+        const autoSeeds = {};
+        for (const [nodeId, nodeValue] of Object.entries(workflow)) {
+            if (!nodeValue || typeof nodeValue !== "object")
+                continue;
+            const inputs = nodeValue.inputs;
+            if (!inputs || typeof inputs !== "object")
+                continue;
+            if (typeof inputs.seed === "number" && inputs.seed === -1) {
+                const val = Math.floor(Math.random() * 2_147_483_647);
+                inputs.seed = val;
+                autoSeeds[nodeId] = val;
+            }
+        }
+        return autoSeeds;
+    }
+    async processQueue() {
+        if (this.processing) {
+            return;
+        }
+        this.processing = true;
+        try {
+            while (true) {
+                const reservation = await this.queue.reserve();
+                if (!reservation) {
+                    break;
+                }
+                const job = this.jobStore.get(reservation.payload.jobId);
+                if (!job) {
+                    await this.queue.commit(reservation.reservationId);
+                    continue;
+                }
+                const lease = this.clientManager.claim(job);
+                if (!lease) {
+                    await this.queue.retry(reservation.reservationId, { delayMs: job.options.retryDelayMs });
+                    this.scheduleProcess(job.options.retryDelayMs);
+                    break;
+                }
+                this.runJob({ reservation, job, clientId: lease.clientId, release: lease.release }).catch((error) => {
+                    console.error("[WorkflowPool] Unhandled job error", error);
+                });
+            }
+        }
+        finally {
+            this.processing = false;
+        }
+    }
+    async runJob(ctx) {
+        const { reservation, job, clientId, release } = ctx;
+        const managed = this.clientManager.getClient(clientId);
+        const client = managed?.client;
+        if (!client) {
+            await this.queue.retry(reservation.reservationId, { delayMs: job.options.retryDelayMs });
+            release({ success: false });
+            return;
+        }
+        job.status = "running";
+        job.clientId = clientId;
+        job.attempts += 1;
+        reservation.payload.attempts = job.attempts;
+        job.startedAt = Date.now();
+        this.dispatchEvent(new CustomEvent("job:started", { detail: { job } }));
+        const workflowPayload = cloneDeep(reservation.payload.workflow);
+        if (job.attachments?.length) {
+            for (const attachment of job.attachments) {
+                const filename = attachment.filename ?? `${job.jobId}-${attachment.nodeId}-${attachment.inputName}.bin`;
+                const blob = attachment.file instanceof Buffer ? new Blob([new Uint8Array(attachment.file)]) : attachment.file;
+                await client.ext.file.uploadImage(blob, filename, { override: true });
+                const node = workflowPayload[attachment.nodeId];
+                if (node?.inputs) {
+                    node.inputs[attachment.inputName] = filename;
+                }
+            }
+        }
+        const autoSeeds = this.applyAutoSeed(workflowPayload);
+        let wfInstance = Workflow.from(workflowPayload);
+        if (job.options.includeOutputs?.length) {
+            for (const nodeId of job.options.includeOutputs) {
+                if (nodeId) {
+                    wfInstance = wfInstance.output(nodeId);
+                }
+            }
+        }
+        wfInstance.inferDefaultOutputs?.();
+        // Use stored metadata if available (from Workflow instance), otherwise extract from recreated instance
+        const outputNodeIds = reservation.payload.workflowMeta?.outputNodeIds ??
+            wfInstance.outputNodeIds ??
+            job.options.includeOutputs ?? [];
+        const outputAliases = reservation.payload.workflowMeta?.outputAliases ??
+            wfInstance.outputAliases ?? {};
+        let promptBuilder = new PromptBuilder(wfInstance.json, wfInstance.inputPaths ?? [], outputNodeIds);
+        for (const nodeId of outputNodeIds) {
+            const alias = outputAliases[nodeId] ?? nodeId;
+            promptBuilder = promptBuilder.setOutputNode(alias, nodeId);
+        }
+        const wrapper = new CallWrapper(client, promptBuilder);
+        let pendingSettled = false;
+        let resolvePending;
+        let rejectPending;
+        const pendingPromise = new Promise((resolve, reject) => {
+            resolvePending = () => {
+                if (!pendingSettled) {
+                    pendingSettled = true;
+                    resolve();
+                }
+            };
+            rejectPending = (err) => {
+                if (!pendingSettled) {
+                    pendingSettled = true;
+                    reject(err);
+                }
+            };
+        });
+        let resolveCompletion;
+        let rejectCompletion;
+        const completionPromise = new Promise((resolve, reject) => {
+            resolveCompletion = resolve;
+            rejectCompletion = reject;
+        });
+        wrapper.onProgress((progress, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            this.dispatchEvent(new CustomEvent("job:progress", {
+                detail: { jobId: job.jobId, clientId, progress }
+            }));
+        });
+        wrapper.onPreview((blob, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            this.dispatchEvent(new CustomEvent("job:preview", {
+                detail: { jobId: job.jobId, clientId, blob }
+            }));
+        });
+        wrapper.onPreviewMeta((payload, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            this.dispatchEvent(new CustomEvent("job:preview_meta", {
+                detail: { jobId: job.jobId, clientId, payload }
+            }));
+        });
+        wrapper.onOutput((key, data, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            this.dispatchEvent(new CustomEvent("job:output", {
+                detail: { jobId: job.jobId, clientId, key: String(key), data }
+            }));
+        });
+        wrapper.onPending((promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            this.dispatchEvent(new CustomEvent("job:accepted", { detail: { job } }));
+            resolvePending?.();
+        });
+        wrapper.onStart((promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+        });
+        wrapper.onFinished((data, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            job.status = "completed";
+            job.lastError = undefined;
+            const resultPayload = {};
+            for (const nodeId of outputNodeIds) {
+                const alias = outputAliases[nodeId] ?? nodeId;
+                // CallWrapper uses alias keys when mapOutputKeys is configured, fallback to nodeId
+                const nodeResult = data[alias];
+                const fallbackResult = data[nodeId];
+                const finalResult = nodeResult !== undefined ? nodeResult : fallbackResult;
+                resultPayload[alias] = finalResult;
+            }
+            resultPayload._nodes = [...outputNodeIds];
+            resultPayload._aliases = { ...outputAliases };
+            if (job.promptId) {
+                resultPayload._promptId = job.promptId;
+            }
+            if (Object.keys(autoSeeds).length) {
+                resultPayload._autoSeeds = { ...autoSeeds };
+            }
+            job.result = resultPayload;
+            job.completedAt = Date.now();
+            this.dispatchEvent(new CustomEvent("job:completed", { detail: { job } }));
+            resolveCompletion?.();
+        });
+        wrapper.onFailed((error, promptId) => {
+            if (!job.promptId && promptId) {
+                job.promptId = promptId;
+            }
+            job.lastError = error;
+            rejectPending?.(error);
+            rejectCompletion?.(error);
+        });
+        try {
+            const exec = wrapper.run();
+            await pendingPromise;
+            this.activeJobs.set(job.jobId, {
+                reservation,
+                job,
+                clientId,
+                release,
+                cancel: async () => {
+                    try {
+                        if (job.promptId) {
+                            await client.ext.queue.interrupt(job.promptId);
+                        }
+                    }
+                    finally {
+                        this.activeJobs.delete(job.jobId);
+                        await this.queue.discard(reservation.reservationId, new Error("cancelled"));
+                        release({ success: false });
+                    }
+                }
+            });
+            const result = await exec;
+            if (result === false) {
+                // Execution failed - try to get the error from completionPromise rejection
+                try {
+                    await completionPromise;
+                }
+                catch (err) {
+                    throw err;
+                }
+                throw job.lastError ?? new Error("Execution failed");
+            }
+            await completionPromise;
+            await this.queue.commit(reservation.reservationId);
+            release({ success: true });
+        }
+        catch (error) {
+            const latestStatus = this.jobStore.get(job.jobId)?.status;
+            if (latestStatus === "cancelled") {
+                release({ success: false });
+                return;
+            }
+            job.lastError = error;
+            job.status = "failed";
+            this.clientManager.recordFailure(clientId, job, error);
+            const remainingAttempts = job.options.maxAttempts - job.attempts;
+            const willRetry = remainingAttempts > 0;
+            this.dispatchEvent(new CustomEvent("job:failed", {
+                detail: { job, willRetry }
+            }));
+            if (willRetry) {
+                const delay = this.opts.retryBackoffMs ?? job.options.retryDelayMs;
+                this.dispatchEvent(new CustomEvent("job:retrying", { detail: { job, delayMs: delay } }));
+                job.status = "queued";
+                job.clientId = undefined;
+                job.promptId = undefined;
+                job.startedAt = undefined;
+                job.completedAt = undefined;
+                job.result = undefined;
+                await this.queue.retry(reservation.reservationId, { delayMs: delay });
+                this.dispatchEvent(new CustomEvent("job:queued", { detail: { job } }));
+                this.scheduleProcess(delay);
+                release({ success: false });
+            }
+            else {
+                job.completedAt = Date.now();
+                await this.queue.discard(reservation.reservationId, error);
+                release({ success: false });
+            }
+        }
+        finally {
+            this.activeJobs.delete(job.jobId);
+            void this.processQueue();
+        }
+    }
+}
+//# sourceMappingURL=WorkflowPool.js.map

package/dist/pool/WorkflowPool.js.map

@@ -0,0 +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;AA8E7C,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,EAAE;YACpD,qBAAqB,EAAE,IAAI,EAAE,qBAAqB,IAAI,KAAK;SAC5D,CAAC,CAAC;QACH,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,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC7B,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"}

package/dist/pool/client/ClientManager.d.ts

@@ -0,0 +1,71 @@
+import type { ComfyApi } from "../../client.js";
+import { TypedEventTarget } from "../../typed-event-target.js";
+import type { JobRecord } from "../types/job.js";
+import type { FailoverStrategy } from "../failover/Strategy.js";
+import type { WorkflowPoolEventMap } from "../types/events.js";
+export interface ManagedClient {
+    client: ComfyApi;
+    id: string;
+    online: boolean;
+    busy: boolean;
+    lastError?: unknown;
+    lastSeenAt: number;
+    supportedWorkflows: Set<string>;
+}
+interface ClientLease {
+    client: ComfyApi;
+    clientId: string;
+    release: (opts?: {
+        success?: boolean;
+    }) => void;
+}
+export declare class ClientManager extends TypedEventTarget<WorkflowPoolEventMap> {
+    private clients;
+    private strategy;
+    private healthCheckInterval;
+    private readonly healthCheckIntervalMs;
+    /**
+     * Create a new ClientManager for managing ComfyUI client connections.
+     *
+     * @param strategy - Failover strategy for handling client failures
+     * @param opts - Configuration options
+     * @param opts.healthCheckIntervalMs - Interval (ms) for health check pings to keep connections alive.
+     *   Set to 0 to disable. Default: 30000 (30 seconds).
+     */
+    constructor(strategy: FailoverStrategy, opts?: {
+        /**
+         * Interval in milliseconds for health check pings.
+         * Health checks keep idle connections alive by periodically polling client status.
+         * @default 30000 (30 seconds)
+         */
+        healthCheckIntervalMs?: number;
+    });
+    private emitBlocked;
+    private emitUnblocked;
+    initialize(clients: ComfyApi[]): Promise<void>;
+    addClient(client: ComfyApi): Promise<void>;
+    list(): ManagedClient[];
+    getClient(clientId: string): ManagedClient | undefined;
+    claim(job: JobRecord): ClientLease | null;
+    recordFailure(clientId: string, job: JobRecord, error: unknown): void;
+    /**
+     * Start periodic health check to keep connections alive and detect issues early.
+     * Pings idle clients by polling their queue status.
+     */
+    private startHealthCheck;
+    /**
+     * Perform health check on all clients.
+     * For idle clients, polls queue status to keep WebSocket alive and detect connection issues.
+     */
+    private performHealthCheck;
+    /**
+     * Stop health check interval (called during shutdown).
+     */
+    stopHealthCheck(): void;
+    /**
+     * Cleanup resources when destroying the manager.
+     */
+    destroy(): void;
+}
+export {};
+//# sourceMappingURL=ClientManager.d.ts.map

package/dist/pool/client/ClientManager.d.ts.map

@@ -0,0 +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;IACnC,OAAO,CAAC,mBAAmB,CAA+B;IAC1D,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAS;IAE/C;;;;;;;OAOG;gBACS,QAAQ,EAAE,gBAAgB,EAAE,IAAI,CAAC,EAAE;QAC7C;;;;WAIG;QACH,qBAAqB,CAAC,EAAE,MAAM,CAAA;KAC/B;IAMD,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,aAAa;IAIf,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAO9C,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;IAkB9D;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAYxB;;;OAGG;YACW,kBAAkB;IAkBhC;;OAEG;IACH,eAAe,IAAI,IAAI;IAOvB;;OAEG;IACH,OAAO,IAAI,IAAI;CAGhB"}