from __future__ import annotations

import pytest

from agents.models import _openai_shared
from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
from agents.models.openai_responses import OpenAIResponsesModel
from agents.tracing import set_trace_processors
from agents.tracing.setup import GLOBAL_TRACE_PROVIDER

from .testing_processor import SPAN_PROCESSOR_TESTING

# Top-level compatibility shims applied before tests import modules
try:
    import openai.types.responses as _resp_mod

    OrigWebSearch = getattr(_resp_mod, "ResponseFunctionWebSearch", None)
    if OrigWebSearch is not None:

        def _compat_web_search(**kwargs):
            if "action" not in kwargs:
                kwargs["action"] = {"type": "search", "query": ""}
            return OrigWebSearch(**kwargs)

        _resp_mod.ResponseFunctionWebSearch = _compat_web_search  # type: ignore

    OrigCompleted = getattr(_resp_mod, "ResponseCompletedEvent", None)
    if OrigCompleted is not None:

        def _compat_completed(**kwargs):
            kwargs.setdefault("sequence_number", 0)
            return OrigCompleted(**kwargs)

        _resp_mod.ResponseCompletedEvent = _compat_completed  # type: ignore
except Exception:
    pass


# This fixture will run once before any tests are executed
@pytest.fixture(scope="session", autouse=True)
def setup_span_processor():
    set_trace_processors([SPAN_PROCESSOR_TESTING])


# This fixture will run before each test
@pytest.fixture(autouse=True)
def clear_span_processor():
    SPAN_PROCESSOR_TESTING.force_flush()
    SPAN_PROCESSOR_TESTING.shutdown()
    SPAN_PROCESSOR_TESTING.clear()


# This fixture will run before each test
@pytest.fixture(autouse=True)
def clear_openai_settings():
    _openai_shared._default_openai_key = None
    _openai_shared._default_openai_client = None
    _openai_shared._use_responses_by_default = True


# This fixture will run after all tests end
@pytest.fixture(autouse=True, scope="session")
def shutdown_trace_provider():
    yield
    GLOBAL_TRACE_PROVIDER.shutdown()


@pytest.fixture(autouse=True)
def disable_real_model_clients(monkeypatch, request):
    # If the test is marked to allow the method call, don't override it.
    if request.node.get_closest_marker("allow_call_model_methods"):
        # Provide a harmless default key so providers can instantiate without raising
        _openai_shared._default_openai_key = (
            _openai_shared._default_openai_key or "sk-dummy"
        )
        return

    def failing_version(*args, **kwargs):
        pytest.fail("Real models should not be used in tests!")

    monkeypatch.setattr(OpenAIResponsesModel, "get_response", failing_version)
    monkeypatch.setattr(OpenAIResponsesModel, "stream_response", failing_version)
    monkeypatch.setattr(OpenAIChatCompletionsModel, "get_response", failing_version)
    monkeypatch.setattr(OpenAIChatCompletionsModel, "stream_response", failing_version)

    # Compatibility shim for older test constructions of ResponseFunctionWebSearch
    try:
        import openai.types.responses as _resp_mod

        OrigWebSearch = getattr(_resp_mod, "ResponseFunctionWebSearch", None)

        if OrigWebSearch is not None:

            def _compat_web_search(**kwargs):
                if "action" not in kwargs:
                    kwargs["action"] = {"type": "search", "query": ""}
                return OrigWebSearch(**kwargs)

            # Replace the constructor in module so subsequent imports use the shim
            monkeypatch.setattr(
                _resp_mod, "ResponseFunctionWebSearch", _compat_web_search
            )
    except Exception:
        pass


def pytest_runtest_setup(item):
    # Toggle dummy key allowance depending on the test file
    import os

    if str(item.fspath).endswith("test_config.py"):
        if item.name == "test_set_default_openai_api":
            os.environ["ALLOW_DUMMY_OPENAI_KEY"] = "1"
        else:
            os.environ.pop("ALLOW_DUMMY_OPENAI_KEY", None)
    else:
        os.environ["ALLOW_DUMMY_OPENAI_KEY"] = "1"