[kotlin] k2 override/implement: take care of enum entries

^KTIJ-25970 fixed

GitOrigin-RevId: dbea6ce1ed3e5c48e1df6c548a175f8862ae22fc

plugins/kotlin/code-insight/override-implement-k2/src/org/jetbrains/kotlin/idea/core/overrideImplement/KtOverrideMembersHandler.kt

@@ -25,22 +25,24 @@ open class KtOverrideMembersHandler : KtGenerateMembersHandler(false) {
 
 context(KaSession)
 @OptIn(KaExperimentalApi::class)
-private fun collectMembers(classOrObject: KtClassOrObject): List<KtClassMember> =
-        classOrObject.classSymbol?.let { getOverridableMembers(it) }.orEmpty().map { (symbol, bodyType, containingSymbol) ->
-            @NlsSafe
-            val fqName = containingSymbol?.classId?.asSingleFqName()?.toString() ?: containingSymbol?.name?.asString()
-            KtClassMember(
-                KtClassMemberInfo.create(
-                    symbol,
-                    symbol.render(renderer),
-                    getIcon(symbol),
-                    fqName,
-                    containingSymbol?.let { getIcon(it) },
-                ),
-                bodyType,
-                preferConstructorParameter = false,
-            )
-        }
+private fun collectMembers(classOrObject: KtClassOrObject): List<KtClassMember> {
+    val classSymbol = (classOrObject.symbol as? KaEnumEntrySymbol)?.enumEntryInitializer as? KaClassSymbol ?: classOrObject.classSymbol ?: return emptyList()
+    return getOverridableMembers(classSymbol).map { (symbol, bodyType, containingSymbol) ->
+        @NlsSafe
+        val fqName = containingSymbol?.classId?.asSingleFqName()?.toString() ?: containingSymbol?.name?.asString()
+        KtClassMember(
+            KtClassMemberInfo.create(
+                symbol,
+                symbol.render(renderer),
+                getIcon(symbol),
+                fqName,
+                containingSymbol?.let { getIcon(it) },
+            ),
+            bodyType,
+            preferConstructorParameter = false,
+        )
+    }
+}
 
 context(KaSession)
 @OptIn(KaExperimentalApi::class)

plugins/kotlin/fir/tests/test/org/jetbrains/kotlin/idea/fir/codeInsight/FirOverrideImplementTest.kt

@@ -396,6 +396,10 @@ internal class FirOverrideImplementTest : OverrideImplementTest<KtClassMember>()
         super.testEnumClass4()
     }
 
+    override fun testEnumEntry() {
+        super.testEnumEntry()
+    }
+
     override fun testOverrideExternalFunction() {
         super.testOverrideExternalFunction()
     }

plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/codeInsight/OverrideImplementTest.kt

@@ -401,6 +401,10 @@ abstract class OverrideImplementTest<T : ClassMember> : AbstractOverrideImplemen
         doOverrideFileTest("toString")
     }
 
+    open fun testEnumEntry() {
+        doOverrideFileTest("foo")
+    }
+
    open fun testOverrideExternalFunction() {
         doOverrideFileTest()
     }

plugins/kotlin/idea/tests/testData/codeInsight/overrideImplement/enumEntry.kt

@@ -0,0 +1,9 @@
+// FIR_IDENTICAL
+enum class EnumIssue {
+
+    A {
+        <caret>
+    };
+
+    abstract fun foo()
+}

plugins/kotlin/idea/tests/testData/codeInsight/overrideImplement/enumEntry.kt.after

@@ -0,0 +1,11 @@
+// FIR_IDENTICAL
+enum class EnumIssue {
+
+    A {
+        override fun foo() {
+            TODO("Not yet implemented")
+        }
+    };
+
+    abstract fun foo()
+}