"""HTTP / WebSocket 接口：用 TestClient 起完整的应用，Hub 经进程内假 broker 连模拟板子。"""
import time

import pytest
from fastapi.testclient import TestClient

from prrack.api import create_app
from prrack.bus import MemBroker, MemBus
from prrack.db import Db, now_ms
from prrack.hub import Hub
from prrack.recovery import RecoveryManager
from prrack.simboard import SimBoard

B = "b81f3fc9c8bc"


@pytest.fixture
def env(tmp_path):
    broker = MemBroker()
    db = Db(str(tmp_path / "a.db"))
    hub = Hub(db, MemBus(broker), attempt_timeout=0.2)
    rec = RecoveryManager(hub, now_ms)
    sim = SimBoard(MemBus(broker), B, present=(True, True, True, False, True))
    dist = tmp_path / "dist"
    (dist / "assets").mkdir(parents=True)
    (dist / "index.html").write_text("<html>rack</html>", encoding="utf-8")
    async def up():
        await hub.start()
        sim.start()
        sim.tick(now_ms())

    app = create_app(hub, rec, str(dist), startup=[up], shutdown=[hub.stop])
    with TestClient(app) as c:
        deadline = time.time() + 2
        while B not in hub.latest and time.time() < deadline:
            time.sleep(0.01)
        yield c, hub, sim, db


def test_boards_and_snapshot(env):
    c, hub, sim, db = env
    boards = c.get("/api/boards").json()
    assert [b["id"] for b in boards] == [B]
    assert boards[0]["online"] is True and len(boards[0]["phones"]) == 5
    assert c.get("/api/boards/nope").status_code == 404


def test_command_records_the_portal_user(env):
    c, hub, sim, db = env
    r = c.post("/api/boards/%s/cmd" % B, json={"cmd": "power", "rail": 1, "on": False},
               headers={"X-Auth-User": "zhouyang"})
    assert r.status_code == 200 and r.json()["ok"] is True
    assert sim.rails[0].en is False
    a = c.get("/api/audit").json()[0]
    assert a["user"] == "zhouyang" and a["request"]["cmd"] == "power" and a["ok"] == 1
    # 没有门户头（服务器本机直接 curl）记作 local
    c.post("/api/boards/%s/cmd" % B, json={"cmd": "power", "rail": 1, "on": True})
    assert c.get("/api/audit").json()[0]["user"] == "local"


def test_bad_commands_never_reach_the_board(env):
    c, hub, sim, db = env
    n = len(sim.received)
    for body in ({"cmd": "wifi", "ssid": "x"}, {"cmd": "set_voltage", "rail": 1, "mv": 4300},
                 {"cmd": "key", "rail": 1, "keys": "pwr", "ms": 60000}, {"cmd": "rm -rf"}):
        r = c.post("/api/boards/%s/cmd" % B, json=body)
        assert r.status_code == 400, body
    assert len(sim.received) == n


def test_board_rejection_is_passed_through(env):
    c, hub, sim, db = env
    r = c.post("/api/boards/%s/cmd" % B, json={"cmd": "power_cycle", "rail": 4})
    assert r.status_code == 200 and r.json()["err"] == "empty"


def test_timeout_is_504(env):
    c, hub, sim, db = env
    sim.drop_next_replies = 99
    r = c.post("/api/boards/%s/cmd" % B, json={"cmd": "power", "rail": 2, "on": True})
    assert r.status_code == 504


def test_phone_settings_toggle_auto_recovery(env):
    c, hub, sim, db = env
    r = c.patch("/api/boards/%s/rails/3" % B, json={"label": "A-03", "auto_recover": True})
    assert r.status_code == 200
    ph = [p for p in r.json()["phones"] if p["rail"] == 3][0]
    assert ph["label"] == "A-03" and ph["auto_recover"] == 1
    rec = c.get("/api/recovery").json()
    assert rec["params"]["window_s"] == 300
    r = c.put("/api/recovery/params", json={"window_s": 120, "flat_ua": -5, "bogus": 1})
    assert r.json()["window_s"] == 120 and r.json()["flat_ua"] == 5000  # 负数不收


def test_history_after_telemetry(env):
    c, hub, sim, db = env
    for _ in range(3):
        sim.tick(now_ms())
    time.sleep(0.1)
    hub.flush()
    h = c.get("/api/boards/%s/rails/1/history" % B, params={"start": now_ms() - 60000}).json()
    assert h["resolution"] == "raw" and len(h["points"]) >= 1
    assert h["points"][0][2] is not None


def test_websocket_hello_and_live_tele(env):
    c, hub, sim, db = env
    with c.websocket_connect("/api/ws", headers={"X-Auth-User": "zhouyang"}) as ws:
        hello = ws.receive_json()
        assert hello["t"] == "hello" and hello["user"] == "zhouyang" and hello["boards"][0]["id"] == B
        sim.tick(now_ms())
        msg = ws.receive_json()
        while msg["t"] != "tele":
            msg = ws.receive_json()
        assert msg["board"] == B and len(msg["status"]["rails"]) == 5


def test_spa_fallback_is_not_cached(env):
    c, hub, sim, db = env
    r = c.get("/some/deep/route")
    assert r.status_code == 200 and "rack" in r.text
    assert r.headers["cache-control"] == "no-cache"


def test_a_real_websocket_library_is_installed():
    """TestClient 自带 WS 实现，上面那条 WS 用例在没装 websockets 的机器上也是绿的；
    而真 uvicorn 缺了它会把 /api/ws 当普通 GET（落到 SPA 兜底，回 200 HTML）。
    deploy.sh 在服务器上跑这套测试，所以缺库会在换上去之前被拦住。"""
    from prrack.main import ws_library_ok
    assert ws_library_ok()


def test_board_history_raw_and_minute(env):
    c, hub, sim, db = env
    base = now_ms() // 60000 * 60000 - 3 * 60000
    rows = [(base + i * 1000, B, 12000, 1500000, 35000 + i, 40000 - i, 300) for i in range(120)]
    db.many("INSERT INTO board_tele VALUES (?,?,?,?,?,?,?)", rows)
    h = c.get("/api/boards/%s/history" % B, params={"start": base, "end": base + 120000}).json()
    assert h["resolution"] == "raw" and len(h["points"]) >= 100
    p = h["points"][0]
    assert p[1] == 35000 and p[4] == 40000 and p[7] == 300 and p[8] == 12000
    # 分钟聚合：两个整分钟，每分钟 min/max 就是那一分钟里的首尾
    assert db.downsample(base + 180000) > 0
    m = db.board_history(B, base, base + 120000 + 1)  # 跨度小也能强制走聚合：t0 早于原始保留期时
    rows_min = db.q("SELECT * FROM board_tele_min WHERE board_id=? ORDER BY ts", (B,))
    assert [r["t0_min"] for r in rows_min][:2] == [35000, 35060]
    assert [r["t0_max"] for r in rows_min][:2] == [35059, 35119]
    assert rows_min[0]["t1_min"] == 40000 - 59 and rows_min[0]["fan_avg"] == 300
    assert m["resolution"] == "raw"
    old = db.board_history(B, base - 10 * 86400000, base + 120000)
    # 10 天跨度 ⇒ 桶约 9 分钟，两分钟落进同一桶：min 取两分钟的最小、max 取最大
    assert old["resolution"] == "minute" and old["points"][0][2] == 35000
    assert max(p[3] for p in old["points"]) == 35119
    assert c.get("/api/boards/nope/history", params={"start": 0}).status_code == 404