Use Style class in Paragraph

2 files changed, 26 insertions, 5 deletions

diff --git a/src/main/java/org/zwobble/mammoth/documents/ParagraphElement.java b/src/main/java/org/zwobble/mammoth/documents/ParagraphElement.java
index dcf8dad..22d9306 100644
--- a/src/main/java/org/zwobble/mammoth/documents/ParagraphElement.java
+++ b/src/main/java/org/zwobble/mammoth/documents/ParagraphElement.java
@@ -4,16 +4,16 @@ import java.util.List;
 import java.util.Optional;
 
 public class ParagraphElement implements DocumentElement, HasChildren {
-    private final Optional<String> styleId;
+    private final Optional<Style> style;
     private final List<DocumentElement> children;
 
-    public ParagraphElement(Optional<String> styleId, List<DocumentElement> children) {
-        this.styleId = styleId;
+    public ParagraphElement(Optional<Style> style, List<DocumentElement> children) {
+        this.style = style;
         this.children = children;
     }
 
-    public Optional<String> getStyleId() {
-        return styleId;
+    public Optional<Style> getStyle() {
+        return style;
     }
 
     public List<DocumentElement> getChildren() {
diff --git a/src/main/java/org/zwobble/mammoth/documents/Style.java b/src/main/java/org/zwobble/mammoth/documents/Style.java
new file mode 100644
index 0000000..82a25de
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/documents/Style.java
@@ -0,0 +1,21 @@
+package org.zwobble.mammoth.documents;
+
+import java.util.Optional;
+
+public class Style {
+    private final String styleId;
+    private final Optional<String> name;
+
+    public Style(String styleId, Optional<String> name) {
+        this.styleId = styleId;
+        this.name = name;
+    }
+
+    public String getStyleId() {
+        return styleId;
+    }
+
+    public Optional<String> getName() {
+        return name;
+    }
+}