documentation/lib/commands/serve.js

a documentation generator

'use strict';

var errorPage = require('../serve/error_page');
var fs = require('fs');
var path = require('path');
var chokidar = require('chokidar');
var sharedOptions = require('./shared_options');
var Server = require('../serve/server');
var _ = require('lodash');
var getPort = require('get-port');
var documentation = require('../');

module.exports.command = 'serve [input..]';
module.exports.description = 'generate, update, and display HTML documentation';
/**
 * Add yargs parsing for the serve command
 * @param {Object} yargs module instance
 * @returns {Object} yargs with options
 * @private
 */
module.exports.builder = Object.assign({}, sharedOptions.sharedOutputOptions, sharedOptions.sharedInputOptions, {
  port: {
    describe: 'preferred port for the local server',
    type: 'number',
    default: 4001
  }
});

/**
 * Wrap the documentation build command along with a server, making it possible
 * to preview changes live
 * @private
 * @param {Object} argv cli input
 * @returns {undefined} has side effects
 */
module.exports.handler = function serve(argv) {
  argv._handled = true;

  if (!argv.input.length) {
    try {
      argv.input = [JSON.parse(fs.readFileSync(path.resolve('package.json'), 'utf8')).main || 'index.js'];
    } catch (e) {
      throw new Error('documentation was given no files and was not run in a module directory');
    }
  }

  getPort(argv.port).then(function (port) {
    var server = new Server(port);
    var watcher = void 0;

    server.on('listening', function () {
      process.stdout.write(`documentation.js serving on port ${port}\n`);
    });

    function updateWatcher() {
      if (!watcher) {
        watcher = chokidar.watch(argv.input);
        watcher.on('all', _.debounce(updateServer, 300));
      }

      documentation.expandInputs(argv.input, argv).then(function (files) {
        watcher.add(files.map(function (data) {
          return typeof data === 'string' ? data : data.file;
        }));
      }).catch(function (err) {
        /* eslint no-console: 0 */
        return server.setFiles([errorPage(err)]).start();
      });
    }

    function updateServer() {
      documentation.build(argv.input, argv).then(function (comments) {
        return documentation.formats.html(comments, argv);
      }).then(function (files) {
        if (argv.watch) {
          updateWatcher();
        }
        server.setFiles(files).start();
      }).catch(function (err) {
        return server.setFiles([errorPage(err)]).start();
      });
    }

    updateServer();
  });
};