Fix enum lookup through shadowed envs

Author: mgomes

vibes/enum_test.go

@@ -173,6 +173,32 @@ end
 	}
 }
 
+func TestLookupEnumInEnvSkipsNonEnumShadowBindings(t *testing.T) {
+	enumDef, err := compileEnumDef(&EnumStmt{
+		Name: "Status",
+		Members: []EnumMemberStmt{
+			{Name: "Draft"},
+		},
+	})
+	if err != nil {
+		t.Fatalf("compile enum: %v", err)
+	}
+
+	root := newEnv(nil)
+	root.Define("Status", NewEnum(enumDef))
+
+	shadow := newEnv(root)
+	shadow.Define("Status", NewString("shadow"))
+
+	got, ok := lookupEnumInEnv(shadow, "Status")
+	if !ok {
+		t.Fatalf("expected lookup to resolve parent enum")
+	}
+	if got != enumDef {
+		t.Fatalf("expected parent enum def, got %#v", got)
+	}
+}
+
 func TestEnumModuleExportsAndTypedCalls(t *testing.T) {
 	engine := moduleTestEngine(t)
 	script := compileScriptWithEngine(t, engine, `def run()

vibes/execution_types_normalize.go

@@ -273,14 +273,17 @@ func resolveEnumType(ty *TypeExpr, ctx typeContext) (*EnumDef, error) {
 }
 
 func lookupEnumInEnv(env *Env, name string) (*EnumDef, bool) {
-	if env == nil {
-		return nil, false
-	}
-	val, ok := env.Get(name)
-	if !ok || val.Kind() != KindEnum {
-		return nil, false
+	for scope := env; scope != nil; scope = scope.parent {
+		val, ok := scope.values[name]
+		if !ok {
+			continue
+		}
+		if val.Kind() != KindEnum {
+			continue
+		}
+		return val.Enum(), true
 	}
-	return val.Enum(), true
+	return nil, false
 }
 
 func errorAsTypeMismatch(err error, target **typeMismatchError) bool {