Defer generation of child nodes until necessary

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/HtmlPathElement.java b/src/main/java/org/zwobble/mammoth/internal/styles/HtmlPathElement.java
index 08221eb..4c4ff98 100644
--- a/src/main/java/org/zwobble/mammoth/internal/styles/HtmlPathElement.java
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/HtmlPathElement.java
@@ -5,6 +5,7 @@ import org.zwobble.mammoth.internal.html.HtmlNode;
 
 import java.util.List;
 import java.util.Map;
+import java.util.function.Supplier;
 
 import static org.zwobble.mammoth.internal.util.Lists.list;
 import static org.zwobble.mammoth.internal.util.Maps.map;
@@ -32,7 +33,11 @@ public class HtmlPathElement {
         this.isCollapsible = isCollapsible;
     }
 
-    public List<HtmlNode> wrap(List<HtmlNode> nodes) {
+    public Supplier<List<HtmlNode>> wrap(Supplier<List<HtmlNode>> generateNodes) {
+        return () -> wrapNodes(generateNodes.get());
+    }
+
+    private List<HtmlNode> wrapNodes(List<HtmlNode> nodes) {
         return list(new HtmlElement(tagNames, attributes, nodes, isCollapsible));
     }
 }