ss-tools-duck 1.0.0

package/duck_video.js

@@ -0,0 +1,201 @@
+/**
+ * duck_video.js
+ *
+ * 视频合成 + 解帧工具（对应 Python _images_to_video 功能）
+ * 依赖：fluent-ffmpeg（需要系统安装 ffmpeg）
+ *
+ * 提供：
+ *   imagesToMp4({ frames, fps, audioPath, outputPath }) → Promise<Buffer>
+ *   mp4ToFrames({ mp4Input, outputDir }) → Promise<Buffer[]>
+ *   encodeVideoFrames({ frames, fps, audioPath, password, title, compress, outputPath })
+ *     帧序列 → 鸭子图（一步封装）
+ */
+
+'use strict';
+
+const fs         = require('fs');
+const path       = require('path');
+const os         = require('os');
+const crypto     = require('crypto');
+const ffmpeg     = require('fluent-ffmpeg');
+const sharp      = require('sharp');
+const { encodeMp4 } = require('./duck_encode');
+
+// ─── 工具：创建临时目录 ───────────────────────────────────────────────────────
+
+function makeTempDir() {
+  const dir = path.join(os.tmpdir(), 'duck_ffmpeg_' + crypto.randomBytes(6).toString('hex'));
+  fs.mkdirSync(dir, { recursive: true });
+  return dir;
+}
+
+function removeTempDir(dir) {
+  try {
+    fs.rmSync(dir, { recursive: true, force: true });
+  } catch { /* ignore */ }
+}
+
+// ─── 判断 ffmpeg 是否可用 ─────────────────────────────────────────────────────
+
+async function checkFfmpegAvailable() {
+  return new Promise((resolve) => {
+    ffmpeg.getAvailableFormats((err) => resolve(!err));
+  });
+}
+
+// ─── 图片帧 → MP4 Buffer ──────────────────────────────────────────────────────
+
+/**
+ * 将图片帧数组合成 MP4（H.264, yuv420p，与 Python MoviePy 输出兼容）
+ *
+ * @param {object}          opts
+ * @param {Buffer[]}        opts.frames       PNG / JPEG Buffer 数组
+ * @param {number}          opts.fps          帧率
+ * @param {string}          [opts.audioPath]  音频文件路径（WAV/MP3/AAC）
+ * @param {string}          [opts.outputPath] 若提供则同时写文件
+ * @returns {Promise<Buffer>}  MP4 字节
+ */
+async function imagesToMp4({ frames, fps, audioPath = null, outputPath = null }) {
+  if (!frames || frames.length === 0) throw new Error('frames 数组不能为空');
+
+  const tmpDir    = makeTempDir();
+  const frameDir  = path.join(tmpDir, 'frames');
+  fs.mkdirSync(frameDir);
+
+  try {
+    // ── 写出帧图片 ──────────────────────────────────────────────────────────
+    for (let i = 0; i < frames.length; i++) {
+      const framePath = path.join(frameDir, `frame_${String(i).padStart(6, '0')}.png`);
+      // 确保是 PNG 格式（sharp 自动转换）
+      const pngBuf = await sharp(frames[i]).png().toBuffer();
+      fs.writeFileSync(framePath, pngBuf);
+    }
+
+    // ── 合成 MP4 ────────────────────────────────────────────────────────────
+    const tmpMp4 = path.join(tmpDir, 'output.mp4');
+
+    await new Promise((resolve, reject) => {
+      let cmd = ffmpeg()
+        .input(path.join(frameDir, 'frame_%06d.png'))
+        .inputFPS(fps)
+        .videoCodec('libx264')
+        .outputOptions([
+          '-pix_fmt yuv420p',
+          '-crf 16',
+          '-preset medium',
+          '-profile:v high',
+          '-movflags +faststart',
+        ]);
+
+      if (audioPath && fs.existsSync(audioPath)) {
+        cmd = cmd
+          .input(audioPath)
+          .audioCodec('aac')
+          .outputOptions(['-shortest']); // 音频/视频取最短
+      }
+
+      cmd
+        .output(tmpMp4)
+        .on('end', resolve)
+        .on('error', reject)
+        .run();
+    });
+
+    const mp4Buffer = fs.readFileSync(tmpMp4);
+
+    if (outputPath) {
+      fs.mkdirSync(path.dirname(outputPath), { recursive: true });
+      fs.writeFileSync(outputPath, mp4Buffer);
+    }
+
+    return mp4Buffer;
+
+  } finally {
+    removeTempDir(tmpDir);
+  }
+}
+
+// ─── MP4 → 帧数组 ─────────────────────────────────────────────────────────────
+
+/**
+ * 将 MP4 解帧为 PNG Buffer 数组
+ *
+ * @param {object}        opts
+ * @param {Buffer|string} opts.mp4Input   MP4 Buffer 或文件路径
+ * @param {number}        [opts.fps]      抽帧帧率（默认：原始帧率全提取）
+ * @returns {Promise<{ frames: Buffer[], fps: number }>}
+ */
+async function mp4ToFrames({ mp4Input, fps = null }) {
+  const tmpDir = makeTempDir();
+  let   tmpMp4 = null;
+
+  try {
+    // 若是 Buffer，先写临时文件
+    if (Buffer.isBuffer(mp4Input)) {
+      tmpMp4 = path.join(tmpDir, 'input.mp4');
+      fs.writeFileSync(tmpMp4, mp4Input);
+    } else {
+      tmpMp4 = mp4Input;
+    }
+
+    const frameDir = path.join(tmpDir, 'frames');
+    fs.mkdirSync(frameDir);
+
+    // 读取原始帧率
+    const meta = await new Promise((resolve, reject) => {
+      ffmpeg.ffprobe(tmpMp4, (err, data) => err ? reject(err) : resolve(data));
+    });
+    const videoStream = meta.streams.find(s => s.codec_type === 'video');
+    const srcFps      = videoStream
+      ? eval(videoStream.r_frame_rate) // "30/1" → 30
+      : 30;
+    const outFps      = fps || srcFps;
+
+    // 提取帧
+    await new Promise((resolve, reject) => {
+      ffmpeg(tmpMp4)
+        .outputOptions([`-vf fps=${outFps}`])
+        .output(path.join(frameDir, 'frame_%06d.png'))
+        .on('end', resolve)
+        .on('error', reject)
+        .run();
+    });
+
+    const frameFiles = fs.readdirSync(frameDir)
+      .filter(f => f.endsWith('.png'))
+      .sort();
+    const frames = frameFiles.map(f => fs.readFileSync(path.join(frameDir, f)));
+
+    return { frames, fps: outFps };
+
+  } finally {
+    removeTempDir(tmpDir);
+  }
+}
+
+// ─── 一步封装：帧序列 → 鸭子图 ──────────────────────────────────────────────
+
+/**
+ * 视频帧序列 → 先合成 MP4 → 再编码为鸭子图
+ *
+ * @param {object}         opts
+ * @param {Buffer[]}       opts.frames      图片帧 Buffer 数组
+ * @param {number}         opts.fps         帧率
+ * @param {string}         [opts.audioPath] 音频文件路径
+ * @param {string}         [opts.password]
+ * @param {string}         [opts.title]
+ * @param {number}         [opts.compress]
+ * @param {string}         [opts.outputPath]
+ * @returns {Promise<{ imageBuffer: Buffer, filePath?: string }>}
+ */
+async function encodeVideoFrames({ frames, fps, audioPath = null, password = '', title = '', compress = 2, outputPath = null }) {
+  const mp4Buffer = await imagesToMp4({ frames, fps, audioPath });
+  return encodeMp4({ mp4Input: mp4Buffer, password, title, compress, outputPath });
+}
+
+module.exports = {
+  checkFfmpegAvailable,
+  imagesToMp4,
+  mp4ToFrames,
+  encodeVideoFrames,
+};

package/index.d.ts

@@ -0,0 +1,196 @@
+/**
+ * SS_tools Duck 隐写工具库 — TypeScript 类型声明
+ */
+
+// ─── 公共选项类型 ─────────────────────────────────────────────────────────────
+
+/** 压缩比（影响每像素隐藏的 bit 数）：2 = 最低信息量，8 = 最高 */
+export type CompressLevel = 2 | 6 | 8;
+
+/** 编码共有选项 */
+export interface EncodeOptions {
+  /** 加密密码；留空或省略则不加密 */
+  password?: string;
+  /** 显示在鸭子图右下角的标题 */
+  title?: string;
+  /** 压缩比，默认 2 */
+  compress?: CompressLevel;
+  /** 输出文件路径；省略则返回 Buffer 不写文件 */
+  outputPath?: string;
+}
+
+/** 编码结果 */
+export interface EncodeResult {
+  /** 生成的鸭子图 PNG 字节 */
+  imageBuffer: Buffer;
+  /** 写出的文件路径（仅当传入 outputPath 时存在） */
+  filePath?: string;
+}
+
+// ─── 解码相关 ─────────────────────────────────────────────────────────────────
+
+/** 解码结果 */
+export interface DecodeResult {
+  /** 还原的原始字节（文本为 UTF-8 编码的 bytes） */
+  data: Buffer;
+  /** 文件扩展名，如 "txt" / "png" / "mp4" / "bin" */
+  ext: string;
+  /** 若 ext === "txt"，已解码为字符串 */
+  text?: string;
+  /** 若 ext === "png"，PNG 图片 Buffer */
+  imageBuffer?: Buffer;
+  /** 若 ext === "mp4" 或 "mp4.binpng"，MP4 视频 Buffer */
+  mp4Buffer?: Buffer;
+  /** 若传入 outputDir，还原文件的输出路径 */
+  filePath?: string;
+}
+
+// ─── 编码 API ─────────────────────────────────────────────────────────────────
+
+/** 单张图片 → 鸭子图 */
+export function encodeImage(opts: EncodeOptions & { image: Buffer }): Promise<EncodeResult>;
+
+/** 文本字符串 → 鸭子图 */
+export function encodeText(opts: EncodeOptions & { text: string }): Promise<EncodeResult>;
+
+/**
+ * 任意二进制数据 → 鸭子图（通用接口）
+ * @param opts.rawBytes  原始字节
+ * @param opts.ext       扩展名（如 "pdf"、"zip"）
+ */
+export function encodeBytes(opts: EncodeOptions & { rawBytes: Buffer; ext: string }): Promise<EncodeResult>;
+
+/**
+ * 多帧图片序列 → 独立鸭子图序列（每帧一个文件）
+ * @param opts.outputDir  输出目录；文件命名为 duck_000.png, duck_001.png …
+ */
+export function encodeImageSequence(opts: EncodeOptions & {
+  images: Buffer[];
+  outputDir?: string;
+}): Promise<{ imageBuffers: Buffer[]; filePaths?: string[] }>;
+
+/**
+ * MP4 视频（Buffer 或文件路径）→ 鸭子图（binpng 格式）
+ * @param opts.mp4Input  MP4 字节 Buffer 或文件路径字符串
+ */
+export function encodeMp4(opts: EncodeOptions & { mp4Input: Buffer | string }): Promise<EncodeResult>;
+
+// ─── 解码 API ─────────────────────────────────────────────────────────────────
+
+/** 从鸭子图 Buffer 或文件路径中还原隐藏数据 */
+export function decodeDuckImage(opts: {
+  duckImage: Buffer | string;
+  /** 解密密码（无密码可省略） */
+  password?: string;
+  /** 还原文件的输出目录 */
+  outputDir?: string;
+}): Promise<DecodeResult>;
+
+/** 便捷接口：从文件路径解码 */
+export function decodeFromFile(
+  duckFilePath: string,
+  password?: string,
+  outputDir?: string
+): Promise<DecodeResult>;
+
+/** 便捷接口：从 Buffer 解码 */
+export function decodeFromBuffer(
+  duckBuffer: Buffer,
+  password?: string,
+  outputDir?: string
+): Promise<DecodeResult>;
+
+// ─── 视频合成 API ─────────────────────────────────────────────────────────────
+
+/** 检查系统 ffmpeg 是否可用 */
+export function checkFfmpegAvailable(): Promise<boolean>;
+
+/**
+ * 图片帧数组 → MP4 Buffer
+ * @param opts.frames      PNG/JPEG Buffer 数组（按顺序）
+ * @param opts.fps         帧率
+ * @param opts.audioPath   混音文件路径（WAV/MP3/AAC，可选）
+ * @param opts.outputPath  若提供则同时写出文件
+ */
+export function imagesToMp4(opts: {
+  frames: Buffer[];
+  fps: number;
+  audioPath?: string | null;
+  outputPath?: string | null;
+}): Promise<Buffer>;
+
+/**
+ * MP4 解帧为 PNG Buffer 数组
+ * @param opts.mp4Input  MP4 Buffer 或文件路径
+ * @param opts.fps       抽帧帧率（省略则按原始帧率全提取）
+ */
+export function mp4ToFrames(opts: {
+  mp4Input: Buffer | string;
+  fps?: number | null;
+}): Promise<{ frames: Buffer[]; fps: number }>;
+
+/**
+ * 帧序列 → 合成 MP4 → 编码为鸭子图（一步封装）
+ */
+export function encodeVideoFrames(opts: EncodeOptions & {
+  frames: Buffer[];
+  fps: number;
+  audioPath?: string | null;
+}): Promise<EncodeResult>;
+
+// ─── 底层工具 API ─────────────────────────────────────────────────────────────
+
+/** 计算指定 payload 长度所需的画布像素尺寸 */
+export function requiredCanvasSize(payloadBits: number, compress?: CompressLevel): number;
+
+/**
+ * 将字节数组转为二进制图 PNG（3 bytes = 1 pixel，用于存储 MP4 原始字节）
+ * @param bytes   原始字节
+ * @param size    目标图像尺寸（正方形，默认 512）
+ */
+export function bytesToBinaryImage(bytes: Buffer, size?: number): Promise<Buffer>;
+
+/** 从二进制图 PNG 中还原字节（去除末尾零填充） */
+export function binaryImageToBytes(pngBuffer: Buffer): Promise<Buffer>;
+
+/** 构造文件头二进制数据 */
+export function buildFileHeader(opts: {
+  ext: string;
+  dataLength: number;
+  password?: string;
+}): Buffer;
+
+/** 解析文件头 */
+export function parseHeader(buf: Buffer, password?: string): {
+  ext: string;
+  dataStart: number;
+  dataLength: number;
+};
+
+/** 将 payload 隐写到 PNG 图像中，返回新的 PNG Buffer */
+export function embedPayloadLSB(
+  pngBuffer: Buffer,
+  payloadBits: number[],
+  compress?: CompressLevel
+): Promise<Buffer>;
+
+/** 从 PNG 图像中提取隐写数据（尝试指定的 k 值） */
+export function extractPayloadWithK(
+  pngBuffer: Buffer,
+  k: CompressLevel
+): Promise<{ bits: number[]; bytesFromBits: (bits: number[]) => Buffer }>;
+
+/** 完整流程：构造 header → embed → 返回鸭子图 Buffer */
+export function exportDuckPayload(opts: {
+  rawBytes: Buffer;
+  ext: string;
+  password?: string;
+  title?: string;
+  compress?: CompressLevel;
+}): Promise<Buffer>;
+
+/** 用 SVG 生成鸭子图底图（返回 PNG Buffer） */
+export function buildDuckImageBuffer(size: number, title?: string): Promise<Buffer>;
+
+/** 生成鸭子 SVG 字符串 */
+export function buildDuckImageSVG(size: number, title?: string): string;

package/index.js

@@ -0,0 +1,88 @@
+/**
+ * index.js — SS_tools Duck 隐写工具库（Node.js 版）
+ *
+ * 公开 API：
+ *
+ *   编码（隐藏数据）
+ *   ─────────────────────────────────────────────
+ *   encodeImage({ image, password, title, compress, outputPath })
+ *     单张图片 → 鸭子图
+ *
+ *   encodeText({ text, password, title, compress, outputPath })
+ *     文本字符串 → 鸭子图
+ *
+ *   encodeBytes({ rawBytes, ext, password, title, compress, outputPath })
+ *     任意二进制数据 → 鸭子图（通用接口）
+ *
+ *   encodeImageSequence({ images, password, title, compress, outputDir })
+ *     多帧图片 → 独立鸭子图序列
+ *
+ *   encodeMp4({ mp4Input, password, title, compress, outputPath })
+ *     MP4 文件 → 鸭子图
+ *
+ *   解码（还原数据）
+ *   ─────────────────────────────────────────────
+ *   decodeDuckImage({ duckImage, password, outputDir })
+ *     鸭子图（Buffer 或路径） → { data, ext, text?, imageBuffer?, mp4Buffer?, filePath? }
+ *
+ *   decodeFromFile(duckFilePath, password?, outputDir?)
+ *     便捷：文件路径解码
+ *
+ *   decodeFromBuffer(duckBuffer, password?, outputDir?)
+ *     便捷：Buffer 解码
+ *
+ *   视频合成（来自 duck_video，需系统安装 ffmpeg）
+ *   ─────────────────────────────────────────────
+ *   imagesToMp4({ frames, fps, audioPath, outputPath })
+ *     图片帧 Buffer 数组 → MP4 Buffer
+ *
+ *   mp4ToFrames({ mp4Input, fps })
+ *     MP4 → 帧 Buffer 数组
+ *
+ *   encodeVideoFrames({ frames, fps, audioPath, password, title, compress, outputPath })
+ *     帧序列 → 鸭子图（一步封装）
+ *
+ *   checkFfmpegAvailable()
+ *     检查系统 ffmpeg 是否可用
+ *
+ *   底层工具（来自 duck_payload_exporter）
+ *   ─────────────────────────────────────────────
+ *   buildFileHeader / parseHeader
+ *   requiredCanvasSize
+ *   embedPayloadLSB / extractPayloadWithK
+ *   bytesToBinaryImage / binaryImageToBytes
+ *   exportDuckPayload
+ */
+
+'use strict';
+
+const encode   = require('./duck_encode');
+const decode   = require('./duck_decode');
+const exporter = require('./duck_payload_exporter');
+const video    = require('./duck_video');
+
+module.exports = {
+  // ── 编码 ──────────────────────────────────────────────────────────────────
+  ...encode,
+
+  // ── 解码 ──────────────────────────────────────────────────────────────────
+  ...decode,
+
+  // ── 视频合成 ───────────────────────────────────────────────────────────────
+  checkFfmpegAvailable: video.checkFfmpegAvailable,
+  imagesToMp4:          video.imagesToMp4,
+  mp4ToFrames:          video.mp4ToFrames,
+  encodeVideoFrames:    video.encodeVideoFrames,
+
+  // ── 底层工具（按需使用） ───────────────────────────────────────────────────
+  buildFileHeader:      exporter.buildFileHeader,
+  parseHeader:          exporter.parseHeader,
+  requiredCanvasSize:   exporter.requiredCanvasSize,
+  embedPayloadLSB:      exporter.embedPayloadLSB,
+  extractPayloadWithK:  exporter.extractPayloadWithK,
+  bytesToBinaryImage:   exporter.bytesToBinaryImage,
+  binaryImageToBytes:   exporter.binaryImageToBytes,
+  exportDuckPayload:    exporter.exportDuckPayload,
+  buildDuckImageBuffer: exporter.buildDuckImageBuffer,
+  buildDuckImageSVG:    exporter.buildDuckImageSVG,
+};

package/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "ss-tools-duck",
+  "version": "1.0.0",
+  "description": "Node.js implementation of SS_tools duck steganography encoder/decoder",
+  "main": "index.js",
+  "types": "index.d.ts",
+  "bin": {
+    "duck-encode": "./duck-cli.js",
+    "duck-decode": "./duck-cli.js",
+    "duck-cli": "./duck-cli.js"
+  },
+  "scripts": {
+    "test": "node test.js",
+    "cli": "node duck-cli.js"
+  },
+  "dependencies": {
+    "fluent-ffmpeg": "^2.1.3",
+    "glob": "^11.0.0",
+    "sharp": "^0.33.0"
+  }
+}
+