Handle flexible type in asMemberOf

google · Sep 19, 2024 · 9faf216 · 9faf216
1 parent 7699b6e
commit 9faf216
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/kotlin-analysis-api/src/main/kotlin/com/google/devtools/ksp/impl/symbol/kotlin/util.kt b/kotlin-analysis-api/src/main/kotlin/com/google/devtools/ksp/impl/symbol/kotlin/util.kt
@@ -606,11 +606,13 @@ internal fun KaValueParameterSymbol.getDefaultValue(): KaAnnotationValue? {
 
 @OptIn(KaExperimentalApi::class)
 internal fun fillInDeepSubstitutor(context: KaType, substitutorBuilder: KaSubstitutorBuilder) {
-    if (context !is KaClassType) {
-        return
+    val unwrappedType = when (context) {
+        is KaClassType -> context
+        is KaFlexibleType -> context.lowerBound as? KaClassType ?: return
+        else -> return
     }
-    val parameters = context.symbol.typeParameters
-    val arguments = context.typeArguments
+    val parameters = unwrappedType.symbol.typeParameters
+    val arguments = unwrappedType.typeArguments
     if (parameters.size != arguments.size) {
         throw IllegalStateException("invalid substitution for $context")
     }