// Node.js 18+ | @modelcontextprotocol/sdk ^1.0.0

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

// 1. 创建 Server 实例
const server = new McpServer({
  name: 'calculator',
  version: '1.0.0',
})

// 2. 注册一个 Tool
server.registerTool(
  'add',
  {
    title: '加法计算',
    description: '计算两个数字的和',
    inputSchema: {
      a: z.number().describe('第一个数字'),
      b: z.number().describe('第二个数字'),
    },
    outputSchema: {
      result: z.number(),
    },
  },
  async ({ a, b }) => {
    const result = a + b
    return {
      content: [{ type: 'text', text: `${a} + ${b} = ${result}` }],
      structuredContent: { result },
    }
  }
)

// 3. 使用 stdio 传输启动 Server
const transport = new StdioServerTransport()
await server.connect(transport)
console.error('Calculator MCP Server running on stdio')

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

const server = new McpServer({
  name: 'calculator',
  version: '1.0.0',
})

// 通用的双参数数学运算 schema
const mathSchema = {
  inputSchema: {
    a: z.number().describe('第一个数字'),
    b: z.number().describe('第二个数字'),
  },
  outputSchema: {
    result: z.number(),
  },
}

// 加法
server.registerTool(
  'add',
  { title: '加法', description: '计算 a + b', ...mathSchema },
  async ({ a, b }) => {
    const result = a + b
    return {
      content: [{ type: 'text', text: `${a} + ${b} = ${result}` }],
      structuredContent: { result },
    }
  }
)

// 减法
server.registerTool(
  'subtract',
  { title: '减法', description: '计算 a - b', ...mathSchema },
  async ({ a, b }) => {
    const result = a - b
    return {
      content: [{ type: 'text', text: `${a} - ${b} = ${result}` }],
      structuredContent: { result },
    }
  }
)

// 乘法
server.registerTool(
  'multiply',
  { title: '乘法', description: '计算 a × b', ...mathSchema },
  async ({ a, b }) => {
    const result = a * b
    return {
      content: [{ type: 'text', text: `${a} × ${b} = ${result}` }],
      structuredContent: { result },
    }
  }
)

// 除法（带错误处理）
server.registerTool(
  'divide',
  { title: '除法', description: '计算 a ÷ b', ...mathSchema },
  async ({ a, b }) => {
    if (b === 0) {
      return {
        content: [{ type: 'text', text: '错误：除数不能为零' }],
        isError: true,
      }
    }
    const result = a / b
    return {
      content: [{ type: 'text', text: `${a} ÷ ${b} = ${result}` }],
      structuredContent: { result },
    }
  }
)

// 启动 Server
const transport = new StdioServerTransport()
await server.connect(transport)
console.error('Calculator MCP Server running on stdio')