Handle until

Author: kddnewton

lib/syntax_tree/visitor/compiler.rb

@@ -1997,6 +1997,24 @@ def visit_undef(node)
         end
       end
 
+      def visit_until(node)
+        jumps = []
+
+        jumps << builder.jump(-1)
+        builder.putnil
+        builder.pop
+        jumps << builder.jump(-1)
+
+        label = builder.label
+        visit(node.statements)
+        builder.pop
+        jumps.each { |jump| jump[1] = builder.label }
+
+        visit(node.predicate)
+        builder.branchunless(label)
+        builder.putnil if last_statement?
+      end
+
       def visit_var_field(node)
         case node.value
         when CVar, IVar

test/compiler_test.rb

@@ -260,6 +260,9 @@ class CompilerTest < Minitest::Test
       "if foo then bar elsif baz then qux end",
       "foo if bar",
       "foo while bar",
+      "while foo do bar end",
+      "foo until bar",
+      "until foo do bar end",
       "for i in [1, 2, 3] do i end",
       "foo ? bar : baz",
       # Constructed values