ss-tools-duck 1.0.0

package/duck_decode.js

@@ -0,0 +1,161 @@
+/**
+ * duck_decode.js
+ *
+ * Node.js 解码器：从鸭子图还原隐藏的数据
+ * 对应 Python duck_decode_node.py 的核心逻辑（去除 ComfyUI 节点壳）
+ *
+ * 还原类型：
+ *   - 图片（ext = "png"）→ PNG Buffer
+ *   - 文本（ext = "txt"）→ string
+ *   - MP4（ext = "mp4" 或 "mp4.binpng"）→ MP4 Buffer
+ *   - 其他任意二进制（ext = 任意）→ Buffer
+ */
+
+'use strict';
+
+const fs   = require('fs');
+const path = require('path');
+const sharp = require('sharp');
+
+const {
+  extractPayloadWithK,
+  parseHeader,
+  binaryImageToBytes,
+} = require('./duck_payload_exporter');
+
+// ─── 主解码函数 ───────────────────────────────────────────────────────────────
+
+/**
+ * 从鸭子图中解码出隐藏的数据
+ *
+ * 自动尝试 LSB k=2、k=6、k=8 三种模式（与 Python 一致）
+ *
+ * @param {object}         opts
+ * @param {Buffer|string}  opts.duckImage   鸭子图（PNG Buffer 或文件路径）
+ * @param {string}         [opts.password]  解密密码（无加密时留空）
+ * @param {string}         [opts.outputDir] 若提供则把还原文件写入此目录
+ * @returns {Promise<DecodeResult>}
+ *
+ * @typedef {object} DecodeResult
+ * @property {Buffer}  data        原始数据字节
+ * @property {string}  ext         文件扩展名
+ * @property {string}  [filePath]  写出的文件路径（outputDir 不为空时有值）
+ * @property {string}  [text]      ext='txt' 时的 UTF-8 文本内容
+ * @property {Buffer}  [imageBuffer] ext='png' 时的 PNG Buffer（即 data 本身）
+ * @property {Buffer}  [mp4Buffer]   ext='mp4' 或 'mp4.binpng' 时的 MP4 Buffer
+ */
+async function decodeDuckImage({ duckImage, password = '', outputDir = null }) {
+  // ── 加载图片 → 原始 RGB 像素 ─────────────────────────────────────────────
+  const inputBuffer = typeof duckImage === 'string'
+    ? await fs.promises.readFile(duckImage)
+    : duckImage;
+
+  const { data: rawPixels, info } = await sharp(inputBuffer)
+    .toColorspace('srgb')
+    .removeAlpha()
+    .raw()
+    .toBuffer({ resolveWithObject: true });
+
+  const { width, height } = info;
+
+  // ── 自动探测 LSB 位数（k = 2 → 6 → 8） ──────────────────────────────────
+  let payload  = null;
+  let parsed   = null;
+  let lastErr  = null;
+
+  for (const k of [2, 6, 8]) {
+    try {
+      payload = extractPayloadWithK(rawPixels, width, height, k);
+      parsed  = parseHeader(payload, password);
+      break; // 成功则退出循环
+    } catch (err) {
+      lastErr = err;
+      payload = null;
+      parsed  = null;
+    }
+  }
+
+  if (!parsed) {
+    throw lastErr || new Error('解析失败，无法从图片中提取有效数据');
+  }
+
+  const { data, ext } = parsed;
+  const result        = { data, ext };
+
+  // ── 按扩展名分发处理 ──────────────────────────────────────────────────────
+  if (ext.toLowerCase() === 'txt') {
+    // 文本：解码 UTF-8，回退 GBK（Node.js 原生不支持 GBK，回退时保留原始 Buffer）
+    let text = '';
+    try {
+      text = data.toString('utf-8');
+    } catch {
+      try {
+        // 如果系统安装了 iconv-lite，可以尝试 GBK 解码
+        const iconv = require('iconv-lite');
+        text = iconv.decode(data, 'gbk');
+      } catch {
+        text = `[无法解码文本内容，原始数据已存储于 data 字段]`;
+      }
+    }
+    result.text = text;
+
+  } else if (ext.toLowerCase().endsWith('.binpng') || ext.toLowerCase() === 'mp4') {
+    // 视频：binpng → 还原 MP4 字节，或直接使用 MP4 字节
+    let mp4Bytes;
+    if (ext.toLowerCase().endsWith('.binpng')) {
+      // binpng 格式：data 本身就是一张 PNG（像素 = MP4 字节）
+      mp4Bytes = await binaryImageToBytes(data);
+    } else {
+      mp4Bytes = data;
+    }
+    result.mp4Buffer = mp4Bytes;
+
+  } else if (ext.toLowerCase() === 'png') {
+    result.imageBuffer = data;
+
+  }
+  // 其他格式（jpg、zip 等）：data 即原始字节，调用方自行处理
+
+  // ── 可选：写出文件 ────────────────────────────────────────────────────────
+  if (outputDir) {
+    await fs.promises.mkdir(outputDir, { recursive: true });
+    const safeName = 'duck_recovered.' + (ext.endsWith('.binpng') ? 'mp4' : ext.replace(/[/\\?%*:|"<>]/g, '_'));
+    const filePath = path.join(outputDir, safeName);
+
+    const bytesToWrite = result.mp4Buffer || result.imageBuffer || data;
+    await fs.promises.writeFile(filePath, bytesToWrite);
+    result.filePath = filePath;
+  }
+
+  return result;
+}
+
+/**
+ * 从本地文件解码（便捷方法）
+ *
+ * @param {string}  duckFilePath   鸭子图文件路径
+ * @param {string}  [password]
+ * @param {string}  [outputDir]
+ * @returns {Promise<DecodeResult>}
+ */
+async function decodeFromFile(duckFilePath, password = '', outputDir = null) {
+  return decodeDuckImage({ duckImage: duckFilePath, password, outputDir });
+}
+
+/**
+ * 从 Buffer 解码（便捷方法）
+ *
+ * @param {Buffer}  duckBuffer     鸭子图 PNG Buffer
+ * @param {string}  [password]
+ * @param {string}  [outputDir]
+ * @returns {Promise<DecodeResult>}
+ */
+async function decodeFromBuffer(duckBuffer, password = '', outputDir = null) {
+  return decodeDuckImage({ duckImage: duckBuffer, password, outputDir });
+}
+
+module.exports = {
+  decodeDuckImage,
+  decodeFromFile,
+  decodeFromBuffer,
+};

package/duck_encode.js

@@ -0,0 +1,219 @@
+/**
+ * duck_encode.js
+ *
+ * Node.js 编码器：把图片 / 文本数据隐藏进鸭子图
+ * 对应 Python duck_encode_node.py 的核心逻辑（去除 ComfyUI 节点壳）
+ *
+ * 支持的输入类型：
+ *   - 单张图片（PNG/JPEG Buffer 或文件路径）
+ *   - 多张图片（Buffer 数组 —— 输出图片序列，每帧独立生成一张鸭子图）
+ *   - 纯文本字符串
+ *
+ * 注意：视频合成（对应 Python _images_to_video）依赖 ffmpeg，本文件不包含，
+ *       如需合成视频请在外部使用 fluent-ffmpeg 将帧序列合成 MP4 后，
+ *       把 MP4 Buffer 通过 encodeBytes({ rawBytes, ext: 'mp4.binpng', ... }) 编码。
+ */
+
+'use strict';
+
+const fs   = require('fs');
+const path = require('path');
+const sharp = require('sharp');
+
+const {
+  buildFileHeader,
+  exportDuckPayload,
+  bytesToBinaryImage,
+  requiredCanvasSize,
+  embedPayloadLSB,
+  buildDuckImageBuffer,
+} = require('./duck_payload_exporter');
+
+// ─── 工具：图片 Buffer → PNG 字节 ────────────────────────────────────────────
+
+/**
+ * 读取图片文件 / Buffer，统一转成 PNG Buffer
+ */
+async function toPngBuffer(input) {
+  if (typeof input === 'string') {
+    return sharp(input).toColorspace('srgb').removeAlpha().png().toBuffer();
+  }
+  if (Buffer.isBuffer(input)) {
+    return sharp(input).toColorspace('srgb').removeAlpha().png().toBuffer();
+  }
+  throw new TypeError(`不支持的输入类型：${typeof input}`);
+}
+
+// ─── 主编码函数 ───────────────────────────────────────────────────────────────
+
+/**
+ * 把单张图片编码为鸭子图
+ *
+ * @param {object}         opts
+ * @param {Buffer|string}  opts.image      输入图片（Buffer 或文件路径）
+ * @param {string}         [opts.password] 加密密码
+ * @param {string}         [opts.title]    鸭子图标题
+ * @param {number}         [opts.compress] 压缩模式 2/6/8（默认 2）
+ * @param {string}         [opts.outputPath] 若提供则保存 PNG 文件
+ * @returns {Promise<{ imageBuffer: Buffer, filePath?: string }>}
+ */
+async function encodeImage({ image, password = '', title = '', compress = 2, outputPath = null }) {
+  const pngBytes = await toPngBuffer(image);
+  const result   = await exportDuckPayload({
+    rawBytes: pngBytes,
+    password,
+    ext: 'png',
+    compress,
+    title,
+  });
+
+  if (outputPath) {
+    await fs.promises.writeFile(outputPath, result.imageBuffer);
+    return { imageBuffer: result.imageBuffer, filePath: outputPath };
+  }
+  return { imageBuffer: result.imageBuffer };
+}
+
+/**
+ * 把纯文本编码为鸭子图
+ *
+ * @param {object}  opts
+ * @param {string}  opts.text           待隐藏的文本（UTF-8）
+ * @param {string}  [opts.password]     加密密码
+ * @param {string}  [opts.title]        鸭子图标题
+ * @param {number}  [opts.compress]     2/6/8（默认 2）
+ * @param {string}  [opts.outputPath]   保存路径
+ * @returns {Promise<{ imageBuffer: Buffer, filePath?: string }>}
+ */
+async function encodeText({ text, password = '', title = '', compress = 2, outputPath = null }) {
+  const rawBytes = Buffer.from(text, 'utf-8');
+  const result   = await exportDuckPayload({
+    rawBytes,
+    password,
+    ext: 'txt',
+    compress,
+    title,
+  });
+
+  if (outputPath) {
+    await fs.promises.writeFile(outputPath, result.imageBuffer);
+    return { imageBuffer: result.imageBuffer, filePath: outputPath };
+  }
+  return { imageBuffer: result.imageBuffer };
+}
+
+/**
+ * 把任意二进制数据编码为鸭子图（通用接口）
+ *
+ * @param {object}  opts
+ * @param {Buffer}  opts.rawBytes       原始字节
+ * @param {string}  opts.ext            扩展名，如 "png" / "txt" / "mp4.binpng"
+ * @param {string}  [opts.password]
+ * @param {string}  [opts.title]
+ * @param {number}  [opts.compress]
+ * @param {string}  [opts.outputPath]
+ * @returns {Promise<{ imageBuffer: Buffer, filePath?: string }>}
+ */
+async function encodeBytes({ rawBytes, ext, password = '', title = '', compress = 2, outputPath = null }) {
+  const result = await exportDuckPayload({ rawBytes, password, ext, compress, title });
+
+  if (outputPath) {
+    await fs.promises.writeFile(outputPath, result.imageBuffer);
+    return { imageBuffer: result.imageBuffer, filePath: outputPath };
+  }
+  return { imageBuffer: result.imageBuffer };
+}
+
+/**
+ * 把多张图片编码为「图片序列」（每帧一张独立的鸭子图）
+ * 所有帧使用统一的画布尺寸（对应 Python combine_video=False 分支）
+ *
+ * @param {object}           opts
+ * @param {Array<Buffer|string>} opts.images  图片数组
+ * @param {string}           [opts.password]
+ * @param {string}           [opts.title]
+ * @param {number}           [opts.compress]
+ * @param {string}           [opts.outputDir]  若提供则逐帧保存 PNG
+ * @returns {Promise<{ frames: Buffer[], filePaths?: string[] }>}
+ */
+async function encodeImageSequence({ images, password = '', title = '', compress = 2, outputDir = null }) {
+  if (!images || images.length === 0) throw new Error('images 数组不能为空');
+
+  const lsbBits = compress >= 8 ? 8 : compress >= 6 ? 6 : 2;
+
+  // 第一步：把所有帧转为 PNG 并构建文件头，计算最大所需尺寸
+  const pngList    = await Promise.all(images.map(img => toPngBuffer(img)));
+  const headerList = pngList.map(png => buildFileHeader(png, password, 'png'));
+  const maxBits    = Math.max(...headerList.map(h => (h.length + 4) * 8));
+  const unifiedSize = requiredCanvasSize(maxBits, lsbBits);
+
+  // 第二步：逐帧嵌入（使用统一画布尺寸）
+  const frames    = [];
+  const filePaths = [];
+
+  for (let i = 0; i < pngList.length; i++) {
+    const frameTitle  = `${title} (${i + 1}/${pngList.length})`;
+    const duckPng     = await buildDuckImageBuffer(unifiedSize, frameTitle);
+    const resultFrame = await embedPayloadLSB(duckPng, headerList[i], lsbBits);
+
+    frames.push(resultFrame);
+
+    if (outputDir) {
+      const outPath = path.join(outputDir, `duck_seq_${String(i).padStart(5, '0')}.png`);
+      await fs.promises.mkdir(outputDir, { recursive: true });
+      await fs.promises.writeFile(outPath, resultFrame);
+      filePaths.push(outPath);
+    }
+  }
+
+  return outputDir ? { frames, filePaths } : { frames };
+}
+
+/**
+ * 把已有的 MP4 文件编码为鸭子图
+ * （MP4 字节先转为 binpng 格式再走 LSB 嵌入，与 Python 视频路径对齐）
+ *
+ * @param {object}       opts
+ * @param {Buffer|string} opts.mp4Input  MP4 Buffer 或文件路径
+ * @param {string}        [opts.password]
+ * @param {string}        [opts.title]
+ * @param {number}        [opts.compress]
+ * @param {string}        [opts.outputPath]
+ * @returns {Promise<{ imageBuffer: Buffer, filePath?: string }>}
+ */
+async function encodeMp4({ mp4Input, password = '', title = '', compress = 2, outputPath = null }) {
+  let mp4Bytes;
+  if (typeof mp4Input === 'string') {
+    mp4Bytes = await fs.promises.readFile(mp4Input);
+  } else if (Buffer.isBuffer(mp4Input)) {
+    mp4Bytes = mp4Input;
+  } else {
+    throw new TypeError('mp4Input 必须是 Buffer 或文件路径');
+  }
+
+  // MP4 字节 → binpng PNG Buffer（中间存储格式）
+  const binPng      = await bytesToBinaryImage(mp4Bytes, 512);
+  const binPngBytes = binPng;
+
+  const result = await exportDuckPayload({
+    rawBytes: binPngBytes,
+    password,
+    ext: 'mp4.binpng',
+    compress,
+    title,
+  });
+
+  if (outputPath) {
+    await fs.promises.writeFile(outputPath, result.imageBuffer);
+    return { imageBuffer: result.imageBuffer, filePath: outputPath };
+  }
+  return { imageBuffer: result.imageBuffer };
+}
+
+module.exports = {
+  encodeImage,
+  encodeText,
+  encodeBytes,
+  encodeImageSequence,
+  encodeMp4,
+};