"""Util functions for the dwd_weather_warnings integration."""

from __future__ import annotations

from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from .exceptions import EntityNotFoundError


def get_position_data(
    hass: HomeAssistant, registry_id: str
) -> tuple[float, float] | None:
    """Extract longitude and latitude from a device tracker."""
    registry = er.async_get(hass)
    registry_entry = registry.async_get(registry_id)
    if registry_entry is None:
        raise EntityNotFoundError(f"Failed to find registry entry {registry_id}")

    entity = hass.states.get(registry_entry.entity_id)
    if entity is None:
        raise EntityNotFoundError(f"Failed to find entity {registry_entry.entity_id}")

    latitude = entity.attributes.get(ATTR_LATITUDE)
    if not latitude:
        raise AttributeError(
            f"Failed to find attribute '{ATTR_LATITUDE}' in {registry_entry.entity_id}",
            ATTR_LATITUDE,
        )

    longitude = entity.attributes.get(ATTR_LONGITUDE)
    if not longitude:
        raise AttributeError(
            f"Failed to find attribute '{ATTR_LONGITUDE}' in {registry_entry.entity_id}",
            ATTR_LONGITUDE,
        )

    return (latitude, longitude)