From 761378436749986f806ec1547a9a90c3fd380413 Mon Sep 17 00:00:00 2001
From: Mick Vleeshouwer <mick@imick.nl>
Date: Wed, 30 Mar 2022 01:07:34 -0700
Subject: [PATCH] Extend Overkiz diagnostics and implement device diagnostics
 (#68859)

---
 .../components/overkiz/diagnostics.py         | 50 ++++++++++++++++---
 1 file changed, 44 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/overkiz/diagnostics.py b/homeassistant/components/overkiz/diagnostics.py
index c5596a9cf3d..77ca0227579 100644
--- a/homeassistant/components/overkiz/diagnostics.py
+++ b/homeassistant/components/overkiz/diagnostics.py
@@ -1,21 +1,59 @@
 """Provides diagnostics for Overkiz."""
 from __future__ import annotations
 
-from typing import Any, cast
+from typing import Any
+
+from pyoverkiz.obfuscate import obfuscate_id
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry
 
 from . import HomeAssistantOverkizData
-from .const import DOMAIN
+from .const import CONF_HUB, DOMAIN
 
 
 async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
 ) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
-    data: HomeAssistantOverkizData = hass.data[DOMAIN][entry.entry_id]
-    client = data.coordinator.client
-    setup = await client.get_diagnostic_data()
+    entry_data: HomeAssistantOverkizData = hass.data[DOMAIN][entry.entry_id]
+    client = entry_data.coordinator.client
 
-    return cast(dict, setup)
+    data = {
+        "setup": await client.get_diagnostic_data(),
+        "server": entry.data[CONF_HUB],
+        "execution_history": [
+            repr(execution) for execution in await client.get_execution_history()
+        ],
+    }
+
+    return data
+
+
+async def async_get_device_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a device entry."""
+    entry_data: HomeAssistantOverkizData = hass.data[DOMAIN][entry.entry_id]
+    client = entry_data.coordinator.client
+
+    device_url = min(device.identifiers)[1]
+
+    data = {
+        "device": {
+            "controllable_name": device.hw_version,
+            "firmware": device.sw_version,
+            "device_url": obfuscate_id(device_url),
+            "model": device.model,
+        },
+        "setup": await client.get_diagnostic_data(),
+        "server": entry.data[CONF_HUB],
+        "execution_history": [
+            repr(execution)
+            for execution in await client.get_execution_history()
+            if any(command.device_url == device_url for command in execution.commands)
+        ],
+    }
+
+    return data