Extract supported_os from config and push to the registry service

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,
+                "supported_os": node_config.project.supported_os,
             },
             "node_version": {
                 "version": node_config.project.version,

comfy_cli/registry/config_parser.py

@@ -157,6 +157,9 @@ def extract_node_configuration(
     urls_data = project_data.get("urls", {})
     comfy_data = data.get("tool", {}).get("comfy", {})
 
+    classifiers = project_data.get("classifiers", [])
+    supported_os = [c for c in classifiers if c.startswith("Operating System :: ")]
+
     license_data = project_data.get("license", {})
     if isinstance(license_data, str):
         license = License(text=license_data)
@@ -190,6 +193,7 @@ def extract_node_configuration(
             repository=urls_data.get("Repository", ""),
             issues=urls_data.get("Issues", ""),
         ),
+        supported_os=supported_os,  # Add the OS classifiers
     )
 
     comfy = ComfyConfig(

comfy_cli/registry/types.py

@@ -69,6 +69,7 @@ class ProjectConfig:
     dependencies: List[str] = field(default_factory=list)
     license: License = field(default_factory=License)
     urls: URLs = field(default_factory=URLs)
+    supported_os: List[str] = field(default_factory=list)
 
 
 @dataclass

tests/comfy_cli/registry/test_config_parser.py

@@ -127,3 +127,27 @@ def test_extract_license_incorrect_format():
         assert result is not None, "Expected PyProjectConfig, got None"
         assert isinstance(result, PyProjectConfig)
         assert result.project.license == License(text="MIT")
+
+
+def test_extract_node_configuration_with_os_classifiers():
+    mock_data = {
+        "project": {
+            "classifiers": [
+                "Operating System :: OS Independent",
+                "Operating System :: Microsoft :: Windows",
+                "Programming Language :: Python :: 3",
+                "Topic :: Software Development",
+            ]
+        }
+    }
+    with (
+        patch("os.path.isfile", return_value=True),
+        patch("builtins.open", mock_open()),
+        patch("tomlkit.load", return_value=mock_data),
+    ):
+        result = extract_node_configuration("fake_path.toml")
+
+        assert result is not None
+        assert len(result.project.supported_os) == 2
+        assert "Operating System :: OS Independent" in result.project.supported_os
+        assert "Operating System :: Microsoft :: Windows" in result.project.supported_os