]> git.r.bdr.sh - rbdr/serpentity/commitdiff
Yields nodeCollection instead of nodes array
authorBen Beltran <redacted>
Sat, 21 Apr 2018 08:11:47 +0000 (03:11 -0500)
committerBen Beltran <redacted>
Sat, 21 Apr 2018 08:11:47 +0000 (03:11 -0500)
lib/serpentity.js
lib/serpentity/node_collection.js

index 0d932ac9b8bec2a8b7f419581849db08d8290039..60fb23cb6956cb4922239a5155e806d00586e476 100644 (file)
@@ -223,7 +223,7 @@ const Serpentity = class Serpentity {
       nodeCollection.add(entity);
     }
 
       nodeCollection.add(entity);
     }
 
-    return nodeCollection.nodes;
+    return nodeCollection;
   }
 
   /*
   }
 
   /*
index 73998b123500f36a6a3b828b672d10720235bfe8..7c2c627e13424acf4ca880f6cf90f1be0e964ce5 100644 (file)
@@ -93,4 +93,13 @@ const NodeCollection = class NodeCollection extends Events {
   }
 };
 
   }
 };
 
+
+/*
+ * Make the node collection iterable without returning the array directly
+ */
+NodeCollection.prototype[Symbol.iterator] = function * () {
+
+  yield* this.nodes;
+};
+
 module.exports = NodeCollection;
 module.exports = NodeCollection;