From a506a6d3679c282cfd3a2f0257f5859d420feb22 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 20 Feb 2016 15:02:54 +0000
Subject: Allow paragraph matchers by style name

---
 src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(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 98ae351..b33e0db 100644
--- a/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
+++ b/src/main/java/org/zwobble/mammoth/styles/StyleMapBuilder.java
@@ -1,11 +1,11 @@
 package org.zwobble.mammoth.styles;
 
-import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableList;
 
 public class StyleMapBuilder {
     private HtmlPath underline;
     private HtmlPath strikethrough;
-    private final ImmutableMap.Builder<String, HtmlPath> paragraphStyles = ImmutableMap.builder();
+    private final ImmutableList.Builder<StyleMapping> paragraphStyles = ImmutableList.builder();
 
     public StyleMapBuilder() {
         this.underline = HtmlPath.EMPTY;
@@ -22,8 +22,8 @@ public class StyleMapBuilder {
         return this;
     }
 
-    public StyleMapBuilder mapParagraph(String styleId, HtmlPath path) {
-        paragraphStyles.put(styleId, path);
+    public StyleMapBuilder mapParagraph(ParagraphMatcher paragraphMatcher, HtmlPath path) {
+        paragraphStyles.add(new StyleMapping(paragraphMatcher, path));
         return this;
     }
 
-- 
cgit v1.2.3