Improve ebook generation (toc, ...)

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/generate/page/index.js b/lib/generate/page/index.js
index c2f4484..7e6826d 100644
--- a/lib/generate/page/index.js
+++ b/lib/generate/page/index.js
@@ -18,6 +18,11 @@ swig.setFilter('code', function(code, lang) {
     }
 });
 
+// Convert a level into a deep level
+swig.setFilter('lvl', function(lvl) {
+    return lvl.split(".").length;
+});
+
 
 /*
  *  This generator will generate a simple index.html which can be converted as a PDF
@@ -66,11 +71,14 @@ Generator.prototype.finish = function() {
     var basePath = ".";
     var output = path.join(this.options.output, "index.html");
 
+    var progress = parse.progress(this.options.navigation, "README.md");
+
     return Q()
     // Generate html
     .then(function(pages) {
         return that._writeTemplate(that.template, {
             pages: that.pages,
+            progress: progress,
 
             basePath: basePath,
             staticBase: path.join(basePath, "gitbook"),