RT-Prompt-MCP

#!/usr/bin/env node /** @format */ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js" import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js" import { z } from "zod" // 导入我们的提示词补充服务 import { getBackendPrompts, getFrontendPrompts, getGeneralPrompts, getUiDesignPrompts, getRongtongBackendCrudSuggestions } from "./promptSuggestions.js" // 导入飞书文档获取函数 import { getContentByName } from "./feishu.js" // 导入coding内容获取函数 import { getCodingContent } from "./coding.js" // 创建MCP服务器实例 const server = new McpServer({ name: "rt-prompt-mcp", version: "1.0.0", description: "MCP Server for prompt engineering suggestions", }) // 修改获取后端相关提示词的工具 server.tool( "get_backend_suggestions", { context: z.string().optional().describe("当前上下文或任务描述"), databaseType: z.string().optional().describe("数据库类型，如MySQL、PostgreSQL等"), language: z.string().optional().describe("编程语言，如Java、Python等"), }, async ({ context, databaseType, language }) => { const suggestions = getBackendPrompts(context, databaseType, language) return { content: [ { type: "text", text: suggestions, }, ], } } ) // 修改获取前端相关提示词的工具 server.tool( "get_frontend_suggestions", { context: z.string().optional().describe("当前上下文或任务描述"), framework: z.string().optional().describe("前端框架，如React、Vue等"), deviceType: z.string().optional().describe("设备类型，如移动端、桌面端等"), }, async ({ context, framework, deviceType }) => { const suggestions = getFrontendPrompts(context, framework, deviceType) return { content: [ { type: "text", text: suggestions, }, ], } } ) // 修改获取通用提示词补充的工具 server.tool( "get_general_suggestions", { context: z.string().optional().describe("当前上下文或任务描述"), taskType: z.string().optional().describe("任务类型，如代码生成、文档生成等"), }, async ({ context, taskType }) => { const suggestions = getGeneralPrompts(context, taskType) return { content: [ { type: "text", text: suggestions, }, ], } } ) // 添加获取UI设计图转化的提示词补充工具 server.tool( "get_ui_design_suggestions", { context: z.string().optional().describe("当前上下文或任务描述"), designType: z.string().optional().describe("设计类型，如线框图、高保真原型图等"), platform: z.string().optional().describe("平台类型，如Web、iOS、Android等"), }, async ({ context, designType, platform }, extra) => { const suggestions = await getUiDesignPrompts(context, designType, platform) return { content: [ { type: "text", text: suggestions, }, ], } } ) // 新增：注册荣通后端 CRUD 工具 server.tool( "get_rt_crud_suggestions", { base_path: z .string() .optional() .describe("可选的 Java/Kotlin 根包路径，例如 'com.example.myapp' 或 'cn.teamy'。如果提供，将替换提示中默认的 'cn.teamy'。请使用点分隔。"), }, async ({ base_path }) => { const suggestionsArray = getRongtongBackendCrudSuggestions(base_path) return { content: [ { type: "text", text: suggestionsArray.length > 0 ? suggestionsArray[0] : "", }, ], } } ) // 新增：从飞书文档获取提示词工具 server.tool( "get_feishu_prompt", { prompt_name: z.string().describe("提示词名称，如'UI转化提示词'、'AI生成UI-3D风格'等"), }, async ({ prompt_name }) => { try { // 调用飞书API获取对应名称的内容 const content = await getContentByName(prompt_name) // 检查内容是否是coding.net链接 if (content && content.trim().includes("coding.net")) { // 使用正则表达式提取链接 const urlMatch = content.match(/(https?:\/\/[^\s]+coding\.net[^\s]+)/i) if (urlMatch && urlMatch[1]) { // 使用getCodingContent获取链接内容 const codingContent = await getCodingContent(urlMatch[1]) return { content: [ { type: "text", text: codingContent || `无法从链接获取内容: ${urlMatch[1]}`, }, ], } } } return { content: [ { type: "text", text: content || `未找到名为"${prompt_name}"的提示词`, }, ], } } catch (error) { return { content: [ { type: "text", text: `获取提示词失败: ${error instanceof Error ? error.message : String(error)}`, }, ], } } } ) // 资源: 提供关于如何使用此服务的信息 server.resource("info", "info://usage", async (uri) => ({ contents: [ { uri: uri.href, text: `# RT-Prompt-MCP 使用指南 本服务提供专业的提示词补充功能，可以帮助LLM生成更符合要求的内容。 ## 可用工具 1. get_backend_suggestions: 获取后端开发相关的提示词补充 2. get_frontend_suggestions: 获取前端开发相关的提示词补充 3. get_general_suggestions: 获取通用场景的提示词补充 4. get_ui_design_suggestions: 获取UI设计图转化相关的提示词补充 5. get_rt_crud_suggestions: 获取荣通后端标准 CRUD 开发规范提示词 6. get_feishu_prompt: 从飞书文档获取指定名称的提示词 ## 使用示例 例如，当您需要生成数据库表结构时，可以使用： \`\`\` 使用 get_backend_suggestions 工具，并提供以下参数: - context: "创建用户和订单的数据库表结构" - databaseType: "MySQL" - language: "SQL" \`\`\` 例如，当您需要将UI设计图转化为代码时，可以使用： \`\`\` 使用 get_ui_design_suggestions 工具，并提供以下参数: - context: "将设计图转换为响应式界面" - designType: "高保真原型图" - platform: "Web" \`\`\` 例如，当您需要获取飞书文档中的特定提示词时，可以使用： \`\`\` 使用 get_feishu_prompt 工具，并提供以下参数: - prompt_name: "UI转化提示词" \`\`\` `, }, ], })) // 启动服务器 async function main() { const transport = new StdioServerTransport() await server.connect(transport) } main().catch((error) => { process.exit(1) })