Add homeassistant.set_floor_aliases service

custom_components/spook/ectoplasms/homeassistant/services/set_floor_aliases.py

@@ -0,0 +1,39 @@
+"""Spook - Your homie."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import voluptuous as vol
+
+from homeassistant.components.homeassistant import DOMAIN
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import config_validation as cv, floor_registry as fr
+
+from ....services import AbstractSpookAdminService
+
+if TYPE_CHECKING:
+    from homeassistant.core import ServiceCall
+
+
+class SpookService(AbstractSpookAdminService):
+    """Home Assistant service to set the aliases of a floor."""
+
+    domain = DOMAIN
+    service = "set_floor_aliases"
+    schema = {
+        vol.Required("floor_id"): cv.string,
+        vol.Required("aliases"): vol.All(cv.ensure_list, [cv.string]),
+    }
+
+    async def async_handle_service(self, call: ServiceCall) -> None:
+        """Handle the service call."""
+        floor_registry = fr.async_get(self.hass)
+        if not floor_registry.async_get_floor(call.data["floor_id"]):
+            msg = f"Floor {call.data['floor_id']} not found"
+            raise HomeAssistantError(msg)
+
+        floor_registry.async_update(
+            call.data["floor_id"],
+            aliases=set(call.data["aliases"]),
+        )

custom_components/spook/services.yaml

@@ -372,6 +372,24 @@ homeassistant_remove_alias_from_floor:
         object:
 
 homeassistant_set_floor_aliases:
+  name: Sets aliases for a floor 👻
+  description: >-
+    Sets aliases for a floor. Overwrite and removed any existing aliases,
+    fully replacing them with the new ones.
+  fields:
+    floor_id:
+      name: Floor ID
+      description: The ID of the floor to set the aliases for.
+      required: true
+      selector:
+        text:
+    aliases:
+      name: Aliases
+      description: The aliases to set for the floor.
+      required: true
+      selector:
+        object:
+
 homeassistant_ignore_all_discovered:
   name: Ignore all currently discovered devices 👻
   description: >-