Validate core_files.yaml base_platforms completeness (#162826)

script/hassfest/__main__.py

@@ -15,6 +15,7 @@ from . import (
     conditions,
     config_flow,
     config_schema,
+    core_files,
     dependencies,
     dhcp,
     docker,
@@ -62,6 +63,7 @@ INTEGRATION_PLUGINS = [
     config_flow,  # This needs to run last, after translations are processed
 ]
 HASS_PLUGINS = [
+    core_files,
     docker,
     mypy_config,
     metadata,

script/hassfest/core_files.py

@@ -0,0 +1,50 @@
+"""Validate .core_files.yaml base_platforms alignment with entity platforms."""
+
+import re
+
+from homeassistant.util.yaml import load_yaml_dict
+
+from .model import Config, Integration
+
+# Non-entity-platform components that belong in base_platforms
+EXTRA_BASE_PLATFORMS = {"diagnostics"}
+
+_COMPONENT_RE = re.compile(r"homeassistant/components/([^/]+)/\*\*")
+
+
+def validate(integrations: dict[str, Integration], config: Config) -> None:
+    """Validate that base_platforms in .core_files.yaml matches entity platforms."""
+    if config.specific_integrations:
+        return
+
+    core_files_path = config.root / ".core_files.yaml"
+    core_files = load_yaml_dict(str(core_files_path))
+
+    base_platform_entries = {
+        match.group(1)
+        for entry in core_files["base_platforms"]
+        if (match := _COMPONENT_RE.match(entry))
+    }
+
+    entity_platforms = {
+        integration.domain
+        for integration in integrations.values()
+        if integration.manifest.get("integration_type") == "entity"
+        and integration.domain != "tag"
+    }
+
+    expected = entity_platforms | EXTRA_BASE_PLATFORMS
+
+    for domain in sorted(expected - base_platform_entries):
+        config.add_error(
+            "core_files",
+            f"Entity platform '{domain}' is missing from "
+            "base_platforms in .core_files.yaml",
+        )
+
+    for domain in sorted(base_platform_entries - expected):
+        config.add_error(
+            "core_files",
+            f"'{domain}' in base_platforms in .core_files.yaml "
+            "is not an entity platform or in EXTRA_BASE_PLATFORMS",
+        )