diff --git a/app/subgraph_manager.py b/app/subgraph_manager.py index 6a8f586a4..ceabf8a0f 100644 --- a/app/subgraph_manager.py +++ b/app/subgraph_manager.py @@ -1,6 +1,8 @@ from __future__ import annotations from typing import TypedDict +import json +import logging import os import folder_paths import glob @@ -40,6 +42,7 @@ class SubgraphManager: def __init__(self): self.cached_custom_node_subgraphs: dict[SubgraphEntry] | None = None self.cached_blueprint_subgraphs: dict[SubgraphEntry] | None = None + self.distribution = os.environ.get("DISTRIBUTION", "localhost") def _create_entry(self, file: str, source: str, node_pack: str) -> tuple[str, SubgraphEntry]: """Create a subgraph entry from a file path. Expects normalized path (forward slashes).""" @@ -90,15 +93,56 @@ class SubgraphManager: return subgraphs_dict async def get_blueprint_subgraphs(self, force_reload=False): - """Load subgraphs from the blueprints directory.""" + """Load subgraphs from the blueprints directory using index.json for discovery.""" if not force_reload and self.cached_blueprint_subgraphs is not None: return self.cached_blueprint_subgraphs subgraphs_dict: dict[SubgraphEntry] = {} blueprints_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'blueprints') - if os.path.exists(blueprints_dir): + index_path = os.path.join(blueprints_dir, "index.json") + if os.path.isfile(index_path): + try: + with open(index_path, "r", encoding="utf-8") as f: + categories = json.load(f) + except (json.JSONDecodeError, OSError) as e: + logging.error("Failed to load blueprint index %s: %s", index_path, e) + categories = [] + + if not isinstance(categories, list): + logging.error("Blueprint index.json is not a list: %s", index_path) + categories = [] + + for category in categories: + module_name = category.get("moduleName", "default") + for blueprint in category.get("blueprints", []): + name = blueprint.get("name") + if not name: + logging.warning("Blueprint entry missing 'name' in category '%s', skipping", module_name) + continue + + include_on = blueprint.get("includeOnDistributions") + if include_on is not None and self.distribution not in include_on: + continue + + file_by_dist = blueprint.get("fileByDistribution", {}) + filename = file_by_dist.get(self.distribution, f"{name}.json") + filepath = os.path.realpath(os.path.join(blueprints_dir, filename)) + if not filepath.startswith(os.path.realpath(blueprints_dir) + os.sep): + logging.warning("Blueprint path escapes blueprints directory: %s", filepath) + continue + + if not os.path.isfile(filepath): + logging.warning("Blueprint file not found: %s", filepath) + continue + + entry_id, entry = self._create_entry(filepath, Source.templates, module_name) + subgraphs_dict[entry_id] = entry + elif os.path.exists(blueprints_dir): + logging.warning("No blueprint index.json found at %s, falling back to glob", index_path) for file in glob.glob(os.path.join(blueprints_dir, "*.json")): + if os.path.basename(file) == "index.json": + continue file = file.replace('\\', '/') entry_id, entry = self._create_entry(file, Source.templates, "comfyui") subgraphs_dict[entry_id] = entry