Improve error message for failed binding validations when the paramet…

…erand return type do not match
square · Dec 16, 2022 · 3ee77e4 · 3ee77e4
1 parent a4a932c
commit 3ee77e4
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 5 deletions.
diff --git a/compiler/src/main/java/com/squareup/anvil/compiler/codegen/dagger/BindsMethodValidator.kt b/compiler/src/main/java/com/squareup/anvil/compiler/codegen/dagger/BindsMethodValidator.kt
@@ -7,6 +7,7 @@ import com.squareup.anvil.compiler.codegen.PrivateCodeGenerator
 import com.squareup.anvil.compiler.daggerBindsFqName
 import com.squareup.anvil.compiler.daggerModuleFqName
 import com.squareup.anvil.compiler.internal.reference.AnvilCompilationExceptionFunctionReference
+import com.squareup.anvil.compiler.internal.reference.ClassReference
 import com.squareup.anvil.compiler.internal.reference.FunctionReference
 import com.squareup.anvil.compiler.internal.reference.allSuperTypeClassReferences
 import com.squareup.anvil.compiler.internal.reference.classAndInnerClassReferences
@@ -75,28 +76,43 @@ internal class BindsMethodValidator : PrivateCodeGenerator() {
     )
 
     if (!function.parameterMatchesReturnType() && !function.receiverMatchesReturnType()) {
+      val returnType = function.returnType().asClassReference().shortName
+      val paramSuperTypes = (function.parameterSuperTypes() ?: function.receiverSuperTypes())!!
+        .map { it.shortName }
+        .toList()
+
       throw AnvilCompilationExceptionFunctionReference(
-        message = "@Binds methods' parameter type must be assignable to the return type",
+        message = "@Binds methods' parameter type must be assignable to the return type. " +
+          "Expected return type of $returnType but impl parameter of type " +
+          "${paramSuperTypes.first()} only has the following supertypes: $paramSuperTypes",
         functionReference = function
       )
     }
   }
 
   private fun FunctionReference.Psi.parameterMatchesReturnType(): Boolean {
+    return parameterSuperTypes()
+      ?.contains(returnType().asClassReference())
+      ?: false
+  }
+
+  private fun FunctionReference.Psi.parameterSuperTypes(): Sequence<ClassReference>? {
     return parameters.singleOrNull()
       ?.type()
       ?.asClassReference()
       ?.allSuperTypeClassReferences(includeSelf = true)
+  }
+
+  private fun FunctionReference.Psi.receiverMatchesReturnType(): Boolean {
+    return receiverSuperTypes()
       ?.contains(returnType().asClassReference())
       ?: false
   }
 
-  private fun FunctionReference.Psi.receiverMatchesReturnType(): Boolean {
+  private fun FunctionReference.Psi.receiverSuperTypes(): Sequence<ClassReference>? {
     return function.receiverTypeReference
       ?.toTypeReference(declaringClass)
       ?.asClassReference()
       ?.allSuperTypeClassReferences(includeSelf = true)
-      ?.contains(returnType().asClassReference())
-      ?: false
   }
 }
diff --git a/compiler/src/test/java/com/squareup/anvil/compiler/dagger/BindsMethodValidatorTest.kt b/compiler/src/test/java/com/squareup/anvil/compiler/dagger/BindsMethodValidatorTest.kt
@@ -37,7 +37,9 @@ class BindsMethodValidatorTest(
       import dagger.Module
       import javax.inject.Inject
 
-      class Foo @Inject constructor()
+      interface Lorem
+      open class Ipsum
+      class Foo @Inject constructor() : Ipsum(), Lorem
       interface Bar
 
       @Module
@@ -51,6 +53,12 @@ class BindsMethodValidatorTest(
       assertThat(messages).contains(
         "@Binds methods' parameter type must be assignable to the return type"
       )
+      if (!useDagger) {
+        assertThat(messages).contains(
+          "Expected return type of Bar but impl parameter of type Foo only has the following " +
+            "supertypes: [Foo, Ipsum, Lorem]"
+        )
+      }
     }
   }