Improve error messaging for error types used as annotation arguments (#…

…80)

CHANGELOG.md

@@ -3,6 +3,8 @@
 **Unreleased**
 --------------
 
+- **Enhancement**: Improve error messaging for error types used as annotation arguments.
+
 0.3.3
 -----
 

compiler-utils/api/compiler-utils.api

@@ -112,6 +112,7 @@ public final class com/squareup/anvil/compiler/internal/ksp/KspUtilKt {
 	public static final fun anySymbolsWithAnnotations (Lcom/google/devtools/ksp/processing/Resolver;Ljava/util/Collection;)Z
 	public static final fun atLeastOneAnnotation (Lcom/google/devtools/ksp/symbol/KSClassDeclaration;Ljava/lang/String;Lcom/google/devtools/ksp/symbol/KSType;)Ljava/util/List;
 	public static synthetic fun atLeastOneAnnotation$default (Lcom/google/devtools/ksp/symbol/KSClassDeclaration;Ljava/lang/String;Lcom/google/devtools/ksp/symbol/KSType;ILjava/lang/Object;)Ljava/util/List;
+	public static final fun checkErrorType (Lcom/google/devtools/ksp/symbol/KSType;Lcom/google/devtools/ksp/symbol/KSNode;)Lcom/google/devtools/ksp/symbol/KSType;
 	public static final fun contextualToClassName (Lcom/google/devtools/ksp/symbol/KSType;Lcom/google/devtools/ksp/symbol/KSNode;)Lcom/squareup/kotlinpoet/ClassName;
 	public static final fun contextualToClassName (Lcom/google/devtools/ksp/symbol/KSTypeReference;)Lcom/squareup/kotlinpoet/ClassName;
 	public static final fun contextualToTypeName (Lcom/google/devtools/ksp/symbol/KSType;Lcom/google/devtools/ksp/symbol/KSNode;Lcom/squareup/kotlinpoet/ksp/TypeParameterResolver;)Lcom/squareup/kotlinpoet/TypeName;

compiler-utils/src/main/java/com/squareup/anvil/compiler/internal/ksp/KSAnnotationExtensions.kt

@@ -166,7 +166,17 @@ public fun KSAnnotation.classNameArgumentAt(
 public inline fun <reified T> KSAnnotation.argumentOfTypeAt(
   name: String,
 ): T? {
-  return argumentOfTypeWithMapperAt<T, T>(name) { _, value ->
+  return argumentOfTypeWithMapperAt<T, T>(name) { arg, value ->
+    when {
+      value is KSType -> {
+        value.checkErrorType(arg)
+      }
+      value is List<*> && value.firstOrNull() is KSType -> {
+        for (element in value) {
+          (element as KSType).checkErrorType(arg)
+        }
+      }
+    }
     value
   }
 }

compiler-utils/src/main/java/com/squareup/anvil/compiler/internal/ksp/KspUtil.kt

@@ -387,18 +387,19 @@ public fun KSType.contextualToClassName(origin: KSNode): ClassName {
   return toClassName()
 }
 
-private fun KSType.checkErrorType(origin: KSNode) {
+@PublishedApi
+internal fun KSType.checkErrorType(origin: KSNode): KSType {
   val (type, node) = if (isError) {
     this to origin
   } else {
     arguments.asSequence()
-      .mapNotNull {
-        it.type?.let {
-          it.resolve() to it
+      .mapNotNull { arg ->
+        arg.type?.let { argType ->
+          argType.resolve() to argType
         }
       }
       .firstOrNull { it.first.isError }
-  } ?: return
+  } ?: return this
 
   val message = buildString {
     appendLine(