Introduce HtmlTag class

1 files changed, 11 insertions, 11 deletions

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 e855f26..9999fbc 100644
--- a/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
+++ b/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
@@ -4,28 +4,28 @@ import java.util.List;
 import java.util.Map;
 
 public class HtmlElement implements HtmlNode {
-    private final List<String> tagNames;
-    private final Map<String, String> attributes;
+    private final HtmlTag tag;
     private final List<HtmlNode> children;
-    private final boolean collapsible;
 
-    public HtmlElement(List<String> tagNames, Map<String, String> attributes, List<HtmlNode> children, boolean collapsible) {
-        this.tagNames = tagNames;
-        this.attributes = attributes;
+    public HtmlElement(HtmlTag tag, List<HtmlNode> children) {
+        this.tag = tag;
         this.children = children;
-        this.collapsible = collapsible;
+    }
+
+    public HtmlTag getTag() {
+        return tag;
     }
 
     public List<String> getTagNames() {
-        return tagNames;
+        return tag.getTagNames();
     }
 
     public String getTagName() {
-        return tagNames.get(0);
+        return getTagNames().get(0);
     }
 
     public Map<String, String> getAttributes() {
-        return attributes;
+        return tag.getAttributes();
     }
 
     public List<HtmlNode> getChildren() {
@@ -33,7 +33,7 @@ public class HtmlElement implements HtmlNode {
     }
 
     public boolean isCollapsible() {
-        return collapsible;
+        return tag.isCollapsible();
     }
 
     @Override