Add accessor for user defined effect infos

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
index 85efa16b..fd45f29d 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
@@ -88,6 +88,12 @@ public record PatioCodeGenContext(
             }
         }
 
+        public UserDefinedEffectInfo userDefinedEffectInfo(UserDefinedEffect effect) {
+            return typeInfos.userDefinedEffectInfo(effect)
+                // TODO: throw better error
+                .orElseThrow(() -> new RuntimeException("No effect info for: " + effect));
+        }
+
         public int effectTag(UserDefinedEffect effect) {
             var tag = this.effectTags.get(effect);
             if (tag == null) {
@@ -307,6 +313,10 @@ public record PatioCodeGenContext(
         return globalContext.structTag(type);
     }
 
+    public UserDefinedEffectInfo userDefinedEffectInfo(UserDefinedEffect effect) {
+        return this.globalContext.userDefinedEffectInfo(effect);
+    }
+
     public int effectTag(UserDefinedEffect effect) {
         return globalContext.effectTag(effect);
     }