rt-prompt-mcp/dist/index.js

MCP Server for prompt engineering suggestions

#!/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);
});