fix(file.js, doctoc.js): don't match mdx files unless --syntax mdx

Author: I-3B

doctoc.js

@@ -114,7 +114,7 @@ for (var i = 0; i < argv._.length; i++) {
 
   if (stat.isDirectory()) {
     console.log ('\nDocToccing "%s" and its sub directories for %s.', target, mode);
-    files = file.findMarkdownFiles(target);
+    files = file.findMarkdownFiles(target, syntax);
   } else {
     console.log ('\nDocToccing single file "%s" for %s.', target, mode);
     files = [{ path: target }];

lib/file.js

@@ -2,58 +2,59 @@ var path  =  require('path')
  ,  fs  =  require('fs')
  ,  _   =  require('underscore');
 
-var markdownExts = ['.md', '.markdown', '.mdx'];
+var markdownExts = ['.md', '.markdown'];
+var mdxExts = ['.mdx'];
 var ignoredDirs  = ['.', '..', '.git', 'node_modules'];
 
-function separateFilesAndDirs(fileInfos) {
+function separateFilesAndDirs(fileInfos, syntax = "md") {;
   return {
     directories :  _(fileInfos).filter(function (x) {
       return x.isDirectory() && !_(ignoredDirs).include(x.name);
     }),
-    markdownFiles :  _(fileInfos).filter(function (x) { 
-      return x.isFile() && _(markdownExts).include(path.extname(x.name)); 
+    markdownFiles :  _(fileInfos).filter(function (x) {
+      return x.isFile() && _(syntax === 'mdx'? mdxExts : markdownExts).include(path.extname(x.name));
     })
   };
 }
 
-function findRec(currentPath) {
+function findRec(currentPath, syntax) {
   function getStat (entry) {
     var target = path.join(currentPath, entry),
       stat = fs.statSync(target);
 
-    return  _(stat).extend({ 
+    return  _(stat).extend({
       name: entry,
       path: target
     });
   }
-  
+
   function process (fileInfos) {
-    var res = separateFilesAndDirs(fileInfos);
+    var res = separateFilesAndDirs(fileInfos, syntax);
     var tgts = _(res.directories).pluck('path');
 
-    if (res.markdownFiles.length > 0) 
+    if (res.markdownFiles.length > 0)
       console.log('\nFound %s in "%s"', _(res.markdownFiles).pluck('name').join(', '), currentPath);
-    else 
+    else
       console.log('\nFound nothing in "%s"', currentPath);
 
-    return { 
+    return {
       markdownFiles :  res.markdownFiles,
       subdirs     :  tgts
     };
   }
 
   var stats                  =  _(fs.readdirSync(currentPath)).map(getStat)
     , res                    =  process(stats)
-    , markdownsInSubdirs     =  _(res.subdirs).map(findRec)
+    , markdownsInSubdirs     =  _(res.subdirs).map((subdir)=> findRec(subdir, syntax))
     , allMarkdownsHereAndSub =  res.markdownFiles.concat(markdownsInSubdirs);
 
   return _(allMarkdownsHereAndSub).flatten();
 }
 
 // Finds all markdown files in given directory and its sub-directories
-// @param {String  } dir - the absolute directory to search in 
-exports.findMarkdownFiles = function(dir) {
-  return findRec(dir);
+// @param {String  } dir - the absolute directory to search in
+exports.findMarkdownFiles = function(dir, syntax) {
+  return findRec(dir, syntax);
 };
 
 /* Example: