Implement basic effect handling in the runtime

1 files changed, 6 insertions, 3 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenEffects.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenEffects.java
index ad5d64de..dc97456e 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenEffects.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenEffects.java
@@ -42,7 +42,8 @@ public class PatioCodeGenEffects {
     private static final PatioFunctionDeclaration PUSH_HANDLER_DECLARATION = new PatioFunctionDeclaration(
         "shed__runtime__effects_push_handler",
         List.of(
-            new PatioLocal.Param("effectTag", EFFECT_TAG_TYPE)
+            new PatioLocal.Param("effectTag", EFFECT_TAG_TYPE),
+            new PatioLocal.Param("methodCount", PatioTypes.INT)
         ),
         EFFECT_HANDLER_TYPE,
         Sources.RUNTIME
@@ -229,6 +230,8 @@ public class PatioCodeGenEffects {
         Source source,
         PatioCodeGenContext context
     ) {
+        var effectInfo = context.userDefinedEffectInfo(effect);
+
         context = context
             .ensureCompilationUnitElement(PUSH_HANDLER_DECLARATION)
             .ensureCompilationUnitElement(SET_HANDLER_METHOD_DECLARATION)
@@ -245,12 +248,12 @@ public class PatioCodeGenEffects {
             Optional.of(effectHandler),
             PUSH_HANDLER_DECLARATION.name(),
             List.of(
-                new PatioInt(context.effectTag(effect))
+                new PatioInt(context.effectTag(effect)),
+                new PatioInt(effectInfo.methodCount())
             ),
             source
         ));
 
-        var effectInfo = context.userDefinedEffectInfo(effect);
         for (var handlerMethodInfo : effectInfo.methods()) {
             context = context.addInstruction(new PatioCall(
                 Optional.empty(),