Encode images as base64

1 files changed, 16 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/DocumentConverter.java b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
index dce76ad..da10324 100644
--- a/src/main/java/org/zwobble/mammoth/DocumentConverter.java
+++ b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
@@ -1,9 +1,12 @@
 package org.zwobble.mammoth;
 
+import com.google.common.io.ByteStreams;
 import org.zwobble.mammoth.documents.*;
 import org.zwobble.mammoth.html.Html;
 import org.zwobble.mammoth.html.HtmlNode;
 
+import java.io.IOException;
+import java.util.Base64;
 import java.util.List;
 import java.util.Map;
 
@@ -127,7 +130,19 @@ public class DocumentConverter {
 
             @Override
             public List<HtmlNode> visit(Image image) {
-                throw new UnsupportedOperationException();
+                // TODO: handle empty content type
+                return image.getContentType()
+                    .map(contentType -> {
+                        try {
+                            String base64 = Base64.getEncoder().encodeToString(ByteStreams.toByteArray(image.open()));
+                            String src = "data:" + contentType + ";base64," + base64;
+                            return list(Html.selfClosingElement("img", map("src", src)));
+                        } catch (IOException exception) {
+                            // TODO: return a result with a warning
+                            throw new RuntimeException(exception);
+                        }
+                    })
+                    .orElse(list());
             }
         });
     }