parser: fix 'val in array' as condition in for stmt (fix #14440) (#14451)

master
yuyi 2022-05-19 01:38:53 +08:00 committed by GitHub
parent 805a7d9713
commit b482c0512b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 1 deletions

View File

@ -82,7 +82,8 @@ fn (mut p Parser) for_stmt() ast.Stmt {
}
p.close_scope()
return for_c_stmt
} else if p.peek_tok.kind in [.key_in, .comma]
} else if (p.peek_tok.kind in [.key_in, .comma] && !(p.tok.kind == .name
&& p.peek_tok.kind == .key_in && p.scope.known_var(p.tok.lit)))
|| (p.tok.kind == .key_mut && p.peek_token(2).kind in [.key_in, .comma]) {
// `for i in vals`, `for i in start .. end`, `for mut user in users`, `for i, mut user in users`
mut val_is_mut := p.tok.kind == .key_mut

View File

@ -0,0 +1,10 @@
type TokenValue = rune | u64
fn test_for_cond() {
val := `+`
for val in [TokenValue(`+`), TokenValue(`-`)] {
println('ok')
break
}
assert true
}