Add option "--timing" to mesure gitbook performances

2 files changed, 14 insertions, 2 deletions

diff --git a/lib/cli/build.js b/lib/cli/build.js
index e64ab44..023901e 100644
--- a/lib/cli/build.js
+++ b/lib/cli/build.js
@@ -1,5 +1,6 @@
 var Parse = require('../parse');
 var Output = require('../output');
+var timing = require('../utils/timing');
 
 var options = require('./options');
 var getBook = require('./getBook');
@@ -11,7 +12,8 @@ module.exports = {
     description: 'build a book',
     options: [
         options.log,
-        options.format
+        options.format,
+        options.timing
     ],
     exec: function(args, kwargs) {
         var book = getBook(args, kwargs);
@@ -24,6 +26,9 @@ module.exports = {
             return Output.generate(Generator, resultBook, {
                 root: outputFolder
             });
+        })
+        .fin(function() {
+            if (kwargs.timing) timing.dump(book.getLogger());
         });
     }
 };
diff --git a/lib/cli/options.js b/lib/cli/options.js
index 26581cb..ddcb5c5 100644
--- a/lib/cli/options.js
+++ b/lib/cli/options.js
@@ -17,7 +17,14 @@ var formatOption = {
     defaults: 'website'
 };
 
+var timingOption = {
+    name: 'timing',
+    description: 'Print timing debug information',
+    defaults: false
+};
+
 module.exports = {
     log: logOptions,
-    format: formatOption
+    format: formatOption,
+    timing: timingOption
 };