Add duration to final log for generation

1 files changed, 11 insertions, 2 deletions

diff --git a/lib/output/generatePage.js b/lib/output/generatePage.js
index 113bf22..aa3b376 100644
--- a/lib/output/generatePage.js
+++ b/lib/output/generatePage.js
@@ -4,6 +4,7 @@ var error = require('../utils/error');
 var Parse = require('../parse');
 var Templating = require('../templating');
 var createTemplateEngine = require('./createTemplateEngine');
+var callPageHook = require('./callPageHook');
 
 /**
     Prepare and generate HTML for a page
@@ -28,10 +29,13 @@ function generatePage(output, page) {
             }));
         }
 
-        return Promise(resultPage.getContent())
+        // Call hook "page:before"
+        return callPageHook('page:before', output, resultPage)
 
         // Escape code blocks with raw tags
-        .then(parser.page.prepare)
+        .then(function(currentPage) {
+            return parser.page.prepare(currentPage.getContent());
+        })
 
         // Render templating syntax
         .then(function(content) {
@@ -49,6 +53,11 @@ function generatePage(output, page) {
         // Return new page
         .then(function(content) {
             return resultPage.set('content', content);
+        })
+
+        // Call final hook
+        .then(function(currentPage) {
+            return callPageHook('page', output, currentPage);
         });
     });
 }