fixed #195

Author: satyr

lib/lexer.js

@@ -137,10 +137,11 @@ exports.doID = function(code, index){
     return this.token('LITERAL', id, true).length;
   case 'for':
     this.seenFor = true;
-    // fallthrough
-  case 'then':
     this.wantBy = false;
     break;
+  case 'then':
+    this.seenFor = this.wantBy = false;
+    break;
   case 'catch':
   case 'function':
     id = '';
@@ -461,7 +462,7 @@ exports.doLine = function(code, index){
       this.newline();
     }
   }
-  this.wantBy = false;
+  this.seenFor = this.wantBy = false;
   return length;
 };
 exports.doSpace = function(code, lastIndex){
@@ -684,6 +685,7 @@ exports.doLiteral = function(code, index){
     break;
   case '=>':
     this.unline();
+    this.seenFor = false;
     if (that = this.doInlinedent(code, index + 2)) {
       return 1 + that;
     }

src/lexer.co

@@ -105,8 +105,8 @@ exports import
     case \return \throw                                then tag = \HURL
     case \break  \continue                             then tag = \JUMP
     case \this \eval \super then return @token(\LITERAL id, true)length
-    case \for  then @seenFor = true; fallthrough
-    case \then then @wantBy  = false
+    case \for  then @seenFor = true; @wantBy = false
+    case \then then @seenFor = @wantBy = false
     case \catch \function then id = ''
     case \in \of
       if @seenFor
@@ -330,7 +330,7 @@ exports import
                    IN OF TO BY FROM EXTENDS IMPLEMENTS ]>
         return length
       if delta then @indent delta else @newline!
-    @wantBy = false
+    @seenFor = @wantBy = false
     length
 
   # Consumes non-newline whitespaces and/or a line comment.
@@ -440,6 +440,7 @@ exports import
     case \..     then @adi!; tag = \ID; val = up || \constructor
     case \=>
       @unline!
+      @seenFor = false
       return 1 + that if @doInlinedent code, index+2
       tag = \THEN
     default if \< is val.charAt 0

test/loop.co

@@ -210,6 +210,14 @@ continue for, o.count!_ in [1 2]
 eq o.i, 4
 
 
+# [#195](https://github.com/satyr/coco/issues/195)
+for [0]
+  ok 0 in {0}
+  for [1] then ok 1 of [1]
+  for [2] =>   ok 2 of [2]
+  ok 3 not in []
+
+
 ### Line folding before/after `for` prepositions
 for x
 in   {2}