[#183] Add Match.attrNames(): Set<String>

jOOQ · Feb 17, 2023 · d77fec5 · d77fec5
1 parent 3ab02e9
commit d77fec5
Show file tree

Hide file tree

Showing 6 changed files with 214 additions and 138 deletions.
diff --git a/jOOX-java-8/src/main/java/org/joox/Impl.java b/jOOX-java-8/src/main/java/org/joox/Impl.java
@@ -69,9 +69,11 @@
 import javax.xml.xpath.XPathFactory;
 import javax.xml.xpath.XPathVariableResolver;
 
+import org.w3c.dom.Attr;
 import org.w3c.dom.Document;
 import org.w3c.dom.DocumentFragment;
 import org.w3c.dom.Element;
+import org.w3c.dom.NamedNodeMap;
 import org.w3c.dom.Node;
 import org.w3c.dom.NodeList;
 
@@ -1197,6 +1199,23 @@ public final Impl prepend(Element... content) {
         return this;
     }
 
+    @Override
+    public final Set<String> attrNames() {
+        Set<String> result = new LinkedHashSet<>();
+
+        for (Element e : this) {
+            NamedNodeMap m = e.getAttributes();
+            int length = m.getLength();
+
+            for (int i = 0; i < length; i++) {
+                Attr a = (Attr) m.item(i);
+                result.add(a.getName());
+            }
+        }
+
+        return result;
+    }
+
     @Override
     public final String attr(String name) {
         if (size() > 0)

diff --git a/jOOX-java-8/src/main/java/org/joox/Match.java b/jOOX-java-8/src/main/java/org/joox/Match.java
@@ -23,6 +23,7 @@
 import java.util.Comparator;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 
 import javax.xml.transform.Result;
 import javax.xml.transform.Source;
@@ -1752,6 +1753,11 @@ public interface Match extends Iterable<Element> {
     // Manipulation of attributes
     // ---------------------------------------------------------------------
 
+    /**
+     * Get the set of available attribute names in the set of matched elements.
+     */
+    Set<String> attrNames();
+
     /**
      * Get an attribute from the first element in the set of matched elements,
      * or <code>null</code> if the first element does not have that attribute.