129 lines
3.0 KiB
JavaScript
129 lines
3.0 KiB
JavaScript
/*
|
||
* argv.js: Simple memory-based store for command-line arguments.
|
||
*
|
||
* (C) 2011, Charlie Robbins and the Contributors.
|
||
*
|
||
*/
|
||
|
||
var util = require('util'),
|
||
common = require('../common'),
|
||
Memory = require('./memory').Memory;
|
||
|
||
//
|
||
// ### function Argv (options)
|
||
// #### @options {Object} Options for this instance.
|
||
// Constructor function for the Argv nconf store, a simple abstraction
|
||
// around the Memory store that can read command-line arguments.
|
||
//
|
||
var Argv = exports.Argv = function (options, usage) {
|
||
Memory.call(this, options);
|
||
|
||
options = options || {};
|
||
this.type = 'argv';
|
||
this.readOnly = options.readOnly !== undefined? options.readOnly : true;
|
||
this.options = options;
|
||
this.usage = usage;
|
||
|
||
if(typeof options.readOnly === 'boolean') {
|
||
this.readOnly = options.readOnly;
|
||
delete options.readOnly;
|
||
// FIXME; should not mutate options!!!!
|
||
} else {
|
||
this.readOnly = true;
|
||
}
|
||
|
||
if(typeof options.parseValues === 'boolean') {
|
||
this.parseValues = options.parseValues;
|
||
delete options.parseValues;
|
||
} else {
|
||
this.parseValues = false;
|
||
}
|
||
if (typeof options.transform === 'function') {
|
||
this.transform = options.transform;
|
||
delete options.transform;
|
||
} else {
|
||
this.transform = false;
|
||
}
|
||
if (typeof options.separator === 'string' || options.separator instanceof RegExp) {
|
||
this.separator = options.separator;
|
||
delete options.separator;
|
||
} else {
|
||
this.separator = '';
|
||
}
|
||
};
|
||
|
||
// Inherit from the Memory store
|
||
util.inherits(Argv, Memory);
|
||
|
||
//
|
||
// ### function loadSync ()
|
||
// Loads the data passed in from `process.argv` into this instance.
|
||
//
|
||
Argv.prototype.loadSync = function () {
|
||
this.loadArgv();
|
||
return this.store;
|
||
};
|
||
|
||
//
|
||
// ### function loadArgv ()
|
||
// Loads the data passed in from the command-line arguments
|
||
// into this instance.
|
||
//
|
||
Argv.prototype.loadArgv = function () {
|
||
var self = this,
|
||
yargs, argv;
|
||
|
||
yargs = isYargs(this.options) ?
|
||
this.options :
|
||
typeof this.options === 'object' ?
|
||
require('yargs')(process.argv.slice(2)).options(this.options) :
|
||
require('yargs')(process.argv.slice(2));
|
||
|
||
if (typeof this.usage === 'string') { yargs.usage(this.usage) }
|
||
|
||
argv = yargs.argv
|
||
|
||
if (!argv) {
|
||
return;
|
||
}
|
||
|
||
if (this.transform) {
|
||
argv = common.transform(argv, this.transform);
|
||
}
|
||
|
||
var tempWrite = false;
|
||
|
||
if(this.readOnly) {
|
||
this.readOnly = false;
|
||
tempWrite = true;
|
||
}
|
||
Object.keys(argv).forEach(function (key) {
|
||
var val = argv[key];
|
||
|
||
if (typeof val !== 'undefined') {
|
||
if (self.parseValues) {
|
||
val = common.parseValues(val);
|
||
}
|
||
|
||
if (self.separator) {
|
||
self.set(common.key.apply(common, key.split(self.separator)), val);
|
||
}
|
||
else {
|
||
self.set(key, val);
|
||
}
|
||
}
|
||
});
|
||
|
||
this.showHelp = yargs.showHelp
|
||
this.help = yargs.help
|
||
|
||
if (tempWrite) {
|
||
this.readOnly = true;
|
||
}
|
||
return this.store;
|
||
};
|
||
|
||
function isYargs(obj) {
|
||
return (typeof obj === 'function' || typeof obj === 'object') && ('argv' in obj);
|
||
}
|