Normalize summary in gitbook

2 files changed, 5 insertions, 1 deletions

diff --git a/lib/book.js b/lib/book.js
index 27a609c..343a340 100644
--- a/lib/book.js
+++ b/lib/book.js
@@ -6,6 +6,7 @@ var fs = require("./utils/fs");
 var parseNavigation = require("./utils/navigation");
 var parseProgress = require("./utils/progress");
 var pageUtil = require("./utils/page");
+var summaryUtil = require("./utils/summary");
 
 var Configuration = require("./configuration");
 var TemplateEngine = require("./template");
@@ -270,7 +271,7 @@ Book.prototype.parseSummary = function() {
 		});
 	})
 	.then(function(summary) {
-		that.summary = summary;
+		that.summary = summaryUtil.normalize(summary);
 		that.navigation = parseNavigation(that.summary, that.files);
 	});
 };
diff --git a/lib/generators/site.js b/lib/generators/site.js
index 5eb8782..242e153 100644
--- a/lib/generators/site.js
+++ b/lib/generators/site.js
@@ -75,6 +75,9 @@ Generator.prototype.prepareTemplates = function() {
 
     // Add filter
     this.env.addFilter("contentLink", this.contentLink.bind(this));
+    this.env.addFilter('lvl', function(lvl) {
+        return lvl.split(".").length;
+    });
 
     // Add extension
     this.env.addExtension('ParentExtension', new ParentExtension());