@playcraft/cli 0.0.19 → 0.0.22

package/dist/commands/image.js

@@ -0,0 +1,470 @@
+import { mkdirSync, readFileSync, statSync, writeFileSync } from 'fs';
+import { basename, dirname } from 'path';
+import sharp from 'sharp';
+import { AgentApiClient } from '../utils/agent-api-client.js';
+/** 单文件输入上限（字节），防止误处理巨型文件导致 OOM */
+const DEFAULT_MAX_INPUT_BYTES = 512 * 1024 * 1024;
+function getMaxInputBytes() {
+    const raw = process.env.PLAYCRAFT_IMAGE_MAX_INPUT_BYTES;
+    if (raw) {
+        const n = Number.parseInt(raw, 10);
+        if (Number.isFinite(n) && n > 0)
+            return n;
+    }
+    return DEFAULT_MAX_INPUT_BYTES;
+}
+function assertInputWithinLimit(filePath, label = 'input') {
+    const max = getMaxInputBytes();
+    const st = statSync(filePath);
+    if (!st.isFile()) {
+        console.error(`Error: ${label} is not a regular file: ${filePath}`);
+        process.exit(1);
+    }
+    if (st.size > max) {
+        console.error(`Error: ${label} file too large (${st.size} B > ${max} B). Raise PLAYCRAFT_IMAGE_MAX_INPUT_BYTES if needed.`);
+        process.exit(1);
+    }
+}
+function ensureDir(filePath) {
+    mkdirSync(dirname(filePath), { recursive: true });
+}
+function writeOutput(path, buf) {
+    ensureDir(path);
+    writeFileSync(path, buf);
+}
+function sizeLabelBytes(byteLength) {
+    const kb = byteLength / 1024;
+    return kb >= 1024 ? `${(kb / 1024).toFixed(2)}MB` : `${Math.round(kb)}KB`;
+}
+function sizeLabel(buf) {
+    return sizeLabelBytes(buf.length);
+}
+function handleError(err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    console.error(`Error: ${msg}`);
+    process.exit(1);
+}
+/** CLI --fit  → sharp resize `fit`（非法则退出并提示） */
+function parseResizeFitOption(raw) {
+    const v = (raw ?? 'cover').toLowerCase();
+    if (v === 'contain' ||
+        v === 'cover' ||
+        v === 'fill' ||
+        v === 'inside' ||
+        v === 'outside') {
+        return v;
+    }
+    console.error('Error: --fit must be contain|cover|fill|inside|outside');
+    process.exit(1);
+}
+export function registerImageCommands(program) {
+    const img = program
+        .command('image')
+        .description('素材工具：本地图片处理（sharp）；remove-background 调用后端 API');
+    // ─── resize ──────────────────────────────────────────────────
+    img.command('resize')
+        .description('缩放图片（按比例或指定宽高）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--width <n>', '目标宽度（像素）', parseInt)
+        .option('--height <n>', '目标高度（像素）', parseInt)
+        .option('--scale <factor>', '缩放比例（如 0.5 = 50%）', parseFloat)
+        .option('--fit <mode>', 'contain|cover|fill|inside|outside', 'cover')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const meta = await sharp(opts.input).metadata();
+            const w = meta.width ?? 64;
+            const h = meta.height ?? 64;
+            let targetW;
+            let targetH;
+            if (opts.scale !== undefined) {
+                targetW = Math.round(w * opts.scale);
+                targetH = Math.round(h * opts.scale);
+            }
+            else {
+                targetW = opts.width;
+                targetH = opts.height;
+            }
+            if (!targetW && !targetH) {
+                console.error('Error: provide --scale, --width, or --height');
+                process.exit(1);
+            }
+            const fit = parseResizeFitOption(opts.fit);
+            ensureDir(opts.output);
+            await sharp(opts.input)
+                .resize(targetW, targetH, { fit })
+                .toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Resized ${w}x${h} → ${targetW ?? '?'}x${targetH ?? '?'} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── crop ────────────────────────────────────────────────────
+    img.command('crop')
+        .description('裁剪图片（指定区域）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .requiredOption('--x <n>', '左边距（像素）', parseInt)
+        .requiredOption('--y <n>', '上边距（像素）', parseInt)
+        .requiredOption('--width <n>', '裁剪宽度（像素）', parseInt)
+        .requiredOption('--height <n>', '裁剪高度（像素）', parseInt)
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input)
+                .extract({ left: opts.x, top: opts.y, width: opts.width, height: opts.height })
+                .toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Cropped to ${opts.width}x${opts.height} at (${opts.x},${opts.y}) — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── rotate ──────────────────────────────────────────────────
+    img.command('rotate')
+        .description('旋转图片')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .requiredOption('--angle <n>', '旋转角度（顺时针，如 90/180/270 或任意整数）', parseInt)
+        .option('--background <color>', '旋转后露出的背景色（CSS hex，如 #000000）', '#00000000')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input)
+                .rotate(opts.angle, { background: opts.background })
+                .toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Rotated ${opts.angle}° — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── flip ────────────────────────────────────────────────────
+    img.command('flip')
+        .description('镜像翻转图片')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--direction <dir>', 'horizontal（左右）或 vertical（上下）', 'horizontal')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            let pipeline = sharp(opts.input);
+            if (opts.direction === 'horizontal') {
+                pipeline = pipeline.flop();
+            }
+            else if (opts.direction === 'vertical') {
+                pipeline = pipeline.flip();
+            }
+            else {
+                console.error('Error: --direction must be horizontal or vertical');
+                process.exit(1);
+            }
+            ensureDir(opts.output);
+            await pipeline.toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Flipped ${opts.direction} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── pad ─────────────────────────────────────────────────────
+    img.command('pad')
+        .description('扩展画布，在图片四周添加边距（保持原图居中）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--top <n>', '上边距（像素）', parseInt, 0)
+        .option('--bottom <n>', '下边距（像素）', parseInt, 0)
+        .option('--left <n>', '左边距（像素）', parseInt, 0)
+        .option('--right <n>', '右边距（像素）', parseInt, 0)
+        .option('--all <n>', '四周统一边距（覆盖单独方向设置）', parseInt)
+        .option('--background <color>', '背景色（CSS hex）', '#00000000')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const pad = opts.all !== undefined
+                ? { top: opts.all, bottom: opts.all, left: opts.all, right: opts.all }
+                : { top: opts.top, bottom: opts.bottom, left: opts.left, right: opts.right };
+            ensureDir(opts.output);
+            await sharp(opts.input)
+                .extend({ ...pad, background: opts.background })
+                .toFile(opts.output);
+            const meta = await sharp(opts.output).metadata();
+            const outSz = statSync(opts.output).size;
+            console.log(`Padded to ${meta.width}x${meta.height} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── convert ─────────────────────────────────────────────────
+    img.command('convert')
+        .description('转换图片格式（png/jpg/webp/avif）并可调整压缩质量')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径（扩展名决定格式，或用 --format 强制指定）')
+        .option('--format <fmt>', '目标格式 (png|jpg|webp|avif)，默认从输出路径扩展名推断')
+        .option('--quality <n>', '压缩质量 1-100（jpg/webp/avif 有效）', parseInt, 85)
+        .option('--lossless', '无损压缩（webp/avif 有效）')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const origSize = statSync(opts.input).size;
+            const fmt = (opts.format ?? opts.output.split('.').pop() ?? 'png').toLowerCase();
+            let pipeline = sharp(opts.input);
+            switch (fmt) {
+                case 'jpg':
+                case 'jpeg':
+                    pipeline = pipeline.jpeg({ quality: opts.quality });
+                    break;
+                case 'webp':
+                    pipeline = pipeline.webp({ quality: opts.quality, lossless: !!opts.lossless });
+                    break;
+                case 'avif':
+                    pipeline = pipeline.avif({ quality: opts.quality, lossless: !!opts.lossless });
+                    break;
+                case 'png':
+                default:
+                    pipeline = pipeline.png();
+                    break;
+            }
+            ensureDir(opts.output);
+            await pipeline.toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            const ratio = origSize > 0 ? ((1 - outSz / origSize) * 100).toFixed(1) : '0.0';
+            console.log(`Converted to ${fmt} — ${opts.output} (${sizeLabelBytes(outSz)}, ${ratio}% reduction)`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── grayscale ───────────────────────────────────────────────
+    img.command('grayscale')
+        .description('图片灰度化')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input).grayscale().toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Grayscale — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── blur ────────────────────────────────────────────────────
+    img.command('blur')
+        .description('高斯模糊')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--sigma <n>', '模糊强度（0.3–1000，越大越模糊）', parseFloat, 3)
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input).blur(opts.sigma).toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Blurred (sigma=${opts.sigma}) — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── tint ────────────────────────────────────────────────────
+    img.command('tint')
+        .description('叠加色调（做角色不同颜色变体）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .requiredOption('--color <hex>', '叠加颜色（CSS hex，如 #ff0000）')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input).tint(opts.color).toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Tinted with ${opts.color} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── trim ────────────────────────────────────────────────────
+    img.command('trim')
+        .description('自动裁掉透明或纯色边界')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--threshold <n>', '颜色容差（0-255，默认 10）', parseInt, 10)
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const before = await sharp(opts.input).metadata();
+            ensureDir(opts.output);
+            await sharp(opts.input).trim({ threshold: opts.threshold }).toFile(opts.output);
+            const after = await sharp(opts.output).metadata();
+            const outSz = statSync(opts.output).size;
+            console.log(`Trimmed ${before.width}x${before.height} → ${after.width}x${after.height} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── negate ──────────────────────────────────────────────────
+    img.command('negate')
+        .description('图片反色（颜色取反）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--no-alpha', '不反转透明通道')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            ensureDir(opts.output);
+            await sharp(opts.input).negate({ alpha: opts.alpha !== false }).toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Negated — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── overlay ─────────────────────────────────────────────────
+    img.command('overlay')
+        .description('将一张图叠在另一张图的指定位置（合成）')
+        .requiredOption('--base <path>', '底图路径')
+        .requiredOption('--overlay <path>', '叠加图路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--x <n>', '叠加图 X 偏移（像素，默认居中）', parseInt)
+        .option('--y <n>', '叠加图 Y 偏移（像素，默认居中）', parseInt)
+        .option('--gravity <g>', '对齐方向（center|north|south|east|west...，--x/--y 优先）', 'center')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.base, 'base');
+            assertInputWithinLimit(opts.overlay, 'overlay');
+            const overlayOpts = { input: opts.overlay };
+            if (opts.x !== undefined && opts.y !== undefined) {
+                overlayOpts.left = opts.x;
+                overlayOpts.top = opts.y;
+            }
+            else {
+                overlayOpts.gravity = opts.gravity;
+            }
+            ensureDir(opts.output);
+            await sharp(opts.base).composite([overlayOpts]).toFile(opts.output);
+            const meta = await sharp(opts.output).metadata();
+            const outSz = statSync(opts.output).size;
+            console.log(`Composited — ${meta.width}x${meta.height} — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── pixelate ────────────────────────────────────────────────
+    img.command('pixelate')
+        .description('像素化图片（像素风格效果）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出图片路径')
+        .option('--pixel-size <n>', '像素块大小（默认 8）', parseInt, 8)
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const meta = await sharp(opts.input).metadata();
+            const w = meta.width ?? 64;
+            const h = meta.height ?? 64;
+            const smallW = Math.max(1, Math.floor(w / opts.pixelSize));
+            const smallH = Math.max(1, Math.floor(h / opts.pixelSize));
+            ensureDir(opts.output);
+            await sharp(opts.input)
+                .resize(smallW, smallH, { kernel: 'nearest' })
+                .resize(w, h, { kernel: 'nearest' })
+                .toFile(opts.output);
+            const outSz = statSync(opts.output).size;
+            console.log(`Pixelated (block=${opts.pixelSize}px) — ${opts.output} (${sizeLabelBytes(outSz)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── sprite-sheet ────────────────────────────────────────────
+    img.command('sprite-sheet')
+        .description('将多张图合并为精灵图，并输出帧坐标 JSON（完整本地实现）')
+        .requiredOption('--inputs <paths>', '输入图片路径列表，逗号分隔')
+        .requiredOption('--output <basePath>', '输出基路径（自动生成 .png 和 .json）')
+        .option('--columns <n>', '列数（默认自动按平方根计算）', parseInt)
+        .option('--padding <n>', '帧间距（像素，默认 0）', parseInt, 0)
+        .option('--cell-width <n>', '统一格子宽度（不填则用第一张图宽度）', parseInt)
+        .option('--cell-height <n>', '统一格子高度（不填则用第一张图高度）', parseInt)
+        .action(async (opts) => {
+        try {
+            const paths = opts.inputs.split(',').map((p) => p.trim()).filter(Boolean);
+            if (!paths.length) {
+                console.error('Error: --inputs is empty');
+                process.exit(1);
+            }
+            for (let i = 0; i < paths.length; i++) {
+                assertInputWithinLimit(paths[i], `inputs[${i}]`);
+            }
+            const firstMeta = await sharp(paths[0]).metadata();
+            const cellW = opts.cellWidth ?? firstMeta.width ?? 64;
+            const cellH = opts.cellHeight ?? firstMeta.height ?? 64;
+            const padding = opts.padding;
+            const cols = opts.columns ?? Math.ceil(Math.sqrt(paths.length));
+            const rows = Math.ceil(paths.length / cols);
+            const totalW = cols * cellW + (cols - 1) * padding;
+            const totalH = rows * cellH + (rows - 1) * padding;
+            const composites = [];
+            const frames = {};
+            for (let i = 0; i < paths.length; i++) {
+                const col = i % cols;
+                const row = Math.floor(i / cols);
+                const x = col * (cellW + padding);
+                const y = row * (cellH + padding);
+                const frameName = basename(paths[i]) || `frame_${i}`;
+                const resized = await sharp(paths[i]).resize(cellW, cellH).toBuffer();
+                composites.push({ input: resized, left: x, top: y });
+                frames[frameName] = { x, y, w: cellW, h: cellH };
+            }
+            const canvas = await sharp({
+                create: { width: totalW, height: totalH, channels: 4, background: { r: 0, g: 0, b: 0, alpha: 0 } },
+            }).png().toBuffer();
+            const sheetBuf = await sharp(canvas).composite(composites).png().toBuffer();
+            const basePath = opts.output.replace(/\.(png|json)$/i, '');
+            const pngPath = `${basePath}.png`;
+            const jsonPath = `${basePath}.json`;
+            writeOutput(pngPath, sheetBuf);
+            writeOutput(jsonPath, Buffer.from(JSON.stringify({ frames, meta: { image: pngPath, size: { w: totalW, h: totalH }, scale: 1 } }, null, 2), 'utf-8'));
+            console.log(`Sprite sheet: ${pngPath} (${cols}x${rows} grid, ${paths.length} frames, ${sizeLabel(sheetBuf)})`);
+            console.log(`Frame data: ${jsonPath}`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+    // ─── remove-background（AI 背景移除，调用 backend）────────────
+    img.command('remove-background')
+        .description('AI 背景移除，输出透明 PNG（模型运行在 backend，无需沙箱加载大模型）')
+        .requiredOption('--input <path>', '输入图片路径')
+        .requiredOption('--output <path>', '输出路径（必须是 .png，保留透明通道）')
+        .action(async (opts) => {
+        try {
+            assertInputWithinLimit(opts.input);
+            const imageBase64 = readFileSync(opts.input).toString('base64');
+            const client = new AgentApiClient();
+            const result = await client.post('/remove-background', { imageBase64 });
+            const buf = Buffer.from(result.imageBase64, 'base64');
+            writeOutput(opts.output, buf);
+            const origSize = statSync(opts.input).size;
+            console.log(`Background removed: ${opts.output} (${sizeLabel(buf)}, orig ${sizeLabelBytes(origSize)})`);
+        }
+        catch (e) {
+            handleError(e);
+        }
+    });
+}