@zzp123/mcp-zentao 1.2.2 → 1.3.1

package/scripts/upload-clipboard-image.ps1

@@ -0,0 +1,128 @@
+# 从剪贴板获取图片并上传到禅道
+# 使用方法: .\upload-clipboard-image.ps1
+#
+# 参数:
+# -ZentaoUrl: 禅道系统URL
+# -Username: 用户名
+# -Password: 密码
+# -Uid: 可选的UID参数
+
+param(
+    [Parameter(Mandatory=$true)]
+    [string]$ZentaoUrl,
+
+    [Parameter(Mandatory=$true)]
+    [string]$Username,
+
+    [Parameter(Mandatory=$true)]
+    [string]$Password,
+
+    [Parameter(Mandatory=$false)]
+    [string]$Uid = ""
+)
+
+# 添加必要的程序集
+Add-Type -AssemblyName System.Windows.Forms
+Add-Type -AssemblyName System.Drawing
+
+Write-Host "正在读取剪贴板图片..." -ForegroundColor Cyan
+
+# 从剪贴板获取图片
+$img = [System.Windows.Forms.Clipboard]::GetImage()
+
+if ($img -eq $null) {
+    Write-Host "错误: 剪贴板中没有图片" -ForegroundColor Red
+    exit 1
+}
+
+Write-Host "✓ 成功读取剪贴板图片 ($($img.Width)x$($img.Height))" -ForegroundColor Green
+
+# 转换为 base64
+$ms = New-Object System.IO.MemoryStream
+$img.Save($ms, [System.Drawing.Imaging.ImageFormat]::Png)
+$base64 = [Convert]::ToBase64String($ms.ToArray())
+$ms.Close()
+
+Write-Host "✓ 图片已转换为 base64" -ForegroundColor Green
+
+# 创建 img 文件夹
+$imgDir = Join-Path (Get-Location) "img"
+if (-not (Test-Path $imgDir)) {
+    New-Item -ItemType Directory -Path $imgDir | Out-Null
+    Write-Host "✓ 创建 img 文件夹: $imgDir" -ForegroundColor Green
+}
+
+# 保存到本地
+$filename = "clipboard_$([DateTimeOffset]::Now.ToUnixTimeMilliseconds()).png"
+$localPath = Join-Path $imgDir $filename
+$bytes = [Convert]::FromBase64String($base64)
+[System.IO.File]::WriteAllBytes($localPath, $bytes)
+Write-Host "✓ 图片已保存到: $localPath" -ForegroundColor Green
+
+# 获取 Token
+Write-Host "正在连接禅道系统..." -ForegroundColor Cyan
+$passwordMd5 = [System.BitConverter]::ToString(
+    [System.Security.Cryptography.MD5]::Create().ComputeHash(
+        [System.Text.Encoding]::UTF8.GetBytes($Password)
+    )
+).Replace("-", "").ToLower()
+
+$tokenUrl = "$ZentaoUrl/zentao/api.php/v1/tokens"
+$tokenBody = @{
+    account = $Username
+    password = $passwordMd5
+} | ConvertTo-Json
+
+try {
+    $tokenResponse = Invoke-RestMethod -Uri $tokenUrl -Method Post -Body $tokenBody -ContentType "application/json"
+    $token = $tokenResponse.token
+    Write-Host "✓ 成功获取 Token" -ForegroundColor Green
+} catch {
+    Write-Host "错误: 无法获取 Token - $($_.Exception.Message)" -ForegroundColor Red
+    exit 1
+}
+
+# 上传文件
+Write-Host "正在上传到禅道..." -ForegroundColor Cyan
+
+$uploadUrl = "$ZentaoUrl/zentao/api.php/v1/files"
+if ($Uid) {
+    $uploadUrl += "?uid=$Uid"
+}
+
+# 创建 multipart/form-data
+$boundary = [System.Guid]::NewGuid().ToString()
+$LF = "`r`n"
+
+$bodyLines = (
+    "--$boundary",
+    "Content-Disposition: form-data; name=`"imgFile`"; filename=`"$filename`"",
+    "Content-Type: image/png$LF",
+    [System.Text.Encoding]::GetEncoding("iso-8859-1").GetString($bytes),
+    "--$boundary--$LF"
+) -join $LF
+
+try {
+    $response = Invoke-RestMethod -Uri $uploadUrl -Method Post `
+        -Headers @{ Token = $token } `
+        -ContentType "multipart/form-data; boundary=$boundary" `
+        -Body $bodyLines
+
+    Write-Host ""
+    Write-Host "========================================" -ForegroundColor Green
+    Write-Host "✓ 上传成功！" -ForegroundColor Green
+    Write-Host "========================================" -ForegroundColor Green
+    Write-Host "文件ID: $($response.id)"
+    Write-Host "访问URL: $($response.url)"
+    Write-Host "本地路径: $localPath"
+    Write-Host "========================================" -ForegroundColor Green
+
+} catch {
+    Write-Host ""
+    Write-Host "========================================" -ForegroundColor Red
+    Write-Host "✗ 上传失败" -ForegroundColor Red
+    Write-Host "========================================" -ForegroundColor Red
+    Write-Host "错误信息: $($_.Exception.Message)" -ForegroundColor Red
+    Write-Host "========================================" -ForegroundColor Red
+    exit 1
+}

package/scripts/upload-clipboard-image.py

@@ -0,0 +1,196 @@
+#!/usr/bin/env python3
+"""
+从剪贴板获取图片并上传到禅道
+
+依赖安装:
+    pip install pillow requests
+
+使用方法:
+    python upload-clipboard-image.py
+
+环境变量:
+    ZENTAO_URL: 禅道系统URL
+    ZENTAO_USERNAME: 用户名
+    ZENTAO_PASSWORD: 密码
+    ZENTAO_UID: 可选的UID参数
+"""
+
+import os
+import sys
+import hashlib
+import requests
+import base64
+from io import BytesIO
+from pathlib import Path
+from datetime import datetime
+
+try:
+    from PIL import ImageGrab, Image
+except ImportError:
+    print("错误: 请先安装 Pillow 库")
+    print("运行: pip install pillow")
+    sys.exit(1)
+
+
+def get_clipboard_image():
+    """从剪贴板获取图片"""
+    print("正在读取剪贴板图片...")
+
+    try:
+        # 尝试获取剪贴板图片
+        img = ImageGrab.grabclipboard()
+
+        if img is None:
+            print("错误: 剪贴板中没有图片")
+            sys.exit(1)
+
+        # 如果是文件路径列表，读取第一个图片文件
+        if isinstance(img, list):
+            img_path = img[0]
+            if os.path.isfile(img_path):
+                img = Image.open(img_path)
+            else:
+                print(f"错误: 无法打开文件 {img_path}")
+                sys.exit(1)
+
+        print(f"✓ 成功读取剪贴板图片 ({img.width}x{img.height})")
+        return img
+
+    except Exception as e:
+        print(f"错误: 无法读取剪贴板图片 - {e}")
+        sys.exit(1)
+
+
+def image_to_base64(img):
+    """将图片转换为 base64"""
+    buffer = BytesIO()
+    img.save(buffer, format='PNG')
+    img_bytes = buffer.getvalue()
+    return base64.b64encode(img_bytes).decode('utf-8'), img_bytes
+
+
+def get_zentao_token(url, username, password):
+    """获取禅道 Token"""
+    print("正在连接禅道系统...")
+
+    # MD5 加密密码
+    password_md5 = hashlib.md5(password.encode()).hexdigest()
+
+    token_url = f"{url}/zentao/api.php/v1/tokens"
+    data = {
+        "account": username,
+        "password": password_md5
+    }
+
+    try:
+        response = requests.post(token_url, json=data)
+        response.raise_for_status()
+        token = response.json().get('token')
+
+        if not token:
+            print("错误: 无法获取 Token")
+            sys.exit(1)
+
+        print("✓ 成功获取 Token")
+        return token
+
+    except requests.exceptions.RequestException as e:
+        print(f"错误: 无法连接禅道系统 - {e}")
+        sys.exit(1)
+
+
+def upload_to_zentao(url, token, img_bytes, filename, uid=None):
+    """上传图片到禅道"""
+    print("正在上传到禅道...")
+
+    upload_url = f"{url}/zentao/api.php/v1/files"
+    if uid:
+        upload_url += f"?uid={uid}"
+
+    headers = {
+        "Token": token
+    }
+
+    files = {
+        'imgFile': (filename, img_bytes, 'image/png')
+    }
+
+    try:
+        response = requests.post(upload_url, headers=headers, files=files)
+        response.raise_for_status()
+        result = response.json()
+
+        print("")
+        print("=" * 40)
+        print("✓ 上传成功！")
+        print("=" * 40)
+        print(f"文件ID: {result.get('id')}")
+        print(f"访问URL: {result.get('url')}")
+        print("=" * 40)
+
+        return result
+
+    except requests.exceptions.RequestException as e:
+        print("")
+        print("=" * 40)
+        print("✗ 上传失败")
+        print("=" * 40)
+        print(f"错误信息: {e}")
+        print("=" * 40)
+        sys.exit(1)
+
+
+def main():
+    # 检查环境变量
+    zentao_url = os.environ.get('ZENTAO_URL')
+    username = os.environ.get('ZENTAO_USERNAME')
+    password = os.environ.get('ZENTAO_PASSWORD')
+    uid = os.environ.get('ZENTAO_UID')
+
+    if not all([zentao_url, username, password]):
+        print("错误: 请设置环境变量 ZENTAO_URL, ZENTAO_USERNAME, ZENTAO_PASSWORD")
+        print("")
+        print("Windows示例:")
+        print("  set ZENTAO_URL=https://your-zentao-url")
+        print("  set ZENTAO_USERNAME=your-username")
+        print("  set ZENTAO_PASSWORD=your-password")
+        print("  python upload-clipboard-image.py")
+        print("")
+        print("Mac/Linux示例:")
+        print("  export ZENTAO_URL=https://your-zentao-url")
+        print("  export ZENTAO_USERNAME=your-username")
+        print("  export ZENTAO_PASSWORD=your-password")
+        print("  python upload-clipboard-image.py")
+        sys.exit(1)
+
+    # 获取剪贴板图片
+    img = get_clipboard_image()
+
+    # 转换为 base64
+    base64_str, img_bytes = image_to_base64(img)
+    print("✓ 图片已转换为 base64")
+
+    # 创建 img 文件夹
+    img_dir = Path.cwd() / 'img'
+    img_dir.mkdir(exist_ok=True)
+
+    # 保存到本地
+    timestamp = int(datetime.now().timestamp() * 1000)
+    filename = f"clipboard_{timestamp}.png"
+    local_path = img_dir / filename
+
+    with open(local_path, 'wb') as f:
+        f.write(img_bytes)
+
+    print(f"✓ 图片已保存到: {local_path}")
+
+    # 获取 Token
+    token = get_zentao_token(zentao_url, username, password)
+
+    # 上传到禅道
+    result = upload_to_zentao(zentao_url, token, img_bytes, filename, uid)
+    print(f"本地路径: {local_path}")
+
+
+if __name__ == '__main__':
+    main()

package/scripts/upload.bat.example

@@ -0,0 +1,41 @@
+@echo off
+REM ========================================
+REM 剪贴板图片上传到禅道 - 配置示例
+REM ========================================
+REM
+REM 使用方法:
+REM 1. 复制此文件为 upload.bat
+REM 2. 修改下面的配置信息
+REM 3. 双击运行或在命令行运行
+REM
+
+REM ========================================
+REM 配置区域 - 请修改为你的禅道信息
+REM ========================================
+
+set ZENTAO_URL=https://your-zentao-url
+set ZENTAO_USERNAME=your-username
+set ZENTAO_PASSWORD=your-password
+set ZENTAO_UID=
+
+REM ========================================
+REM 选择上传方式
+REM ========================================
+
+REM 方式1: 使用 Node.js (推荐)
+node "%~dp0upload-clipboard-image.js"
+
+REM 方式2: 使用 PowerShell (Windows)
+REM powershell -ExecutionPolicy Bypass -File "%~dp0upload-clipboard-image.ps1" ^
+REM     -ZentaoUrl "%ZENTAO_URL%" ^
+REM     -Username "%ZENTAO_USERNAME%" ^
+REM     -Password "%ZENTAO_PASSWORD%" ^
+REM     -Uid "%ZENTAO_UID%"
+
+REM 方式3: 使用 Python
+REM python "%~dp0upload-clipboard-image.py"
+
+REM ========================================
+REM 按任意键退出
+REM ========================================
+pause