Add command "serve"

1 files changed, 6 insertions, 5 deletions

diff --git a/lib/cli/build.js b/lib/cli/build.js
index 49931bd..978bc87 100644
--- a/lib/cli/build.js
+++ b/lib/cli/build.js
@@ -1,10 +1,12 @@
 var path = require('path');
 
+var Parse = require('../parse');
+var Output = require('../output');
+
 var options = require('./options');
 var getBook = require('./getBook');
+var getOutputFolder = require('./getOutputFolder');
 
-var Parse = require('../parse');
-var Output = require('../output');
 
 module.exports = {
     name: 'build [book] [output]',
@@ -15,13 +17,12 @@ module.exports = {
     ],
     exec: function(args, kwargs) {
         var book = getBook(args, kwargs);
+        var outputFolder = getOutputFolder(args);
+
         var Generator = Output.getGenerator(kwargs.format);
 
         return Parse.parseBook(book)
         .then(function(resultBook) {
-            var defaultOutputRoot = path.join(resultBook.getRoot(), '_book');
-            var outputFolder = args[1]? path.resolve(process.cwd(), args[1]) : defaultOutputRoot;
-
             return Output.generate(Generator, resultBook, {
                 root: outputFolder
             });