added switches for ac and dc on off

custom_components/bluetti_bt/switch.py

@@ -0,0 +1,135 @@
+"""Bluetti BT switches."""
+
+from __future__ import annotations
+
+import logging
+
+from homeassistant.components.switch import SwitchEntity, SwitchDeviceClass
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.const import (
+    CONF_ADDRESS,
+    CONF_NAME,
+    EntityCategory,
+)
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+)
+
+from bluetti_mqtt.bluetooth import build_device
+from bluetti_mqtt.mqtt_client import (
+    NORMAL_DEVICE_FIELDS,
+    DC_INPUT_FIELDS,
+    MqttFieldType,
+)
+
+from . import device_info as dev_info, get_unique_id
+from .const import CONTROL_FIELDS, DOMAIN, ADDITIONAL_DEVICE_FIELDS
+from .coordinator import PollingCoordinator, DummyDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    """Setup switch entities."""
+
+    device_name = entry.data.get(CONF_NAME)
+    address = entry.data.get(CONF_ADDRESS)
+    if address is None:
+        _LOGGER.error("Device has no address")
+
+    # Generate device info
+    _LOGGER.info("Creating sensors for device with address %s", address)
+    device_info = dev_info(entry)
+
+    # Add sensors according to device_info
+    bluetti_device = build_device(address, device_name)
+    bluetti_device = DummyDevice(bluetti_device)
+
+    sensors_to_add = []
+    all_fields = NORMAL_DEVICE_FIELDS
+    all_fields.update(DC_INPUT_FIELDS)
+    all_fields.update(ADDITIONAL_DEVICE_FIELDS)
+    for field_key, field_config in all_fields.items():
+        if bluetti_device.has_field(field_key):
+            if field_config.type == MqttFieldType.BOOL:
+                if field_config.setter is True and field_key in CONTROL_FIELDS:
+                    sensors_to_add.append(
+                        BluettiSwitch(
+                            hass.data[DOMAIN][entry.entry_id],
+                            device_info,
+                            address,
+                            field_key,
+                            field_config.home_assistant_extra.get(CONF_NAME, ""),
+                        )
+                    )
+
+    async_add_entities(sensors_to_add)
+
+
+class BluettiSwitch(CoordinatorEntity, SwitchEntity):
+    """Bluetti universal switch."""
+
+    def __init__(
+        self,
+        coordinator: PollingCoordinator,
+        device_info: DeviceInfo,
+        address,
+        response_key: str,
+        name: str,
+        category: EntityCategory | None = None,
+    ):
+        """Init entity."""
+        super().__init__(coordinator)
+
+        e_name = f"{device_info.get('name')} {name}"
+        self._address = address
+        self._response_key = response_key
+
+        self._attr_device_info = device_info
+        self._attr_name = e_name
+        self._attr_available = False
+        self._attr_unique_id = get_unique_id(e_name)
+        self._attr_entity_category = category
+        self._attr_device_class = SwitchDeviceClass.OUTLET
+
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        """Handle updated data from the coordinator."""
+        _LOGGER.debug("Updating state of %s", self._attr_unique_id)
+        if not isinstance(self.coordinator.data, dict):
+            _LOGGER.error(
+                "Invalid data from coordinator (sensor.%s)", self._attr_unique_id
+            )
+            self._attr_available = False
+            return
+
+        response_data = self.coordinator.data.get(self._response_key)
+        if response_data is None:
+            self._attr_available = False
+            return
+
+        if not isinstance(response_data, bool):
+            _LOGGER.warning(
+                "Invalid response data type from coordinator (sensor.%s): %s",
+                self._attr_unique_id,
+                response_data,
+            )
+            self._attr_available = False
+            return
+
+        self._attr_available = True
+        self._attr_is_on = self.coordinator.data[self._response_key] is True
+        self.async_write_ha_state()
+
+    async def async_turn_on(self, **kwargs):
+        """Turn the entity on."""
+        _LOGGER.warning("Turn on")
+
+    async def async_turn_off(self, **kwargs):
+        """Turn the entity off."""
+        _LOGGER.warning("Turn off")