From 64a8c2f5eba93d3de3ad29a34f2f3d22b0122827 Mon Sep 17 00:00:00 2001
From: Piotr Machowski <PiotrMachowski@users.noreply.github.com>
Date: Sat, 30 Sep 2023 04:37:31 +0200
Subject: [PATCH 1/3] Add dedicated jinja functions

---
 README.md                             | 102 +++++++++++++-------------
 custom_components/saver/__init__.py   | 101 ++++++++++++++++++++++---
 custom_components/saver/manifest.json |   2 +-
 custom_components/saver/services.yaml |  43 +++++++++--
 4 files changed, 181 insertions(+), 67 deletions(-)

diff --git a/README.md b/README.md
index 43abcb6..d2333db 100644
--- a/README.md
+++ b/README.md
@@ -22,18 +22,18 @@
 [community_forum]: https://community.home-assistant.io/t/custom-component-saver/204249
 
 
-[![Open your Home Assistant instance and start setting up a new integration.](https://my.home-assistant.io/badges/config_flow_start.svg)](https://my.home-assistant.io/redirect/config_flow_start/?domain=saver)
-
 # Saver
 
 This custom component allows you to save current state of any entity and use its data later to restore it.
 
-Additionally you can create simple variables and use their values in scripts.
+Additionally, you can create simple variables and use their values in scripts.
 
 ## Installation
 
 ### Using [HACS](https://hacs.xyz/) (recommended)
 
+[![Open your Home Assistant instance and open a repository inside the Home Assistant Community Store.](https://my.home-assistant.io/badges/hacs_repository.svg)](https://my.home-assistant.io/redirect/hacs_repository/?owner=PiotrMachowski&repository=Home-Assistant-custom-components-Saver&category=Integration)
+
 This integration can be installed using HACS.
 To do it search for `Saver` in *Integrations* section.
 
@@ -50,15 +50,19 @@ rm saver.zip
 
 ## Configuration
 
-### GUI
+### Using UI
+
+[![Open your Home Assistant instance and start setting up a new integration.](https://my.home-assistant.io/badges/config_flow_start.svg)](https://my.home-assistant.io/redirect/config_flow_start/?domain=saver)
+
 From the Home Assistant front page go to **Configuration** and then select **Integrations** from the list.
 
-Use the plus button in the bottom right to add a new integration called **Saver**.
+Use the "plus" button in the bottom right to add a new integration called **Saver**.
 
 The success dialog will appear or an error will be displayed in the popup.
 
 ### YAML
-In configuration.yaml:
+
+Add following code in configuration.yaml:
 ```yaml
 saver:
 ```
@@ -73,28 +77,6 @@ data:
   entity_id: cover.living_room
 ```
 
-### Restore state
-Executes the script using saved state of the entity.
-
-To use state of an entity you have to use a following value in a template: `state`.
-
-To use state attribute (in this case `current_position`) of an entity you have to use a following value in a template: `attr_current_position`.
-
-```yaml
-service: saver.restore_state
-data:
-  entity_id: cover.living_room
-  restore_script:
-    - service: cover.set_cover_position
-      data_template:
-        entity_id: cover.living_room
-        position: "{{ '{{ attr_current_position | int }}' }}"
-    - service: input_text.set_value
-      data_template:
-        entity_id: input_text.cover_description
-        value: "Cover is now {{ '{{ state }}' }}"
-```
-
 ### Delete saved state
 Deletes a saved state for an entity.
 ```yaml
@@ -120,7 +102,41 @@ data:
   name: counter
 ```
 
-### Execute script
+### Clear
+Deletes all saved data.
+```yaml
+service: saver.clear
+```
+
+### Deprecated services
+
+<details>
+<summary>Show</summary>
+
+
+#### Restore state
+Executes the script using saved state of the entity.
+
+To use state of an entity you have to use a following value in a template: `state`.
+
+To use state attribute (in this case `current_position`) of an entity you have to use a following value in a template: `attr_current_position`.
+
+```yaml
+service: saver.restore_state
+data:
+  entity_id: cover.living_room
+  restore_script:
+    - service: cover.set_cover_position
+      data_template:
+        entity_id: cover.living_room
+        position: "{{ '{{ attr_current_position | int }}' }}"
+    - service: input_text.set_value
+      data_template:
+        entity_id: input_text.cover_description
+        value: "Cover is now {{ '{{ state }}' }}"
+```
+
+#### Execute script
 Executes a script using all saved entities and variables.
 
 To use state of an entity (in this case `cover.living_room`) you have to use a following value in a template: `cover_living_room_state`.
@@ -145,28 +161,16 @@ data:
         value: "Counter has value {{ '{{ counter }}' }}"
 ```
 
-### Clear
-Deletes all saved data.
-```yaml
-service: saver.clear
-```
 
-## Using in templates
-It is possible to use saved data in templates via `saver.saver` entity:
+</details>
+
+
+## Using saved values in templates
+It is possible to use saved data in templates using `saver_entity` and `saver_variable` functions:
 ```yaml
-script:
-  - service: cover.set_cover_position
-    data_template:
-      entity_id: cover.living_room
-      position: "{{ state_attr('saver.saver', 'entities')['cover.living_room'].attributes.current_position | int }}"
-  - service: input_text.set_value
-    data_template:
-      entity_id: input_text.cover_description
-      value: "Cover is now {{ state_attr('saver.saver', 'entities')['cover.living_room'].state }}"
-  - service: input_text.set_value
-    data_template:
-      entity_id: input_text.counter_description
-      value: "Counter has value {{ state_attr('saver.saver', 'variables')['counter'] }}"
+{{ saver_entity("sun.sun") }}  # returns saved state of "sun.sun" entity
+{{ saver_entity("sun.sun", "azimuth") }}  # returns "azimuth" attribute of saved "sun.sun" entity
+{{ saver_variable("counter") }}  # returns saved variable "counter"
 ```
 
 ## Events
diff --git a/custom_components/saver/__init__.py b/custom_components/saver/__init__.py
index 9beb7de..4fa85aa 100644
--- a/custom_components/saver/__init__.py
+++ b/custom_components/saver/__init__.py
@@ -1,9 +1,11 @@
 import logging
+from typing import Any, Callable
 
-from homeassistant.core import Context
+from homeassistant.core import Context, HomeAssistant
 from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.script import Script, SCRIPT_MODE_PARALLEL
 from homeassistant.helpers.entity_component import EntityComponent
+from homeassistant.helpers.template import _get_state_if_valid, Template, TemplateEnvironment
 
 from .const import *
 
@@ -21,10 +23,89 @@ async def async_setup_entry(hass, config_entry):
     return result
 
 
+class SaverVariableTemplate:
+    def __init__(self, hass: HomeAssistant, entity_id: str) -> None:
+        self._hass = hass
+        self._entity_id = entity_id
+
+    def __call__(self, variable: str) -> Any:
+        saver_state = _get_state_if_valid(self._hass, self._entity_id)
+        if saver_state is None:
+            return None
+        variables = saver_state.attributes["variables"]
+        if variable in variables:
+            return variables[variable]
+        return None
+
+    def __repr__(self):
+        return "<template SaverVariable>"
+
+
+class SaverEntityTemplate:
+    def __init__(self, hass: HomeAssistant, entity_id: str) -> None:
+        self._hass = hass
+        self._entity_id = entity_id
+
+    def __call__(self, entity_id: str, attribute: str | None = None) -> Any:
+        saver_state = _get_state_if_valid(self._hass, self._entity_id)
+        if saver_state is None:
+            return None
+        entities = saver_state.attributes["entities"]
+        if entity_id not in entities:
+            return None
+        state = entities[entity_id]
+        state_val = state["state"] if isinstance(state, dict) else state.state
+        attrs = state["attributes"] if isinstance(state, dict) else state.attributes
+        if attribute is None:
+            return state_val
+        if attribute not in attrs:
+            return None
+        return attrs[attribute]
+
+    def __repr__(self):
+        return "<template SaverEntityTemplate>"
+
+
+def setup_templates(hass: HomeAssistant):
+    def is_safe_callable(self: TemplateEnvironment, obj):
+        # noinspection PyUnresolvedReferences
+        return (isinstance(obj, (SaverVariableTemplate, SaverEntityTemplate))
+                or self.saver_original_is_safe_callable_old(obj))
+
+    def patch_environment(env: TemplateEnvironment):
+        env.globals["saver_variable"] = SaverVariableTemplate(hass, f"{DOMAIN}.{DOMAIN}")
+        env.globals["saver_entity"] = SaverEntityTemplate(hass, f"{DOMAIN}.{DOMAIN}")
+
+    def patched_init(
+        self: TemplateEnvironment,
+        hass_param: HomeAssistant | None,
+        limited: bool | None = False,
+        strict: bool | None = False,
+        log_fn: Callable[[int, str], None] | None = None,
+    ):
+        # noinspection PyUnresolvedReferences
+        self.saver_original__init__(hass_param, limited, strict, log_fn)
+        patch_environment(self)
+
+    TemplateEnvironment.saver_original__init__ = TemplateEnvironment.__init__
+    TemplateEnvironment.__init__ = patched_init
+    TemplateEnvironment.saver_original_is_safe_callable_old = TemplateEnvironment.is_safe_callable
+    TemplateEnvironment.is_safe_callable = is_safe_callable
+
+    tpl = Template("", hass)
+    tpl._strict = False
+    tpl._limited = False
+    patch_environment(tpl._env)
+    tpl._strict = True
+    tpl._limited = False
+    patch_environment(tpl._env)
+
+
 def setup_entry(hass, config_entry):
     component = EntityComponent(_LOGGER, DOMAIN, hass)
     saver_entity = SaverEntity()
     component.add_entities([saver_entity])
+    setup_templates(hass)
 
     def clear(call):
         saver_entity.clear()
@@ -130,15 +211,11 @@ def restore(self, entity_id, restore_script, delete):
         self.schedule_update_ha_state()
 
     def save(self, entity_id):
-        tmp = {**self._entities_db}
-        tmp[entity_id] = self.hass.states.get(entity_id)
-        self._entities_db = tmp
+        self._entities_db = {**self._entities_db, entity_id: self.hass.states.get(entity_id)}
         self.schedule_update_ha_state()
 
     def set_variable(self, variable, value):
-        tmp = {**self._variables_db}
-        tmp[variable] = value
-        self._variables_db = tmp
+        self._variables_db = {**self._variables_db, variable: value}
         self.schedule_update_ha_state()
 
     @property
@@ -154,10 +231,12 @@ def state(self):
 
     async def async_added_to_hass(self):
         state = await self.async_get_last_state()
-        if state is not None \
-                and state.attributes is not None \
-                and "variables" in state.attributes and not isinstance(state.attributes["entities"], list) \
-                and "entities" in state.attributes and not isinstance(state.attributes["variables"], list):
+        if (
+            state is not None
+            and state.attributes is not None
+            and "variables" in state.attributes and not isinstance(state.attributes["entities"], list)
+            and "entities" in state.attributes and not isinstance(state.attributes["variables"], list)
+        ):
             self._variables_db = state.attributes["variables"]
             self._entities_db = state.attributes["entities"]
 
diff --git a/custom_components/saver/manifest.json b/custom_components/saver/manifest.json
index 9b710a0..7d5866b 100644
--- a/custom_components/saver/manifest.json
+++ b/custom_components/saver/manifest.json
@@ -8,5 +8,5 @@
   "iot_class": "calculated",
   "issue_tracker": "https://github.com/PiotrMachowski/Home-Assistant-custom-components-Saver/issues",
   "requirements": [],
-  "version": "v1.1.5"
+  "version": "v1.2.0"
 }
\ No newline at end of file
diff --git a/custom_components/saver/services.yaml b/custom_components/saver/services.yaml
index 72923b3..370c7e6 100644
--- a/custom_components/saver/services.yaml
+++ b/custom_components/saver/services.yaml
@@ -7,52 +7,83 @@ delete:
     Deletes a saved state for an entity.
   fields:
     entity_id:
+      name: Entity
       description: ID of entity to delete from saver.
-      example: light.patio
+      example: sun.sun
+      required: true
+      selector:
+        entity:
 
 delete_variable:
   description: >
     Deletes a saved variable.
   fields:
     name:
+      name: Variable
       description: Name of the variable to delete from saver.
       example: counter
+      required: true
+      selector:
+        text:
 
 execute:
   description: >
-    Executes a script using all saved entities and variables.
+    (Deprecated) Executes a script using all saved entities and variables.
   fields:
     script:
-      description: The script to excute using saved data.
+      name: Script
+      description: The script to execute using saved data.
+      required: true
 
 set_variable:
   description: >
     Sets the value to the variable.
   fields:
     name:
+      name: Variable
       description: Name of the variable to save.
       example: counter
+      required: true
+      selector:
+        text:
     value:
+      name: Value
       description: The new value
       example: 3
+      required: true
+      selector:
+        text:
 
 save_state:
   description: >
     Saves the state and parameters of the entity.
   fields:
     entity_id:
+      name: Entity
       description: ID of the entity that should be saved.
-      example: light.patio
+      example: sun.sun
+      required: true
+      selector:
+        entity:
 
 restore_state:
   description: >
-    Executes the script using saved state of the entity.
+    (Deprecated) Executes the script using saved state of the entity.
   fields:
     entity_id:
+      name: Entity
       description: ID of the entity to restore.
-      example: light.patio
+      example: sun.sun
+      required: true
+      selector:
+        entity:
     restore_script:
+      name: Restore script
       description: Script used to restore values.
+      required: true
     delete_after_run:
+      name: Delete after run
       description: "Deletes the saved state after an execution of the script. Default: true [Optional]"
       example: false
+      selector:
+        boolean:

From 8f759be0f0d7b97ce1d65b59750370e4bc1f7cdb Mon Sep 17 00:00:00 2001
From: Piotr Machowski <PiotrMachowski@users.noreply.github.com>
Date: Sat, 30 Sep 2023 05:10:34 +0200
Subject: [PATCH 2/3] Allow saving and deleting multiple entities at once

---
 custom_components/saver/__init__.py    | 69 ++++++++++++++------------
 custom_components/saver/config_flow.py |  5 +-
 custom_components/saver/const.py       | 11 +++-
 custom_components/saver/services.yaml  |  2 +
 4 files changed, 52 insertions(+), 35 deletions(-)

diff --git a/custom_components/saver/__init__.py b/custom_components/saver/__init__.py
index 4fa85aa..2dbb2bf 100644
--- a/custom_components/saver/__init__.py
+++ b/custom_components/saver/__init__.py
@@ -1,7 +1,8 @@
 import logging
-from typing import Any, Callable
+from typing import Any, Callable, Sequence
 
-from homeassistant.core import Context, HomeAssistant
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import Context, HomeAssistant, ServiceCall
 from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.script import Script, SCRIPT_MODE_PARALLEL
 from homeassistant.helpers.entity_component import EntityComponent
@@ -10,9 +11,10 @@
 from .const import *
 
 _LOGGER = logging.getLogger(__name__)
+CONFIG_SCHEMA = SAVER_SCHEMA
 
 
-def setup(hass, config):
+def setup(hass, config) -> bool:
     if DOMAIN not in config:
         return True
     return setup_entry(hass, config)
@@ -37,7 +39,7 @@ def __call__(self, variable: str) -> Any:
             return variables[variable]
         return None
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return "<template SaverVariable>"
 
 
@@ -62,17 +64,17 @@ def __call__(self, entity_id: str, attribute: str | None = None) -> Any:
             return None
         return attrs[attribute]
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return "<template SaverEntityTemplate>"
 
 
-def setup_templates(hass: HomeAssistant):
-    def is_safe_callable(self: TemplateEnvironment, obj):
+def setup_templates(hass: HomeAssistant) -> None:
+    def is_safe_callable(self: TemplateEnvironment, obj) -> bool:
         # noinspection PyUnresolvedReferences
         return (isinstance(obj, (SaverVariableTemplate, SaverEntityTemplate))
                 or self.saver_original_is_safe_callable_old(obj))
 
-    def patch_environment(env: TemplateEnvironment):
+    def patch_environment(env: TemplateEnvironment) -> None:
         env.globals["saver_variable"] = SaverVariableTemplate(hass, f"{DOMAIN}.{DOMAIN}")
         env.globals["saver_entity"] = SaverEntityTemplate(hass, f"{DOMAIN}.{DOMAIN}")
 
@@ -82,7 +84,7 @@ def patched_init(
         limited: bool | None = False,
         strict: bool | None = False,
         log_fn: Callable[[int, str], None] | None = None,
-    ):
+    ) -> None:
         # noinspection PyUnresolvedReferences
         self.saver_original__init__(hass_param, limited, strict, log_fn)
         patch_environment(self)
@@ -101,35 +103,35 @@ def patched_init(
     patch_environment(tpl._env)
 
 
-def setup_entry(hass, config_entry):
+def setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
     component = EntityComponent(_LOGGER, DOMAIN, hass)
     saver_entity = SaverEntity()
     component.add_entities([saver_entity])
     setup_templates(hass)
 
-    def clear(call):
+    def clear(_call: ServiceCall) -> None:
         saver_entity.clear()
         hass.bus.fire('event_saver_cleared')
 
-    def delete(call):
+    def delete(call: ServiceCall) -> None:
         data = call.data
         entity_id = data[CONF_ENTITY_ID]
         saver_entity.delete(entity_id)
         hass.bus.fire('event_saver_deleted_entity', {'entity_id': entity_id})
 
-    def delete_variable(call):
+    def delete_variable(call: ServiceCall) -> None:
         data = call.data
         variable = data[CONF_NAME]
         saver_entity.delete_variable(variable)
         hass.bus.fire('event_saver_deleted_variable', {'variable': variable})
 
-    def execute(call):
+    def execute(call: ServiceCall) -> None:
         data = call.data
         script = data[CONF_SCRIPT]
         saver_entity.execute(script)
         hass.bus.fire('event_saver_executed', {'script': script})
 
-    def restore_state(call):
+    def restore_state(call: ServiceCall) -> None:
         data = call.data
         entity_id = data[CONF_ENTITY_ID]
         restore_script = data[CONF_RESTORE_SCRIPT]
@@ -137,13 +139,13 @@ def restore_state(call):
         saver_entity.restore(entity_id, restore_script, should_delete)
         hass.bus.fire('event_saver_restored', {'entity_id': entity_id})
 
-    def save_state(call):
+    def save_state(call: ServiceCall) -> None:
         data = call.data
         entity_id = data[CONF_ENTITY_ID]
         saver_entity.save(entity_id)
         hass.bus.fire('event_saver_saved_entity', {'entity_id': entity_id})
 
-    def set_variable(call):
+    def set_variable(call) -> None:
         data = call.data
         name = data[CONF_NAME]
         value = data[CONF_VALUE]
@@ -163,32 +165,33 @@ def set_variable(call):
 
 class SaverEntity(RestoreEntity):
 
-    def __init__(self):
+    def __init__(self) -> None:
         self._entities_db = {}
         self._variables_db = {}
 
     @property
-    def name(self):
+    def name(self) -> str:
         return DOMAIN
 
-    def clear(self):
+    def clear(self) -> None:
         self._entities_db = {}
         self._variables_db = {}
         self.schedule_update_ha_state()
 
-    def delete(self, entity_id):
+    def delete(self, entity_ids: list[str]) -> None:
         tmp = {**self._entities_db}
-        tmp.pop(entity_id)
+        for entity_id in entity_ids:
+            tmp.pop(entity_id)
         self._entities_db = tmp
         self.schedule_update_ha_state()
 
-    def delete_variable(self, variable):
+    def delete_variable(self, variable: str) -> None:
         tmp = {**self._variables_db}
         tmp.pop(variable)
         self._variables_db = tmp
         self.schedule_update_ha_state()
 
-    def execute(self, script):
+    def execute(self, script: Sequence[dict[str, Any]]) -> None:
         script = Script(self.hass, script, self.name, DOMAIN, script_mode=SCRIPT_MODE_PARALLEL)
         variables = {}
         variables.update(self._variables_db)
@@ -197,7 +200,7 @@ def execute(self, script):
         script.run(variables=variables, context=Context())
         self.schedule_update_ha_state()
 
-    def restore(self, entity_id, restore_script, delete):
+    def restore(self, entity_id: str, restore_script: Sequence[dict[str, Any]], delete: bool) -> None:
         if entity_id not in self._entities_db:
             return
         old = self._entities_db[entity_id]
@@ -210,26 +213,28 @@ def restore(self, entity_id, restore_script, delete):
         script.run(variables=variables, context=Context())
         self.schedule_update_ha_state()
 
-    def save(self, entity_id):
-        self._entities_db = {**self._entities_db, entity_id: self.hass.states.get(entity_id)}
+    def save(self, entity_ids: list[str]) -> None:
+        self._entities_db = {**self._entities_db}
+        for entity_id in entity_ids:
+            self._entities_db[entity_id] = self.hass.states.get(entity_id)
         self.schedule_update_ha_state()
 
-    def set_variable(self, variable, value):
+    def set_variable(self, variable: str, value: Any) -> None:
         self._variables_db = {**self._variables_db, variable: value}
         self.schedule_update_ha_state()
 
     @property
-    def state_attributes(self):
+    def state_attributes(self) -> dict[str, Any]:
         return {
             "entities": self._entities_db,
             "variables": self._variables_db
         }
 
     @property
-    def state(self):
+    def state(self) -> int:
         return len(self._entities_db) + len(self._variables_db)
 
-    async def async_added_to_hass(self):
+    async def async_added_to_hass(self) -> None:
         state = await self.async_get_last_state()
         if (
             state is not None
@@ -241,7 +246,7 @@ async def async_added_to_hass(self):
             self._entities_db = state.attributes["entities"]
 
     @staticmethod
-    def convert_to_variables(state, entity_id=None):
+    def convert_to_variables(state: Any, entity_id: str | None = None) -> dict:
         prefix = ""
         state_val = state["state"] if isinstance(state, dict) else state.state
         attrs = state["attributes"] if isinstance(state, dict) else state.attributes
diff --git a/custom_components/saver/config_flow.py b/custom_components/saver/config_flow.py
index af433a2..eb842fb 100644
--- a/custom_components/saver/config_flow.py
+++ b/custom_components/saver/config_flow.py
@@ -1,4 +1,7 @@
+from typing import Any
+
 from homeassistant import config_entries
+from homeassistant.data_entry_flow import FlowResult
 
 from .const import DOMAIN
 
@@ -8,7 +11,7 @@ class SaverFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
     VERSION = 1
     CONNECTION_CLASS = config_entries.CONN_CLASS_LOCAL_PUSH
 
-    async def async_step_user(self, user_input=None):
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult:
         await self.async_set_unique_id(DOMAIN)
         self._abort_if_unique_id_configured()
         if user_input is not None:
diff --git a/custom_components/saver/const.py b/custom_components/saver/const.py
index a7e0ccf..c8ebddd 100644
--- a/custom_components/saver/const.py
+++ b/custom_components/saver/const.py
@@ -5,6 +5,13 @@
 
 DOMAIN = 'saver'
 
+SAVER_SCHEMA = vol.Schema(
+    {
+        DOMAIN: vol.Schema({})
+    },
+    extra=vol.ALLOW_EXTRA
+)
+
 CONF_DELETE_AFTER_RUN = 'delete_after_run'
 CONF_RESTORE_SCRIPT = 'restore_script'
 CONF_SCRIPT = 'script'
@@ -16,7 +23,7 @@
 
 SERVICE_DELETE = 'delete'
 SERVICE_DELETE_SCHEMA = vol.Schema({
-    vol.Required(CONF_ENTITY_ID): cv.entity_id
+    vol.Required(CONF_ENTITY_ID): cv.entity_ids
 })
 
 SERVICE_DELETE_VARIABLE = 'delete_variable'
@@ -38,7 +45,7 @@
 
 SERVICE_SAVE_STATE = 'save_state'
 SERVICE_SAVE_STATE_SCHEMA = vol.Schema({
-    vol.Required(CONF_ENTITY_ID): cv.entity_id
+    vol.Required(CONF_ENTITY_ID): cv.entity_ids
 })
 
 SERVICE_SET_VARIABLE = 'set_variable'
diff --git a/custom_components/saver/services.yaml b/custom_components/saver/services.yaml
index 370c7e6..4ca659b 100644
--- a/custom_components/saver/services.yaml
+++ b/custom_components/saver/services.yaml
@@ -13,6 +13,7 @@ delete:
       required: true
       selector:
         entity:
+          multiple: true
 
 delete_variable:
   description: >
@@ -65,6 +66,7 @@ save_state:
       required: true
       selector:
         entity:
+          multiple: true
 
 restore_state:
   description: >

From 2bda5ac4cf4cac765c7ed77291a30bef1d96ba91 Mon Sep 17 00:00:00 2001
From: Piotr Machowski <PiotrMachowski@users.noreply.github.com>
Date: Sat, 30 Sep 2023 05:15:53 +0200
Subject: [PATCH 3/3] Add service names

---
 custom_components/saver/services.yaml | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/custom_components/saver/services.yaml b/custom_components/saver/services.yaml
index 4ca659b..551dae8 100644
--- a/custom_components/saver/services.yaml
+++ b/custom_components/saver/services.yaml
@@ -1,8 +1,10 @@
 clear:
+  name: Clear storage
   description: >
     Deletes all saved data.
 
 delete:
+  name: Delete entity
   description: >
     Deletes a saved state for an entity.
   fields:
@@ -16,6 +18,7 @@ delete:
           multiple: true
 
 delete_variable:
+  name: Delete variable
   description: >
     Deletes a saved variable.
   fields:
@@ -28,6 +31,7 @@ delete_variable:
         text:
 
 execute:
+  name: Execute
   description: >
     (Deprecated) Executes a script using all saved entities and variables.
   fields:
@@ -37,6 +41,7 @@ execute:
       required: true
 
 set_variable:
+  name: Set variable
   description: >
     Sets the value to the variable.
   fields:
@@ -56,6 +61,7 @@ set_variable:
         text:
 
 save_state:
+  name: Save state
   description: >
     Saves the state and parameters of the entity.
   fields:
@@ -69,6 +75,7 @@ save_state:
           multiple: true
 
 restore_state:
+  name: Restore state
   description: >
     (Deprecated) Executes the script using saved state of the entity.
   fields: