Allow HTML elements to have a choice of tag names

1 files changed, 8 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/mammoth/html/HtmlElement.java b/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
index cc81aa2..b6dab22 100644
--- a/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
+++ b/src/main/java/org/zwobble/mammoth/html/HtmlElement.java
@@ -4,20 +4,24 @@ import java.util.List;
 import java.util.Map;
 
 public class HtmlElement implements HtmlNode {
-    private final String tagName;
+    private final List<String> tagNames;
     private final Map<String, String> attributes;
     private final List<HtmlNode> children;
     private final boolean collapsible;
 
-    public HtmlElement(String tagName, Map<String, String> attributes, List<HtmlNode> children, boolean collapsible) {
-        this.tagName = tagName;
+    public HtmlElement(List<String> tagNames, Map<String, String> attributes, List<HtmlNode> children, boolean collapsible) {
+        this.tagNames = tagNames;
         this.attributes = attributes;
         this.children = children;
         this.collapsible = collapsible;
     }
 
+    public List<String> getTagNames() {
+        return tagNames;
+    }
+
     public String getTagName() {
-        return tagName;
+        return tagNames.get(0);
     }
 
     public Map<String, String> getAttributes() {