From 332b1a6657e2eb5646a21cd9561a4d08a0e93aa6 Mon Sep 17 00:00:00 2001
From: Tomaae <23486452+tomaae@users.noreply.github.com>
Date: Thu, 1 May 2025 13:19:52 +0200
Subject: [PATCH] Added wireless host signal information, fixes #413

---
 custom_components/mikrotik_router/coordinator.py    | 13 ++++++++++++-
 .../mikrotik_router/device_tracker_types.py         |  4 ++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/custom_components/mikrotik_router/coordinator.py b/custom_components/mikrotik_router/coordinator.py
index 377a49a..4460efb 100644
--- a/custom_components/mikrotik_router/coordinator.py
+++ b/custom_components/mikrotik_router/coordinator.py
@@ -2056,6 +2056,10 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
                 {"name": "interface", "default": "unknown"},
                 {"name": "ap", "type": "bool"},
                 {"name": "uptime"},
+                {"name": "signal-strength"},
+                {"name": "tx-ccq"},
+                {"name": "tx-rate"},
+                {"name": "rx-rate"},
             ],
         )
 
@@ -2094,7 +2098,14 @@ class MikrotikCoordinator(DataUpdateCoordinator[None]):
                 wireless_detected[uid] = True
                 self.ds["host"][uid]["available"] = True
                 self.ds["host"][uid]["last-seen"] = utcnow()
-                for key in ["mac-address", "interface"]:
+                for key in [
+                    "mac-address",
+                    "interface",
+                    "signal-strength",
+                    "tx-ccq",
+                    "tx-rate",
+                    "rx-rate",
+                ]:
                     self.ds["host"][uid][key] = vals[key]
 
         # Add hosts from DHCP
diff --git a/custom_components/mikrotik_router/device_tracker_types.py b/custom_components/mikrotik_router/device_tracker_types.py
index d8b1ec3..55c6bac 100644
--- a/custom_components/mikrotik_router/device_tracker_types.py
+++ b/custom_components/mikrotik_router/device_tracker_types.py
@@ -16,6 +16,10 @@ DEVICE_ATTRIBUTES_HOST = [
     "authorized",
     "bypassed",
     "last-seen",
+    "signal-strength",
+    "tx-ccq",
+    "tx-rate",
+    "rx-rate",
 ]