@zzp123/mcp-zentao 1.2.0 → 1.2.2

package/dist/index.js

@@ -932,21 +932,107 @@ server.tool("getModules", {
 });
 // 文件相关工具
 server.tool("uploadFile", {
-    filePath: z.string(),
+    filePath: z.string().optional(),
+    base64Data: z.string().optional(),
+    filename: z.string().optional(),
     uid: z.string().optional()
-}, async ({ filePath, uid }) => {
+}, async ({ filePath, base64Data, filename, uid }) => {
     if (!zentaoApi)
         throw new Error("Please initialize Zentao API first");
     const fs = await import('fs');
     const path = await import('path');
-    const fileBuffer = fs.readFileSync(filePath);
-    const filename = path.basename(filePath);
+    let fileBuffer;
+    let finalFilename;
+    let savedPath;
+    // 如果提供了 base64 数据（复制的图片）
+    if (base64Data) {
+        // 创建 img 文件夹（如果不存在）
+        const imgDir = path.join(process.cwd(), 'img');
+        if (!fs.existsSync(imgDir)) {
+            fs.mkdirSync(imgDir, { recursive: true });
+        }
+        // 处理 base64 数据
+        const matches = base64Data.match(/^data:image\/(\w+);base64,(.+)$/);
+        if (matches) {
+            const ext = matches[1];
+            const data = matches[2];
+            fileBuffer = Buffer.from(data, 'base64');
+            finalFilename = filename || `image_${Date.now()}.${ext}`;
+        }
+        else {
+            // 直接的 base64 数据，没有 data URL 前缀
+            fileBuffer = Buffer.from(base64Data, 'base64');
+            finalFilename = filename || `image_${Date.now()}.png`;
+        }
+        // 保存到 img 文件夹
+        savedPath = path.join(imgDir, finalFilename);
+        fs.writeFileSync(savedPath, fileBuffer);
+    }
+    // 如果提供了文件路径
+    else if (filePath) {
+        fileBuffer = fs.readFileSync(filePath);
+        finalFilename = path.basename(filePath);
+    }
+    else {
+        throw new Error("必须提供 filePath 或 base64Data 参数");
+    }
     const result = await zentaoApi.uploadFile({
         file: fileBuffer,
-        filename,
+        filename: finalFilename,
         uid
     });
-    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
+    const response = {
+        upload: result,
+    };
+    if (savedPath) {
+        response.savedPath = savedPath;
+    }
+    return { content: [{ type: "text", text: JSON.stringify(response, null, 2) }] };
+});
+server.tool("uploadImageFromClipboard", {
+    base64Data: z.string(),
+    filename: z.string().optional(),
+    uid: z.string().optional()
+}, async ({ base64Data, filename, uid }) => {
+    if (!zentaoApi)
+        throw new Error("Please initialize Zentao API first");
+    const fs = await import('fs');
+    const path = await import('path');
+    // 创建 img 文件夹（如果不存在）
+    const imgDir = path.join(process.cwd(), 'img');
+    if (!fs.existsSync(imgDir)) {
+        fs.mkdirSync(imgDir, { recursive: true });
+    }
+    let fileBuffer;
+    let finalFilename;
+    // 处理 base64 数据
+    const matches = base64Data.match(/^data:image\/(\w+);base64,(.+)$/);
+    if (matches) {
+        const ext = matches[1];
+        const data = matches[2];
+        fileBuffer = Buffer.from(data, 'base64');
+        finalFilename = filename || `clipboard_${Date.now()}.${ext}`;
+    }
+    else {
+        // 直接的 base64 数据，没有 data URL 前缀
+        fileBuffer = Buffer.from(base64Data, 'base64');
+        finalFilename = filename || `clipboard_${Date.now()}.png`;
+    }
+    // 保存到 img 文件夹
+    const savedPath = path.join(imgDir, finalFilename);
+    fs.writeFileSync(savedPath, fileBuffer);
+    // 上传到禅道
+    const result = await zentaoApi.uploadFile({
+        file: fileBuffer,
+        filename: finalFilename,
+        uid
+    });
+    const response = {
+        upload: result,
+        savedPath: savedPath,
+        message: `图片已保存到本地并上传到禅道`
+    };
+    return { content: [{ type: "text", text: JSON.stringify(response, null, 2) }] };
 });
 server.tool("downloadFile", {
     fileId: z.number(),

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@zzp123/mcp-zentao",
-    "version": "1.2.0",
+    "version": "1.2.2",
     "description": "禅道项目管理系统的高级API集成包，提供任务管理、Bug跟踪等功能的完整封装，专为Cursor IDE设计的MCP扩展",
     "main": "dist/index.js",
     "type": "module",
@@ -42,7 +42,7 @@
     },
     "dependencies": {
         "@modelcontextprotocol/sdk": "^1.6.1",
-        "@zzp123/mcp-zentao": "^1.1.0",
+        "@zzp123/mcp-zentao": "^1.2.1",
         "axios": "^1.6.7",
         "chalk": "^4.1.2",
         "form-data": "^4.0.4",