@becrafter/prompt-manager 0.1.22 → 0.2.2

package/packages/server/toolm/tool-execution.service.js

@@ -1,6 +1,6 @@
 /**
  * 工具执行服务
- * 
+ *
  * 职责：
  * 1. 处理工具执行模式
  * 2. 创建执行上下文
@@ -21,7 +21,7 @@ import { generateHelpInfo } from './tool-manual-generator.service.js';
  */
 export async function executeTool(toolName, parameters) {
   logger.info(`执行工具: ${toolName}`, { parameters });
-  
+
   try {
     const tool = toolLoaderService.getTool(toolName);
     const toolModule = tool.module;
@@ -31,9 +31,9 @@ export async function executeTool(toolName, parameters) {
 
     // 2. 创建工具执行上下文
     const toolContext = await createToolContext(toolName, toolModule);
-    
+
     // 3. 记录执行开始
-    toolContext.api.logger.info('执行开始', { 
+    toolContext.api.logger.info('执行开始', {
       tool: toolName,
       parameters: Object.keys(parameters)
     });
@@ -42,35 +42,33 @@ export async function executeTool(toolName, parameters) {
     let result;
     try {
       result = await toolContext.execute(parameters);
-      
+
       // 5. 记录执行完成
-      toolContext.api.logger.info('执行完成', { 
+      toolContext.api.logger.info('执行完成', {
         tool: toolName,
         success: true
       });
-      
     } catch (error) {
       // 记录执行错误
-      toolContext.api.logger.error('执行失败', { 
+      toolContext.api.logger.error('执行失败', {
         tool: toolName,
         error: error.message,
         stack: error.stack
       });
       throw error;
     }
-    
+
     logger.info(`工具执行成功: ${toolName}`);
-    
+
     // 6. 格式化并返回结果
     const content = formatToolResult(result, toolName);
     return { content };
-    
   } catch (error) {
     logger.error(`工具执行失败: ${toolName}`, { error: error.message });
-    
+
     const tool = toolLoaderService.getTool(toolName);
     const helpInfo = generateErrorHelpInfo(toolName, error, tool, parameters);
-    
+
     return createTextResponse(helpInfo);
   }
 }
@@ -82,40 +80,40 @@ export async function executeTool(toolName, parameters) {
 /**
  * 格式化工具执行结果
  * 根据结果类型自动识别并转换为 MCP 格式的内容
- * 
+ *
  * @param {any} result - 工具执行结果
  * @param {string} toolName - 工具名称
  * @returns {Array} MCP 格式的内容数组
  */
 function formatToolResult(result, toolName) {
   const content = [];
-  
+
   // 尝试提取图片数据
   const imageContent = extractImageContent(result);
   if (imageContent.length > 0) {
     content.push(...imageContent);
-    
+
     // 对于截图操作，只返回图片
     if (result?.method === 'takeScreenshot') {
       logger.info('截图成功，返回图片数据');
       return content;
     }
-    
+
     // 其他操作有图片时，添加文本信息
     if (result?.text) {
       content.push(createTextContent(result.text));
     } else if (result?.method) {
       content.push(createTextContent(`操作成功：${result.method}`));
     }
-    
+
     return content;
   }
-  
+
   // 处理 file-reader 工具的其他格式（二进制、媒体元信息等）
   if (result?.format && result?.content) {
     return [createJsonTextContent(result, toolName)];
   }
-  
+
   // 默认情况：返回 JSON 格式
   return [createJsonTextContent(result, toolName)];
 }
@@ -126,17 +124,17 @@ function formatToolResult(result, toolName) {
  * 1. chrome-devtools: result.images 数组
  * 2. filesystem: result.base64 + result.mimeType (图片类型)
  * 3. file-reader: result.format === 'image' + result.content.base64
- * 
+ *
  * @param {any} result - 工具执行结果
  * @returns {Array} 图片内容数组
  */
 function extractImageContent(result) {
   const images = [];
-  
+
   if (!result) {
     return images;
   }
-  
+
   // 情况1: chrome-devtools 工具返回的图片数组
   if (Array.isArray(result.images) && result.images.length > 0) {
     for (const image of result.images) {
@@ -146,18 +144,18 @@ function extractImageContent(result) {
     }
     return images;
   }
-  
+
   // 情况2: filesystem 工具的 read_media_file 返回的 base64 + mimeType
   if (result.base64 && result.mimeType && result.mimeType.startsWith('image/')) {
     images.push(createImageContent(result.base64, result.mimeType));
     logger.info('读取图片文件成功，返回图片数据');
     return images;
   }
-  
+
   // 情况3: file-reader 工具返回的图片格式
   if (result.format === 'image' && result.content?.base64 && result.content?.mimeType) {
     images.push(createImageContent(result.content.base64, result.content.mimeType));
-    
+
     // 如果有描述信息，也一并返回
     if (result.content.description) {
       images.push(createTextContent(result.content.description));
@@ -165,59 +163,63 @@ function extractImageContent(result) {
     logger.info('读取图片文件成功，返回图片数据');
     return images;
   }
-  
+
   return images;
 }
 
 /**
  * 创建图片类型的内容
- * 
+ *
  * @param {string} data - base64 编码的图片数据
  * @param {string} mimeType - 图片 MIME 类型
  * @returns {object} MCP 图片内容对象
  */
 function createImageContent(data, mimeType) {
   return {
-    type: "image",
-    data: data,
-    mimeType: mimeType
+    type: 'image',
+    data,
+    mimeType
   };
 }
 
 /**
  * 创建文本类型的内容
- * 
+ *
  * @param {string} text - 文本内容
  * @returns {object} MCP 文本内容对象
  */
 function createTextContent(text) {
   return {
-    type: "text",
-    text: text
+    type: 'text',
+    text
   };
 }
 
 /**
  * 创建 JSON 格式的文本内容
- * 
+ *
  * @param {any} result - 工具执行结果
  * @param {string} toolName - 工具名称
  * @returns {object} MCP 文本内容对象
  */
 function createJsonTextContent(result, toolName) {
   return createTextContent(
-    JSON.stringify({
-      success: true,
-      tool: toolName,
-      mode: 'execute',
-      result: result
-    }, null, 2)
+    JSON.stringify(
+      {
+        success: true,
+        tool: toolName,
+        mode: 'execute',
+        result
+      },
+      null,
+      2
+    )
   );
 }
 
 /**
  * 创建文本类型的响应
- * 
+ *
  * @param {string} text - 文本内容
  * @returns {object} MCP 格式的响应对象
  */
@@ -229,7 +231,7 @@ function createTextResponse(text) {
 
 /**
  * 生成错误帮助信息
- * 
+ *
  * @param {string} toolName - 工具名称
  * @param {Error} error - 错误对象
  * @param {object} tool - 工具对象
@@ -241,7 +243,7 @@ function generateErrorHelpInfo(toolName, error, tool, parameters) {
   if (isValidationErrorType(error.message)) {
     return generateHelpInfo(toolName, error, tool, parameters);
   }
-  
+
   // 检查是否是业务错误
   const businessErrors = tool?.businessErrors || [];
   for (const businessError of businessErrors) {
@@ -249,14 +251,14 @@ function generateErrorHelpInfo(toolName, error, tool, parameters) {
       return generateHelpInfo(toolName, error, tool, parameters, businessError);
     }
   }
-  
+
   // 其他错误也返回帮助信息
   return generateHelpInfo(toolName, error, tool, parameters);
 }
 
 /**
  * 判断是否是参数验证错误
- * 
+ *
  * @param {string} errorMessage - 错误消息
  * @returns {boolean} 是否是参数验证错误
  */
@@ -271,7 +273,6 @@ function isValidationErrorType(errorMessage) {
     /参数.*类型错误/i,
     /参数.*格式错误/i
   ];
-  
+
   return validationPatterns.some(pattern => pattern.test(errorMessage));
 }
-

package/packages/server/toolm/tool-loader.service.js

@@ -1,6 +1,6 @@
 /**
  * 工具加载服务
- * 
+ *
  * 职责：
  * 1. 扫描并加载 ~/.prompt-manager/tools 和 packages/resources/tools 目录下的所有工具
  * 2. 验证工具是否符合标准接口规范
@@ -11,26 +11,22 @@
 
 import fs from 'fs-extra';
 import path from 'path';
-import os from 'os';
-import { fileURLToPath } from 'url';
 import { logger } from '../utils/logger.js';
 import { pathExists } from './tool-utils.js';
 import { generateManual as generateManualFromService } from './tool-manual-generator.service.js';
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
+import { config } from '../utils/config.js';
 
 class ToolLoaderService {
   constructor() {
     // 工具缓存：toolName -> toolModule
     this.toolCache = new Map();
-    
+
     // 工具目录列表（所有工具都在沙箱环境中）
     this.toolDirectories = [
       // 沙箱工具目录（系统工具和用户工具都在这里）
-      path.join(os.homedir(), '.prompt-manager', 'toolbox')
+      config.getToolboxDir()
     ];
-    
+
     // 已初始化标志
     this.initialized = false;
   }
@@ -45,14 +41,14 @@ class ToolLoaderService {
     }
 
     logger.info('初始化工具加载器...');
-    
+
     // 确保工具箱目录存在
-    const toolboxDir = path.join(os.homedir(), '.prompt-manager', 'toolbox');
+    const toolboxDir = config.getToolboxDir();
     await fs.ensureDir(toolboxDir);
-    
+
     // 扫描并加载所有工具
     await this.scanAndLoadTools();
-    
+
     this.initialized = true;
     logger.info(`工具加载器初始化完成，共加载 ${this.toolCache.size} 个工具`);
   }
@@ -62,14 +58,14 @@ class ToolLoaderService {
    */
   async scanAndLoadTools() {
     for (const toolsDir of this.toolDirectories) {
-      if (!await pathExists(toolsDir)) {
+      if (!(await pathExists(toolsDir))) {
         logger.debug(`工具目录不存在，跳过: ${toolsDir}`);
         continue;
       }
 
       try {
         const entries = await fs.readdir(toolsDir, { withFileTypes: true });
-        
+
         for (const entry of entries) {
           if (!entry.isDirectory()) {
             continue;
@@ -78,7 +74,7 @@ class ToolLoaderService {
           const toolName = entry.name;
           const toolDir = path.join(toolsDir, toolName);
           const toolFile = path.join(toolDir, `${toolName}.tool.js`);
-          
+
           if (await pathExists(toolFile)) {
             try {
               await this.loadTool(toolName, toolFile);
@@ -129,7 +125,7 @@ class ToolLoaderService {
   validateToolInterface(toolName, tool) {
     // 必需的方法
     const requiredMethods = ['execute'];
-    
+
     // 推荐的方法
     const recommendedMethods = ['getMetadata', 'getSchema', 'getDependencies', 'getBusinessErrors'];
 
@@ -216,4 +212,3 @@ class ToolLoaderService {
 
 // 导出单例实例
 export const toolLoaderService = new ToolLoaderService();
-

package/packages/server/toolm/tool-logger.service.js

@@ -1,6 +1,6 @@
 /**
  * 工具日志记录服务
- * 
+ *
  * 职责：
  * 1. 为每个工具提供独立的日志记录器
  * 2. 日志自动写入到工具的 run.log 文件
@@ -10,9 +10,9 @@
 
 import fs from 'fs-extra';
 import path from 'path';
-import os from 'os';
 import { logger } from '../utils/logger.js';
 import { pathExists } from './tool-utils.js';
+import { config } from '../utils/config.js';
 
 // 日志队列，用于批量写入
 const logQueues = new Map();
@@ -38,27 +38,25 @@ export function getLogger(toolName) {
   if (!logQueues.has(toolName)) {
     logQueues.set(toolName, []);
   }
-  
+
   const logQueue = logQueues.get(toolName);
-  
+
   return {
     info: (message, meta) => log(toolName, 'INFO', message, meta),
     warn: (message, meta) => log(toolName, 'WARN', message, meta),
     error: (message, meta) => log(toolName, 'ERROR', message, meta),
     debug: (message, meta) => log(toolName, 'DEBUG', message, meta)
   };
-  
+
   function log(toolName, level, message, meta = {}) {
     const timestamp = new Date().toISOString();
-    const metaStr = meta && Object.keys(meta).length > 0 
-      ? ' ' + JSON.stringify(meta)
-      : '';
-    
+    const metaStr = meta && Object.keys(meta).length > 0 ? ` ${JSON.stringify(meta)}` : '';
+
     const logEntry = `[${timestamp}] [${level}] ${message}${metaStr}\n`;
-    
+
     // 添加到队列
     logQueue.push(logEntry);
-    
+
     // 如果队列达到一定大小，立即刷新
     if (logQueue.length >= 10) {
       flushLogQueue(toolName).catch(error => {
@@ -84,38 +82,35 @@ async function flushLogQueue(toolName) {
   if (!logQueue || logQueue.length === 0) {
     return;
   }
-  
-  const toolDir = path.join(os.homedir(), '.prompt-manager', 'toolbox', toolName);
+
+  const toolDir = config.getToolDir(toolName);
   const logFilePath = path.join(toolDir, 'run.log');
-  
+
   try {
     // 确保目录存在
     await fs.ensureDir(toolDir);
-    
+
     // 读取现有日志（如果存在）
     let existingLogs = '';
     if (await pathExists(logFilePath)) {
       existingLogs = await fs.readFile(logFilePath, 'utf-8');
     }
-    
+
     // 清理过期日志
     const cleanedLogs = cleanOldLogs(existingLogs);
-    
+
     // 合并新日志
     const newLogs = logQueue.join('');
     const allLogs = cleanedLogs + newLogs;
-    
+
     // 检查日志文件大小
-    const logsToWrite = allLogs.length > MAX_LOG_SIZE
-      ? keepRecentLogs(allLogs)
-      : allLogs;
-    
+    const logsToWrite = allLogs.length > MAX_LOG_SIZE ? keepRecentLogs(allLogs) : allLogs;
+
     // 写入日志文件
     await fs.writeFile(logFilePath, logsToWrite, 'utf-8');
-    
+
     // 清空队列
     logQueue.length = 0;
-    
   } catch (error) {
     logger.error(`写入工具日志失败: ${toolName}`, { error: error.message });
   }
@@ -130,15 +125,15 @@ function cleanOldLogs(logs) {
   const lines = logs.split('\n');
   const now = Date.now();
   const retentionMs = LOG_RETENTION_HOURS * 60 * 60 * 1000;
-  
+
   const validLines = [];
-  
+
   for (const line of lines) {
     if (!line.trim()) {
       validLines.push(line);
       continue;
     }
-    
+
     // 提取时间戳
     const timestampMatch = line.match(/^\[([^\]]+)\]/);
     if (timestampMatch) {
@@ -156,7 +151,7 @@ function cleanOldLogs(logs) {
       validLines.push(line);
     }
   }
-  
+
   return validLines.join('\n');
 }
 
@@ -168,7 +163,7 @@ function cleanOldLogs(logs) {
 function keepRecentLogs(logs) {
   const lines = logs.split('\n');
   const recentLines = lines.slice(-MAX_LOG_LINES);
-  return recentLines.join('\n') + '\n';
+  return `${recentLines.join('\n')}\n`;
 }
 
 /**
@@ -176,27 +171,27 @@ function keepRecentLogs(logs) {
  */
 export function startLogCleanupTask() {
   setInterval(async () => {
-    const toolboxDir = path.join(os.homedir(), '.prompt-manager', 'toolbox');
-    
-    if (!await pathExists(toolboxDir)) {
+    const toolboxDir = config.getToolboxDir();
+
+    if (!(await pathExists(toolboxDir))) {
       return;
     }
-    
+
     try {
       const entries = await fs.readdir(toolboxDir, { withFileTypes: true });
-      
+
       for (const entry of entries) {
         if (!entry.isDirectory()) {
           continue;
         }
-        
+
         const toolName = entry.name;
         const logFilePath = path.join(toolboxDir, toolName, 'run.log');
-        
+
         if (await pathExists(logFilePath)) {
           const logs = await fs.readFile(logFilePath, 'utf-8');
           const cleanedLogs = cleanOldLogs(logs);
-          
+
           if (cleanedLogs.length < logs.length) {
             await fs.writeFile(logFilePath, cleanedLogs, 'utf-8');
             logger.debug(`清理工具日志: ${toolName}`);
@@ -207,7 +202,7 @@ export function startLogCleanupTask() {
       logger.error('定期清理日志失败', { error: error.message });
     }
   }, CLEANUP_INTERVAL);
-  
+
   logger.info('工具日志清理任务已启动');
 }
 
@@ -220,4 +215,3 @@ export async function flushAllLogQueues() {
     await flushLogQueue(toolName);
   }
 }
-

package/packages/server/toolm/tool-manager.handler.js

@@ -1,20 +1,19 @@
 /**
  * 统一工具管理处理器
- * 
+ *
  * 职责：
  * 1. 实现 mcp__promptmanager__toolm 工具的处理逻辑
  * 2. 路由到不同的模式处理器
  * 3. 统一错误处理
  */
 
-import { logger } from '../utils/logger.js';
 import { toolLoaderService } from './tool-loader.service.js';
 import { parseToolYaml } from './tool-yaml-parser.service.js';
-import { 
-  handleManualMode, 
-  handleExecuteMode, 
-  handleConfigureMode, 
-  handleLogMode 
+import {
+  handleManualMode,
+  handleExecuteMode,
+  handleConfigureMode,
+  handleLogMode
 } from './tool-mode-handlers.service.js';
 
 /**
@@ -24,7 +23,8 @@ import {
  */
 export async function handleToolM(args) {
   const { yaml: yamlInput } = args;
-  
+
+  /* eslint-disable no-useless-catch */
   try {
     // 初始化工具加载器（如果尚未初始化）
     if (!toolLoaderService.initialized) {
@@ -33,10 +33,13 @@ export async function handleToolM(args) {
 
     // 解析 YAML 配置
     const { toolName, mode, parameters } = parseToolYaml(yamlInput);
-    
+
     // 检查工具是否存在
     if (!toolLoaderService.hasTool(toolName)) {
-      const availableTools = toolLoaderService.getAllTools().map(t => t.name).join(', ');
+      const availableTools = toolLoaderService
+        .getAllTools()
+        .map(t => t.name)
+        .join(', ');
       throw new Error(`工具 '${toolName}' 不存在\n可用工具: ${availableTools}`);
     }
 
@@ -44,20 +47,19 @@ export async function handleToolM(args) {
     switch (mode) {
       case 'manual':
         return handleManualMode(toolName);
-      
+
       case 'execute':
         return await handleExecuteMode(toolName, parameters);
-      
+
       case 'configure':
         return await handleConfigureMode(toolName, parameters);
-      
+
       case 'log':
         return await handleLogMode(toolName, parameters);
-      
+
       default:
         throw new Error(`不支持的模式: ${mode}\n支持的模式: manual, execute, configure, log`);
     }
-
   } catch (error) {
     // 错误已经在 parseToolYaml 中处理，直接抛出
     throw error;