Improve generic support in the interpreter

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodSignature.java b/src/main/java/org/zwobble/couscous/ast/MethodSignature.java
index 060d2bc..715dade 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodSignature.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodSignature.java
@@ -1,9 +1,12 @@
 package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.types.Type;
+import org.zwobble.couscous.types.Types;
 
 import java.util.List;
 
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
+
 public class MethodSignature {
     public static MethodSignature signature(String name, List<Type> arguments, Type returnType) {
         return new MethodSignature(name, arguments, returnType);
@@ -31,6 +34,13 @@ public class MethodSignature {
         return returnType;
     }
 
+    public MethodSignature generic() {
+        return new MethodSignature(
+            name,
+            eagerMap(arguments, Types::generic),
+            Types.generic(returnType));
+    }
+
     @Override
     public String toString() {
         return "MethodSignature(" +