@becrafter/prompt-manager 0.0.15 → 0.0.18

package/packages/server/app.js

@@ -0,0 +1,163 @@
+import express from 'express';
+import cors from 'cors';
+import path from 'path';
+import { randomUUID } from 'node:crypto';
+import { util } from './utils/util.js';
+import { config } from './utils/config.js';
+import { logger } from './utils/logger.js';
+import { adminRouter } from './api/admin.routes.js';
+import { openRouter } from './api/open.routes.js';
+import { getMcpServer } from './mcp/mcp.server.js';
+import { isInitializeRequest } from '@modelcontextprotocol/sdk/types.js';
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import { InMemoryEventStore } from '@modelcontextprotocol/sdk/examples/shared/inMemoryEventStore.js';
+
+
+const app = express();
+const adminUiRoot = util.getAdminUiRoot();
+
+// 全局中间件
+app.use(cors());
+app.use(express.json({ limit: '8mb' }));
+app.use(express.urlencoded({ extended: true }));
+
+
+// 为管理员界面提供静态文件服务 - 根路径
+app.use(config.adminPath, express.static(adminUiRoot));
+
+// 为管理员界面提供根路径访问（当用户访问 /admin 时显示 admin.html）
+app.get(config.adminPath, (req, res) => {
+  res.sendFile(path.join(adminUiRoot, 'admin.html'));
+});
+
+// 为管理员界面提供根路径访问（当用户访问 /admin/ 时显示 admin.html）
+app.get(config.adminPath + '/', (req, res) => {
+  res.sendFile(path.join(adminUiRoot, 'admin.html'));
+});
+
+
+// 注册后台API
+app.use('/adminapi', adminRouter);
+
+// 注册开放API
+app.use('/openapi', openRouter);
+
+
+const transports = {};
+const mcpServers = {}; // 存储每个会话的MCP服务器实例
+
+// 挂载MCP流式服务（独立路径前缀，避免冲突）
+app.all('/mcp', (req, res) => {
+  try {
+    let transport;
+    const sessionId = req.headers['mcp-session-id'] || '';
+    
+    if (sessionId && transports[sessionId]) {
+      const existingTransport = transports[sessionId];
+      // Check if the transport is of the correct type
+      if (existingTransport instanceof StreamableHTTPServerTransport) {
+        // Reuse existing transport
+        transport = existingTransport;
+      } else {
+        // Transport exists but is not a StreamableHTTPServerTransport (could be SSEServerTransport)
+        res.status(400).json({
+            jsonrpc: '2.0',
+            error: {
+                code: -32000,
+                message: 'Bad Request: Session exists but uses a different transport protocol'
+            },
+            id: null
+        });
+        return;
+      }
+    } else if (!sessionId && req.method === 'POST' && isInitializeRequest(req.body)) {
+      const eventStore = new InMemoryEventStore();
+      transport = new StreamableHTTPServerTransport({
+          sessionIdGenerator: () => randomUUID(),
+          eventStore, // Enable resumability
+          onsessioninitialized: sessionId => {
+              // Store the transport by session ID when session is initialized
+              console.log(`StreamableHTTP session initialized with ID: ${sessionId}`);
+              transports[sessionId] = transport;
+              
+              // 为新会话创建MCP服务器实例
+              mcpServers[sessionId] = getMcpServer();
+              mcpServers[sessionId].connect(transport);
+          }
+      });
+
+      // Set up onclose handler to clean up transport when closed
+      transport.onclose = () => {
+          const sid = transport.sessionId;
+          if (sid && transports[sid]) {
+              console.log(`Transport closed for session ${sid}, removing from transports map`);
+              delete transports[sid];
+              // 清理MCP服务器实例
+              if (mcpServers[sid]) {
+                  delete mcpServers[sid];
+              }
+          }
+      };
+
+      // 添加错误处理
+      transport.onerror = (error) => {
+          console.error('MCP Transport error:', error);
+      };
+    } else {
+      // Invalid request - no session ID or not initialization request
+      res.status(400).json({
+          jsonrpc: '2.0',
+          error: {
+              code: -32000,
+              message: 'Bad Request: No valid session ID provided'
+          },
+          id: null
+      });
+      return;
+    }
+    
+    // Handle the request with the transport
+    transport.handleRequest(req, res, req.body);
+  } catch (error) {
+    logger.error('Error handling MCP request: ' + error.message);
+    return res.status(500).json({
+        jsonrpc: '2.0',
+        error: {
+            code: -32603,
+            message: 'Internal server error'
+        },
+        id: null
+    });
+  }
+});
+
+// 错误处理中间件
+app.use((err, req, res, next) => {
+  console.error(`[服务器错误]: ${err.message}`);
+  res.status(500).send('Internal Server Error')
+});
+
+// 404处理
+app.use((req, res) => {
+  res.status(404).send('Not Found')
+});
+
+// 导出清理函数
+export function cleanupMcpSessions() {
+  console.log('Cleaning up MCP sessions');
+  // 清理所有活动的传输和服务器实例
+  for (const sessionId in transports) {
+    try {
+      if (transports[sessionId] && typeof transports[sessionId].close === 'function') {
+        transports[sessionId].close();
+      }
+    } catch (error) {
+      console.error(`Error closing transport for session ${sessionId}:`, error);
+    }
+    delete transports[sessionId];
+  }
+  
+  // 注意：mcpServers 对象在 app.js 中不可访问，需要在 server.js 中处理
+}
+
+export default app;

package/packages/server/{mcp.js → mcp/mcp.handler.js}

@@ -1,9 +1,10 @@
 // 导入自定义模块
-import { config } from './config.js';
-import { logger } from './logger.js';
+import { config } from '../utils/config.js';
+import { logger } from '../utils/logger.js';
+import { util } from '../utils/util.js';
 
 // 处理 get_prompt 工具调用
-export async function handleGetPrompt(args, promptManager) {
+export async function handleGetPrompt(args) {
   // 注意：这里为了兼容性，我们同时支持prompt_id和name参数
   const promptId = args.prompt_id || args.name;
   
@@ -11,6 +12,7 @@ export async function handleGetPrompt(args, promptManager) {
     throw new Error("缺少必需参数: prompt_id 或 name");
   }
   
+  const promptManager = await util.getPromptManager();
   const prompt = promptManager.getPrompt(promptId);
   if (!prompt) {
     throw new Error(`未找到ID为 "${promptId}" 的prompt`);
@@ -41,12 +43,13 @@ export async function handleGetPrompt(args, promptManager) {
 }
 
 // 处理 search_prompts 工具调用
-export async function handleSearchPrompts(args, promptManager) {
+export async function handleSearchPrompts(args) {
   // 注意：这里为了兼容性，我们同时支持title和name参数
   const searchTerm = args.title || args.name;
   
   const logLevel = config.getLogLevel();
-  const allPrompts = promptManager.getPrompts();
+  const promptManager = await util.getPromptManager();
+  let allPrompts = (await promptManager.loadPrompts()).prompts || [];
 
   // 如果搜索词为空，则返回所有提示词
   if (!searchTerm) {
@@ -67,7 +70,7 @@ export async function handleSearchPrompts(args, promptManager) {
     prompt.hasArguments = prompt.arguments && prompt.arguments.length > 0;
     return {
       prompt: prompt,
-      score: calculateSimilarityScore(searchTerm, prompt),
+      score: util.calculateSimilarityScore(searchTerm, prompt),
     };
   })
   .filter(result => result.score > 0) // 只返回有匹配的结果
@@ -98,9 +101,10 @@ export async function handleSearchPrompts(args, promptManager) {
 /**
  * 处理 reload_prompts 工具调用
  */
-export async function handleReloadPrompts(promptManager) {
+export async function handleReloadPrompts(args) {
   logger.info('重新加载prompts...');
   
+  const promptManager = await util.getPromptManager();
   const result = await promptManager.reloadPrompts();
   
   return {
@@ -125,8 +129,6 @@ export async function handleReloadPrompts(promptManager) {
 function formatResults(results = []) {
   if (!Array.isArray(results)) return [];
 
-  // console.log(results);
-
   return results.map(result => {
     const prompt = result.prompt ? result.prompt : result;
     const baseItem = {
@@ -161,74 +163,4 @@ function convertToText(result) {
       }
     ]
   };
-}
-
-/**
- * 计算搜索关键词与prompt的相似度得分
- * @param {string} searchTerm - 搜索关键词
- * @param {Object} prompt - prompt对象
- * @returns {number} 相似度得分 (0-100)
- */
-function calculateSimilarityScore(searchTerm, prompt) {
-  let totalScore = 0;
-  const searchLower = searchTerm ? searchTerm.toLowerCase() : '';
-  
-  // 搜索字段权重配置（专注于内容搜索，不包含ID检索）
-  const fieldWeights = {
-    name: 60,         // 名称权重高，是主要匹配字段
-    description: 40   // 描述权重适中，是辅助匹配字段
-  };
-  
-  // 计算name匹配得分
-  if (prompt && prompt.name && typeof prompt.name === 'string') {
-    const nameScore = getStringMatchScore(searchLower, prompt.name.toLowerCase());
-    totalScore += nameScore * fieldWeights.name;
-  }
-  
-  // 计算description匹配得分
-  if (prompt.description) {
-    const descScore = getStringMatchScore(searchLower, prompt.description.toLowerCase());
-    totalScore += descScore * fieldWeights.description;
-  }
-  
-  // 标准化得分到0-100范围
-  const maxPossibleScore = Object.values(fieldWeights).reduce((sum, weight) => sum + weight, 0);
-  return Math.round((totalScore / maxPossibleScore) * 100);
-}
-
-/**
- * 计算两个字符串的匹配得分
- * @param {string} search - 搜索词 (已转小写)
- * @param {string} target - 目标字符串 (已转小写)
- * @returns {number} 匹配得分 (0-1)
- */
-function getStringMatchScore(search, target) {
-  if (!search || !target) return 0;
-  
-  // 完全匹配得分最高
-  if (target === search) return 1.0;
-  
-  // 完全包含得分较高
-  if (target.includes(search)) return 0.8;
-  
-  // 部分词匹配
-  const searchWords = search.split(/\s+/).filter(word => word.length > 0);
-  const targetWords = target.split(/\s+/).filter(word => word.length > 0);
-  
-  let matchedWords = 0;
-  for (const searchWord of searchWords) {
-    for (const targetWord of targetWords) {
-      if (targetWord.includes(searchWord) || searchWord.includes(targetWord)) {
-        matchedWords++;
-        break;
-      }
-    }
-  }
-  
-  if (searchWords.length > 0) {
-    const wordMatchRatio = matchedWords / searchWords.length;
-    return wordMatchRatio * 0.6; // 部分词匹配得分
-  }
-  
-  return 0;
 }

package/packages/server/mcp/mcp.server.js

@@ -0,0 +1,82 @@
+import { config } from '../utils/config.js';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import {
+    handleGetPrompt,
+    handleSearchPrompts,
+    handleReloadPrompts
+} from './mcp.handler.js';
+
+class Server {
+    constructor() {
+        this.server = new McpServer(
+            {
+                name: 'Prompt Management Server',
+                version: config.getServerVersion()
+            },
+            { capabilities: { logging: {} } }
+        );
+    }
+
+    registerTools(tools) {
+        for (const tool of tools) {
+            this.server.tool(tool.name, tool.description, tool.inputSchema, tool.handler);
+        }
+    }
+
+    getServer() {
+        return this.server;
+    }
+}
+
+export const getMcpServer = () => {
+    const mcpServer = new Server();
+    mcpServer.registerTools([
+        {
+            name: 'get_prompt',
+            description: 'Retrieve the complete content of a specific prompt by its ID, including all messages, arguments, and metadata.',
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    prompt_id: {
+                        type: 'string',
+                        description: 'the unique identifier of the prompt to retrieve'
+                    }
+                },
+                required: ['prompt_id']
+            },
+            handler: async (args) => {
+                return handleGetPrompt(args);
+            }
+        },
+        {
+            name: 'search_prompts',
+            description: 'For keyword search matching, retrieve or return all prompts. When a corresponding prompt word is matched, utilize the prompt ID to invoke the get_prompt tool to query the complete content of the prompt word.',
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    name: {
+                        type: 'string',
+                        description: 'Optional name filter for prompts'
+                    }
+                },
+                required: []
+            },
+            handler: async (args) => {
+                return handleSearchPrompts(args);
+            }
+        },
+        // {
+        //     name: 'reload_prompts',
+        //     description: 'Force a reload of all preset prompts to overwrite the cache.',
+        //     inputSchema: {
+        //         type: 'object',
+        //         properties: {},
+        //         required: []
+        //     },
+        //     handler: async (args) => {
+        //         return handleReloadPrompts(args);
+        //     }
+        // }
+    ]);
+    return mcpServer.getServer();
+};

package/packages/server/middlewares/auth.middleware.js

@@ -0,0 +1,34 @@
+import { config } from '../utils/config.js';
+
+/**
+     * 管理员API中间件
+     * @param {*} req 
+     * @param {*} res 
+     * @param {*} next 
+     * @returns 
+     */
+export function adminAuthMiddleware(req, res, next) {
+    // 检查是否启用了管理员功能
+    if (!config.adminEnable) {
+        return res.status(404).json({ error: 'Admin功能未启用' });
+    }
+
+    // 检查Authorization请求头是否存在且格式正确
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ')) {
+        return res.status(401).json({ error: '未提供认证令牌' });
+    }
+
+    // 提取Bearer token
+    const token = authHeader.substring(7);
+
+    // 验证令牌是否在配置的管理员列表中
+    const admin = config.admins.find(a => a.token === token);
+    if (!admin) {
+        return res.status(401).json({ error: '无效的认证令牌' });
+    }
+
+    // 将管理员信息附加到请求对象，供后续中间件使用
+    req.admin = admin;
+    next();
+}