mb-rrvideo-server 1.0.2

package/src/utils/video.js

@@ -0,0 +1,234 @@
+/**
+ * 视频处理工具模块
+ * 负责执行 rrvideos 命令进行视频转换和合并
+ */
+
+const { spawn } = require('child_process');
+const { execSync } = require('child_process');
+const path = require('path');
+
+/**
+ * 转义命令行参数
+ */
+function escapeArg(str) {
+  if (str == null) return '';
+  return `"${String(str).replace(/"/g, '\\"')}"`;
+}
+
+/**
+ * 执行 rrvideos 命令
+ * @param {string} command - 完整的命令字符串
+ * @param {function} onLog - 日志回调函数 (type, data)
+ * @param {object} customEnv - 自定义环境变量
+ * @returns {ChildProcess} 子进程对象
+ */
+function runRrvideosCommand(command, onLog, customEnv) {
+  if (process.env.TEST_MODE === '1') {
+    // 测试模式：模拟进度与完成
+    setTimeout(() => onLog && onLog('stdout', '[TEST] progress: 10%'), 50);
+    setTimeout(() => onLog && onLog('stdout', '[TEST] progress: 50%'), 100);
+    setTimeout(() => onLog && onLog('stdout', '[TEST] progress: 90%'), 150);
+    setTimeout(() => onLog && onLog('close', { code: 0, signal: null }), 220);
+    return { kill: () => {}, pid: 12345 };
+  }
+
+  const spawnOptions = {
+    shell: true,
+    stdio: ['ignore', 'pipe', 'pipe'],
+  };
+  
+  // 使用自定义环境变量或继承当前进程环境
+  if (customEnv) {
+    spawnOptions.env = customEnv;
+  } else {
+    spawnOptions.env = Object.assign({}, process.env);
+  }
+  
+  const child = spawn(command, spawnOptions);
+
+  // 流式捕获 stdout/stderr
+  if (child.stdout) {
+    child.stdout.on('data', (data) => {
+      const output = String(data).trim();
+      if (output) {
+        onLog && onLog('stdout', output);
+      }
+    });
+  }
+  
+  if (child.stderr) {
+    child.stderr.on('data', (data) => {
+      const output = String(data).trim();
+      if (output) {
+        onLog && onLog('stderr', output);
+      }
+    });
+  }
+
+  // close 事件带上 signal
+  child.on('close', (code, signal) => {
+    onLog && onLog('close', { code, signal });
+  });
+  
+  return child;
+}
+
+/**
+ * 获取 rrvideos 命令路径
+ * @returns {string} rrvideos 可执行文件路径
+ */
+function getRrvideosPath() {
+  try {
+    const rrvideosPath = execSync('which rrvideos', { encoding: 'utf-8' }).trim();
+    return rrvideosPath;
+  } catch (e) {
+    throw new Error('rrvideos 命令未找到，请确保已安装 mb-rrvideo');
+  }
+}
+
+/**
+ * 获取 node 命令路径（与 rrvideos 同目录）
+ * @returns {string} node 可执行文件路径
+ */
+function getNodePath() {
+  const rrvideosPath = getRrvideosPath();
+  const rrvideosDir = path.dirname(rrvideosPath);
+  return path.join(rrvideosDir, 'node');
+}
+
+/**
+ * 生成临时 rrvideo 配置文件
+ * @param {object} rrvideoConfig - rrvideo 配置对象（从 config.rrvideo 读取）
+ * @returns {string} 临时配置文件路径
+ */
+function generateTempConfigFile(rrvideoConfig) {
+  const fs = require('fs');
+  const os = require('os');
+  const path = require('path');
+  
+  // 创建临时目录
+  const tempDir = path.join(os.tmpdir(), 'rrvideo-server');
+  if (!fs.existsSync(tempDir)) {
+    fs.mkdirSync(tempDir, { recursive: true });
+  }
+  
+  // 生成唯一的临时文件名
+  const tempFile = path.join(tempDir, `rrvideo-config-${Date.now()}-${Math.random().toString(36).slice(2)}.json`);
+  
+  // 写入配置（rrvideos 期望的是扁平的 JSON，不需要 rrvideo 包裹层）
+  fs.writeFileSync(tempFile, JSON.stringify(rrvideoConfig, null, 2));
+  
+  return tempFile;
+}
+
+/**
+ * 删除临时配置文件
+ * @param {string} configPath - 临时配置文件路径
+ */
+function cleanupTempConfigFile(configPath) {
+  try {
+    const fs = require('fs');
+    if (configPath && fs.existsSync(configPath)) {
+      fs.unlinkSync(configPath);
+    }
+  } catch (e) {
+    console.error(`清理临时配置文件失败: ${e.message}`);
+  }
+}
+
+/**
+ * 构建转换命令
+ * @param {Array} inputList - 输入文件路径列表
+ * @param {Array} outputList - 输出文件路径列表
+ * @param {object|string} config - 配置对象或配置文件路径
+ * @returns {object} { command: string, tempConfigFile: string|null }
+ */
+function buildConvertCommand(inputList, outputList, config) {
+  const nodePath = getNodePath();
+  const rrvideosPath = getRrvideosPath();
+  
+  let command = `${nodePath} --expose-gc ${rrvideosPath}`;
+  inputList.forEach((p) => { command += ` --input ${escapeArg(p)}`; });
+  outputList.forEach((p) => { command += ` --output ${escapeArg(p)}`; });
+  command += ` --format mp4`;
+  
+  let tempConfigFile = null;
+  
+  if (config) {
+    if (typeof config === 'string') {
+      // 如果是字符串，当作文件路径直接使用
+      command += ` --config ${escapeArg(config)}`;
+    } else if (typeof config === 'object') {
+      // 如果是对象，生成临时配置文件
+      tempConfigFile = generateTempConfigFile(config);
+      command += ` --config ${escapeArg(tempConfigFile)}`;
+    }
+  }
+  
+  return { command, tempConfigFile };
+}
+
+/**
+ * 构建合并命令
+ * @param {Array} videos - 视频文件路径列表
+ * @param {string} output - 输出文件路径
+ * @param {object} options - 合并选项
+ * @returns {string} 完整的命令字符串
+ */
+function buildMergeCommand(videos, output, options = {}) {
+  const nodePath = getNodePath();
+  const rrvideosPath = getRrvideosPath();
+  
+  const format = options.format || 'mp4';
+  const method = options.method || 'concat';
+  const autoMode = options.auto_mode || 'smart';
+  const headers = options.headers || {};
+  
+  let command = `${nodePath} --expose-gc ${rrvideosPath} merge --output ${escapeArg(output)} --format ${escapeArg(format)} --method ${escapeArg(method)} --auto-mode ${escapeArg(autoMode)}`;
+  
+  videos.forEach((v) => {
+    command += ` --videos ${escapeArg(v)}`;
+  });
+  
+  Object.keys(headers).forEach((k) => {
+    const pair = `${k}:${headers[k]}`;
+    command += ` --headers ${escapeArg(pair)}`;
+  });
+  
+  return command;
+}
+
+/**
+ * 设置文件权限
+ * @param {string} directoryPath - 目录路径
+ * @returns {Promise<void>}
+ */
+function setPermissions(directoryPath) {
+  return new Promise((resolve, reject) => {
+    const { exec } = require('child_process');
+    const command = `chmod -R o+r ${directoryPath}/*`;
+    exec(command, (error, stdout, stderr) => {
+      if (error) {
+        console.error(`chmod执行错误: ${error}`);
+        reject(error);
+        return;
+      }
+      if (stderr) {
+        console.error(`chmod标准错误输出: ${stderr}`);
+      }
+      console.log(`chmod标准输出: ${stdout}`);
+      resolve(stdout);
+    });
+  });
+}
+
+module.exports = {
+  runRrvideosCommand,
+  getRrvideosPath,
+  getNodePath,
+  buildConvertCommand,
+  buildMergeCommand,
+  setPermissions,
+  escapeArg,
+  cleanupTempConfigFile
+};