yzsd-image 1.0.0

package/PUBLISH.md

@@ -0,0 +1,138 @@
+# 发布到 npm 的步骤
+
+## 发布前准备
+
+1. **确保你有 npm 账号**
+   ```bash
+   npm login
+   ```
+
+2. **启用双因素认证 (2FA)**
+   - 访问 https://www.npmjs.com/settings/[你的用户名]/tfa
+   - 启用 2FA (使用 Google Authenticator 或其他 2FA 应用)
+   - npm 现在要求所有发布者启用 2FA
+
+3. **检查 package.json 中的信息**
+   - 确认 `name` 是唯一的(可以先在 npmjs.com 搜索)
+   - 填写 `author` 字段
+   - 确认 `version` 号
+
+4. **测试构建**
+   ```bash
+   npm run build
+   ```
+
+## 发布步骤
+
+```bash
+cd D:\Workspace\Projects+++++\yzsd-image\yzsd-image
+
+# 登录 npm (如果还没登录)
+npm login
+
+# 发布到 npm
+npm publish
+```
+
+## 发布后使用
+
+用户可以通过以下方式使用:
+
+### 方式 1: npx (推荐,无需安装)
+
+在 Claude Desktop 配置文件中添加:
+
+**Windows**: `%APPDATA%\Claude\claude_desktop_config.json`
+**macOS**: `~/Library/Application Support/Claude/claude_desktop_config.json`
+
+```json
+{
+  "mcpServers": {
+    "yzsd-image": {
+      "command": "npx",
+      "args": ["-y", "yzsd-image"],
+      "env": {
+        "YZSD_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+### 方式 2: 全局安装
+
+```bash
+npm install -g yzsd-image
+```
+
+然后配置:
+
+```json
+{
+  "mcpServers": {
+    "yzsd-image": {
+      "command": "yzsd-image",
+      "env": {
+        "YZSD_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+## 更新版本
+
+修改代码后发布新版本:
+
+```bash
+# 更新补丁版本 (1.0.0 -> 1.0.1)
+npm version patch
+
+# 更新次版本 (1.0.0 -> 1.1.0)
+npm version minor
+
+# 更新主版本 (1.0.0 -> 2.0.0)
+npm version major
+
+# 发布
+npm publish
+```
+
+## 项目结构说明
+
+```
+yzsd-image/
+├── src/
+│   └── index.ts          # MCP server 主文件
+├── dist/                 # 编译后的文件 (npm 发布这个)
+│   └── index.js
+├── package.json          # npm 配置
+├── tsconfig.json         # TypeScript 配置
+├── README.md             # 使用文档
+├── .gitignore
+└── .npmignore
+```
+
+## 功能说明
+
+### 提供的 MCP 工具
+
+1. **generate_image** - 文生图
+   - prompt: 图片描述
+   - aspectRatio: 宽高比 (1:1, 16:9, 9:16, 等)
+   - size: 分辨率 (1K, 2K, 4K)
+   - outputPath: 输出路径
+
+2. **edit_image** - 图生图编辑
+   - inputPath: 输入图片路径
+   - prompt: 编辑描述
+   - aspectRatio: 输出宽高比
+   - outputPath: 输出路径
+
+### API 端点
+
+连接到: `https://token.1001xr.asia:8443`
+
+API 端点:
+- `/v1/images/generate` - 生成图片
+- `/v1/images/edit` - 编辑图片

package/README.md

@@ -0,0 +1,146 @@
+# yzsd-image
+
+MCP server for YZSD AI image generation. Generate and edit images using AI through the Model Context Protocol.
+
+## Features
+
+- **Text-to-Image**: Generate images from text descriptions
+- **Image-to-Image**: Edit existing images with AI
+- **Multiple Aspect Ratios**: 1:1, 16:9, 9:16, 4:3, 3:4, 3:2, 2:3, 21:9, 5:4, 4:5
+- **3 Resolution Levels**: 1K (fast), 2K (recommended), 4K (high quality)
+- **MCP Compatible**: Works with Claude Desktop, Cline, and other MCP clients
+
+## Installation
+
+### Using npx (Recommended)
+
+No installation needed! Configure directly in your MCP client:
+
+```json
+{
+  "mcpServers": {
+    "yzsd-image": {
+      "command": "npx",
+      "args": ["-y", "yzsd-image"],
+      "env": {
+        "YZSD_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+### Global Installation
+
+```bash
+npm install -g yzsd-image
+```
+
+Then configure:
+
+```json
+{
+  "mcpServers": {
+    "yzsd-image": {
+      "command": "yzsd-image",
+      "env": {
+        "YZSD_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+## Configuration
+
+### Claude Desktop
+
+Edit your Claude Desktop config file:
+
+**macOS**: `~/Library/Application Support/Claude/claude_desktop_config.json`
+**Windows**: `%APPDATA%\Claude\claude_desktop_config.json`
+
+Add:
+
+```json
+{
+  "mcpServers": {
+    "yzsd-image": {
+      "command": "npx",
+      "args": ["-y", "yzsd-image"],
+      "env": {
+        "YZSD_API_KEY": "your-api-key-here"
+      }
+    }
+  }
+}
+```
+
+### Cline (VS Code Extension)
+
+Add to your Cline MCP settings:
+
+```json
+{
+  "yzsd-image": {
+    "command": "npx",
+    "args": ["-y", "yzsd-image"],
+    "env": {
+      "YZSD_API_KEY": "your-api-key-here"
+    }
+  }
+}
+```
+
+## Usage
+
+Once configured, you can use the tools in your MCP client:
+
+### Generate Image
+
+```
+Generate a beautiful landscape image with mountains and sunset, 16:9 aspect ratio, 2K resolution
+```
+
+The MCP server provides the `generate_image` tool with parameters:
+- `prompt` (required): Text description
+- `aspectRatio` (optional): Default "1:1"
+- `size` (optional): "1K", "2K", or "4K", default "2K"
+- `outputPath` (optional): Where to save the image
+
+### Edit Image
+
+```
+Edit the image at ./photo.jpg, change the background to a beach scene
+```
+
+The MCP server provides the `edit_image` tool with parameters:
+- `inputPath` (required): Path to input image
+- `prompt` (required): Edit description
+- `aspectRatio` (optional): Default "1:1"
+- `outputPath` (optional): Where to save edited image
+
+## API Endpoint
+
+This MCP server connects to: `https://token.1001xr.asia:8443`
+
+## Development
+
+```bash
+# Clone the repository
+git clone <your-repo-url>
+cd yzsd-image
+
+# Install dependencies
+npm install
+
+# Build
+npm run build
+
+# Test locally
+node dist/index.js
+```
+
+## License
+
+MIT

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,213 @@
+#!/usr/bin/env node
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import axios from "axios";
+import * as fs from "fs/promises";
+import * as path from "path";
+const API_BASE_URL = "https://token.1001xr.asia:8443";
+const VALID_ASPECT_RATIOS = [
+    "1:1", "16:9", "9:16", "4:3", "3:4",
+    "3:2", "2:3", "21:9", "5:4", "4:5"
+];
+const VALID_SIZES = ["1K", "2K", "4K"];
+class YZSDImageServer {
+    server;
+    apiKey;
+    constructor() {
+        this.server = new Server({
+            name: "yzsd-image",
+            version: "1.0.0",
+        }, {
+            capabilities: {
+                tools: {},
+            },
+        });
+        this.apiKey = process.env.YZSD_API_KEY;
+        this.setupHandlers();
+    }
+    setupHandlers() {
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+            tools: this.getTools(),
+        }));
+        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+            const { name, arguments: args } = request.params;
+            try {
+                switch (name) {
+                    case "generate_image":
+                        return await this.handleGenerateImage(args);
+                    case "edit_image":
+                        return await this.handleEditImage(args);
+                    default:
+                        throw new Error(`Unknown tool: ${name}`);
+                }
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: `Error: ${errorMessage}`,
+                        },
+                    ],
+                    isError: true,
+                };
+            }
+        });
+    }
+    getTools() {
+        return [
+            {
+                name: "generate_image",
+                description: "Generate an AI image from text prompt using YZSD API. Supports multiple aspect ratios (1:1, 16:9, 9:16, 4:3, 3:4, 3:2, 2:3, 21:9, 5:4, 4:5) and resolutions (1K, 2K, 4K).",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        prompt: {
+                            type: "string",
+                            description: "Text description of the image to generate",
+                        },
+                        aspectRatio: {
+                            type: "string",
+                            description: "Aspect ratio for the image (default: 1:1)",
+                            enum: VALID_ASPECT_RATIOS,
+                        },
+                        size: {
+                            type: "string",
+                            description: "Resolution size: 1K (fast), 2K (recommended), 4K (high quality)",
+                            enum: VALID_SIZES,
+                        },
+                        outputPath: {
+                            type: "string",
+                            description: "Output file path (default: ./output.png)",
+                        },
+                        apiKey: {
+                            type: "string",
+                            description: "YZSD API key (optional, can use YZSD_API_KEY env var)",
+                        },
+                    },
+                    required: ["prompt"],
+                },
+            },
+            {
+                name: "edit_image",
+                description: "Edit an existing image using AI based on text prompt. Upload a local image and describe the modifications you want.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        inputPath: {
+                            type: "string",
+                            description: "Path to the input image file",
+                        },
+                        prompt: {
+                            type: "string",
+                            description: "Description of how to edit the image",
+                        },
+                        aspectRatio: {
+                            type: "string",
+                            description: "Output aspect ratio (default: 1:1)",
+                            enum: VALID_ASPECT_RATIOS,
+                        },
+                        outputPath: {
+                            type: "string",
+                            description: "Output file path (default: ./edited.png)",
+                        },
+                        apiKey: {
+                            type: "string",
+                            description: "YZSD API key (optional, can use YZSD_API_KEY env var)",
+                        },
+                    },
+                    required: ["inputPath", "prompt"],
+                },
+            },
+        ];
+    }
+    resolveApiKey(providedKey) {
+        const key = providedKey || this.apiKey;
+        if (!key) {
+            throw new Error("API key not found. Set YZSD_API_KEY environment variable or provide apiKey parameter.");
+        }
+        return key;
+    }
+    async handleGenerateImage(args) {
+        const { prompt, aspectRatio = "1:1", size = "2K", outputPath = "./output.png", apiKey, } = args;
+        const key = this.resolveApiKey(apiKey);
+        if (!VALID_ASPECT_RATIOS.includes(aspectRatio)) {
+            throw new Error(`Invalid aspect ratio. Must be one of: ${VALID_ASPECT_RATIOS.join(", ")}`);
+        }
+        if (!VALID_SIZES.includes(size)) {
+            throw new Error(`Invalid size. Must be one of: ${VALID_SIZES.join(", ")}`);
+        }
+        const response = await axios.post(`${API_BASE_URL}/v1/images/generate`, {
+            prompt,
+            aspect_ratio: aspectRatio,
+            size,
+        }, {
+            headers: {
+                "Authorization": `Bearer ${key}`,
+                "Content-Type": "application/json",
+            },
+            timeout: size === "4K" ? 120000 : size === "2K" ? 60000 : 30000,
+        });
+        const imageData = response.data.image || response.data.data?.[0]?.b64_json;
+        if (!imageData) {
+            throw new Error("No image data in response");
+        }
+        const buffer = Buffer.from(imageData, "base64");
+        const outputDir = path.dirname(outputPath);
+        await fs.mkdir(outputDir, { recursive: true });
+        await fs.writeFile(outputPath, buffer);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✓ Image generated successfully\n\nPrompt: ${prompt}\nAspect Ratio: ${aspectRatio}\nSize: ${size}\nOutput: ${outputPath}`,
+                },
+            ],
+        };
+    }
+    async handleEditImage(args) {
+        const { inputPath, prompt, aspectRatio = "1:1", outputPath = "./edited.png", apiKey, } = args;
+        const key = this.resolveApiKey(apiKey);
+        if (!VALID_ASPECT_RATIOS.includes(aspectRatio)) {
+            throw new Error(`Invalid aspect ratio. Must be one of: ${VALID_ASPECT_RATIOS.join(", ")}`);
+        }
+        const imageBuffer = await fs.readFile(inputPath);
+        const base64Image = imageBuffer.toString("base64");
+        const response = await axios.post(`${API_BASE_URL}/v1/images/edit`, {
+            image: base64Image,
+            prompt,
+            aspect_ratio: aspectRatio,
+        }, {
+            headers: {
+                "Authorization": `Bearer ${key}`,
+                "Content-Type": "application/json",
+            },
+            timeout: 120000,
+        });
+        const imageData = response.data.image || response.data.data?.[0]?.b64_json;
+        if (!imageData) {
+            throw new Error("No image data in response");
+        }
+        const buffer = Buffer.from(imageData, "base64");
+        const outputDir = path.dirname(outputPath);
+        await fs.mkdir(outputDir, { recursive: true });
+        await fs.writeFile(outputPath, buffer);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✓ Image edited successfully\n\nInput: ${inputPath}\nPrompt: ${prompt}\nAspect Ratio: ${aspectRatio}\nOutput: ${outputPath}`,
+                },
+            ],
+        };
+    }
+    async run() {
+        const transport = new StdioServerTransport();
+        await this.server.connect(transport);
+        console.error("YZSD Image MCP Server running on stdio");
+    }
+}
+const server = new YZSDImageServer();
+server.run().catch(console.error);

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "yzsd-image",
+  "version": "1.0.0",
+  "description": "MCP server for YZSD AI image generation",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "yzsd-image": "dist/index.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "prepare": "npm run build",
+    "start": "node dist/index.js"
+  },
+  "keywords": [
+    "mcp",
+    "ai",
+    "image-generation",
+    "yzsd"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.4",
+    "axios": "^1.7.9"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.5",
+    "typescript": "^5.7.3"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}