Add FirScope::getCallableNames function

compiler/fir/java/src/org/jetbrains/kotlin/fir/java/enhancement/SignatureEnhancement.kt

@@ -55,6 +55,28 @@ class FirSignatureEnhancement(
         FirJavaEnhancementContext(session) { null }.copyWithNewDefaultTypeQualifiers(typeQualifierResolver, jsr305State, owner.annotations)
 
     private val enhancements = mutableMapOf<FirCallableSymbol<*>, FirCallableSymbol<*>>()
+    private val overriddenFunctions = mutableMapOf<FirFunctionSymbol<*>, Collection<FirFunctionSymbol<*>>>()
+
+    override fun processPropertiesByName(name: Name, processor: (FirVariableSymbol<*>) -> Unit) {
+        useSiteMemberScope.processPropertiesByName(name) process@{ original ->
+
+            val field = enhancements.getOrPut(original) { enhance(original, name) }
+            processor(field as FirVariableSymbol<*>)
+        }
+
+        return super.processPropertiesByName(name, processor)
+    }
+
+    override fun getCallableNames(): Set<Name> {
+        return useSiteMemberScope.getCallableNames() + super.getCallableNames()
+    }
+
+    override fun processFunctionsByName(name: Name, processor: (FirFunctionSymbol<*>) -> Unit) {
+        useSiteMemberScope.processFunctionsByName(name) process@{ original ->
+
+            val function = enhancements.getOrPut(original) { enhance(original, name) }
+            processor(function as FirFunctionSymbol<*>)
+        }
 
     fun enhancedFunction(
         function: FirFunctionSymbol<*>,

compiler/fir/java/src/org/jetbrains/kotlin/fir/java/scopes/JavaClassUseSiteMemberScope.kt

@@ -45,6 +45,10 @@ class JavaClassUseSiteMemberScope(
         }
     }
 
+    override fun getCallableNames(): Set<Name> {
+        return declaredMemberScope.getCallableNames() + superTypesScope.getCallableNames()
+    }
+
     private fun generateAccessorSymbol(
         functionSymbol: FirFunctionSymbol<*>,
         syntheticPropertyName: Name,

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/AbstractFirUseSiteMemberScope.kt

@@ -130,4 +130,8 @@ abstract class AbstractFirUseSiteMemberScope(
     override fun processDeclaredConstructors(processor: (FirConstructorSymbol) -> Unit) {
         declaredMemberScope.processDeclaredConstructors(processor)
     }
+
+    override fun getCallableNames(): Set<Name> {
+        return declaredMemberScope.getCallableNames() + superTypesScope.getCallableNames()
+    }
 }

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/FirClassDeclaredMemberScope.kt

@@ -76,6 +76,10 @@ class FirClassDeclaredMemberScope(
     ) {
         nestedClassifierScope?.processClassifiersByNameWithSubstitution(name, processor)
     }
+
+    override fun getCallableNames(): Set<Name> {
+        return callablesIndex.keys
+    }
 }
 
 

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/FirTypeIntersectionScope.kt

@@ -341,6 +341,10 @@ class FirTypeIntersectionScope private constructor(
         return ProcessorAction.NEXT
     }
 
+    override fun getCallableNames(): Set<Name> {
+        return scopes.flatMapTo(mutableSetOf()) { it.getCallableNames() }
+    }
+
     companion object {
         fun prepareIntersectionScope(
             session: FirSession,

compiler/fir/tree/src/org/jetbrains/kotlin/fir/scopes/FirCompositeScope.kt

@@ -44,4 +44,8 @@ class FirCompositeScope(private val scopes: Iterable<FirScope>) : FirScope() {
     override fun processPropertiesByName(name: Name, processor: (FirVariableSymbol<*>) -> Unit) {
         return processComposite(FirScope::processPropertiesByName, name, processor)
     }
+
+    override fun getCallableNames(): Set<Name> {
+        return scopes.flatMapTo(mutableSetOf()) { it.getCallableNames() }
+    }
 }

compiler/fir/tree/src/org/jetbrains/kotlin/fir/scopes/FirScope.kt

@@ -34,6 +34,7 @@ abstract class FirScope {
 
     open fun mayContainName(name: Name) = true
 
+    open fun getCallableNames(): Set<Name> = emptySet()
 }
 
 fun FirTypeScope.processOverriddenFunctionsAndSelf(