checker: fix comptime ident checking on defer stmt (fix #20719) (#20723)

vlang · Feb 4, 2024 · ec7e058 · ec7e058
1 parent abd2b64
commit ec7e058
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/vlib/v/checker/checker.v b/vlib/v/checker/checker.v
@@ -2036,7 +2036,8 @@ fn (mut c Checker) stmt(mut node ast.Stmt) {
 			for i, ident in node.defer_vars {
 				mut id := ident
 				if mut id.info is ast.IdentVar {
-					if id.comptime && id.name in ast.valid_comptime_not_user_defined {
+					if id.comptime && (id.tok_kind == .question
+						|| id.name in ast.valid_comptime_not_user_defined) {
 						node.defer_vars[i] = ast.Ident{
 							scope: unsafe { nil }
 							name: ''

diff --git a/vlib/v/tests/defer_if_comptime_test.v b/vlib/v/tests/defer_if_comptime_test.v
@@ -0,0 +1,9 @@
+fn test_main() {
+	defer {
+		$if foo ? {
+			assert false
+		} $else {
+			assert true
+		}
+	}
+}