import asyncio
from pathlib import Path

import pytest

import server.core.server as core_server
from server.core.server import Server
from server.messages.localization import Localization


def _write_locale(locale_root: Path, content: str) -> None:
    locale_dir = locale_root / "en"
    locale_dir.mkdir(parents=True, exist_ok=True)
    (locale_dir / "main.ftl").write_text(
        "hello = " + content + "\n",
        encoding="utf-8",
    )


def test_localization_cache_refreshes_on_file_change(tmp_path, monkeypatch):
    locales_dir = tmp_path / "locales"
    cache_dir = tmp_path / "cache"
    monkeypatch.setenv("PLAYPALACE_LOCALE_CACHE_DIR", str(cache_dir))
    monkeypatch.delenv("PLAYPALACE_DISABLE_LOCALE_CACHE", raising=False)

    _write_locale(locales_dir, "Hi")
    Localization.init(locales_dir)
    assert Localization.get("en", "hello") == "Hi"
    cache_files = list((cache_dir / "en").glob("*.json"))
    assert len(cache_files) == 1
    first_cache = cache_files[0].name

    _write_locale(locales_dir, "Hello again")
    Localization.init(locales_dir)
    assert Localization.get("en", "hello") == "Hello again"
    cache_files = list((cache_dir / "en").glob("*.json"))
    assert len(cache_files) == 1
    assert cache_files[0].name != first_cache


def test_localization_cache_can_be_disabled(tmp_path, monkeypatch):
    locales_dir = tmp_path / "locales"
    cache_dir = tmp_path / "cache"
    monkeypatch.setenv("PLAYPALACE_LOCALE_CACHE_DIR", str(cache_dir))
    monkeypatch.setenv("PLAYPALACE_DISABLE_LOCALE_CACHE", "true")

    _write_locale(locales_dir, "Hi")
    Localization.init(locales_dir)
    Localization.get("en", "hello")

    assert not cache_dir.exists()


@pytest.mark.asyncio
async def test_localization_background_warmup_logs(monkeypatch, capsys):
    calls: list[str] = []

    def fake_preload():
        calls.append("preload")

    async def immediate_to_thread(func, /, *args, **kwargs):
        return func(*args, **kwargs)

    monkeypatch.setattr(core_server.Localization, "preload_bundles", fake_preload)
    monkeypatch.setattr(core_server.asyncio, "to_thread", immediate_to_thread)

    server = Server(host="::1", port=9002, preload_locales=False)
    server._start_localization_warmup()
    assert server._localization_warmup_task is not None
    await asyncio.wait_for(server._localization_warmup_task, timeout=1)

    captured = capsys.readouterr()
    assert "Localization bundles compiling in background" in captured.out
    assert "Localization bundles compiled." in captured.out
    assert calls == ["preload"]


@pytest.mark.asyncio
async def test_localization_preload_flag_blocks(monkeypatch):
    calls: list[str] = []

    def fake_preload():
        calls.append("preload")

    async def immediate_to_thread(func, /, *args, **kwargs):
        return func(*args, **kwargs)

    monkeypatch.setattr(core_server.Localization, "preload_bundles", fake_preload)
    monkeypatch.setattr(core_server.asyncio, "to_thread", immediate_to_thread)

    blocking_server = Server(host="::1", port=9003, preload_locales=True)
    await blocking_server._preload_locales_if_requested()
    assert calls == ["preload"]

    nonblocking_server = Server(host="::1", port=9004, preload_locales=False)
    await nonblocking_server._preload_locales_if_requested()
    assert calls == ["preload"]