Handle params in method references

1 files changed, 49 insertions, 4 deletions

diff --git a/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java b/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
index 1fe776d4..d6626b37 100644
--- a/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/shid/ShidCodeGen.java
@@ -13,6 +13,7 @@ import org.shedlang.compiler.util.P;
 import java.text.MessageFormat;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Stream;
 
 public class ShidCodeGen {
     private ShidCodeGen() {
@@ -407,6 +408,8 @@ public class ShidCodeGen {
             case METHOD -> {
                 // TODO: handle not a struct type
                 var structType = ((ShidFieldLoad.Receiver.Struct) receiver).type();
+                // TODO: handle not a function type
+                var methodType = (FunctionType) node.type();
 
                 var receiverLocalNameResult = context.generateShidVal("receiver");
                 var receiverLocalName = receiverLocalNameResult.value();
@@ -416,17 +419,59 @@ public class ShidCodeGen {
                     new ShidLocalStore(receiverLocalName, node.source())
                 );
 
-                context = context.enterBlock(List.of(), node.source());
+                var shedPositionalParamNames = new ArrayList<String>();
+                for (
+                    var positionalParamIndex = 0;
+                    positionalParamIndex < methodType.positionalParams().size();
+                    positionalParamIndex++
+                ) {
+                    shedPositionalParamNames.add("arg" + positionalParamIndex);
+                }
+
+                var boundVariables = Stream.concat(
+                    shedPositionalParamNames.stream(),
+                    methodType.namedParams().stream()
+                        .map(namedParam -> namedParam.name())
+                )
+                    .map(BoundVariable::param)
+                    .toList();
+
+                context = context.enterBlock(boundVariables, node.source());
 
                 context = context.addInstruction(
                     new ShidMethodLoad(structType, node.memberName(), node.source())
                 );
+
                 context = context.addInstruction(
                     new ShidLocalLoad(receiverLocalName, node.source())
                 );
 
+                var shidPositionalParams = new ArrayList<ShidDefineFunction.Param>();
+                for (var positionalParamName : shedPositionalParamNames) {
+                    var shidParamName = context.shedVariableToLocal(positionalParamName);
+                    shidPositionalParams.add(new ShidDefineFunction.Param(shidParamName, node.source()));
+                    context = context.addInstruction(
+                        new ShidLocalLoad(shidParamName, node.source())
+                    );
+                }
+
+                var shidNamedParams = new ArrayList<ShidDefineFunction.Param>();
+                for (var namedParam : methodType.namedParams()) {
+                    var shidParamName = context.shedVariableToLocal(namedParam.name());
+                    shidNamedParams.add(new ShidDefineFunction.Param(shidParamName, node.source()));
+                    context = context.addInstruction(
+                        new ShidLocalLoad(shidParamName, node.source())
+                    );
+                }
+
                 context = context.addInstruction(
-                    new ShidCall(1, List.of(), node.source())
+                    new ShidCall(
+                        methodType.positionalParams().size() + 1,
+                        methodType.namedParams().stream()
+                            .map(namedParam -> namedParam.name())
+                            .toList(),
+                        node.source()
+                    )
                 );
 
                 context = context.addInstruction(new ShidReturn(node.source()));
@@ -438,8 +483,8 @@ public class ShidCodeGen {
                 yield context.addInstruction(
                     new ShidDefineFunction(
                         node.memberName(),
-                        List.of(),
-                        List.of(),
+                        shidPositionalParams,
+                        shidNamedParams,
                         FreeVariables.builder()
                             .plus(receiverLocalName, node.source())
                             .build(),