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

// 建立用於爬取內部文件的 MCP 伺服器
const server = new McpServer({
  name: "internal-docs",
  version: "1.0.0"
});

const turndownService = new TurndownService();

// 新增爬取內部文件的工具
server.tool("get_doc",
  { url: z.string() },
  async ({ url }) => {
    try {
      const response = await fetch(url);
      const html = await response.text();
      
      // 將 HTML 轉換為 markdown
      const markdown = turndownService.turndown(html);
      
      return {
        content: [{ type: "text", text: markdown }]
      };
    } catch (error) {
      return {
        content: [{ type: "text", text: `爬取 ${url} 時發生錯誤：${error.message}` }]
      };
    }
  }
);

// 開始在 stdin 上接收訊息並在 stdout 上發送訊息
const transport = new StdioServerTransport();
await server.connect(transport);