Add get_conditions_for_target websocket command (home-assistant#157344)

Co-authored-by: Erik Montnemery <[email protected]>

Author: abmantis

homeassistant/components/websocket_api/automation.py

@@ -9,6 +9,9 @@
 from homeassistant.const import CONF_TARGET
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import target as target_helpers
+from homeassistant.helpers.condition import (
+    async_get_all_descriptions as async_get_all_condition_descriptions,
+)
 from homeassistant.helpers.entity import (
     entity_sources,
     get_device_class,
@@ -199,6 +202,16 @@ async def async_get_triggers_for_target(
     )
 
 
+async def async_get_conditions_for_target(
+    hass: HomeAssistant, target_selector: ConfigType, expand_group: bool
+) -> set[str]:
+    """Get conditions for a target."""
+    descriptions = await async_get_all_condition_descriptions(hass)
+    return _async_get_automation_components_for_target(
+        hass, target_selector, expand_group, descriptions
+    )
+
+
 async def async_get_services_for_target(
     hass: HomeAssistant, target_selector: ConfigType, expand_group: bool
 ) -> set[str]:

homeassistant/components/websocket_api/commands.py

@@ -87,7 +87,11 @@
 from homeassistant.util.json import format_unserializable_data
 
 from . import const, decorators, messages
-from .automation import async_get_services_for_target, async_get_triggers_for_target
+from .automation import (
+    async_get_conditions_for_target,
+    async_get_services_for_target,
+    async_get_triggers_for_target,
+)
 from .connection import ActiveConnection
 from .messages import construct_event_message, construct_result_message
 
@@ -109,6 +113,7 @@ def async_register_commands(
     async_reg(hass, handle_execute_script)
     async_reg(hass, handle_extract_from_target)
     async_reg(hass, handle_fire_event)
+    async_reg(hass, handle_get_conditions_for_target)
     async_reg(hass, handle_get_config)
     async_reg(hass, handle_get_services)
     async_reg(hass, handle_get_services_for_target)
@@ -903,6 +908,29 @@ async def handle_get_triggers_for_target(
     connection.send_result(msg["id"], triggers)
 
 
+@decorators.websocket_command(
+    {
+        vol.Required("type"): "get_conditions_for_target",
+        vol.Required("target"): cv.TARGET_FIELDS,
+        vol.Optional("expand_group", default=True): bool,
+    }
+)
+@decorators.async_response
+async def handle_get_conditions_for_target(
+    hass: HomeAssistant, connection: ActiveConnection, msg: dict[str, Any]
+) -> None:
+    """Handle get conditions for target command.
+
+    This command returns all conditions that can be used with any entities that are currently
+    part of a target.
+    """
+    conditions = await async_get_conditions_for_target(
+        hass, msg["target"], msg["expand_group"]
+    )
+
+    connection.send_result(msg["id"], conditions)
+
+
 @decorators.websocket_command(
     {
         vol.Required("type"): "get_services_for_target",