nx-mcp-server 1.0.0

package/src/tools/qrcode.js

@@ -0,0 +1,80 @@
+import QRCode from 'qrcode';
+import sharp from 'sharp';
+import {saveOutput,loadImage} from '../utils/output.js';
+
+export async function generateQrcode(args){
+  const {content,margin=4,darkColor='#000000',lightColor='#ffffff',errorLevel='M',logo,outputDir,watermark}=args;
+  let width=args.width||256;
+
+  // 直接输出 PNG buffer（无需 canvas）
+  const qrBuf=await QRCode.toBuffer(content,{
+    width,
+    margin,
+    color:{dark:darkColor,light:lightColor},
+    errorCorrectionLevel:errorLevel,
+    type:'png',
+  });
+
+  const composites=[];
+
+  // 如果有 logo，先用白色圆角背景遮挡 QR 码图案，再叠加 logo
+  if(logo){
+    const logoSize=Math.floor(width*0.22);
+    const bgPad=Math.floor(logoSize*0.18);
+    const bgSize=logoSize+bgPad*2;
+    const logoImg=await loadImage(logo);
+
+    // 白色圆角背景 —— 清除 QR 码中间图案，让 logo 清晰可辨
+    const whiteBg=Buffer.from(
+      `<svg xmlns="http://www.w3.org/2000/svg" width="${bgSize}" height="${bgSize}"><rect width="${bgSize}" height="${bgSize}" rx="${bgPad}" fill="#ffffff"/></svg>`
+    );
+
+    // Logo 渲染（SVG 填充分色，同时让背景透明）
+    const logoBuf=await logoImg
+      .resize(logoSize,logoSize,{fit:'inside',background:{r:255,g:255,b:255,alpha:0}})
+      .png().toBuffer();
+
+    const bgX=Math.floor((width-bgSize)/2);
+    const bgY=Math.floor((width-bgSize)/2);
+    const logoX=Math.floor((width-logoSize)/2);
+    const logoY=Math.floor((width-logoSize)/2);
+
+    composites.push({input:whiteBg,top:bgY,left:bgX});
+    composites.push({input:logoBuf,top:logoY,left:logoX});
+  }
+
+  // 文字水印（可选，一站式生成美化二维码）
+  if(watermark&&watermark.text){
+    const wm=watermark;
+    const wmFontSize=wm.fontSize||14;
+    const wmColor=wm.color||darkColor;
+    const wmOpacity=wm.opacity!=null?wm.opacity:0.4;
+    const wmPos=wm.position||'bottom';
+
+    const textY=wmPos==='top'?wmFontSize+8:width-wmFontSize/2;
+    const alpha=Math.round(wmOpacity*255).toString(16).padStart(2,'0');
+
+    const wmSvg=Buffer.from(
+      `<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${width}"><text x="${width/2}" y="${textY}" font-size="${wmFontSize}" fill="${wmColor}${alpha}" text-anchor="middle" font-family="sans-serif">${escapeXml(wm.text)}</text></svg>`
+    );
+    composites.push({input:wmSvg,top:0,left:0});
+  }
+
+  const finalBuf=composites.length>0
+    ?await sharp(qrBuf).composite(composites.map(c=>({...c,blend:'over'}))).png().toBuffer()
+    :qrBuf;
+
+  const saved=await saveOutput(finalBuf,'qrcode.png',outputDir,'url');
+  return {
+    success:true,
+    ...saved,
+    content,
+    format:'qrcode',
+    hasLogo:!!logo,
+    hasWatermark:!!(watermark&&watermark.text),
+  };
+}
+
+function escapeXml(s){
+  return s.replace(/&/g,'&amp;').replace(/</g,'&lt;').replace(/>/g,'&gt;').replace(/"/g,'&quot;');
+}

package/src/tools/smart.js

@@ -0,0 +1,52 @@
+import {tongyiText2Image} from '../providers/tongyi.js';
+import {saveOutput} from '../utils/output.js';
+import axios from 'axios';
+
+export async function generateSmart(args){
+  const images=await tongyiText2Image({
+    apiKey:args.apiKey||'',
+    prompt:args.prompt,
+    negativePrompt:args.negativePrompt||'',
+    width:args.width||1024,
+    height:args.height||1024,
+    style:args.style||'',
+    model:args.model||'wanx-v1',
+    numImages:args.numImages||1,
+  });
+
+  const results=[];
+  for(const url of images.images){
+    // 始终下载到本地，确保用户能直接看到图片
+    const res=await axios.get(url,{responseType:'arraybuffer',timeout:60000});
+    const buf=Buffer.from(res.data);
+    const saved=await saveOutput(buf,'ai_generated.png',args.outputDir,'url');
+
+    if(args.outputFormat==='base64'){
+      const b64=buf.toString('base64');
+      results.push({
+        ...saved,
+        base64:`data:image/png;base64,${b64}`,
+        mimeType:'image/png',
+        width:args.width||1024,
+        height:args.height||1024,
+      });
+    }else{
+      results.push({
+        ...saved,
+        width:args.width||1024,
+        height:args.height||1024,
+      });
+    }
+  }
+
+  return {
+    success:true,
+    model:images.model,
+    prompt:images.prompt,
+    style:args.style||'',
+    size:`${args.width||1024}×${args.height||1024}`,
+    count:results.length,
+    images:results,
+    message:`成功生成 ${results.length} 张图片，已保存到本地 output 目录`,
+  };
+}

package/src/tools/stitch.js

@@ -0,0 +1,68 @@
+import sharp from 'sharp';
+import {saveOutput,loadImage} from '../utils/output.js';
+
+export async function stitchImages(args){
+  const {sources,columns=2,gap=0,background='#ffffff',align='center',format='png',outputDir}=args;
+
+  if(!sources||sources.length<2){
+    throw new Error('至少需要 2 张图片进行拼接');
+  }
+
+  // 加载所有图片
+  const images=[];
+  for(const src of sources){
+    const img=await loadImage(src);
+    const meta=await img.metadata();
+    const buf=await img.png().toBuffer();
+    images.push({buffer:buf,width:meta.width,height:meta.height});
+  }
+
+  // 计算网格布局
+  const rows=Math.ceil(images.length/columns);
+  const cellWidth=Math.max(...images.map(i=>i.width))+gap;
+  const cellHeight=Math.max(...images.map(i=>i.height))+gap;
+  const totalWidth=cellWidth*columns+gap;
+  const totalHeight=cellHeight*rows+gap;
+
+  // 拼接
+  const composites=[];
+  for(let i=0;i<images.length;i++){
+    const img=images[i];
+    const row=Math.floor(i/columns);
+    const col=i%columns;
+    const cellX=gap+col*cellWidth;
+    const cellY=gap+row*cellHeight;
+
+    // 在单元格内对齐
+    let left=cellX,top=cellY;
+    if(align==='center'){
+      left+=Math.floor((cellWidth-gap-img.width)/2);
+      top+=Math.floor((cellHeight-gap-img.height)/2);
+    }else if(align==='bottom'){
+      top+=cellHeight-gap-img.height;
+    }
+
+    composites.push({input:img.buffer,top,left});
+  }
+
+  const pipeline=sharp({create:{width:totalWidth,height:totalHeight,channels:4,background}})
+    .composite(composites.map(c=>({...c,blend:'over'})));
+
+  let result;
+  if(format==='jpeg') result=pipeline.jpeg();
+  else if(format==='webp') result=pipeline.webp();
+  else result=pipeline.png();
+
+  const buffer=await result.toBuffer();
+  const filename=`stitch.${format}`;
+  const saved=await saveOutput(buffer,filename,outputDir,'url');
+
+  return {
+    success:true,
+    ...saved,
+    grid:`${columns}×${rows}`,
+    totalImages:images.length,
+    totalWidth,
+    totalHeight,
+  };
+}

package/src/tools/watermark.js

@@ -0,0 +1,88 @@
+import sharp from 'sharp';
+import {saveOutput,loadImage} from '../utils/output.js';
+
+const POSITIONS={
+  topLeft:{gravity:'northwest'},
+  topRight:{gravity:'northeast'},
+  bottomLeft:{gravity:'southwest'},
+  bottomRight:{gravity:'southeast'},
+  center:{gravity:'centre'},
+};
+
+export async function addWatermark(args){
+  const {src,type,content,imageSrc,position='bottomRight',margin=20,opacity=0.5,fontSize=36,color='#ffffff',angle=-30,outputDir}=args;
+
+  const base=await loadImage(src);
+  const meta=await base.metadata();
+
+  if(type==='image'&&imageSrc){
+    // 图片水印
+    const wmImg=await loadImage(imageSrc);
+    const wmMeta=await wmImg.metadata();
+    const maxWmSize=Math.min(meta.width,meta.height)*0.2;
+    const scale=Math.min(maxWmSize/wmMeta.width,maxWmSize/wmMeta.height,1);
+    const wmBuf=await wmImg
+      .resize(Math.round(wmMeta.width*scale),Math.round(wmMeta.height*scale))
+      .ensureAlpha(opacity)
+      .png().toBuffer();
+
+    const pos=POSITIONS[position]||POSITIONS.bottomRight;
+    const result=await base
+      .composite([{input:wmBuf,...pos,top:margin,left:margin,blend:'over'}])
+      .png().toBuffer();
+
+    const saved=await saveOutput(result,'watermarked.png',outputDir,'url');
+    return {success:true,...saved,type:'image'};
+  }
+
+  if(type==='text'&&content){
+    // 文字水印
+    const textSvg=await createTextOverlay(meta.width,meta.height,content,position,margin,opacity,fontSize,color,angle);
+    const result=await base
+      .composite([{input:textSvg,top:0,left:0,blend:'over'}])
+      .png().toBuffer();
+
+    const saved=await saveOutput(result,'watermarked.png',outputDir,'url');
+    return {success:true,...saved,type:'text',content};
+  }
+
+  throw new Error(`水印类型 ${type} 需要对应的 content 或 imageSrc`);
+}
+
+async function createTextOverlay(width,height,text,position,margin,opacity,fontSize,color,angle){
+  const alpha=Math.round(opacity*255).toString(16).padStart(2,'0');
+  const textColor=`${color}${alpha}`;
+
+  let svg='';
+  if(position==='tile'){
+    // 平铺模式：旋转文字重复排列
+    const spacing=fontSize*4;
+    svg=`<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}">`;
+    for(let y=spacing;y<height+spacing;y+=spacing){
+      for(let x=-spacing;x<width+spacing;x+=spacing*2){
+        svg+=`<text x="${x}" y="${y}" font-size="${fontSize}" fill="${textColor}" transform="rotate(${angle},${x},${y})" opacity="${opacity}">${escapeXml(text)}</text>`;
+      }
+    }
+    svg+='</svg>';
+  }else{
+    // 固定位置
+    const pos=POSITIONS[position]||POSITIONS.bottomRight;
+    let x=margin,y=margin;
+    if(pos.gravity.includes('east')) x=width-margin;
+    else if(pos.gravity.includes('west')) x=margin;
+    if(pos.gravity.includes('south')) y=height-margin;
+    else if(pos.gravity.includes('north')) y=margin+fontSize;
+
+    const anchor=pos.gravity.includes('west')?'start':pos.gravity.includes('east')?'end':'middle';
+    if(anchor==='middle') x=width/2;
+    if(pos.gravity.includes('north')&&pos.gravity.includes('south')) y=height/2;
+
+    svg=`<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}"><text x="${x}" y="${y}" font-size="${fontSize}" fill="${textColor}" text-anchor="${anchor}" opacity="${opacity}">${escapeXml(text)}</text></svg>`;
+  }
+
+  return Buffer.from(svg);
+}
+
+function escapeXml(s){
+  return s.replace(/&/g,'&amp;').replace(/</g,'&lt;').replace(/>/g,'&gt;').replace(/"/g,'&quot;');
+}

package/src/utils/gradient.js

@@ -0,0 +1,27 @@
+// 生成渐变背景的 SVG overlay
+export function gradientSvg(width,height,config){
+  const {type='linear',direction='to bottom',stops}=config;
+  if(!stops||!stops.length) return null;
+
+  let gradientDef;
+  if(type==='linear'){
+    const dirs={
+      'to bottom':{x1:'0%',y1:'0%',x2:'0%',y2:'100%'},
+      'to right':{x1:'0%',y1:'0%',x2:'100%',y2:'0%'},
+      'to bottom right':{x1:'0%',y1:'0%',x2:'100%',y2:'100%'},
+      'to bottom left':{x1:'100%',y1:'0%',x2:'0%',y2:'100%'},
+      'to top':{x1:'0%',y1:'100%',x2:'0%',y2:'0%'},
+    };
+    const d=dirs[direction]||dirs['to bottom'];
+    gradientDef=`<linearGradient id="g" ${Object.entries(d).map(([k,v])=>`${k}="${v}"`).join(' ')}>${stops.map(s=>`<stop offset="${s.offset||'0%'}" stop-color="${s.color}"/>`).join('')}</linearGradient>`;
+  }else{
+    gradientDef=`<radialGradient id="g" cx="50%" cy="50%">${stops.map(s=>`<stop offset="${s.offset||'0%'}" stop-color="${s.color}"/>`).join('')}</radialGradient>`;
+  }
+
+  return Buffer.from(`<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}"><defs>${gradientDef}</defs><rect width="${width}" height="${height}" fill="url(#g)"/></svg>`);
+}
+
+// 创建纯色 SVG 背景
+export function solidBgSvg(width,height,color){
+  return Buffer.from(`<svg xmlns="http://www.w3.org/2000/svg" width="${width}" height="${height}"><rect width="${width}" height="${height}" fill="${color}"/></svg>`);
+}

package/src/utils/output.js

@@ -0,0 +1,73 @@
+import {writeFile,mkdir} from 'fs/promises';
+import {join,dirname} from 'path';
+import {existsSync} from 'fs';
+
+const DEFAULT_OUTPUT_DIR=join(process.cwd(),'output');
+
+// 确保输出目录存在，返回实际输出路径
+export async function resolveOutput(filename,outputDir){
+  const dir=outputDir||DEFAULT_OUTPUT_DIR;
+  if(!existsSync(dir)) await mkdir(dir,{recursive:true});
+  const ts=Date.now();
+  const stem=filename.replace(/\.[^.]+$/,'');
+  const ext=filename.split('.').pop();
+  return join(dir,`${stem}_${ts}.${ext}`);
+}
+
+// 将 buffer 写入文件，返回路径和 base64
+export async function saveOutput(buffer,filename,outputDir,returnType='url'){
+  const filePath=await resolveOutput(filename,outputDir);
+  await writeFile(filePath,buffer);
+
+  const result={path:filePath,size:buffer.length};
+
+  if(returnType==='base64'){
+    result.base64=`data:image/${filename.split('.').pop()};base64,${buffer.toString('base64')}`;
+  }
+  if(returnType==='url'){
+    result.url=`file://${filePath.replace(/\\/g,'/')}`;
+  }
+
+  return result;
+}
+
+// 从路径或 URL 加载图片为 sharp 对象
+import sharp from 'sharp';
+import axios from 'axios';
+
+export async function loadImage(src){
+  if(src.startsWith('http://')||src.startsWith('https://')){
+    const res=await axios.get(src,{responseType:'arraybuffer'});
+    return sharp(Buffer.from(res.data));
+  }
+  return sharp(src);
+}
+
+// 解析颜色字符串为 {r,g,b,a} 或 sharp 可接受的格式
+export function parseColor(color){
+  if(!color||color==='transparent') return {r:0,g:0,b:0,alpha:0};
+  if(color.startsWith('#')){
+    const hex=color.slice(1);
+    if(hex.length===3){
+      const [r,g,b]=hex.split('').map(c=>parseInt(c+c,16));
+      return {r,g,b,alpha:1};
+    }
+    if(hex.length===6){
+      return {
+        r:parseInt(hex.slice(0,2),16),
+        g:parseInt(hex.slice(2,4),16),
+        b:parseInt(hex.slice(4,6),16),
+        alpha:1,
+      };
+    }
+    if(hex.length===8){
+      return {
+        r:parseInt(hex.slice(0,2),16),
+        g:parseInt(hex.slice(2,4),16),
+        b:parseInt(hex.slice(4,6),16),
+        alpha:parseInt(hex.slice(6,8),16)/255,
+      };
+    }
+  }
+  return {r:255,g:255,b:255,alpha:1};
+}