"""D02 기준 지정(PLAN 529-3 · 560-2) — 경위도 ↔ TM 변환 · 전체 두 점 정본 API · 옛 연도 행 접기
· 연도 잇기(D03 [저장]이 씀).

./venv/Scripts/python.exe -m pytest resources/tester/test_529_3_plan_points.py -q
"""

import json

import pytest
from fastapi import FastAPI, HTTPException
from fastapi.testclient import TestClient

from common_util.common_util_auth import require_company
from common_util.common_util_crs import lonlat_to_tm, tm_epsg_for_lon, tm_to_lonlat
from D02_PlanBase import D02_PlanBase_Router as router_mod

PID = "12953f5a-5ba1-42ad-a427-8ac857a58bc2"
URL = f"/api/projects/{PID}/plan/points"


# --- 좌표 변환 ---


@pytest.mark.parametrize("epsg, lon0", [(5185, 125.0), (5186, 127.0), (5187, 129.0)])
def test_origin_is_false_origin(epsg, lon0) -> None:
    x, y = lonlat_to_tm(lon0, 38.0, epsg)  # 원점 = (200 000, 600 000)
    assert abs(x - 200000.0) < 0.001 and abs(y - 600000.0) < 0.001


@pytest.mark.parametrize("lon, lat", [(127.3851, 36.3504), (125.7, 34.1), (129.36, 36.02)])
def test_round_trip_within_1mm(lon, lat) -> None:
    epsg = tm_epsg_for_lon(lon)
    x, y = lonlat_to_tm(lon, lat, epsg)
    back = lonlat_to_tm(*tm_to_lonlat(x, y, epsg), epsg)
    assert abs(back[0] - x) < 0.001 and abs(back[1] - y) < 0.001


def test_belt_by_longitude() -> None:
    assert [tm_epsg_for_lon(v) for v in (124.5, 125.999, 126.0, 127.999, 128.0, 131.8)] == [
        5185,
        5185,
        5186,
        5186,
        5187,
        5187,
    ]


# --- 지점 정본 API ---


A, B, C = (
    {"lon": 127.10, "lat": 37.50},
    {"lon": 127.12, "lat": 37.51},
    {"lon": 127.15, "lat": 37.52},
)


@pytest.fixture
def api(monkeypatch):
    state = {
        "project": {"project_kind": "PLAN", "project_year": 2026, "project_year_end": 2027},
        "rows": {},
        "session": {"role": "USER", "email": "kim@forest.go.kr", "company_id": 1},
    }

    async def load(project_id):
        return state["project"], {y: dict(r) for y, r in state["rows"].items()}

    async def store_base(project_id, base):
        state["project"] = {**state["project"], "plan_base_json": json.dumps(base)}

    monkeypatch.setattr(router_mod, "_load", load)
    monkeypatch.setattr(router_mod, "_store_base", store_base)
    app = FastAPI()
    app.include_router(router_mod.router)
    app.dependency_overrides[require_company] = lambda: state["session"]
    client = TestClient(app)
    client.state = state
    return client


def test_put_two_points(api) -> None:
    got = api.put(URL, json={"start": A, "end": C})
    assert got.status_code == 200
    data = got.json()
    assert data["epsg"] == 5186
    assert (data["start"]["x"], data["start"]["y"]) == pytest.approx(
        lonlat_to_tm(127.10, 37.50, 5186)
    )
    assert abs(data["end"]["lon"] - C["lon"]) < 1e-8 and abs(data["end"]["lat"] - C["lat"]) < 1e-8
    stored = json.loads(api.state["project"]["plan_base_json"])
    assert set(stored) == {"epsg", "start", "end"} and set(stored["start"]) == {"x", "y"}
    assert api.state["rows"] == {}  # 연도 행은 D02 가 안 만듦


def test_put_one_point_and_empty(api) -> None:
    data = api.put(URL, json={"start": A}).json()
    assert data["start"] and data["end"] is None and data["epsg"] == 5186
    data = api.put(URL, json={}).json()
    assert data == {"status": "success", "epsg": None, "start": None, "end": None}


def test_put_forest_only(api) -> None:
    api.state["session"] = {"role": "USER", "email": "lee@company.com", "company_id": 1}
    assert api.put(URL, json={"start": A, "end": B}).status_code == 403
    api.state["session"] = {"role": "SYSTEM_ADMIN", "email": "admin@x.com", "company_id": None}
    assert api.put(URL, json={"start": A, "end": B}).status_code == 200


def test_get_folds_old_year_rows(api) -> None:
    epsg = 5186
    a, b, c = (lonlat_to_tm(p["lon"], p["lat"], epsg) for p in (A, B, C))
    for year, s, e in ((2026, a, b), (2027, b, c)):
        api.state["rows"][year] = {
            "year": year,
            **{"start_x": s[0], "start_y": s[1], "end_x": e[0], "end_y": e[1]},
            "epsg": epsg,
        }
    data = api.get(URL).json()
    assert data["epsg"] == epsg
    assert (data["start"]["x"], data["end"]["x"]) == pytest.approx((a[0], c[0]))


def test_get_empty_new_project(api) -> None:
    assert api.get(URL).json() == {"status": "success", "epsg": None, "start": None, "end": None}


def test_chain_years_for_d03() -> None:
    items = [
        router_mod.YearPoints(year=2026, start=A, end=B),
        router_mod.YearPoints(year=2027, start=A, end=C),
    ]
    rows = router_mod.chain_years(2026, 2027, items)
    assert rows[1]["start"].lon == B["lon"]  # 시점 = 직전 종점
    with pytest.raises(HTTPException):
        router_mod.chain_years(2026, 2027, items[:1])