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.project.urls.banner,
                 "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

@@ -31,6 +31,7 @@ def create_comfynode_config():
 
     urls = tomlkit.table()
     urls["Repository"] = ""
+    urls["Banner"] = ""
 
     project.add("urls", urls)
     document.add("project", project)
@@ -282,6 +283,7 @@ def extract_node_configuration(
             documentation=urls_data.get("Documentation", ""),
             repository=urls_data.get("Repository", ""),
             issues=urls_data.get("Issues", ""),
+            banner=urls_data.get("Banner", ""),
         ),
         supported_os=supported_os,
         supported_accelerators=supported_accelerators,

comfy_cli/registry/types.py

@@ -37,6 +37,7 @@ class URLs:
     documentation: str = ""
     repository: str = ""
     issues: str = ""
+    banner: 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": {
@@ -73,6 +74,7 @@ def test_extract_node_configuration_success(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",
         )
         assert result.tool_comfy.publisher_id == "test-publisher"
         assert result.tool_comfy.display_name == "Test Project"