Update gen sidebar

Author: sy-records

lib/cli.js

@@ -111,7 +111,7 @@ require('yargs')
   })
   .command({
     command: 'gen <path>',
-    aliases: 't',
+    aliases: 'g',
     desc: chalk.gray(y18n.__('gen')),
     builder: yargs =>
       yargs.options({

lib/commands/gen.js

@@ -1,21 +1,23 @@
 'use strict'
 
 const fs = require('fs')
+const os = require('os')
 const {cwd, exists} = require('../util')
 const chalk = require('chalk')
 const path = require('path')
+const ignoreFiles = ['_navbar', '_coverpage', '_sidebar']
 
 // eslint-disable-next-line
 module.exports = function (path = '', sidebar) {
   const cwdPath = cwd(path || '.')
 
   if (exists(cwdPath)) {
     if (sidebar) {
-      console.log(sidebar)
       const sidebarPath = cwdPath + '/' + sidebar || '_sidebar.md'
 
       if (!exists(sidebarPath)) {
         genSidebar(cwdPath, sidebarPath)
+        console.log(chalk.green(`Generate sidebar file '${sidebar}' success.`))
         return true
       }
 
@@ -28,27 +30,48 @@ module.exports = function (path = '', sidebar) {
 }
 
 function genSidebar(cwdPath, sidebarPath) {
-  let node = ''
-  travel(cwdPath, function (pathname) {
+  let tree = ''
+  let lastPath = ''
+  let nodeName = ''
+  getDirFiles(cwdPath, function (pathname) {
     path.relative(pathname, cwdPath)
     pathname = pathname.replace(cwdPath + '/', '')
-    node += '* [' + path.basename(pathname, '.md') + '](' + pathname + ')\n'
+    let filename = path.basename(pathname, '.md')
+    let splitPath = pathname.split(path.sep)
+
+    if (ignoreFiles.indexOf(filename) === -1) {
+      nodeName = '- [' + filename + '](' + pathname + ')' + os.EOL
+    }
+
+    if (splitPath.length > 1) {
+      if (splitPath[0] !== lastPath) {
+        lastPath = splitPath[0]
+        tree += os.EOL + '- ' + splitPath[0] + os.EOL
+      }
+
+      tree += '  ' + nodeName
+    } else {
+      if (lastPath !== '') {
+        lastPath = ''
+        tree += os.EOL
+      }
+
+      tree += nodeName
+    }
   })
-  fs.writeFile(sidebarPath, node, 'utf8', err => {
+  fs.writeFile(sidebarPath, tree, 'utf8', err => {
     if (err) {
-      throw err
+      console.log(chalk.red(`Generate sidebar file error, message: ${err.message}`))
     }
-
-    console.log('gen success')
   })
 }
 
-function travel(dir, callback) {
+function getDirFiles(dir, callback) {
   fs.readdirSync(dir).forEach(function (file) {
     let pathname = path.join(dir, file)
 
     if (fs.statSync(pathname).isDirectory()) {
-      travel(pathname, callback)
+      getDirFiles(pathname, callback)
     } else if (path.extname(file) === '.md') {
       callback(pathname)
     }