comfyui-node 1.0.0

package/dist/pool.js

@@ -0,0 +1,467 @@
+import { delay } from "./tools.js";
+import { TypedEventTarget } from "./typed-event-target.js";
+/**
+ * Represents the mode for picking clients from a queue.
+ *
+ * - "PICK_ZERO": Picks the client which has zero queue remaining. This is the default mode. (For who using along with ComfyUI web interface)
+ * - "PICK_LOWEST": Picks the client which has the lowest queue remaining.
+ * - "PICK_ROUTINE": Picks the client in a round-robin manner.
+ */
+export var EQueueMode;
+(function (EQueueMode) {
+    /**
+     * Picks the client which has zero queue remaining. This is the default mode. (For who using along with ComfyUI web interface)
+     */
+    EQueueMode[EQueueMode["PICK_ZERO"] = 0] = "PICK_ZERO";
+    /**
+     * Picks the client which has the lowest queue remaining.
+     */
+    EQueueMode[EQueueMode["PICK_LOWEST"] = 1] = "PICK_LOWEST";
+    /**
+     * Picks the client in a round-robin manner.
+     */
+    EQueueMode[EQueueMode["PICK_ROUTINE"] = 2] = "PICK_ROUTINE";
+})(EQueueMode || (EQueueMode = {}));
+export class ComfyPool extends TypedEventTarget {
+    clients = [];
+    clientStates = [];
+    mode = EQueueMode.PICK_ZERO;
+    jobQueue = [];
+    routineIdx = 0;
+    maxQueueSize = 1000;
+    poolMonitoringInterval;
+    claimTimeoutMs = -1;
+    constructor(clients, 
+    /**
+     * The mode for picking clients from the queue. Defaults to "PICK_ZERO".
+     */
+    mode = EQueueMode.PICK_ZERO, opts) {
+        super();
+        this.mode = mode;
+        if (opts && typeof opts.maxQueueSize === "number") {
+            if (opts.maxQueueSize < 0) {
+                throw new Error("maxQueueSize cannot be negative");
+            }
+            this.maxQueueSize = opts.maxQueueSize;
+        }
+        if (typeof opts?.claimTimeoutMs === "number") {
+            this.claimTimeoutMs = opts.claimTimeoutMs;
+        }
+        this.poolMonitoringInterval = setInterval(() => {
+            this.processJobQueue().catch((err) => {
+                console.error("[ComfyPool] Error processing job queue:", err);
+            });
+        }, 5000);
+        this.initPool(clients).then(() => {
+            this.dispatchEvent(new CustomEvent("init"));
+        }).catch(reason => {
+            console.error("[ComfyPool] Error initializing pool:", reason);
+            this.dispatchEvent(new CustomEvent("error", { detail: reason }));
+        });
+    }
+    async initPool(clients) {
+        for (const client of clients) {
+            await this.addClient(client);
+        }
+        await this.processJobQueue();
+    }
+    // Use inherited on/off which return unsubscriber; provide chain helpers if desired.
+    chainOn(type, callback, options) {
+        super.on(type, callback, options);
+        return this;
+    }
+    chainOff(type, callback, options) {
+        super.off(type, callback, options);
+        return this;
+    }
+    /**
+     * Removes all event listeners from the pool.
+     */
+    removeAllListeners() {
+        // No internal registry now; rely on GC for one-off handlers or keep a manual registry later if needed.
+    }
+    /**
+     * Adds a client to the pool.
+     *
+     * @param client - The client to be added.
+     * @returns Promise<void>
+     */
+    async addClient(client) {
+        const index = this.clients.push(client) - 1;
+        this.clientStates.push({
+            id: client.id,
+            queueRemaining: 0,
+            locked: false,
+            online: false
+        });
+        await this.initializeClient(client, index);
+        this.dispatchEvent(new CustomEvent("added", { detail: { client, clientIdx: index } }));
+    }
+    /**
+     * Destroys the pool and all its clients.
+     * Ensures all connections, timers and event listeners are properly closed.
+     */
+    destroy() {
+        // Cancel any pending jobs
+        this.jobQueue = [];
+        // Destroy all clients properly and ensure they're cleaned up
+        this.clients.forEach((client, index) => {
+            try {
+                client.destroy();
+            }
+            catch (e) {
+                console.error(`[ComfyPool] Error destroying client ${client.id}:`, e);
+            }
+        });
+        // Clear arrays
+        this.clients = [];
+        this.clientStates = [];
+        // Remove all event listeners
+        this.removeAllListeners();
+        if (this.poolMonitoringInterval) {
+            clearInterval(this.poolMonitoringInterval);
+            this.poolMonitoringInterval = undefined;
+        }
+    }
+    /**
+     * Removes a client from the pool.
+     *
+     * @param client - The client to be removed.
+     * @returns void
+     */
+    removeClient(client) {
+        const index = this.clients.indexOf(client);
+        this.removeClientByIndex(index);
+    }
+    /**
+     * Removes a client from the pool by its index.
+     *
+     * @param index - The index of the client to remove.
+     * @returns void
+     * @fires removed - Fires a "removed" event with the removed client and its index as detail.
+     */
+    removeClientByIndex(index) {
+        if (index >= 0 && index < this.clients.length) {
+            const client = this.clients.splice(index, 1)[0];
+            client.destroy();
+            this.clientStates.splice(index, 1);
+            this.dispatchEvent(new CustomEvent("removed", { detail: { client, clientIdx: index } }));
+        }
+    }
+    /**
+     * Changes the mode of the queue.
+     *
+     * @param mode - The new mode to set for the queue.
+     * @returns void
+     */
+    changeMode(mode) {
+        this.mode = mode;
+        this.dispatchEvent(new CustomEvent("change_mode", { detail: { mode } }));
+    }
+    /**
+     * Picks a ComfyApi client from the pool based on the given index.
+     *
+     * @param idx - The index of the client to pick. Defaults to 0 if not provided.
+     * @returns The picked ComfyApi client.
+     */
+    pick(idx = 0) {
+        return this.clients[idx];
+    }
+    /**
+     * Retrieves a `ComfyApi` object from the pool based on the provided ID.
+     * @param id - The ID of the `ComfyApi` object to retrieve.
+     * @returns The `ComfyApi` object with the matching ID, or `undefined` if not found.
+     */
+    pickById(id) {
+        return this.clients.find((c) => c.id === id);
+    }
+    /**
+     * Executes a job using the provided client and optional client index.
+     *
+     * @template T The type of the result returned by the job.
+     * @param {Function} job The job to be executed.
+     * @param {number} [weight] The weight of the job.
+     * @param {Object} [clientFilter] An object containing client filtering options.
+     * @param {Object} [options] Additional options for job execution.
+     * @returns {Promise<T>} A promise that resolves with the result of the job.
+     */
+    run(job, weight, clientFilter, options) {
+        const enableFailover = options?.enableFailover !== false; // Default to true
+        const retryDelay = options?.retryDelay || 1000;
+        return new Promise(async (resolve, reject) => {
+            let excludedIds = clientFilter?.excludeIds ? [...clientFilter.excludeIds] : [];
+            let attempt = 0;
+            const onlineClients = this.clientStates.filter((c) => c.online);
+            const maxRetries = options?.maxRetries || onlineClients.length;
+            let lastError = null;
+            const tryExecute = async () => {
+                attempt++;
+                const fn = async (client, idx) => {
+                    this.dispatchEvent(new CustomEvent("executing", { detail: { client, clientIdx: idx } }));
+                    try {
+                        const result = await job(client, idx);
+                        this.dispatchEvent(new CustomEvent("executed", { detail: { client, clientIdx: idx } }));
+                        resolve(result);
+                    }
+                    catch (e) {
+                        lastError = e;
+                        console.error(`[ComfyPool] Job failed on client ${client.id} (attempt ${attempt}/${maxRetries}):`, e);
+                        // If failover is enabled and we have more attempts, exclude this client and retry
+                        if (enableFailover && attempt < maxRetries && onlineClients.length > excludedIds.length) {
+                            excludedIds.push(client.id);
+                            this.dispatchEvent(new CustomEvent("execution_error", {
+                                detail: { client, clientIdx: idx, error: e, willRetry: true, attempt, maxRetries }
+                            }));
+                            // Wait before retrying
+                            setTimeout(() => {
+                                tryExecute().catch(reject);
+                            }, retryDelay);
+                        }
+                        else {
+                            // No more retries or failover disabled, reject with the error
+                            this.dispatchEvent(new CustomEvent("execution_error", {
+                                detail: { client, clientIdx: idx, error: e, willRetry: false, attempt, maxRetries }
+                            }));
+                            reject(e);
+                        }
+                    }
+                };
+                try {
+                    await this.claim(fn, weight, {
+                        includeIds: clientFilter?.includeIds,
+                        excludeIds: excludedIds
+                    }, (err) => {
+                        // onError from claim (e.g. timeout acquiring client)
+                        lastError = err;
+                        reject(err);
+                    });
+                }
+                catch (claimError) {
+                    reject(lastError || claimError);
+                }
+            };
+            // Start the first attempt
+            tryExecute().catch(reject);
+        });
+    }
+    /**
+     * Executes a batch of asynchronous jobs concurrently and returns an array of results.
+     *
+     * @template T - The type of the result returned by each job.
+     * @param jobs - An array of functions that represent the asynchronous jobs to be executed.
+     * @param weight - An optional weight value to assign to each job.
+     * @param clientFilter - An optional object containing client filtering options.
+     * @returns A promise that resolves to an array of results, in the same order as the jobs array.
+     */
+    batch(jobs, weight, clientFilter) {
+        return Promise.all(jobs.map((task) => this.run(task, weight, clientFilter)));
+    }
+    async initializeClient(client, index) {
+        this.dispatchEvent(new CustomEvent("loading_client", {
+            detail: { client, clientIdx: index }
+        }));
+        const states = this.clientStates[index];
+        client.on("status", (ev) => {
+            if (states.online === false) {
+                this.dispatchEvent(new CustomEvent("connected", { detail: { client, clientIdx: index } }));
+            }
+            states.online = true;
+            if (ev.detail.status.exec_info && ev.detail.status.exec_info.queue_remaining !== states.queueRemaining) {
+                if (ev.detail.status.exec_info.queue_remaining > 0) {
+                    this.dispatchEvent(new CustomEvent("have_job", {
+                        detail: { client, remain: states.queueRemaining }
+                    }));
+                }
+                if (ev.detail.status.exec_info.queue_remaining === 0) {
+                    this.dispatchEvent(new CustomEvent("idle", { detail: { client } }));
+                }
+            }
+            states.queueRemaining = ev.detail.status.exec_info.queue_remaining;
+            if (this.mode !== EQueueMode.PICK_ZERO) {
+                states.locked = false;
+            }
+        });
+        client.on("terminal", (ev) => {
+            this.dispatchEvent(new CustomEvent("terminal", {
+                detail: {
+                    clientIdx: index,
+                    ...ev.detail
+                }
+            }));
+        });
+        client.on("disconnected", () => {
+            states.online = false;
+            states.locked = false;
+            this.dispatchEvent(new CustomEvent("disconnected", {
+                detail: { client, clientIdx: index }
+            }));
+        });
+        client.on("reconnected", () => {
+            states.online = true;
+            states.locked = false;
+            this.dispatchEvent(new CustomEvent("reconnected", {
+                detail: { client, clientIdx: index }
+            }));
+        });
+        client.on("execution_success", (ev) => {
+            states.locked = false;
+        });
+        client.on("execution_interrupted", (ev) => {
+            states.locked = false;
+            this.dispatchEvent(new CustomEvent("execution_interrupted", {
+                detail: {
+                    client,
+                    clientIdx: index
+                }
+            }));
+        });
+        client.on("execution_error", (ev) => {
+            states.locked = false;
+            this.dispatchEvent(new CustomEvent("execution_error", {
+                detail: {
+                    client,
+                    clientIdx: index,
+                    error: new Error(ev.detail.exception_type, { cause: ev.detail })
+                }
+            }));
+        });
+        client.on("queue_error", (ev) => {
+            states.locked = false;
+        });
+        client.on("auth_error", (ev) => {
+            this.dispatchEvent(new CustomEvent("auth_error", {
+                detail: { client, clientIdx: index, res: ev.detail }
+            }));
+        });
+        client.on("auth_success", (ev) => {
+            this.dispatchEvent(new CustomEvent("auth_success", {
+                detail: { client, clientIdx: index }
+            }));
+        });
+        client.on("connection_error", (ev) => {
+            this.dispatchEvent(new CustomEvent("connection_error", {
+                detail: { client, clientIdx: index, res: ev.detail }
+            }));
+        });
+        /**
+         * Wait for the client to be ready before start using it
+         * Note: init() now returns the client instance and sets isReady=true internally
+         */
+        await client.init();
+        // No need to call waitForReady() as init() already does that
+        this.bindClientSystemMonitor(client, index);
+        this.dispatchEvent(new CustomEvent("ready", { detail: { client, clientIdx: index } }));
+    }
+    async bindClientSystemMonitor(client, index) {
+        if (client.ext.monitor.isSupported) {
+            client.ext.monitor.on("system_monitor", (ev) => {
+                this.dispatchEvent(new CustomEvent("system_monitor", {
+                    detail: {
+                        client,
+                        data: ev.detail,
+                        clientIdx: index
+                    }
+                }));
+            });
+        }
+    }
+    pushJobByWeight(item) {
+        const idx = this.jobQueue.findIndex((job) => job.weight > item.weight);
+        if (idx === -1) {
+            return this.jobQueue.push(item);
+        }
+        else {
+            this.jobQueue.splice(idx, 0, item);
+            return idx;
+        }
+    }
+    async claim(fn, weight, clientFilter, onError) {
+        if (this.jobQueue.length >= this.maxQueueSize) {
+            throw new Error("Job queue limit reached");
+        }
+        const inputWeight = weight === undefined ? this.jobQueue.length : weight;
+        const idx = this.pushJobByWeight({
+            weight: inputWeight,
+            fn,
+            excludeClientIds: clientFilter?.excludeIds,
+            includeClientIds: clientFilter?.includeIds,
+            onError
+        });
+        this.dispatchEvent(new CustomEvent("add_job", {
+            detail: { jobIdx: idx, weight: inputWeight }
+        }));
+        await this.processJobQueue();
+    }
+    async getAvailableClient(includeIds, excludeIds, timeout = -1) {
+        let tries = 1;
+        const start = Date.now();
+        while (true) {
+            if (timeout > 0 && Date.now() - start > timeout) {
+                throw new Error("Timeout waiting for an available client");
+            }
+            if (tries < 100)
+                tries++;
+            let index = -1;
+            const acceptedClients = this.clientStates.filter((c) => {
+                if (!c.online)
+                    return false;
+                if (includeIds && includeIds.length > 0) {
+                    return includeIds.includes(c.id);
+                }
+                if (excludeIds && excludeIds.length > 0) {
+                    return !excludeIds.includes(c.id);
+                }
+                return true;
+            });
+            switch (this.mode) {
+                case EQueueMode.PICK_ZERO:
+                    index = acceptedClients.findIndex((c) => c.queueRemaining === 0 && !c.locked && c.id);
+                    break;
+                case EQueueMode.PICK_LOWEST:
+                    const queueSizes = acceptedClients.map((state) => state.online ? state.queueRemaining : Number.MAX_SAFE_INTEGER);
+                    index = queueSizes.indexOf(Math.min(...queueSizes));
+                    break;
+                case EQueueMode.PICK_ROUTINE:
+                    index = this.routineIdx++ % acceptedClients.length;
+                    this.routineIdx = this.routineIdx % acceptedClients.length;
+                    break;
+            }
+            if (index !== -1 && acceptedClients[index]) {
+                const trueIdx = this.clientStates.findIndex((c) => c.id === acceptedClients[index].id);
+                this.clientStates[trueIdx].locked = true;
+                return this.clients[trueIdx];
+            }
+            await delay(Math.min(tries * 10));
+        }
+    }
+    async processJobQueue() {
+        if (this.jobQueue.length === 0) {
+            return;
+        }
+        while (this.jobQueue.length > 0) {
+            const job = this.jobQueue.shift();
+            if (!job)
+                continue;
+            try {
+                const client = await this.getAvailableClient(job.includeClientIds, job.excludeClientIds, this.claimTimeoutMs);
+                const clientIdx = this.clients.indexOf(client);
+                await job.fn(client, clientIdx);
+            }
+            catch (error) {
+                console.error("[ComfyPool] Error processing job:", error);
+                try {
+                    job?.onError?.(error);
+                }
+                catch { }
+            }
+        }
+    }
+}
+// Test-only helpers (non-breaking; ignored at runtime usage)
+export const __TEST_ONLY__ = {
+    snapshotQueue(pool) {
+        const queue = pool.jobQueue;
+        return queue.map((j) => ({ weight: j.weight, include: j.includeClientIds, exclude: j.excludeClientIds }));
+    }
+};
+//# sourceMappingURL=pool.js.map

package/dist/pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pool.js","sourceRoot":"","sources":["../src/pool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAmB3D;;;;;;GAMG;AACH,MAAM,CAAN,IAAY,UAaX;AAbD,WAAY,UAAU;IACpB;;OAEG;IACH,qDAAW,CAAA;IACX;;OAEG;IACH,yDAAa,CAAA;IACb;;OAEG;IACH,2DAAc,CAAA;AAChB,CAAC,EAbW,UAAU,KAAV,UAAU,QAarB;AAED,MAAM,OAAO,SAAU,SAAQ,gBAAoC;IAC1D,OAAO,GAAe,EAAE,CAAC;IACxB,YAAY,GAKf,EAAE,CAAC;IAEA,IAAI,GAAe,UAAU,CAAC,SAAS,CAAC;IACxC,QAAQ,GAAmB,EAAE,CAAC;IAC9B,UAAU,GAAW,CAAC,CAAC;IACd,YAAY,GAAW,IAAI,CAAC;IACrC,sBAAsB,CAA6C;IACnE,cAAc,GAAW,CAAC,CAAC,CAAC;IAEpC,YACE,OAAmB;IACnB;;OAEG;IACH,OAAmB,UAAU,CAAC,SAAS,EACvC,IASC;QAED,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QAEjB,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;YAClD,IAAI,IAAI,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QACxC,CAAC;QACD,IAAI,OAAO,IAAI,EAAE,cAAc,KAAK,QAAQ,EAAE,CAAC;YAC7C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC5C,CAAC;QAED,IAAI,CAAC,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7C,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACnC,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,GAAG,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,CAAC;QAET,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAC/B,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;YAChB,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,MAAM,CAAC,CAAC;YAC9D,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAmB;QAChC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;QACD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;IAC/B,CAAC;IAED,oFAAoF;IAC7E,OAAO,CAAqC,IAAO,EAAE,QAAgD,EAAE,OAA2C;QACvJ,KAAK,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IACM,QAAQ,CAAqC,IAAO,EAAE,QAAgD,EAAE,OAAwC;QACrJ,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,QAAe,EAAE,OAAO,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACI,kBAAkB;QACvB,uGAAuG;IACzG,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,MAAgB;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;YACrB,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,cAAc,EAAE,CAAC;YACjB,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;IACzF,CAAC;IAED;;;OAGG;IACH,OAAO;QAEL,0BAA0B;QAC1B,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QAEnB,6DAA6D;QAC7D,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YACrC,IAAI,CAAC;gBACH,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,KAAK,CAAC,uCAAuC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YACxE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,eAAe;QACf,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,6BAA6B;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAE1B,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAChC,aAAa,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC3C,IAAI,CAAC,sBAAsB,GAAG,SAAS,CAAC;QAC1C,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,MAAgB;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;;;;;OAMG;IACH,mBAAmB,CAAC,KAAa;QAC/B,IAAI,KAAK,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,UAAU,CAAC,IAAgB;QACzB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3E,CAAC;IAED;;;;;OAKG;IACH,IAAI,CAAC,MAAc,CAAC;QAClB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,EAAU;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;;;;;OASG;IACH,GAAG,CACD,GAAyD,EACzD,MAAe,EACf,YASC,EACD,OAgBC;QAED,MAAM,cAAc,GAAG,OAAO,EAAE,cAAc,KAAK,KAAK,CAAC,CAAC,kBAAkB;QAC5E,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,CAAC;QAE/C,OAAO,IAAI,OAAO,CAAI,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,WAAW,GAAG,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/E,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YAChE,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,aAAa,CAAC,MAAM,CAAC;YACnE,IAAI,SAAS,GAAY,IAAI,CAAC;YAE1B,MAAM,UAAU,GAAG,KAAK,IAAmB,EAAE;gBAC3C,OAAO,EAAE,CAAC;gBAEV,MAAM,EAAE,GAAG,KAAK,EAAE,MAAgB,EAAE,GAAY,EAAE,EAAE;oBAClD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;oBACzF,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;wBACtC,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,UAAU,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;wBACxF,OAAO,CAAC,MAAM,CAAC,CAAC;oBAClB,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,SAAS,GAAG,CAAC,CAAC;wBACd,OAAO,CAAC,KAAK,CAAC,oCAAoC,MAAM,CAAC,EAAE,aAAa,OAAO,IAAI,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC;wBAEtG,kFAAkF;wBAClF,IAAI,cAAc,IAAI,OAAO,GAAG,UAAU,IAAI,aAAa,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;4BACxF,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;4BAC5B,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,iBAAiB,EAAE;gCACjC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE;6BACnF,CAAC,CACH,CAAC;4BAEF,uBAAuB;4BACvB,UAAU,CAAC,GAAG,EAAE;gCACd,UAAU,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;4BAC7B,CAAC,EAAE,UAAU,CAAC,CAAC;wBACjB,CAAC;6BAAM,CAAC;4BACN,8DAA8D;4BAC9D,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,iBAAiB,EAAE;gCACjC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE;6BACpF,CAAC,CACH,CAAC;4BACF,MAAM,CAAC,CAAC,CAAC,CAAC;wBACZ,CAAC;oBACH,CAAC;gBACH,CAAC,CAAC;gBAEF,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE;wBAC3B,UAAU,EAAE,YAAY,EAAE,UAAU;wBACpC,UAAU,EAAE,WAAW;qBACxB,EAAE,CAAC,GAAG,EAAE,EAAE;wBACT,qDAAqD;wBACrD,SAAS,GAAG,GAAG,CAAC;wBAChB,MAAM,CAAC,GAAG,CAAC,CAAC;oBACd,CAAC,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBACpB,MAAM,CAAC,SAAS,IAAI,UAAU,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC,CAAC;YAEF,0BAA0B;YAC1B,UAAU,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CACH,IAAiE,EACjE,MAAe,EACf,YASC;QAED,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/E,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAgB,EAAE,KAAa;QAC5D,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,gBAAgB,EAAE;YAChC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE;SACrC,CAAC,CACH,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE;YACzB,IAAI,MAAM,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;gBAC5B,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;YAC7F,CAAC;YACD,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;YACrB,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,KAAK,MAAM,CAAC,cAAc,EAAE,CAAC;gBACvG,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;oBACnD,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,UAAU,EAAE;wBAC1B,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,cAAc,EAAE;qBAClD,CAAC,CACH,CAAC;gBACJ,CAAC;gBACD,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,KAAK,CAAC,EAAE,CAAC;oBACrD,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YACD,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC;YACnE,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC;gBACvC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACxB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE;YAC3B,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,UAAU,EAAE;gBAC1B,MAAM,EAAE;oBACN,SAAS,EAAE,KAAK;oBAChB,GAAG,EAAE,CAAC,MAAM;iBACb;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;YAC7B,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACtB,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,cAAc,EAAE;gBAC9B,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE;aACrC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;YAC5B,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC;YACrB,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,aAAa,EAAE;gBAC7B,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE;aACrC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,EAAE,EAAE,EAAE;YACpC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QACxB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,uBAAuB,EAAE,CAAC,EAAE,EAAE,EAAE;YACxC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,uBAAuB,EAAE;gBACvC,MAAM,EAAE;oBACN,MAAM;oBACN,SAAS,EAAE,KAAK;iBACjB;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE;YAClC,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,iBAAiB,EAAE;gBACjC,MAAM,EAAE;oBACN,MAAM;oBACN,SAAS,EAAE,KAAK;oBAChB,KAAK,EAAE,IAAI,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC;iBACjE;aACF,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE;YAC9B,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC;QACxB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;YAC7B,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,YAAY,EAAE;gBAC5B,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,MAAM,EAAE;aACrD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,EAAE,EAAE,EAAE;YAC/B,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,cAAc,EAAE;gBAC9B,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE;aACrC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,EAAE,EAAE,EAAE;YACnC,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,kBAAkB,EAAE;gBAClC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,MAAM,EAAE;aACrD,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QACH;;;WAGG;QACH,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAEpB,6DAA6D;QAC7D,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC5C,IAAI,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;IACzF,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,MAAgB,EAAE,KAAa;QACnE,IAAI,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,gBAAgB,EAAE;oBAChC,MAAM,EAAE;wBACN,MAAM;wBACN,IAAI,EAAE,EAAE,CAAC,MAAM;wBACf,SAAS,EAAE,KAAK;qBACjB;iBACF,CAAC,CACH,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,IAAa;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QACvE,IAAI,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YACnC,OAAO,GAAG,CAAC;QACb,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,EAA2D,EAC3D,MAAe,EACf,YAGC,EACD,OAAgC;QAEhC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,WAAW,GAAG,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QACzE,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC;YAC/B,MAAM,EAAE,WAAW;YACnB,EAAE;YACF,gBAAgB,EAAE,YAAY,EAAE,UAAU;YAC1C,gBAAgB,EAAE,YAAY,EAAE,UAAU;YAC1C,OAAO;SACR,CAAC,CAAC;QACH,IAAI,CAAC,aAAa,CAChB,IAAI,WAAW,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE;SAC7C,CAAC,CACH,CAAC;QACF,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,UAAqB,EAAE,UAAqB,EAAE,OAAO,GAAG,CAAC,CAAC;QACzF,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,OAAO,IAAI,EAAE,CAAC;YACZ,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,OAAO,EAAE,CAAC;gBAChD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,KAAK,GAAG,GAAG;gBAAE,KAAK,EAAE,CAAC;YACzB,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC;YACf,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBACrD,IAAI,CAAC,CAAC,CAAC,MAAM;oBAAE,OAAO,KAAK,CAAC;gBAC5B,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACnC,CAAC;gBACD,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpC,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;YACH,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,UAAU,CAAC,SAAS;oBACvB,KAAK,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;oBACtF,MAAM;gBACR,KAAK,UAAU,CAAC,WAAW;oBACzB,MAAM,UAAU,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAC/C,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAC9D,CAAC;oBACF,KAAK,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC;oBACpD,MAAM;gBACR,KAAK,UAAU,CAAC,YAAY;oBAC1B,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,eAAe,CAAC,MAAM,CAAC;oBACnD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC;oBAC3D,MAAM;YACV,CAAC;YACD,IAAI,KAAK,KAAK,CAAC,CAAC,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;gBACvF,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC;gBACzC,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;YACD,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAE3B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YAClC,IAAI,CAAC,GAAG;gBAAE,SAAS;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC9G,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAClC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;gBAC1D,IAAI,CAAC;oBACH,GAAG,EAAE,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;gBAAC,MAAM,CAAC,CAAC,CAAC;YACb,CAAC;QACH,CAAC;IAEH,CAAC;CACF;AAED,6DAA6D;AAC7D,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,aAAa,CAAC,IAAe;QAC3B,MAAM,KAAK,GAAe,IAA2C,CAAC,QAAQ,CAAC;QAC/E,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,gBAAgB,EAAE,OAAO,EAAE,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;IAC5G,CAAC;CACF,CAAC"}

package/dist/prompt-builder.d.ts

@@ -0,0 +1,131 @@
+import { NodeData, OSType } from "./types/api.js";
+import { DeepKeys, Simplify } from "./types/tool.js";
+export declare class PromptBuilder<I extends string, O extends string, T extends NodeData> {
+    prompt: T;
+    mapInputKeys: Partial<Record<I, string | string[]>>;
+    mapOutputKeys: Partial<Record<O, string>>;
+    bypassNodes: (keyof T)[];
+    constructor(prompt: T, inputKeys: I[], outputKeys: O[]);
+    /**
+     * Creates a new instance of the PromptBuilder with the same prompt, input keys, and output keys.
+     *
+     * @returns A new instance of the PromptBuilder.
+     */
+    clone(): PromptBuilder<I, O, T>;
+    /**
+     * Marks a node to be bypassed at generation.
+     *
+     * @param node Node which will be bypassed.
+     */
+    bypass(node: keyof T): PromptBuilder<I, O, T>;
+    /**
+     * Marks multiple nodes to be bypassed at generation.
+     *
+     * @param nodes Array of nodes which will be bypassed.
+     */
+    bypass(nodes: (keyof T)[]): PromptBuilder<I, O, T>;
+    /**
+     * Unmarks a node from bypass at generation.
+     *
+     * @param node Node to reverse bypass on.
+     */
+    reinstate(node: keyof T): PromptBuilder<I, O, T>;
+    /**
+     * Unmarks a collection of nodes from bypass at generation.
+     *
+     * @param nodes Array of nodes to reverse bypass on.
+     */
+    reinstate(nodes: (keyof T)[]): PromptBuilder<I, O, T>;
+    /**
+     * Sets the input node for a given key. Can be map multiple keys to the same input.
+     *
+     * @param input - The input node to set.
+     * @param key - The key(s) to associate with the input node. Can be array of keys.
+     * @returns This builder instance.
+     */
+    setInputNode(input: I, key: DeepKeys<T> | Array<DeepKeys<T>>): PromptBuilder<I, O, T>;
+    /**
+     * Sets the raw input node for the given input and key. This will bypass the typing check. Use for dynamic nodes.
+     *
+     * @param input - The input node to be set.
+     * @param key - The key associated with the input node.
+     * @returns The current instance for method chaining.
+     */
+    setRawInputNode(input: I, key: string | string[]): PromptBuilder<I, O, T>;
+    /**
+     * Appends raw input node keys to the map of input keys. This will bypass the typing check. Use for dynamic nodes.
+     *
+     * @param input - The input node to which the keys will be appended.
+     * @param key - The key or array of keys to append to the input node.
+     * @returns A clone of the current instance with the updated input keys.
+     */
+    appendRawInputNode(input: I, key: string | string[]): PromptBuilder<I, O, T>;
+    /**
+     * Appends mapped key into the input node.
+     *
+     * @param input - The input node to append.
+     * @param key - The key(s) to associate with the input node. Can be array of keys.
+     * @returns The updated prompt builder.
+     */
+    appendInputNode(input: I, key: DeepKeys<T> | Array<DeepKeys<T>>): PromptBuilder<I, O, T>;
+    /**
+     * Sets the output node for a given key. This will bypass the typing check. Use for dynamic nodes.
+     *
+     * @param output - The output node to set.
+     * @param key - The key to associate with the output node.
+     * @returns This builder instance.
+     */
+    setRawOutputNode(output: O, key: string): PromptBuilder<I, O, T>;
+    /**
+     * Sets the output node for a given key.
+     *
+     * @param output - The output node to set.
+     * @param key - The key to associate with the output node.
+     * @returns This builder instance.
+     */
+    setOutputNode(output: O, key: DeepKeys<T>): PromptBuilder<I, O, T>;
+    /**
+     * Sets the value for a specific input key in the prompt builder.
+     *
+     * @template V - The type of the value being set.
+     * @param {I} key - The input key.
+     * @param {V} value - The value to set.
+     * @param {OSType} [encodeOs] - The OS type to encode the path.
+     * @returns A new prompt builder with the updated value.
+     * @throws {Error} - If the key is not found.
+     */
+    input<V = string | number | undefined>(key: I, value: V, encodeOs?: OSType): Simplify<PromptBuilder<I, O, T>>;
+    /**
+     * Sets the value for a any input key in the prompt builder.
+     *
+     * @template V - The type of the value being set.
+     * @param {string} key - The input key.
+     * @param {V} value - The value to set.
+     * @param {OSType} [encodeOs] - The OS type to encode the path.
+     * @returns A new prompt builder with the updated value.
+     * @throws {Error} - If the key is not found.
+     */
+    inputRaw<V = string | number | undefined>(key: string, value: V, encodeOs?: OSType): Simplify<PromptBuilder<I, O, T>>;
+    /**
+     * @deprecated Please call `input` directly instead
+     */
+    get caller(): this;
+    /**
+     * Gets the workflow object of the prompt builder.
+     */
+    get workflow(): Simplify<T>;
+    /**
+     * Validates that all declared output keys are mapped to existing nodes in the workflow.
+     * Throws an Error listing missing mappings.
+     */
+    validateOutputMappings(): this;
+    /**
+     * Detect simple circular references in node inputs (node referencing itself directly or via immediate tuple link).
+     * More advanced graph cycles can be added later.
+     */
+    validateNoImmediateCycles(): this;
+    /** Serialize builder state (excluding functions) */
+    toJSON(): any;
+    static fromJSON<I extends string, O extends string, T extends NodeData>(data: any): PromptBuilder<I, O, T>;
+}
+//# sourceMappingURL=prompt-builder.d.ts.map

package/dist/prompt-builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-builder.d.ts","sourceRoot":"","sources":["../src/prompt-builder.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAErD,qBAAa,aAAa,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,QAAQ;IAC/E,MAAM,EAAE,CAAC,CAAC;IACV,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC,CAAC,CAAM;IACzD,aAAa,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAM;IAC/C,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,CAAM;gBAElB,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE;IAWtD;;;;OAIG;IACH,KAAK,IAAI,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAY/B;;;;OAIG;IACH,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAE7C;;;;OAIG;IACH,MAAM,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAWlD;;;;OAIG;IACH,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAEhD;;;;OAIG;IACH,SAAS,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAcrD;;;;;;OAMG;IACH,YAAY,CAAC,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAI5D;;;;;;OAMG;IACH,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE;IAKhD;;;;;;OAMG;IACH,kBAAkB,CAAC,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE;IASnD;;;;;;OAMG;IACH,eAAe,CAAC,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAI/D;;;;;;OAMG;IACH,gBAAgB,CAAC,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,MAAM;IAKvC;;;;;;OAMG;IACH,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;IAIzC;;;;;;;;;OASG;IACH,KAAK,CAAC,CAAC,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,CAAC,EAAE,MAAM,GAkCzD,QAAQ,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAGjD;;;;;;;;;OASG;IACH,QAAQ,CAAC,CAAC,GAAG,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,CAAC,EAAE,MAAM,GA4BjE,QAAQ,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAGjD;;OAEG;IACH,IAAI,MAAM,SAET;IAED;;OAEG;IACH,IAAI,QAAQ,IACY,QAAQ,CAAC,CAAC,CAAC,CAClC;IAED;;;OAGG;IACH,sBAAsB,IAAI,IAAI;IAc9B;;;OAGG;IACH,yBAAyB,IAAI,IAAI;IAajC,oDAAoD;IACpD,MAAM,IAAI,GAAG;IASb,MAAM,CAAC,QAAQ,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,QAAQ,EAAE,IAAI,EAAE,GAAG,GAAG,aAAa,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC;CASzG"}