From 07209bbf7b6b4da453f40c3431c9d305bf282130 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 16 Apr 2016 12:02:19 +0100
Subject: Move Base64 encoding into its own type

---
 .../zwobble/mammoth/internal/conversion/DocumentToHtml.java   | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
index deab0d3..268d942 100644
--- a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
+++ b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
@@ -6,14 +6,15 @@ import org.zwobble.mammoth.internal.html.HtmlNode;
 import org.zwobble.mammoth.internal.results.InternalResult;
 import org.zwobble.mammoth.internal.styles.HtmlPath;
 import org.zwobble.mammoth.internal.styles.StyleMap;
+import org.zwobble.mammoth.internal.util.Base64Encoding;
 import org.zwobble.mammoth.internal.util.Lists;
 
 import java.io.IOException;
+import java.io.InputStream;
 import java.util.*;
 
 import static org.zwobble.mammoth.internal.util.Lists.*;
 import static org.zwobble.mammoth.internal.util.Maps.map;
-import static org.zwobble.mammoth.internal.util.Streams.toByteArray;
 
 public class DocumentToHtml {
     public static InternalResult<List<HtmlNode>> convertToHtml(Document document, DocumentToHtmlOptions options) {
@@ -207,7 +208,7 @@ public class DocumentToHtml {
                         try {
                             Map<String, String> attributes = new HashMap<>();
 
-                            String base64 = Base64.getEncoder().encodeToString(toByteArray(image.open()));
+                            String base64 = imageToBase64(image);
                             String src = "data:" + contentType + ";base64," + base64;
                             attributes.put("src", src);
 
@@ -224,6 +225,12 @@ public class DocumentToHtml {
         });
     }
 
+    private static String imageToBase64(Image image) throws IOException {
+        try (InputStream stream = image.open()) {
+            return Base64Encoding.streamToBase64(stream);
+        }
+    }
+
     private String generateNoteHtmlId(NoteType noteType, String noteId) {
         return generateId(noteTypeToIdFragment(noteType) + "-" + noteId);
     }
-- 
cgit v1.2.3