56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from homeassistant.components import script
|
|
from homeassistant.const import (
|
|
EVENT_COMPONENT_LOADED,
|
|
EVENT_SERVICE_REGISTERED,
|
|
EVENT_SERVICE_REMOVED,
|
|
)
|
|
|
|
from ....entity_filtering import (
|
|
async_filter_known_services,
|
|
async_find_services_in_sequence,
|
|
async_get_all_services,
|
|
)
|
|
from ....repairs import AbstractSpookEntityComponentUnknownReferencesRepair
|
|
|
|
if TYPE_CHECKING:
|
|
from typing import Any
|
|
|
|
|
|
class SpookRepair(AbstractSpookEntityComponentUnknownReferencesRepair):
|
|
"""Spook repair tries to find unknown referenced services in scripts."""
|
|
|
|
domain = script.DOMAIN
|
|
repair = "script_unknown_service_references"
|
|
inspect_events = {
|
|
EVENT_COMPONENT_LOADED,
|
|
EVENT_SERVICE_REGISTERED,
|
|
EVENT_SERVICE_REMOVED,
|
|
}
|
|
inspect_config_entry_changed = True
|
|
inspect_on_reload = True
|
|
|
|
unavailable_entity_class = script.UnavailableScriptEntity
|
|
entity_label = "script"
|
|
reference_label = "services"
|
|
edit_url_pattern = "/config/script/edit/{unique_id}"
|
|
|
|
_known_services: set[str]
|
|
|
|
async def _async_setup_inspection(self) -> None:
|
|
"""Cache known services for this inspection cycle."""
|
|
self._known_services = async_get_all_services(self.hass)
|
|
|
|
async def _async_compute_unknown_references(self, entity: Any) -> set[str]:
|
|
"""Return unknown services called by ``entity``."""
|
|
return async_filter_known_services(
|
|
self.hass,
|
|
services=async_find_services_in_sequence(entity.script.sequence),
|
|
known_services=self._known_services,
|
|
)
|