"""
BAN YARO — Wetter via Open-Meteo
- get_weather_summary(): Push-Job, prüft 5 deutsche Städte
- get_weather_for_location(): API-Endpoint, beliebiger Standort mit TTL-Cache
"""

import time
import logging
import httpx
from datetime import datetime

logger = logging.getLogger(__name__)

# WMO-Wettercodes → (Beschreibung, Phosphor-Icon-Name)
_WMO = {
    0:  ('Klar',                  'sun'),
    1:  ('Überwiegend klar',      'cloud-sun'),
    2:  ('Teilweise bewölkt',     'cloud-sun'),
    3:  ('Bedeckt',               'cloud'),
    45: ('Nebel',                 'cloud-fog'),
    48: ('Gefrierender Nebel',    'cloud-fog'),
    51: ('Leichter Nieselregen',  'cloud-rain'),
    53: ('Nieselregen',           'cloud-rain'),
    55: ('Starker Nieselregen',   'cloud-rain'),
    61: ('Leichter Regen',        'cloud-rain'),
    63: ('Regen',                 'cloud-rain'),
    65: ('Starker Regen',         'cloud-rain'),
    71: ('Leichter Schnee',       'snowflake'),
    73: ('Schnee',                'snowflake'),
    75: ('Starker Schnee',        'snowflake'),
    77: ('Schneekörner',          'snowflake'),
    80: ('Leichte Schauer',       'cloud-rain'),
    81: ('Schauer',               'cloud-rain'),
    82: ('Starke Schauer',        'cloud-rain'),
    85: ('Schneeschauer',         'snowflake'),
    86: ('Starke Schneeschauer',  'snowflake'),
    95: ('Gewitter',              'cloud-lightning'),
    96: ('Gewitter mit Hagel',    'cloud-lightning'),
    99: ('Schweres Gewitter',     'cloud-lightning'),
}

# TTL-Cache: (round(lat,1), round(lon,1)) → (timestamp, data)
_location_cache: dict = {}
_CACHE_TTL = 1800  # 30 Minuten


async def get_weather_for_location(lat: float, lon: float) -> dict:
    """Holt aktuelles Wetter für einen Standort. 30-min TTL-Cache."""
    key = (round(lat, 1), round(lon, 1))
    now = time.time()
    if key in _location_cache:
        ts, cached = _location_cache[key]
        if now - ts < _CACHE_TTL:
            return cached

    url = (
        "https://api.open-meteo.com/v1/forecast"
        f"?latitude={lat}&longitude={lon}"
        "&current=temperature_2m,apparent_temperature,weathercode,windspeed_10m,is_day"
        "&daily=precipitation_probability_max,uv_index_max"
        "&timezone=Europe%2FBerlin&forecast_days=1"
    )
    async with httpx.AsyncClient(timeout=8.0) as client:
        resp = await client.get(url)
        resp.raise_for_status()
        raw = resp.json()

    cur   = raw.get('current', {})
    daily = raw.get('daily', {})

    temp       = cur.get('temperature_2m')
    feels_like = cur.get('apparent_temperature')
    wcode      = cur.get('weathercode', 0)
    wind       = cur.get('windspeed_10m')
    is_day     = cur.get('is_day', 1)
    precip     = (daily.get('precipitation_probability_max') or [None])[0]
    uv         = (daily.get('uv_index_max') or [None])[0]

    desc, icon = _WMO.get(wcode, ('Unbekannt', 'cloud'))
    if wcode == 0 and not is_day:
        icon = 'moon'

    month = datetime.now().month
    zecken = None
    if temp is not None and temp > 7.0 and 3 <= month <= 10:
        zecken = 'hoch' if temp > 20 else ('mittel' if temp > 12 else 'niedrig')

    data = {
        'temp_c':        temp,
        'feels_like_c':  feels_like,
        'weathercode':   wcode,
        'desc':          desc,
        'icon':          icon,
        'wind_kmh':      wind,
        'precip_prob':   precip,
        'uv_index':      uv,
        'is_day':        bool(is_day),
        'zecken_warnung': zecken,
    }
    _location_cache[key] = (now, data)
    return data

# Wichtige deutsche Städte als Stichprobe
CITIES = [
    {"name": "Berlin",    "lat": 52.52, "lon": 13.41},
    {"name": "München",   "lat": 48.14, "lon": 11.58},
    {"name": "Hamburg",   "lat": 53.55, "lon": 10.00},
    {"name": "Frankfurt", "lat": 50.11, "lon":  8.68},
    {"name": "Köln",      "lat": 50.94, "lon":  6.96},
]

# WMO-Wettercodes 95–99 = Gewitter
THUNDERSTORM_CODES = {95, 96, 99}


async def get_weather_summary() -> dict:
    """
    Holt Tagesprognose für mehrere deutsche Städte von Open-Meteo.
    Gibt zurück: {"max_temp_c": float, "thunderstorm": bool}
    """
    max_temp = -999.0
    thunderstorm = False

    async with httpx.AsyncClient(timeout=10.0) as client:
        for city in CITIES:
            url = (
                "https://api.open-meteo.com/v1/forecast"
                f"?latitude={city['lat']}&longitude={city['lon']}"
                "&daily=temperature_2m_max,precipitation_probability_max,weathercode"
                "&timezone=Europe%2FBerlin&forecast_days=1"
            )
            try:
                resp = await client.get(url)
                resp.raise_for_status()
                data = resp.json()

                daily = data.get("daily", {})
                temps = daily.get("temperature_2m_max", [None])
                precip_probs = daily.get("precipitation_probability_max", [None])
                weathercodes = daily.get("weathercode", [None])

                temp = temps[0] if temps else None
                precip_prob = precip_probs[0] if precip_probs else None
                weathercode = weathercodes[0] if weathercodes else None

                if temp is not None and temp > max_temp:
                    max_temp = temp

                if (
                    precip_prob is not None and precip_prob > 50
                    and weathercode is not None and int(weathercode) in THUNDERSTORM_CODES
                ):
                    thunderstorm = True
                    logger.info(f"Gewitter erkannt: {city['name']} (Code {weathercode}, {precip_prob}% Niederschlag)")

            except Exception as e:
                logger.warning(f"Wetter-Abruf für {city['name']} fehlgeschlagen: {e}")

    if max_temp == -999.0:
        max_temp = 0.0

    logger.info(f"Wetter-Zusammenfassung: max_temp={max_temp}°C, thunderstorm={thunderstorm}")
    return {"max_temp_c": max_temp, "thunderstorm": thunderstorm}


# ---------------------------------------------------------------------------
# 7-Tage-Vorhersage
# ---------------------------------------------------------------------------
import asyncio  # noqa: E402 — appended section

_forecast_cache: dict = {}
_FORECAST_TTL = 3600  # 1 Stunde

_WDAY_DE = ['Mo', 'Di', 'Mi', 'Do', 'Fr', 'Sa', 'So']


def _wind_dir(deg: float) -> str:
    dirs = ['N', 'NO', 'O', 'SO', 'S', 'SW', 'W', 'NW']
    idx = round(deg / 45) % 8
    return dirs[idx]


def _asphalt_temp(air_max: float, uv_max: float) -> tuple[float, str]:
    bonus = min(uv_max * 3.0, 30.0)
    asphalt = air_max + bonus
    if asphalt <= 30:
        warn = 'safe'
    elif asphalt <= 40:
        warn = 'warm'
    elif asphalt <= 55:
        warn = 'hot'
    else:
        warn = 'danger'
    return round(asphalt, 1), warn


def _pollen_lvl(val: float | None) -> dict:
    if val is None:
        return {'level': 0, 'label': 'keine'}
    if val < 5:
        return {'level': 1, 'label': 'niedrig'}
    if val < 25:
        return {'level': 2, 'label': 'mittel'}
    if val < 100:
        return {'level': 3, 'label': 'hoch'}
    return {'level': 4, 'label': 'sehr hoch'}


async def get_forecast(lat: float, lon: float) -> dict:
    """7-Tage-Wettervorhersage inkl. Pollen, Asphalttemperatur, Zecken. 1h TTL-Cache."""
    key = (round(lat, 2), round(lon, 2))
    now = time.time()
    if key in _forecast_cache:
        ts, cached = _forecast_cache[key]
        if now - ts < _FORECAST_TTL:
            return cached

    forecast_url = (
        "https://api.open-meteo.com/v1/forecast"
        f"?latitude={lat}&longitude={lon}"
        "&daily=temperature_2m_max,temperature_2m_min,apparent_temperature_max,"
        "apparent_temperature_min,precipitation_probability_max,precipitation_sum,"
        "weathercode,windspeed_10m_max,winddirection_10m_dominant,uv_index_max,"
        "sunrise,sunset"
        "&timezone=auto&forecast_days=7"
    )
    pollen_url = (
        "https://air-quality-api.open-meteo.com/v1/air-quality"
        f"?latitude={lat}&longitude={lon}"
        "&hourly=alder_pollen,birch_pollen,grass_pollen,mugwort_pollen,ragweed_pollen"
        "&timezone=auto&forecast_days=7"
    )

    async with httpx.AsyncClient(timeout=10.0) as client:
        forecast_task = client.get(forecast_url)
        pollen_task   = client.get(pollen_url)
        forecast_resp, pollen_resp = await asyncio.gather(
            forecast_task, pollen_task, return_exceptions=True
        )

    # --- Forecast (required) ---
    if isinstance(forecast_resp, Exception):
        raise forecast_resp
    forecast_resp.raise_for_status()
    raw = forecast_resp.json()

    daily    = raw.get('daily', {})
    timezone = raw.get('timezone', 'auto')

    dates       = daily.get('time', [])
    temp_max    = daily.get('temperature_2m_max', [])
    temp_min    = daily.get('temperature_2m_min', [])
    feels_max   = daily.get('apparent_temperature_max', [])
    feels_min   = daily.get('apparent_temperature_min', [])
    precip_prob = daily.get('precipitation_probability_max', [])
    precip_sum  = daily.get('precipitation_sum', [])
    wcodes      = daily.get('weathercode', [])
    wind_kmh    = daily.get('windspeed_10m_max', [])
    wind_deg    = daily.get('winddirection_10m_dominant', [])
    uv_index    = daily.get('uv_index_max', [])
    sunrises    = daily.get('sunrise', [])
    sunsets     = daily.get('sunset', [])

    # --- Pollen (optional) ---
    pollen_daily: dict | None = None
    if not isinstance(pollen_resp, Exception):
        try:
            pollen_resp.raise_for_status()
            praw    = pollen_resp.json()
            hourly  = praw.get('hourly', {})
            htimes  = hourly.get('time', [])
            # aggregate hourly → daily max per type
            pollen_types = {
                'erle':    hourly.get('alder_pollen', []),
                'birke':   hourly.get('birch_pollen', []),
                'graeser': hourly.get('grass_pollen', []),
                'beifuss': hourly.get('mugwort_pollen', []),
                'ambrosia': hourly.get('ragweed_pollen', []),
            }
            # build date → max mapping per type
            pollen_daily = {ptype: {} for ptype in pollen_types}
            for i, ts_str in enumerate(htimes):
                day_str = ts_str[:10]  # 'YYYY-MM-DD'
                for ptype, vals in pollen_types.items():
                    v = vals[i] if i < len(vals) else None
                    if v is not None:
                        prev = pollen_daily[ptype].get(day_str)
                        pollen_daily[ptype][day_str] = max(prev, v) if prev is not None else v
        except Exception as e:
            logger.warning(f"Pollen-Abruf fehlgeschlagen: {e}")
            pollen_daily = None

    # --- Assemble days ---
    days = []
    for i, date_str in enumerate(dates):
        wcode     = int(wcodes[i]) if i < len(wcodes) and wcodes[i] is not None else 0
        desc, icon = _WMO.get(wcode, ('Unbekannt', 'cloud'))

        t_max  = temp_max[i]  if i < len(temp_max)  else None
        t_min  = temp_min[i]  if i < len(temp_min)  else None
        f_max  = feels_max[i] if i < len(feels_max) else None
        f_min  = feels_min[i] if i < len(feels_min) else None
        pp     = precip_prob[i] if i < len(precip_prob) else None
        ps     = precip_sum[i]  if i < len(precip_sum)  else None
        wk     = wind_kmh[i]   if i < len(wind_kmh)   else None
        wd_deg = wind_deg[i]   if i < len(wind_deg)   else None
        uv     = uv_index[i]   if i < len(uv_index)   else None

        # Sunrise / Sunset → HH:MM only (format: "2025-05-02T06:12")
        sunrise_raw = sunrises[i] if i < len(sunrises) else None
        sunset_raw  = sunsets[i]  if i < len(sunsets)  else None
        sunrise_hm  = sunrise_raw[11:16] if sunrise_raw and len(sunrise_raw) >= 16 else sunrise_raw
        sunset_hm   = sunset_raw[11:16]  if sunset_raw  and len(sunset_raw)  >= 16 else sunset_raw

        # Weekday
        try:
            dt_obj = datetime.strptime(date_str, '%Y-%m-%d')
            wday   = _WDAY_DE[dt_obj.weekday()]
        except Exception:
            wday = ''

        # Asphalt
        asphalt_t, asphalt_w = _asphalt_temp(t_max or 0.0, uv or 0.0)

        # Zecken
        month  = datetime.strptime(date_str, '%Y-%m-%d').month
        zecken = None
        if t_max is not None and t_max > 7.0 and 3 <= month <= 10:
            zecken = 'hoch' if t_max > 20 else ('mittel' if t_max > 12 else 'niedrig')

        # Pollen
        if pollen_daily is not None:
            pollen_out = {
                pt: _pollen_lvl(pollen_daily[pt].get(date_str))
                for pt in ('erle', 'birke', 'graeser', 'beifuss', 'ambrosia')
            }
        else:
            pollen_out = None

        days.append({
            'date':         date_str,
            'wday':         wday,
            'weathercode':  wcode,
            'desc':         desc,
            'icon':         icon,
            'temp_max':     t_max,
            'temp_min':     t_min,
            'feels_max':    f_max,
            'feels_min':    f_min,
            'precip_prob':  pp,
            'precip_sum':   ps,
            'wind_kmh':     wk,
            'wind_dir':     _wind_dir(wd_deg) if wd_deg is not None else None,
            'wind_dir_deg': wd_deg,
            'uv_index':     uv,
            'sunrise':      sunrise_hm,
            'sunset':       sunset_hm,
            'asphalt_temp': asphalt_t,
            'asphalt_warn': asphalt_w,
            'pollen':       pollen_out,
            'zecken':       zecken,
            'thunderstorm': wcode in {95, 96, 99},
            'paw_cold':     wcode in {71, 73, 75, 77, 85, 86} or (t_min is not None and t_min < 0),
        })

    result = {'timezone': timezone, 'days': days}
    _forecast_cache[key] = (now, result)
    _log_forecast(round(lat, 1), round(lon, 1), days)
    return result


def _log_forecast(lat_r: float, lon_r: float, days: list) -> None:
    """Speichert jeden Forecast-Tag in weather_log (INSERT OR IGNORE — kein Überschreiben)."""
    if not days:
        return
    try:
        import json
        from database import db
        with db() as conn:
            for d in days:
                pollen = d.get('pollen') or {}
                conn.execute("""
                    INSERT OR IGNORE INTO weather_log
                        (date, lat_r, lon_r,
                         temp_max, temp_min, feels_max,
                         precip_prob, precip_sum,
                         wind_kmh, wind_dir, uv_index,
                         weathercode, weatherdesc,
                         sunrise, sunset,
                         asphalt_temp, asphalt_warn, zecken,
                         pollen_erle, pollen_birke, pollen_graeser,
                         pollen_beifuss, pollen_ambrosia,
                         forecast_json)
                    VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
                """, (
                    d['date'], lat_r, lon_r,
                    d.get('temp_max'), d.get('temp_min'), d.get('feels_max'),
                    d.get('precip_prob'), d.get('precip_sum'),
                    d.get('wind_kmh'), d.get('wind_dir'), d.get('uv_index'),
                    d.get('weathercode'), d.get('desc'),
                    d.get('sunrise'), d.get('sunset'),
                    d.get('asphalt_temp'), d.get('asphalt_warn'), d.get('zecken'),
                    pollen.get('erle',    {}).get('level'),
                    pollen.get('birke',   {}).get('level'),
                    pollen.get('graeser', {}).get('level'),
                    pollen.get('beifuss', {}).get('level'),
                    pollen.get('ambrosia',{}).get('level'),
                    json.dumps(d, ensure_ascii=False),
                ))
    except Exception as e:
        logger.warning(f"weather_log insert fehlgeschlagen: {e}")