Fix generation for page format

1 files changed, 3 insertions, 21 deletions

diff --git a/lib/generate/page/index.js b/lib/generate/page/index.js
index 7f730ea..ac815e7 100644
--- a/lib/generate/page/index.js
+++ b/lib/generate/page/index.js
@@ -7,7 +7,7 @@ var hljs = require('highlight.js');
 
 var fs = require("../fs");
 var parse = require("../../parse");
-var BaseGenerator = require("../generator");
+var BaseGenerator = require("../site");
 
 // Swig filter: highlight coloration
 swig.setFilter('code', function(code, lang) {
@@ -71,30 +71,12 @@ Generator.prototype.finish = function() {
     return Q()
     // Generate html
     .then(function(pages) {
-        return that.template({
-            title: that.options.title,
-            description: that.options.description,
-
-            githubAuthor: that.options.github.split("/")[0],
-            githubId: that.options.github,
-            githubHost: that.options.githubHost,
-
-            summary: that.options.summary,
-            allNavigation: that.options.navigation,
-            
+        return that._writeTemplate(that.template, {
             pages: that.pages,
 
             basePath: basePath,
             staticBase: path.join(basePath, "gitbook"),
-        });
-    })
-
-    // Write html to index.html
-    .then(function(html) {
-        return fs.writeFile(
-            output,
-            html
-        );
+        }, output);
     })
 
     // Copy assets