Push enum variant selection into ArbitraryValueAnalysis

2 files changed, 5 insertions, 5 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/analysis/ArbitraryValueAnalysis.java b/src/main/java/org/zwobble/hobgoblin/compiler/analysis/ArbitraryValueAnalysis.java
index b803b3d..03cb732 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/analysis/ArbitraryValueAnalysis.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/analysis/ArbitraryValueAnalysis.java
@@ -39,7 +39,7 @@ public class ArbitraryValueAnalysis {
             }
         }
 
-        record EnumFirstVariant(EnumType type) implements ArbitraryValue {
+        record EnumVariantValue(EnumType type, EnumVariant variant) implements ArbitraryValue {
             @Override
             public void collectTypeParams(Set<TypeParam> typeParams) {
             }
@@ -104,7 +104,7 @@ public class ArbitraryValueAnalysis {
             }
 
             case EnumType enumType ->
-                new ArbitraryValue.EnumFirstVariant(enumType);
+                new ArbitraryValue.EnumVariantValue(enumType, typesInfo.enumVariants(enumType).getFirst());
 
             case SimpleNativeType simpleNativeType ->
                 new ArbitraryValue.NativeFunctionCall(simpleNativeType);
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
index 1fe01c9..8e85e91 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
@@ -671,12 +671,12 @@ public class JavaTypesGenerator implements Generator {
                 );
             }
 
-            case ArbitraryValueAnalysis.ArbitraryValue.EnumFirstVariant enumFirstVariant -> {
-                var enumType = enumFirstVariant.type();
+            case ArbitraryValueAnalysis.ArbitraryValue.EnumVariantValue enumVariant -> {
+                var enumType = enumVariant.type();
 
                 yield new JavaStaticFieldAccess(
                     generateTypeRef(enumType, context),
-                    generateEnumConstantName(context.enumVariants(enumType).getFirst().name())
+                    generateEnumConstantName(enumVariant.variant().name())
                 );
             }