import json
import sys
from pathlib import Path
from typing import Any, Dict, List, Optional

from mcp_server import MCPServer, MCPTools, Request, Response

# 自定义工具函数
def calculate_bmi(height: float, weight: float) -> str:
    """计算BMI值并返回健康状态评估"""
    bmi = weight / (height * height)
    
    if bmi < 18.5:
        status = "偏瘦"
    elif bmi < 24:
        status = "正常"
    elif bmi < 28:
        status = "偏胖"
    else:
        status = "肥胖"
    
    return f"您的BMI值为: {bmi:.2f}，健康状态: {status}"

def list_files(directory: str) -> List[str]:
    """列出指定目录下的所有文件"""
    path = Path(directory)
    if not path.exists() or not path.is_dir():
        return ["目录不存在或不是有效目录"]
    
    return [str(file.name) for file in path.iterdir()]

def get_weather(city: str) -> str:
    """获取指定城市的天气信息"""
    # 实际项目中应调用气象API
    weather_data = {
        "北京": "晴朗，23°C",
        "上海": "多云，25°C",
        "广州": "小雨，28°C",
        "深圳": "晴朗，30°C"
    }
    
    return f"{city}的天气: {weather_data.get(city, '未知天气信息')}"

def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
    """将一种货币转换为另一种货币"""
    # 实际项目中应调用汇率API
    rates = {
        "CNY": {"USD": 0.14, "EUR": 0.13, "JPY": 20.82},
        "USD": {"CNY": 7.15, "EUR": 0.92, "JPY": 149.08},
        "EUR": {"CNY": 7.80, "USD": 1.09, "JPY": 162.60},
        "JPY": {"CNY": 0.048, "USD": 0.0067, "EUR": 0.0062}
    }
    
    if from_currency not in rates or to_currency not in rates[from_currency]:
        return "不支持的货币转换"
    
    converted = amount * rates[from_currency][to_currency]
    return f"{amount} {from_currency} = {converted:.2f} {to_currency}"

def handle_request(request: Request) -> Response:
    """处理MCP请求并返回响应"""
    query = request.query
    
    # 构建回复
    answer = f"收到您的问题: {query}\n\n"
    answer += "我可以帮助您处理代码问题，您也可以使用以下工具函数：\n"
    answer += "1. calculate_bmi - 计算BMI值\n"
    answer += "2. list_files - 列出目录文件\n"
    answer += "3. get_weather - 获取城市天气\n"
    answer += "4. convert_currency - 货币转换\n"
    
    return Response(answer=answer)

def main():
    """主函数，启动MCP服务器"""
    # 创建工具函数映射
    tools = MCPTools()
    tools.register("calculate_bmi", calculate_bmi)
    tools.register("list_files", list_files)
    tools.register("get_weather", get_weather)
    tools.register("convert_currency", convert_currency)
    
    # 创建并启动MCP服务器
    server = MCPServer(
        handle_request=handle_request,
        tools=tools,
        project_managed=True,  # 启用项目管理
        name="多功能助手"      # 服务器名称
    )
    
    # 启动服务器
    server.start()
    print("MCP服务器已启动，等待请求...")

if __name__ == "__main__":
    main()

import mcp

@mcp.tool()
def calculate_bmi(height: float, weight: float) -> str:
    """计算BMI值并返回健康状态评估"""
    bmi = weight / (height * height)
    if bmi < 18.5:
        status = "偏瘦"
    elif bmi < 24:
        status = "正常"
    elif bmi < 28:
        status = "偏胖"
    else:
        status = "肥胖"
    return f"您的BMI值为: {bmi:.2f}，健康状态: {status}"    
    # ... 其他工具函数

if __name__ == "__main__":
    print("多功能助手启动...")
    try:
        mcp.run(transport="stdio")
    except Exception as e:
        print(f"多功能助手启动失败：{str(e)}")

# 定义工具链
def analyze_project_health(project_dir):
    """分析项目健康状况的工具链"""
    # 1. 列出项目文件
    files = list_files(project_dir)
    
    # 2. 分析代码复杂度
    complexity_report = analyze_complexity(project_dir)
    
    # 3. 检查依赖更新
    dependencies_report = check_dependencies(project_dir)
    
    # 4. 汇总报告
    return {
        "files_count": len(files),
        "complexity": complexity_report,
        "dependencies": dependencies_report,
        "health_score": calculate_health_score(complexity_report, dependencies_report)
    }

# 注册复合工具
tools.register("project_health", analyze_project_health)