@lmcc-dev/mult-fetch-mcp-server 1.0.1

package/dist/src/lib/server/prompts.js

@@ -0,0 +1,207 @@
+/**
+ * Author: Martin <lmccc.dev@gmail.com>
+ * Co-Author: AI Assistant (Claude)
+ * Description: This code was collaboratively developed by Martin and AI Assistant.
+ */
+import { ListPromptsRequestSchema, GetPromptRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { log, COMPONENTS } from '../logger.js';
+import { t } from '../i18n/index.js';
+// 定义提示模板 (Define prompt templates)
+const PROMPTS = {
+    "fetch-website": {
+        name: "mult-fetch-mcp:prompt:fetch-website",
+        description: t('prompts.fetchWebsite.description'),
+        arguments: [
+            {
+                name: "url",
+                description: t('prompts.url.description'),
+                required: true
+            },
+            {
+                name: "format",
+                description: t('prompts.format.description'),
+                required: false
+            },
+            {
+                name: "useBrowser",
+                description: t('prompts.useBrowser.description'),
+                required: false
+            }
+        ]
+    },
+    "extract-content": {
+        name: "mult-fetch-mcp:prompt:extract-content",
+        description: t('prompts.extractContent.description'),
+        arguments: [
+            {
+                name: "url",
+                description: t('prompts.url.description'),
+                required: true
+            },
+            {
+                name: "selector",
+                description: t('prompts.selector.description'),
+                required: false
+            },
+            {
+                name: "dataType",
+                description: t('prompts.dataType.description'),
+                required: false
+            }
+        ]
+    },
+    "debug-fetch": {
+        name: "mult-fetch-mcp:prompt:debug-fetch",
+        description: t('prompts.debugFetch.description'),
+        arguments: [
+            {
+                name: "url",
+                description: t('prompts.url.description'),
+                required: true
+            },
+            {
+                name: "error",
+                description: t('prompts.error.description'),
+                required: true
+            }
+        ]
+    }
+};
+/**
+ * 注册提示相关处理程序到服务器 (Register prompt handlers to server)
+ * @param server MCP服务器实例 (MCP server instance)
+ */
+export function registerPrompts(server) {
+    // 注册提示列表处理程序
+    server.setRequestHandler(ListPromptsRequestSchema, async (request) => {
+        const debug = request.params?.debug === true;
+        log('prompts.list.request', debug, { params: request.params }, COMPONENTS.PROMPTS);
+        return {
+            prompts: Object.values(PROMPTS)
+        };
+    });
+    // 注册获取提示处理程序
+    server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+        const promptName = request.params.name;
+        const args = request.params.arguments || {};
+        const debug = typeof args.debug === 'boolean' ? args.debug : false;
+        log('prompts.get.request', debug, { promptName, args }, COMPONENTS.PROMPTS);
+        // 检查提示是否存在
+        if (!PROMPTS[promptName]) {
+            throw new Error(t('prompts.notFound') + `: ${promptName}`);
+        }
+        // 检查必需参数
+        const prompt = PROMPTS[promptName];
+        const requiredArgs = prompt.arguments?.filter(arg => arg.required) || [];
+        for (const arg of requiredArgs) {
+            if (args[arg.name] === undefined) {
+                throw new Error(t('prompts.missingRequiredArg') + `: ${arg.name}`);
+            }
+        }
+        // 根据提示名称和参数生成消息
+        if (promptName === "fetch-website") {
+            const url = args.url;
+            const format = args.format || "html";
+            let useBrowser = false;
+            if (args.useBrowser !== undefined) {
+                if (typeof args.useBrowser === 'boolean') {
+                    useBrowser = args.useBrowser;
+                }
+                else if (typeof args.useBrowser === 'string') {
+                    const useBrowserStr = args.useBrowser.toLowerCase().trim();
+                    useBrowser = useBrowserStr === 'true' || useBrowserStr === 'yes' || useBrowserStr === '是' || useBrowserStr === '1';
+                }
+                else if (typeof args.useBrowser === 'number') {
+                    useBrowser = args.useBrowser !== 0;
+                }
+            }
+            log('prompts.useBrowserValue', debug, { original: args.useBrowser, parsed: useBrowser }, COMPONENTS.PROMPTS);
+            let toolName = "fetch_html";
+            if (format === "json")
+                toolName = "fetch_json";
+            if (format === "text")
+                toolName = "fetch_txt";
+            if (format === "markdown")
+                toolName = "fetch_markdown";
+            return {
+                description: t('prompts.fetchWebsite.result'),
+                messages: [
+                    {
+                        role: "user",
+                        content: {
+                            type: "text",
+                            text: t('prompts.fetchWebsite.message')
+                        }
+                    },
+                    {
+                        role: "assistant",
+                        content: {
+                            type: "text",
+                            text: t('prompts.fetchWebsite.response')
+                        }
+                    },
+                    {
+                        role: "user",
+                        content: {
+                            type: "text",
+                            text: `${t('prompts.fetchWebsite.instruction')}: ${url}\n\n` +
+                                `${t('prompts.fetchWebsite.formatInstruction')}: ${format}\n\n` +
+                                `${t('prompts.fetchWebsite.browserInstruction')}: ${useBrowser ? t('prompts.yes') : t('prompts.no')}`
+                        }
+                    }
+                ]
+            };
+        }
+        else if (promptName === "extract-content") {
+            const url = args.url;
+            const selector = args.selector || "main";
+            const dataType = args.dataType || "text";
+            return {
+                description: t('prompts.extractContent.result'),
+                messages: [
+                    {
+                        role: "user",
+                        content: {
+                            type: "text",
+                            text: `${t('prompts.extractContent.message')}: ${url}\n\n` +
+                                `${t('prompts.extractContent.selectorInstruction')}: ${selector}\n\n` +
+                                `${t('prompts.extractContent.dataTypeInstruction')}: ${dataType}`
+                        }
+                    }
+                ]
+            };
+        }
+        else if (promptName === "debug-fetch") {
+            const url = args.url;
+            const error = args.error;
+            return {
+                description: t('prompts.debugFetch.result'),
+                messages: [
+                    {
+                        role: "user",
+                        content: {
+                            type: "text",
+                            text: `${t('prompts.debugFetch.message')}: ${url}\n\n` +
+                                `${t('prompts.debugFetch.errorDetails')}: ${error}\n\n` +
+                                `${t('prompts.debugFetch.instruction')}`
+                        }
+                    }
+                ]
+            };
+        }
+        // 默认情况下返回一个通用消息
+        return {
+            description: t('prompts.generic.result'),
+            messages: [
+                {
+                    role: "user",
+                    content: {
+                        type: "text",
+                        text: `${t('prompts.generic.message')}: ${promptName}\n\n` +
+                            `${t('prompts.generic.args')}: ${JSON.stringify(args, null, 2)}`
+                    }
+                }
+            ]
+        };
+    });
+}

package/dist/src/lib/server/resources.js

@@ -0,0 +1,171 @@
+/**
+ * Author: Martin <lmccc.dev@gmail.com>
+ * Co-Author: AI Assistant (Claude)
+ * Description: This code was collaboratively developed by Martin and AI Assistant.
+ */
+import { ListResourcesRequestSchema, ReadResourceRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { log, getLogFilePath, clearLogFile, COMPONENTS } from '../logger.js';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+// 获取项目根目录路径 (Get project root directory path)
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const projectRoot = path.resolve(__dirname, '../../..');
+/**
+ * 注册资源相关处理程序到服务器 (Register resource handlers to server)
+ * @param server MCP服务器实例 (MCP server instance)
+ */
+export function registerResources(server) {
+    // 注册资源列表处理程序
+    server.setRequestHandler(ListResourcesRequestSchema, async (request) => {
+        const debug = request.params?.debug === true;
+        log('resources.list.request', debug, { params: request.params }, COMPONENTS.RESOURCES);
+        // 返回实际可用的资源列表
+        return {
+            resources: [
+                // 项目文档资源 (Project documentation resources)
+                {
+                    name: 'mult-fetch-mcp:log:debug-log',
+                    uri: 'file:///logs/debug',
+                    description: 'Debug log file containing all debug messages from the server'
+                },
+                // 清理日志资源 (Clear log resource)
+                {
+                    name: 'mult-fetch-mcp:log:clear-debug-log',
+                    uri: 'file:///logs/clear',
+                    description: 'Clear the debug log file'
+                }
+            ],
+            resourceTemplates: [
+                // 源代码文件模板 (Source code file template)
+                {
+                    uriTemplate: 'file:///src/{path}',
+                    description: 'Source code files in the project',
+                    name: 'mult-fetch-mcp:src:file'
+                },
+                // 文档文件模板 (Documentation file template)
+                {
+                    uriTemplate: 'file:///docs/{path}',
+                    description: 'Documentation files in the project',
+                    name: 'mult-fetch-mcp:docs:file'
+                }
+            ]
+        };
+    });
+    // 注册资源读取处理程序
+    server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
+        const uri = request.params.uri;
+        const debug = request.params?.debug === true;
+        log('resources.read.request', debug, { uri }, COMPONENTS.RESOURCES);
+        try {
+            // 解析URI并读取相应文件
+            const filePath = parseResourceUri(uri);
+            if (!filePath) {
+                throw new Error(`resources.invalidUri`);
+            }
+            const content = await readFileContent(filePath);
+            const mimeType = getMimeType(filePath);
+            return {
+                contents: [
+                    {
+                        uri,
+                        mimeType,
+                        text: content
+                    }
+                ]
+            };
+        }
+        catch (error) {
+            log('resources.readError', debug, { uri, error: error instanceof Error ? error.message : String(error) }, COMPONENTS.RESOURCES);
+            throw new Error(`resources.notFound: ${uri}`);
+        }
+    });
+}
+/**
+ * 解析资源URI为文件路径 (Parse resource URI to file path)
+ * @param uri 资源URI (Resource URI)
+ * @returns 文件路径或null (File path or null)
+ */
+function parseResourceUri(uri) {
+    // 处理日志文件URI
+    if (uri === 'file:///logs/debug') {
+        return getLogFilePath();
+    }
+    // 处理清理日志文件URI
+    if (uri === 'file:///logs/clear') {
+        return 'CLEAR_LOG'; // 特殊标记，表示清理日志文件
+    }
+    // 处理预定义的URI模式
+    if (uri.startsWith('file:///docs/')) {
+        const resourceName = uri.substring('file:///docs/'.length);
+        if (resourceName === 'readme') {
+            return path.join(projectRoot, 'README.md');
+        }
+        else if (resourceName === 'package') {
+            return path.join(projectRoot, 'package.json');
+        }
+        else {
+            return path.join(projectRoot, resourceName);
+        }
+    }
+    else if (uri.startsWith('file:///src/')) {
+        const resourceName = uri.substring('file:///src/'.length);
+        if (resourceName === 'index') {
+            return path.join(projectRoot, 'index.ts');
+        }
+        else if (resourceName === 'client') {
+            return path.join(projectRoot, 'src', 'client.ts');
+        }
+        else {
+            return path.join(projectRoot, 'src', resourceName);
+        }
+    }
+    return null;
+}
+/**
+ * 读取文件内容 (Read file content)
+ * @param filePath 文件路径 (File path)
+ * @returns 文件内容 (File content)
+ */
+async function readFileContent(filePath) {
+    try {
+        // 处理清理日志文件的特殊标记
+        if (filePath === 'CLEAR_LOG') {
+            clearLogFile();
+            return '日志文件已清理 (Log file has been cleared)';
+        }
+        return fs.readFileSync(filePath, 'utf-8');
+    }
+    catch (error) {
+        log('resources.fileReadError', true, { filePath, error: error instanceof Error ? error.message : String(error) }, COMPONENTS.RESOURCES);
+        throw new Error('resources.fileReadError');
+    }
+}
+/**
+ * 根据文件路径获取MIME类型 (Get MIME type based on file path)
+ * @param filePath 文件路径 (File path)
+ * @returns MIME类型 (MIME type)
+ */
+function getMimeType(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    switch (ext) {
+        case '.md':
+            return 'text/markdown';
+        case '.json':
+            return 'application/json';
+        case '.ts':
+            return 'text/typescript';
+        case '.js':
+            return 'text/javascript';
+        case '.html':
+            return 'text/html';
+        case '.css':
+            return 'text/css';
+        case '.txt':
+        case '.log':
+            return 'text/plain';
+        default:
+            return 'application/octet-stream';
+    }
+}