Normalize .md links to .html when rendering pages

1 files changed, 27 insertions, 0 deletions

diff --git a/lib/parse/renderer.js b/lib/parse/renderer.js
new file mode 100644
index 0000000..66aad46
--- /dev/null
+++ b/lib/parse/renderer.js
@@ -0,0 +1,27 @@
+var inherits = require('util').inherits;
+
+var marked = require('marked');
+
+
+function GitBookRenderer(options) {
+    if(!(this instanceof GitBookRenderer)) {
+        return new GitBookRenderer(options);
+    }
+    GitBookRenderer.super_.call(this, options);
+}
+inherits(GitBookRenderer, marked.Renderer);
+
+
+GitBookRenderer.prototype.link = function(href, title, text) {
+    // Replace .md extensions by .html
+    return GitBookRenderer.super_.prototype.link.call(
+        this,
+        href.replace(/\.md$/, '.html'),
+        title,
+        text
+    );
+};
+
+
+// Exports
+module.exports = GitBookRenderer;