Read linked blips

3 files changed, 33 insertions, 6 deletions

diff --git a/src/main/java/org/zwobble/mammoth/Mammoth.java b/src/main/java/org/zwobble/mammoth/Mammoth.java
index 0651944..63a313f 100644
--- a/src/main/java/org/zwobble/mammoth/Mammoth.java
+++ b/src/main/java/org/zwobble/mammoth/Mammoth.java
@@ -8,6 +8,7 @@ import org.zwobble.mammoth.xml.XmlElement;
 
 import java.io.File;
 import java.io.IOException;
+import java.io.InputStream;
 import java.util.zip.ZipFile;
 
 import static com.google.common.collect.Iterables.transform;
@@ -21,12 +22,19 @@ public class Mammoth {
             Numbering numbering = Numbering.EMPTY;
             Relationships relationships = Relationships.EMPTY;
             ContentTypes contentTypes = ContentTypes.DEFAULT;
+            FileReader fileReader = new FileReader() {
+                @Override
+                public InputStream getInputStream(String uri) throws IOException {
+                    throw new UnsupportedOperationException();
+                }
+            };
             DocumentXmlReader reader = new DocumentXmlReader(new BodyXmlReader(
                 styles,
                 numbering,
                 relationships,
                 contentTypes,
-                zipFile));
+                zipFile,
+                fileReader));
             return reader.readElement(documentXml)
                 .map(Mammoth::convertToHtml);
         } catch (IOException e) {
diff --git a/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java b/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
index e911a63..0d49809 100644
--- a/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
@@ -27,25 +27,27 @@ public class BodyXmlReader {
     private static final Set<String> IMAGE_TYPES_SUPPORTED_BY_BROWSERS = ImmutableSet.of(
         "image/png", "image/gif", "image/jpeg", "image/svg+xml", "image/tiff");
 
-
     private final Styles styles;
     private final Numbering numbering;
     private final Relationships relationships;
     private final ContentTypes contentTypes;
     private final DocxFile file;
+    private final FileReader fileReader;
 
     public BodyXmlReader(
         Styles styles,
         Numbering numbering,
         Relationships relationships,
         ContentTypes contentTypes,
-        DocxFile file)
+        DocxFile file,
+        FileReader fileReader)
     {
         this.styles = styles;
         this.numbering = numbering;
         this.relationships = relationships;
         this.contentTypes = contentTypes;
         this.file = file;
+        this.fileReader = fileReader;
     }
 
     public ReadResult readElement(XmlElement element) {
@@ -276,9 +278,18 @@ public class BodyXmlReader {
     }
 
     private ReadResult readBlip(XmlElement blip, Optional<String> altText) {
-        String relationshipId = blip.getAttribute("r:embed");
-        String imagePath = relationshipIdToDocxPath(relationshipId);
-        return readImage(imagePath, altText, () -> file.getInputStream(imagePath));
+        Optional<String> embedRelationshipId = blip.getAttributeOrNone("r:embed");
+        Optional<String> linkRelationshipId = blip.getAttributeOrNone("r:link");
+        if (embedRelationshipId.isPresent()) {
+            String imagePath = relationshipIdToDocxPath(embedRelationshipId.get());
+            return readImage(imagePath, altText, () -> file.getInputStream(imagePath));
+        } else if (linkRelationshipId.isPresent()) {
+            String imagePath = relationships.findRelationshipById(linkRelationshipId.get()).getTarget();
+            return readImage(imagePath, altText, () -> fileReader.getInputStream(imagePath));
+        } else {
+            // TODO: emit warning
+            return ReadResult.EMPTY_SUCCESS;
+        }
     }
 
     private ReadResult readImage(String imagePath, Optional<String> altText, InputStreamSupplier open) {
diff --git a/src/main/java/org/zwobble/mammoth/docx/FileReader.java b/src/main/java/org/zwobble/mammoth/docx/FileReader.java
new file mode 100644
index 0000000..9bb5b25
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/docx/FileReader.java
@@ -0,0 +1,8 @@
+package org.zwobble.mammoth.docx;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+public interface FileReader {
+    InputStream getInputStream(String uri) throws IOException;
+}