Wire up comment conversion code

5 files changed, 33 insertions, 6 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/BodyXmlReader.java b/src/main/java/org/zwobble/mammoth/internal/docx/BodyXmlReader.java
index 6dd1134..0c514fb 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/BodyXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/BodyXmlReader.java
@@ -116,6 +116,7 @@ public class BodyXmlReader {
             case "w:del":
             case "w:footnoteRef":
             case "w:endnoteRef":
+            case "w:annotationRef":
             case "w:pPr":
             case "w:rPr":
             case "w:tblPr":
diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/DocumentReader.java b/src/main/java/org/zwobble/mammoth/internal/docx/DocumentReader.java
index b53c9dd..0452c63 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/DocumentReader.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/DocumentReader.java
@@ -1,5 +1,6 @@
 package org.zwobble.mammoth.internal.docx;
 
+import org.zwobble.mammoth.internal.documents.Comment;
 import org.zwobble.mammoth.internal.documents.Document;
 import org.zwobble.mammoth.internal.documents.Notes;
 import org.zwobble.mammoth.internal.results.InternalResult;
@@ -9,6 +10,7 @@ import org.zwobble.mammoth.internal.xml.XmlElement;
 
 import java.io.IOException;
 import java.nio.file.Path;
+import java.util.List;
 import java.util.Optional;
 
 import static org.zwobble.mammoth.internal.util.Lists.list;
@@ -28,13 +30,21 @@ public class DocumentReader {
             Relationships relationships = readRelationships(zipFile, name);
             return new BodyXmlReader(styles, numbering, relationships, contentTypes, zipFile, fileReader);
         };
-        return readNotes(zipFile, bodyReaders)
-            .flatMap(notes -> {
-                DocumentXmlReader reader = new DocumentXmlReader(bodyReaders.forName("document"), notes);
+        return InternalResult.flatMap(
+            readNotes(zipFile, bodyReaders),
+            readComments(zipFile, bodyReaders),
+            (notes, comments) -> {
+                DocumentXmlReader reader = new DocumentXmlReader(bodyReaders.forName("document"), notes, comments);
                 return reader.readElement(parseOfficeXml(zipFile, "word/document.xml"));
             });
     }
 
+    private static InternalResult<List<Comment>> readComments(DocxFile file, BodyReaders bodyReaders) {
+        return tryParseOfficeXml(file, "word/comments.xml")
+            .map(new CommentXmlReader(bodyReaders.forName("comments"))::readElement)
+            .orElse(InternalResult.success(list()));
+    }
+
     private static InternalResult<Notes> readNotes(DocxFile file, BodyReaders bodyReaders) {
         return InternalResult.map(
             tryParseOfficeXml(file, "word/footnotes.xml")
diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/DocumentXmlReader.java b/src/main/java/org/zwobble/mammoth/internal/docx/DocumentXmlReader.java
index bd7247e..25855f3 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/DocumentXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/DocumentXmlReader.java
@@ -1,26 +1,29 @@
 package org.zwobble.mammoth.internal.docx;
 
+import org.zwobble.mammoth.internal.documents.Comment;
 import org.zwobble.mammoth.internal.documents.Document;
 import org.zwobble.mammoth.internal.documents.Notes;
 import org.zwobble.mammoth.internal.results.InternalResult;
 import org.zwobble.mammoth.internal.xml.XmlElement;
 import org.zwobble.mammoth.internal.xml.XmlElementLike;
 
-import static org.zwobble.mammoth.internal.util.Lists.list;
+import java.util.List;
 
 public class DocumentXmlReader {
     private final BodyXmlReader bodyReader;
     private final Notes notes;
+    private final List<Comment> comments;
 
-    public DocumentXmlReader(BodyXmlReader bodyReader, Notes notes) {
+    public DocumentXmlReader(BodyXmlReader bodyReader, Notes notes, List<Comment> comments) {
         this.bodyReader = bodyReader;
         this.notes = notes;
+        this.comments = comments;
     }
 
     public InternalResult<Document> readElement(XmlElement element) {
         XmlElementLike body = element.findChildOrEmpty("w:body");
         return bodyReader.readElements(body.getChildren())
             .toResult()
-            .map(children -> new Document(children, notes, list()));
+            .map(children -> new Document(children, notes, comments));
     }
 }
diff --git a/src/main/java/org/zwobble/mammoth/internal/results/InternalResult.java b/src/main/java/org/zwobble/mammoth/internal/results/InternalResult.java
index b2e892b..d18f278 100644
--- a/src/main/java/org/zwobble/mammoth/internal/results/InternalResult.java
+++ b/src/main/java/org/zwobble/mammoth/internal/results/InternalResult.java
@@ -23,6 +23,17 @@ public class InternalResult<T> {
             lazyFlatMap(results, result -> result.warnings));
     }
 
+    public static <T1, T2, R> InternalResult<R> flatMap(
+        InternalResult<T1> first,
+        InternalResult<T2> second,
+        BiFunction<T1, T2, InternalResult<R>> function)
+    {
+        InternalResult<R> intermediateResult = function.apply(first.value, second.value);
+        return new InternalResult<>(
+            intermediateResult.value,
+            lazyConcat(lazyConcat(first.warnings, second.warnings), intermediateResult.warnings));
+    }
+
     public static <T1, T2, R> InternalResult<R> map(
         InternalResult<T1> first,
         InternalResult<T2> second,
diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/DefaultStyles.java b/src/main/java/org/zwobble/mammoth/internal/styles/DefaultStyles.java
index 3aaf655..c9c5b5b 100644
--- a/src/main/java/org/zwobble/mammoth/internal/styles/DefaultStyles.java
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/DefaultStyles.java
@@ -26,6 +26,8 @@ public class DefaultStyles {
         "r[style-name='footnote reference'] =>",
         "p[style-name='endnote text'] => p",
         "r[style-name='endnote reference'] =>",
+        "p[style-name='annotation text'] => p",
+        "r[style-name='annotation reference'] =>",
 
         // LibreOffice
         "p[style-name='Footnote'] => p",