Do all HTML writing in HtmlWriter

1 files changed, 12 insertions, 11 deletions

diff --git a/src/main/java/org/zwobble/mammoth/html/HtmlElement.java b/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
index 8111507..637b009 100644
--- a/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
+++ b/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
@@ -14,19 +14,20 @@ public class HtmlElement implements HtmlNode {
         this.children = children;
     }
 
-    @Override
-    public void write(StringBuilder builder) {
-        builder.append("<").append(tagName);
-
-        HtmlWriter.generateAttributes(attributes, builder);
+    public String getTagName() {
+        return tagName;
+    }
 
-        builder.append(">");
+    public Map<String, String> getAttributes() {
+        return attributes;
+    }
 
-        children.forEach(node -> node.write(builder));
+    public List<HtmlNode> getChildren() {
+        return children;
+    }
 
-        builder
-            .append("</")
-            .append(tagName)
-            .append(">");
+    @Override
+    public void visit(Visitor visitor) {
+        visitor.visit(this);
     }
 }