fix for(key in obj)

[org.ibex.core.git] / src / org / ibex / js / Parser.java

diff --git a/src/org/ibex/js/Parser.java b/src/org/ibex/js/Parser.java
index ded2556..a4621db 100644 (file)
--- a/src/org/ibex/js/Parser.java
+++ b/src/org/ibex/js/Parser.java
@@ -835,37 +835,58 @@ class Parser extends Lexer implements ByteCodes {
             pushBackToken(tok, varName);
             
             if (forIn) {
-                b.add(parserLine, NEWSCOPE);                             // for-loops always create new scopes
-                b.add(parserLine, LITERAL, varName);                     // declare the new variable
-                b.add(parserLine, DECLARE);
-                    
-                b.add(parserLine, LOOP);                                 // we actually only add this to ensure that BREAK works
-                b.add(parserLine, POP);                                  // discard the first-iteration indicator
-                int size = b.size;
+                b.add(parserLine, NEWSCOPE);
+                b.add(parserLine, DECLARE, varName);
+                b.add(parserLine, POP);
+                
                 consume(NAME);
                 consume(IN);
-                startExpr(b, -1);
-                b.add(parserLine, PUSHKEYS);                             // push the keys as an array; check the length
-                b.add(parserLine, LITERAL, "length");
-                b.add(parserLine, GET);
+                startExpr(b,-1);
                 consume(RP);
-                    
-                b.add(parserLine, LITERAL, JS.N(1));              // decrement the length
+                b.add(parserLine, PUSHKEYS);
+                b.add(parserLine, DUP); 
+                b.add(parserLine, LITERAL, "length"); 
+                b.add(parserLine, GET);
+                
+                int size = b.size;
+                b.add(parserLine, LOOP);
+                b.add(parserLine, POP);
+                b.add(parserLine, SWAP, JS.N(3));
+                
+                b.add(parserLine, LITERAL, JS.N(1));
                 b.add(parserLine, SUB);
                 b.add(parserLine, DUP);
-                b.add(parserLine, LITERAL, JS.ZERO);              // see if we've exhausted all the elements
+                b.add(parserLine, LITERAL, JS.ZERO);
                 b.add(parserLine, LT);
-                b.add(parserLine, JF, JS.N(2));
-                b.add(parserLine, BREAK);                                // if we have, then BREAK
-                b.add(parserLine, GET_PRESERVE);                         // get the key out of the keys array
+                
+                b.add(parserLine, JF, JS.N(5));
+                b.add(parserLine, SWAP, JS.N(3));
+                b.add(parserLine, SWAP, JS.N(3));
+                b.add(parserLine, SWAP, JS.N(3));
+                b.add(parserLine, BREAK);
+                
+                b.add(parserLine, GET_PRESERVE);
+                b.add(parserLine, TOPSCOPE);
+                b.add(parserLine, SWAP);
                 b.add(parserLine, LITERAL, varName);
-                b.add(parserLine, PUT);                                  // write it to this[varName]                          
-                parseStatement(b, null);                                 // do some stuff
-                b.add(parserLine, CONTINUE);                             // continue if we fall out the bottom
-
-                b.set(size - 1, JS.N(b.size - size + 1));       // BREAK to here
-                b.add(parserLine, OLDSCOPE);                             // restore the scope
-                    
+                b.add(parserLine, SWAP);
+                b.add(parserLine, PUT);
+                b.add(parserLine, POP);
+                b.add(parserLine, POP);
+                b.add(parserLine, SWAP, JS.N(3));
+                b.add(parserLine, SWAP, JS.N(3));
+                b.add(parserLine, SWAP, JS.N(3));
+                
+                parseStatement(b, null);
+                
+                b.add(parserLine, CONTINUE);
+                b.set(size, JS.N(b.size - size));
+                
+                b.add(parserLine, POP); // N
+                b.add(parserLine, POP); // KEYS
+                b.add(parserLine, POP); // OBJ
+                b.add(parserLine, OLDSCOPE);
+                                          
             } else {
                 if (hadVar) pushBackToken(VAR, null);                    // yeah, this actually matters
                 b.add(parserLine, NEWSCOPE);                             // grab a fresh scope