From b22cb29811f269bc7b6cea94e1a8ef23f0f2e53b Mon Sep 17 00:00:00 2001
From: rospogrigio <rospogrigio@gmail.com>
Date: Fri, 13 May 2022 09:57:58 +0200
Subject: [PATCH] Introduced diagnostics

---
 custom_components/localtuya/diagnostics.py | 35 ++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 custom_components/localtuya/diagnostics.py

diff --git a/custom_components/localtuya/diagnostics.py b/custom_components/localtuya/diagnostics.py
new file mode 100644
index 0000000..a18d5d6
--- /dev/null
+++ b/custom_components/localtuya/diagnostics.py
@@ -0,0 +1,35 @@
+"""Diagnostics support for LocalTuya."""
+from __future__ import annotations
+from typing import Any
+
+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."""
+    data = {}
+    data = {**entry.data}
+    # print("DATA is {}".format(data))
+
+    # censoring private information
+    # data["token"] = re.sub(r"[^\-]", "*", data["token"])
+    # data["userId"] = re.sub(r"[^\-]", "*", data["userId"])
+    return data
+
+
+async def async_get_device_diagnostics(
+    hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a device entry."""
+    # dev_id = next(iter(device.identifiers))[1]
+
+    data = {}
+    # data["device"] = device
+    # data["log"] = hass.data[DOMAIN][AIRBNK_DEVICES][dev_id].logger.retrieve_log()
+    return data