ES6-ify

src/cli/utils/commander.js

@@ -1,11 +1,13 @@
-var program = require('commander');
+import { Command } from 'commander';
 
-var _definitions;
-var _reverseDefinitions;
-var _defaults;
-program.loadDefinitions = function(definitions) {
+let _definitions;
+let _reverseDefinitions;
+let _defaults;
+
+Command.prototype.loadDefinitions = function(definitions) {
   _definitions = definitions;
-  Object.keys(definitions).reduce(function(program, opt){
+  
+  Object.keys(definitions).reduce((program, opt) => {
     if (typeof definitions[opt] == "object") {
       const additionalOptions = definitions[opt];
       if (additionalOptions.required === true) {
@@ -14,15 +16,17 @@ program.loadDefinitions = function(definitions) {
         return program.option(`--${opt} [${opt}]`, additionalOptions.help, additionalOptions.action);
       }
     }
-    return program.option(`--${opt} [${opt}]`)
-  }, program);
-  _defaults = Object.keys(definitions).reduce(function(defs, opt) {
+    return program.option(`--${opt} [${opt}]`);
+  }, this);
+  
+  _defaults = Object.keys(definitions).reduce((defs, opt) => {
     if(_definitions[opt].default) {
       defs[opt] = _definitions[opt].default;
     }
     return defs;
   }, {});
-  _reverseDefinitions = Object.keys(definitions).reduce(function(object, key){
+  
+  _reverseDefinitions = Object.keys(definitions).reduce((object, key) => {
       let value = definitions[key];
       if (typeof value == "object") {
         value = value.env;
@@ -34,21 +38,21 @@ program.loadDefinitions = function(definitions) {
    }, {});
    
    /* istanbul ignore next */
-   program.on('--help', function(){
+   this.on('--help', function(){
     console.log('  Configure From Environment:');
     console.log('');
-    Object.keys(_reverseDefinitions).forEach(function(key){
+    Object.keys(_reverseDefinitions).forEach((key) => {
       console.log(`    $ ${key}='${_reverseDefinitions[key]}'`);
     });
     console.log('');
   });
 }
 
-var envParser = function(env = {}) {
-  return Object.keys(_reverseDefinitions).reduce(function(options, key){
+function parseEnvironment(env = {}) {
+  return Object.keys(_reverseDefinitions).reduce((options, key) => {
     if (env[key]) {
       const originalKey = _reverseDefinitions[key];
-      let action = function(option) {return option;}
+      let action = (option) => (option);
       if (typeof _definitions[originalKey] === "object") {
         action = _definitions[originalKey].action || action;
       }
@@ -58,23 +62,24 @@ var envParser = function(env = {}) {
   }, {});
 }
 
-program._parse = program.parse;
+Command.prototype.setValuesIfNeeded = function(options) {
+  Object.keys(options).forEach((key) => {
+   if (!this[key]) {
+     this[key] = options[key];
+   } 
+  });
+} 
 
-program.parse = function(args, env) {
-  program._parse(args);
+Command.prototype._parse = Command.prototype.parse;
+
+Command.prototype.parse = function(args, env) {
+  this._parse(args);
   // Parse the environment first
-  var envOptions = envParser(env);
+  const envOptions = parseEnvironment(env);
+  
   // Load the env if not passed from command line
-  Object.keys(envOptions).forEach(function(key){
-   if (!program[key]) {
-     program[key] = envOptions[key];
-   } 
-  });
-  Object.keys(_defaults).forEach(function(key){
-    if (!program[key]) {
-     program[key] = _defaults[key];
-   } 
-  });
+  this.setValuesIfNeeded(envOptions);
+  this.setValuesIfNeeded(_defaults);
 }
 
-module.exports =  program;
+export default new Command();