Page output a simple html string

2 files changed, 45 insertions, 16 deletions

diff --git a/lib/page/html.js b/lib/page/html.js
index f828d11..b19d5ed 100644
--- a/lib/page/html.js
+++ b/lib/page/html.js
@@ -4,6 +4,7 @@ var domSerializer = require('dom-serializer');
 var slug = require('github-slugid');
 
 var Promise = require('../utils/promise');
+var location = require('../utils/location');
 
 // Render a cheerio DOM as html
 function renderDOM($, dom, options) {
@@ -18,7 +19,10 @@ function HTMLPipeline(htmlString, opts) {
     _.bindAll(this);
 
     this.opts = _.defaults(opts || {}, {
-        convertImages: true
+        convertImages: true,
+
+        // Calcul new href for a relative link
+        onRelativeLink: _.identity
     });
 
     this.$ = cheerio.load(htmlString, {
@@ -31,15 +35,38 @@ function HTMLPipeline(htmlString, opts) {
     });
 }
 
+// Normalize links
+HTMLPipeline.prototype.normalizeLinks = function() {
+    var that = this;
+
+    this.$('a').each(function() {
+        var $a = that.$(this);
+
+        var href = $a.attr('href');
+        if (!href) return;
+
+        if (location.isAnchor(href)) {
+            // Don't "change" anchor links
+        } else if (location.isRelative(href)) {
+            $a.attr('href', that.opts.onRelativeLink(href));
+        } else {
+            // External links
+            $a.attr('target', '_blank');
+        }
+
+    });
+};
+
 // Add ID to headings
 HTMLPipeline.prototype.addHeadingIDs = function() {
     var that = this;
 
     this.$('h1,h2,h3,h4,h5,h6').each(function() {
-        // Already has an ID?
-        if (that.$(this).attr('id')) return;
+        var $h = that.$(this);
 
-        that.$(this).attr('id', slug(that.$(this).text()));
+        // Already has an ID?
+        if ($h.attr('id')) return;
+        $h.attr('id', slug($h.text()));
     });
 };
 
@@ -48,6 +75,7 @@ HTMLPipeline.prototype.output = function() {
     var that = this;
 
     return Promise()
+    .then(this.normalizeLinks)
     .then(this.addHeadingIDs)
     .then(function() {
         return renderDOM(that.$);
diff --git a/lib/page/index.js b/lib/page/index.js
index 8f8819c..e7a4fec 100644
--- a/lib/page/index.js
+++ b/lib/page/index.js
@@ -3,7 +3,6 @@ var path = require('path');
 var parsers = require('gitbook-parsers');
 
 var error = require('../utils/error');
-var Promise = require('../utils/promise');
 var HTMLPipeline = require('./html');
 
 /*
@@ -99,21 +98,23 @@ Page.prototype.parse = function(opts) {
     // Render markup using the parser
     .then(function() {
         return that.parser.page(that.content)
-        .then(that.update);
+        .then(function(out) {
+            var content = _.pluck(out.sections, 'content').join('\n');
+            that.update(content);
+        });
     })
 
     // Normalize HTML output
     .then(function() {
-        return Promise.map(that.content.sections, function(section) {
-            var pipeline = new HTMLPipeline(section.content, opts);
-
-            return pipeline.output()
-            .then(function(content) {
-                return {
-                    content: content
-                };
-            });
-        });
+        var pipelineOpts = _.extend({
+            onRelativeLink: function(href) {
+                console.log('href', href);
+            }
+        }, opts);
+        var pipeline = new HTMLPipeline(that.content, pipelineOpts);
+
+        return pipeline.output()
+        .then(that.update);
     });
 };