Add accessor for effect tags

1 files changed, 14 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 155e391f..85efa16b 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
@@ -87,6 +87,16 @@ public record PatioCodeGenContext(
                 return tag;
             }
         }
+
+        public int effectTag(UserDefinedEffect effect) {
+            var tag = this.effectTags.get(effect);
+            if (tag == null) {
+                // TODO: throw better error
+                throw new RuntimeException("No tag for: " + effect);
+            } else {
+                return tag;
+            }
+        }
     }
 
     public record CompilationUnitContext(
@@ -297,6 +307,10 @@ public record PatioCodeGenContext(
         return globalContext.structTag(type);
     }
 
+    public int effectTag(UserDefinedEffect effect) {
+        return globalContext.effectTag(effect);
+    }
+
     public PatioCodeGenResult<String> allocateCompilationUnitElementName(
         PatioCodeGenCompilationUnitElementSpace space,
         String name