Convert all children to an array

Author: Joris-van-der-Wel

lib/SymbolTree.js

@@ -227,6 +227,29 @@ class SymbolTree {
                 return null;
         }
 
+        /**
+         * Append all children of the given object to an array.
+         *
+         * `O(n)`
+         * @param {Object} object
+         * @param {Object[]} [array=[]]
+         * @return {Object[]}
+         */
+        childrenToArray(object, array) {
+                if (!array) {
+                        array = [];
+                }
+
+                object = this._node(object).first;
+
+                while (object) {
+                        array.push(object);
+                        object = this._node(object).next;
+                }
+
+                return array;
+        }
+
         /**
          * Remove the object from this tree.
          * Has no effect if already removed.
@@ -414,4 +437,4 @@ class SymbolTree {
         }
 }
 
-module.exports = SymbolTree;
+module.exports = SymbolTree;

test/SymbolTree.js

@@ -613,3 +613,27 @@ test('following with skipChildren', function(t) {
 
         t.end();
 });
+
+test('childrenToArray', function(t) {
+        const tree = new SymbolTree();
+        const a = {};
+        const aa = {};
+        const ab = {};
+        const aba = {};
+        const ac = {};
+        const b = {};
+
+        tree.insertLast(aa, a);
+        tree.insertLast(ab, a);
+        tree.insertLast(aba, ab);
+        tree.insertLast(ac, a);
+        tree.insertAfter(b, a);
+
+        t.deepEqual([aa, ab, ac], tree.childrenToArray(a));
+
+        const arr = ['a', 5];
+        tree.childrenToArray(a, arr);
+        t.deepEqual(['a', 5, aa, ab, ac], arr);
+
+        t.end();
+});