comfyui-mcp 0.1.0

package/dist/services/mermaid-parser.js

@@ -0,0 +1,464 @@
+import { ValidationError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+/**
+ * Parse a mermaid flowchart string into structured nodes and edges.
+ */
+export function parseMermaid(mermaidText) {
+    // Strip code fence if present
+    let text = mermaidText.trim();
+    if (text.startsWith("```")) {
+        text = text.replace(/^```(?:mermaid)?\s*\n?/, "").replace(/\n?```\s*$/, "");
+    }
+    const lines = text.split("\n").map((l) => l.trim());
+    const nodes = new Map();
+    const edges = [];
+    let direction = "LR";
+    for (const line of lines) {
+        // Parse direction from flowchart declaration
+        const dirMatch = line.match(/^flowchart\s+(LR|TB|TD|RL)$/);
+        if (dirMatch) {
+            direction = dirMatch[1] === "TD" ? "TB" : dirMatch[1];
+            continue;
+        }
+        // Skip subgraph/end/style/linkStyle lines
+        if (line.startsWith("subgraph ") ||
+            line === "end" ||
+            line.startsWith("style ") ||
+            line.startsWith("linkStyle ") ||
+            line === "") {
+            continue;
+        }
+        // Try to parse as edge: sourceId -->|TYPE| targetId  OR  sourceId --> targetId
+        const edgeMatch = line.match(/^(\S+)\s+--+>(?:\|([^|]*)\|)?\s*(\S+)$/);
+        if (edgeMatch) {
+            edges.push({
+                sourceId: edgeMatch[1],
+                targetId: edgeMatch[3],
+                dataType: edgeMatch[2] || undefined,
+            });
+            continue;
+        }
+        // Try to parse as node definition with various shapes
+        const nodeResult = parseNodeLine(line);
+        if (nodeResult) {
+            nodes.set(nodeResult.id, nodeResult);
+        }
+    }
+    if (nodes.size === 0) {
+        throw new ValidationError("Could not parse any nodes from mermaid text. Expected format: flowchart LR/TB with node definitions.");
+    }
+    return { nodes, edges, direction };
+}
+/**
+ * Parse a single node definition line.
+ * Handles all shape variants:
+ *   id["label"]        — rectangle (loading, utility)
+ *   id(["label"])       — stadium (conditioning)
+ *   id{{"label"}}       — hexagon (sampling)
+ *   id("label")         — rounded (image)
+ *   id((("label")))     — triple circle (output)
+ *   id(("label"))       — double circle
+ */
+function parseNodeLine(line) {
+    // Match: id followed by shape-wrapped label
+    // The id is one or more non-whitespace chars at the start (before the shape opener)
+    const patterns = [
+        // Triple circle: id((("label")))
+        /^(\S+?)\(\(\("(.+?)"\)\)\)$/,
+        // Double circle: id(("label"))
+        /^(\S+?)\(\("(.+?)"\)\)$/,
+        // Hexagon: id{{"label"}}
+        /^(\S+?)\{\{"(.+?)"\}\}$/,
+        // Stadium: id(["label"])
+        /^(\S+?)\(\["(.+?)"\]\)$/,
+        // Rounded: id("label")
+        /^(\S+?)\("(.+?)"\)$/,
+        // Rectangle: id["label"]
+        /^(\S+?)\["(.+?)"\]$/,
+    ];
+    for (const pattern of patterns) {
+        const match = line.match(pattern);
+        if (match) {
+            const id = match[1];
+            const rawLabel = match[2];
+            return parseLabel(id, rawLabel);
+        }
+    }
+    return null;
+}
+/**
+ * Parse the inner label text.
+ * Format: "ClassName\nkey:value key2:value2"
+ * or: "Title\nkey:value key2:value2"
+ */
+function parseLabel(id, rawLabel) {
+    // Unescape mermaid HTML entities
+    const label = rawLabel.replace(/#quot;/g, '"');
+    // Split on literal \n (the escaped newline in mermaid syntax)
+    const parts = label.split("\\n");
+    const title = parts[0].trim();
+    const values = {};
+    if (parts.length > 1) {
+        // The rest is space-separated key:value pairs
+        const valuesStr = parts.slice(1).join(" ");
+        // Parse key:value pairs. Values can contain colons (filenames like model.safetensors),
+        // so we match key up to first colon, then value up to next key or end.
+        // Pattern: word chars + colon + value (up to next space-followed-by-word-colon, or end)
+        const kvRegex = /(\w+):(.+?)(?=\s+\w+:|$)/g;
+        let kvMatch;
+        while ((kvMatch = kvRegex.exec(valuesStr)) !== null) {
+            values[kvMatch[1]] = kvMatch[2].trim();
+        }
+    }
+    return { id, label: title, values };
+}
+/**
+ * Resolve a parsed mermaid graph into a valid ComfyUI workflow JSON
+ * using node definitions from /object_info.
+ */
+export function resolveWorkflow(parsed, objectInfo) {
+    const workflow = {};
+    const warnings = [];
+    // Build lookup: display_name → class_type
+    const displayNameMap = new Map();
+    for (const [className, def] of Object.entries(objectInfo)) {
+        if (def.display_name) {
+            displayNameMap.set(def.display_name.toLowerCase(), className);
+        }
+        // Also map the class name itself (case-insensitive)
+        displayNameMap.set(className.toLowerCase(), className);
+    }
+    // Phase 1: Resolve each node's class_type
+    const nodeClassTypes = new Map();
+    for (const [id, node] of parsed.nodes) {
+        const resolved = resolveClassType(node.label, objectInfo, displayNameMap);
+        if (resolved) {
+            nodeClassTypes.set(id, resolved);
+        }
+        else {
+            // Try inferring from connection data type signature
+            const inferred = inferClassTypeFromConnections(id, parsed.edges, objectInfo);
+            if (inferred) {
+                nodeClassTypes.set(id, inferred);
+                logger.info(`Node ${id}: Inferred class_type "${inferred}" from connection signature for label "${node.label}"`);
+            }
+            else {
+                warnings.push(`Node ${id}: Could not resolve "${node.label}" to a known ComfyUI node type. Using label as class_type.`);
+                nodeClassTypes.set(id, node.label);
+            }
+        }
+    }
+    // Phase 2: Create workflow nodes with empty inputs
+    for (const [id, node] of parsed.nodes) {
+        const classType = nodeClassTypes.get(id);
+        workflow[id] = {
+            class_type: classType,
+            inputs: {},
+        };
+        // If the label differs from class_type, preserve it as _meta.title
+        if (node.label !== classType) {
+            workflow[id]._meta = { title: node.label };
+        }
+    }
+    // Phase 3: Wire connections using edge data types + object_info
+    for (const edge of parsed.edges) {
+        const sourceClassType = nodeClassTypes.get(edge.sourceId);
+        const targetClassType = nodeClassTypes.get(edge.targetId);
+        if (!sourceClassType || !targetClassType) {
+            warnings.push(`Edge ${edge.sourceId} → ${edge.targetId}: source or target node not found, skipping.`);
+            continue;
+        }
+        const sourceDef = objectInfo[sourceClassType];
+        const targetDef = objectInfo[targetClassType];
+        if (!sourceDef || !targetDef) {
+            warnings.push(`Edge ${edge.sourceId} → ${edge.targetId}: missing object_info for ${!sourceDef ? sourceClassType : targetClassType}. Using best guess.`);
+            // Best effort: put connection as first matching input
+            if (edge.dataType) {
+                const inputName = guessInputNameByType(edge.dataType, targetClassType);
+                if (inputName) {
+                    workflow[edge.targetId].inputs[inputName] = [edge.sourceId, 0];
+                }
+            }
+            continue;
+        }
+        // Find the output index on the source that produces this data type
+        let outputIndex = 0;
+        if (edge.dataType && sourceDef.output) {
+            const idx = sourceDef.output.indexOf(edge.dataType);
+            if (idx >= 0) {
+                outputIndex = idx;
+            }
+            else {
+                warnings.push(`Edge ${edge.sourceId} → ${edge.targetId}: data type "${edge.dataType}" not found in ${sourceClassType} outputs [${sourceDef.output.join(", ")}]. Using index 0.`);
+            }
+        }
+        // Find the input name on the target that accepts this data type.
+        // Use the source node's title as a hint for disambiguation
+        // (e.g., "Positive Prompt" → positive, "Negative Prompt" → negative).
+        const sourceTitle = parsed.nodes.get(edge.sourceId)?.label ?? "";
+        const inputName = findInputForType(targetDef, edge.dataType, workflow[edge.targetId].inputs, sourceTitle);
+        if (inputName) {
+            workflow[edge.targetId].inputs[inputName] = [edge.sourceId, outputIndex];
+        }
+        else {
+            warnings.push(`Edge ${edge.sourceId} → ${edge.targetId}: could not find input on ${targetClassType} that accepts "${edge.dataType}".`);
+        }
+    }
+    // Phase 4: Fill widget values from parsed labels
+    for (const [id, node] of parsed.nodes) {
+        const classType = nodeClassTypes.get(id);
+        const def = objectInfo[classType];
+        if (!def)
+            continue;
+        for (const [key, rawValue] of Object.entries(node.values)) {
+            // Don't overwrite connections
+            if (isConnectionValue(workflow[id].inputs[key]))
+                continue;
+            // Coerce value to the correct type based on objectInfo
+            const coerced = coerceValue(key, rawValue, def);
+            if (coerced !== undefined) {
+                workflow[id].inputs[key] = coerced;
+            }
+            else {
+                // Fall back to string
+                workflow[id].inputs[key] = rawValue;
+            }
+        }
+    }
+    // Phase 5: Fill remaining required inputs with defaults from objectInfo
+    for (const [id] of parsed.nodes) {
+        const classType = nodeClassTypes.get(id);
+        const def = objectInfo[classType];
+        if (!def)
+            continue;
+        fillDefaults(workflow[id], def, warnings, id);
+    }
+    return { workflow, warnings };
+}
+/**
+ * Infer class_type by matching a node's connection signature against all known node types.
+ * E.g., a node that receives CLIP and outputs CONDITIONING is very likely CLIPTextEncode.
+ */
+function inferClassTypeFromConnections(nodeId, edges, objectInfo) {
+    // Collect incoming and outgoing data types for this node
+    const incomingTypes = new Set();
+    const outgoingTypes = new Set();
+    for (const edge of edges) {
+        if (edge.targetId === nodeId && edge.dataType) {
+            incomingTypes.add(edge.dataType);
+        }
+        if (edge.sourceId === nodeId && edge.dataType) {
+            outgoingTypes.add(edge.dataType);
+        }
+    }
+    if (incomingTypes.size === 0 && outgoingTypes.size === 0)
+        return null;
+    // Score each node type by how well it matches the signature
+    let bestMatch = null;
+    let bestScore = 0;
+    for (const [className, def] of Object.entries(objectInfo)) {
+        let score = 0;
+        // Check outgoing types match outputs
+        if (outgoingTypes.size > 0) {
+            const outputSet = new Set(def.output);
+            for (const t of outgoingTypes) {
+                if (outputSet.has(t))
+                    score += 2;
+            }
+        }
+        // Check incoming types match inputs
+        if (incomingTypes.size > 0) {
+            const allInputs = {
+                ...def.input.required,
+                ...def.input.optional,
+            };
+            const inputTypes = new Set(Object.values(allInputs).map((spec) => typeof spec[0] === "string" ? spec[0] : ""));
+            for (const t of incomingTypes) {
+                if (inputTypes.has(t))
+                    score += 1;
+            }
+        }
+        if (score > bestScore) {
+            bestScore = score;
+            bestMatch = className;
+        }
+    }
+    // Require a minimum score to avoid false matches
+    return bestScore >= 2 ? bestMatch : null;
+}
+/**
+ * Resolve a label to a known class_type.
+ */
+function resolveClassType(label, objectInfo, displayNameMap) {
+    // Direct match against class names
+    if (objectInfo[label])
+        return label;
+    // Case-insensitive match
+    const lower = label.toLowerCase();
+    const fromDisplay = displayNameMap.get(lower);
+    if (fromDisplay)
+        return fromDisplay;
+    // Fuzzy: try removing spaces (e.g., "Checkpoint Loader Simple" → "CheckpointLoaderSimple")
+    const noSpaces = label.replace(/\s+/g, "");
+    if (objectInfo[noSpaces])
+        return noSpaces;
+    // Try display name without spaces
+    const fromDisplayNoSpaces = displayNameMap.get(noSpaces.toLowerCase());
+    if (fromDisplayNoSpaces)
+        return fromDisplayNoSpaces;
+    return null;
+}
+/**
+ * Find an input on the target node that accepts the given data type.
+ * Avoids assigning to an input that already has a connection.
+ * Uses sourceTitle as a hint when multiple inputs accept the same type
+ * (e.g., KSampler has both "positive" and "negative" CONDITIONING inputs).
+ */
+function findInputForType(targetDef, dataType, currentInputs, sourceTitle) {
+    if (!dataType)
+        return null;
+    const allInputs = {
+        ...targetDef.input.required,
+        ...targetDef.input.optional,
+    };
+    // Collect all matching, unwired inputs
+    const candidates = [];
+    for (const [inputName, spec] of Object.entries(allInputs)) {
+        if (isConnectionValue(currentInputs[inputName]))
+            continue;
+        const acceptedType = spec[0];
+        const matches = (typeof acceptedType === "string" && acceptedType === dataType) ||
+            (Array.isArray(acceptedType) && acceptedType.includes(dataType));
+        if (matches) {
+            candidates.push(inputName);
+        }
+    }
+    if (candidates.length === 0)
+        return null;
+    if (candidates.length === 1)
+        return candidates[0];
+    // Multiple candidates — use sourceTitle to pick the best match.
+    // E.g., title "Positive Prompt" should match input named "positive",
+    // and "Negative Prompt" should match "negative".
+    if (sourceTitle) {
+        const titleLower = sourceTitle.toLowerCase();
+        for (const inputName of candidates) {
+            if (titleLower.includes(inputName.toLowerCase())) {
+                return inputName;
+            }
+        }
+    }
+    // Fall back to first available
+    return candidates[0];
+}
+/**
+ * Fallback: guess input name from data type for common patterns.
+ */
+function guessInputNameByType(dataType, _targetClassType) {
+    const typeToInput = {
+        MODEL: "model",
+        CLIP: "clip",
+        VAE: "vae",
+        CONDITIONING: "positive", // ambiguous but common
+        LATENT: "latent_image",
+        IMAGE: "images",
+        MASK: "mask",
+        CONTROL_NET: "control_net",
+        UPSCALE_MODEL: "upscale_model",
+    };
+    return typeToInput[dataType] ?? null;
+}
+/**
+ * Coerce a string value to the expected type based on objectInfo.
+ */
+function coerceValue(key, rawValue, def) {
+    const spec = findInputSpec(key, def);
+    if (!spec)
+        return undefined;
+    const [typeOrEnum, options] = spec;
+    // Enum type (array of allowed string values)
+    if (Array.isArray(typeOrEnum)) {
+        // Return the raw value if it's in the enum, otherwise closest match
+        if (typeOrEnum.includes(rawValue))
+            return rawValue;
+        // Try case-insensitive
+        const found = typeOrEnum.find((v) => typeof v === "string" && v.toLowerCase() === rawValue.toLowerCase());
+        return found ?? rawValue;
+    }
+    switch (typeOrEnum) {
+        case "INT": {
+            const n = Number(rawValue);
+            return Number.isFinite(n) ? Math.round(n) : undefined;
+        }
+        case "FLOAT": {
+            const n = Number(rawValue);
+            return Number.isFinite(n) ? n : undefined;
+        }
+        case "BOOLEAN":
+            return rawValue === "true" || rawValue === "1";
+        case "STRING":
+            return rawValue;
+        default:
+            // Unknown type — might be a connection type like MODEL, CLIP
+            // Don't set scalar values for connection types
+            return undefined;
+    }
+}
+/**
+ * Find the input spec for a given key in a node definition.
+ */
+function findInputSpec(key, def) {
+    if (def.input.required?.[key])
+        return def.input.required[key];
+    if (def.input.optional?.[key])
+        return def.input.optional[key];
+    return null;
+}
+/**
+ * Fill required inputs that weren't set by connections or parsed values.
+ */
+function fillDefaults(node, def, warnings, nodeId) {
+    const required = def.input.required ?? {};
+    for (const [inputName, spec] of Object.entries(required)) {
+        // Skip if already set
+        if (node.inputs[inputName] !== undefined)
+            continue;
+        const [typeOrEnum, options] = spec;
+        // Connection types (MODEL, CLIP, etc.) — can't fill with defaults, skip
+        if (typeof typeOrEnum === "string" &&
+            typeOrEnum === typeOrEnum.toUpperCase() &&
+            typeOrEnum.length > 1 &&
+            !["INT", "FLOAT", "STRING", "BOOLEAN"].includes(typeOrEnum)) {
+            continue;
+        }
+        // Enum: use first option as default
+        if (Array.isArray(typeOrEnum) && typeOrEnum.length > 0) {
+            node.inputs[inputName] = typeOrEnum[0];
+            continue;
+        }
+        // Use explicit default if available
+        if (options && "default" in options) {
+            node.inputs[inputName] = options.default;
+            continue;
+        }
+        // No default available
+        logger.debug(`Node ${nodeId} (${node.class_type}): required input "${inputName}" has no value or default`);
+    }
+    // Also fill optional inputs that have defaults (for completeness)
+    const optional = def.input.optional ?? {};
+    for (const [inputName, spec] of Object.entries(optional)) {
+        if (node.inputs[inputName] !== undefined)
+            continue;
+        const [, options] = spec;
+        if (options && "default" in options) {
+            node.inputs[inputName] = options.default;
+        }
+    }
+}
+function isConnectionValue(value) {
+    return (Array.isArray(value) &&
+        value.length === 2 &&
+        typeof value[0] === "string" &&
+        typeof value[1] === "number");
+}
+//# sourceMappingURL=mermaid-parser.js.map

package/dist/services/mermaid-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mermaid-parser.js","sourceRoot":"","sources":["../../src/services/mermaid-parser.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AA6B5C;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,WAAmB;IAC9C,8BAA8B;IAC9B,IAAI,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC5C,MAAM,KAAK,GAAiB,EAAE,CAAC;IAC/B,IAAI,SAAS,GAAgB,IAAI,CAAC;IAElC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,6CAA6C;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAC3D,IAAI,QAAQ,EAAE,CAAC;YACb,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAE,QAAQ,CAAC,CAAC,CAAiB,CAAC;YACvE,SAAS;QACX,CAAC;QAED,0CAA0C;QAC1C,IACE,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;YAC5B,IAAI,KAAK,KAAK;YACd,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YACzB,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;YAC7B,IAAI,KAAK,EAAE,EACX,CAAC;YACD,SAAS;QACX,CAAC;QAED,+EAA+E;QAC/E,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,wCAAwC,CACzC,CAAC;QACF,IAAI,SAAS,EAAE,CAAC;YACd,KAAK,CAAC,IAAI,CAAC;gBACT,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;gBACtB,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;gBACtB,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS;aACpC,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,sDAAsD;QACtD,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,UAAU,EAAE,CAAC;YACf,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACrB,MAAM,IAAI,eAAe,CACvB,sGAAsG,CACvG,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACrC,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CAAC,IAAY;IACjC,4CAA4C;IAC5C,oFAAoF;IACpF,MAAM,QAAQ,GAAG;QACf,iCAAiC;QACjC,6BAA6B;QAC7B,+BAA+B;QAC/B,yBAAyB;QACzB,yBAAyB;QACzB,yBAAyB;QACzB,yBAAyB;QACzB,yBAAyB;QACzB,uBAAuB;QACvB,qBAAqB;QACrB,yBAAyB;QACzB,qBAAqB;KACtB,CAAC;IAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACpB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC1B,OAAO,UAAU,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;GAIG;AACH,SAAS,UAAU,CAAC,EAAU,EAAE,QAAgB;IAC9C,iCAAiC;IACjC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAE/C,8DAA8D;IAC9D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAE9B,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,8CAA8C;QAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3C,uFAAuF;QACvF,uEAAuE;QACvE,wFAAwF;QACxF,MAAM,OAAO,GAAG,2BAA2B,CAAC;QAC5C,IAAI,OAA+B,CAAC;QACpC,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AACtC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAC7B,MAAqB,EACrB,UAAsB;IAEtB,MAAM,QAAQ,GAAiB,EAAE,CAAC;IAClC,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,0CAA0C;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,KAAK,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC1D,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;YACrB,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,WAAW,EAAE,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;QACD,oDAAoD;QACpD,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAED,0CAA0C;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;QAC1E,IAAI,QAAQ,EAAE,CAAC;YACb,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,oDAAoD;YACpD,MAAM,QAAQ,GAAG,6BAA6B,CAC5C,EAAE,EACF,MAAM,CAAC,KAAK,EACZ,UAAU,CACX,CAAC;YACF,IAAI,QAAQ,EAAE,CAAC;gBACb,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;gBACjC,MAAM,CAAC,IAAI,CACT,QAAQ,EAAE,0BAA0B,QAAQ,0CAA0C,IAAI,CAAC,KAAK,GAAG,CACpG,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CACX,QAAQ,EAAE,wBAAwB,IAAI,CAAC,KAAK,4DAA4D,CACzG,CAAC;gBACF,cAAc,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;IACH,CAAC;IAED,mDAAmD;IACnD,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAC1C,QAAQ,CAAC,EAAE,CAAC,GAAG;YACb,UAAU,EAAE,SAAS;YACrB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,mEAAmE;QACnE,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,gEAAgE;IAChE,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAChC,MAAM,eAAe,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1D,MAAM,eAAe,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE1D,IAAI,CAAC,eAAe,IAAI,CAAC,eAAe,EAAE,CAAC;YACzC,QAAQ,CAAC,IAAI,CACX,QAAQ,IAAI,CAAC,QAAQ,MAAM,IAAI,CAAC,QAAQ,8CAA8C,CACvF,CAAC;YACF,SAAS;QACX,CAAC;QAED,MAAM,SAAS,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC;QAE9C,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,EAAE,CAAC;YAC7B,QAAQ,CAAC,IAAI,CACX,QAAQ,IAAI,CAAC,QAAQ,MAAM,IAAI,CAAC,QAAQ,6BAA6B,CAAC,SAAS,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,qBAAqB,CACzI,CAAC;YACF,sDAAsD;YACtD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,SAAS,GAAG,oBAAoB,CAAC,IAAI,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;gBACvE,IAAI,SAAS,EAAE,CAAC;oBACd,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,mEAAmE;QACnE,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,IAAI,CAAC,QAAQ,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpD,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC;gBACb,WAAW,GAAG,GAAG,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CACX,QAAQ,IAAI,CAAC,QAAQ,MAAM,IAAI,CAAC,QAAQ,gBAAgB,IAAI,CAAC,QAAQ,kBAAkB,eAAe,aAAa,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAClK,CAAC;YACJ,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,2DAA2D;QAC3D,sEAAsE;QACtE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;QACjE,MAAM,SAAS,GAAG,gBAAgB,CAChC,SAAS,EACT,IAAI,CAAC,QAAQ,EACb,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,EAC9B,WAAW,CACZ,CAAC;QAEF,IAAI,SAAS,EAAE,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC3E,CAAC;aAAM,CAAC;YACN,QAAQ,CAAC,IAAI,CACX,QAAQ,IAAI,CAAC,QAAQ,MAAM,IAAI,CAAC,QAAQ,6BAA6B,eAAe,kBAAkB,IAAI,CAAC,QAAQ,IAAI,CACxH,CAAC;QACJ,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAC1C,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,GAAG;YAAE,SAAS;QAEnB,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,8BAA8B;YAC9B,IAAI,iBAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAAE,SAAS;YAE1D,uDAAuD;YACvD,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;YAChD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,sBAAsB;gBACtB,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IAED,wEAAwE;IACxE,KAAK,MAAM,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAChC,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAC1C,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,GAAG;YAAE,SAAS;QAEnB,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;AAChC,CAAC;AAED;;;GAGG;AACH,SAAS,6BAA6B,CACpC,MAAc,EACd,KAAmB,EACnB,UAAsB;IAEtB,yDAAyD;IACzD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IACxC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAExC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9C,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9C,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEtE,4DAA4D;IAC5D,IAAI,SAAS,GAAkB,IAAI,CAAC;IACpC,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC1D,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,qCAAqC;QACrC,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtC,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC9B,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;oBAAE,KAAK,IAAI,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG;gBAChB,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ;gBACrB,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ;aACtB,CAAC;YACF,MAAM,UAAU,GAAG,IAAI,GAAG,CACxB,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACpC,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAC3C,CACF,CAAC;YACF,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC9B,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;oBAAE,KAAK,IAAI,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;YACtB,SAAS,GAAG,KAAK,CAAC;YAClB,SAAS,GAAG,SAAS,CAAC;QACxB,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,OAAO,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;AAC3C,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CACvB,KAAa,EACb,UAAsB,EACtB,cAAmC;IAEnC,mCAAmC;IACnC,IAAI,UAAU,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEpC,yBAAyB;IACzB,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAClC,MAAM,WAAW,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,WAAW;QAAE,OAAO,WAAW,CAAC;IAEpC,2FAA2F;IAC3F,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC3C,IAAI,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAE1C,kCAAkC;IAClC,MAAM,mBAAmB,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;IACvE,IAAI,mBAAmB;QAAE,OAAO,mBAAmB,CAAC;IAEpD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;GAKG;AACH,SAAS,gBAAgB,CACvB,SAAyB,EACzB,QAA4B,EAC5B,aAAsC,EACtC,WAAoB;IAEpB,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE3B,MAAM,SAAS,GAAG;QAChB,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ;QAC3B,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ;KAC5B,CAAC;IAEF,uCAAuC;IACvC,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,KAAK,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1D,IAAI,iBAAiB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAAE,SAAS;QAE1D,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,OAAO,GACX,CAAC,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,QAAQ,CAAC;YAC/D,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnE,IAAI,OAAO,EAAE,CAAC;YACZ,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;IAElD,gEAAgE;IAChE,qEAAqE;IACrE,iDAAiD;IACjD,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,WAAW,CAAC,WAAW,EAAE,CAAC;QAC7C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACjD,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,+BAA+B;IAC/B,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,SAAS,oBAAoB,CAC3B,QAAgB,EAChB,gBAAwB;IAExB,MAAM,WAAW,GAA2B;QAC1C,KAAK,EAAE,OAAO;QACd,IAAI,EAAE,MAAM;QACZ,GAAG,EAAE,KAAK;QACV,YAAY,EAAE,UAAU,EAAE,uBAAuB;QACjD,MAAM,EAAE,cAAc;QACtB,KAAK,EAAE,QAAQ;QACf,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,aAAa;QAC1B,aAAa,EAAE,eAAe;KAC/B,CAAC;IACF,OAAO,WAAW,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;AACvC,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAClB,GAAW,EACX,QAAgB,EAChB,GAAmB;IAEnB,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACrC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAE5B,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAEnC,6CAA6C;IAC7C,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9B,oEAAoE;QACpE,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;QACnD,uBAAuB;QACvB,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAC3B,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,QAAQ,CAAC,WAAW,EAAE,CAC3E,CAAC;QACF,OAAO,KAAK,IAAI,QAAQ,CAAC;IAC3B,CAAC;IAED,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,KAAK,CAAC,CAAC,CAAC;YACX,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC3B,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACxD,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC3B,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5C,CAAC;QACD,KAAK,SAAS;YACZ,OAAO,QAAQ,KAAK,MAAM,IAAI,QAAQ,KAAK,GAAG,CAAC;QACjD,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB;YACE,6DAA6D;YAC7D,+CAA+C;YAC/C,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CACpB,GAAW,EACX,GAAmB;IAEnB,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC9D,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC;QAAE,OAAO,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC9D,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CACnB,IAAkB,EAClB,GAAmB,EACnB,QAAkB,EAClB,MAAc;IAEd,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;IAE1C,KAAK,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzD,sBAAsB;QACtB,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,SAAS;YAAE,SAAS;QAEnD,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;QAEnC,wEAAwE;QACxE,IACE,OAAO,UAAU,KAAK,QAAQ;YAC9B,UAAU,KAAK,UAAU,CAAC,WAAW,EAAE;YACvC,UAAU,CAAC,MAAM,GAAG,CAAC;YACrB,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,EAC3D,CAAC;YACD,SAAS;QACX,CAAC;QAED,oCAAoC;QACpC,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YACvC,SAAS;QACX,CAAC;QAED,oCAAoC;QACpC,IAAI,OAAO,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;YACzC,SAAS;QACX,CAAC;QAED,uBAAuB;QACvB,MAAM,CAAC,KAAK,CACV,QAAQ,MAAM,KAAK,IAAI,CAAC,UAAU,sBAAsB,SAAS,2BAA2B,CAC7F,CAAC;IACJ,CAAC;IAED,kEAAkE;IAClE,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzD,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,SAAS;YAAE,SAAS;QACnD,MAAM,CAAC,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;QACzB,IAAI,OAAO,IAAI,SAAS,IAAI,OAAO,EAAE,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC3C,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,CACL,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACpB,KAAK,CAAC,MAAM,KAAK,CAAC;QAClB,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ;QAC5B,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CAC7B,CAAC;AACJ,CAAC"}

package/dist/services/model-resolver.d.ts

@@ -0,0 +1,25 @@
+export declare const MODEL_SUBDIRS: readonly ["checkpoints", "loras", "vae", "upscale_models", "controlnet", "embeddings", "clip", "diffusers", "gligen", "hypernetworks", "photomaker", "style_models", "unet"];
+export type ModelType = (typeof MODEL_SUBDIRS)[number];
+export interface HFModelResult {
+    id: string;
+    modelId: string;
+    author: string;
+    tags: string[];
+    downloads: number;
+    likes: number;
+    lastModified: string;
+}
+export interface LocalModel {
+    name: string;
+    path: string;
+    size: number;
+    modified: string;
+    type: string;
+}
+export declare function searchHuggingFaceModels(query: string, options?: {
+    filter?: string;
+    limit?: number;
+}): Promise<HFModelResult[]>;
+export declare function listLocalModels(modelType?: string): Promise<LocalModel[]>;
+export declare function downloadModel(url: string, targetSubfolder: string, filename?: string): Promise<string>;
+//# sourceMappingURL=model-resolver.d.ts.map

package/dist/services/model-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.d.ts","sourceRoot":"","sources":["../../src/services/model-resolver.ts"],"names":[],"mappings":"AASA,eAAO,MAAM,aAAa,8KAchB,CAAC;AAEX,MAAM,MAAM,SAAS,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,CAAC,CAAC;AAEvD,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AASD,wBAAsB,uBAAuB,CAC3C,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAO,GAChD,OAAO,CAAC,aAAa,EAAE,CAAC,CAoC1B;AAED,wBAAsB,eAAe,CACnC,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,UAAU,EAAE,CAAC,CAqCvB;AAED,wBAAsB,aAAa,CACjC,GAAG,EAAE,MAAM,EACX,eAAe,EAAE,MAAM,EACvB,QAAQ,CAAC,EAAE,MAAM,GAChB,OAAO,CAAC,MAAM,CAAC,CAqCjB"}

package/dist/services/model-resolver.js

@@ -0,0 +1,123 @@
+import { readdir, stat, mkdir } from "node:fs/promises";
+import { createWriteStream } from "node:fs";
+import { pipeline } from "node:stream/promises";
+import { Readable } from "node:stream";
+import { join, basename } from "node:path";
+import { config } from "../config.js";
+import { ModelError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+export const MODEL_SUBDIRS = [
+    "checkpoints",
+    "loras",
+    "vae",
+    "upscale_models",
+    "controlnet",
+    "embeddings",
+    "clip",
+    "diffusers",
+    "gligen",
+    "hypernetworks",
+    "photomaker",
+    "style_models",
+    "unet",
+];
+function getModelsRoot() {
+    if (!config.comfyuiPath) {
+        throw new ModelError("COMFYUI_PATH is not configured. Set the COMFYUI_PATH environment variable.");
+    }
+    return join(config.comfyuiPath, "models");
+}
+export async function searchHuggingFaceModels(query, options = {}) {
+    const { filter, limit = 10 } = options;
+    const params = new URLSearchParams({
+        search: query,
+        limit: String(limit),
+    });
+    if (filter)
+        params.set("filter", filter);
+    const headers = {};
+    if (config.huggingfaceToken) {
+        headers["Authorization"] = `Bearer ${config.huggingfaceToken}`;
+    }
+    const url = `https://huggingface.co/api/models?${params}`;
+    logger.debug("HuggingFace API request", { url });
+    const res = await fetch(url, { headers });
+    if (!res.ok) {
+        const body = await res.text().catch(() => "");
+        throw new ModelError(`HuggingFace API ${res.status}: ${res.statusText}`, { url, status: res.status, body });
+    }
+    const data = (await res.json());
+    return data.map((m) => ({
+        id: String(m.id ?? m._id ?? ""),
+        modelId: String(m.modelId ?? m.id ?? ""),
+        author: String(m.author ?? ""),
+        tags: Array.isArray(m.tags) ? m.tags.map(String) : [],
+        downloads: Number(m.downloads ?? 0),
+        likes: Number(m.likes ?? 0),
+        lastModified: String(m.lastModified ?? ""),
+    }));
+}
+export async function listLocalModels(modelType) {
+    const modelsRoot = getModelsRoot();
+    const dirsToScan = modelType
+        ? [modelType]
+        : [...MODEL_SUBDIRS];
+    const results = [];
+    for (const dir of dirsToScan) {
+        const dirPath = join(modelsRoot, dir);
+        let entries;
+        try {
+            entries = await readdir(dirPath);
+        }
+        catch {
+            // Directory doesn't exist -- skip silently
+            continue;
+        }
+        for (const entry of entries) {
+            const filePath = join(dirPath, entry);
+            try {
+                const info = await stat(filePath);
+                if (!info.isFile())
+                    continue;
+                results.push({
+                    name: entry,
+                    path: filePath,
+                    size: info.size,
+                    modified: info.mtime.toISOString(),
+                    type: dir,
+                });
+            }
+            catch {
+                // Skip files we can't stat
+            }
+        }
+    }
+    return results;
+}
+export async function downloadModel(url, targetSubfolder, filename) {
+    const modelsRoot = getModelsRoot();
+    const targetDir = join(modelsRoot, targetSubfolder);
+    // Ensure target directory exists
+    await mkdir(targetDir, { recursive: true });
+    const resolvedFilename = filename ?? (basename(new URL(url).pathname) || "model.safetensors");
+    const targetPath = join(targetDir, resolvedFilename);
+    logger.info(`Downloading model to ${targetPath}`, { url });
+    const headers = {};
+    if (config.huggingfaceToken && url.includes("huggingface.co")) {
+        headers["Authorization"] = `Bearer ${config.huggingfaceToken}`;
+    }
+    const res = await fetch(url, { headers });
+    if (!res.ok) {
+        throw new ModelError(`Download failed: ${res.status} ${res.statusText}`, { url, status: res.status });
+    }
+    if (!res.body) {
+        throw new ModelError("Download response has no body", { url });
+    }
+    const nodeStream = Readable.fromWeb(res.body);
+    const fileStream = createWriteStream(targetPath);
+    await pipeline(nodeStream, fileStream);
+    const info = await stat(targetPath);
+    logger.info(`Download complete: ${resolvedFilename} (${(info.size / 1024 / 1024).toFixed(1)} MB)`);
+    return targetPath;
+}
+//# sourceMappingURL=model-resolver.js.map

package/dist/services/model-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.js","sourceRoot":"","sources":["../../src/services/model-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,aAAa;IACb,OAAO;IACP,KAAK;IACL,gBAAgB;IAChB,YAAY;IACZ,YAAY;IACZ,MAAM;IACN,WAAW;IACX,QAAQ;IACR,eAAe;IACf,YAAY;IACZ,cAAc;IACd,MAAM;CACE,CAAC;AAsBX,SAAS,aAAa;IACpB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,MAAM,IAAI,UAAU,CAAC,4EAA4E,CAAC,CAAC;IACrG,CAAC;IACD,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,KAAa,EACb,UAA+C,EAAE;IAEjD,MAAM,EAAE,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;QACjC,MAAM,EAAE,KAAK;QACb,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;KACrB,CAAC,CAAC;IACH,IAAI,MAAM;QAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAEzC,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAC5B,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,CAAC,gBAAgB,EAAE,CAAC;IACjE,CAAC;IAED,MAAM,GAAG,GAAG,qCAAqC,MAAM,EAAE,CAAC;IAC1D,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IAEjD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1C,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,IAAI,UAAU,CAClB,mBAAmB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,EAClD,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,CAClC,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAmC,CAAC;IAElE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACtB,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC;QAC/B,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;QACxC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,CAAC;QAC9B,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;QACrD,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC;QACnC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QAC3B,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,YAAY,IAAI,EAAE,CAAC;KAC3C,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,SAAkB;IAElB,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,MAAM,UAAU,GAAa,SAAS;QACpC,CAAC,CAAC,CAAC,SAAS,CAAC;QACb,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC;IAEvB,MAAM,OAAO,GAAiB,EAAE,CAAC;IAEjC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QACtC,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;QAAC,MAAM,CAAC;YACP,2CAA2C;YAC3C,SAAS;QACX,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACtC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAClC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;oBAAE,SAAS;gBAC7B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,KAAK;oBACX,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE;oBAClC,IAAI,EAAE,GAAG;iBACV,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,2BAA2B;YAC7B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAW,EACX,eAAuB,EACvB,QAAiB;IAEjB,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;IAEpD,iCAAiC;IACjC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE5C,MAAM,gBAAgB,GAAG,QAAQ,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,mBAAmB,CAAC,CAAC;IAC9F,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAErD,MAAM,CAAC,IAAI,CAAC,wBAAwB,UAAU,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IAE3D,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,MAAM,CAAC,gBAAgB,IAAI,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC9D,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,CAAC,gBAAgB,EAAE,CAAC;IACjE,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1C,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,UAAU,CAClB,oBAAoB,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,EAAE,EAClD,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAC5B,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QACd,MAAM,IAAI,UAAU,CAAC,+BAA+B,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,IAAgD,CAAC,CAAC;IAC1F,MAAM,UAAU,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACjD,MAAM,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAEvC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC;IACpC,MAAM,CAAC,IAAI,CAAC,sBAAsB,gBAAgB,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IAEnG,OAAO,UAAU,CAAC;AACpB,CAAC"}