@zzp123/mcp-zentao 1.3.0 → 1.3.1

package/dist/index.js

@@ -990,49 +990,100 @@ server.tool("uploadFile", {
     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 }) => {
+}, async ({ 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}`;
+    const { execSync } = await import('child_process');
+    try {
+        // 读取系统剪贴板图片
+        let base64Data;
+        let fileBuffer;
+        let finalFilename;
+        // Windows: 使用 PowerShell 读取剪贴板
+        if (process.platform === 'win32') {
+            const psScript = `
+                Add-Type -AssemblyName System.Windows.Forms
+                $img = [System.Windows.Forms.Clipboard]::GetImage()
+                if ($img -ne $null) {
+                    $ms = New-Object System.IO.MemoryStream
+                    $img.Save($ms, [System.Drawing.Imaging.ImageFormat]::Png)
+                    [Convert]::ToBase64String($ms.ToArray())
+                } else {
+                    Write-Error "NoImage"
+                }
+            `;
+            const result = execSync(`powershell -Command "${psScript.replace(/\n/g, ' ')}"`, {
+                encoding: 'utf8',
+                maxBuffer: 10 * 1024 * 1024,
+                stdio: ['pipe', 'pipe', 'pipe']
+            }).trim();
+            if (!result || result.includes('NoImage') || result.includes('Error')) {
+                throw new Error('剪贴板中没有图片。请先复制图片，不要粘贴到输入框，直接告诉我"上传"。');
+            }
+            base64Data = result;
+            fileBuffer = Buffer.from(base64Data, 'base64');
+            finalFilename = filename || `clipboard_${Date.now()}.png`;
+        }
+        // macOS: 使用 pngpaste
+        else if (process.platform === 'darwin') {
+            const tempFile = path.join(process.cwd(), '.temp_clipboard.png');
+            try {
+                execSync(`pngpaste "${tempFile}"`);
+                fileBuffer = fs.readFileSync(tempFile);
+                fs.unlinkSync(tempFile);
+                finalFilename = filename || `clipboard_${Date.now()}.png`;
+            }
+            catch {
+                throw new Error('剪贴板中没有图片。请先安装 pngpaste: brew install pngpaste');
+            }
+        }
+        // Linux: 使用 xclip
+        else {
+            try {
+                const buffer = execSync('xclip -selection clipboard -t image/png -o', {
+                    maxBuffer: 10 * 1024 * 1024
+                });
+                fileBuffer = buffer;
+                finalFilename = filename || `clipboard_${Date.now()}.png`;
+            }
+            catch {
+                throw new Error('剪贴板中没有图片。请先安装 xclip: sudo apt-get install xclip');
+            }
+        }
+        // 创建 img 文件夹
+        const imgDir = path.join(process.cwd(), 'img');
+        if (!fs.existsSync(imgDir)) {
+            fs.mkdirSync(imgDir, { recursive: true });
+        }
+        // 保存到 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) }] };
     }
-    else {
-        // 直接的 base64 数据，没有 data URL 前缀
-        fileBuffer = Buffer.from(base64Data, 'base64');
-        finalFilename = filename || `clipboard_${Date.now()}.png`;
+    catch (error) {
+        return {
+            content: [{
+                    type: "text",
+                    text: `错误: ${error.message}\n\n提示：\n1. 确保已复制图片到剪贴板\n2. 不要粘贴到输入框，直接说"上传剪贴板图片"\n3. 或使用命令行: upload.bat`
+                }],
+            isError: true
+        };
     }
-    // 保存到 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.3.0",
+    "version": "1.3.1",
     "description": "禅道项目管理系统的高级API集成包，提供任务管理、Bug跟踪等功能的完整封装，专为Cursor IDE设计的MCP扩展",
     "main": "dist/index.js",
     "type": "module",

package/scripts/get-clipboard-base64.js

@@ -0,0 +1,93 @@
+#!/usr/bin/env node
+
+/**
+ * 获取剪贴板图片的 base64 编码
+ * 可以在 Claude Code 中使用 Bash 工具调用此脚本
+ */
+
+import { execSync } from 'child_process';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import fs from 'fs';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+async function getClipboardBase64() {
+    try {
+        let base64Data;
+
+        // Windows: 使用 PowerShell
+        if (process.platform === 'win32') {
+            const psScript = `
+                Add-Type -AssemblyName System.Windows.Forms
+                $img = [System.Windows.Forms.Clipboard]::GetImage()
+                if ($img -ne $null) {
+                    $ms = New-Object System.IO.MemoryStream
+                    $img.Save($ms, [System.Drawing.Imaging.ImageFormat]::Png)
+                    [Convert]::ToBase64String($ms.ToArray())
+                } else {
+                    Write-Error "No image in clipboard"
+                    exit 1
+                }
+            `;
+
+            try {
+                base64Data = execSync(`powershell -Command "${psScript.replace(/\n/g, ' ')}"`, {
+                    encoding: 'utf8',
+                    maxBuffer: 10 * 1024 * 1024,
+                    stdio: ['pipe', 'pipe', 'pipe']
+                }).trim();
+
+                if (!base64Data || base64Data.includes('Error')) {
+                    console.error('ERROR: No image in clipboard');
+                    process.exit(1);
+                }
+
+                // 输出完整的 data URL
+                console.log(`data:image/png;base64,${base64Data}`);
+
+            } catch (error) {
+                console.error('ERROR: Failed to read clipboard image');
+                process.exit(1);
+            }
+        }
+        // macOS: 使用 pngpaste
+        else if (process.platform === 'darwin') {
+            try {
+                execSync('which pngpaste', { stdio: 'ignore' });
+                const tempFile = join(__dirname, '.temp_clipboard.png');
+                execSync(`pngpaste "${tempFile}"`);
+                const buffer = fs.readFileSync(tempFile);
+                base64Data = buffer.toString('base64');
+                fs.unlinkSync(tempFile);
+                console.log(`data:image/png;base64,${base64Data}`);
+            } catch (error) {
+                console.error('ERROR: No image in clipboard or pngpaste not installed');
+                console.error('Install: brew install pngpaste');
+                process.exit(1);
+            }
+        }
+        // Linux: 使用 xclip
+        else {
+            try {
+                execSync('which xclip', { stdio: 'ignore' });
+                const buffer = execSync('xclip -selection clipboard -t image/png -o', {
+                    maxBuffer: 10 * 1024 * 1024
+                });
+                base64Data = buffer.toString('base64');
+                console.log(`data:image/png;base64,${base64Data}`);
+            } catch (error) {
+                console.error('ERROR: No image in clipboard or xclip not installed');
+                console.error('Install: sudo apt-get install xclip');
+                process.exit(1);
+            }
+        }
+
+    } catch (error) {
+        console.error('ERROR:', error.message);
+        process.exit(1);
+    }
+}
+
+getClipboardBase64();