add ffmpeg arguments

Author: sugiruu

cli/Command.js

@@ -78,7 +78,9 @@ class Command {
       output: this.argv.output || '.',
       format: this.argv.format,
       audioonly: this.argv.audioonly,
-      quality: this.argv.quality
+      quality: this.argv.quality,
+      inputParams: this.argv.inputParam,
+      outputParams: this.argv.outputParam
     })
 
     let downloadBar

cli/index.js

@@ -7,7 +7,7 @@ const Command = require('./Command.js')
 
 const argv = yargs
   .usage('Usage: $0 <command> [options]')
-  .example('$0 -i myaudio.mp3 -t parse.txt -o /home/Music -m title=Test')
+  .example('$0 -i myvideo.avi -f mp4 -t parse.txt -o /home/Videos -m title=Test -p:o "\'-vf scale=320:-1\'"')
   .alias('a', 'audioonly')
   .alias('o', 'output')
   .alias('t', 'template')
@@ -17,6 +17,8 @@ const argv = yargs
   .alias('f', 'format')
   .alias('s', 'sections')
   .alias('q', 'quality')
+  .alias('p:i', 'inputParam')
+  .alias('p:o', 'outputParam')
   .describe('o', 'Output path')
   .describe('t', 'Template text file')
   .describe('i', 'Input file or YouTube URL')
@@ -26,13 +28,17 @@ const argv = yargs
   .describe('a', 'Force download only audio files when using a url as input')
   .describe('s', 'Sections to split')
   .describe('q', 'Download quality (highest/lowest/highestaudio/lowestaudio/highestvideo/lowestvideo)')
+  .describe('p:i', 'Additional FFMpeg input parameters')
+  .describe('p:o', 'Additional FFMpeg output parameters')
   .default('f', 'mp3')
   .default('t', 'templ.txt')
   .default('i', 'input.mp3')
   .default('a', false)
   .default('c', 3)
   .default('q', 'highest')
   .array('m').default('m', [])
+  .array('p:i').default('p:i', [])
+  .array('p:o').default('p:o', [])
   .array('s')
   .help('h').alias('h', 'help')
   .argv

lib/MediaSplit.js

@@ -28,6 +28,8 @@ class MediaSplit extends EventEmitter {
    * @param {string} options.format - Output format (mp3, m4a, flac, etc)
    * @param {boolean} options.audioonly - Force download only audio files when using a url as input
    * @param {string} options.quality - Download quality
+   * @param {string[]} options.inputParams - FFMpeg additional input parameters
+   * @param {string[]} options.outputParams - FFMpeg additional output parameters
    * @return MediaSplit
    */
   constructor (options = {}) {
@@ -43,10 +45,20 @@ class MediaSplit extends EventEmitter {
       output: '.',
       format: 'mp3',
       audioonly: false,
-      quality: 'highest'
+      quality: 'highest',
+      inputParams: [],
+      outputParams: []
     }
 
-    this._options = Object.assign(defaults, options)
+    this._options = Object.assign(
+      defaults,
+      options,
+      {
+        inputParams: this._sanitizeParams(options.inputParams),
+        outputParams: this._sanitizeParams(options.outputParams)
+      }
+    )
+
     this._inputFile = ''
     this._downloadOptions = {}
   }
@@ -327,6 +339,7 @@ class MediaSplit extends EventEmitter {
             '-hide_banner',
             '-loglevel', 'repeat+error',
             '-y',
+            ...this._options.inputParams,
             '-i', this._inputFile,
             '-ss', section.start
           ]
@@ -353,6 +366,9 @@ class MediaSplit extends EventEmitter {
             args.push('-metadata', `${name}=${value}`)
           }
 
+          // Additional ffmpeg output arguments
+          args.push(...this._options.outputParams)
+
           // And finally the output path
           args.push(path.join(this._options.output, section.name))
 
@@ -373,6 +389,15 @@ class MediaSplit extends EventEmitter {
     return Promise.all(promises)
   }
 
+  /**
+   * @param {string[]} params
+   */
+  _sanitizeParams (params) {
+    return (params || [])
+      .map((x) => x.replace(/^("|')/gim, '').replace(/("|')$/gim, ''))
+      .flatMap(x => x.split(' '))
+  }
+
   _checkAccess (dir, mode) {
     return new Promise((resolve, reject) => fs.access(dir, mode, (err) => err ? reject(err) : resolve(dir)))
   }

spec/Command.spec.js

@@ -0,0 +1,13 @@
+'use strict'
+
+const expect = require('chai').expect
+const spawnSync = require('child_process').spawnSync
+const path = require('path')
+
+describe('Command', () => {
+  it('should return 0 code', () => {
+    const binPath = path.resolve('cli/index.js')
+    const proc = spawnSync('node', [ binPath, 'h' ])
+    expect(proc.status).to.be.equal(0)
+  })
+})