Add style mappings for runs

1 files changed, 12 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java b/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
index b33e0db..cdc3f31 100644
--- a/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
+++ b/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
@@ -1,11 +1,14 @@
 package org.zwobble.mammoth.styles;
 
 import com.google.common.collect.ImmutableList;
+import org.zwobble.mammoth.documents.Paragraph;
+import org.zwobble.mammoth.documents.Run;
 
 public class StyleMapBuilder {
     private HtmlPath underline;
     private HtmlPath strikethrough;
-    private final ImmutableList.Builder<StyleMapping> paragraphStyles = ImmutableList.builder();
+    private final ImmutableList.Builder<StyleMapping<Paragraph>> paragraphStyles = ImmutableList.builder();
+    private final ImmutableList.Builder<StyleMapping<Run>> runStyles = ImmutableList.builder();
 
     public StyleMapBuilder() {
         this.underline = HtmlPath.EMPTY;
@@ -22,12 +25,17 @@ public class StyleMapBuilder {
         return this;
     }
 
-    public StyleMapBuilder mapParagraph(ParagraphMatcher paragraphMatcher, HtmlPath path) {
-        paragraphStyles.add(new StyleMapping(paragraphMatcher, path));
+    public StyleMapBuilder mapParagraph(ParagraphMatcher matcher, HtmlPath path) {
+        paragraphStyles.add(new StyleMapping<Paragraph>(matcher, path));
+        return this;
+    }
+
+    public StyleMapBuilder mapRun(RunMatcher matcher, HtmlPath path) {
+        runStyles.add(new StyleMapping<Run>(matcher, path));
         return this;
     }
 
     public StyleMap build() {
-        return new StyleMap(underline, strikethrough, paragraphStyles.build());
+        return new StyleMap(underline, strikethrough, paragraphStyles.build(), runStyles.build());
     }
 }