From 8e959ae5b51e1b6ea28fc0a5d2db72ef3aa684bd Mon Sep 17 00:00:00 2001
From: Felipe Pena <felipensp@gmail.com>
Date: Sun, 16 Apr 2023 08:36:51 -0300
Subject: [PATCH] parser: fix wrong nodes generated for $something (#17969)

---
 vlib/v/checker/tests/invalid_comptime_test.out | 5 +++++
 vlib/v/checker/tests/invalid_comptime_test.vv  | 7 +++++++
 vlib/v/parser/parser.v                         | 4 +++-
 3 files changed, 15 insertions(+), 1 deletion(-)
 create mode 100644 vlib/v/checker/tests/invalid_comptime_test.out
 create mode 100644 vlib/v/checker/tests/invalid_comptime_test.vv

diff --git a/vlib/v/checker/tests/invalid_comptime_test.out b/vlib/v/checker/tests/invalid_comptime_test.out
new file mode 100644
index 0000000000..f6c5c9c714
--- /dev/null
+++ b/vlib/v/checker/tests/invalid_comptime_test.out
@@ -0,0 +1,5 @@
+vlib/v/checker/tests/invalid_comptime_test.vv:1:1: error: unexpected token `$`
+    1 | $fn func(a int) int {
+      | ^
+    2 |     return a + a
+    3 | }
diff --git a/vlib/v/checker/tests/invalid_comptime_test.vv b/vlib/v/checker/tests/invalid_comptime_test.vv
new file mode 100644
index 0000000000..8c5ca83632
--- /dev/null
+++ b/vlib/v/checker/tests/invalid_comptime_test.vv
@@ -0,0 +1,7 @@
+$fn func(a int) int {
+    return a + a
+}
+
+fn main() {
+    println(func(5))
+}
\ No newline at end of file
diff --git a/vlib/v/parser/parser.v b/vlib/v/parser/parser.v
index 6239f70a3a..0c7be27ae9 100644
--- a/vlib/v/parser/parser.v
+++ b/vlib/v/parser/parser.v
@@ -754,7 +754,7 @@ pub fn (mut p Parser) top_stmt() ast.Stmt {
 				if p.peek_tok.kind == .key_for {
 					comptime_for_stmt := p.comptime_for()
 					return p.other_stmts(comptime_for_stmt)
-				} else {
+				} else if p.peek_tok.kind == .key_if {
 					if_expr := p.if_expr(true)
 					cur_stmt := ast.ExprStmt{
 						expr: if_expr
@@ -765,6 +765,8 @@ pub fn (mut p Parser) top_stmt() ast.Stmt {
 					} else {
 						return p.other_stmts(cur_stmt)
 					}
+				} else {
+					return p.unexpected()
 				}
 			}
 			.hash {