Generate stack IR for effect field access

2 files changed, 31 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
index 4ca4de8e..8110e6da 100644
--- a/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
+++ b/src/main/java/org/shedlang/compiler/stackir/StackCompiler.java
@@ -190,6 +190,14 @@ public class StackCompiler {
             case StructType structType ->
                 new FieldLoadInstruction.Receiver.Struct(structType);
 
+            case TypeLevelValueType typeLevelValueType -> {
+                if (typeLevelValueType.value() instanceof UserDefinedEffect effect) {
+                    yield new FieldLoadInstruction.Receiver.Effect(effect);
+                } else {
+                    throw new RuntimeException("not a type with fields");
+                }
+            }
+
             default ->
                 throw new RuntimeException("not a type with fields");
         };
diff --git a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFieldAccessTests.java b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFieldAccessTests.java
index fee566cb..23d87b61 100644
--- a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFieldAccessTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFieldAccessTests.java
@@ -3,6 +3,7 @@ package org.shedlang.compiler.stackir;
 import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.typed.Typed;
 import org.shedlang.compiler.stackir.instructions.FieldLoadInstruction;
+import org.shedlang.compiler.types.Effects;
 import org.shedlang.compiler.types.NamespaceName;
 import org.shedlang.compiler.types.Types;
 
@@ -52,4 +53,26 @@ public class StackCompilerFieldAccessTests {
             isFieldLoadInstruction(structType, "name")
         ));
     }
+
+    @Test
+    public void userDefinedEffectFieldAccessIsCompiledToFieldLoad() {
+        var effect = Effects.userDefined(NamespaceName.of("Example", "Project"), "State");
+        var node = Typed.fieldAccess(
+            Typed.reference("State", Types.effectType(effect)),
+            "read",
+            Types.STRING
+        );
+        var context = StackCompiler.Context.EMPTY
+            .addShedVariable("State");
+
+        var result = StackCompiler.compileExpression(node, context);
+
+        assertThat(result.instructions(), isSequence(
+            isLocalLoadInstruction("State__0"),
+            isFieldLoadInstruction(
+                new FieldLoadInstruction.Receiver.Effect(effect),
+                "read"
+            )
+        ));
+    }
 }