From 9ac722cd96111afa9ec0522af4f6dbf95a62b7b4 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 20 Feb 2016 14:39:16 +0000
Subject: Add style mappings for paragraphs by style ID

---
 src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java')

diff --git a/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java b/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
index 93a2c8c..c815e1b 100644
--- a/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
+++ b/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
@@ -1,10 +1,13 @@
 package org.zwobble.mammoth.styles;
 
+import com.google.common.collect.ImmutableMap;
+
 import static org.zwobble.mammoth.util.MammothLists.list;
 
 public class StyleMapBuilder {
     private HtmlPath underline;
     private HtmlPath strikethrough;
+    private final ImmutableMap.Builder<String, HtmlPath> paragraphStyles = ImmutableMap.builder();
 
     public StyleMapBuilder() {
         this.underline = new HtmlPath(list());
@@ -29,7 +32,12 @@ public class StyleMapBuilder {
         return this;
     }
 
+    public StyleMapBuilder mapParagraph(String styleId, HtmlPath path) {
+        paragraphStyles.put(styleId, path);
+        return this;
+    }
+
     public StyleMap build() {
-        return new StyleMap(underline, strikethrough);
+        return new StyleMap(underline, strikethrough, paragraphStyles.build());
     }
 }
-- 
cgit v1.2.3