from __future__ import annotations

from types import SimpleNamespace

import pytest

from opensquilla.engine.runtime import TurnRunner
from opensquilla.engine.types import AgentConfig
from opensquilla.gateway.config import GatewayConfig


class _SessionConfigManager:
    def __init__(self, config: object | None) -> None:
        self.config = config

    def get_session_config(self, session_key: str) -> object | None:
        return self.config


def test_resolve_agent_max_iterations_prefers_explicit_value(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "222")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(SimpleNamespace(agent_max_iterations=111)),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test", 444) == 444
    assert runner._last_agent_max_iterations_source == "explicit argument"


def test_resolve_agent_max_iterations_accepts_explicit_zero(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "222")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(SimpleNamespace(agent_max_iterations=111)),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test", 0) == 0
    assert runner._last_agent_max_iterations_source == "explicit argument"


def test_resolve_agent_max_iterations_prefers_session_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "222")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(SimpleNamespace(agent_max_iterations=111)),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test") == 111
    assert runner._last_agent_max_iterations_source == "session config"


def test_resolve_agent_max_iterations_accepts_session_config_zero(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "222")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(SimpleNamespace(agent_max_iterations=0)),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test") == 0
    assert runner._last_agent_max_iterations_source == "session config"


def test_resolve_agent_max_iterations_prefers_env_over_gateway_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "222")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(None),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test") == 222
    assert runner._last_agent_max_iterations_source == "env OPENSQUILLA_AGENT_MAX_ITERATIONS"


def test_resolve_agent_max_iterations_accepts_env_zero(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "0")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(None),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test") == 0
    assert runner._last_agent_max_iterations_source == "env OPENSQUILLA_AGENT_MAX_ITERATIONS"


def test_resolve_agent_max_iterations_uses_gateway_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.delenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", raising=False)
    runner = TurnRunner(provider_selector=None, config=GatewayConfig(agent_max_iterations=333))

    assert runner._resolve_agent_max_iterations("agent:main:test") == 333
    assert runner._last_agent_max_iterations_source == "gateway config"


def test_resolve_agent_max_iterations_uses_agent_default_without_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.delenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", raising=False)
    runner = TurnRunner(provider_selector=None, config=None)

    assert runner._resolve_agent_max_iterations("agent:main:test") == AgentConfig().max_iterations
    assert runner._last_agent_max_iterations_source == "AgentConfig default"


def test_resolve_agent_max_iterations_invalid_env_falls_through(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_MAX_ITERATIONS", "not-an-int")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(None),
        config=GatewayConfig(agent_max_iterations=333),
    )

    assert runner._resolve_agent_max_iterations("agent:main:test") == 333
    assert runner._last_agent_max_iterations_source == "gateway config"


def test_resolve_agent_max_iterations_rejects_invalid_explicit_value() -> None:
    runner = TurnRunner(provider_selector=None, config=GatewayConfig())

    with pytest.raises(ValueError, match="max_iterations"):
        runner._resolve_agent_max_iterations("agent:main:test", -1)