refactor(args): Improve the way we use yargs to simplify the markdown-toc option

tmcw · tmcw · commit dd9c03a19ed7 · 2016-12-28T22:25:45.000-05:00
diff --git a/lib/commands/build.js b/lib/commands/build.js
@@ -26,9 +26,9 @@ module.exports.builder = extend({},
       default: 'json',
       choices: ['json', 'md', 'remark', 'html']
     },
-    'no-markdown-toc': {
+    'markdown-toc': {
       describe: 'include a table of contents in markdown output',
-      default: 'stdout',
+      default: true,
       type: 'boolean'
     },
     output: {
@@ -69,7 +69,7 @@ module.exports.handler = function build(argv, callback) {
       version: argv['project-version'] || (argv.package || {}).version,
       theme: argv.theme,
       paths: argv.paths,
-      'no-markdown-toc': argv['no-markdown-toc'],
+      markdownToc: argv.markdownToc,
       hljs: argv.hljs || {}
     };
 
diff --git a/lib/output/markdown_ast.js b/lib/output/markdown_ast.js
@@ -210,7 +210,7 @@ function markdownAST(comments, options, callback) {
 
   var root = rerouteLinks(linkerStack.link,
     u('root', generatorComment
-      .concat(options['no-markdown-toc'] ? [] : tableOfContentsHeading)
+      .concat(options.markdownToc ? tableOfContentsHeading : [])
       .concat(comments.reduce(function (memo, comment) {
         return memo.concat(generate(2, comment));
       }, []))));
diff --git a/test/test.js b/test/test.js
@@ -157,7 +157,7 @@ test('outputs', function (ttt) {
 
         if (file.match(/es6.input.js/)) {
           tt.test('no markdown TOC', function (t) {
-            outputMarkdown(_.cloneDeep(result), { 'no-markdown-toc': true }, function (err, result) {
+            outputMarkdown(_.cloneDeep(result), { markdownToc: false }, function (err, result) {
               t.ifError(err);
               var outputfile = file.replace('.input.js', '.output-toc.md');
               if (UPDATE) {
