import base64
from pathlib import Path
import sys
import types

import pytest

from repeater.data_acquisition.sqlite_handler import SQLiteHandler


def _make_handler(tmp_path: Path) -> SQLiteHandler:
    return SQLiteHandler(tmp_path)


def test_api_token_crud_cycle(tmp_path):
    h = _make_handler(tmp_path)

    token_id = h.create_api_token("svc-a", "hash-a")
    assert isinstance(token_id, int)

    verified = h.verify_api_token("hash-a")
    assert verified is not None
    assert verified["id"] == token_id
    assert verified["name"] == "svc-a"

    listed = h.list_api_tokens()
    assert any(t["id"] == token_id for t in listed)

    assert h.revoke_api_token(token_id) is True
    assert h.verify_api_token("hash-a") is None
    assert h.revoke_api_token(token_id) is False


def test_transport_key_crud_cycle(tmp_path):
    h = _make_handler(tmp_path)

    key_id = h.create_transport_key(
        name="root",
        flood_policy="allow",
        transport_key="dGVzdC1rZXk=",  # base64('test-key')
    )
    assert isinstance(key_id, int)

    row = h.get_transport_key_by_id(key_id)
    assert row is not None
    assert row["name"] == "root"
    assert row["flood_policy"] == "allow"

    # No fields to update returns False by design.
    assert h.update_transport_key(key_id) is False

    assert h.update_transport_key(key_id, name="child", flood_policy="deny") is True
    row2 = h.get_transport_key_by_id(key_id)
    assert row2 is not None
    assert row2["name"] == "child"
    assert row2["flood_policy"] == "deny"

    all_rows = h.get_transport_keys()
    assert len(all_rows) == 1
    assert all_rows[0]["id"] == key_id

    assert h.delete_transport_key(key_id) is True
    assert h.get_transport_key_by_id(key_id) is None
    assert h.delete_transport_key(key_id) is False


def test_generate_transport_key_uses_implicit_hashtag_region(tmp_path, monkeypatch):
    h = _make_handler(tmp_path)

    captured = {}
    fake_transport_keys = types.ModuleType("pymc_core.protocol.transport_keys")

    def _fake_get_auto_key_for(name: str) -> bytes:
        captured["name"] = name
        return b"0123456789abcdef"

    fake_transport_keys.get_auto_key_for = _fake_get_auto_key_for

    fake_protocol = types.ModuleType("pymc_core.protocol")
    fake_protocol.transport_keys = fake_transport_keys

    fake_core = types.ModuleType("pymc_core")
    fake_core.protocol = fake_protocol

    monkeypatch.setitem(sys.modules, "pymc_core", fake_core)
    monkeypatch.setitem(sys.modules, "pymc_core.protocol", fake_protocol)
    monkeypatch.setitem(sys.modules, "pymc_core.protocol.transport_keys", fake_transport_keys)

    generated = h.generate_transport_key("eu")
    generated_bytes = base64.b64decode(generated)

    assert captured["name"] == "eu"
    assert generated_bytes == b"0123456789abcdef"
    assert len(generated_bytes) == 16


def test_room_messages_and_sync_flow(tmp_path):
    h = _make_handler(tmp_path)

    room_hash = "0x42"
    a_pub = "a" * 64
    b_pub = "b" * 64

    m1 = h.insert_room_message(room_hash, a_pub, "hello", post_timestamp=100.0)
    m2 = h.insert_room_message(room_hash, b_pub, "world", post_timestamp=200.0)
    assert isinstance(m1, int)
    assert isinstance(m2, int)

    assert h.get_room_message_count(room_hash) == 2

    # get_room_messages sorts by post_timestamp DESC
    msgs = h.get_room_messages(room_hash, limit=10, offset=0)
    assert len(msgs) == 2
    assert msgs[0]["message_text"] == "world"

    since = h.get_messages_since(room_hash, since_timestamp=150.0, limit=10)
    assert len(since) == 1
    assert since[0]["message_text"] == "world"

    unsynced_for_a = h.get_unsynced_messages(room_hash, client_pubkey=a_pub, sync_since=0.0)
    assert len(unsynced_for_a) == 1
    assert unsynced_for_a[0]["author_pubkey"] == b_pub

    assert h.get_unsynced_count(room_hash, client_pubkey=a_pub, sync_since=0.0) == 1

    # Client sync upsert/get/list
    assert h.upsert_client_sync(room_hash, a_pub, sync_since=50.0, last_activity=123.0) is True
    sync = h.get_client_sync(room_hash, a_pub)
    assert sync is not None
    assert sync["sync_since"] == 50.0

    clients = h.get_all_room_clients(room_hash)
    assert len(clients) == 1
    assert clients[0]["client_pubkey"] == a_pub

    assert h.delete_room_message(room_hash, int(m1)) is True
    assert h.delete_room_message(room_hash, int(m1)) is False

    deleted = h.clear_room_messages(room_hash)
    assert deleted == 1
    assert h.get_room_message_count(room_hash) == 0


def test_store_and_delete_advert(tmp_path):
    h = _make_handler(tmp_path)

    h.store_advert(
        {
            "timestamp": 123.0,
            "pubkey": "pk1",
            "node_name": "node-1",
            "is_repeater": True,
            "route_type": 1,
            "contact_type": "neighbor",
            "latitude": 1.0,
            "longitude": 2.0,
            "rssi": -88,
            "snr": 7.5,
            "is_new_neighbor": True,
            "zero_hop": True,
        }
    )

    with h._connect() as conn:
        row = conn.execute("SELECT id FROM adverts WHERE pubkey = ?", ("pk1",)).fetchone()
    assert row is not None
    advert_id = int(row[0])

    assert h.delete_advert(advert_id) is True
    assert h.delete_advert(advert_id) is False


def test_verify_api_token_last_used_throttle(tmp_path, monkeypatch):
    h = _make_handler(tmp_path)
    h._api_token_last_used_interval_sec = 300

    now = {"v": 1000.0}

    monkeypatch.setattr("repeater.data_acquisition.sqlite_handler.time.time", lambda: now["v"])

    token_id = h.create_api_token("svc-throttle", "hash-throttle")
    assert token_id > 0

    assert h.verify_api_token("hash-throttle") is not None
    with h._connect() as conn:
        first = conn.execute(
            "SELECT last_used FROM api_tokens WHERE id = ?", (token_id,)
        ).fetchone()[0]
    assert first == 1000.0

    now["v"] = 1010.0
    assert h.verify_api_token("hash-throttle") is not None
    with h._connect() as conn:
        second = conn.execute(
            "SELECT last_used FROM api_tokens WHERE id = ?", (token_id,)
        ).fetchone()[0]
    assert second == 1000.0

    now["v"] = 1401.0
    assert h.verify_api_token("hash-throttle") is not None
    with h._connect() as conn:
        third = conn.execute(
            "SELECT last_used FROM api_tokens WHERE id = ?", (token_id,)
        ).fetchone()[0]
    assert third == 1401.0


def test_store_advert_zero_hop_signal_handling(tmp_path):
    h = _make_handler(tmp_path)

    h.store_advert(
        {
            "timestamp": 10.0,
            "pubkey": "pk-z",
            "node_name": "node-z",
            "is_repeater": False,
            "route_type": 1,
            "contact_type": "neighbor",
            "rssi": -80,
            "snr": 5.0,
            "is_new_neighbor": True,
            "zero_hop": True,
        }
    )

    # Multi-hop update must preserve previous zero-hop signal quality.
    h.store_advert(
        {
            "timestamp": 20.0,
            "pubkey": "pk-z",
            "node_name": "node-z-2",
            "is_repeater": False,
            "route_type": 2,
            "contact_type": "neighbor",
            "rssi": -50,
            "snr": 9.0,
            "is_new_neighbor": False,
            "zero_hop": False,
        }
    )
    with h._connect() as conn:
        row = conn.execute(
            "SELECT rssi, snr, zero_hop, advert_count FROM adverts WHERE pubkey = ?",
            ("pk-z",),
        ).fetchone()
    assert row is not None
    assert row[0] == -80
    assert row[1] == 5.0
    assert bool(row[2]) is True
    assert row[3] == 2

    # New zero-hop update should refresh signal quality.
    h.store_advert(
        {
            "timestamp": 30.0,
            "pubkey": "pk-z",
            "node_name": "node-z-3",
            "is_repeater": False,
            "route_type": 1,
            "contact_type": "neighbor",
            "rssi": -60,
            "snr": 6.5,
            "is_new_neighbor": False,
            "zero_hop": True,
        }
    )
    with h._connect() as conn:
        row2 = conn.execute(
            "SELECT rssi, snr, zero_hop, advert_count FROM adverts WHERE pubkey = ?",
            ("pk-z",),
        ).fetchone()
    assert row2 is not None
    assert row2[0] == -60
    assert row2[1] == 6.5
    assert bool(row2[2]) is True
    assert row2[3] == 3


def test_sync_transport_keys_validation_and_tree_apply(tmp_path, monkeypatch):
    h = _make_handler(tmp_path)

    with pytest.raises(ValueError, match="must be a list"):
        h.sync_transport_keys({"bad": True})

    with pytest.raises(ValueError, match="Duplicate node_id"):
        h.sync_transport_keys(
            [
                {"node_id": "1", "name": "a", "flood_policy": "allow"},
                {"node_id": "1", "name": "b", "flood_policy": "deny"},
            ]
        )

    with pytest.raises(ValueError, match="Parent node 'missing'"):
        h.sync_transport_keys(
            [
                {
                    "node_id": "c1",
                    "name": "child",
                    "flood_policy": "allow",
                    "parent_node_id": "missing",
                }
            ]
        )

    monkeypatch.setattr(h, "generate_transport_key", lambda _name: "GEN-KEY")
    applied = h.sync_transport_keys(
        [
            {
                "node_id": "root",
                "name": "root-name",
                "flood_policy": "allow",
                "transport_key": "ROOT-KEY",
            },
            {
                "node_id": "child",
                "name": "child-name",
                "flood_policy": "deny",
                "parent_node_id": "root",
                "transport_key": None,
            },
        ]
    )

    assert applied == {"applied_nodes": 2, "generated_keys": 1}

    with h._connect() as conn:
        rows = conn.execute(
            "SELECT id, name, flood_policy, transport_key, parent_id FROM transport_keys ORDER BY id"
        ).fetchall()
    assert len(rows) == 2
    assert rows[0][1] == "root-name"
    assert rows[0][3] == "ROOT-KEY"
    assert rows[1][1] == "child-name"
    assert rows[1][2] == "deny"
    assert rows[1][3] == "GEN-KEY"
    assert rows[1][4] == rows[0][0]