From dc6e1277e8255a37727951b78ba5c8fd52de12ed Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 28 Nov 2016 16:21:49 +0000
Subject: Use break style maps in document conversion

---
 src/main/java/org/zwobble/mammoth/internal/styles/StyleMap.java      | 5 +++++
 .../java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java    | 5 +++++
 2 files changed, 10 insertions(+)

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

diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMap.java b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMap.java
index 0e8ed6d..9856d70 100644
--- a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMap.java
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMap.java
@@ -96,4 +96,9 @@ public class StyleMap {
         return tryFind(runStyles, styleMapping -> styleMapping.matches(run))
             .map(StyleMapping::getHtmlPath);
     }
+
+    public Optional<HtmlPath> getBreakHtmlPath(Break breakElement) {
+        return tryFind(breakStyles, styleMapping -> styleMapping.matches(breakElement))
+            .map(StyleMapping::getHtmlPath);
+    }
 }
diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
index 2d66f2a..4972883 100644
--- a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
@@ -61,6 +61,11 @@ public class StyleMapBuilder {
         return this;
     }
 
+    public StyleMapBuilder mapBreak(BreakMatcher matcher, HtmlPath path) {
+        breakStyles.add(new StyleMapping<>(matcher, path));
+        return this;
+    }
+
     public StyleMap build() {
         return new StyleMap(bold, italic, underline, strikethrough, commentReference, paragraphStyles, runStyles, breakStyles);
     }
-- 
cgit v1.2.3