Obfuscated diagnostics; fixed IP auto-update; cleaned logs

custom_components/localtuya/diagnostics.py

@@ -3,11 +3,26 @@ from __future__ import annotations
 from typing import Any
 
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_DEVICES
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntry
 
-from .const import DOMAIN, DATA_CLOUD
+
+from homeassistant.const import (
+    CONF_CLIENT_ID,
+    CONF_CLIENT_SECRET,
+    CONF_DEVICES,
+)
+
+from .const import (
+    DOMAIN,
+    DATA_CLOUD,
+    CONF_USER_ID,
+    CONF_LOCAL_KEY,
+)
+
+CLOUD_DEVICES = "cloud_devices"
+DEVICE_CONFIG = "device_config"
+DEVICE_CLOUD_INFO = "device_cloud_info"
 
 
 async def async_get_config_entry_diagnostics(
@@ -15,14 +30,20 @@ async def async_get_config_entry_diagnostics(
 ) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
     data = {}
-    data = {**entry.data}
-    # print("DATA is {}".format(data))
+    data = entry.data.copy()
     tuya_api = hass.data[DOMAIN][DATA_CLOUD]
-    data["cloud_devices"] = tuya_api._device_list
-
     # censoring private information
-    # data["token"] = re.sub(r"[^\-]", "*", data["token"])
-    # data["userId"] = re.sub(r"[^\-]", "*", data["userId"])
+    for field in [CONF_CLIENT_ID, CONF_CLIENT_SECRET, CONF_USER_ID]:
+        data[field] = f"{data[field][0:3]}...{data[field][-3:]}"
+    for dev_id, dev in data[CONF_DEVICES].items():
+        local_key = data[CONF_DEVICES][dev_id][CONF_LOCAL_KEY]
+        local_key_obfuscated = f"{local_key[0:3]}...{local_key[-3:]}"
+        data[CONF_DEVICES][dev_id][CONF_LOCAL_KEY] = local_key_obfuscated
+    data[CLOUD_DEVICES] = tuya_api._device_list
+    for dev_id, dev in data[CLOUD_DEVICES].items():
+        local_key = data[CLOUD_DEVICES][dev_id][CONF_LOCAL_KEY]
+        local_key_obfuscated = f"{local_key[0:3]}...{local_key[-3:]}"
+        data[CLOUD_DEVICES][dev_id][CONF_LOCAL_KEY] = local_key_obfuscated
     return data
 
 
@@ -32,9 +53,14 @@ async def async_get_device_diagnostics(
     """Return diagnostics for a device entry."""
     data = {}
     dev_id = list(device.identifiers)[0][1].split("_")[-1]
-    data["device_config"] = entry.data[CONF_DEVICES][dev_id]
+    data[DEVICE_CONFIG] = entry.data[CONF_DEVICES][dev_id]
+    local_key = data[DEVICE_CONFIG][CONF_LOCAL_KEY]
+    data[DEVICE_CONFIG][CONF_LOCAL_KEY] = f"{local_key[0:3]}...{local_key[-3:]}"
+
     tuya_api = hass.data[DOMAIN][DATA_CLOUD]
     if dev_id in tuya_api._device_list:
-        data["device_cloud_info"] = tuya_api._device_list[dev_id]
+        data[DEVICE_CLOUD_INFO] = tuya_api._device_list[dev_id]
+        local_key = data[DEVICE_CLOUD_INFO][CONF_LOCAL_KEY]
+        data[DEVICE_CLOUD_INFO][CONF_LOCAL_KEY] = f"{local_key[0:3]}...{local_key[-3:]}"
     # data["log"] = hass.data[DOMAIN][CONF_DEVICES][dev_id].logger.retrieve_log()
     return data