【ffmpeg】解决fluentffmpeg使用ffprobe无效问题

前言

　　在使用fluent-ffmpeg时，ffprobe方法无论添加什么选项都只返回视频的元信息。

 　　如下图：下图是获取视频信息的函数

　　

 　　如下图：下图为调用并打印出视频信息

　　

 　　

 　　

　　使用ffprobe不管添加什么参数，第一张图添加了 ['-v', 'quiet', '-select_streams', 'v', '-show_entries', 'frame=pkt_pts_time,pict_type'] （获取IBP帧时间点）, 但是获取到的结果还是一些视频的元信息。

查看源代码

　　在 fluent-ffmpeg/lib/ffprobe.js 中：

var ffprobe = spawn(path, ['-show_streams', '-show_format'].concat(options, src));

　　这段代码可以看出，是在使用ffprobe执行命令，并且默认添加了 '-show_streams' 和 '-show_format' 选项，而options则是调用者传进来的，两者进行合并。这里既然合并并执行了，为何没效呢，说明问题不在这，继续看。

　　

　　　　ffprobe.stdout.on('data', function(data) {
        console.log(data.toString())
        stdout += data;
      });

      ffprobe.stdout.on('close', function() {
        stdoutClosed = true;
        handleExit();
      });

      ffprobe.stderr.on('data', function(data) {
        stderr += data;
      });

      ffprobe.stderr.on('close', function() {
        stderrClosed = true;
        handleExit();
      });

　　这些代码则是使用 spawn 执行的事件监听，可以直接看看执行的结果

　　

　　可以看到，确实是有结果的。说明在处理这些数据的时候没有处理这些数据，只处理了他原本默认选项的数据。

　　再看 spawn 的 close事件，调用了 handleExit 

function handleExit(err) {
        if (err) {
          exitError = err;
        }

        if (processExited && stdoutClosed && stderrClosed) {
          if (exitError) {
            if (stderr) {
              exitError.message += '\n' + stderr;
            }

            return handleCallback(exitError);
          }

          // Process output
          var data = parseFfprobeOutput(stdout);
          
          // Handle legacy output with "TAG:x" and "DISPOSITION:x" keys
          [data.format].concat(data.streams).forEach(function(target) {
            if (target) {
              var legacyTagKeys = Object.keys(target).filter(legacyTag);

              if (legacyTagKeys.length) {
                target.tags = target.tags || {};

                legacyTagKeys.forEach(function(tagKey) {
                  target.tags[tagKey.substr(4)] = target[tagKey];
                  delete target[tagKey];
                });
              }

              var legacyDispositionKeys = Object.keys(target).filter(legacyDisposition);

              if (legacyDispositionKeys.length) {
                target.disposition = target.disposition || {};

                legacyDispositionKeys.forEach(function(dispositionKey) {
                  target.disposition[dispositionKey.substr(12)] = target[dispositionKey];
                  delete target[dispositionKey];
                });
              }
            }
          });

          handleCallback(null, data);
        }
      }

　　具体关键代码为，上面代码片段的 16 行，parseFfprobeOutput， 解析ffprobe的输出

function parseFfprobeOutput(out) {
  var lines = out.split(/\r\n|\r|\n/);

  lines = lines.filter(function (line) {
    return line.length > 0;
  });

  var data = {
    streams: [],
    format: {},
    chapters: []
13   };

  function parseBlock(name) {
    var data = {};

    var line = lines.shift();
    while (typeof line !== 'undefined') {
      if (line.toLowerCase() == '[/'+name+']') {
        return data;
      } else if (line.match(/^\[/)) {
        line = lines.shift();
        continue;
      }

      var kv = line.match(/^([^=]+)=(.*)$/);
      if (kv) {
        if (!(kv[1].match(/^TAG:/)) && kv[2].match(/^[0-9]+(\.[0-9]+)?$/)) {
          data[kv[1]] = Number(kv[2]);
        } else {
          data[kv[1]] = kv[2];
        }
      }

      line = lines.shift();
    }

    return data;
  }

  var line = lines.shift();
  while (typeof line !== 'undefined') {
    if (line.match(/^\[stream/i)) {
      var stream = parseBlock('stream');
      data.streams.push(stream);
    } else if (line.match(/^\[chapter/i)) {
      var chapter = parseBlock('chapter');
      data.chapters.push(chapter);
    51      52 53     } else if (line.toLowerCase() === '[format]') {
      data.format = parseBlock('format');
    }

    line = lines.shift();
  }

  return data;
}

　　这里代码就是具体解析ffprobe执行命令后输出的字符串的函数，不管有什么结果，都只解析了 stream、chapter、format 三个字段的值。

修改代码

• 手动修改

　　只需将 while 循环里的代码修改即可。

　　修改前：

while (typeof line !== 'undefined') {
     if (line.match(/^\[stream/i)) {
       var stream = parseBlock('stream');
       data.streams.push(stream);
     } else if (line.match(/^\[chapter/i)) {
       var chapter = parseBlock('chapter');
       data.chapters.push(chapter);
     
      
      
     } else if (line.toLowerCase() === '[format]') {
       data.format = parseBlock('format');
     }
 
     line = lines.shift();
   }

　　修改后：

while (typeof line !== 'undefined') {

    if (line.match(/^\[stream/i)) {
      var stream = parseBlock('stream');
      data.streams.push(stream);
    } else if (line.match(/^\[chapter/i)) {
      var chapter = parseBlock('chapter');
      data.chapters.push(chapter);
    } else if (line.toLowerCase() === '[format]') {
      data.format = parseBlock('format');
    } else if (line.match(/^\[[^\/].*?/i)) {

      let name = line.slice(1,-1).toLowerCase()
      if(!data[name] || !(data[name] instanceof Array)) data[name] = []
      var res = parseBlock(name)
      data[name].push(res)
    }

    line = lines.shift();
  }

　　上面是手动的修改 fluent-ffmpeg内的源代码，每次安装 fluent-ffmpeg 都要重新修改非常麻烦。

•  自动修改

　　原理是读取 fluent-ffmpeg/lib/ffprobe.js 文件的代码字符串，将代码字符串转换为 AST，再修改 AST，最后将 AST 转换为代码，再将代码写到 ffprobe.js 文件中。

require('fluent-ffmpeg/lib/ffprobe.js')  // 导入 ffprobe
const esprima = require('esprima')
const escodegen = require('escodegen')
const estraverse = require('estraverse')
const fs = require('fs')

const sourcePath = module.children[0].id  // 用 module 获取到 ffprobe 的路径（得先导入ffprobe）


// 修改后的代码的字符串
const newParseCode = `
    while (typeof line !== 'undefined') {
 
     if (line.match(/^\\[stream/i)) {
      var stream = parseBlock('stream');
       data.streams.push(stream);
    } else if (line.match(/^\\[chapter/i)) {
      var chapter = parseBlock('chapter');
      data.chapters.push(chapter);
     } else if (line.toLowerCase() === '[format]') {
       data.format = parseBlock('format');
    } else if (line.match(/^\\[[^\\/].*?/i)) {

       let name = line.slice(1,-1).toLowerCase()
       if(!data[name] || !(data[name] instanceof Array)) data[name] = []
      var res = parseBlock(name)
       data[name].push(res)
    }

    line = lines.shift();
   }
`

// 读取 ffprobe 的源代码为字符串
const oldParseCode = fs.readFileSync(sourcePath).toString()

// 将修改后的代码字符串转换为 AST
const newParseAST = esprima.parseScript(newParseCode)

// 将 ffprobe 源代码字符串转换为 AST
var oldParseAST = esprima.parseScript(oldParseCode)

// 用 estraverse 找到 parseFfprobeOutput 函数的位置
estraverse.traverse(oldParseAST, {
    enter: (node) => {

        if (node.type == 'FunctionDeclaration' && node.id.name == 'parseFfprobeOutput') {

            // 再找到 while 循环的位置
            estraverse.replace(node, {
                enter: (node, parent) => {
                    if (node.type == 'WhileStatement' && parent.body.length > 4) {

                        // 直接将 while 循环位置的 AST 进行替换为修改后代码字符串的 AST
                        return newParseAST
                    }
                }

            })


            return
        }
    }
})

// 用 escodegen 将 AST 转换为代码
const code = escodegen.generate(oldParseAST)

70 // 将代码字符串写到文件中

fs.writeFileSync(sourcePath, code)

　　这样只需引入写好的这段代码即可。