import requests
import uvicorn
from mcp.server.fastmcp.server import FastMCP
from typing import Optional
import logging

# 配置日志记录
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 1. 使用 FastMCP 创建一个 Server 实例
mcp = FastMCP(
    "APICallerServer",
    title="API Caller Server",
    description="A server that provides a tool to call APIs.",
    version="0.2.0"  # a new version
)

# 2. 使用 @mcp.tool() 装饰器来定义一个工具
@mcp.tool()
def api_caller(method: str, url: str, headers: Optional[dict] = None, params: Optional[dict] = None, json_body: Optional[dict] = None) -> dict:
    """
    一个通用的API调用工具，可以发送HTTP请求。
    """
    logging.info(f"api_caller: Received request -> method={method}, url={url}, params={params}, json_body={json_body}")
    try:
        response = requests.request(
            method=method,
            url=url,
            headers=headers,
            params=params,
            json=json_body
        )
        response.raise_for_status()  # 如果状态码是 4xx 或 5xx，则引发HTTPError

        logging.info(f"api_caller: Request to {url} successful with status code {response.status_code}")

        # 尝试将响应解析为JSON，如果失败则作为纯文本返回
        try:
            response_body = response.json()
        except requests.exceptions.JSONDecodeError:
            response_body = response.text
        
        return {
            "status_code": response.status_code,
            "headers": dict(response.headers),
            "body": response_body
        }
    except requests.exceptions.RequestException as e:
        logging.error(f"api_caller: Request to {url} failed. Error: {e}", exc_info=True)
        return {
            "error": "APIRequestError",
            "message": str(e)
        }

# 3. (可选) 如果直接运行此文件，则启动服务器
if __name__ == "__main__":
    # FastMCP对象本身不是ASGI应用，但它的 streamable_http_app() 方法会返回一个
    uvicorn.run(mcp.streamable_http_app(), host="127.0.0.1", port=8001)