mcp-osp-prompt 1.0.0

package/resource-manager.js

@@ -0,0 +1,358 @@
+/**
+ * 资源管理器模块
+ * 
+ * 提供统一的资源访问API，用于获取projects和rules内容
+ * Task 2.1 - 资源访问接口层
+ */
+
+import fs from 'fs/promises';
+import path from 'path';
+import { CFG } from './config.js';
+
+/**
+ * 读取资源缓存
+ * @returns {Object|null} 缓存的资源对象
+ */
+export async function readResourceCache() {
+  const cacheFile = path.join(CFG.cacheDir, 'resource-cache.json');
+  
+  try {
+    const data = await fs.readFile(cacheFile, 'utf8');
+    const resources = JSON.parse(data);
+    
+    if (resources && resources.prompts && resources.projects && resources.rules) {
+      return resources;
+    }
+    
+    return null;
+  } catch (error) {
+    console.warn(`[ResourceManager] No resource cache found: ${error.message}`);
+    return null;
+  }
+}
+
+/**
+ * 获取项目列表
+ * @returns {Array} 项目文档列表
+ */
+export async function getProjectsList() {
+  try {
+    const cache = await readResourceCache();
+    
+    if (!cache || !cache.projects) {
+      console.warn('[ResourceManager] No projects found in cache');
+      return [];
+    }
+    
+    console.log(`[ResourceManager] Found ${cache.projects.length} projects`);
+    return cache.projects;
+  } catch (error) {
+    console.error(`[ResourceManager] Failed to get projects list: ${error.message}`);
+    return [];
+  }
+}
+
+/**
+ * 获取项目文档内容
+ * @param {String} projectName - 项目文件名 (如 'kiki-framework-wiki.md')
+ * @returns {String|null} 项目文档内容
+ */
+export async function getProjectContent(projectName) {
+  try {
+    // 项目文档缓存在 projects 子目录
+    const cacheFile = path.join(CFG.cacheDir, 'projects', projectName);
+    
+    const content = await fs.readFile(cacheFile, 'utf8');
+    console.log(`[ResourceManager] Loaded project: ${projectName} (${content.length} bytes)`);
+    
+    return content;
+  } catch (error) {
+    console.warn(`[ResourceManager] Failed to load project ${projectName}: ${error.message}`);
+    
+    // 尝试从远程拉取（TODO: 在后续优化中实现）
+    return null;
+  }
+}
+
+/**
+ * 根据语言获取规则文件信息
+ * @param {String} language - 语言名称 (如 'java', 'python')
+ * @returns {Object|null} 规则文件信息
+ */
+export async function getRulesByLanguage(language) {
+  try {
+    const cache = await readResourceCache();
+    
+    if (!cache || !cache.rules) {
+      console.warn('[ResourceManager] No rules found in cache');
+      return null;
+    }
+    
+    // 查找匹配的规则文件
+    const rule = cache.rules.find(r => r.language === language);
+    
+    if (rule) {
+      console.log(`[ResourceManager] Found rule for language: ${language}`);
+      return rule;
+    }
+    
+    console.warn(`[ResourceManager] No rule found for language: ${language}`);
+    return null;
+  } catch (error) {
+    console.error(`[ResourceManager] Failed to get rules for ${language}: ${error.message}`);
+    return null;
+  }
+}
+
+/**
+ * 获取规则文档内容
+ * @param {String} ruleFilename - 规则文件名 (如 'java-code-rules.md')
+ * @returns {String|null} 规则文档内容
+ */
+export async function getRuleContent(ruleFilename) {
+  try {
+    // 规则文档缓存在 rules 子目录
+    const cacheFile = path.join(CFG.cacheDir, 'rules', ruleFilename);
+    
+    const content = await fs.readFile(cacheFile, 'utf8');
+    console.log(`[ResourceManager] Loaded rule: ${ruleFilename} (${content.length} bytes)`);
+    
+    return content;
+  } catch (error) {
+    console.warn(`[ResourceManager] Failed to load rule ${ruleFilename}: ${error.message}`);
+    
+    // 尝试从远程拉取（TODO: 在后续优化中实现）
+    return null;
+  }
+}
+
+/**
+ * 根据语言获取规则内容（便捷函数）
+ * @param {String} language - 语言名称
+ * @returns {String|null} 规则内容
+ */
+export async function getRuleContentByLanguage(language) {
+  try {
+    const ruleInfo = await getRulesByLanguage(language);
+    
+    if (!ruleInfo) {
+      return null;
+    }
+    
+    return await getRuleContent(ruleInfo.filename);
+  } catch (error) {
+    console.error(`[ResourceManager] Failed to get rule content for ${language}: ${error.message}`);
+    return null;
+  }
+}
+
+/**
+ * 获取所有可用的规则语言列表
+ * @returns {Array} 语言名称数组
+ */
+export async function getAvailableLanguages() {
+  try {
+    const cache = await readResourceCache();
+    
+    if (!cache || !cache.rules) {
+      return [];
+    }
+    
+    const languages = cache.rules
+      .map(r => r.language)
+      .filter(Boolean);
+    
+    console.log(`[ResourceManager] Available languages: ${languages.join(', ')}`);
+    return languages;
+  } catch (error) {
+    console.error(`[ResourceManager] Failed to get available languages: ${error.message}`);
+    return [];
+  }
+}
+
+/**
+ * 检查资源是否已同步
+ * @returns {Boolean} true表示资源已同步
+ */
+export async function isResourcesSynced() {
+  try {
+    const cache = await readResourceCache();
+    return cache !== null;
+  } catch (error) {
+    return false;
+  }
+}
+
+/**
+ * 获取资源同步时间
+ * @returns {String|null} ISO时间字符串
+ */
+export async function getLastSyncTime() {
+  try {
+    const cache = await readResourceCache();
+    return cache ? cache.lastSync : null;
+  } catch (error) {
+    return null;
+  }
+}
+
+/**
+ * 🟢 GREEN: 读取文件的前几行作为摘要
+ * @param {string} filePath - 文件路径
+ * @param {number} maxLines - 最多读取的行数
+ * @returns {string} 摘要内容
+ */
+async function readFileSummary(filePath, maxLines = 2) {
+  try {
+    const content = await fs.readFile(filePath, 'utf-8');
+    const allLines = content.split('\n');
+    
+    // 去掉markdown标题符号和空行，获取前N行有效内容
+    const validLines = allLines
+      .map(line => line.replace(/^#+\s*/, '').trim())
+      .filter(line => line.length > 0)
+      .slice(0, maxLines); // 取前N行有效内容
+    
+    // 每行限制120字符，用换行符连接
+    const summary = validLines
+      .map(line => line.substring(0, 120))
+      .join('\n  '); // 第二行缩进以便区分
+    
+    return summary || 'No description available';
+  } catch (error) {
+    return 'Description not available';
+  }
+}
+
+/**
+ * 🟢 GREEN: 格式化resources列表为轻量级文本（供prompt使用）
+ * @param {Array<string>} resourceTypes - 需要的resource类型 ['projects', 'rules']
+ * @returns {string} 格式化的文本
+ */
+export async function formatResourcesList(resourceTypes = ['projects', 'rules']) {
+  try {
+    const cache = await readResourceCache();
+    if (!cache) {
+      return '# Available Resources\n\n(No resources cached yet)\n';
+    }
+    
+    let text = '# Available Resources\n\n';
+    
+    // 格式化projects
+    if (resourceTypes.includes('projects') && cache.projects && cache.projects.length > 0) {
+      text += '## Projects Documentation\n';
+      for (const project of cache.projects) {
+        // 尝试读取文件摘要
+        const projectPath = path.join(CFG.cacheDir, 'projects', project.filename);
+        const summary = await readFileSummary(projectPath);
+        text += `- **${project.filename}**: ${summary}\n`;
+      }
+      text += '\n';
+    }
+    
+    // 格式化rules
+    if (resourceTypes.includes('rules') && cache.rules && cache.rules.length > 0) {
+      text += '## Rules & Standards\n';
+      for (const rule of cache.rules) {
+        // 尝试读取文件摘要
+        const rulePath = path.join(CFG.cacheDir, 'rules', rule.filename);
+        const summary = await readFileSummary(rulePath);
+        text += `- **${rule.filename}**: ${summary}\n`;
+      }
+      text += '\n';
+    }
+    
+    text += '---\n\n';
+    text += '*To load full content of a resource, use the `load-resource` tool when needed.*\n\n';
+    
+    return text;
+  } catch (error) {
+    console.error('[ResourceManager] Failed to format resources list:', error.message);
+    return '# Available Resources\n\n(Error loading resources)\n';
+  }
+}
+
+/**
+ * 🟢 GREEN: 加载resource的完整内容
+ * @param {string} resourceName - 资源名称（不含路径，如 "java-rules.md"）
+ * @returns {Object} { success: boolean, content: string, error?: string }
+ */
+export async function loadResourceContent(resourceName) {
+  try {
+    const cache = await readResourceCache();
+    if (!cache) {
+      return { success: false, error: 'No resources cached' };
+    }
+    
+    // 查找resource在哪个类型中
+    let resourcePath = null;
+    let resourceType = null;
+    
+    // 检查projects
+    if (cache.projects.some(p => p.filename === resourceName)) {
+      resourcePath = path.join(CFG.cacheDir, 'projects', resourceName);
+      resourceType = 'project';
+    }
+    // 检查rules
+    else if (cache.rules.some(r => r.filename === resourceName)) {
+      resourcePath = path.join(CFG.cacheDir, 'rules', resourceName);
+      resourceType = 'rule';
+    }
+    
+    if (!resourcePath) {
+      return { success: false, error: `Resource "${resourceName}" not found in cache` };
+    }
+    
+    // 读取文件内容
+    const content = await fs.readFile(resourcePath, 'utf-8');
+    return {
+      success: true,
+      content: content,
+      type: resourceType,
+      filename: resourceName
+    };
+  } catch (error) {
+    return { success: false, error: error.message };
+  }
+}
+
+/**
+ * 获取资源统计信息
+ * @returns {Object} 统计信息
+ */
+export async function getResourceStats() {
+  try {
+    const cache = await readResourceCache();
+    
+    if (!cache) {
+      return {
+        synced: false,
+        prompts: 0,
+        projects: 0,
+        rules: 0,
+        total: 0,
+        lastSync: null
+      };
+    }
+    
+    return {
+      synced: true,
+      prompts: cache.prompts?.length || 0,
+      projects: cache.projects?.length || 0,
+      rules: cache.rules?.length || 0,
+      total: (cache.prompts?.length || 0) + (cache.projects?.length || 0) + (cache.rules?.length || 0),
+      lastSync: cache.lastSync
+    };
+  } catch (error) {
+    console.error(`[ResourceManager] Failed to get resource stats: ${error.message}`);
+    return {
+      synced: false,
+      prompts: 0,
+      projects: 0,
+      rules: 0,
+      total: 0,
+      lastSync: null
+    };
+  }
+}
+

package/server.js

@@ -0,0 +1,305 @@
+#!/usr/bin/env node
+import process from 'process';
+import { initializePrompts, getToolsConfiguration } from './prompt-manager.js';
+import { handleDevTool, handleDevManual, handleDevFeedback, handleLoadResource } from './tools.js';
+import { CFG } from './config.js';
+import { 
+  createSuccessResponse, 
+  createErrorResponse, 
+  MCP_ERROR_CODES, 
+  isNotification,
+  createToolDescription,
+
+  withErrorHandling,
+  SERVER_INFO,
+  SERVER_CAPABILITIES
+} from './utils.js';
+
+// 🛡️ 增强错误处理 - 防止进程崩溃导致Cursor显示红色状态
+const debug = process.env.DEBUG_LOG === 'true';
+
+process.on('uncaughtException', (error) => {
+  console.error('🚨 [MCP-Server] Uncaught Exception:', error.message);
+  if (debug) {
+    console.error(error.stack);
+  }
+  // 不要退出进程，继续服务
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('🚨 [MCP-Server] Unhandled Rejection:', reason);
+  // 不要退出进程，继续服务
+});
+
+// 📊 内存监控 - 防止内存泄漏
+if (debug) {
+  setInterval(() => {
+    const usage = process.memoryUsage();
+    if (usage.heapUsed > 100 * 1024 * 1024) { // 100MB
+      console.warn('⚠️ [MCP-Server] High memory usage:', Math.round(usage.heapUsed / 1024 / 1024) + 'MB');
+    }
+  }, 300000); // 每5分钟检查一次
+}
+
+// 🔄 优雅关闭处理
+process.on('SIGTERM', () => {
+  console.log('📴 [MCP-Server] Received SIGTERM, shutting down gracefully');
+  process.exit(0);
+});
+
+process.on('SIGINT', () => {
+  console.log('📴 [MCP-Server] Received SIGINT, shutting down gracefully');
+  process.exit(0);
+});
+
+// 全局工具配置和就绪状态
+let globalToolsConfig = [];
+let isServerReady = false;
+
+// 同步初始化函数 - 阻塞式等待初始化完成
+async function initializeServerSync() {
+  try {
+    console.error('[MCP-Server] Starting synchronous initialization...');
+    
+    // 同步初始化prompt管理器
+    globalToolsConfig = await initializePrompts();
+    
+    // 标记服务器就绪
+    isServerReady = true;
+    console.error(`[MCP-Server] ✅ Server ready with ${globalToolsConfig.length} tools`);
+    
+    return true;
+  } catch (error) {
+    console.error('[MCP-Server] ❌ Initialization failed:', error.message);
+    console.error('[MCP-Server] Server will not accept requests');
+    isServerReady = false;
+    
+    // 🟢 GREEN: Task 2.1 - 优雅降级而不是崩溃
+    console.error('[MCP-Server] ⚠️  Entering degraded mode due to initialization failure');
+    
+    // 设置基本的降级工具配置
+    globalToolsConfig = getBasicFallbackTools();
+    isServerReady = true; // 允许服务器继续运行
+    
+    console.error(`[MCP-Server] ✅ Degraded mode active with ${globalToolsConfig.length} basic tools`);
+    return false; // 表示降级模式
+  }
+}
+
+const handleRequest = withErrorHandling(async (req) => {
+  if (isNotification(req)) return null;
+  
+  // 检查服务器是否就绪
+  if (!isServerReady && (req.method === 'tools/list' || req.method === 'tools/call')) {
+    return createErrorResponse(req.id, MCP_ERROR_CODES.INTERNAL_ERROR, 
+      'MCP server still initializing, please wait');
+  }
+  
+  switch (req.method) {
+    case 'initialize':
+      return createSuccessResponse(req.id, {
+        protocolVersion: '2024-11-05',
+        capabilities: SERVER_CAPABILITIES,
+        serverInfo: SERVER_INFO
+      });
+      
+
+    case 'tools/list':
+      return createSuccessResponse(req.id, { 
+        tools: getToolsList() 
+      });
+      
+    case 'tools/call':
+      return await handleToolCall(req);
+      
+    default:
+      return createErrorResponse(req.id, MCP_ERROR_CODES.METHOD_NOT_FOUND, 
+        `Method ${req.method} not found`);
+  }
+});
+
+
+
+// 🟢 GREEN: Task 2.1 - 降级模式下的基础工具
+function getBasicFallbackTools() {
+  return [
+    {
+      name: 'dev-manual',
+      type: 'handler',
+      handler: 'handleDevManual',
+      description: '手动选择开发类型（降级模式）',
+      schema: {
+        type: 'object',
+        properties: {
+          random_string: { description: 'Dummy parameter for no-parameter tools', type: 'string' }
+        },
+        required: ['random_string']
+      }
+    },
+    {
+      name: 'dev-feedback',
+      type: 'handler',
+      handler: 'handleDevFeedback',
+      description: '反馈确认工具（降级模式）',
+      schema: {
+        type: 'object',
+        properties: {
+          title: { type: 'string' },
+          message: { type: 'string' },
+          options: { type: 'array', items: { type: 'string' } }
+        },
+        required: ['title', 'message']
+      }
+    }
+  ];
+}
+
+function getToolsList() {
+  // Generate tools from unified prompt manager
+  if (!isServerReady) {
+    console.warn('[MCP-Server] Warning: getToolsList called before initialization completed');
+    return [];
+  }
+  
+  return globalToolsConfig.map(toolConfig => 
+    createToolDescription(
+      toolConfig.name,
+      toolConfig.description,
+      toolConfig.schema
+    )
+  );
+}
+
+async function handleToolCall(req) {
+  const name = req.params?.name;
+  const args = req.params?.arguments || {};
+  let resultText;
+
+  
+  // Find tool in global configuration
+  const tool = globalToolsConfig.find(t => t.name === name);
+  if (!tool) {
+    return createErrorResponse(req.id, MCP_ERROR_CODES.INVALID_PARAMS, 'Unknown tool');
+  }
+  
+  try {
+    if (tool.type === 'handler') {
+      // Handle special tools with dedicated handlers
+      switch (tool.handler) {
+        case 'handleDevManual':
+          resultText = await handleDevManual();
+          break;
+        case 'handleDevFeedback':
+          const feedbackResult = await handleDevFeedback(args);
+          
+          // Format feedback result for AI consumption
+          if (typeof feedbackResult === 'object' && feedbackResult.hasUserInput) {
+            resultText = `📝 **用户反馈处理结果**
+
+**用户选择:** ${feedbackResult.action}
+**具体建议:** "${feedbackResult.input}"
+
+🔄 **AI指令:** 请根据用户的具体建议调整当前的执行计划。用户明确要求进行修改，请：
+1. 分析用户的建议内容
+2. 调整相应的实现方案 
+3. 说明具体如何应用这个建议
+4. 继续执行调整后的计划
+
+⚠️ **MANDATORY NEXT STEP REMINDER:** 在完成下一个实施步骤后，必须再次调用dev-feedback工具汇报进度并获取确认。
+
+**重要:** 不要忽略用户的建议，必须基于此建议修改后续的执行步骤。`;
+          } else {
+            resultText = `✅ **用户确认:** ${feedbackResult}
+
+继续按原计划执行。
+
+⚠️ **MANDATORY NEXT STEP REMINDER:** 在完成下一个实施步骤后，必须再次调用dev-feedback工具汇报进度并获取确认。每个步骤完成后都需要调用feedback工具 - 这是强制要求！`;
+          }
+          break;
+        case 'handleLoadResource':
+          resultText = await handleLoadResource(args);
+          break;
+        default:
+          throw new Error(`Unknown handler: ${tool.handler}`);
+      }
+    } else if (tool.type === 'prompt') {
+      // Handle prompt-based tools
+      const promptResult = await handleDevTool(name, args);
+      
+      // Add mandatory feedback reminder to all prompt-based tools
+      resultText = `${promptResult}
+
+⚠️ **CRITICAL WORKFLOW REQUIREMENT:**
+1. 在制定好plan后，第一步开始执行前，必须调用dev-feedback工具将计划展示给用户确认
+2. 每个实施步骤完成后，必须调用dev-feedback工具汇报进度并获取确认后再继续下一步
+3. 这是强制要求，不可跳过！每次调用feedback后会自动提醒下一步也需要再次调用feedback
+
+**请严格遵守此工作流程，确保每个步骤都有用户确认。**`;
+    } else {
+      throw new Error(`Unknown tool type: ${tool.type}`);
+    }
+    
+    return createSuccessResponse(req.id, {
+      content: [{ type: 'text', text: resultText }]
+    });
+  } catch (error) {
+    // 简洁的错误日志记录
+    console.error(`[MCP错误] 工具: ${name}, 错误: ${error.message}`);
+    
+    // 特殊处理：某些"错误"实际上是正常的用户交互结果
+    if (error.message.includes('用户要求调整计划')) {
+      console.log(`[MCP信息] 用户计划调整请求: ${error.message}`);
+      return createSuccessResponse(req.id, {
+        content: [{ type: 'text', text: error.message }]
+      });
+    }
+    
+    // 提供简洁直观的错误信息
+    let errorMsg = error.message;
+    
+    // 针对常见错误场景提供更清晰的信息
+    if (error.message.includes('fetch fail') || error.message.includes('API')) {
+      errorMsg = `网络请求失败 - ${error.message}`;
+    } else if (error.message.includes('RESOURCE_PATH')) {
+      errorMsg = `配置错误 - ${error.message}`;
+    } else if (error.message.includes('Unknown prompt file')) {
+      errorMsg = `文件未找到 - ${error.message}`;
+    } else if (error.message.includes('429') || error.message.includes('API速率限制')) {
+      // 429错误不应该导致MCP崩溃，这是正常的速率限制
+      errorMsg = `缓存已生效，临时API限制 - 功能正常可用`;
+    }
+    
+    return createErrorResponse(req.id, MCP_ERROR_CODES.INTERNAL_ERROR, errorMsg);
+  }
+}
+
+// 🟢 GREEN: Task 2.1 - 支持优雅降级的初始化
+const initSuccess = await initializeServerSync();
+if (!initSuccess) {
+  console.error('[MCP-Server] ⚠️  Running in degraded mode');
+}
+
+if (CFG.debug) {
+  console.error('[MCP-Server] Debug mode enabled');
+}
+
+// 初始化完成后才开始监听stdin
+process.stdin.setEncoding('utf8');
+let buffer = '';
+process.stdin.on('data', chunk => {
+  buffer += chunk;
+  let idx;
+  while ((idx = buffer.indexOf('\n')) >= 0) {
+    const line = buffer.slice(0, idx);
+    buffer = buffer.slice(idx + 1);
+    if (!line.trim()) continue;
+    let req;
+    try { req = JSON.parse(line); } catch { continue; }
+    handleRequest(req).then(res => {
+      if (res !== undefined && res !== null) process.stdout.write(JSON.stringify(res) + '\n');
+    }).catch(err => {
+      const id = req?.id;
+      if (id !== undefined) process.stdout.write(JSON.stringify({ jsonrpc: '2.0', id, error: { code: -32603, message: err.message } }) + '\n');
+    });
+  }
+});