From d9a34a0836a7e41a975017fb021f5af90d954ae5 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 23 Jan 2018 21:26:38 +0000
Subject: Find main document part using _rels/.rels

---
 src/main/java/org/zwobble/mammoth/internal/archives/Archive.java     | 1 +
 .../java/org/zwobble/mammoth/internal/archives/InMemoryArchive.java  | 5 +++++
 .../java/org/zwobble/mammoth/internal/archives/ZippedArchive.java    | 5 +++++
 3 files changed, 11 insertions(+)

(limited to 'src/main/java/org/zwobble/mammoth/internal/archives')

diff --git a/src/main/java/org/zwobble/mammoth/internal/archives/Archive.java b/src/main/java/org/zwobble/mammoth/internal/archives/Archive.java
index 4440234..2f5a4ae 100644
--- a/src/main/java/org/zwobble/mammoth/internal/archives/Archive.java
+++ b/src/main/java/org/zwobble/mammoth/internal/archives/Archive.java
@@ -7,4 +7,5 @@ import java.util.Optional;
 
 public interface Archive extends Closeable {
     Optional<InputStream> tryGetInputStream(String name) throws IOException;
+    boolean exists(String name);
 }
diff --git a/src/main/java/org/zwobble/mammoth/internal/archives/InMemoryArchive.java b/src/main/java/org/zwobble/mammoth/internal/archives/InMemoryArchive.java
index 9ad7134..0844e8d 100644
--- a/src/main/java/org/zwobble/mammoth/internal/archives/InMemoryArchive.java
+++ b/src/main/java/org/zwobble/mammoth/internal/archives/InMemoryArchive.java
@@ -45,6 +45,11 @@ public class InMemoryArchive implements MutableArchive {
             .map(ByteArrayInputStream::new);
     }
 
+    @Override
+    public boolean exists(String name) {
+        return entries.containsKey(name);
+    }
+
     @Override
     public void writeEntry(String path, String content) {
         entries.put(path, content.getBytes(StandardCharsets.UTF_8));
diff --git a/src/main/java/org/zwobble/mammoth/internal/archives/ZippedArchive.java b/src/main/java/org/zwobble/mammoth/internal/archives/ZippedArchive.java
index 853ced9..47be074 100644
--- a/src/main/java/org/zwobble/mammoth/internal/archives/ZippedArchive.java
+++ b/src/main/java/org/zwobble/mammoth/internal/archives/ZippedArchive.java
@@ -28,6 +28,11 @@ public class ZippedArchive implements Archive {
         }
     }
 
+    @Override
+    public boolean exists(String name) {
+        return zipFile.getEntry(name) != null;
+    }
+
     @Override
     public void close() throws IOException {
         zipFile.close();
-- 
cgit v1.2.3