Convert svg images to png in ebook

1 files changed, 72 insertions, 3 deletions

diff --git a/lib/utils/page.js b/lib/utils/page.js
index 525722e..e526afb 100644
--- a/lib/utils/page.js
+++ b/lib/utils/page.js
@@ -1,8 +1,11 @@
+var Q = require('q');
 var _ = require('lodash');
 var path = require('path');
 var cheerio = require('cheerio');
 
 var links = require('./links');
+var imgUtils = require('./images');
+var fs = require('./fs');
 
 function replaceText($, el, search, replace, text_only ) {
     return $(el).each(function(){
@@ -61,6 +64,7 @@ function pregQuote( str ) {
 // Adapt an html snippet to be relative to a base folder
 function normalizeHtml(src, options) {
 	var $ = cheerio.load(src);
+    var toConvert = [];
 
 	$("img").each(function() {
         var src = $(this).attr("src");
@@ -70,6 +74,31 @@ function normalizeHtml(src, options) {
         	src = links.toAbsolute(src, options.base, options.output);
         }
 
+        // Convert if needed
+        if (options.convertImages) {
+            var ext = path.extname(src);
+            if (_.contains(imgUtils.INVALID, ext)) {
+                // Replace extension
+                var dest = path.join(path.dirname(src), path.basename(src, ext)+".png");
+
+                // Absolute with input
+                dest = path.resolve(options.book.root, dest);
+
+                // Get a name that doesn't exists
+                dest = fs.getUniqueFilename(dest);
+
+                // Reset as relative to book
+                dest = path.relative(options.book.root, dest);
+
+                options.book.log.debug.ln("detect invalid image (will be converted to png):", src);
+                toConvert.push({
+                    source: src,
+                    dest: dest
+                });
+                src = dest;
+            }
+        }
+
         $(this).attr("src", src);
     });
 
@@ -109,13 +138,40 @@ function normalizeHtml(src, options) {
         });
     });
 
-	return $.html();
+	return {
+        html: $.html(),
+        images: toConvert
+    };
+};
+
+// Convert svg images to png
+function convertImages(images, options) {
+    return _.reduce(images, function(prev, image) {
+        return prev.then(function() {
+            var imgin = path.resolve(options.book.root, image.source);
+            var imgout = path.resolve(options.book.options.output, image.dest);
+
+            options.book.log.debug("convert image", image.source, "to", image.dest, "...");
+            return imgUtils.convert(imgin, imgout)
+            .then(function() {
+                options.book.log.debug.ok();
+            }, function() {
+                options.book.log.debug.fail();
+            });
+        });
+    }, Q());
 };
 
 
 // Adapt page content to be relative to a base folder
 function normalizePage(sections, options) {
 	options = _.defaults(options || {}, {
+        // Current book
+        book: null,
+
+        // Do we need to convert svg?
+        convertImages: false,
+
         // Current file path
         input: ".",
 
@@ -132,11 +188,24 @@ function normalizePage(sections, options) {
         glossary: []
 	});
 
-	return _.map(sections, function(section) {
+    // List of images to convert
+    var toConvert = [];
+
+	sections = _.map(sections, function(section) {
 		if (section.type != "normal") return section;
-		section.content = normalizeHtml(section.content, options);
+
+        var out = normalizeHtml(section.content, options);;
+
+        toConvert = toConvert.concat(out.images);
+		section.content = out.html;
 		return section;
 	});
+
+    return Q()
+    .then(function() {
+        return convertImages(toConvert, options);
+    })
+    .thenResolve(sections);
 };
 
 // Extract text from sections