import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { z } from "zod";

interface WeatherData {
  city: string;
  temperature: number;
  condition: string;
  forecast: {
    date: string;
    condition: string;
    high: number;
    low: number;
  }[];
  lastUpdated: string;
}

// 模拟天气数据存储
const weatherDatabase: Record<string, WeatherData> = {
  "北京": {
    city: "北京",
    temperature: 22,
    condition: "晴朗",
    forecast: [
      { date: "2025-04-08", condition: "晴朗", high: 24, low: 15 },
      { date: "2025-04-09", condition: "大雪", high: 23, low: 14 },
      { date: "2025-04-10", condition: "大风", high: 20, low: 12 }
    ],
    lastUpdated: new Date().toISOString()
  },
  "上海": {
    city: "上海",
    temperature: 24,
    condition: "多云",
    forecast: [
      { date: "2025-04-08", condition: "多云", high: 25, low: 17 },
      { date: "2025-04-09", condition: "阴天", high: 24, low: 16 },
      { date: "2025-04-10", condition: "小雨", high: 22, low: 15 }
    ],
    lastUpdated: new Date().toISOString()
  },
  "广州": {
    city: "广州",
    temperature: 28,
    condition: "晴朗",
    forecast: [
      { date: "2025-04-08", condition: "晴朗", high: 30, low: 21 },
      { date: "2025-04-09", condition: "晴朗", high: 29, low: 21 },
      { date: "2025-04-10", condition: "多云", high: 28, low: 20 }
    ],
    lastUpdated: new Date().toISOString()
  }
};

// 创建 MCP 服务器
const server = new McpServer({
  name: "天气服务",
  version: "1.0.0",
  description: "提供城市天气信息的MCP服务"
});

// 定义资源：获取城市天气数据
server.resource(
  "weatherData",
  "weather://{city}",
  async (uri) => {
    // 从URI中提取城市名称
    const city = uri.pathname.split('/').pop() || "";
    
    if (!weatherDatabase[city]) {
      return {
        contents: [{
          uri: uri.href,
          text: `未找到城市 ${city} 的天气数据`
        }],
        isError: true
      };
    }

    return {
      contents: [{
        uri: uri.href,
        text: JSON.stringify(weatherDatabase[city], null, 2)
      }]
    };
  }
);

// 定义资源：获取所有支持的城市列表
server.resource(
  "cityList",
  "weather://cities",
  async (uri) => {
    return {
      contents: [{
        uri: uri.href,
        text: JSON.stringify(Object.keys(weatherDatabase), null, 2)
      }]
    };
  }
);

// 定义工具：获取天气预报
server.tool(
  "getWeatherForecast",
  { city: z.string().describe("要查询天气预报的城市名") },
  async ({ city }) => {
    if (!weatherDatabase[city]) {
      return {
        content: [{ 
          type: "text", 
          text: `城市 ${city} 的天气数据不存在` 
        }],
        isError: true
      };
    }

    const weather = weatherDatabase[city];
    return {
      content: [{ 
        type: "text", 
        text: `${city}天气预报:\n当前温度: ${weather.temperature}°C, 天气状况: ${weather.condition}\n未来三天预报:\n${
          weather.forecast.map(day => `${day.date}: ${day.condition}, ${day.low}°C - ${day.high}°C`).join('\n')
        }\n最后更新时间: ${weather.lastUpdated}`
      }]
    };
  }
);

// 定义工具：更新城市天气
server.tool(
  "updateWeather",
  { 
    city: z.string().describe("要更新天气的城市名"),
    temperature: z.number().describe("当前温度(°C)"),
    condition: z.string().describe("天气状况描述")
  },
  async ({ city, temperature, condition }) => {
    if (!weatherDatabase[city]) {
      // 如果城市不存在，则创建新记录
      weatherDatabase[city] = {
        city,
        temperature,
        condition,
        forecast: [
          { date: "2025-04-08", condition: condition, high: temperature + 2, low: temperature - 5 },
          { date: "2025-04-09", condition: condition, high: temperature + 1, low: temperature - 6 },
          { date: "2025-04-10", condition: condition, high: temperature, low: temperature - 7 }
        ],
        lastUpdated: new Date().toISOString()
      };
    } else {
      // 更新现有记录
      weatherDatabase[city].temperature = temperature;
      weatherDatabase[city].condition = condition;
      weatherDatabase[city].lastUpdated = new Date().toISOString();
    }

    return {
      content: [{ 
        type: "text", 
        text: `已成功更新${city}的天气数据: 温度 ${temperature}°C, 天气状况 ${condition}` 
      }]
    };
  }
);

// 定义提示词：天气查询
server.prompt(
  "weatherQuery",
  { city: z.string().describe("城市名称") },
  ({ city }) => ({
    messages: [{
      role: "user",
      content: {
        type: "text",
        text: `请提供${city}的详细天气信息，包括温度、天气状况和未来几天的预报。`
      }
    }]
  })
);

// 定义提示词：天气建议
server.prompt(
  "weatherAdvice",
  { 
    city: z.string().describe("城市名称"),
    activity: z.string().describe("计划的活动")
  },
  ({ city, activity }) => ({
    messages: [{
      role: "user",
      content: {
        type: "text",
        text: `请根据${city}的天气情况，给我关于${activity}的建议。考虑温度、天气状况等因素。`
      }
    }]
  })
);

// 启动服务器
async function main(){
  const transport = new StdioServerTransport();
  console.log("天气服务启动中...");
  
  try {
    await server.connect(transport);
    console.log("天气服务已连接");
  } catch (error) {
    console.error("服务器连接失败:", error);
  }
}

main().catch(console.error);

{
  "mcpServers": {
    "weather-mcp-server": {
      "type": "command",
      "command": "node /Users/jayzx/CursorProjects/cursor-mcp-demo/quickstart-resources/weather-mcp-server/build/index.js"
    }
  }
}