fix(parser): accept bare NEXT and read its identifier only from the same line

Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>

1 files changed, 9 insertions, 10 deletions

diff --git a/lib/Parser/ParserControlFlow.cc b/lib/Parser/ParserControlFlow.cc
index feedf44..7caf555 100644
--- a/lib/Parser/ParserControlFlow.cc
+++ b/lib/Parser/ParserControlFlow.cc
@@ -124,18 +124,17 @@ std::unique_ptr<StmtAST> Parser::ParseForStmt() {
         return nullptr;
     }
     getNextToken();
-    
-    if (CurTok != tok_identifier) {
-        cps::sourceError(Lex.getLine(), "expected identifier after NEXT (e.g., NEXT %s)", VarName.c_str());
-        return nullptr;
-    }
 
-    if (!sameIdentifier(Lex.IdentifierStr, VarName)) {
-        cps::sourceError(Lex.getLine(), "NEXT identifier '%s' does not match FOR variable '%s'",
-                Lex.IdentifierStr.c_str(), VarName.c_str());
-        return nullptr;
+    // The identifier after NEXT is optional; one on the next line starts a
+    // new statement.
+    if (CurTok == tok_identifier && !StartsLine) {
+        if (!sameIdentifier(Lex.IdentifierStr, VarName)) {
+            cps::sourceError(Lex.getLine(), "NEXT identifier '%s' does not match FOR variable '%s'",
+                    Lex.IdentifierStr.c_str(), VarName.c_str());
+            return nullptr;
+        }
+        getNextToken();
     }
-    getNextToken();
 
     return std::make_unique<ForStmtAST>(VarName, std::move(Start), std::move(End), std::move(Step), std::move(Body));
 }