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);