@dyno181cm.nexsoft/zentao_mcp 1.0.2 → 1.1.0

package/README.md

@@ -9,6 +9,7 @@ A Model Context Protocol (MCP) server for integrating with the Zentao API. This
   - `zentao_get_execution_tasks`: Retrieve tasks in an execution (sprint), with optional filtering by module ID.
   - `zentao_get_product_bugs`: Fetch a list of bugs associated with a specific product.
   - `zentao_get_bug_details`: Get detailed information about a specific bug.
+  - `zentao_download_attachment`: Download file attachments from ZenTao (e.g., bug/task images or videos).
 
 ## Requirements
 
@@ -105,3 +106,8 @@ If a user clones the repository locally, they can configure their client to run
 - **`zentao_get_bug_details`**
   - **Inputs:**
     - `bugId` (string | number) - Required. Bug ID.
+
+- **`zentao_download_attachment`**
+  - **Inputs:**
+    - `fileId` (string | number) - Required. File ID to download.
+    - `extension` (string) - Optional. File extension (e.g., mp4, png).

package/build/tools.js

@@ -1,4 +1,6 @@
 import { z } from "zod";
+import * as os from "os";
+import * as path from "path";
 import { ZentaoClient } from "./zentaoClient.js";
 const client = new ZentaoClient();
 export function registerTools(server) {
@@ -38,4 +40,25 @@ export function registerTools(server) {
             content: [{ type: "text", text: JSON.stringify(data, null, 2) }],
         };
     });
+    server.registerTool("zentao_download_attachment", {
+        description: "Download a file attachment from ZenTao and save it locally",
+        inputSchema: {
+            fileId: z.union([z.string(), z.number()]).describe("File ID to download"),
+            extension: z.string().optional().describe("Optional file extension (e.g., mp4, png)"),
+        }
+    }, async ({ fileId, extension }) => {
+        const ext = extension ? (extension.startsWith('.') ? extension : `.${extension}`) : '';
+        const targetPath = path.join(os.tmpdir(), `zentao_file_${fileId}${ext}`);
+        try {
+            const savedPath = await client.downloadFile(fileId, targetPath);
+            return {
+                content: [{ type: "text", text: `File downloaded successfully to: ${savedPath}` }],
+            };
+        }
+        catch (error) {
+            return {
+                content: [{ type: "text", text: `Failed to download file: ${error.message}` }],
+            };
+        }
+    });
 }

package/build/zentaoClient.js

@@ -1,5 +1,6 @@
 import axios from 'axios';
 import * as dotenv from 'dotenv';
+import * as fs from 'fs';
 dotenv.config();
 export class ZentaoClient {
     client;
@@ -98,4 +99,24 @@ export class ZentaoClient {
         const res = await this.client.get(`/bugs/${bugId}`);
         return res.data;
     }
+    async downloadFile(fileId, targetPath) {
+        const writer = fs.createWriteStream(targetPath);
+        const res = await this.client.get(`/files/${fileId}`, {
+            responseType: 'stream',
+        });
+        return new Promise((resolve, reject) => {
+            res.data.pipe(writer);
+            let error = null;
+            writer.on('error', err => {
+                error = err;
+                writer.close();
+                reject(err);
+            });
+            writer.on('close', () => {
+                if (!error) {
+                    resolve(targetPath);
+                }
+            });
+        });
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dyno181cm.nexsoft/zentao_mcp",
-  "version": "1.0.2",
+  "version": "1.1.0",
   "description": "Zentao MCP Server",
   "main": "build/index.js",
   "bin": {