mcp-cos-upload 1.0.0 → 1.0.2

package/README.md

@@ -54,14 +54,19 @@ Upload a file to Tencent Cloud COS.
 
 | Parameter | Type | Required | Description |
 |-----------|------|----------|-------------|
-| `url` | string | ❌ | URL to download and upload (mutually exclusive with `content`) |
-| `content` | string | ❌ | Text content to upload (mutually exclusive with `url`) |
+| `url` | string | ❌ | URL to download and upload |
+| `filepath` | string | ❌ | Local file path to upload |
+| `content` | string | ❌ | Text content to upload |
 | `bucket` | string | ❌ | COS bucket name (uses env default if not provided) |
 | `region` | string | ❌ | COS region (uses env default if not provided) |
 | `key` | string | ❌ | Object key (auto-generated if not provided) |
 | `folder` | string | ❌ | Folder prefix (overrides `COS_KEY_PREFIX`) |
 | `filename` | string | ❌ | Custom filename |
 | `ext` | string | ❌ | File extension (e.g., `png`, `jpg`, `svg`) |
+| `compress` | boolean | ❌ | Enable image compression (default: `true`) |
+| `quality` | number | ❌ | Compression quality 1-100 (default: `80`, for JPEG/WebP/PNG) |
+
+> Note: `url`, `filepath`, and `content` are mutually exclusive - provide only one.
 
 **Example:**
 
@@ -69,6 +74,20 @@ Upload a file to Tencent Cloud COS.
 Upload this image to COS: https://example.com/image.png
 ```
 
+## Image Compression
+
+Supports automatic compression for uploaded images:
+
+| Format | Compression |
+|--------|-------------|
+| PNG | Lossless/lossy compression |
+| JPEG | Quality adjustment with mozjpeg |
+| WebP | Quality adjustment |
+| GIF | Optimization |
+| SVG | SVGO optimization (removes redundant code) |
+
+Compression is enabled by default. Use `compress: false` to upload original files.
+
 ## License
 
 MIT

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-cos-upload",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "MCP server for uploading files to Tencent Cloud COS",
   "type": "module",
   "main": "src/index.js",
@@ -11,24 +11,18 @@
   "keywords": [
     "mcp",
     "cos",
-    "image-upload",
-    "figma"
+    "tencent-cloud",
+    "upload"
   ],
-  "author": "",
   "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": ""
-  },
   "engines": {
     "node": ">=18.0.0"
   },
-  "scripts": {
-    "start": "node src/index.js"
-  },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.12.0",
     "cos-nodejs-sdk-v5": "^2.15.4",
+    "sharp": "^0.34.5",
+    "svgo": "^4.0.0",
     "zod": "^3.24.0"
   }
 }

package/src/index.js

@@ -8,6 +8,10 @@ import {
   CallToolRequestSchema,
 } from "@modelcontextprotocol/sdk/types.js";
 import { z } from "zod";
+import sharp from "sharp";
+import { optimize } from "svgo";
+import { readFile } from "fs/promises";
+import { extname, basename } from "path";
 import {
   createCosClient,
   DEFAULT_BUCKET,
@@ -25,9 +29,12 @@ const CosUploadArgsSchema = z.object({
   key: z.string().optional().describe("对象 Key，不传则自动生成"),
   folder: z.string().optional().describe("可选目录前缀（覆盖 env 中的 COS_KEY_PREFIX）"),
   filename: z.string().optional().describe("可选文件名，例如使用 Figma 节点名称"),
-  ext: z.string().optional().describe("可选扩展名，如 png/jpg/svg"),
-  content: z.string().optional().describe("要上传的文本内容（和 url 二选一）"),
-  url: z.string().url().optional().describe("远程文件 URL（和 content 二选一，常用于 Figma 导出图片）"),
+  ext: z.string().optional().describe("可选扩展名，如 png/jpg/svg，不传则从 filename/url/filepath 推断，默认 png"),
+  content: z.string().optional().describe("要上传的文本内容（和 url/filepath 三选一）"),
+  url: z.string().url().optional().describe("远程文件 URL（和 content/filepath 三选一，常用于 Figma 导出图片）"),
+  filepath: z.string().optional().describe("本地文件路径（和 url/content 三选一）"),
+  compress: z.boolean().optional().default(true).describe("是否压缩图片，默认 true"),
+  quality: z.number().min(1).max(100).optional().default(80).describe("压缩质量 1-100，默认 80（仅对 JPEG/WebP/PNG 有效）"),
 });
 
 // MCP 客户端的 JSON Schema
@@ -39,9 +46,12 @@ const cosUploadInputSchema = {
     key: { type: "string", description: "对象 Key，不传则自动生成" },
     folder: { type: "string", description: "可选目录前缀（覆盖 env 中的 COS_KEY_PREFIX）" },
     filename: { type: "string", description: "可选文件名，例如使用 Figma 节点名称" },
-    ext: { type: "string", description: "可选扩展名，如 png/jpg/svg，不传则从 filename/url 推断，默认 png" },
-    content: { type: "string", description: "要上传的文本内容（和 url 二选一）" },
-    url: { type: "string", description: "远程文件 URL（和 content 二选一，常用于 Figma 导出图片）" },
+    ext: { type: "string", description: "可选扩展名，如 png/jpg/svg，不传则从 filename/url/filepath 推断，默认 png" },
+    content: { type: "string", description: "要上传的文本内容（和 url/filepath 三选一）" },
+    url: { type: "string", description: "远程文件 URL（和 content/filepath 三选一，常用于 Figma 导出图片）" },
+    filepath: { type: "string", description: "本地文件路径（和 url/content 三选一）" },
+    compress: { type: "boolean", description: "是否压缩图片，默认 true" },
+    quality: { type: "number", description: "压缩质量 1-100，默认 80（仅对 JPEG/WebP/PNG 有效）" },
   },
   required: [],
 };
@@ -78,23 +88,48 @@ async function main() {
       return { isError: true, content: [{ type: "text", text: "参数校验失败: " + (err?.message || JSON.stringify(err)) }] };
     }
 
-    let { bucket, region, key, folder, filename, ext, content, url } = args;
+    let { bucket, region, key, folder, filename, ext, content, url, filepath, compress, quality } = args;
 
     bucket = bucket || DEFAULT_BUCKET;
     region = region || DEFAULT_REGION;
+    compress = compress !== false; // 默认开启压缩
+    quality = quality || 80;
 
     if (!bucket || !region) {
       return { isError: true, content: [{ type: "text", text: "❌ bucket 和 region 必须指定。请在 env 配置 COS_DEFAULT_BUCKET / COS_DEFAULT_REGION，或在调用时传入。" }] };
     }
 
-    if (!content && !url) {
-      return { isError: true, content: [{ type: "text", text: "必须提供 content 或 url 其中一个参数。" }] };
+    if (!content && !url && !filepath) {
+      return { isError: true, content: [{ type: "text", text: "必须提供 content、url 或 filepath 其中一个参数。" }] };
     }
 
+    // 扩展名到 MIME 类型映射
+    const extToMimeType = {
+      png: "image/png", jpg: "image/jpeg", jpeg: "image/jpeg",
+      gif: "image/gif", webp: "image/webp", svg: "image/svg+xml",
+      txt: "text/plain", json: "application/json",
+    };
+
     let body;
     let contentType = null;
     if (content) {
       body = content;
+    } else if (filepath) {
+      // 读取本地文件
+      try {
+        body = await readFile(filepath);
+        const fileExt = extname(filepath).slice(1).toLowerCase();
+        contentType = extToMimeType[fileExt] || "";
+        // 从文件路径提取文件名（如果未指定）
+        if (!filename) {
+          filename = basename(filepath, extname(filepath));
+        }
+        if (!ext) {
+          ext = fileExt;
+        }
+      } catch (fileErr) {
+        return { isError: true, content: [{ type: "text", text: `读取本地文件失败: ${fileErr?.message || String(fileErr)}` }] };
+      }
     } else {
       const res = await fetch(url);
       if (!res.ok) {
@@ -104,6 +139,95 @@ async function main() {
       body = Buffer.from(await res.arrayBuffer());
     }
 
+    // 图片压缩处理
+    let originalSize = Buffer.isBuffer(body) ? body.length : Buffer.byteLength(body);
+    let compressedSize = originalSize;
+    let wasCompressed = false;
+
+    if (compress && Buffer.isBuffer(body)) {
+      const mimeType = (contentType || "").split(";")[0].trim().toLowerCase();
+
+      try {
+        if (mimeType === "image/svg+xml" || mimeType === "image/svg") {
+          // SVG 压缩：使用 svgo
+          const svgString = body.toString("utf-8");
+          const result = optimize(svgString, {
+            multipass: true,
+            plugins: [
+              "preset-default",
+              "removeDimensions",
+              { name: "removeViewBox", active: false },
+            ],
+          });
+          body = Buffer.from(result.data, "utf-8");
+          compressedSize = body.length;
+          wasCompressed = true;
+        } else if (["image/png", "image/jpeg", "image/jpg", "image/webp", "image/gif"].includes(mimeType)) {
+          // 位图压缩：使用 sharp，参考 TinyPNG 的智能有损压缩策略
+          const sharpInstance = sharp(body);
+
+          if (mimeType === "image/png") {
+            // PNG: 使用调色板模式 + 颜色量化（类似 TinyPNG 的核心算法）
+            body = await sharpInstance
+              .png({
+                palette: true,           // 启用调色板模式，大幅减小文件
+                quality,                 // 颜色量化质量
+                compressionLevel: 9,     // 最大压缩级别
+                effort: 10,              // 最大压缩努力
+              })
+              .toBuffer();
+          } else if (mimeType === "image/jpeg" || mimeType === "image/jpg") {
+            // JPEG: 使用 mozjpeg 编码器 + 优化参数
+            body = await sharpInstance
+              .jpeg({
+                quality,
+                mozjpeg: true,                // 使用 mozjpeg 编码器
+                trellisQuantisation: true,    // 网格量化，提高压缩效率
+                overshootDeringing: true,     // 减少振铃效应
+                optimiseScans: true,          // 优化扫描
+              })
+              .toBuffer();
+          } else if (mimeType === "image/webp") {
+            // WebP: 智能压缩
+            body = await sharpInstance
+              .webp({
+                quality,
+                effort: 6,                // 压缩努力 (0-6)
+                smartSubsample: true,     // 智能子采样
+              })
+              .toBuffer();
+          } else if (mimeType === "image/gif") {
+            // GIF: 优化
+            body = await sharpInstance.gif({ effort: 10 }).toBuffer();
+          }
+          compressedSize = body.length;
+          wasCompressed = true;
+        }
+      } catch (compressErr) {
+        // 压缩失败时使用原始数据，不中断上传
+        console.error("[mcp-cos-upload] 压缩失败，使用原始文件:", compressErr.message);
+      }
+    }
+
+    // 生成随机字符串（6-8位）
+    const generateRandomStr = (len = 6) => {
+      const chars = "abcdefghijklmnopqrstuvwxyz0123456789";
+      let result = "";
+      for (let i = 0; i < len; i++) {
+        result += chars.charAt(Math.floor(Math.random() * chars.length));
+      }
+      return result;
+    };
+
+    // 获取当前日期目录（YYYY-MM-DD）
+    const getDateFolder = () => {
+      const now = new Date();
+      const year = now.getFullYear();
+      const month = String(now.getMonth() + 1).padStart(2, "0");
+      const day = String(now.getDate()).padStart(2, "0");
+      return `${year}-${month}-${day}`;
+    };
+
     // 自动生成 key
     if (!key) {
       if (!ext) {
@@ -132,8 +256,16 @@ async function main() {
         }
       }
 
+      // 添加随机字符防止文件名冲突
+      const randomSuffix = generateRandomStr(6);
+      const finalFilename = `${filename}_${randomSuffix}`;
+
+      // 构建 key: prefix/日期目录/文件名.扩展名
       const prefix = (folder || KEY_PREFIX || "").replace(/^\/+|\/+$/g, "");
-      const keyParts = prefix ? [prefix, `${filename}.${ext}`] : [`${filename}.${ext}`];
+      const dateFolder = getDateFolder();
+      const keyParts = prefix
+        ? [prefix, dateFolder, `${finalFilename}.${ext}`]
+        : [dateFolder, `${finalFilename}.${ext}`];
       key = keyParts.join("/");
     }
 
@@ -152,12 +284,17 @@ async function main() {
         ? `https://${CDN_DOMAIN}/${encodeURI(key)}`
         : `https://${bucket}.cos.${region}.myqcloud.com/${encodeURI(key)}`;
 
+      const compressionInfo = wasCompressed
+        ? `压缩: ${(originalSize / 1024).toFixed(1)}KB → ${(compressedSize / 1024).toFixed(1)}KB (${((1 - compressedSize / originalSize) * 100).toFixed(1)}% 减少)`
+        : `压缩: 未压缩`;
+
       const summaryText = [
         "COS 上传成功 ✅",
         `Bucket: ${bucket}`,
         `Region: ${region}`,
         `Key: ${key}`,
         `URL: ${cdnUrl}`,
+        compressionInfo,
         `耗时: ${elapsed}ms`,
       ].join("\n");