From 2bcc21ecbb7cd0208101f7f5f99b644e2563d4b6 Mon Sep 17 00:00:00 2001
From: Maikel Punie <maikel.punie@gmail.com>
Date: Mon, 14 Feb 2022 19:25:14 +0100
Subject: [PATCH] Add velbus diagnostics (#65426)

---
 .coveragerc                                   |  1 +
 .../components/velbus/diagnostics.py          | 57 +++++++++++++++++++
 2 files changed, 58 insertions(+)
 create mode 100644 homeassistant/components/velbus/diagnostics.py

diff --git a/.coveragerc b/.coveragerc
index 5419a8247ad..b59986fbe64 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -1339,6 +1339,7 @@ omit =
     homeassistant/components/velbus/climate.py
     homeassistant/components/velbus/const.py
     homeassistant/components/velbus/cover.py
+    homeassistant/components/velbus/diagnostics.py
     homeassistant/components/velbus/light.py
     homeassistant/components/velbus/sensor.py
     homeassistant/components/velbus/switch.py
diff --git a/homeassistant/components/velbus/diagnostics.py b/homeassistant/components/velbus/diagnostics.py
new file mode 100644
index 00000000000..f6015abd1f8
--- /dev/null
+++ b/homeassistant/components/velbus/diagnostics.py
@@ -0,0 +1,57 @@
+"""Diagnostics support for Velbus."""
+from __future__ import annotations
+
+from typing import Any
+
+from velbusaio.channels import Channel as VelbusChannel
+from velbusaio.module import Module as VelbusModule
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry
+
+from .const import DOMAIN
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+    controller = hass.data[DOMAIN][entry.entry_id]["cntrl"]
+    data: dict[str, Any] = {"entry": entry.as_dict(), "modules": []}
+    for module in controller.get_modules().values():
+        data["modules"].append(_build_module_diagnostics_info(module))
+    return data
+
+
+async def async_get_device_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a device entry."""
+    controller = hass.data[DOMAIN][entry.entry_id]["cntrl"]
+    channel = list(next(iter(device.identifiers)))[1]
+    modules = controller.get_modules()
+    return _build_module_diagnostics_info(modules[int(channel)])
+
+
+def _build_module_diagnostics_info(module: VelbusModule) -> dict[str, Any]:
+    """Build per module diagnostics info."""
+    data: dict[str, Any] = {
+        "type": module.get_type_name(),
+        "address": module.get_addresses(),
+        "name": module.get_name(),
+        "sw_version": module.get_sw_version(),
+        "is_loaded": module.is_loaded(),
+        "channels": _build_channels_diagnostics_info(module.get_channels()),
+    }
+    return data
+
+
+def _build_channels_diagnostics_info(
+    channels: dict[str, VelbusChannel]
+) -> dict[str, Any]:
+    """Build diagnostics info for all channels."""
+    data: dict[str, Any] = {}
+    for channel in channels.values():
+        data[str(channel.get_channel_number())] = channel.get_channel_info()
+    return data