Fix generic issue with return type signature

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/mapping/ProcyonEntryFactory.java b/src/main/java/cuchaz/enigma/mapping/ProcyonEntryFactory.java
index 8cbca13d..941caf9d 100644
--- a/src/main/java/cuchaz/enigma/mapping/ProcyonEntryFactory.java
+++ b/src/main/java/cuchaz/enigma/mapping/ProcyonEntryFactory.java
@@ -34,7 +34,16 @@ public class ProcyonEntryFactory {
                 builder.append(paramType.getErasedSignature());
         }
         builder.append(")");
-        builder.append(methodReference.getReturnType().getErasedSignature());
+
+        // TODO: Fix Procyon render
+        TypeReference returnType = methodReference.getReturnType();
+        if (returnType.getErasedSignature().equals("Ljava/lang/Object;") && returnType.hasExtendsBound() && returnType.getExtendsBound() instanceof  CompoundTypeReference)
+        {
+            List<TypeReference> interfaces = ((CompoundTypeReference) returnType.getExtendsBound()).getInterfaces();
+            interfaces.forEach((inter) -> builder.append(inter.getErasedSignature()));
+        }
+        else
+            builder.append(returnType.getErasedSignature());
         return builder.toString();
     }