Add base for calling hook on page/global

1 files changed, 45 insertions, 2 deletions

diff --git a/lib/output/generateBook.js b/lib/output/generateBook.js
index fb03418..c35719b 100644
--- a/lib/output/generateBook.js
+++ b/lib/output/generateBook.js
@@ -1,4 +1,5 @@
 var Output = require('../models/output');
+var Config = require('../models/config');
 var Promise = require('../utils/promise');
 
 var callHook = require('./callHook');
@@ -28,7 +29,7 @@ var generatePages = require('./generatePages');
     @param {Book} book
     @param {Object} options
 
-    @return {Promise}
+    @return {Promise<Output>}
 */
 function generateBook(generator, book, options) {
     options = generator.Options(options);
@@ -39,8 +40,50 @@ function generateBook(generator, book, options) {
     .then(preparePlugins)
     .then(preparePages)
     .then(prepareAssets)
+
+    .then(
+        callHook.bind(null,
+            'config',
+            function(output) {
+                var book = output.getBook();
+                var config = book.getConfig();
+                var values = config.getValues();
+
+                return values.toJS();
+            },
+            function(output, result) {
+                var book = output.getBook();
+                var config = book.getConfig();
+
+                config = Config.updateValues(config, result);
+                book = book.set('config', config);
+                return output.set('book', book);
+            }
+        )
+    )
+
+    .then(callHook.bind(null, 'init'))
+
+    .then(function(output) {
+        if (!generator.onInit) {
+            return output;
+        }
+
+        return generator.onInit(output);
+    })
+
     .then(generateAssets.bind(null, generator))
-    .then(generatePages.bind(null, generator));
+    .then(generatePages.bind(null, generator))
+
+    .then(callHook.bind(null, 'finish:before'))
+
+    .then(function(output) {
+        if (!generator.onFinish) {
+            return output;
+        }
+
+        return generator.onFinish(output);
+    });
 }
 
 module.exports = generateBook;