import time
from types import SimpleNamespace
from pathlib import Path
import pytest

import sys

sys.path.append(str(Path(__file__).resolve().parents[1]))

import main


def test_inactivity_triggers_lock(monkeypatch):
    locked = {"locked": False, "unlocked": False}

    def update_activity():
        pm.last_activity = time.time()

    def lock_vault():
        locked["locked"] = True

    def unlock_vault():
        locked["unlocked"] = True
        update_activity()

    pm = SimpleNamespace(
        is_dirty=False,
        last_update=time.time(),
        last_activity=time.time() - 1.0,
        nostr_client=SimpleNamespace(close_client_pool=lambda: None),
        handle_add_password=lambda: None,
        handle_retrieve_entry=lambda: None,
        handle_modify_entry=lambda: None,
        update_activity=update_activity,
        lock_vault=lock_vault,
        unlock_vault=unlock_vault,
    )

    monkeypatch.setattr(main, "timed_input", lambda *_: "7")

    with pytest.raises(SystemExit):
        main.display_menu(pm, sync_interval=1000, inactivity_timeout=0.1)

    assert locked["locked"]
    assert locked["unlocked"]


def test_input_timeout_triggers_lock(monkeypatch):
    """Ensure locking occurs if no input is provided before timeout."""
    locked = {"locked": 0, "unlocked": 0}

    def update_activity():
        pm.last_activity = time.time()

    def lock_vault():
        locked["locked"] += 1

    def unlock_vault():
        locked["unlocked"] += 1
        update_activity()

    pm = SimpleNamespace(
        is_dirty=False,
        last_update=time.time(),
        last_activity=time.time(),
        nostr_client=SimpleNamespace(close_client_pool=lambda: None),
        handle_add_password=lambda: None,
        handle_retrieve_entry=lambda: None,
        handle_modify_entry=lambda: None,
        update_activity=update_activity,
        lock_vault=lock_vault,
        unlock_vault=unlock_vault,
    )

    responses = iter([TimeoutError(), "7"])

    def fake_input(*_args, **_kwargs):
        val = next(responses)
        if isinstance(val, Exception):
            raise val
        return val

    monkeypatch.setattr(main, "timed_input", fake_input)

    with pytest.raises(SystemExit):
        main.display_menu(pm, sync_interval=1000, inactivity_timeout=0.1)

    assert locked["locked"] == 1
    assert locked["unlocked"] == 1