boolean() with passing test

yargs · Feb 13, 2011 · 0e380b9 · 0e380b9
1 parent 62644d4
commit 0e380b9
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -58,6 +58,24 @@ function Argv (args, cwd) {
         }
     }
 
+    self.boolean = function (bools) {
+        if (!Array.isArray(bools)) {
+            bools = [].slice.call(arguments);
+        }
+
+        bools.forEach(function (name) {
+            if (self.argv[name] === undefined) {
+                self.argv[name] = false;
+            }
+            else {
+                self.argv._.push(self.argv[name]);
+                self.argv[name] = true;
+            }
+        });
+
+        return self;
+    };
+
     for (var i = 0; i < args.length; i++) {
         var arg = args[i];
 

diff --git a/test/parse.js b/test/parse.js
@@ -159,3 +159,9 @@ exports.nums = function (assert) {
     assert.eql(typeof argv.w, 'string');
     assert.eql(typeof argv._[0], 'number');
 };
+
+exports['flag boolean'] = function (assert) {
+    var parse = optimist([ '-t', 'moo' ]).boolean(['t']).argv;
+    assert.eql(parse, { t : true, _ : [ 'moo' ], $0 : 'expresso' });
+    assert.eql(typeof parse.t, 'boolean');
+};