Merge pull request #6259 from odin-lang/bill/range-init

`for init; x in y {}` style loops (proof of concept)

1 files changed, 7 insertions, 1 deletions

diff --git a/src/check_stmt.cpp b/src/check_stmt.cpp
index 8f17e66c4..bfa68ca78 100644
--- a/src/check_stmt.cpp
+++ b/src/check_stmt.cpp
@@ -1705,11 +1705,17 @@ gb_internal void check_range_stmt(CheckerContext *ctx, Ast *node, u32 mod_flags)
 
 	TEMPORARY_ALLOCATOR_GUARD();
 
-	u32 new_flags = mod_flags | Stmt_BreakAllowed | Stmt_ContinueAllowed;
 
 	check_open_scope(ctx, node);
 	check_label(ctx, rs->label, node);
 
+	Operand init = {};
+	if (rs->init != nullptr) {
+		check_stmt(ctx, rs->init, mod_flags);
+	}
+
+	u32 new_flags = mod_flags | Stmt_BreakAllowed | Stmt_ContinueAllowed;
+
 	auto vals = array_make<Type *>(temporary_allocator(), 0, 2);
 	auto entities = array_make<Entity *>(temporary_allocator(), 0, 2);
 	bool is_map = false;