Props Iterator

brainkim · Oct 12, 2020 · 79e1298 · 79e1298
1 parent 274b914
commit 79e1298
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/crank.js b/crank.js
@@ -292,12 +292,24 @@ class Context {
 
     // flags
     this._isUpdating = false;
+    this._isIterating = false;
     this._isDone = false;
   }
 
   refresh() {
     return stepCtx(this);
   }
+
+  *[Symbol.iterator]() {
+    while (!this._isDone) {
+      if (this._isIterating) {
+        throw new Error("Context iterated twice without a yield");
+      }
+
+      this._isIterating = true;
+      yield this._el.props;
+    }
+  }
 }
 
 function stepCtx(ctx) {
@@ -313,6 +325,7 @@ function stepCtx(ctx) {
   }
 
   const iteration = ctx._iter.next();
+  ctx._isIterating = false;
   if (iteration.done) {
     ctx._isDone = true;
   }