nx-mcp-server 1.0.4 → 1.0.6

package/src/index.js

@@ -1,232 +1,26 @@
 #!/usr/bin/env node
 import {Server} from '@modelcontextprotocol/sdk/server/index.js';
 import {StdioServerTransport} from '@modelcontextprotocol/sdk/server/stdio.js';
-import {CallToolRequestSchema, ListToolsRequestSchema, isInitializeRequest} from '@modelcontextprotocol/sdk/types.js';
-import {generateSmart} from './tools/smart.js';
-import {generateBasic} from './tools/basic.js';
-import {generateQrcode} from './tools/qrcode.js';
-import {generateBarcode} from './tools/barcode.js';
-import {compressImage} from './tools/compress.js';
-import {convertFormat} from './tools/convert.js';
-import {stitchImages} from './tools/stitch.js';
-import {extractColors} from './tools/colors.js';
-import {addWatermark} from './tools/watermark.js';
-
-// ===== 工具注册表 =====
-const tools=[
-  {
-    name:'generate_smart',
-    description:'智能生图：调用通义万相 API 文生图，自动下载保存到本地 output 目录',
-    inputSchema:{
-      type:'object',
-      properties:{
-        apiKey:{type:'string',description:'通义千问 API Key（可选，未传则读取环境变量 DASHSCOPE_API_KEY）'},
-        prompt:{type:'string',description:'图像描述提示词'},
-        negativePrompt:{type:'string',description:'负面提示词'},
-        width:{type:'integer',description:'宽度(px)',default:1024},
-        height:{type:'integer',description:'高度(px)',default:1024},
-        style:{type:'string',description:'风格预设',enum:['photorealistic','anime','oil-painting','watercolor','3d-render','sketch','flat-illustration']},
-        model:{type:'string',description:'模型名称',default:'wanx-v1'},
-        numImages:{type:'integer',description:'生成数量',default:1,minimum:1,maximum:4},
-        outputFormat:{type:'string',description:'输出格式',enum:['url','base64','file'],default:'url'},
-        outputDir:{type:'string',description:'输出目录'},
-      },
-      required:['prompt'],
-    },
-  },
-  {
-    name:'generate_basic',
-    description:'普通生图：本地 Canvas 合成图形',
-    inputSchema:{
-      type:'object',
-      properties:{
-        width:{type:'integer',description:'画布宽度(px)',default:800},
-        height:{type:'integer',description:'画布高度(px)',default:600},
-        background:{type:'string',description:'背景：纯色(hex)、渐变 JSON、transparent',default:'#ffffff'},
-        shapes:{type:'array',description:'图形列表',items:{type:'object',properties:{
-          type:{type:'string',enum:['rect','circle','ellipse','line','polygon']},
-          x:{type:'number'},y:{type:'number'},w:{type:'number'},h:{type:'number'},
-          r:{type:'number',description:'圆/椭圆半径或rx'},
-          ry:{type:'number',description:'椭圆ry'},
-          points:{type:'array',description:'多边形/线段端点',items:{type:'object',properties:{x:{type:'number'},y:{type:'number'}},required:['x','y']}},
-          fill:{type:'string',description:'填充色'},
-          stroke:{type:'string',description:'描边色'},
-          strokeWidth:{type:'number'},
-        },required:['type']}},
-        texts:{type:'array',description:'文字列表',items:{type:'object',properties:{
-          content:{type:'string'},x:{type:'number'},y:{type:'number'},
-          fontSize:{type:'number',default:24},color:{type:'string',default:'#000000'},
-          fontFamily:{type:'string',default:'sans-serif'},align:{type:'string',enum:['left','center','right']},
-          maxWidth:{type:'number'},
-        },required:['content','x','y']}},
-        images:{type:'array',description:'叠加图片',items:{type:'object',properties:{
-          src:{type:'string',description:'图片路径/URL'},x:{type:'number'},y:{type:'number'},
-          width:{type:'number'},height:{type:'number'},opacity:{type:'number',minimum:0,maximum:1},
-        },required:['src']}},
-        border:{type:'object',properties:{width:{type:'number'},color:{type:'string'},radius:{type:'number'}}},
-        format:{type:'string',enum:['png','jpeg','webp'],default:'png'},
-        quality:{type:'integer',minimum:1,maximum:100,default:90},
-        outputDir:{type:'string',description:'输出目录'},
-      },
-      required:['width','height'],
-    },
-  },
-  {
-    name:'generate_qrcode',
-    description:'生成二维码（支持中心logo+文字水印，一站式美化）',
-    inputSchema:{
-      type:'object',
-      properties:{
-        content:{type:'string',description:'编码内容'},
-        width:{type:'integer',description:'尺寸(px)',default:256},
-        margin:{type:'integer',description:'边距',default:4},
-        darkColor:{type:'string',default:'#000000'},
-        lightColor:{type:'string',default:'#ffffff'},
-        errorLevel:{type:'string',enum:['L','M','Q','H'],default:'M'},
-        logo:{type:'string',description:'中心logo图片路径(可选，自动加白色圆角背景遮挡QR图案)'},
-        watermark:{type:'object',description:'文字水印(可选)',properties:{
-          text:{type:'string',description:'水印文字'},
-          position:{type:'string',enum:['top','bottom'],default:'bottom'},
-          fontSize:{type:'integer',default:14},
-          color:{type:'string',description:'文字颜色'},
-          opacity:{type:'number',minimum:0,maximum:1,default:0.4},
-        }},
-        outputDir:{type:'string'},
-      },
-      required:['content'],
-    },
-  },
-  {
-    name:'generate_barcode',
-    description:'生成条形码',
-    inputSchema:{
-      type:'object',
-      properties:{
-        content:{type:'string',description:'编码内容(数字/字母)'},
-        format:{type:'string',enum:['CODE128','CODE39','EAN13','EAN8','UPC','ITF'],default:'CODE128'},
-        width:{type:'integer',description:'宽度(px)',default:400},
-        height:{type:'integer',description:'高度(px)',default:100},
-        displayValue:{type:'boolean',description:'是否显示编码文字',default:true},
-        fontSize:{type:'integer',default:16},
-        outputDir:{type:'string'},
-      },
-      required:['content'],
-    },
-  },
-  {
-    name:'compress_image',
-    description:'压缩图片（质量/尺寸）',
-    inputSchema:{
-      type:'object',
-      properties:{
-        src:{type:'string',description:'源图片路径/URL'},
-        quality:{type:'integer',minimum:1,maximum:100,description:'JPEG/WebP质量',default:80},
-        maxWidth:{type:'integer',description:'最大宽度(等比缩放)'},
-        maxHeight:{type:'integer',description:'最大高度(等比缩放)'},
-        format:{type:'string',enum:['jpeg','webp','png'],description:'输出格式',default:'jpeg'},
-        outputDir:{type:'string'},
-      },
-      required:['src'],
-    },
-  },
-  {
-    name:'convert_format',
-    description:'图片格式互转（PNG↔JPEG↔WebP↔BMP↔TIFF）',
-    inputSchema:{
-      type:'object',
-      properties:{
-        src:{type:'string',description:'源图片路径/URL'},
-        format:{type:'string',enum:['png','jpeg','webp','bmp','tiff','avif'],description:'目标格式'},
-        quality:{type:'integer',minimum:1,maximum:100,default:90},
-        background:{type:'string',description:'透明图转不透明格式时的背景色',default:'#ffffff'},
-        outputDir:{type:'string'},
-      },
-      required:['src','format'],
-    },
-  },
-  {
-    name:'stitch_images',
-    description:'多图拼接/网格布局',
-    inputSchema:{
-      type:'object',
-      properties:{
-        sources:{type:'array',description:'图片路径列表',items:{type:'string'}},
-        columns:{type:'integer',description:'列数',default:2},
-        gap:{type:'integer',description:'间距(px)',default:0},
-        background:{type:'string',default:'#ffffff'},
-        align:{type:'string',enum:['top','center','bottom'],default:'center'},
-        format:{type:'string',enum:['png','jpeg','webp'],default:'png'},
-        outputDir:{type:'string'},
-      },
-      required:['sources'],
-    },
-  },
-  {
-    name:'extract_colors',
-    description:'提取图片主色调/调色板',
-    inputSchema:{
-      type:'object',
-      properties:{
-        src:{type:'string',description:'图片路径/URL'},
-        count:{type:'integer',description:'提取颜色数量',default:5,minimum:2,maximum:20},
-        format:{type:'string',enum:['hex','rgb','hsl'],default:'hex'},
-      },
-      required:['src'],
-    },
-  },
-  {
-    name:'add_watermark',
-    description:'添加文字/图片水印',
-    inputSchema:{
-      type:'object',
-      properties:{
-        src:{type:'string',description:'源图片路径/URL'},
-        type:{type:'string',enum:['text','image'],description:'水印类型'},
-        content:{type:'string',description:'文字水印内容'},
-        imageSrc:{type:'string',description:'图片水印路径'},
-        position:{type:'string',enum:['topLeft','topRight','bottomLeft','bottomRight','center','tile'],default:'bottomRight'},
-        margin:{type:'integer',default:20},
-        opacity:{type:'number',minimum:0,maximum:1,default:0.5},
-        fontSize:{type:'integer',default:36},
-        color:{type:'string',default:'#ffffff'},
-        fontSize:{type:'integer',default:36},
-        angle:{type:'integer',description:'旋转角度(仅tile模式)',default:-30},
-        outputDir:{type:'string'},
-      },
-      required:['src','type'],
-    },
-  },
-];
-
-// ===== 工具路由 =====
-const handlers={
-  generate_smart:generateSmart,
-  generate_basic:generateBasic,
-  generate_qrcode:generateQrcode,
-  generate_barcode:generateBarcode,
-  compress_image:compressImage,
-  convert_format:convertFormat,
-  stitch_images:stitchImages,
-  extract_colors:extractColors,
-  add_watermark:addWatermark,
-};
+import {CallToolRequestSchema,ListToolsRequestSchema,isInitializeRequest} from '@modelcontextprotocol/sdk/types.js';
+import {definitions} from './tools/definitions.js';
+import {handlers} from './tools/handlers.js';
 
 // ===== 创建 MCP Server =====
 function createServer(){
   const server=new Server(
-    {name:'nx-mcp-server',version:'1.0.0'},
+    {name:'nx-mcp-server',version:'1.0.5'},
     {capabilities:{tools:{}}},
   );
 
   server.setRequestHandler(ListToolsRequestSchema,async()=>{
-    return {tools};
+    return {tools:definitions};
   });
 
   server.setRequestHandler(CallToolRequestSchema,async(request)=>{
     const {name,arguments:args}=request.params;
     const handler=handlers[name];
     if(!handler){
-      return {content:[{type:'text',text:`Unknown tool: ${name}`}],isError:true};
+      return {content:[{type:'text',text:`未知工具: ${name}`}],isError:true};
     }
     try{
       const result=await handler(args);
@@ -235,7 +29,7 @@ function createServer(){
       };
     }catch(err){
       return {
-        content:[{type:'text',text:`Error: ${err.message}`}],
+        content:[{type:'text',text:`错误: ${err.message}`}],
         isError:true,
       };
     }
@@ -252,7 +46,7 @@ async function startHttp(){
 
   const PORT=parseInt(process.env.MCP_PORT||'3000',10);
   const app=createMcpExpressApp({host:'0.0.0.0'});
-  const transports={};// sessionId → transport
+  const transports={};
 
   const mcpPostHandler=async (req,res)=>{
     const sessionId=req.headers['mcp-session-id'];
@@ -261,7 +55,6 @@ async function startHttp(){
     if(sessionId&&transports[sessionId]){
       transport=transports[sessionId];
     }else if(!sessionId&&isInitializeRequest(req.body)){
-      // 新会话初始化
       const eventStore={
         events:[],
         getLastEventId:()=>eventStore.events.length.toString(),
@@ -312,7 +105,6 @@ async function startHttp(){
   app.get('/mcp',mcpGetHandler);
   app.delete('/mcp',mcpDeleteHandler);
 
-  // 健康检查
   app.get('/health',(_req,res)=>{res.json({status:'ok',uptime:process.uptime()});});
 
   app.listen(PORT,()=>{
@@ -320,7 +112,7 @@ async function startHttp(){
   });
 }
 
-// ===== Stdio 模式（本地 Claude Code） =====
+// ===== Stdio 模式（本地 Claude Code / Cherry Studio） =====
 async function startStdio(){
   const server=createServer();
   const transport=new StdioServerTransport();

package/src/nx-api.js

@@ -0,0 +1,209 @@
+import {getConfig} from './config.js';
+import {readFileSync,existsSync,writeFileSync,mkdirSync} from 'fs';
+import {basename,dirname,join} from 'path';
+
+// 判断是否为本地文件路径
+function isFilePath(str){
+  return /^[A-Za-z]:[\\/]/.test(str)||str.startsWith('/');
+}
+
+// 从不同来源加载图片 Buffer
+async function loadImageBuffer(src){
+  // 本地文件路径
+  if(isFilePath(src)){
+    if(!existsSync(src)) throw new Error(`文件不存在: ${src}`);
+    const buf=readFileSync(src);
+    const ext=basename(src).split('.').pop()||'png';
+    const mime=ext==='jpg'?'jpeg':ext;
+    return {buffer:buf,name:basename(src),mime:`image/${mime}`};
+  }
+  // data URL
+  if(src.startsWith('data:')){
+    const [head,base64]=src.split(',');
+    if(!base64) throw new Error('data URL 格式无效');
+    const mime=head.match(/:(.*?);/)?.[1]||'image/png';
+    const ext=(mime.split('/')[1]||'png').replace('jpeg','jpg');
+    const buf=Buffer.from(base64,'base64');
+    return {buffer:buf,name:`image.${ext}`,mime};
+  }
+  // HTTP(S) URL
+  const res=await fetch(src);
+  if(!res.ok) throw new Error(`下载失败 HTTP ${res.status}`);
+  const buf=Buffer.from(await res.arrayBuffer());
+  const ext=src.split('.').pop().split('?')[0]||'png';
+  const contentType=res.headers.get('content-type')||'';
+  const mime=contentType||`image/${ext==='jpg'?'jpeg':ext}`;
+  const name=src.split('/').pop().split('?')[0]||`image.${ext}`;
+  return {buffer:buf,name,mime};
+}
+
+// 下载 CDN 压缩图并写入本地
+async function downloadToFile(cdnUrl,localPath){
+  const res=await fetch(cdnUrl);
+  if(!res.ok) throw new Error(`下载 CDN 失败 HTTP ${res.status}`);
+  const buf=Buffer.from(await res.arrayBuffer());
+  const dir=dirname(localPath);
+  if(!existsSync(dir)) mkdirSync(dir,{recursive:true});
+  writeFileSync(localPath,buf);
+  return buf.length;
+}
+
+// ===== 图片压缩 API 客户端 =====
+// 接口文档: md/compressImage-api.md
+// POST /v1/nx/compressImage
+// 两种请求模式：
+//   模式A（JSON）：纯 HTTP(S) urls → Content-Type: application/json  → body: {urls,quality}
+//   模式B（FormData）：本地文件 / dataUrl / files → multipart/form-data → body: FormData{files,quality}
+
+function isHttpUrl(str){
+  return str&&(str.startsWith('http://')||str.startsWith('https://'));
+}
+
+async function compressImage({urls=[],files=[],quality=90,output=null,apiKey=null}){
+  const {api}=getConfig();
+
+  // API Key：参数传入 > 环境变量 NX_API_KEY
+  const key=apiKey||api.apiKey;
+  if(!key){
+    throw new Error('未配置 API Key，请设置环境变量 NX_API_KEY 或传入 apiKey 参数');
+  }
+
+  // 合并所有来源，保持顺序和原始引用
+  const sources=[...urls,...files];
+  if(!sources.length) throw new Error('请提供 urls 或 files 参数，至少需要一个');
+
+  // 判断请求模式：全是 HTTP URL → JSON 模式；否则 → FormData 模式
+  const allHttp=sources.every(u=>isHttpUrl(u));
+  const useJson=allHttp;
+  if(!allHttp){
+    const hasLocal=sources.some(u=>isFilePath(u));
+    if(hasLocal) console.warn('💡 检测到本地文件路径，仅本地 MCP Server 可用。远程部署请传 HTTP(S) 图片链接。');
+  }
+
+  const apiUrl=`${api.baseUrl}/v1/nx/compressImage`;
+  const controller=new AbortController();
+  const timer=setTimeout(()=>controller.abort(),180000);
+
+  let body,headers;
+  if(useJson){
+    // ===== 模式A：JSON 请求（纯 HTTP URL） =====
+    headers={
+      'Content-Type':'application/json',
+      'Authorization':`Bearer ${key}`,
+    };
+    body=JSON.stringify({urls:sources,quality});
+  }else{
+    // ===== 模式B：FormData 请求（本地文件/dataUrl/files） =====
+    const fd=new FormData();
+    const loaded=await Promise.allSettled(sources.map(async (src)=>{
+      try{
+        const {buffer,name,mime}=await loadImageBuffer(src);
+        return {blob:new Blob([buffer],{type:mime}),name,src};
+      }catch(e){
+        throw new Error(`加载 ${src} 失败: ${e.message}`);
+      }
+    }));
+    loaded.forEach((r)=>{
+      if(r.status==='fulfilled'){
+        fd.append('files',r.value.blob,r.value.name);
+      }else{
+        throw new Error(r.reason.message);
+      }
+    });
+    fd.append('quality',String(quality));
+    headers={'Authorization':`Bearer ${key}`};
+    body=fd;
+  }
+
+  let res;
+  try{
+    res=await fetch(apiUrl,{method:'POST',headers,body,signal:controller.signal});
+  }catch(err){
+    clearTimeout(timer);
+    if(err.name==='AbortError'){
+      throw new Error('压缩请求超时 (180s)，图片可能过大，建议换小图重试');
+    }
+    throw new Error(`网络请求失败: ${err.message}`);
+  }
+  clearTimeout(timer);
+
+  const data=await res.json();
+
+  if(data.code!==0){
+    const errType=data.error_type||'unknown';
+    const errMap={
+      missing_auth:'未提供 Authorization 头',
+      auth_failed:'API Key 无效',
+      auth_exception:'认证服务异常，请重试',
+      missing_params:'缺少 urls 或 files 参数',
+    };
+    const hint=errMap[errType]||data.message||'未知错误';
+    throw new Error(`压缩API错误 [${errType}]: ${hint}`);
+  }
+
+  const rawItems=(data.data&&data.data.items)||[];
+  const successCount=rawItems.filter(it=>!it.error).length;
+  const failCount=rawItems.filter(it=>it.error).length;
+
+  // 构建 items，originalUrl 保留原始输入
+  const items=rawItems.map((it,i)=>({
+    originalUrl:sources[i]||it.original_url,
+    compressedUrl:it.compressed_url||null,
+    originalSize:it.original_size,
+    compressedSize:it.compressed_size,
+    ratio:it.ratio||'0%',
+    error:it.error||null,
+  }));
+
+  // ===== 处理 output =====
+  let outputResult=null;
+  if(output&&items.length){
+    const compressed=items.filter(it=>it.compressedUrl);
+    if(output==='overwrite'){
+      let overwritten=0;
+      for(const it of compressed){
+        if(isFilePath(it.originalUrl)){
+          try{
+            const size=await downloadToFile(it.compressedUrl,it.originalUrl);
+            it.outputPath=it.originalUrl;
+            it.outputSize=size;
+            overwritten++;
+          }catch(e){
+            it.outputError=e.message;
+          }
+        }
+      }
+      outputResult={mode:'overwrite',count:overwritten};
+    }else{
+      let saved=0;
+      for(const it of compressed){
+        try{
+          const name=basename(it.originalUrl)||`compressed_${Date.now()}`;
+          const outPath=join(output,name);
+          const size=await downloadToFile(it.compressedUrl,outPath);
+          it.outputPath=outPath;
+          it.outputSize=size;
+          saved++;
+        }catch(e){
+          it.outputError=e.message;
+        }
+      }
+      outputResult={mode:'save',dir:output,count:saved};
+    }
+  }
+
+  return {
+    success:failCount===0,
+    message:data.message||`压缩完成: 成功 ${successCount}, 失败 ${failCount}`,
+    mode:useJson?'json':'formdata',
+    items,
+    summary:{
+      total:items.length,
+      success:successCount,
+      failed:failCount,
+    },
+    ...(outputResult&&{output:outputResult}),
+  };
+}
+
+export {compressImage};

package/src/tools/definitions.js

@@ -0,0 +1,19 @@
+// MCP 工具定义
+
+const definitions=[
+  {
+    name:'nx_compress',
+    description:'远程智能压缩图片。两种方式：①传 urls（HTTP链接或本地路径）；②传 files（dataUrl 格式）。支持 PNG/JPEG/BMP/WebP/TGA。支持 output 参数直接写入本地文件',
+    inputSchema:{
+      type:'object',
+      properties:{
+        urls:{type:'array',description:'[方式一] 图片URL列表，支持 HTTP(S) 链接、本地路径、data URL。远程调用请传 HTTP(S) 链接',items:{type:'string'}},
+        files:{type:'array',description:'[方式二] 图片 dataUrl 列表（data:image/...;base64,...）。仅 FormData 模式，适合本地大图',items:{type:'string'}},
+        quality:{type:'integer',description:'压缩质量 1-100，数值越大质量越高',minimum:1,maximum:100,default:90},
+        output:{type:'string',description:'输出方式："overwrite" 覆盖原文件（仅本地路径有效），或指定输出目录路径如 E:/output/'},
+      },
+    },
+  },
+];
+
+export {definitions};

package/src/tools/handlers.js

@@ -0,0 +1,15 @@
+import {compressImage} from '../nx-api.js';
+
+// ===== 工具处理函数 =====
+
+async function handleCompress(args){
+  return compressImage({urls:args.urls,files:args.files,quality:args.quality,output:args.output});
+}
+
+// ===== 工具路由表 =====
+
+const handlers={
+  nx_compress:handleCompress,
+};
+
+export {handlers};

package/.claude/settings.json

@@ -1,11 +0,0 @@
-{
-  "mcpServers": {
-    "nx-mcp": {
-      "command": "node",
-      "args": ["src/index.js"],
-      "env": {
-        "DASHSCOPE_API_KEY": ""
-      }
-    }
-  }
-}