Fix break in Enumerator

Now we support infinite enumerators
opal · Nov 8, 2013 · 69cdfa1 · 69cdfa1
1 parent 1585e3d
commit 69cdfa1
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/opal/core/enumerator.rb b/opal/core/enumerator.rb
@@ -9,13 +9,31 @@ def initialize(enumerator, block, to)
     end
 
     def yield(*values)
-      @to.call(*values)
+      %x{
+        if ($opal.$yieldX(#@to, values) === $breaker) {
+          throw $breaker;
+        }
+      }
+
+      self
     end
 
     alias << yield
 
     def call
-      @block.call(self)
+      %x{
+        try {
+          #@block(self)
+        }
+        catch (e) {
+          if (e === $breaker) {
+            return $breaker;
+          }
+          else {
+            throw e;
+          }
+        }
+      }
     end
   end