Parse banner url from toml file and send it to registry service while publishing a new node version

Author: coderfromthenorth93

comfy_cli/registry/api.py

@@ -54,6 +54,7 @@ def publish_node_version(self, node_config: PyProjectConfig, token) -> PublishNo
                 "name": node_config.tool_comfy.display_name,
                 "license": license_json,
                 "repository": node_config.project.urls.repository,
+                "banner_url": node_config.tool_comfy.banner_url,
                 "supported_os": node_config.project.supported_os,
                 "supported_accelerators": node_config.project.supported_accelerators,
                 "supported_comfyui_version": node_config.project.supported_comfyui_version,

comfy_cli/registry/config_parser.py

@@ -295,6 +295,7 @@ def extract_node_configuration(
         icon=comfy_data.get("Icon", ""),
         models=[Model(location=m["location"], model_url=m["model_url"]) for m in comfy_data.get("Models", [])],
         includes=comfy_data.get("includes", []),
+        banner_url=comfy_data.get("Banner", ""),
     )
 
     return PyProjectConfig(project=project, tool_comfy=comfy)

comfy_cli/registry/types.py

@@ -52,6 +52,7 @@ class ComfyConfig:
     icon: str = ""
     models: List[Model] = field(default_factory=list)
     includes: List[str] = field(default_factory=list)
+    banner_url: str = ""
 
 
 @dataclass

tests/comfy_cli/registry/test_config_parser.py

@@ -31,6 +31,7 @@ def mock_toml_data():
                 "Documentation": "https://docs.example.com",
                 "Repository": "https://github.com/example/test-project",
                 "Issues": "https://github.com/example/test-project/issues",
+                "Banner": "https://example.com/banner.png",
             },
         },
         "tool": {
@@ -77,6 +78,7 @@ def test_extract_node_configuration_success(mock_toml_data):
         assert result.tool_comfy.publisher_id == "test-publisher"
         assert result.tool_comfy.display_name == "Test Project"
         assert result.tool_comfy.icon == "icon.png"
+        assert result.tool_comfy.banner_url == "https://example.com/banner.png"
         assert len(result.tool_comfy.models) == 2
         assert result.tool_comfy.models[0] == Model(location="model1.bin", model_url="https://example.com/model1")