Merge branch 'feature/migrate-sensor-entities-to-configs' into dev

Author: ooree23

custom_components/recalbox/__init__.py

@@ -19,7 +19,7 @@
 import hashlib
 
 _LOGGER = logging.getLogger(__name__)
-PLATFORMS = ["switch", "sensor"]
+PLATFORMS = ["switch", "sensor", "number"]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

custom_components/recalbox/number.py

@@ -0,0 +1,42 @@
+from homeassistant.components.number import NumberEntity
+from homeassistant.helpers.entity import EntityCategory
+from .const import DOMAIN
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    entities = [
+        RecalboxPortNumber(hass, config_entry, "api_port_os", "Port API OS", "mdi:api", 80),
+        RecalboxPortNumber(hass, config_entry, "api_port_gamesmanager", "Port API Games", "mdi:api", 81),
+        RecalboxPortNumber(hass, config_entry, "udp_recalbox", "Port UDP Recalbox", "mdi:remote", 1337),
+        RecalboxPortNumber(hass, config_entry, "udp_retroarch", "Port UDP RetroArch", "mdi:remote", 55355),
+        RecalboxPortNumber(hass, config_entry, "api_port_kodi", "Port API Kodi", "mdi:kodi", 8081),
+    ]
+    async_add_entities(entities)
+
+class RecalboxPortNumber(NumberEntity):
+    _attr_entity_category = EntityCategory.CONFIG
+    _attr_has_entity_name = True
+    _attr_native_min_value = 1
+    _attr_native_max_value = 65535
+    _attr_native_step = 1
+
+    def __init__(self, hass, config_entry, key, name, icon, defaultValue):
+        self.hass = hass
+        self._config_entry = config_entry
+        self._key = key
+        self._attr_name = name
+        self._attr_icon = icon
+        self._attr_unique_id = f"{config_entry.entry_id}_port_{key}"
+        self._default = defaultValue
+
+    @property
+    def native_value(self):
+        return self._config_entry.options.get(self._key, self._default)
+
+    async def async_set_native_value(self, value: float):
+        new_options = dict(self._config_entry.options)
+        new_options[self._key] = int(value)
+        self.hass.config_entries.async_update_entry(self._config_entry, options=new_options)
+
+    @property
+    def device_info(self):
+        return {"identifiers": {(DOMAIN, self._config_entry.entry_id)}}

custom_components/recalbox/sensor.py

@@ -9,19 +9,13 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     # On crée une liste d'entités à ajouter
     entities = [
         RecalboxDiagnosticSensor(config_entry, "host", "Host", "mdi:ip-network"),
-        RecalboxDiagnosticSensor(config_entry, "api_port_os", "Port API OS", "mdi:api", 80),
-        RecalboxDiagnosticSensor(config_entry, "api_port_gamesmanager", "Port API Games", "mdi:api", 81),
-        RecalboxDiagnosticSensor(config_entry, "udp_recalbox", "Port UDP Recalbox", "mdi:remote", 1337),
-        RecalboxDiagnosticSensor(config_entry, "udp_retroarch", "Port UDP RetroArch", "mdi:remote", 55355),
-        RecalboxDiagnosticSensor(config_entry, "api_port_kodi", "Port API Kodi", "mdi:kodi", 8081),
-        RecalboxDiagnosticSensor(config_entry, "only_ip_v4", "Force mDNS IP v4 only", "mdi:dns", True),
     ]
     async_add_entities(entities)
 
 
 class RecalboxDiagnosticSensor(SensorEntity):
     """Classe générique pour les diagnostics de la Recalbox."""
-    _attr_entity_category = EntityCategory.CONFIG
+    _attr_entity_category = EntityCategory.DIAGNOSTIC
     _attr_has_entity_name = True
 
     def __init__(self, config_entry, key, name, icon, default=None):

custom_components/recalbox/switch.py

@@ -3,6 +3,7 @@
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed, CoordinatorEntity
 from homeassistant.helpers import device_registry as dr
 from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity import EntityCategory
 from .const import DOMAIN
 from .translations_service import RecalboxTranslator
 from .api import RecalboxAPI
@@ -30,7 +31,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     # On crée l'entité en lui passant l'objet config_entry (qui contient l'IP)
     new_recalbox_entity = RecalboxEntity(hass, config_entry, api, coordinator)
     hass.data[DOMAIN]["instances"][config_entry.entry_id]["sensor_entity"] = new_recalbox_entity # pour la retrouver ailleurs plus facilement
-    async_add_entities([new_recalbox_entity])
+    async_add_entities([new_recalbox_entity, RecalboxForceV4Switch(hass, config_entry)])
 
 
 
@@ -397,3 +398,36 @@ async def async_added_to_hass(self):
             _LOGGER.debug("Premier ping échoué au démarrage : on laisse la recalbox sur OFF")
 
 
+
+
+# --------- SWITCH ON CONFIG -----------
+
+class RecalboxForceV4Switch(SwitchEntity):
+    _attr_entity_category = EntityCategory.CONFIG
+    _attr_has_entity_name = True
+    _attr_name = "Force mDNS IP v4 only"
+    _attr_icon = "mdi:dns"
+
+    def __init__(self, hass, config_entry):
+        self.hass = hass
+        self._config_entry = config_entry
+        self._attr_unique_id = f"{config_entry.entry_id}_only_ip_v4"
+
+    @property
+    def is_on(self):
+        return self._config_entry.options.get("only_ip_v4", True)
+
+    async def async_turn_on(self, **kwargs):
+        await self._update_config(True)
+
+    async def async_turn_off(self, **kwargs):
+        await self._update_config(False)
+
+    async def _update_config(self, value):
+        new_options = dict(self._config_entry.options)
+        new_options["only_ip_v4"] = value
+        self.hass.config_entries.async_update_entry(self._config_entry, options=new_options)
+
+    @property
+    def device_info(self):
+        return {"identifiers": {(DOMAIN, self._config_entry.entry_id)}}