"""Diagnostics support for SmartThings."""

from __future__ import annotations

import asyncio
from dataclasses import asdict
from typing import Any

from pysmartthings import DeviceEvent

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

from . import SmartThingsConfigEntry
from .const import DOMAIN

EVENT_WAIT_TIME = 5


async def async_get_config_entry_diagnostics(
    hass: HomeAssistant,
    entry: SmartThingsConfigEntry,
) -> dict[str, Any]:
    """Return diagnostics for a config entry."""
    client = entry.runtime_data.client
    return {"devices": await client.get_raw_devices()}


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

    device_status = await client.get_raw_device_status(device_id)
    device_info = await client.get_raw_device(device_id)

    events: list[DeviceEvent] = []

    def register_event(event: DeviceEvent) -> None:
        events.append(event)

    listener = client.add_device_event_listener(device_id, register_event)

    await asyncio.sleep(EVENT_WAIT_TIME)

    listener()

    return {
        "events": [asdict(event) for event in events],
        "status": device_status,
        "info": device_info,
    }