Files
kami-parse-server/src/cli/utils/commander.js

86 lines
2.4 KiB
JavaScript
Raw Normal View History

2016-02-24 08:50:04 -05:00
import { Command } from 'commander';
2016-02-19 20:16:14 -05:00
2016-02-24 08:50:04 -05:00
let _definitions;
let _reverseDefinitions;
let _defaults;
Command.prototype.loadDefinitions = function(definitions) {
2016-02-19 20:16:14 -05:00
_definitions = definitions;
2016-02-24 08:50:04 -05:00
Object.keys(definitions).reduce((program, opt) => {
2016-02-19 20:16:14 -05:00
if (typeof definitions[opt] == "object") {
const additionalOptions = definitions[opt];
if (additionalOptions.required === true) {
return program.option(`--${opt} <${opt}>`, additionalOptions.help, additionalOptions.action);
} else {
return program.option(`--${opt} [${opt}]`, additionalOptions.help, additionalOptions.action);
}
}
2016-02-24 08:50:04 -05:00
return program.option(`--${opt} [${opt}]`);
}, this);
_defaults = Object.keys(definitions).reduce((defs, opt) => {
if(_definitions[opt].default) {
defs[opt] = _definitions[opt].default;
}
return defs;
}, {});
2016-02-24 08:50:04 -05:00
_reverseDefinitions = Object.keys(definitions).reduce((object, key) => {
2016-02-19 20:16:14 -05:00
let value = definitions[key];
if (typeof value == "object") {
value = value.env;
}
if (value) {
object[value] = key;
}
return object;
}, {});
2016-02-23 10:56:01 -05:00
/* istanbul ignore next */
2016-02-24 08:50:04 -05:00
this.on('--help', function(){
2016-02-19 20:16:14 -05:00
console.log(' Configure From Environment:');
console.log('');
2016-02-24 08:50:04 -05:00
Object.keys(_reverseDefinitions).forEach((key) => {
2016-02-19 20:16:14 -05:00
console.log(` $ ${key}='${_reverseDefinitions[key]}'`);
});
console.log('');
});
}
2016-02-24 08:50:04 -05:00
function parseEnvironment(env = {}) {
return Object.keys(_reverseDefinitions).reduce((options, key) => {
2016-02-19 20:16:14 -05:00
if (env[key]) {
const originalKey = _reverseDefinitions[key];
2016-02-24 08:50:04 -05:00
let action = (option) => (option);
2016-02-19 20:16:14 -05:00
if (typeof _definitions[originalKey] === "object") {
action = _definitions[originalKey].action || action;
}
options[_reverseDefinitions[key]] = action(env[key]);
}
return options;
}, {});
}
2016-02-24 08:50:04 -05:00
Command.prototype.setValuesIfNeeded = function(options) {
Object.keys(options).forEach((key) => {
if (!this[key]) {
this[key] = options[key];
}
});
}
Command.prototype._parse = Command.prototype.parse;
2016-02-19 20:16:14 -05:00
2016-02-24 08:50:04 -05:00
Command.prototype.parse = function(args, env) {
this._parse(args);
2016-02-19 20:16:14 -05:00
// Parse the environment first
2016-02-24 08:50:04 -05:00
const envOptions = parseEnvironment(env);
2016-02-19 20:16:14 -05:00
// Load the env if not passed from command line
2016-02-24 08:50:04 -05:00
this.setValuesIfNeeded(envOptions);
this.setValuesIfNeeded(_defaults);
2016-02-19 20:16:14 -05:00
}
2016-02-24 08:50:04 -05:00
export default new Command();