Fix #629: add generator name in template context

2 files changed, 39 insertions, 1 deletions

diff --git a/lib/template.js b/lib/template.js
index f43073b..28b11c6 100644
--- a/lib/template.js
+++ b/lib/template.js
@@ -318,7 +318,8 @@ TemplateEngine.prototype.renderString = function(content, context, options) {
 
         // infos about gitbook
         gitbook: {
-            version: pkg.version
+            version: pkg.version,
+            generator: this.book.options.generator
         }
     });
     options = _.defaults(options || {}, {
diff --git a/test/templating.js b/test/templating.js
new file mode 100644
index 0000000..9165d7a
--- /dev/null
+++ b/test/templating.js
@@ -0,0 +1,37 @@
+var path = require('path');
+var _ = require('lodash');
+var assert = require('assert');
+var fs = require("fs");
+
+var pkg = require("../package.json");
+
+describe('Templating', function () {
+    before(function(done) {
+        testGeneration(books[0], "website", function(output) {}, done);
+    });
+
+    var testTpl = function(str, args, options) {
+        return books[0].template.renderString(str, args, options)
+        .then(books[0].template.postProcess)
+    };
+
+    it('should correctly have access to generator', function(done) {
+        qdone(
+            testTpl('{{ gitbook.generator }}')
+            .then(function(content) {
+                assert.equal(content, "website");
+            }),
+            done
+        );
+    });
+
+    it('should correctly have access to gitbook version', function(done) {
+        qdone(
+            testTpl('{{ gitbook.version }}')
+            .then(function(content) {
+                assert.equal(content, pkg.version.toString());
+            }),
+            done
+        );
+    });
+});