【ffmpeg】解決fluent-ffmpeg使用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)

　　這樣只需引入寫好的這段代碼即可。