Use break style maps in document conversion

1 files changed, 9 insertions, 5 deletions

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