ExpectActualUtils: repairSuperTypeList should consider type parameters

idea/src/org/jetbrains/kotlin/idea/quickfix/expectactual/ExpectActualUtils.kt

@@ -217,32 +217,37 @@ private fun KtPsiFactory.repairSuperTypeList(
     context: BindingContext
 ): Collection<String> {
     val superNames = linkedSetOf<String>()
-    generated.superTypeListEntries.zip(original.superTypeListEntries).forEach { (generatedEntry, originalEntry) ->
-        val superType = context[BindingContext.TYPE, originalEntry.typeReference]
-        val superClassDescriptor = superType?.constructor?.declarationDescriptor as? ClassDescriptor ?: return@forEach
-        if (generateExpectClass && !checker.checkAccessibility(superType)) {
-            generatedEntry.delete()
-            return@forEach
-        }
+    val typeParametersFqName = context[BindingContext.DECLARATION_TO_DESCRIPTOR, original]
+        ?.safeAs<ClassDescriptor>()
+        ?.declaredTypeParameters?.mapNotNull { it.fqNameOrNull()?.asString() }.orEmpty()
 
-        superType.fqName?.shortName()?.asString()?.let { superNames += it }
-        if (generateExpectClass) {
-            if (generatedEntry !is KtSuperTypeCallEntry) return@forEach
-        } else {
-            if (generatedEntry !is KtSuperTypeEntry) return@forEach
-        }
-
-        if (superClassDescriptor.kind == ClassKind.CLASS || superClassDescriptor.kind == ClassKind.ENUM_CLASS) {
-            val entryText = IdeDescriptorRenderers.SOURCE_CODE.renderType(superType)
-            val newGeneratedEntry = if (generateExpectClass) {
-                createSuperTypeEntry(entryText)
-            } else {
-                createSuperTypeCallEntry("$entryText()")
+    checker.runInContext(checker.existingTypeNames + typeParametersFqName) {
+        generated.superTypeListEntries.zip(original.superTypeListEntries).forEach { (generatedEntry, originalEntry) ->
+            val superType = context[BindingContext.TYPE, originalEntry.typeReference]
+            val superClassDescriptor = superType?.constructor?.declarationDescriptor as? ClassDescriptor ?: return@forEach
+            if (generateExpectClass && !checker.checkAccessibility(superType)) {
+                generatedEntry.delete()
+                return@forEach
+            }
+
+            superType.fqName?.shortName()?.asString()?.let { superNames += it }
+            if (generateExpectClass) {
+                if (generatedEntry !is KtSuperTypeCallEntry) return@forEach
+            } else {
+                if (generatedEntry !is KtSuperTypeEntry) return@forEach
+            }
+
+            if (superClassDescriptor.kind == ClassKind.CLASS || superClassDescriptor.kind == ClassKind.ENUM_CLASS) {
+                val entryText = IdeDescriptorRenderers.SOURCE_CODE.renderType(superType)
+                val newGeneratedEntry = if (generateExpectClass) {
+                    createSuperTypeEntry(entryText)
+                } else {
+                    createSuperTypeCallEntry("$entryText()")
+                }
+                generatedEntry.replace(newGeneratedEntry).safeAs<KtElement>()?.addToBeShortenedDescendantsToWaitingSet()
             }
-            generatedEntry.replace(newGeneratedEntry).safeAs<KtElement>()?.addToBeShortenedDescendantsToWaitingSet()
         }
     }
-
     if (generated.superTypeListEntries.isEmpty()) generated.getSuperTypeList()?.delete()
     return superNames
 }

idea/testData/multiModuleQuickFix/createExpect/classWithSuperClassAndTypeParameter/header/My.kt

@@ -0,0 +1,7 @@
+// My: to be implemented
+// DISABLE-ERRORS
+
+@Target(AnnotationTarget.TYPE, AnnotationTarget.VALUE_PARAMETER)
+annotation class SimpleA
+
+class Common<T>

idea/testData/multiModuleQuickFix/createExpect/classWithSuperClassAndTypeParameter/header/My.kt.after

@@ -0,0 +1,8 @@
+// My: to be implemented
+// DISABLE-ERRORS
+
+@Target(AnnotationTarget.TYPE, AnnotationTarget.VALUE_PARAMETER)
+annotation class SimpleA
+
+class Common<T>
+expect class My<T> : Common<@SimpleA T>

idea/testData/multiModuleQuickFix/createExpect/classWithSuperClassAndTypeParameter/jvm/My.kt

@@ -0,0 +1,4 @@
+// "Create expected class in common module testModule_Common" "true"
+// DISABLE-ERRORS
+
+actual class <caret>My<T> : Common<@SimpleA T>()

idea/testData/multiModuleQuickFix/createExpect/classWithSuperClassAndTypeParameter/jvm/My.kt.after

@@ -0,0 +1,4 @@
+// "Create expected class in common module testModule_Common" "true"
+// DISABLE-ERRORS
+
+actual class My<T> : Common<@SimpleA T>()

idea/testData/multiModuleQuickFix/createExpect/topLevelFunctionWithAnnotations/common/SimpleA.kt

@@ -1 +1,2 @@
+@Target(AnnotationTarget.TYPE, AnnotationTarget.VALUE_PARAMETER)
 annotation class SimpleA

idea/testData/multiModuleQuickFix/createExpect/topLevelFunctionWithAnnotations/common/myExtension.kt.after

@@ -1 +1 @@
-expect fun @receiver:SimpleA String.myExtension()
+expect fun <T : @SimpleA List<@SimpleA List<@SimpleA String>>> @receiver:SimpleA String.myExtension(@SimpleA a: @SimpleA List<@SimpleA List<@SimpleA String>>)

idea/testData/multiModuleQuickFix/createExpect/topLevelFunctionWithAnnotations/jvm/Utils.kt

@@ -1,4 +1,6 @@
 // "Create expected function in common module testModule_Common" "true"
 // DISABLE-ERRORS
 
-actual fun @receiver:SimpleA String.<caret>myExtension() { println(this) }
+actual fun <T : @SimpleA List<@SimpleA List<@SimpleA String>>>@receiver:SimpleA String.<caret>myExtension(@SimpleA a: @SimpleA List<@SimpleA List<@SimpleA String>>) {
+    println(this)
+}

idea/testData/multiModuleQuickFix/createExpect/topLevelFunctionWithAnnotations/jvm/Utils.kt.after

@@ -1,4 +1,6 @@
 // "Create expected function in common module testModule_Common" "true"
 // DISABLE-ERRORS
 
-actual fun @receiver:SimpleA String.<caret>myExtension() { println(this) }
+actual fun <T : @SimpleA List<@SimpleA List<@SimpleA String>>>@receiver:SimpleA String.myExtension(@SimpleA a: @SimpleA List<@SimpleA List<@SimpleA String>>) {
+    println(this)
+}

idea/tests/org/jetbrains/kotlin/idea/quickfix/QuickFixMultiModuleTestGenerated.java

@@ -332,6 +332,11 @@ public class QuickFixMultiModuleTestGenerated extends AbstractQuickFixMultiModul
             runTest("idea/testData/multiModuleQuickFix/createExpect/class/");
         }
 
+        @TestMetadata("classWithSuperClassAndTypeParameter")
+        public void testClassWithSuperClassAndTypeParameter() throws Exception {
+            runTest("idea/testData/multiModuleQuickFix/createExpect/classWithSuperClassAndTypeParameter/");
+        }
+
         @TestMetadata("classWithSuperTypeFromOtherPackage")
         public void testClassWithSuperTypeFromOtherPackage() throws Exception {
             runTest("idea/testData/multiModuleQuickFix/createExpect/classWithSuperTypeFromOtherPackage/");