Optionally skip children when looking for following siblings

Author: Joris-van-der-Wel

lib/SymbolTree.js

@@ -150,10 +150,11 @@ module.exports = class SymbolTree {
          * @param {Object} [treeRoot] If set, `treeRoot` must be an inclusive ancestor
          *        of the return value (or else null is returned). This check _assumes_
          *        that `root` is also an inclusive ancestor of the given `node`
+         * @param {Boolean} [skipChildren=false] If set, ignore the childen of `object`
          * @returns {?Object}
          */
-        following(object, treeRoot) {
-                if (this._node(object).first) {
+        following(object, treeRoot, skipChildren) {
+                if (!skipChildren && this._node(object).first) {
                         return this._node(object).first;
                 }
 

test/SymbolTree.js

@@ -579,3 +579,17 @@ test('following using a specified root', function(t) {
 
         t.end();
 });
+
+test('following with skipChildren', function(t) {
+        const tree = new SymbolTree();
+        const a = {};
+        const aa = {};
+        const b = {};
+
+        tree.insertLast(aa, a);
+        tree.insertAfter(b, a);
+
+        t.equal(b, tree.following(a, null, true));
+
+        t.end();
+});