Allow binding of parameterised types

2 files changed, 49 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
index f70ee57..a770108 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaTypes.java
@@ -2,10 +2,11 @@ package org.zwobble.couscous.frontends.java;
 
 import com.google.common.collect.ImmutableSet;
 import org.eclipse.jdt.core.dom.*;
-import org.zwobble.couscous.types.ParameterizedType;
 import org.zwobble.couscous.types.*;
+import org.zwobble.couscous.types.ParameterizedType;
 import org.zwobble.couscous.types.Type;
 import org.zwobble.couscous.types.TypeParameter;
+import org.zwobble.couscous.util.Optionals;
 
 import java.util.List;
 import java.util.Optional;
@@ -14,6 +15,7 @@ import java.util.Set;
 import static com.google.common.collect.Iterables.transform;
 import static java.util.Arrays.asList;
 import static org.zwobble.couscous.types.Types.erasure;
+import static org.zwobble.couscous.util.Casts.tryCast;
 import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
 public class JavaTypes {
@@ -84,10 +86,33 @@ public class JavaTypes {
     public static Type bind(Type genericType, Type concreteType) {
         if (genericType.equals(concreteType)) {
             return concreteType;
-        } else if (!(genericType instanceof TypeParameter)) {
-            throw new RuntimeException("Type parameter was " + genericType);
+        }
+        return Optionals.flatMap(
+            tryCast(ParameterizedType.class, genericType),
+            tryCast(ParameterizedType.class, concreteType),
+            JavaTypes::bindParameterizedType)
+            .orElseGet(() -> {
+                if (!(genericType instanceof TypeParameter)) {
+                    return concreteType;
+                } else {
+                    return new BoundTypeParameter((TypeParameter) genericType, concreteType);
+                }
+            });
+    }
+
+    private static Optional<Type> bindParameterizedType(
+        ParameterizedType genericType,
+        ParameterizedType concreteType)
+    {
+        if (genericType.getRawType().equals(concreteType.getRawType())) {
+            return Optional.of(new ParameterizedType(
+                genericType.getRawType(),
+                eagerMap(
+                    genericType.getParameters(),
+                    concreteType.getParameters(),
+                    JavaTypes::bind)));
         } else {
-            return new BoundTypeParameter((TypeParameter) genericType, concreteType);
+            return Optional.empty();
         }
     }
 }
diff --git a/src/main/java/org/zwobble/couscous/util/Optionals.java b/src/main/java/org/zwobble/couscous/util/Optionals.java
new file mode 100644
index 0000000..979dc5f
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/util/Optionals.java
@@ -0,0 +1,20 @@
+package org.zwobble.couscous.util;
+
+import java.util.Optional;
+import java.util.function.BiFunction;
+
+public class Optionals {
+    private Optionals() {}
+
+    public static <T1, T2, R> Optional<R> flatMap(
+        Optional<T1> first,
+        Optional<T2> second,
+        BiFunction<T1, T2, Optional<R>> function)
+    {
+        if (first.isPresent() && second.isPresent()) {
+            return function.apply(first.get(), second.get());
+        } else {
+            return Optional.empty();
+        }
+    }
+}