Update ContentNode.java

cyanpotion · Dec 16, 2018 · 5b60019 · 5b60019
1 parent edf0dae
commit 5b60019
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/main/java/com/xenoamess/x8l/ContentNode.java b/src/main/java/com/xenoamess/x8l/ContentNode.java
@@ -216,5 +216,29 @@ public List<CommentNode> getCommentNodesFromChildren(int maxSize) {
         return res;
     }
 
+    public String getName() {
+        if (!this.attributesKeyList.isEmpty()) {
+            return this.attributesKeyList.get(0);
+        } else {
+            return "";
+        }
+    }
+
+    public List<ContentNode> getContentNodesFromChildrenThatNameIs(String name) {
+        return this.getContentNodesFromChildrenThatNameIs(name, 0);
+    }
+
+    public List<ContentNode> getContentNodesFromChildrenThatNameIs(String name, int maxSize) {
+        List<ContentNode> res = new ArrayList<ContentNode>();
+        for (TreeNode au : this.children) {
+            if (au instanceof CommentNode && ((ContentNode) au).getName().equals(name)) {
+                res.add((ContentNode) au);
+                if (res.size() == maxSize) {
+                    return res;
+                }
+            }
+        }
+        return res;
+    }
 
 }