Merge pull request #482 from GitbookIO/feature/includes

Add include support, with variables and the whole shebang

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/generate/generator.js b/lib/generate/generator.js
index 8724a4f..84c65d7 100644
--- a/lib/generate/generator.js
+++ b/lib/generate/generator.js
@@ -14,12 +14,35 @@ var BaseGenerator = function(options) {
     this.options.plugins = Plugin.normalizeNames(this.options.plugins);
     this.options.plugins = _.union(this.options.plugins, this.options.defaultsPlugins);
     this.plugins = [];
+
+    // Include variables (not yet loaded)
+    this.variables = {};
 };
 
 BaseGenerator.prototype.callHook = function(name, data) {
     return this.plugins.hook(name, data);
 };
 
+// Sets up generator
+BaseGenerator.prototype.load = function() {
+    return this.loadVariables()
+    .then(this.loadPlugins.bind(this));
+};
+
+BaseGenerator.prototype.loadVariables = function() {
+    var that = this;
+
+    return fs.readFile(path.join(this.options.input, 'variables.json'), 'utf8')
+    .then(function(content) {
+        try {
+            that.variables = JSON.parse(content);
+        } catch(err) {
+            console.log('No variables.json');
+        }
+    })
+    .fail(function() {});
+};
+
 BaseGenerator.prototype.loadPlugins = function() {
     var that = this;