from __future__ import annotations

from types import SimpleNamespace

from adapter.matrix.bot import build_runtime
from adapter.matrix.reconciliation import reconcile_startup_state
from adapter.matrix.store import (
    get_room_meta,
    next_platform_chat_id,
    set_room_meta,
)
from core.store import SQLiteStore
from sdk.mock import MockPlatformClient


async def test_room_agent_id_and_platform_chat_id_survive_restart(tmp_path):
    db = str(tmp_path / "state.db")
    store = SQLiteStore(db)
    await set_room_meta(store, "!room:example.org", {
        "room_type": "chat",
        "agent_id": "agent-1",
        "platform_chat_id": "42",
    })

    store2 = SQLiteStore(db)
    meta = await get_room_meta(store2, "!room:example.org")
    assert meta is not None
    assert meta["agent_id"] == "agent-1"
    assert meta["platform_chat_id"] == "42"


async def test_platform_chat_seq_survives_restart(tmp_path):
    db = str(tmp_path / "state.db")
    store = SQLiteStore(db)
    assert await next_platform_chat_id(store) == "1"
    assert await next_platform_chat_id(store) == "2"
    assert await next_platform_chat_id(store) == "3"

    store2 = SQLiteStore(db)
    assert await next_platform_chat_id(store2) == "4"


async def test_routing_state_survives_restart_and_routes_correctly(tmp_path):
    db = str(tmp_path / "state.db")
    store = SQLiteStore(db)
    await set_room_meta(store, "!convo:example.org", {
        "room_type": "chat",
        "agent_id": "agent-1",
        "platform_chat_id": "10",
    })

    store2 = SQLiteStore(db)
    meta = await get_room_meta(store2, "!convo:example.org")
    assert meta is not None
    assert meta["agent_id"] == "agent-1"
    assert meta["platform_chat_id"] == "10"


async def test_missing_durable_store_starts_clean(tmp_path):
    db = str(tmp_path / "brand_new.db")
    store = SQLiteStore(db)
    assert await get_room_meta(store, "!nonexistent:example.org") is None


async def test_startup_reconciliation_backfills_legacy_platform_chat_id_before_restart_routes(
    tmp_path,
):
    db = str(tmp_path / "state.db")
    store = SQLiteStore(db)
    await set_room_meta(
        store,
        "!chat2:example.org",
        {
            "room_type": "chat",
            "chat_id": "C2",
            "display_name": "Чат 2",
            "matrix_user_id": "@alice:example.org",
            "space_id": "!space:example.org",
        },
    )

    runtime = build_runtime(platform=MockPlatformClient(), store=store)
    client = SimpleNamespace(
        user_id="@bot:example.org",
        rooms={
            "!space:example.org": SimpleNamespace(
                room_id="!space:example.org",
                name="Lambda - Alice",
                display_name="Lambda - Alice",
                users={
                    "@bot:example.org": SimpleNamespace(user_id="@bot:example.org"),
                    "@alice:example.org": SimpleNamespace(user_id="@alice:example.org"),
                },
                space_parents=set(),
            ),
            "!chat2:example.org": SimpleNamespace(
                room_id="!chat2:example.org",
                name="Чат 2",
                display_name="Чат 2",
                users={
                    "@bot:example.org": SimpleNamespace(user_id="@bot:example.org"),
                    "@alice:example.org": SimpleNamespace(user_id="@alice:example.org"),
                },
                space_parents={"!space:example.org"},
            ),
        },
    )

    await reconcile_startup_state(client, runtime)

    store2 = SQLiteStore(db)
    room_meta = await get_room_meta(store2, "!chat2:example.org")
    assert room_meta is not None
    assert room_meta["platform_chat_id"] == "1"
