@zzp123/mcp-zentao 1.1.2 → 1.2.1

package/dist/api/zentaoApi.d.ts

@@ -1,4 +1,4 @@
-import { AssignFeedbackRequest, Bug, BugResolution, BugStatus, Build, CloseFeedbackRequest, CreateBuildRequest, CreateBugRequest, CreateExecutionRequest, CreateFeedbackRequest, CreatePlanRequest, CreateProductRequest, CreateProgramRequest, CreateProjectRequest, CreateStoryRequest, CreateTaskRequest, CreateTestCaseRequest, CreateTicketRequest, CreateUserRequest, Execution, Feedback, Plan, Product, Program, Project, Release, Story, Task, TaskStatus, TaskUpdate, TestCase, Ticket, UpdateBuildRequest, UpdateBugRequest, UpdateExecutionRequest, UpdateFeedbackRequest, UpdatePlanRequest, UpdateProductRequest, UpdateProgramRequest, UpdateProjectRequest, UpdateStoryRequest, UpdateTestCaseRequest, UpdateTicketRequest, UpdateUserRequest, UserDetail, ZentaoConfig } from '../types/zentao';
+import { AssignFeedbackRequest, Bug, BugResolution, BugStatus, Build, CloseFeedbackRequest, CreateBuildRequest, CreateBugRequest, CreateExecutionRequest, CreateFeedbackRequest, CreatePlanRequest, CreateProductRequest, CreateProgramRequest, CreateProjectRequest, CreateStoryRequest, CreateTaskRequest, CreateTestCaseRequest, CreateTicketRequest, CreateUserRequest, Execution, Feedback, FileUploadResponse, Module, ModuleType, Plan, Product, Program, Project, Release, Story, Task, TaskStatus, TaskUpdate, TestCase, Ticket, UpdateBuildRequest, UpdateBugRequest, UpdateExecutionRequest, UpdateFeedbackRequest, UpdatePlanRequest, UpdateProductRequest, UpdateProgramRequest, UpdateProjectRequest, UpdateStoryRequest, UpdateTestCaseRequest, UpdateTicketRequest, UpdateUserRequest, UploadFileRequest, UserDetail, ZentaoConfig } from '../types/zentao';
 export declare class ZentaoAPI {
     private config;
     private client;
@@ -148,4 +148,7 @@ export declare class ZentaoAPI {
     deleteTicket(ticketId: number): Promise<{
         message: string;
     }>;
+    getModules(type: ModuleType, id: number): Promise<Module[]>;
+    uploadFile(uploadRequest: UploadFileRequest): Promise<FileUploadResponse>;
+    downloadFile(fileId: number): Promise<Buffer>;
 }

package/dist/api/zentaoApi.js

@@ -1116,4 +1116,64 @@ export class ZentaoAPI {
             throw error;
         }
     }
+    // 模块相关API
+    async getModules(type, id) {
+        try {
+            console.log(`正在获取模块列表，类型: ${type}, ID: ${id}...`);
+            const params = { type, id };
+            const response = await this.request('GET', '/modules', params);
+            console.log('获取模块列表响应:', response);
+            return response.modules;
+        }
+        catch (error) {
+            console.error('获取模块列表失败:', error);
+            throw error;
+        }
+    }
+    // 文件相关API
+    async uploadFile(uploadRequest) {
+        try {
+            console.log(`正在上传文件: ${uploadRequest.filename}...`);
+            const token = await this.getToken();
+            // 创建FormData
+            const FormData = (await import('form-data')).default;
+            const formData = new FormData();
+            formData.append('imgFile', uploadRequest.file, uploadRequest.filename);
+            const params = uploadRequest.uid ? { uid: uploadRequest.uid } : undefined;
+            const response = await this.client.request({
+                method: 'POST',
+                url: '/files',
+                params,
+                data: formData,
+                headers: {
+                    Token: token,
+                    ...formData.getHeaders()
+                },
+            });
+            console.log('上传文件响应:', response.data);
+            return response.data;
+        }
+        catch (error) {
+            console.error('上传文件失败:', error);
+            throw error;
+        }
+    }
+    async downloadFile(fileId) {
+        try {
+            console.log(`正在下载文件 ${fileId}...`);
+            const token = await this.getToken();
+            const response = await this.client.request({
+                method: 'GET',
+                url: `/files/${fileId}`,
+                headers: { Token: token },
+                responseType: 'arraybuffer'
+            });
+            console.log('下载文件成功');
+            return Buffer.from(response.data);
+        }
+        catch (error) {
+            console.error('下载文件失败:', error);
+            throw error;
+        }
+    }
 }

package/dist/index.js

@@ -920,6 +920,86 @@ server.tool("deleteTicket", { ticketId: z.number() }, async ({ ticketId }) => {
         throw new Error("Please initialize Zentao API first");
     return { content: [{ type: "text", text: JSON.stringify(await zentaoApi.deleteTicket(ticketId), null, 2) }] };
 });
+// 模块相关工具
+server.tool("getModules", {
+    type: z.enum(['story', 'task', 'bug', 'case', 'feedback', 'product']),
+    id: z.number()
+}, async ({ type, id }) => {
+    if (!zentaoApi)
+        throw new Error("Please initialize Zentao API first");
+    const modules = await zentaoApi.getModules(type, id);
+    return { content: [{ type: "text", text: JSON.stringify(modules, null, 2) }] };
+});
+// 文件相关工具
+server.tool("uploadFile", {
+    filePath: z.string().optional(),
+    base64Data: z.string().optional(),
+    filename: z.string().optional(),
+    uid: z.string().optional()
+}, 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');
+    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: finalFilename,
+        uid
+    });
+    const response = {
+        upload: result,
+    };
+    if (savedPath) {
+        response.savedPath = savedPath;
+    }
+    return { content: [{ type: "text", text: JSON.stringify(response, null, 2) }] };
+});
+server.tool("downloadFile", {
+    fileId: z.number(),
+    savePath: z.string()
+}, async ({ fileId, savePath }) => {
+    if (!zentaoApi)
+        throw new Error("Please initialize Zentao API first");
+    const fs = await import('fs');
+    const fileBuffer = await zentaoApi.downloadFile(fileId);
+    fs.writeFileSync(savePath, fileBuffer);
+    return { content: [{ type: "text", text: `文件已下载到: ${savePath}` }] };
+});
 // Start receiving messages on stdin and sending messages on stdout
 const transport = new StdioServerTransport();
 await server.connect(transport).catch(console.error);

package/dist/types/zentao.d.ts

@@ -611,3 +611,27 @@ export interface UpdateTicketRequest {
     type?: string;
     desc?: string;
 }
+export interface Module {
+    id: string;
+    root: string;
+    branch: string;
+    name: string;
+    parent: string;
+    path: string;
+    grade: string;
+    order: string;
+    type: string;
+    owner: string;
+    collector?: string;
+    short?: string;
+}
+export type ModuleType = 'story' | 'task' | 'bug' | 'case' | 'feedback' | 'product';
+export interface FileUploadResponse {
+    id: number;
+    url: string;
+}
+export interface UploadFileRequest {
+    file: Buffer | Blob;
+    filename: string;
+    uid?: string;
+}

package/package.json

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