From d54918c26a0e6eae954eaa2052c7e8ec546c3409 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Tue, 23 Jan 2018 20:58:26 +0000
Subject: Read relationship type

---
 .../java/org/zwobble/mammoth/internal/docx/Relationships.java    | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/main/java/org/zwobble/mammoth/internal/docx/Relationships.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/Relationships.java b/src/main/java/org/zwobble/mammoth/internal/docx/Relationships.java
index d4d6f2e..bfd3fc3 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/Relationships.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/Relationships.java
@@ -10,16 +10,25 @@ public class Relationships {
     public static final Relationships EMPTY = new Relationships(list());
 
     private final Map<String, String> targetsByRelationshipId;
+    private final Map<String, List<String>> targetsByType;
 
     public Relationships(List<Relationship> relationships) {
         this.targetsByRelationshipId = toMap(relationships, relationship -> entry(
             relationship.getRelationshipId(),
             relationship.getTarget()
         ));
+        this.targetsByType = toMultiMap(relationships, relationship -> entry(
+            relationship.getType(),
+            relationship.getTarget()
+        ));
     }
 
     public String findTargetByRelationshipId(String relationshipId) {
         return lookup(targetsByRelationshipId, relationshipId)
             .orElseThrow(() -> new RuntimeException("Could not find relationship '" + relationshipId + "'"));
     }
+
+    public List<String> findTargetsByType(String type) {
+        return lookup(targetsByType, type).orElse(list());
+    }
 }
-- 
cgit v1.2.3