"""Single-URL web-fetch control tool."""

from __future__ import annotations

import re
from typing import Any

from openai4s.tools.base import Tool
from openai4s.tools.taxonomy import resource_key


class WebFetchTool(Tool):
    """Normalize fetch arguments and preserve the host soft-fail contract."""

    name = "web_fetch"
    host_method = "web_fetch"
    description = "Fetch a URL and return its content as markdown/text/html/json."
    parameters = {
        "properties": {
            "url": {
                "type": "string",
                "minLength": 1,
                "description": "URL to fetch.",
            },
            "format": {
                "type": "string",
                "enum": ["markdown", "text", "html", "json"],
                "description": "One of markdown|text|html|json (default markdown).",
            },
            "max_chars": {
                "type": "integer",
                "minimum": 1,
                "maximum": 1000000,
                "description": "Truncate the returned content to this many characters.",
            },
            "timeout": {
                "type": "number",
                "minimum": 1,
                "maximum": 120,
                "description": "Seconds to wait before giving up (default 30).",
            },
            "method": {
                "type": "string",
                "enum": ["GET", "HEAD"],
                "description": (
                    "HEAD asks only whether the resource exists, without "
                    "downloading it."
                ),
            },
            "user_agent": {
                "type": "string",
                "description": (
                    "Override the User-Agent. Some scholarly APIs serve their "
                    "polite pool only to callers that send a contactable one."
                ),
            },
        },
        "required": ["url"],
    }
    needs_network = True
    screen_untrusted_output = True

    def permission_target(self, arguments: Any) -> str:
        if not isinstance(arguments, dict):
            return ""
        url = str(arguments.get("url") or "")
        return re.sub(r"^https?://(www\.)?", "", url).split("/")[0]

    def resource_keys(self, arguments: Any) -> tuple[str, ...]:
        return (resource_key("network", self.permission_target(arguments)),)

    def execute(self, _runtime: Any, arguments: dict) -> dict:
        from openai4s import egress, webtools

        try:
            return webtools.web_fetch(
                arguments.get("url", ""),
                fmt=arguments.get("format", "markdown"),
                timeout=float(arguments.get("timeout") or 30),
                max_chars=int(arguments.get("max_chars") or 20000),
                method=arguments.get("method") or "GET",
                user_agent=arguments.get("user_agent") or None,
            )
        except (
            webtools.NetworkDisabled,
            webtools.SSRFBlocked,
            webtools.ResponseTooLarge,
            egress.EgressBlocked,
        ) as error:
            return {"error": str(error)}
        except Exception as error:  # noqa: BLE001
            return {"error": f"web_fetch: {error}"}


__all__ = ["WebFetchTool"]