1

cyanpotion · Dec 16, 2018 · 83b182a · 83b182a
1 parent a8b297e
commit 83b182a
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -5,5 +5,5 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.xenoamess</groupId>
     <artifactId>x8l</artifactId>
-    <version>0.27</version>
+    <version>0.28</version>
 </project>
diff --git a/src/main/java/com/xenoamess/x8l/ContentNode.java b/src/main/java/com/xenoamess/x8l/ContentNode.java
@@ -6,6 +6,7 @@
 import java.io.Writer;
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 public class ContentNode extends TreeNode {
@@ -24,8 +25,10 @@ public void addAttribute(String key, String value) {
         if (value == null) {
             value = "";
         }
+        if (!this.attributes.containsKey(key)) {
+            attributesKeyList.add(key);
+        }
         attributes.put(key, value);
-        attributesKeyList.add(key);
     }
 
 
@@ -114,4 +117,58 @@ public void output(Writer writer) {
             e.printStackTrace();
         }
     }
+
+    public List<TextNode> getTextNodesFromChildren() {
+        return this.getTextNodesFromChildren(0);
+    }
+
+    public List<TextNode> getTextNodesFromChildren(int maxSize) {
+        List<TextNode> res = new ArrayList<TextNode>();
+        for (TreeNode au : this.children) {
+            if (au instanceof TextNode) {
+                res.add((TextNode) au);
+                if (res.size() == maxSize) {
+                    return res;
+                }
+            }
+        }
+        return res;
+    }
+
+    public List<ContentNode> getContentNodesFromChildren() {
+        return this.getContentNodesFromChildren(0);
+    }
+
+    public List<ContentNode> getContentNodesFromChildren(int maxSize) {
+        List<ContentNode> res = new ArrayList<ContentNode>();
+        for (TreeNode au : this.children) {
+            if (au instanceof ContentNode) {
+                res.add((ContentNode) au);
+                if (res.size() == maxSize) {
+                    return res;
+                }
+            }
+        }
+        return res;
+    }
+
+
+    public List<CommentNode> getCommentNodesFromChildren() {
+        return this.getCommentNodesFromChildren(0);
+    }
+
+    public List<CommentNode> getCommentNodesFromChildren(int maxSize) {
+        List<CommentNode> res = new ArrayList<CommentNode>();
+        for (TreeNode au : this.children) {
+            if (au instanceof CommentNode) {
+                res.add((CommentNode) au);
+                if (res.size() == maxSize) {
+                    return res;
+                }
+            }
+        }
+        return res;
+    }
+
+
 }