mcp-cos-upload 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,74 @@
+# mcp-cos-upload
+
+MCP server for uploading files to Tencent Cloud COS. Supports uploading from URL (e.g., Figma export) or text content.
+
+## Installation
+
+```bash
+npm install mcp-cos-upload
+```
+
+## Usage
+
+### MCP Configuration (Cursor / Claude Desktop)
+
+Add to your MCP configuration:
+
+```json
+{
+  "mcpServers": {
+    "cos-upload": {
+      "command": "npx",
+      "args": ["-y", "mcp-cos-upload"],
+      "env": {
+        "COS_SECRET_ID": "your_secret_id",
+        "COS_SECRET_KEY": "your_secret_key",
+        "COS_DEFAULT_BUCKET": "your_bucket_name",
+        "COS_DEFAULT_REGION": "ap-guangzhou",
+        "COS_KEY_PREFIX": "uploads/",
+        "COS_CDN_DOMAIN": "your-cdn-domain.com"
+      }
+    }
+  }
+}
+```
+
+### Environment Variables
+
+| Variable | Required | Description |
+|----------|----------|-------------|
+| `COS_SECRET_ID` | ✅ | Tencent Cloud SecretId |
+| `COS_SECRET_KEY` | ✅ | Tencent Cloud SecretKey |
+| `COS_DEFAULT_BUCKET` | ✅ | Default COS bucket name |
+| `COS_DEFAULT_REGION` | ✅ | Default COS region (e.g., `ap-guangzhou`) |
+| `COS_KEY_PREFIX` | ❌ | Key prefix for uploaded files (e.g., `uploads/`) |
+| `COS_CDN_DOMAIN` | ❌ | Custom CDN domain (e.g., `cdn.example.com`) |
+
+## Tools
+
+### cos_upload
+
+Upload a file to Tencent Cloud COS.
+
+**Parameters:**
+
+| 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`) |
+| `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`) |
+
+**Example:**
+
+```
+Upload this image to COS: https://example.com/image.png
+```
+
+## License
+
+MIT

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "mcp-cos-upload",
+  "version": "1.0.0",
+  "description": "MCP server for uploading files to Tencent Cloud COS",
+  "type": "module",
+  "main": "src/index.js",
+  "bin": "./src/index.js",
+  "files": [
+    "src/**/*"
+  ],
+  "keywords": [
+    "mcp",
+    "cos",
+    "image-upload",
+    "figma"
+  ],
+  "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",
+    "zod": "^3.24.0"
+  }
+}

package/src/cos.js

@@ -0,0 +1,19 @@
+// mcp-cos-upload - COS client configuration
+import COS from "cos-nodejs-sdk-v5";
+
+export function createCosClient() {
+  const secretId = process.env.COS_SECRET_ID;
+  const secretKey = process.env.COS_SECRET_KEY;
+
+  if (!secretId || !secretKey) {
+    console.error("[mcp-cos-upload] ❌ 缺少 COS_SECRET_ID 或 COS_SECRET_KEY");
+    process.exit(1);
+  }
+
+  return new COS({ SecretId: secretId, SecretKey: secretKey });
+}
+
+export const DEFAULT_BUCKET = process.env.COS_DEFAULT_BUCKET || null;
+export const DEFAULT_REGION = process.env.COS_DEFAULT_REGION || null;
+export const KEY_PREFIX = process.env.COS_KEY_PREFIX || "";
+export const CDN_DOMAIN = process.env.COS_CDN_DOMAIN || null;

package/src/index.js

@@ -0,0 +1,182 @@
+#!/usr/bin/env node
+// mcp-cos-upload - MCP server for uploading files to Tencent Cloud COS
+
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import {
+  ListToolsRequestSchema,
+  CallToolRequestSchema,
+} from "@modelcontextprotocol/sdk/types.js";
+import { z } from "zod";
+import {
+  createCosClient,
+  DEFAULT_BUCKET,
+  DEFAULT_REGION,
+  KEY_PREFIX,
+  CDN_DOMAIN,
+} from "./cos.js";
+
+const cos = createCosClient();
+
+// 工具参数校验（zod）
+const CosUploadArgsSchema = z.object({
+  bucket: z.string().optional().describe("COS Bucket 名，不传则用默认 COS_DEFAULT_BUCKET"),
+  region: z.string().optional().describe("COS 区域，不传则用默认 COS_DEFAULT_REGION"),
+  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 导出图片）"),
+});
+
+// MCP 客户端的 JSON Schema
+const cosUploadInputSchema = {
+  type: "object",
+  properties: {
+    bucket: { type: "string", description: "COS Bucket 名，不传则用默认 COS_DEFAULT_BUCKET" },
+    region: { type: "string", description: "COS 区域，不传则用默认 COS_DEFAULT_REGION" },
+    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 导出图片）" },
+  },
+  required: [],
+};
+
+async function main() {
+  const server = new Server(
+    { name: "mcp-cos-upload", version: "1.0.0" },
+    { capabilities: { tools: {} } }
+  );
+
+  // 列出可用工具
+  server.setRequestHandler(ListToolsRequestSchema, async () => ({
+    tools: [
+      {
+        name: "cos_upload",
+        description: "上传文件到腾讯云 COS。支持从 URL 下载上传或直接上传文本内容，返回 CDN 访问地址。",
+        inputSchema: cosUploadInputSchema,
+      },
+    ],
+  }));
+
+  // 处理工具调用
+  server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: rawArgs } = request.params;
+
+    if (name !== "cos_upload") {
+      return { isError: true, content: [{ type: "text", text: `未知工具: ${name}` }] };
+    }
+
+    let args;
+    try {
+      args = CosUploadArgsSchema.parse(rawArgs || {});
+    } catch (err) {
+      return { isError: true, content: [{ type: "text", text: "参数校验失败: " + (err?.message || JSON.stringify(err)) }] };
+    }
+
+    let { bucket, region, key, folder, filename, ext, content, url } = args;
+
+    bucket = bucket || DEFAULT_BUCKET;
+    region = region || DEFAULT_REGION;
+
+    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 其中一个参数。" }] };
+    }
+
+    let body;
+    let contentType = null;
+    if (content) {
+      body = content;
+    } else {
+      const res = await fetch(url);
+      if (!res.ok) {
+        return { isError: true, content: [{ type: "text", text: `下载 URL 失败: ${res.status} ${res.statusText}` }] };
+      }
+      contentType = res.headers.get("content-type") || "";
+      body = Buffer.from(await res.arrayBuffer());
+    }
+
+    // 自动生成 key
+    if (!key) {
+      if (!ext) {
+        if (contentType) {
+          const contentTypeMap = {
+            "image/png": "png", "image/jpeg": "jpg", "image/jpg": "jpg",
+            "image/gif": "gif", "image/webp": "webp", "image/svg+xml": "svg",
+            "image/svg": "svg", "text/plain": "txt", "application/json": "json",
+          };
+          const mimeType = contentType.split(";")[0].trim().toLowerCase();
+          ext = contentTypeMap[mimeType];
+        }
+        if (!ext) {
+          const fromName = filename || (url ? new URL(url).pathname.split("/").pop() : "") || "";
+          const m = fromName.match(/\.([a-zA-Z0-9]+)$/);
+          ext = m ? m[1] : "png";
+        }
+      }
+
+      if (!filename) {
+        if (url) {
+          const last = new URL(url).pathname.split("/").pop() || "";
+          filename = last.replace(/\.[a-zA-Z0-9]+$/, "") || `img_${Date.now()}`;
+        } else {
+          filename = `img_${Date.now()}`;
+        }
+      }
+
+      const prefix = (folder || KEY_PREFIX || "").replace(/^\/+|\/+$/g, "");
+      const keyParts = prefix ? [prefix, `${filename}.${ext}`] : [`${filename}.${ext}`];
+      key = keyParts.join("/");
+    }
+
+    const start = Date.now();
+    const putObject = () => new Promise((resolve, reject) => {
+      cos.putObject({ Bucket: bucket, Region: region, Key: key, Body: body }, (err, data) => {
+        if (err) return reject(err);
+        resolve(data);
+      });
+    });
+
+    try {
+      const data = await putObject();
+      const elapsed = Date.now() - start;
+      const cdnUrl = CDN_DOMAIN
+        ? `https://${CDN_DOMAIN}/${encodeURI(key)}`
+        : `https://${bucket}.cos.${region}.myqcloud.com/${encodeURI(key)}`;
+
+      const summaryText = [
+        "COS 上传成功 ✅",
+        `Bucket: ${bucket}`,
+        `Region: ${region}`,
+        `Key: ${key}`,
+        `URL: ${cdnUrl}`,
+        `耗时: ${elapsed}ms`,
+      ].join("\n");
+
+      return {
+        content: [
+          { type: "text", text: summaryText },
+          { type: "text", text: JSON.stringify({ bucket, region, key, cdnUrl, raw: data }, null, 2) },
+        ],
+      };
+    } catch (err) {
+      return { isError: true, content: [{ type: "text", text: `COS 上传失败: ${err?.message || String(err)}` }] };
+    }
+  });
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+}
+
+main().catch((err) => {
+  console.error("[mcp-cos-upload] MCP server failed:", err);
+  process.exit(1);
+});