Preliminary support for host tracking #24

custom_components/mikrotik_router/device_tracker.py

@@ -20,7 +20,7 @@ from .const import (
 
 _LOGGER = logging.getLogger(__name__)
 
-DEVICE_ATTRIBUTES = [
+DEVICE_ATTRIBUTES_IFACE = [
     "running",
     "enabled",
     "comment",
@@ -36,6 +36,15 @@ DEVICE_ATTRIBUTES = [
     "default-name",
 ]
 
+DEVICE_ATTRIBUTES_HOST = [
+    "mac-address",
+    "address",
+    "host-name",
+    "status",
+    "last-seen",
+    "interface",
+]
+
 
 # ---------------------------
 #   format_attribute
@@ -80,18 +89,24 @@ def update_items(inst, mikrotik_controller, async_add_entities, tracked):
     """Update tracked device state from the controller."""
     new_tracked = []
 
-    for uid in mikrotik_controller.data["interface"]:
-        if mikrotik_controller.data["interface"][uid]["type"] == "ether":
-            item_id = f"{inst}-{mikrotik_controller.data['interface'][uid]['default-name']}"
+    # Add switches
+    for sid, sid_uid, sid_func in zip(
+        ["interface", "dhcp"],
+        ["default-name", "mac-address"],
+        [
+            MikrotikControllerPortDeviceTracker,
+            MikrotikControllerHostDeviceTracker,
+        ],
+    ):
+        for uid in mikrotik_controller.data[sid]:
+            item_id = f"{inst}-{sid}-{mikrotik_controller.data[sid][uid][sid_uid]}"
             _LOGGER.debug("Updating device_tracker %s", item_id)
             if item_id in tracked:
                 if tracked[item_id].enabled:
                     tracked[item_id].async_schedule_update_ha_state()
                 continue
 
-            tracked[item_id] = MikrotikControllerPortDeviceTracker(
-                inst, uid, mikrotik_controller
-            )
+            tracked[item_id] = sid_func(inst, uid, mikrotik_controller)
             new_tracked.append(tracked[item_id])
 
     if new_tracked:
@@ -186,7 +201,106 @@ class MikrotikControllerPortDeviceTracker(ScannerEntity):
         """Return the port state attributes."""
         attributes = self._attrs
 
-        for variable in DEVICE_ATTRIBUTES:
+        for variable in DEVICE_ATTRIBUTES_IFACE:
+            if variable in self._data:
+                attributes[format_attribute(variable)] = self._data[variable]
+
+        return attributes
+
+
+# ---------------------------
+#   MikrotikControllerHostDeviceTracker
+# ---------------------------
+class MikrotikControllerHostDeviceTracker(ScannerEntity):
+    """Representation of a network device."""
+
+    def __init__(self, inst, uid, mikrotik_controller):
+        """Set up tracked port."""
+        self._inst = inst
+        self._ctrl = mikrotik_controller
+        self._data = mikrotik_controller.data["dhcp"][uid]
+
+        self._attrs = {
+            ATTR_ATTRIBUTION: ATTRIBUTION,
+        }
+
+    @property
+    def entity_registry_enabled_default(self):
+        """Return if the entity should be enabled when first added to the entity registry."""
+        return True
+
+    async def async_added_to_hass(self):
+        """Host entity created."""
+        _LOGGER.debug(
+            "New host tracker %s (%s - %s)",
+            self._inst,
+            self._data["host-name"],
+            self._data["mac-address"],
+        )
+
+    async def async_update(self):
+        """Synchronize state with controller."""
+
+    @property
+    def is_connected(self):
+        """Return true if the host is connected to the network."""
+        return self._data["available"]
+
+    @property
+    def source_type(self):
+        """Return the source type of the host."""
+        return SOURCE_TYPE_ROUTER
+
+    @property
+    def name(self):
+        """Return the name of the host."""
+        return f"{self._inst} {self._data['host-name']}"
+
+    @property
+    def unique_id(self):
+        """Return a unique identifier for this host."""
+        return f"{self._inst.lower()}-{self._data['mac-address']}"
+
+    @property
+    def available(self) -> bool:
+        """Return if controller is available."""
+        return self._ctrl.connected()
+
+    @property
+    def icon(self):
+        """Return the icon."""
+        if self._data["available"]:
+            icon = "mdi:lan-connect"
+        else:
+            icon = "mdi:lan-disconnect"
+
+        return icon
+
+    @property
+    def device_info(self):
+        """Return a host description for device registry."""
+        info = {
+            "identifiers": {
+                (
+                    DOMAIN,
+                    "serial-number",
+                    self._ctrl.data["routerboard"]["serial-number"],
+                    "switch",
+                    "Hosts",
+                )
+            },
+            "manufacturer": self._ctrl.data["resource"]["platform"],
+            "model": self._ctrl.data["resource"]["board-name"],
+            "name": "Hosts",
+        }
+        return info
+
+    @property
+    def device_state_attributes(self):
+        """Return the host state attributes."""
+        attributes = self._attrs
+
+        for variable in DEVICE_ATTRIBUTES_HOST:
             if variable in self._data:
                 attributes[format_attribute(variable)] = self._data[variable]