Add sua as optimistic switch (Closes: #26) (#27)

custom_components/goecharger_mqtt/definitions.py

@@ -172,6 +172,7 @@ class GoEChargerSwitchEntityDescription(
     domain: str = "switch"
     payload_on: str = "true"
     payload_off: str = "false"
+    optimistic: bool = False
 
 
 @dataclass
@@ -2618,6 +2619,15 @@ def transform_code(value, mapping_table) -> str:
         entity_registry_enabled_default=True,
         disabled=False,
     ),
+    GoEChargerSwitchEntityDescription(
+        key="sua",
+        name="Simulate unplugging permanently",
+        optimistic=True,
+        entity_category=EntityCategory.CONFIG,
+        device_class=None,
+        entity_registry_enabled_default=True,
+        disabled=False,
+    ),
 )
 
 NUMBERS: tuple[GoEChargerNumberEntityDescription, ...] = (

custom_components/goecharger_mqtt/switch.py

@@ -39,23 +39,40 @@ def __init__(
         super().__init__(config_entry, description)
 
         self.entity_description = description
+        self._optimistic = self.entity_description.optimistic
 
     @property
     def available(self):
         """Return True if entity is available."""
-        return self._attr_is_on is not None
+        if self._optimistic:
+            return self._topic is not None
+
+        return self._topic is not None
+
+    @property
+    def assumed_state(self):
+        """Return true if we do optimistic updates."""
+        return self._optimistic
 
     async def async_turn_on(self, **kwargs):
         """Turn the switch on."""
         await mqtt.async_publish(
             self.hass, f"{self._topic}/set", self.entity_description.payload_on
         )
+        if self._optimistic:
+            # Optimistically assume that switch has changed state.
+            self._attr_is_on = True
+            self.async_write_ha_state()
 
     async def async_turn_off(self, **kwargs):
         """Turn the switch off."""
         await mqtt.async_publish(
             self.hass, f"{self._topic}/set", self.entity_description.payload_off
         )
+        if self._optimistic:
+            # Optimistically assume that switch has changed state.
+            self._attr_is_on = False
+            self.async_write_ha_state()
 
     async def async_added_to_hass(self):
         """Subscribe to MQTT events."""