From e5f70950620f2178fefe91f33a3401f831f32e7d Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 14 Feb 2016 19:31:02 +0000
Subject: Strip empty nodes from output

---
 src/main/java/org/zwobble/mammoth/html/Html.java | 43 ++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

(limited to 'src/main/java/org/zwobble/mammoth/html/Html.java')

diff --git a/src/main/java/org/zwobble/mammoth/html/Html.java b/src/main/java/org/zwobble/mammoth/html/Html.java
index 0a32e58..7e2a82e 100644
--- a/src/main/java/org/zwobble/mammoth/html/Html.java
+++ b/src/main/java/org/zwobble/mammoth/html/Html.java
@@ -1,5 +1,7 @@
 package org.zwobble.mammoth.html;
 
+import org.zwobble.mammoth.util.MammothLists;
+
 import java.util.List;
 import java.util.Map;
 
@@ -50,4 +52,45 @@ public class Html {
     public static HtmlNode selfClosingElement(String tagName, Map<String, String> attributes) {
         return new HtmlSelfClosingElement(tagName, attributes);
     }
+
+    public static List<HtmlNode> stripEmpty(List<HtmlNode> nodes) {
+        return MammothLists.eagerFlatMap(nodes, node -> stripEmpty(node));
+    }
+
+    private static List<HtmlNode> stripEmpty(HtmlNode node) {
+        return node.accept(new HtmlNode.Mapper<List<HtmlNode>>() {
+            @Override
+            public List<HtmlNode> visit(HtmlElement element) {
+                List<HtmlNode> children = stripEmpty(element.getChildren());
+                if (children.isEmpty()) {
+                    return list();
+                } else {
+                    return list(new HtmlElement(
+                        element.getTagName(),
+                        element.getAttributes(),
+                        children,
+                        element.isCollapsible()));
+                }
+            }
+
+            @Override
+            public List<HtmlNode> visit(HtmlSelfClosingElement element) {
+                return list(element);
+            }
+
+            @Override
+            public List<HtmlNode> visit(HtmlTextNode node) {
+                if (node.getValue().isEmpty()) {
+                    return list();
+                } else {
+                    return list(node);
+                }
+            }
+
+            @Override
+            public List<HtmlNode> visit(HtmlForceWrite forceWrite) {
+                return list(forceWrite);
+            }
+        });
+    }
 }
-- 
cgit v1.2.3