"""
The recovery-wrapped keypair copy (docs/auth-confirm.md §4.3, MNP 0.14).

`bundle_enc_recovery` is a second copy of the identity bundle wrapped under the
account's recovery key. The store has to add the column to a database that
predates it, and a plain re-backup that omits the recovery copy must not erase
one already there.
"""

import aiosqlite
import pytest
from meshbay_node.bundle_store import BundleStore


@pytest.mark.asyncio
async def test_round_trip_with_and_without_recovery(tmp_path):
    store = BundleStore(db_path=tmp_path / "bundles.db")
    await store.open()

    await store.store_keypair("u1", "pass-wrapped-1")
    row = await store.fetch_keypair("u1")
    assert row == {"bundle_enc": "pass-wrapped-1", "bundle_enc_recovery": None}

    await store.store_keypair("u2", "pass-wrapped-2", "recovery-wrapped-2")
    row = await store.fetch_keypair("u2")
    assert row["bundle_enc"] == "pass-wrapped-2"
    assert row["bundle_enc_recovery"] == "recovery-wrapped-2"

    assert await store.fetch_keypair("nobody") is None
    await store.close()


@pytest.mark.asyncio
async def test_re_backup_without_recovery_keeps_the_existing_copy(tmp_path):
    """A passphrase-change re-wrap sends only bundle_enc; the recovery copy stays."""
    store = BundleStore(db_path=tmp_path / "bundles.db")
    await store.open()

    await store.store_keypair("u1", "v1", "recovery-v1")
    await store.store_keypair("u1", "v2")                       # no recovery arg
    row = await store.fetch_keypair("u1")
    assert row["bundle_enc"] == "v2"
    assert row["bundle_enc_recovery"] == "recovery-v1"

    # An explicit new recovery copy does replace it.
    await store.store_keypair("u1", "v3", "recovery-v3")
    row = await store.fetch_keypair("u1")
    assert row == {"bundle_enc": "v3", "bundle_enc_recovery": "recovery-v3"}
    await store.close()


@pytest.mark.asyncio
async def test_migration_adds_the_column_to_an_old_database(tmp_path):
    db_path = tmp_path / "bundles.db"

    # A keypair_bundles table as it looked before MNP 0.14.
    async with aiosqlite.connect(str(db_path)) as db:
        await db.execute(
            "CREATE TABLE keypair_bundles ("
            "  user_id TEXT PRIMARY KEY,"
            "  bundle_enc TEXT NOT NULL,"
            "  stored_at TEXT NOT NULL DEFAULT (datetime('now')))")
        await db.execute(
            "INSERT INTO keypair_bundles (user_id, bundle_enc) VALUES ('old', 'legacy')")
        await db.commit()

    store = BundleStore(db_path=db_path)
    await store.open()   # runs the migration

    row = await store.fetch_keypair("old")
    assert row == {"bundle_enc": "legacy", "bundle_enc_recovery": None}

    await store.store_keypair("old", "legacy", "recovery-now")
    row = await store.fetch_keypair("old")
    assert row["bundle_enc_recovery"] == "recovery-now"
    await store.close()