Read optional attributes of comments

1 files changed, 11 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/CommentXmlReader.java b/src/main/java/org/zwobble/mammoth/internal/docx/CommentXmlReader.java
index 1740ec2..1b6a171 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/CommentXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/CommentXmlReader.java
@@ -28,8 +28,17 @@ public class CommentXmlReader {
             .map(children -> new Comment(
                 commentId,
                 children,
-                Optional.empty(),
-                Optional.empty()
+                readOptionalAttribute(element, "w:author"),
+                readOptionalAttribute(element, "w:initials")
             ));
     }
+
+    private Optional<String> readOptionalAttribute(XmlElement element, String name) {
+        String value = element.getAttributeOrNone(name).orElse("").trim();
+        if (value.isEmpty()) {
+            return Optional.empty();
+        } else {
+            return Optional.of(value);
+        }
+    }
 }