"""여행 일정(1박2일·2박3일) 생성 — 순수 함수 모듈. DB·HTTP 없음.

빌드(payload) 시점에 그 지역의 발행된 지역정보(관광지·맛집·축제)와 업체 좌표로
일정을 **즉석 계산**한다. 저장하지 않는 이유: 재료(local_contents)가 갱신되면
다음 빌드에서 일정도 저절로 최신이 된다 — 따로 저장하면 그 동기화를 또 만들어야 한다.
(docs/superpowers/specs/2026-09-03-local-tourapi-sync-design.md)

★ 지어내지 않는 규칙은 여기도 적용된다.
  재료가 부족하면 채울 수 있는 만큼만 담고, 하루도 못 채우면 일정 자체를 내지 않는다.
  좌표 없는 항목은 거리를 잴 수 없으므로 후보에서 뺀다 — 동선을 보장 못 하는 추천은 틀린 추천이다.

하루의 뼈대: 관광지 2 + 맛집 2(점심·저녁). 진행 중 축제가 있으면 그날 관광지 한 자리를 대신한다.
일자 배분은 업체에서 가까운 순으로 후보를 끊고, 일자 안에서는 최근접 이웃 순서로 동선을 만든다.
"""
from typing import Optional

from common.utils.geo import haversine_km

# 하루 구성 정원. 관광지 자리는 축제가 하나 대신할 수 있다.
_SPOTS_PER_DAY = 2
_MEALS_PER_DAY = 2
# 후보 반경(km). 업체에서 이보다 먼 곳은 '근처'가 아니다 — 1박2일 생활권을 넘는다.
_MAX_RADIUS_KM = 30.0

_STOP_ATTRACTION = "attraction"
_STOP_RESTAURANT = "restaurant"
_STOP_FESTIVAL = "festival"


def _as_float(value) -> Optional[float]:
    try:
        return float(value)
    except (TypeError, ValueError):
        return None


def _candidates(rows: list[dict], stop_type: str,
                base_lat: float, base_lng: float) -> list[dict]:
    """payload 지역정보 행 → 거리 오름차순 후보. 좌표가 없거나 반경 밖이면 뺀다.

    행 모양은 스냅샷 local.contents 의 body(services/external/tour_api._normalize)다 —
    mapx=경도, mapy=위도 (WGS84 문자열).
    """
    out = []
    for row in rows:
        body = row.get("body") or {}
        name = str(row.get("title") or body.get("title") or "").strip()
        lat, lng = _as_float(body.get("mapy")), _as_float(body.get("mapx"))
        if not name or lat is None or lng is None:
            continue
        dist = haversine_km(base_lat, base_lng, lat, lng)
        if dist > _MAX_RADIUS_KM:
            continue
        out.append({"type": stop_type, "name": name, "lat": lat, "lng": lng,
                    "distanceKm": round(dist, 1)})
    out.sort(key=lambda c: c["distanceKm"])
    return out


def _order_by_route(stops: list[dict], base_lat: float, base_lng: float) -> list[dict]:
    """일자 안 동선: 업체에서 출발해 최근접 이웃 순으로 잇는다."""
    remaining = list(stops)
    ordered: list[dict] = []
    lat, lng = base_lat, base_lng
    while remaining:
        nxt = min(remaining, key=lambda s: haversine_km(lat, lng, s["lat"], s["lng"]))
        remaining.remove(nxt)
        ordered.append(nxt)
        lat, lng = nxt["lat"], nxt["lng"]
    return ordered


def _take(pool: list[dict], count: int) -> list[dict]:
    taken, pool[:] = pool[:count], pool[count:]
    return taken


def _plan_days(days: int, attractions: list[dict], restaurants: list[dict],
               festivals: list[dict], base_lat: float, base_lng: float) -> Optional[dict]:
    """일자별 계획. 첫날 하루도 못 채우면 None — 반쪽짜리 일정은 내지 않는다."""
    spots = list(attractions)
    meals = list(restaurants)
    fests = list(festivals)
    plan = []
    for day in range(1, days + 1):
        day_stops = []
        # 축제는 하루 하나까지, 관광지 한 자리를 대신한다.
        fest = _take(fests, 1)
        day_stops += fest
        day_stops += _take(spots, _SPOTS_PER_DAY - len(fest))
        day_stops += _take(meals, _MEALS_PER_DAY)
        if not day_stops:
            break
        plan.append({"day": day, "stops": _order_by_route(day_stops, base_lat, base_lng)})
    if not plan:
        return None
    return {"days": days, "plan": plan}


def build_itineraries(base_lat: Optional[float], base_lng: Optional[float],
                      attractions: list[dict], restaurants: list[dict],
                      festivals: list[dict]) -> list[dict]:
    """업체 좌표 기준 1박2일(2일)·2박3일(3일) 일정. 좌표가 없으면 빈 배열.

    입력 행 모양은 스냅샷 local.contents 항목({title, body:{mapx, mapy, …}})이다.
    """
    if base_lat is None or base_lng is None:
        return []
    spot_pool = _candidates(attractions, _STOP_ATTRACTION, base_lat, base_lng)
    meal_pool = _candidates(restaurants, _STOP_RESTAURANT, base_lat, base_lng)
    fest_pool = _candidates(festivals, _STOP_FESTIVAL, base_lat, base_lng)

    out = []
    for days in (2, 3):
        # 세트마다 독립된 풀 복사 — 1박2일이 소비한 후보가 2박3일에서 빠지면 안 된다.
        built = _plan_days(days, list(spot_pool), list(meal_pool), list(fest_pool),
                           base_lat, base_lng)
        if built:
            out.append(built)
    return out