Use separator during collapse

2 files changed, 8 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/html/Html.java b/src/main/java/org/zwobble/mammoth/internal/html/Html.java
index 4c7a9a9..a3724bf 100644
--- a/src/main/java/org/zwobble/mammoth/internal/html/Html.java
+++ b/src/main/java/org/zwobble/mammoth/internal/html/Html.java
@@ -156,6 +156,10 @@ public class Html {
             tryCast(HtmlElement.class, node),
             (last, next) -> {
                 if (next.isCollapsible() && isMatch(last, next)) {
+                    String separator = next.getSeparator();
+                    if (!separator.isEmpty()) {
+                        last.getChildren().add(Html.text(separator));
+                    }
                     for (HtmlNode child : next.getChildren()) {
                         collapsingAdd(last.getChildren(), child);
                     }
diff --git a/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java b/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
index 9999fbc..4e41b05 100644
--- a/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
+++ b/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
@@ -36,6 +36,10 @@ public class HtmlElement implements HtmlNode {
         return tag.isCollapsible();
     }
 
+    public String getSeparator() {
+        return tag.getSeparator();
+    }
+
     @Override
     public void accept(Visitor visitor) {
         visitor.visit(this);