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

---
 .../mammoth/internal/conversion/DocumentToHtml.java        | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
index c252683..cfa4f56 100644
--- a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
+++ b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
@@ -187,11 +187,15 @@ public class DocumentToHtml {
 
             @Override
             public List<HtmlNode> visit(Break breakElement) {
-                if (breakElement.getType() == Break.Type.LINE) {
-                    return list(Html.element("br"));
-                } else {
-                    return list();
-                }
+                HtmlPath mapping = styleMap.getBreakHtmlPath(breakElement)
+                    .orElseGet(() -> {
+                        if (breakElement.getType() == Break.Type.LINE) {
+                            return HtmlPath.element("br");
+                        } else {
+                            return HtmlPath.EMPTY;
+                        }
+                    });
+                return mapping.wrap(() -> list()).get();
             }
 
             @Override
-- 
cgit v1.2.3