"""Provides diagnostics for EcoWitt."""

from __future__ import annotations

from typing import Any

from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceEntry

from . import EcowittConfigEntry
from .const import DOMAIN


async def async_get_device_diagnostics(
    hass: HomeAssistant, entry: EcowittConfigEntry, device: DeviceEntry
) -> dict[str, Any]:
    """Return diagnostics for a device entry."""
    ecowitt = entry.runtime_data
    station_id = next(item[1] for item in device.identifiers if item[0] == DOMAIN)

    station = ecowitt.stations[station_id]

    return {
        "device": {
            "name": station.station,
            "model": station.model,
            "frequency": station.frequence,  # codespell:ignore frequence
            "version": station.version,
        },
        "raw": ecowitt.last_values[station_id],
        "sensors": {
            sensor.key: sensor.value
            for sensor in ecowitt.sensors.values()
            if sensor.station.key == station_id
        },
    }