Normalize paths in html content

4 files changed, 142 insertions, 9 deletions

diff --git a/lib/generator.js b/lib/generator.js
index 58a9c79..fa6b8b0 100644
--- a/lib/generator.js
+++ b/lib/generator.js
@@ -83,9 +83,4 @@ BaseGenerator.prototype.finish = function() {
     return Q.reject(new Error("Could not finish generation"));
 };
 
-// Modify an extension from a path
-BaseGenerator.prototype.changeExtension = function(filename, newext) {
-    return path.basename(filename, path.extname(filename))+newext;
-}
-
 module.exports = BaseGenerator;
diff --git a/lib/generators/json.js b/lib/generators/json.js
index 2e8c731..65197e0 100644
--- a/lib/generators/json.js
+++ b/lib/generators/json.js
@@ -5,7 +5,8 @@ var _ = require("lodash");
 
 var fs = require("../utils/fs");
 var BaseGenerator = require("../generator");
-
+var links = require("../utils/links");
+var pageUtil = require("../utils/page");
 
 var Generator = function() {
     BaseGenerator.apply(this, arguments);
@@ -21,10 +22,14 @@ Generator.prototype.writeParsedFile = function(page, input) {
     var that = this;
     var json = {
         progress: page.progress,
-        sections: page.sections
+        sections: pageUtil.normalize(page.sections, {
+            navigation: that.book.navigation,
+            base: path.dirname(input) || './',
+            output: path.dirname(input) || './'
+        })
     };
 
-    var output = that.changeExtension(input, ".json");
+    var output = links.changeExtension(input, ".json");
     output = path.join(that.options.output, output);
 
     return fs.writeFile(
@@ -41,7 +46,7 @@ Generator.prototype.langsIndex = function(langs) {
     if (langs.length == 0) return Q.reject("Need at least one language");
 
     var mainLang = _.first(langs).lang;
-    var readme = that.changeExtension(that.book.readmeFile, ".json");
+    var readme = links.changeExtension(that.book.readmeFile, ".json");
 
     return Q()
     .then(function() {
diff --git a/lib/utils/links.js b/lib/utils/links.js
new file mode 100644
index 0000000..e607c94
--- /dev/null
+++ b/lib/utils/links.js
@@ -0,0 +1,67 @@
+var url = require('url');
+var path = require('path');
+
+// Is the link an external link
+var isExternal = function(href) {
+    try {
+        return Boolean(url.parse(href).protocol);
+    } catch(err) { }
+
+    return false;
+};
+
+// Return true if the link is relative
+var isRelative = function(href) {
+    try {
+        var parsed = url.parse(href);
+
+        return !parsed.protocol && parsed.path && parsed.path[0] != '/';
+    } catch(err) {}
+
+    return true;
+};
+
+// Relative to absolute path
+// dir: directory parent of the file currently in rendering process
+// outdir: directory parent from the html output
+
+var toAbsolute = function(_href, dir, outdir) {
+    // Absolute file in source
+    _href = path.join(dir, _href);
+
+    // make it relative to output
+    _href = path.relative(outdir, _href);
+
+    if (process.platform === 'win32') {
+        _href = _href.replace(/\\/g, '/');
+    }
+
+    return _href;
+};
+
+// Join links
+var join = function() {
+    var _href = path.join.apply(path, arguments);
+
+    if (process.platform === 'win32') {
+        _href = _href.replace(/\\/g, '/');
+    }
+
+    return _href;
+};
+
+// Change extension
+var changeExtension = function(filename, newext) {
+    return path.join(
+        path.dirname(filename),
+        path.basename(filename, path.extname(filename))+newext
+    );
+};
+
+module.exports = {
+    isRelative: isRelative,
+    isExternal: isExternal,
+    toAbsolute: toAbsolute,
+    join: join,
+    changeExtension: changeExtension
+};
diff --git a/lib/utils/page.js b/lib/utils/page.js
new file mode 100644
index 0000000..00e4757
--- /dev/null
+++ b/lib/utils/page.js
@@ -0,0 +1,66 @@
+var _ = require('lodash');
+var path = require('path');
+var cheerio = require('cheerio');
+
+var links = require('./links');
+
+// Adapt an html snippet to be relative to a base folder
+function normalizeHtml(src, options) {
+	var $ = cheerio.load(src);
+
+	$("img").each(function() {
+        var src = $(this).attr("src");
+
+        // Transform as absolute
+        if (links.isRelative(src)) {
+        	src = links.toAbsolute(src, options.base, options.output);
+        }
+
+        $(this).attr("src", src);
+    });
+
+    $("a").each(function() {
+        var href = $(this).attr("href");
+
+        if (links.isRelative(href)) {
+        	var absolutePath = path.join(options.base, href);
+
+        	// If is in navigation relative: change extsnio nto be .html
+        	if (options.navigation[absolutePath]) href = links.changeExtension(href, ".html");
+
+        	// Transform as absolute
+        	href = links.toAbsolute(href, options.base, options.output);
+        } else {
+        	$(this).attr("target", "_blank");
+        }
+
+        // Transform extension
+        $(this).attr("href", href);
+    });
+
+	return $.html();
+};
+
+// Adapt page content to be relative to a base folder
+function normalizePage(sections, options) {
+	options = _.defaults(options || {}, {
+		// Navigation to use to transform path
+		navigation: {},
+
+		// Directory parent of the file currently in rendering process
+		base: "./",
+
+		// Directory parent from the html output
+		output: "./"
+	});
+
+	return _.map(sections, function(section) {
+		if (section.type != "normal") return section;
+		section.content = normalizeHtml(section.content, options);
+		return section;
+	});
+};
+
+module.exports = {
+	normalize: normalizePage
+};