from __future__ import annotations

import asyncio
import json

from beaver.tools.builtins import web


class _FakeResponse:
    headers = {"content-type": "text/html"}
    status_code = 200

    def __init__(self, url: str = "https://example.com") -> None:
        self.url = url
        if "duckduckgo.com" in url:
            self.text = '<a class="result__a" href="https://duck.example.com">Duck Example</a>'
        else:
            self.text = (
                '<li class="b_algo"><h2><a href="https://example.com">Example</a></h2>'
                "<p>Example result</p></li>"
            )

    def raise_for_status(self) -> None:
        return None


class _FakeAsyncClient:
    calls: list[dict[str, object]] = []
    urls: list[str] = []
    fail_bing = False

    def __init__(self, **kwargs: object) -> None:
        self.calls.append(kwargs)

    async def __aenter__(self) -> "_FakeAsyncClient":
        return self

    async def __aexit__(self, *args: object) -> None:
        return None

    async def get(self, *args: object, **kwargs: object) -> _FakeResponse:
        url = str(args[0])
        self.urls.append(url)
        if self.fail_bing and "bing.com" in url:
            raise web.httpx.ConnectTimeout("bing unavailable")
        return _FakeResponse(url)


def test_web_tools_use_environment_proxy_settings(monkeypatch) -> None:
    _FakeAsyncClient.calls = []
    monkeypatch.setattr(web.httpx, "AsyncClient", _FakeAsyncClient)

    async def _run() -> None:
        await web.WebFetchTool().execute(url="https://example.com")
        await web.WebSearchTool().execute(query="example")

    asyncio.run(_run())

    assert [call.get("trust_env") for call in _FakeAsyncClient.calls] == [True, True]


def test_web_fetch_uses_short_connect_timeout(monkeypatch) -> None:
    _FakeAsyncClient.calls = []
    _FakeAsyncClient.urls = []
    _FakeAsyncClient.fail_bing = False
    monkeypatch.setattr(web.httpx, "AsyncClient", _FakeAsyncClient)

    asyncio.run(web.WebFetchTool().execute(url="https://example.com"))

    timeout = _FakeAsyncClient.calls[0]["timeout"]
    assert isinstance(timeout, web.httpx.Timeout)
    assert timeout.connect == 5
    assert timeout.read == 12


def test_web_search_uses_reachable_bing_endpoint_first(monkeypatch) -> None:
    _FakeAsyncClient.calls = []
    _FakeAsyncClient.urls = []
    _FakeAsyncClient.fail_bing = False
    monkeypatch.setattr(web.httpx, "AsyncClient", _FakeAsyncClient)

    raw = asyncio.run(web.WebSearchTool().execute(query="weather beijing"))

    payload = json.loads(raw)
    assert payload["success"] is True
    assert payload["engine"] in {"bing", "duckduckgo"}
    assert set(_FakeAsyncClient.urls) == {
        "https://www.bing.com/search?q=weather+beijing",
        "https://duckduckgo.com/html/?q=weather+beijing",
    }

    timeout = _FakeAsyncClient.calls[0]["timeout"]
    assert isinstance(timeout, web.httpx.Timeout)
    assert timeout.connect == 5
    assert timeout.read == 8


def test_web_search_falls_back_when_bing_is_unavailable(monkeypatch) -> None:
    _FakeAsyncClient.calls = []
    _FakeAsyncClient.urls = []
    _FakeAsyncClient.fail_bing = True
    monkeypatch.setattr(web.httpx, "AsyncClient", _FakeAsyncClient)

    raw = asyncio.run(web.WebSearchTool().execute(query="weather beijing"))

    payload = json.loads(raw)
    assert payload["success"] is True
    assert payload["engine"] == "duckduckgo"
    assert set(_FakeAsyncClient.urls) == {
        "https://www.bing.com/search?q=weather+beijing",
        "https://duckduckgo.com/html/?q=weather+beijing",
    }