Move isVoidElement to instance method of HtmlElement

1 files changed, 13 insertions, 0 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 4e41b05..999945e 100644
--- a/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
+++ b/src/main/java/org/zwobble/mammoth/internal/html/HtmlElement.java
@@ -2,6 +2,9 @@ package org.zwobble.mammoth.internal.html;
 
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
+
+import static org.zwobble.mammoth.internal.util.Sets.set;
 
 public class HtmlElement implements HtmlNode {
     private final HtmlTag tag;
@@ -40,6 +43,16 @@ public class HtmlElement implements HtmlNode {
         return tag.getSeparator();
     }
 
+    public boolean isVoid() {
+        return children.isEmpty() && isVoidTag(getTagName());
+    }
+
+    private static final Set<String> VOID_TAG_NAMES = set("img", "br", "hr");
+
+    private static boolean isVoidTag(String tagName) {
+        return VOID_TAG_NAMES.contains(tagName);
+    }
+
     @Override
     public void accept(Visitor visitor) {
         visitor.visit(this);