comfyui-mcp 0.1.0

package/dist/services/registry-client.d.ts

@@ -0,0 +1,28 @@
+export interface RegistrySearchResult {
+    id: string;
+    name: string;
+    description: string;
+    author: string;
+    repository: string;
+    latest_version: string;
+    total_install: number;
+    tags?: string[];
+}
+export interface NodePackDetails extends RegistrySearchResult {
+    versions: Array<{
+        version: string;
+        changelog?: string;
+    }>;
+    nodes: string[];
+    license?: string;
+    created_at: string;
+    updated_at: string;
+}
+export interface SearchNodesOptions {
+    page?: number;
+    limit?: number;
+    tags?: string[];
+}
+export declare function searchNodes(query: string, options?: SearchNodesOptions): Promise<RegistrySearchResult[]>;
+export declare function getNodePackDetails(id: string): Promise<NodePackDetails>;
+//# sourceMappingURL=registry-client.d.ts.map

package/dist/services/registry-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-client.d.ts","sourceRoot":"","sources":["../../src/services/registry-client.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,eAAgB,SAAQ,oBAAoB;IAC3D,QAAQ,EAAE,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACzD,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAiBD,wBAAsB,WAAW,CAC/B,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,kBAAuB,GAC/B,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAgBjC;AAED,wBAAsB,kBAAkB,CACtC,EAAE,EAAE,MAAM,GACT,OAAO,CAAC,eAAe,CAAC,CAI1B"}

package/dist/services/registry-client.js

@@ -0,0 +1,32 @@
+import { RegistryError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+const REGISTRY_BASE = "https://api.comfy.org";
+async function registryFetch(path) {
+    const url = `${REGISTRY_BASE}${path}`;
+    logger.debug("Registry API request", { url });
+    const res = await fetch(url);
+    if (!res.ok) {
+        const body = await res.text().catch(() => "");
+        throw new RegistryError(`Registry API ${res.status}: ${res.statusText}`, { url, status: res.status, body });
+    }
+    return res.json();
+}
+export async function searchNodes(query, options = {}) {
+    const { page = 1, limit = 10 } = options;
+    const params = new URLSearchParams({
+        search: query,
+        page: String(page),
+        limit: String(limit),
+    });
+    const data = await registryFetch(`/nodes?${params}`);
+    // The API may wrap results in a `nodes` array or return them directly
+    const results = Array.isArray(data) ? data : (data.nodes ?? []);
+    logger.info(`Registry search for "${query}" returned ${results.length} results`);
+    return results;
+}
+export async function getNodePackDetails(id) {
+    const data = await registryFetch(`/nodes/${encodeURIComponent(id)}`);
+    logger.info(`Fetched details for node pack "${id}"`);
+    return data;
+}
+//# sourceMappingURL=registry-client.js.map

package/dist/services/registry-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry-client.js","sourceRoot":"","sources":["../../src/services/registry-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,aAAa,GAAG,uBAAuB,CAAC;AA2B9C,KAAK,UAAU,aAAa,CAAI,IAAY;IAC1C,MAAM,GAAG,GAAG,GAAG,aAAa,GAAG,IAAI,EAAE,CAAC;IACtC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;IAE9C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;IAC7B,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,aAAa,CACrB,gBAAgB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,EAC/C,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,CAClC,CAAC;IACJ,CAAC;IACD,OAAO,GAAG,CAAC,IAAI,EAAgB,CAAC;AAClC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,KAAa,EACb,UAA8B,EAAE;IAEhC,MAAM,EAAE,IAAI,GAAG,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;IACzC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;QACjC,MAAM,EAAE,KAAK;QACb,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;KACrB,CAAC,CAAC;IAEH,MAAM,IAAI,GAAG,MAAM,aAAa,CAC9B,UAAU,MAAM,EAAE,CACnB,CAAC;IAEF,sEAAsE;IACtE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAChE,MAAM,CAAC,IAAI,CAAC,wBAAwB,KAAK,cAAc,OAAO,CAAC,MAAM,UAAU,CAAC,CAAC;IACjF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,EAAU;IAEV,MAAM,IAAI,GAAG,MAAM,aAAa,CAAkB,UAAU,kBAAkB,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACtF,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;IACrD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/services/skill-generator.d.ts

@@ -0,0 +1,33 @@
+export interface SkillData {
+    name: string;
+    version: string;
+    description: string;
+    repository: string;
+    nodes: SkillNodeInfo[];
+    examples: SkillExample[];
+}
+export interface SkillNodeInfo {
+    className: string;
+    displayName: string;
+    category: string;
+    inputs: Array<{
+        name: string;
+        type: string;
+        required: boolean;
+        default?: unknown;
+    }>;
+    outputs: Array<{
+        name: string;
+        type: string;
+    }>;
+}
+export interface SkillExample {
+    name: string;
+    description: string;
+    workflow?: unknown;
+}
+export declare function fetchGitHubFile(owner: string, repo: string, path: string): Promise<string>;
+export declare function parseNodeClassMappings(pythonSource: string): Record<string, string>;
+export declare function renderSkillMarkdown(data: SkillData): string;
+export declare function generateSkill(source: string): Promise<string>;
+//# sourceMappingURL=skill-generator.d.ts.map

package/dist/services/skill-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"skill-generator.d.ts","sourceRoot":"","sources":["../../src/services/skill-generator.ts"],"names":[],"mappings":"AAUA,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,aAAa,EAAE,CAAC;IACvB,QAAQ,EAAE,YAAY,EAAE,CAAC;CAC1B;AAED,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,OAAO,CAAC;QAClB,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAChD;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AA4BD,wBAAsB,eAAe,CACnC,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,MAAM,CAAC,CAcjB;AAmED,wBAAgB,sBAAsB,CACpC,YAAY,EAAE,MAAM,GACnB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAwCxB;AA4OD,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,CA4G3D;AAMD,wBAAsB,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CA0DnE"}

package/dist/services/skill-generator.js

@@ -0,0 +1,438 @@
+import { config } from "../config.js";
+import { getObjectInfo } from "../comfyui/client.js";
+import { ComfyUIError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+// ---------------------------------------------------------------------------
+// GitHub helpers
+// ---------------------------------------------------------------------------
+function githubHeaders() {
+    const headers = {
+        Accept: "application/vnd.github.v3+json",
+    };
+    if (config.githubToken) {
+        headers["Authorization"] = `Bearer ${config.githubToken}`;
+    }
+    return headers;
+}
+export async function fetchGitHubFile(owner, repo, path) {
+    const url = `https://api.github.com/repos/${owner}/${repo}/contents/${path}`;
+    const res = await fetch(url, { headers: githubHeaders() });
+    if (!res.ok) {
+        throw new ComfyUIError(`GitHub API error ${res.status} fetching ${path}`, "GITHUB_ERROR");
+    }
+    const data = (await res.json());
+    if (data.encoding === "base64") {
+        return Buffer.from(data.content, "base64").toString("utf-8");
+    }
+    return data.content;
+}
+async function listGitHubDir(owner, repo, path = "") {
+    const url = `https://api.github.com/repos/${owner}/${repo}/contents/${path}`;
+    const res = await fetch(url, { headers: githubHeaders() });
+    if (!res.ok) {
+        throw new ComfyUIError(`GitHub API error ${res.status} listing ${path || "/"}`, "GITHUB_ERROR");
+    }
+    const data = await res.json();
+    if (!Array.isArray(data))
+        return [];
+    return data;
+}
+// ---------------------------------------------------------------------------
+// Registry helper
+// ---------------------------------------------------------------------------
+async function resolveRegistryRepo(registryId) {
+    const url = `https://api.comfy.org/nodes/${encodeURIComponent(registryId)}`;
+    const res = await fetch(url, {
+        headers: { Accept: "application/json" },
+    });
+    if (!res.ok) {
+        throw new ComfyUIError(`ComfyUI Registry error ${res.status} for "${registryId}"`, "REGISTRY_ERROR");
+    }
+    const data = (await res.json());
+    if (!data.repository) {
+        throw new ComfyUIError(`No repository URL found for registry ID "${registryId}"`, "REGISTRY_ERROR");
+    }
+    return data.repository;
+}
+// ---------------------------------------------------------------------------
+// Parse owner/repo from a GitHub URL
+// ---------------------------------------------------------------------------
+function parseGitHubUrl(url) {
+    // Handles: https://github.com/owner/repo, https://github.com/owner/repo.git
+    const match = url.match(/github\.com\/([^/]+)\/([^/?.#]+)/);
+    if (!match) {
+        throw new ComfyUIError(`Cannot parse GitHub owner/repo from "${url}"`, "VALIDATION_ERROR");
+    }
+    return { owner: match[1], repo: match[2].replace(/\.git$/, "") };
+}
+// ---------------------------------------------------------------------------
+// NODE_CLASS_MAPPINGS parser
+// ---------------------------------------------------------------------------
+export function parseNodeClassMappings(pythonSource) {
+    const mappings = {};
+    // Pattern 1: NODE_CLASS_MAPPINGS = { "DisplayName": ClassName, ... }
+    // Handles multiline dicts with various quoting
+    const dictMatch = pythonSource.match(/NODE_CLASS_MAPPINGS\s*=\s*\{([^}]*)\}/s);
+    if (dictMatch) {
+        const body = dictMatch[1];
+        // Match entries like "DisplayName": ClassName or 'DisplayName': ClassName
+        const entryPattern = /["']([^"']+)["']\s*:\s*([A-Za-z_]\w*)/g;
+        let m;
+        while ((m = entryPattern.exec(body)) !== null) {
+            mappings[m[1]] = m[2];
+        }
+    }
+    // Pattern 2: NODE_CLASS_MAPPINGS["DisplayName"] = ClassName
+    const assignPattern = /NODE_CLASS_MAPPINGS\[["']([^"']+)["']\]\s*=\s*([A-Za-z_]\w*)/g;
+    let m;
+    while ((m = assignPattern.exec(pythonSource)) !== null) {
+        mappings[m[1]] = m[2];
+    }
+    // Pattern 3: NODE_CLASS_MAPPINGS.update({ ... })
+    const updateMatch = pythonSource.match(/NODE_CLASS_MAPPINGS\.update\(\s*\{([^}]*)\}\s*\)/s);
+    if (updateMatch) {
+        const body = updateMatch[1];
+        const entryPattern = /["']([^"']+)["']\s*:\s*([A-Za-z_]\w*)/g;
+        let m2;
+        while ((m2 = entryPattern.exec(body)) !== null) {
+            mappings[m2[1]] = m2[2];
+        }
+    }
+    return mappings;
+}
+// ---------------------------------------------------------------------------
+// Collect Python files with NODE_CLASS_MAPPINGS from a repo
+// ---------------------------------------------------------------------------
+async function findNodeMappings(owner, repo) {
+    const allMappings = {};
+    // List root directory
+    const rootEntries = await listGitHubDir(owner, repo);
+    // Gather Python files at root level + common sub-directories
+    const pyFiles = [];
+    const subDirs = [];
+    for (const entry of rootEntries) {
+        if (entry.type === "file" && entry.name.endsWith(".py")) {
+            pyFiles.push(entry.path);
+        }
+        else if (entry.type === "dir") {
+            subDirs.push(entry.path);
+        }
+    }
+    // Check common node directories (first level only to avoid excessive API calls)
+    for (const dir of subDirs.slice(0, 10)) {
+        try {
+            const dirEntries = await listGitHubDir(owner, repo, dir);
+            for (const entry of dirEntries) {
+                if (entry.type === "file" && entry.name.endsWith(".py")) {
+                    pyFiles.push(entry.path);
+                }
+            }
+        }
+        catch {
+            // Skip directories we can't read
+        }
+    }
+    // Fetch Python files and parse NODE_CLASS_MAPPINGS (limit to avoid rate limits)
+    const filesToCheck = pyFiles.slice(0, 30);
+    const fetches = filesToCheck.map(async (path) => {
+        try {
+            const source = await fetchGitHubFile(owner, repo, path);
+            if (source.includes("NODE_CLASS_MAPPINGS")) {
+                const parsed = parseNodeClassMappings(source);
+                Object.assign(allMappings, parsed);
+            }
+        }
+        catch {
+            // Skip files we can't fetch
+        }
+    });
+    await Promise.all(fetches);
+    return allMappings;
+}
+// ---------------------------------------------------------------------------
+// Find example workflows in the repo
+// ---------------------------------------------------------------------------
+async function findExampleWorkflows(owner, repo) {
+    const examples = [];
+    const jsonPaths = [];
+    // Check root and common example dirs
+    const dirsToCheck = ["", "examples", "workflows", "example_workflows"];
+    for (const dir of dirsToCheck) {
+        try {
+            const entries = await listGitHubDir(owner, repo, dir);
+            for (const entry of entries) {
+                if (entry.type === "file" && entry.name.endsWith(".json")) {
+                    jsonPaths.push(entry.path);
+                }
+            }
+        }
+        catch {
+            // Directory may not exist
+        }
+    }
+    // Fetch up to 5 example workflows
+    for (const path of jsonPaths.slice(0, 5)) {
+        try {
+            const content = await fetchGitHubFile(owner, repo, path);
+            const workflow = JSON.parse(content);
+            // Quick heuristic: ComfyUI workflows have nodes with class_type
+            const isWorkflow = typeof workflow === "object" &&
+                workflow !== null &&
+                Object.values(workflow).some((v) => typeof v === "object" &&
+                    v !== null &&
+                    "class_type" in v);
+            if (isWorkflow) {
+                const name = path
+                    .split("/")
+                    .pop()
+                    .replace(/\.json$/, "")
+                    .replace(/[_-]/g, " ");
+                examples.push({
+                    name,
+                    description: `Example workflow from ${path}`,
+                    workflow,
+                });
+            }
+        }
+        catch {
+            // Skip unparseable files
+        }
+    }
+    return examples;
+}
+// ---------------------------------------------------------------------------
+// Fetch README for description
+// ---------------------------------------------------------------------------
+async function fetchReadme(owner, repo) {
+    const candidates = ["README.md", "readme.md", "README.rst", "README"];
+    for (const name of candidates) {
+        try {
+            return await fetchGitHubFile(owner, repo, name);
+        }
+        catch {
+            // Try next
+        }
+    }
+    return null;
+}
+function extractFirstParagraph(markdown) {
+    // Skip leading headings/badges, grab first non-empty text block
+    const lines = markdown.split("\n");
+    const paragraphLines = [];
+    let started = false;
+    for (const line of lines) {
+        const trimmed = line.trim();
+        if (!started) {
+            // Skip headings, blank lines, badge images, HTML
+            if (trimmed === "" ||
+                trimmed.startsWith("#") ||
+                trimmed.startsWith("![") ||
+                trimmed.startsWith("<") ||
+                trimmed.startsWith("[![")) {
+                continue;
+            }
+            started = true;
+        }
+        if (started) {
+            if (trimmed === "" || trimmed.startsWith("#"))
+                break;
+            paragraphLines.push(trimmed);
+        }
+    }
+    return paragraphLines.join(" ").slice(0, 500);
+}
+// ---------------------------------------------------------------------------
+// Match nodes against ComfyUI object_info
+// ---------------------------------------------------------------------------
+function buildNodeInfo(classNameToDisplay, objectInfo) {
+    const nodes = [];
+    for (const [displayName, className] of Object.entries(classNameToDisplay)) {
+        // object_info is keyed by the internal class name
+        const nodeDef = objectInfo[className];
+        const inputs = [];
+        const outputs = [];
+        if (nodeDef) {
+            // Required inputs
+            if (nodeDef.input.required) {
+                for (const [name, spec] of Object.entries(nodeDef.input.required)) {
+                    const type = Array.isArray(spec[0]) ? spec[0].join(" | ") : spec[0];
+                    const defaultVal = spec[1]?.default;
+                    inputs.push({ name, type, required: true, default: defaultVal });
+                }
+            }
+            // Optional inputs
+            if (nodeDef.input.optional) {
+                for (const [name, spec] of Object.entries(nodeDef.input.optional)) {
+                    const type = Array.isArray(spec[0]) ? spec[0].join(" | ") : spec[0];
+                    const defaultVal = spec[1]?.default;
+                    inputs.push({ name, type, required: false, default: defaultVal });
+                }
+            }
+            // Outputs
+            for (let i = 0; i < nodeDef.output.length; i++) {
+                outputs.push({
+                    name: nodeDef.output_name[i] || nodeDef.output[i],
+                    type: nodeDef.output[i],
+                });
+            }
+        }
+        nodes.push({
+            className,
+            displayName: nodeDef?.display_name || displayName,
+            category: nodeDef?.category || "unknown",
+            inputs,
+            outputs,
+        });
+    }
+    // Sort by category then display name for consistent output
+    nodes.sort((a, b) => a.category === b.category
+        ? a.displayName.localeCompare(b.displayName)
+        : a.category.localeCompare(b.category));
+    return nodes;
+}
+// ---------------------------------------------------------------------------
+// Markdown renderer
+// ---------------------------------------------------------------------------
+export function renderSkillMarkdown(data) {
+    const tags = new Set(["comfyui"]);
+    for (const node of data.nodes) {
+        if (node.category && node.category !== "unknown") {
+            // Take the top-level category
+            const top = node.category.split("/")[0].trim().toLowerCase();
+            if (top)
+                tags.add(top);
+        }
+    }
+    const lines = [];
+    // Frontmatter
+    lines.push("---");
+    lines.push(`name: ${data.name}`);
+    lines.push(`version: ${data.version}`);
+    lines.push(`description: ${data.description}`);
+    lines.push(`tags: [${[...tags].join(", ")}]`);
+    lines.push("---");
+    lines.push("");
+    // Title
+    lines.push(`# ${data.name}`);
+    lines.push("");
+    if (data.description) {
+        lines.push(data.description);
+        lines.push("");
+    }
+    if (data.repository) {
+        lines.push(`Repository: ${data.repository}`);
+        lines.push("");
+    }
+    // Nodes
+    lines.push("## Nodes");
+    lines.push("");
+    for (const node of data.nodes) {
+        lines.push(`### ${node.displayName}`);
+        lines.push(`**Class**: \`${node.className}\``);
+        lines.push(`**Category**: ${node.category}`);
+        lines.push("");
+        if (node.inputs.length > 0) {
+            lines.push("**Inputs**:");
+            lines.push("| Name | Type | Required | Default |");
+            lines.push("|------|------|----------|---------|");
+            for (const inp of node.inputs) {
+                const def = inp.default !== undefined ? String(inp.default) : "";
+                lines.push(`| ${inp.name} | ${inp.type} | ${inp.required ? "Yes" : "No"} | ${def} |`);
+            }
+            lines.push("");
+        }
+        if (node.outputs.length > 0) {
+            lines.push("**Outputs**:");
+            lines.push("| Name | Type |");
+            lines.push("|------|------|");
+            for (const out of node.outputs) {
+                lines.push(`| ${out.name} | ${out.type} |`);
+            }
+            lines.push("");
+        }
+    }
+    // Examples
+    if (data.examples.length > 0) {
+        lines.push("## Usage Examples");
+        lines.push("");
+        for (const ex of data.examples) {
+            lines.push(`### ${ex.name}`);
+            lines.push(ex.description);
+            lines.push("");
+            if (ex.workflow && typeof ex.workflow === "object") {
+                const wf = ex.workflow;
+                const nodeTypes = [
+                    ...new Set(Object.values(wf)
+                        .filter((v) => v.class_type)
+                        .map((v) => v.class_type)),
+                ];
+                if (nodeTypes.length > 0) {
+                    lines.push(`Key nodes used: ${nodeTypes.join(", ")}`);
+                    lines.push("");
+                }
+            }
+        }
+    }
+    // Composition Patterns
+    lines.push("## Composition Patterns");
+    lines.push("");
+    lines.push("- To use with LoRA: Insert LoraLoader between CheckpointLoader and the first node that uses MODEL/CLIP");
+    lines.push("- To chain with ControlNet: Add ControlNetLoader and ControlNetApply before the sampler");
+    lines.push("");
+    return lines.join("\n");
+}
+// ---------------------------------------------------------------------------
+// Main entry point
+// ---------------------------------------------------------------------------
+export async function generateSkill(source) {
+    logger.info("Generating skill", { source });
+    // Step 1: Resolve repository URL
+    let repoUrl;
+    if (source.includes("github.com")) {
+        repoUrl = source;
+    }
+    else {
+        // Assume it's a registry ID
+        logger.info("Looking up registry ID", { registryId: source });
+        repoUrl = await resolveRegistryRepo(source);
+    }
+    const { owner, repo } = parseGitHubUrl(repoUrl);
+    logger.info("Resolved repository", { owner, repo });
+    // Step 2: Fetch README for description
+    const readme = await fetchReadme(owner, repo);
+    const description = readme
+        ? extractFirstParagraph(readme)
+        : "ComfyUI custom node pack";
+    // Step 3: Find NODE_CLASS_MAPPINGS in Python files
+    const classNameToDisplay = await findNodeMappings(owner, repo);
+    const nodeCount = Object.keys(classNameToDisplay).length;
+    logger.info("Found node class mappings", { count: nodeCount });
+    if (nodeCount === 0) {
+        logger.warn("No NODE_CLASS_MAPPINGS found in repository");
+    }
+    // Step 4: Match against ComfyUI object_info (best-effort)
+    let objectInfo = {};
+    try {
+        objectInfo = await getObjectInfo();
+    }
+    catch (err) {
+        logger.warn("Could not fetch object_info from ComfyUI (server may be offline)", {
+            error: err instanceof Error ? err.message : String(err),
+        });
+    }
+    const nodes = buildNodeInfo(classNameToDisplay, objectInfo);
+    // Step 5: Find example workflows
+    const examples = await findExampleWorkflows(owner, repo);
+    logger.info("Found example workflows", { count: examples.length });
+    // Step 6: Render SKILL.md
+    const data = {
+        name: repo,
+        version: "1.0.0",
+        description,
+        repository: repoUrl,
+        nodes,
+        examples,
+    };
+    return renderSkillMarkdown(data);
+}
+//# sourceMappingURL=skill-generator.js.map

package/dist/services/skill-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"skill-generator.js","sourceRoot":"","sources":["../../src/services/skill-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAErD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AA8C5C,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAS,aAAa;IACpB,MAAM,OAAO,GAA2B;QACtC,MAAM,EAAE,gCAAgC;KACzC,CAAC;IACF,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5D,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,KAAa,EACb,IAAY,EACZ,IAAY;IAEZ,MAAM,GAAG,GAAG,gCAAgC,KAAK,IAAI,IAAI,aAAa,IAAI,EAAE,CAAC;IAC7E,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;IAC3D,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,YAAY,CACpB,oBAAoB,GAAG,CAAC,MAAM,aAAa,IAAI,EAAE,EACjD,cAAc,CACf,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAsB,CAAC;IACrD,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IACD,OAAO,IAAI,CAAC,OAAO,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,KAAa,EACb,IAAY,EACZ,IAAI,GAAG,EAAE;IAET,MAAM,GAAG,GAAG,gCAAgC,KAAK,IAAI,IAAI,aAAa,IAAI,EAAE,CAAC;IAC7E,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;IAC3D,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,YAAY,CACpB,oBAAoB,GAAG,CAAC,MAAM,YAAY,IAAI,IAAI,GAAG,EAAE,EACvD,cAAc,CACf,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IACpC,OAAO,IAA4B,CAAC;AACtC,CAAC;AAED,8EAA8E;AAC9E,kBAAkB;AAClB,8EAA8E;AAE9E,KAAK,UAAU,mBAAmB,CAAC,UAAkB;IACnD,MAAM,GAAG,GAAG,+BAA+B,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;IAC5E,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;QAC3B,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;KACxC,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,YAAY,CACpB,0BAA0B,GAAG,CAAC,MAAM,SAAS,UAAU,GAAG,EAC1D,gBAAgB,CACjB,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;IAC3D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,IAAI,YAAY,CACpB,4CAA4C,UAAU,GAAG,EACzD,gBAAgB,CACjB,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,CAAC,UAAU,CAAC;AACzB,CAAC;AAED,8EAA8E;AAC9E,qCAAqC;AACrC,8EAA8E;AAE9E,SAAS,cAAc,CAAC,GAAW;IACjC,4EAA4E;IAC5E,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CACrB,kCAAkC,CACnC,CAAC;IACF,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,YAAY,CACpB,wCAAwC,GAAG,GAAG,EAC9C,kBAAkB,CACnB,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC;AACnE,CAAC;AAED,8EAA8E;AAC9E,6BAA6B;AAC7B,8EAA8E;AAE9E,MAAM,UAAU,sBAAsB,CACpC,YAAoB;IAEpB,MAAM,QAAQ,GAA2B,EAAE,CAAC;IAE5C,qEAAqE;IACrE,+CAA+C;IAC/C,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAClC,wCAAwC,CACzC,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC1B,0EAA0E;QAC1E,MAAM,YAAY,GAAG,wCAAwC,CAAC;QAC9D,IAAI,CAAyB,CAAC;QAC9B,OAAO,CAAC,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9C,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED,4DAA4D;IAC5D,MAAM,aAAa,GACjB,+DAA+D,CAAC;IAClE,IAAI,CAAyB,CAAC;IAC9B,OAAO,CAAC,CAAC,GAAG,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACvD,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IAED,iDAAiD;IACjD,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,CACpC,mDAAmD,CACpD,CAAC;IACF,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,YAAY,GAAG,wCAAwC,CAAC;QAC9D,IAAI,EAA0B,CAAC;QAC/B,OAAO,CAAC,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC/C,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,8EAA8E;AAC9E,4DAA4D;AAC5D,8EAA8E;AAE9E,KAAK,UAAU,gBAAgB,CAC7B,KAAa,EACb,IAAY;IAEZ,MAAM,WAAW,GAA2B,EAAE,CAAC;IAE/C,sBAAsB;IACtB,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAErD,6DAA6D;IAC7D,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAa,EAAE,CAAC;IAE7B,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAChC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,gFAAgF;IAChF,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACzD,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;gBAC/B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBACxD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iCAAiC;QACnC,CAAC;IACH,CAAC;IAED,gFAAgF;IAChF,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QAC9C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;YACxD,IAAI,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBAC3C,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;gBAC9C,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,4BAA4B;QAC9B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC3B,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,8EAA8E;AAC9E,qCAAqC;AACrC,8EAA8E;AAE9E,KAAK,UAAU,oBAAoB,CACjC,KAAa,EACb,IAAY;IAEZ,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,SAAS,GAAa,EAAE,CAAC;IAE/B,qCAAqC;IACrC,MAAM,WAAW,GAAG,CAAC,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,mBAAmB,CAAC,CAAC;IAEvE,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACtD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1D,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,0BAA0B;QAC5B,CAAC;IACH,CAAC;IAED,kCAAkC;IAClC,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;YACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACrC,gEAAgE;YAChE,MAAM,UAAU,GACd,OAAO,QAAQ,KAAK,QAAQ;gBAC5B,QAAQ,KAAK,IAAI;gBACjB,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAC1B,CAAC,CAAU,EAAE,EAAE,CACb,OAAO,CAAC,KAAK,QAAQ;oBACrB,CAAC,KAAK,IAAI;oBACV,YAAY,IAAK,CAA6B,CACjD,CAAC;YACJ,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,IAAI,GAAG,IAAI;qBACd,KAAK,CAAC,GAAG,CAAC;qBACV,GAAG,EAAG;qBACN,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;qBACtB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;gBACzB,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI;oBACJ,WAAW,EAAE,yBAAyB,IAAI,EAAE;oBAC5C,QAAQ;iBACT,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,yBAAyB;QAC3B,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,8EAA8E;AAC9E,+BAA+B;AAC/B,8EAA8E;AAE9E,KAAK,UAAU,WAAW,CACxB,KAAa,EACb,IAAY;IAEZ,MAAM,UAAU,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;IACtE,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,OAAO,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACP,WAAW;QACb,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC7C,gEAAgE;IAChE,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,iDAAiD;YACjD,IACE,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;gBACvB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;gBACxB,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;gBACvB,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EACzB,CAAC;gBACD,SAAS;YACX,CAAC;YACD,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,OAAO,KAAK,EAAE,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,MAAM;YACrD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,OAAO,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAChD,CAAC;AAED,8EAA8E;AAC9E,0CAA0C;AAC1C,8EAA8E;AAE9E,SAAS,aAAa,CACpB,kBAA0C,EAC1C,UAAsB;IAEtB,MAAM,KAAK,GAAoB,EAAE,CAAC;IAElC,KAAK,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC1E,kDAAkD;QAClD,MAAM,OAAO,GAA+B,UAAU,CAAC,SAAS,CAAC,CAAC;QAElE,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,MAAM,OAAO,GAA6B,EAAE,CAAC;QAE7C,IAAI,OAAO,EAAE,CAAC;YACZ,kBAAkB;YAClB,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAClE,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBACpE,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;oBACpC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;YACD,kBAAkB;YAClB,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAClE,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;oBACpE,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;oBACpC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;gBACpE,CAAC;YACH,CAAC;YACD,UAAU;YACV,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;oBACjD,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;iBACxB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,KAAK,CAAC,IAAI,CAAC;YACT,SAAS;YACT,WAAW,EAAE,OAAO,EAAE,YAAY,IAAI,WAAW;YACjD,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,SAAS;YACxC,MAAM;YACN,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,2DAA2D;IAC3D,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAClB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ;QACvB,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC;QAC5C,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CACzC,CAAC;IAEF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAE9E,MAAM,UAAU,mBAAmB,CAAC,IAAe;IACjD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAS,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACjD,8BAA8B;YAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC7D,IAAI,GAAG;gBAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,cAAc;IACd,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACjC,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACvC,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC/C,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9C,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,QAAQ;IACR,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,QAAQ;IACR,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACvB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;YACnD,KAAK,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;YACnD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC9B,MAAM,GAAG,GACP,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvD,KAAK,CAAC,IAAI,CACR,KAAK,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,MAAM,GAAG,IAAI,CAC1E,CAAC;YACJ,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC9B,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC/B,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC;YAC9C,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAED,WAAW;IACX,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAC7B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,IAAI,EAAE,CAAC,QAAQ,IAAI,OAAO,EAAE,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACnD,MAAM,EAAE,GAAG,EAAE,CAAC,QAGb,CAAC;gBACF,MAAM,SAAS,GAAG;oBAChB,GAAG,IAAI,GAAG,CACR,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;yBACd,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;yBAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAW,CAAC,CAC7B;iBACF,CAAC;gBACF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,KAAK,CAAC,IAAI,CAAC,mBAAmB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACtD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,uBAAuB;IACvB,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACtC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CACR,wGAAwG,CACzG,CAAC;IACF,KAAK,CAAC,IAAI,CACR,yFAAyF,CAC1F,CAAC;IACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,MAAc;IAChD,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAE5C,iCAAiC;IACjC,IAAI,OAAe,CAAC;IACpB,IAAI,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QAClC,OAAO,GAAG,MAAM,CAAC;IACnB,CAAC;SAAM,CAAC;QACN,4BAA4B;QAC5B,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9D,OAAO,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpD,uCAAuC;IACvC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC9C,MAAM,WAAW,GAAG,MAAM;QACxB,CAAC,CAAC,qBAAqB,CAAC,MAAM,CAAC;QAC/B,CAAC,CAAC,0BAA0B,CAAC;IAE/B,mDAAmD;IACnD,MAAM,kBAAkB,GAAG,MAAM,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/D,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,CAAC;IACzD,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IAE/D,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;IAC5D,CAAC;IAED,0DAA0D;IAC1D,IAAI,UAAU,GAAe,EAAE,CAAC;IAChC,IAAI,CAAC;QACH,UAAU,GAAG,MAAM,aAAa,EAAE,CAAC;IACrC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,IAAI,CAAC,kEAAkE,EAAE;YAC9E,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IAED,MAAM,KAAK,GAAG,aAAa,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;IAE5D,iCAAiC;IACjC,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACzD,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAEnE,0BAA0B;IAC1B,MAAM,IAAI,GAAc;QACtB,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,WAAW;QACX,UAAU,EAAE,OAAO;QACnB,KAAK;QACL,QAAQ;KACT,CAAC;IAEF,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;AACnC,CAAC"}

package/dist/services/workflow-composer.d.ts

@@ -0,0 +1,43 @@
+import type { WorkflowJSON } from "../comfyui/types.js";
+export declare function getNextNodeId(workflow: WorkflowJSON): string;
+export declare const TEMPLATE_NAMES: string[];
+export declare function createWorkflow(template: string, params?: Record<string, unknown>): WorkflowJSON;
+interface SetInputOp {
+    op: "set_input";
+    node_id: string;
+    input_name: string;
+    value: unknown;
+}
+interface AddNodeOp {
+    op: "add_node";
+    class_type: string;
+    inputs?: Record<string, unknown>;
+    id?: string;
+}
+interface RemoveNodeOp {
+    op: "remove_node";
+    node_id: string;
+}
+interface ConnectOp {
+    op: "connect";
+    source_id: string;
+    output_index: number;
+    target_id: string;
+    input_name: string;
+}
+interface InsertBetweenOp {
+    op: "insert_between";
+    source_id: string;
+    output_index: number;
+    target_id: string;
+    input_name: string;
+    new_class_type: string;
+    new_inputs?: Record<string, unknown>;
+}
+export type ModifyOperation = SetInputOp | AddNodeOp | RemoveNodeOp | ConnectOp | InsertBetweenOp;
+export declare function modifyWorkflow(workflow: WorkflowJSON, operations: ModifyOperation[]): {
+    workflow: WorkflowJSON;
+    added_ids: string[];
+};
+export {};
+//# sourceMappingURL=workflow-composer.d.ts.map

package/dist/services/workflow-composer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-composer.d.ts","sourceRoot":"","sources":["../../src/services/workflow-composer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAgB,MAAM,qBAAqB,CAAC;AAKtE,wBAAgB,aAAa,CAAC,QAAQ,EAAE,YAAY,GAAG,MAAM,CAG5D;AAoQD,eAAO,MAAM,cAAc,UAAyB,CAAC;AAErD,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,MAAM,EAChB,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GACnC,YAAY,CAQd;AAID,UAAU,UAAU;IAClB,EAAE,EAAE,WAAW,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,UAAU,SAAS;IACjB,EAAE,EAAE,UAAU,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,UAAU,YAAY;IACpB,EAAE,EAAE,aAAa,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,SAAS;IACjB,EAAE,EAAE,SAAS,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,UAAU,eAAe;IACvB,EAAE,EAAE,gBAAgB,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,MAAM,eAAe,GACvB,UAAU,GACV,SAAS,GACT,YAAY,GACZ,SAAS,GACT,eAAe,CAAC;AA+EpB,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,YAAY,EACtB,UAAU,EAAE,eAAe,EAAE,GAC5B;IAAE,QAAQ,EAAE,YAAY,CAAC;IAAC,SAAS,EAAE,MAAM,EAAE,CAAA;CAAE,CAgCjD"}