Store local variables when looping over bitsets

1 files changed, 33 insertions, 0 deletions

diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index 9f7018f..7170935 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -3550,6 +3550,39 @@ get_locals_for_range_stmt :: proc(
 					)
 				}
 			}
+		case SymbolBitSetValue:
+			if len(stmt.vals) >= 1 {
+				if ident, ok := unwrap_ident(stmt.vals[0]); ok {
+					store_local(
+						ast_context,
+						ident,
+						v.expr,
+						ident.pos.offset,
+						ident.name,
+						ast_context.non_mutable_only,
+						false,
+						true,
+						symbol.pkg,
+						false,
+					)
+				}
+			}
+			if len(stmt.vals) >= 2 {
+				if ident, ok := unwrap_ident(stmt.vals[1]); ok {
+					store_local(
+						ast_context,
+						ident,
+						make_int_ast(ast_context, ident.pos, ident.end),
+						ident.pos.offset,
+						ident.name,
+						ast_context.non_mutable_only,
+						false,
+						true,
+						symbol.pkg,
+						false,
+					)
+				}
+			}
 		}
 	}