"""Diagnostics support for devolo Home Control."""

from __future__ import annotations

from typing import Any

from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant

from . import DevoloHomeControlConfigEntry

TO_REDACT = {CONF_PASSWORD, CONF_USERNAME}


async def async_get_config_entry_diagnostics(
    hass: HomeAssistant, entry: DevoloHomeControlConfigEntry
) -> dict[str, Any]:
    """Return diagnostics for a config entry."""
    device_info = [
        {
            "gateway": {
                "local_connection": gateway.gateway.local_connection,
                "firmware_version": gateway.gateway.firmware_version,
            },
            "devices": [
                {
                    "device_id": device_id,
                    "device_model_uid": properties.device_model_uid,
                    "device_type": properties.device_type,
                    "name": properties.name,
                }
                for device_id, properties in gateway.devices.items()
            ],
        }
        for gateway in entry.runtime_data
    ]

    return {
        "entry": async_redact_data(entry.as_dict(), TO_REDACT),
        "device_info": device_info,
    }