comfyui-node 1.5.0 → 1.6.0

package/dist/pool/SmartPoolV2.js

@@ -0,0 +1,587 @@
+import { randomUUID } from "node:crypto";
+import { TypedEventTarget } from "../typed-event-target.js";
+import { ComfyApi } from "../client.js";
+import { PromptBuilder } from "../prompt-builder.js";
+import { MemoryQueueAdapter } from "./queue/adapters/memory.js";
+import { hashWorkflow } from "./utils/hash.js";
+// ============================================================================
+// MAIN CLASS
+// ============================================================================
+export class SmartPoolV2 extends TypedEventTarget {
+    // Client management
+    clientMap = new Map();
+    // Affinity groups and queues
+    affinityGroups = new Map();
+    defaultQueue;
+    // Job tracking
+    jobStore = new Map();
+    executionContexts = new Map();
+    // Server state
+    idleServers = new Set();
+    serverPerformance = new Map();
+    // Pool configuration
+    options;
+    // Pool state
+    isReady;
+    readyResolve;
+    // =========================================================================
+    // CONSTRUCTOR
+    // =========================================================================
+    constructor(clients, options) {
+        super();
+        this.options = {
+            connectionTimeoutMs: options?.connectionTimeoutMs ?? 10000,
+            jobExecutionTimeoutMs: options?.jobExecutionTimeoutMs ?? 5 * 60 * 1000, // 5 min
+            groupIdleTimeoutMs: options?.groupIdleTimeoutMs ?? 60 * 1000, // 60 sec
+            maxQueueDepth: options?.maxQueueDepth ?? 1000
+        };
+        // Initialize clients
+        for (const client of clients) {
+            if (typeof client === "string") {
+                const apiClient = new ComfyApi(client);
+                this.clientMap.set(apiClient.apiHost, apiClient);
+            }
+            else {
+                this.clientMap.set(client.apiHost, client);
+            }
+        }
+        // Create default queue for unaffinitized jobs
+        this.defaultQueue = this.createAffinityGroup("default", []);
+        // Setup ready promise
+        this.isReady = new Promise((resolve) => {
+            this.readyResolve = resolve;
+        });
+    }
+    // =========================================================================
+    // PUBLIC API
+    // =========================================================================
+    /**
+     * Initialize pool and connect all clients
+     */
+    async connect() {
+        const connectionPromises = [];
+        for (const [url, client] of this.clientMap.entries()) {
+            connectionPromises.push(new Promise((resolve, reject) => {
+                const timeout = setTimeout(() => {
+                    client.abortReconnect();
+                    reject(new Error(`Connection to client ${url} timed out`));
+                }, this.options.connectionTimeoutMs);
+                client
+                    .init(1)
+                    .then(() => {
+                    clearTimeout(timeout);
+                    console.log(`[SmartPoolV2] Connected to ${url}`);
+                    this.idleServers.add(client.apiHost);
+                    resolve();
+                })
+                    .catch((err) => {
+                    clearTimeout(timeout);
+                    reject(err);
+                });
+            }));
+        }
+        await Promise.all(connectionPromises);
+        this.readyResolve?.();
+    }
+    /**
+     * Wait for pool to be ready
+     */
+    ready() {
+        return this.isReady;
+    }
+    /**
+     * Enqueue a workflow - automatically routed by workflow hash
+     * Optional preferredClientIds overrides default routing for this specific job
+     */
+    async enqueue(workflow, options) {
+        const jobId = randomUUID();
+        const workflowHash = workflow.structureHash || hashWorkflow(workflow.json || workflow);
+        const workflowJson = workflow.json || workflow;
+        const outputNodeIds = workflow.outputNodeIds || [];
+        const outputAliases = workflow.outputAliases || {};
+        // Find group by workflow hash, fall back to default
+        let groupId = workflowHash;
+        if (!this.affinityGroups.has(groupId)) {
+            groupId = "default";
+        }
+        const group = this.affinityGroups.get(groupId);
+        if (!group) {
+            throw new Error(`No affinity group for workflow hash "${workflowHash}"`);
+        }
+        // Create job record
+        const jobRecord = {
+            jobId,
+            workflow: workflowJson,
+            workflowHash,
+            options: {
+                maxAttempts: 3,
+                retryDelayMs: 1000,
+                priority: options?.priority ?? 0,
+                // Use per-job preferences if provided, otherwise use group defaults
+                preferredClientIds: options?.preferredClientIds?.length ? options.preferredClientIds : group.preferredServerIds,
+                excludeClientIds: [],
+                metadata: options?.metadata || {}
+            },
+            attempts: 0,
+            enqueuedAt: Date.now(),
+            workflowMeta: {
+                outputNodeIds,
+                outputAliases
+            },
+            status: "queued"
+        };
+        // Store job
+        this.jobStore.set(jobId, jobRecord);
+        // Enqueue to group
+        const payload = jobRecord;
+        await group.queueAdapter.enqueue(payload, { priority: options?.priority ?? 0 });
+        // Emit event
+        this.dispatchEvent(new CustomEvent("job:queued", { detail: { job: jobRecord } }));
+        // Trigger processing immediately (event-driven)
+        setImmediate(() => this.processAffinityGroup(groupId));
+        return jobId;
+    }
+    /**
+     * Set workflow affinity - auto-creates group by workflow hash
+     * Maps workflow hash to preferred servers
+     */
+    setAffinity(workflow, affinity) {
+        const workflowHash = hashWorkflow(workflow);
+        // Create group with hash as ID if doesn't exist
+        if (!this.affinityGroups.has(workflowHash)) {
+            this.createAffinityGroup(workflowHash, affinity.preferredClientIds || []);
+        }
+        const group = this.affinityGroups.get(workflowHash);
+        if (group) {
+            group.workflowHashes.add(workflowHash);
+        }
+    }
+    /**
+     * Get job by ID
+     */
+    getJob(jobId) {
+        return this.jobStore.get(jobId);
+    }
+    /**
+     * Shutdown pool
+     */
+    shutdown() {
+        // Cancel all timeouts
+        for (const group of this.affinityGroups.values()) {
+            if (group.idleTimeoutHandle) {
+                clearTimeout(group.idleTimeoutHandle);
+            }
+        }
+        if (this.defaultQueue?.idleTimeoutHandle) {
+            clearTimeout(this.defaultQueue.idleTimeoutHandle);
+        }
+        // Cancel all job timeouts
+        for (const ctx of this.executionContexts.values()) {
+            if (ctx.timeoutHandle) {
+                clearTimeout(ctx.timeoutHandle);
+            }
+        }
+        // Destroy clients
+        for (const client of this.clientMap.values()) {
+            try {
+                client.destroy();
+            }
+            catch (err) {
+                console.error(`[SmartPoolV2] Error destroying client: ${err}`);
+            }
+        }
+    }
+    /**
+     * Get server performance metrics
+     */
+    getServerPerformance(clientId) {
+        return this.serverPerformance.get(clientId);
+    }
+    // =========================================================================
+    // PRIVATE: AFFINITY GROUP MANAGEMENT
+    // =========================================================================
+    createAffinityGroup(groupId, preferredServerIds) {
+        const group = {
+            id: groupId,
+            preferredServerIds,
+            workflowHashes: new Set(),
+            queueAdapter: new MemoryQueueAdapter(),
+            isProcessing: false,
+            lastJobCompletedMs: Date.now(),
+            jobsEnqueued: 0,
+            jobsCompleted: 0,
+            jobsFailed: 0
+        };
+        this.affinityGroups.set(groupId, group);
+        return group;
+    }
+    // =========================================================================
+    // PRIVATE: QUEUE PROCESSING (EVENT-DRIVEN)
+    // =========================================================================
+    /**
+     * Process affinity group queue - triggered by events only (no polling)
+     */
+    async processAffinityGroup(groupId) {
+        const group = this.affinityGroups.get(groupId);
+        if (!group)
+            return;
+        // Reentrancy guard: if already processing, defer
+        if (group.isProcessing) {
+            if (!group.processingDeferred) {
+                group.processingDeferred = new Promise((resolve) => {
+                    setImmediate(() => {
+                        group.isProcessing = false;
+                        this.processAffinityGroup(groupId).then(resolve);
+                    });
+                });
+            }
+            return group.processingDeferred;
+        }
+        group.isProcessing = true;
+        try {
+            while (true) {
+                // Peek at waiting jobs first
+                const waitingJobs = await group.queueAdapter.peek(100);
+                if (waitingJobs.length === 0) {
+                    break; // No waiting jobs
+                }
+                // Get the first waiting job
+                const jobPayload = waitingJobs[0];
+                const job = this.jobStore.get(jobPayload.jobId);
+                if (!job) {
+                    // Job not found, discard from queue
+                    await group.queueAdapter.discard(jobPayload.jobId, new Error("Job not found"));
+                    continue;
+                }
+                // Find idle servers compatible with this specific job
+                // First check job's preferred clients, then group's preferred servers
+                const preferredServerIds = job.options.preferredClientIds?.length
+                    ? job.options.preferredClientIds
+                    : group.preferredServerIds;
+                const compatibleIdleServers = Array.from(this.idleServers).filter((serverId) => {
+                    // If preferred servers specified (job or group), must match
+                    if (preferredServerIds.length > 0) {
+                        return preferredServerIds.includes(serverId);
+                    }
+                    return true;
+                });
+                if (compatibleIdleServers.length === 0) {
+                    break; // No idle compatible servers for this job
+                }
+                // Sort compatible servers by performance (fastest first)
+                const sortedServers = this.sortServersByPerformance(compatibleIdleServers);
+                const selectedServerId = sortedServers[0];
+                const selectedClient = this.clientMap.get(selectedServerId);
+                if (!selectedClient) {
+                    break;
+                }
+                // Reserve job
+                const reservation = await group.queueAdapter.reserveById(jobPayload.jobId);
+                if (!reservation) {
+                    continue;
+                }
+                // Mark server as no longer idle (synchronous)
+                this.idleServers.delete(selectedServerId);
+                // Enqueue job on server (synchronous, fires in background)
+                await this.enqueueJobOnServer(job, selectedClient, groupId, reservation);
+            }
+        }
+        finally {
+            group.isProcessing = false;
+            // Check for deferred processing
+            if (group.processingDeferred) {
+                group.processingDeferred = undefined;
+            }
+        }
+    }
+    // =========================================================================
+    // PRIVATE: JOB EXECUTION (NO CALLWRAPPER)
+    // =========================================================================
+    /**
+     * Enqueue job on server and manage execution
+     */
+    async enqueueJobOnServer(job, client, groupId, reservation) {
+        const group = this.affinityGroups.get(groupId);
+        if (!group)
+            return;
+        job.attempts += 1;
+        job.status = "running";
+        job.clientId = client.apiHost;
+        job.startedAt = Date.now();
+        try {
+            // Clone workflow to avoid mutations
+            const workflowJson = JSON.parse(JSON.stringify(job.workflow));
+            const outputNodeIds = job.workflowMeta?.outputNodeIds || [];
+            // Auto-randomize seeds
+            try {
+                for (const node of Object.values(workflowJson)) {
+                    const n = node;
+                    if (n?.inputs?.seed === -1) {
+                        n.inputs.seed = Math.floor(Math.random() * 2_147_483_647);
+                    }
+                }
+            }
+            catch {
+                /* non-fatal */
+            }
+            // Build prompt
+            const pb = new PromptBuilder(workflowJson, [], outputNodeIds);
+            for (const nodeId of outputNodeIds) {
+                pb.setOutputNode(nodeId, nodeId);
+            }
+            const promptJson = pb.prompt;
+            // Append to server queue
+            let queueResponse;
+            try {
+                queueResponse = await client.ext.queue.appendPrompt(promptJson);
+            }
+            catch (err) {
+                throw new Error(`Failed to enqueue job: ${err}`);
+            }
+            const promptId = queueResponse.prompt_id;
+            job.promptId = promptId;
+            // Create execution context
+            const ctx = {
+                job,
+                groupId,
+                promptId
+            };
+            this.executionContexts.set(job.jobId, ctx);
+            // Emit accepted event
+            this.dispatchEvent(new CustomEvent("job:accepted", {
+                detail: { job, clientId: client.apiHost }
+            }));
+            this.dispatchEvent(new CustomEvent("job:started", {
+                detail: { job, clientId: client.apiHost, promptId }
+            }));
+            // Set up execution timeout (5 min)
+            ctx.timeoutHandle = setTimeout(() => {
+                console.warn(`[SmartPoolV2] Job ${job.jobId} execution timeout`);
+                this.handleJobTimeout(job.jobId);
+            }, this.options.jobExecutionTimeoutMs);
+            // Set up event listeners (strict prompt_id matching)
+            const outputMap = {};
+            let outputsCollected = 0;
+            const expectedOutputCount = outputNodeIds.length;
+            ctx.executedHandler = (ev) => {
+                // Strict prompt_id check
+                if (ev.detail.prompt_id !== promptId) {
+                    return;
+                }
+                const nodeId = ev.detail.node;
+                const output = ev.detail.output;
+                outputMap[nodeId] = output;
+                outputsCollected++;
+                // All outputs collected?
+                if (outputsCollected === expectedOutputCount) {
+                    this.handleJobCompletion(job.jobId, groupId, outputMap, client.apiHost);
+                }
+            };
+            ctx.executionSuccessHandler = async (ev) => {
+                if (ev.detail.prompt_id !== promptId) {
+                    return;
+                }
+                // Try to get missing outputs from history if needed
+                if (outputsCollected < expectedOutputCount) {
+                    try {
+                        const history = await client.ext.history.getHistory(promptId);
+                        if (history?.outputs) {
+                            for (const [nodeIdStr, nodeOutput] of Object.entries(history.outputs)) {
+                                const nodeId = nodeIdStr;
+                                if (!outputMap[nodeId] && nodeOutput) {
+                                    outputMap[nodeId] = nodeOutput;
+                                    outputsCollected++;
+                                }
+                            }
+                        }
+                    }
+                    catch {
+                        /* non-fatal */
+                    }
+                }
+                // Complete job regardless
+                this.handleJobCompletion(job.jobId, groupId, outputMap, client.apiHost);
+            };
+            ctx.executionErrorHandler = (ev) => {
+                if (ev.detail.prompt_id !== promptId) {
+                    return;
+                }
+                const error = new Error(`Execution error: ${ev.detail.exception_type}`);
+                this.handleJobFailure(job.jobId, groupId, error);
+            };
+            // Attach listeners
+            client.on("executed", ctx.executedHandler);
+            client.on("execution_success", ctx.executionSuccessHandler);
+            client.on("execution_error", ctx.executionErrorHandler);
+            // Commit to queue
+            await group.queueAdapter.commit(reservation.reservationId);
+        }
+        catch (err) {
+            const error = err instanceof Error ? err : new Error(String(err));
+            // Retry or fail
+            if (job.attempts < job.options.maxAttempts) {
+                await group.queueAdapter.retry(reservation.reservationId, {
+                    delayMs: job.options.retryDelayMs
+                });
+            }
+            else {
+                await group.queueAdapter.discard(reservation.reservationId, error);
+                this.handleJobFailure(job.jobId, groupId, error);
+            }
+            // Mark server idle again
+            this.idleServers.add(client.apiHost);
+            this.dispatchEvent(new CustomEvent("server:idle", {
+                detail: { clientId: client.apiHost, groupId }
+            }));
+            // Trigger processing
+            setImmediate(() => this.processAffinityGroup(groupId));
+        }
+    }
+    /**
+     * Handle job completion
+     */
+    handleJobCompletion(jobId, groupId, outputMap, clientId) {
+        const job = this.jobStore.get(jobId);
+        if (!job)
+            return;
+        const ctx = this.executionContexts.get(jobId);
+        if (ctx?.timeoutHandle) {
+            clearTimeout(ctx.timeoutHandle);
+        }
+        // Clean up listeners
+        if (ctx) {
+            const client = this.clientMap.get(clientId);
+            if (client) {
+                if (ctx.executedHandler)
+                    client.off("executed", ctx.executedHandler);
+                if (ctx.executionSuccessHandler)
+                    client.off("execution_success", ctx.executionSuccessHandler);
+                if (ctx.executionErrorHandler)
+                    client.off("execution_error", ctx.executionErrorHandler);
+            }
+        }
+        // Update job
+        job.status = "completed";
+        job.result = outputMap;
+        job.completedAt = Date.now();
+        const executionTimeMs = job.completedAt - (job.startedAt || job.completedAt);
+        this.updateServerPerformance(clientId, executionTimeMs);
+        // Update group stats
+        const group = this.affinityGroups.get(groupId);
+        if (group) {
+            group.jobsCompleted++;
+            group.lastJobCompletedMs = Date.now();
+            // Reset idle timeout for this group
+            if (group.idleTimeoutHandle) {
+                clearTimeout(group.idleTimeoutHandle);
+            }
+            group.idleTimeoutHandle = setTimeout(() => {
+                this.dispatchEvent(new CustomEvent("group:idle-timeout", {
+                    detail: { groupId, reason: "No jobs completed in idle threshold" }
+                }));
+            }, this.options.groupIdleTimeoutMs);
+        }
+        // Mark server idle
+        this.idleServers.add(clientId);
+        this.dispatchEvent(new CustomEvent("server:idle", {
+            detail: { clientId, groupId }
+        }));
+        // Emit completed event
+        this.dispatchEvent(new CustomEvent("job:completed", { detail: { job } }));
+        // Clean up context
+        this.executionContexts.delete(jobId);
+        // Trigger processing
+        setImmediate(() => this.processAffinityGroup(groupId));
+    }
+    /**
+     * Handle job failure
+     */
+    handleJobFailure(jobId, groupId, error) {
+        const job = this.jobStore.get(jobId);
+        if (!job)
+            return;
+        const ctx = this.executionContexts.get(jobId);
+        if (ctx?.timeoutHandle) {
+            clearTimeout(ctx.timeoutHandle);
+        }
+        // Clean up listeners
+        if (ctx && job.clientId) {
+            const client = this.clientMap.get(job.clientId);
+            if (client) {
+                if (ctx.executedHandler)
+                    client.off("executed", ctx.executedHandler);
+                if (ctx.executionSuccessHandler)
+                    client.off("execution_success", ctx.executionSuccessHandler);
+                if (ctx.executionErrorHandler)
+                    client.off("execution_error", ctx.executionErrorHandler);
+            }
+        }
+        job.status = "failed";
+        job.lastError = error;
+        job.completedAt = Date.now();
+        // Update group stats
+        const group = this.affinityGroups.get(groupId);
+        if (group) {
+            group.jobsFailed++;
+        }
+        // Mark server idle
+        if (job.clientId) {
+            this.idleServers.add(job.clientId);
+            this.dispatchEvent(new CustomEvent("server:idle", {
+                detail: { clientId: job.clientId, groupId }
+            }));
+        }
+        // Emit failed event
+        this.dispatchEvent(new CustomEvent("job:failed", {
+            detail: { job, error, willRetry: false }
+        }));
+        // Clean up context
+        this.executionContexts.delete(jobId);
+        // Trigger processing
+        setImmediate(() => this.processAffinityGroup(groupId));
+    }
+    /**
+     * Handle job timeout
+     */
+    handleJobTimeout(jobId) {
+        const job = this.jobStore.get(jobId);
+        if (!job)
+            return;
+        const ctx = this.executionContexts.get(jobId);
+        const groupId = ctx?.groupId || "default";
+        const error = new Error(`Job execution timeout after ${this.options.jobExecutionTimeoutMs}ms`);
+        this.handleJobFailure(jobId, groupId, error);
+    }
+    // =========================================================================
+    // PRIVATE: PERFORMANCE TRACKING
+    // =========================================================================
+    updateServerPerformance(clientId, executionTimeMs) {
+        let metrics = this.serverPerformance.get(clientId);
+        if (!metrics) {
+            metrics = {
+                clientId,
+                totalJobsCompleted: 0,
+                totalExecutionTimeMs: 0,
+                averageExecutionTimeMs: 0
+            };
+            this.serverPerformance.set(clientId, metrics);
+        }
+        metrics.totalJobsCompleted++;
+        metrics.totalExecutionTimeMs += executionTimeMs;
+        metrics.lastJobDurationMs = executionTimeMs;
+        metrics.averageExecutionTimeMs = metrics.totalExecutionTimeMs / metrics.totalJobsCompleted;
+    }
+    sortServersByPerformance(serverIds) {
+        return [...serverIds].sort((a, b) => {
+            const metricsA = this.serverPerformance.get(a);
+            const metricsB = this.serverPerformance.get(b);
+            // Untracked servers go to end
+            if (!metricsA)
+                return 1;
+            if (!metricsB)
+                return -1;
+            return metricsA.averageExecutionTimeMs - metricsB.averageExecutionTimeMs;
+        });
+    }
+}
+//# sourceMappingURL=SmartPoolV2.js.map

package/dist/pool/SmartPoolV2.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SmartPoolV2.js","sourceRoot":"","sources":["../../src/pool/SmartPoolV2.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAExC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAIhE,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAkE/C,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E,MAAM,OAAO,WAAY,SAAQ,gBAAqC;IACpE,oBAAoB;IACZ,SAAS,GAA0B,IAAI,GAAG,EAAE,CAAC;IAErD,6BAA6B;IACrB,cAAc,GAAoC,IAAI,GAAG,EAAE,CAAC;IAC5D,YAAY,CAAsB;IAE1C,eAAe;IACP,QAAQ,GAA0B,IAAI,GAAG,EAAE,CAAC;IAC5C,iBAAiB,GAAoC,IAAI,GAAG,EAAE,CAAC;IAEvE,eAAe;IACP,WAAW,GAAgB,IAAI,GAAG,EAAE,CAAC;IACrC,iBAAiB,GAA0C,IAAI,GAAG,EAAE,CAAC;IAE7E,qBAAqB;IACb,OAAO,CAA+B;IAE9C,aAAa;IACL,OAAO,CAAgB;IACvB,YAAY,CAAc;IAElC,4EAA4E;IAC5E,cAAc;IACd,4EAA4E;IAE5E,YACE,OAA8B,EAC9B,OAA4B;QAE5B,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,OAAO,GAAG;YACb,mBAAmB,EAAE,OAAO,EAAE,mBAAmB,IAAI,KAAK;YAC1D,qBAAqB,EAAE,OAAO,EAAE,qBAAqB,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ;YAChF,kBAAkB,EAAE,OAAO,EAAE,kBAAkB,IAAI,EAAE,GAAG,IAAI,EAAc,SAAS;YACnF,aAAa,EAAE,OAAO,EAAE,aAAa,IAAI,IAAI;SAC9C,CAAC;QAEF,qBAAqB;QACrB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,SAAS,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACvC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAE5D,sBAAsB;QACtB,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACrC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,4EAA4E;IAC5E,aAAa;IACb,4EAA4E;IAE5E;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,kBAAkB,GAAoB,EAAE,CAAC;QAE/C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;YACrD,kBAAkB,CAAC,IAAI,CACrB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACpC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC9B,MAAM,CAAC,cAAc,EAAE,CAAC;oBACxB,MAAM,CAAC,IAAI,KAAK,CAAC,wBAAwB,GAAG,YAAY,CAAC,CAAC,CAAC;gBAC7D,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;gBAErC,MAAM;qBACH,IAAI,CAAC,CAAC,CAAC;qBACP,IAAI,CAAC,GAAG,EAAE;oBACT,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,8BAA8B,GAAG,EAAE,CAAC,CAAC;oBACjD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBACrC,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBACb,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACtC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK;QACH,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO,CACX,QAAuB,EACvB,OAIC;QAED,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,QAAQ,CAAC,aAAa,IAAI,YAAY,CAAE,QAAgB,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC;QAChG,MAAM,YAAY,GAAI,QAAgB,CAAC,IAAI,IAAI,QAAQ,CAAC;QACxD,MAAM,aAAa,GAAI,QAAgB,CAAC,aAAa,IAAI,EAAE,CAAC;QAC5D,MAAM,aAAa,GAAI,QAAgB,CAAC,aAAa,IAAI,EAAE,CAAC;QAE5D,oDAAoD;QACpD,IAAI,OAAO,GAAG,YAAY,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,OAAO,GAAG,SAAS,CAAC;QACtB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,wCAAwC,YAAY,GAAG,CAAC,CAAC;QAC3E,CAAC;QAED,oBAAoB;QACpB,MAAM,SAAS,GAAc;YAC3B,KAAK;YACL,QAAQ,EAAE,YAAY;YACtB,YAAY;YACZ,OAAO,EAAE;gBACP,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,IAAI;gBAClB,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,CAAC;gBAChC,oEAAoE;gBACpE,kBAAkB,EAAE,OAAO,EAAE,kBAAkB,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,KAAK,CAAC,kBAAkB;gBAC/G,gBAAgB,EAAE,EAAE;gBACpB,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,EAAE;aAClC;YACD,QAAQ,EAAE,CAAC;YACX,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;YACtB,YAAY,EAAE;gBACZ,aAAa;gBACb,aAAa;aACd;YACD,MAAM,EAAE,QAAQ;SACjB,CAAC;QAEF,YAAY;QACZ,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAEpC,mBAAmB;QACnB,MAAM,OAAO,GAAuB,SAAS,CAAC;QAC9C,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhF,aAAa;QACb,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;QAElF,gDAAgD;QAChD,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;QAEvD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;OAGG;IACH,WAAW,CACT,QAAgB,EAChB,QAAgD;QAEhD,MAAM,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QAE5C,gDAAgD;QAChD,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3C,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACpD,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAY;QACjB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,sBAAsB;QACtB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC;YACjD,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;gBAC5B,YAAY,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,EAAE,iBAAiB,EAAE,CAAC;YACzC,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;QACpD,CAAC;QAED,0BAA0B;QAC1B,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,EAAE,CAAC;YAClD,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;gBACtB,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,kBAAkB;QAClB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,GAAG,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,oBAAoB,CAAC,QAAgB;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,4EAA4E;IAC5E,qCAAqC;IACrC,4EAA4E;IAEpE,mBAAmB,CACzB,OAAe,EACf,kBAA4B;QAE5B,MAAM,KAAK,GAAuB;YAChC,EAAE,EAAE,OAAO;YACX,kBAAkB;YAClB,cAAc,EAAE,IAAI,GAAG,EAAE;YACzB,YAAY,EAAE,IAAI,kBAAkB,EAAE;YACtC,YAAY,EAAE,KAAK;YACnB,kBAAkB,EAAE,IAAI,CAAC,GAAG,EAAE;YAC9B,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,4EAA4E;IAC5E,2CAA2C;IAC3C,4EAA4E;IAE5E;;OAEG;IACK,KAAK,CAAC,oBAAoB,CAAC,OAAe;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,iDAAiD;QACjD,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;gBAC9B,KAAK,CAAC,kBAAkB,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;oBACjD,YAAY,CAAC,GAAG,EAAE;wBAChB,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC;wBAC3B,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACnD,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;YACL,CAAC;YACD,OAAO,KAAK,CAAC,kBAAkB,CAAC;QAClC,CAAC;QAED,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;QAE1B,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,6BAA6B;gBAC7B,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACvD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7B,MAAM,CAAC,kBAAkB;gBAC3B,CAAC;gBAED,4BAA4B;gBAC5B,MAAM,UAAU,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;gBAClC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAEhD,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,oCAAoC;oBACpC,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;oBAC/E,SAAS;gBACX,CAAC;gBAED,sDAAsD;gBACtD,sEAAsE;gBACtE,MAAM,kBAAkB,GAAG,GAAG,CAAC,OAAO,CAAC,kBAAkB,EAAE,MAAM;oBAC/D,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,kBAAkB;oBAChC,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC;gBAE7B,MAAM,qBAAqB,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE;oBAC7E,4DAA4D;oBAC5D,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAClC,OAAO,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBAC/C,CAAC;oBACD,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CAAC;gBAEH,IAAI,qBAAqB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvC,MAAM,CAAC,0CAA0C;gBACnD,CAAC;gBAED,yDAAyD;gBACzD,MAAM,aAAa,GAAG,IAAI,CAAC,wBAAwB,CAAC,qBAAqB,CAAC,CAAC;gBAC3E,MAAM,gBAAgB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAE5D,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,MAAM;gBACR,CAAC;gBAED,cAAc;gBACd,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC3E,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,SAAS;gBACX,CAAC;gBAED,8CAA8C;gBAC9C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;gBAE1C,2DAA2D;gBAC3D,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,cAAc,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC;YAE3B,gCAAgC;YAChC,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;gBAC7B,KAAK,CAAC,kBAAkB,GAAG,SAAS,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,0CAA0C;IAC1C,4EAA4E;IAE5E;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC9B,GAAc,EACd,MAAgB,EAChB,OAAe,EACf,WAA6B;QAE7B,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC;QAClB,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC;QACvB,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE3B,IAAI,CAAC;YACH,oCAAoC;YACpC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC9D,MAAM,aAAa,GAAG,GAAG,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,CAAC;YAE5D,uBAAuB;YACvB,IAAI,CAAC;gBACH,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;oBAC/C,MAAM,CAAC,GAAG,IAAW,CAAC;oBACtB,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;wBAC3B,CAAC,CAAC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC;oBAC5D,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,eAAe;YACjB,CAAC;YAED,eAAe;YACf,MAAM,EAAE,GAAG,IAAI,aAAa,CAAC,YAAY,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;YAC9D,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;gBACnC,EAAE,CAAC,aAAa,CAAC,MAAa,EAAE,MAAa,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC;YAE7B,yBAAyB;YACzB,IAAI,aAAa,CAAC;YAClB,IAAI,CAAC;gBACH,aAAa,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YAClE,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC;YACzC,GAAG,CAAC,QAAQ,GAAG,QAAQ,CAAC;YAExB,2BAA2B;YAC3B,MAAM,GAAG,GAAwB;gBAC/B,GAAG;gBACH,OAAO;gBACP,QAAQ;aACT,CAAC;YACF,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAE3C,sBAAsB;YACtB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,cAAc,EAAE;gBAC9B,MAAM,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE;aAC1C,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;gBAC7B,MAAM,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE;aACpD,CAAC,CACH,CAAC;YAEF,mCAAmC;YACnC,GAAG,CAAC,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;gBAClC,OAAO,CAAC,IAAI,CAAC,qBAAqB,GAAG,CAAC,KAAK,oBAAoB,CAAC,CAAC;gBACjE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YAEvC,qDAAqD;YACrD,MAAM,SAAS,GAAwB,EAAE,CAAC;YAC1C,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzB,MAAM,mBAAmB,GAAG,aAAa,CAAC,MAAM,CAAC;YAEjD,GAAG,CAAC,eAAe,GAAG,CAAC,EAAe,EAAE,EAAE;gBACxC,yBAAyB;gBACzB,IAAI,EAAE,CAAC,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO;gBACT,CAAC;gBAED,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;gBAC9B,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC;gBAEhC,SAAS,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;gBAC3B,gBAAgB,EAAE,CAAC;gBAEnB,yBAAyB;gBACzB,IAAI,gBAAgB,KAAK,mBAAmB,EAAE,CAAC;oBAC7C,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC,CAAC;YAEF,GAAG,CAAC,uBAAuB,GAAG,KAAK,EAAE,EAAe,EAAE,EAAE;gBACtD,IAAI,EAAE,CAAC,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO;gBACT,CAAC;gBAED,oDAAoD;gBACpD,IAAI,gBAAgB,GAAG,mBAAmB,EAAE,CAAC;oBAC3C,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;wBAC9D,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;4BACrB,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gCACtE,MAAM,MAAM,GAAG,SAAS,CAAC;gCACzB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;oCACrC,SAAS,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC;oCAC/B,gBAAgB,EAAE,CAAC;gCACrB,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,eAAe;oBACjB,CAAC;gBACH,CAAC;gBAED,0BAA0B;gBAC1B,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAC1E,CAAC,CAAC;YAEF,GAAG,CAAC,qBAAqB,GAAG,CAAC,EAAe,EAAE,EAAE;gBAC9C,IAAI,EAAE,CAAC,MAAM,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO;gBACT,CAAC;gBAED,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,oBAAoB,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;gBACxE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACnD,CAAC,CAAC;YAEF,mBAAmB;YACnB,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,GAAG,CAAC,eAAsB,CAAC,CAAC;YAClD,MAAM,CAAC,EAAE,CAAC,mBAAmB,EAAE,GAAG,CAAC,uBAA8B,CAAC,CAAC;YACnE,MAAM,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,CAAC,qBAA4B,CAAC,CAAC;YAE/D,kBAAkB;YAClB,MAAM,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAElE,gBAAgB;YAChB,IAAI,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC3C,MAAM,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,aAAa,EAAE;oBACxD,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,YAAY;iBAClC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;gBACnE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACnD,CAAC;YAED,yBAAyB;YACzB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;gBAC7B,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE;aAC9C,CAAC,CACH,CAAC;YAEF,qBAAqB;YACrB,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CACzB,KAAY,EACZ,OAAe,EACf,SAA8B,EAC9B,QAAgB;QAEhB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,GAAG,EAAE,aAAa,EAAE,CAAC;YACvB,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAClC,CAAC;QAED,qBAAqB;QACrB,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5C,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,GAAG,CAAC,eAAe;oBAAE,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,eAAsB,CAAC,CAAC;gBAC5E,IAAI,GAAG,CAAC,uBAAuB;oBAAE,MAAM,CAAC,GAAG,CAAC,mBAAmB,EAAE,GAAG,CAAC,uBAA8B,CAAC,CAAC;gBACrG,IAAI,GAAG,CAAC,qBAAqB;oBAAE,MAAM,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,CAAC,qBAA4B,CAAC,CAAC;YACjG,CAAC;QACH,CAAC;QAED,aAAa;QACb,GAAG,CAAC,MAAM,GAAG,WAAW,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC;QACvB,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,eAAe,GAAG,GAAG,CAAC,WAAW,GAAG,CAAC,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC;QAC7E,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAExD,qBAAqB;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,aAAa,EAAE,CAAC;YACtB,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEtC,oCAAoC;YACpC,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;gBAC5B,YAAY,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACxC,CAAC;YACD,KAAK,CAAC,iBAAiB,GAAG,UAAU,CAAC,GAAG,EAAE;gBACxC,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,oBAAoB,EAAE;oBACpC,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,qCAAqC,EAAE;iBACnE,CAAC,CACH,CAAC;YACJ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;QAED,mBAAmB;QACnB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/B,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;YAC7B,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE;SAC9B,CAAC,CACH,CAAC;QAEF,uBAAuB;QACvB,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QAE1E,mBAAmB;QACnB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAErC,qBAAqB;QACrB,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,KAAY,EAAE,OAAe,EAAE,KAAY;QAClE,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,GAAG,EAAE,aAAa,EAAE,CAAC;YACvB,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAClC,CAAC;QAED,qBAAqB;QACrB,IAAI,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,GAAG,CAAC,eAAe;oBAAE,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,eAAsB,CAAC,CAAC;gBAC5E,IAAI,GAAG,CAAC,uBAAuB;oBAAE,MAAM,CAAC,GAAG,CAAC,mBAAmB,EAAE,GAAG,CAAC,uBAA8B,CAAC,CAAC;gBACrG,IAAI,GAAG,CAAC,qBAAqB;oBAAE,MAAM,CAAC,GAAG,CAAC,iBAAiB,EAAE,GAAG,CAAC,qBAA4B,CAAC,CAAC;YACjG,CAAC;QACH,CAAC;QAED,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC;QACtB,GAAG,CAAC,SAAS,GAAG,KAAK,CAAC;QACtB,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,qBAAqB;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,CAAC;QAED,mBAAmB;QACnB,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;gBAC7B,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE;aAC5C,CAAC,CACH,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,YAAY,EAAE;YAC5B,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;SACzC,CAAC,CACH,CAAC;QAEF,mBAAmB;QACnB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAErC,qBAAqB;QACrB,YAAY,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,KAAY;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG;YAAE,OAAO;QAEjB,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,GAAG,EAAE,OAAO,IAAI,SAAS,CAAC;QAE1C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,+BAA+B,IAAI,CAAC,OAAO,CAAC,qBAAqB,IAAI,CAAC,CAAC;QAC/F,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,4EAA4E;IAC5E,gCAAgC;IAChC,4EAA4E;IAEpE,uBAAuB,CAAC,QAAgB,EAAE,eAAuB;QACvE,IAAI,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG;gBACR,QAAQ;gBACR,kBAAkB,EAAE,CAAC;gBACrB,oBAAoB,EAAE,CAAC;gBACvB,sBAAsB,EAAE,CAAC;aAC1B,CAAC;YACF,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC7B,OAAO,CAAC,oBAAoB,IAAI,eAAe,CAAC;QAChD,OAAO,CAAC,iBAAiB,GAAG,eAAe,CAAC;QAC5C,OAAO,CAAC,sBAAsB,GAAG,OAAO,CAAC,oBAAoB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAC7F,CAAC;IAEO,wBAAwB,CAAC,SAAmB;QAClD,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAE/C,8BAA8B;YAC9B,IAAI,CAAC,QAAQ;gBAAE,OAAO,CAAC,CAAC;YACxB,IAAI,CAAC,QAAQ;gBAAE,OAAO,CAAC,CAAC,CAAC;YAEzB,OAAO,QAAQ,CAAC,sBAAsB,GAAG,QAAQ,CAAC,sBAAsB,CAAC;QAC3E,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/pool/WorkflowPool.d.ts

@@ -4,10 +4,26 @@ import type { QueueAdapter, QueueStats } from "./queue/QueueAdapter.js";
 import type { FailoverStrategy } from "./failover/Strategy.js";
 import type { JobRecord, WorkflowInput, WorkflowJobOptions, JobId } from "./types/job.js";
 import type { WorkflowPoolEventMap } from "./types/events.js";
+import type { WorkflowAffinity } from "./types/affinity.js";
 /**
  * Configuration options for WorkflowPool.
  */
 export interface WorkflowPoolOpts {
+    /**
+     * An array of workflow affinity rules to establish a default mapping
+     * between workflows and specific clients.
+     *
+     * @example
+     * ```ts
+     * const pool = new WorkflowPool(clients, {
+     *   workflowAffinities: [
+     *     { workflowHash: "hash1", preferredClientIds: ["client-a"] },
+     *     { workflowHash: "hash2", excludeClientIds: ["client-b"] },
+     *   ]
+     * });
+     * ```
+     */
+    workflowAffinities?: WorkflowAffinity[];
     /**
      * Queue adapter for managing job queue operations.
      *
@@ -153,11 +169,18 @@ export declare class WorkflowPool extends TypedEventTarget<WorkflowPoolEventMap>
     private opts;
     private jobStore;
     private jobFailureAnalysis;
+    private affinities;
     private initPromise;
     private processing;
+    private processQueued;
     private activeJobs;
+    private readonly queueDebug;
+    private debugLog;
     constructor(clients: ComfyApi[], opts?: WorkflowPoolOpts);
     ready(): Promise<void>;
+    setAffinity(affinity: WorkflowAffinity): void;
+    removeAffinity(workflowHash: string): boolean;
+    getAffinities(): WorkflowAffinity[];
     enqueue(workflowInput: WorkflowInput, options?: WorkflowJobOptions): Promise<JobId>;
     getJob(jobId: string): JobRecord | undefined;
     cancel(jobId: string): Promise<boolean>;