1 files changed, 56 insertions, 0 deletions
diff --git a/lib/output/modifiers/svgToImg.js b/lib/output/modifiers/svgToImg.js
new file mode 100644
index 0000000..f31b06d
--- /dev/null
+++ b/lib/output/modifiers/svgToImg.js
@@ -0,0 +1,56 @@
+var path = require('path');
+var crc = require('crc');
+var domSerializer = require('dom-serializer');
+
+var editHTMLElement = require('./editHTMLElement');
+var fs = require('../../utils/fs');
+var LocationUtils = require('../../utils/location');
+
+/**
+    Render a cheerio DOM as html
+
+    @param {HTMLDom} $
+    @param {HTMLElement} dom
+    @param {Object}
+    @return {String}
+*/
+function renderDOM($, dom, options) {
+    if (!dom && $._root && $._root.children) {
+        dom = $._root.children;
+    }
+    options = options|| dom.options || $._options;
+    return domSerializer(dom, options);
+}
+
+/**
+    Replace SVG tag by IMG
+
+    @param {String} baseFolder
+    @param {HTMLDom} $
+*/
+function svgToImg(baseFolder, currentFile, $) {
+    var currentDirectory = path.dirname(currentFile);
+
+    return editHTMLElement($, 'svg', function($svg) {
+        var content = '<?xml version="1.0" encoding="UTF-8"?>' +
+            renderDOM($, $svg);
+
+        // We avoid generating twice the same PNG
+        var hash = crc.crc32(content).toString(16);
+        var fileName = hash + '.svg';
+        var filePath = path.join(baseFolder, fileName);
+
+        // Write the svg to the file
+        return fs.assertFile(filePath, function() {
+            return fs.writeFile(filePath, content, 'utf8');
+        })
+
+        // Return as image
+        .then(function() {
+            var src = LocationUtils.relative(currentDirectory, fileName);
+            $svg.replaceWith('<img src="' + src + '" />');
+        });
+    });
+}
+
+module.exports = svgToImg;