fixes #26

bunglegrind · Nov 28, 2024 · c6b9125 · c6b9125
1 parent 100a902
commit c6b9125
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/web-extension/extractHtml.js b/web-extension/extractHtml.js
@@ -515,6 +515,8 @@ chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
         "saveasebook-src",
         "src"
     );
+    const doc = new DOMParser().parseFromString(htmlContent, "text/html");
+    const res = new XMLSerializer().serializeToString(doc);
 
     allImages.forEach(function (tmpImg) {
         imgsPromises.push(promiseAddZip(tmpImg.originalUrl, tmpImg.filename));
@@ -529,7 +531,7 @@ chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
                 styleFileContent: styleFile,
                 styleFileName: "style" + generateRandomNumber() + ".css",
                 images: extractedImages,
-                content: htmlContent
+                content: res
             };
             sendResponse(result);
         }

diff --git a/web-extension/saveEbook.js b/web-extension/saveEbook.js
@@ -1,6 +1,11 @@
 var cssFileName = 'ebook.css';
 var ebookTitle = null;
 
+function filterHeader(content) {
+    const startTag = "<body>";
+    return content.substring(content.indexOf(startTag) + startTag.length);
+}
+
 chrome.runtime.onMessage.addListener((obj, sender, sendResponse) => {
     if (obj.shortcut && obj.shortcut === 'build-ebook') {
         buildEbook(obj.response);
@@ -132,8 +137,7 @@ function _buildEbook(allPages, fromMenu=false) {
             '<title>' + tmpPageTitle+ '</title>' +
             '<link href="../style/' + page.styleFileName + '" rel="stylesheet" type="text/css" />' +
             '</head><body>' +
-            page.content +
-            '</body></html>'
+            filterHeader(page.content)
         );
     });