create device

Author: johnneerdael

custom_components/hyperhdr_control/__init__.py

@@ -3,15 +3,38 @@
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
-from homeassistant.const import Platform
+from homeassistant.const import CONF_HOST, CONF_PORT, Platform
+from homeassistant.helpers import device_registry as dr
+
+from .const import DOMAIN
 
 PLATFORMS: list[Platform] = [Platform.SWITCH, Platform.BUTTON]
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up HyperHDR Control from a config entry."""
+    # Store an instance of the "domain" that includes the host/port
+    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = {
+        "host": entry.data[CONF_HOST],
+        "port": entry.data[CONF_PORT],
+    }
+
+    # Register device
+    device_registry = dr.async_get(hass)
+    device_registry.async_get_or_create(
+        config_entry_id=entry.entry_id,
+        identifiers={(DOMAIN, f"{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}")},
+        manufacturer="HyperHDR",
+        name=f"HyperHDR ({entry.data[CONF_HOST]})",
+        model="HyperHDR LED Controller",
+        sw_version="1.0.0",
+    )
+
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
     return True
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload a config entry."""
-    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) 
+    unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    if unload_ok:
+        hass.data[DOMAIN].pop(entry.entry_id)
+    return unload_ok 

custom_components/hyperhdr_control/button.py

@@ -11,9 +11,10 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, CONF_PORT
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from .const import AVAILABLE_EFFECTS
+from .const import DOMAIN, AVAILABLE_EFFECTS
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -28,21 +29,33 @@ async def async_setup_entry(
 
     entities = []
     for effect in AVAILABLE_EFFECTS:
-        entities.append(HyperHDREffectButton(host, port, effect))
+        entities.append(HyperHDREffectButton(entry.entry_id, host, port, effect))
 
     async_add_entities(entities, True)
 
 class HyperHDREffectButton(ButtonEntity):
     """Representation of a HyperHDR Effect button."""
 
-    def __init__(self, host: str, port: int, effect_name: str) -> None:
+    def __init__(self, entry_id: str, host: str, port: int, effect_name: str) -> None:
         """Initialize the button."""
+        self._entry_id = entry_id
         self._host = host
         self._port = port
         self._effect_name = effect_name
         self._attr_name = f"HyperHDR Effect {effect_name}"
         self._attr_unique_id = f"hyperhdr_effect_{host}_{port}_{effect_name.lower().replace(' ', '_')}"
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return device information about this HyperHDR instance."""
+        return DeviceInfo(
+            identifiers={(DOMAIN, f"{self._host}:{self._port}")},
+            manufacturer="HyperHDR",
+            name=f"HyperHDR ({self._host})",
+            model="HyperHDR LED Controller",
+            sw_version="1.0.0",
+        )
+
     async def async_press(self) -> None:
         """Handle the button press."""
         request_data = {

custom_components/hyperhdr_control/manifest.json

@@ -6,6 +6,6 @@
   "codeowners": [],
   "requirements": ["aiohttp"],
   "iot_class": "local_polling",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "config_flow": true
 } 

custom_components/hyperhdr_control/switch.py

@@ -11,8 +11,11 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, CONF_PORT
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
+from .const import DOMAIN
+
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(
@@ -24,19 +27,31 @@ async def async_setup_entry(
     host = entry.data[CONF_HOST]
     port = entry.data[CONF_PORT]
 
-    async_add_entities([HyperHDRSwitch(host, port)], True)
+    async_add_entities([HyperHDRSwitch(entry.entry_id, host, port)], True)
 
 class HyperHDRSwitch(SwitchEntity):
     """Representation of a HyperHDR Control switch."""
 
-    def __init__(self, host: str, port: int) -> None:
+    def __init__(self, entry_id: str, host: str, port: int) -> None:
         """Initialize the switch."""
+        self._entry_id = entry_id
         self._host = host
         self._port = port
         self._attr_name = "HyperHDR USB Capture"
         self._attr_unique_id = f"hyperhdr_videograbber_{host}_{port}"
         self._attr_is_on = False
 
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return device information about this HyperHDR instance."""
+        return DeviceInfo(
+            identifiers={(DOMAIN, f"{self._host}:{self._port}")},
+            manufacturer="HyperHDR",
+            name=f"HyperHDR ({self._host})",
+            model="HyperHDR LED Controller",
+            sw_version="1.0.0",
+        )
+
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn the entity on."""
         await self._set_state(True)