[LL FIR] KT-57220 LLFirSessionCache: Clean up symbol provider merging

- `SymbolProviderMerger` encapsulates some boilerplate, which should
  make `mergeDependencySymbolProvidersInto` easier to read and also much
  easier to extend.

analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/sessions/LLFirSessionCache.kt

@@ -537,18 +537,29 @@ internal class LLFirSessionCache(private val project: Project) {
         session: FirSession,
         destination: MutableList<FirSymbolProvider>,
     ) {
-        val (kotlinSymbolProviders, remainingSymbolProviders1) = partitionIsInstance<_, LLFirProvider.SymbolProvider>()
+        SymbolProviderMerger(this, destination).apply {
-        destination.addIfNotNull(LLFirCombinedKotlinSymbolProvider.merge(session, project, kotlinSymbolProviders))
+            merge<LLFirProvider.SymbolProvider> { LLFirCombinedKotlinSymbolProvider.merge(session, project, it) }
+            merge<FirExtensionSyntheticFunctionInterfaceProvider> { LLFirCombinedSyntheticFunctionSymbolProvider.merge(session, it) }
+            finish()
+        }
+    }
 
-        val (syntheticFunctionSymbolProviders, remainingSymbolProviders2) =
+    private class SymbolProviderMerger(
-            remainingSymbolProviders1.partitionIsInstance<_, FirExtensionSyntheticFunctionInterfaceProvider>()
+        symbolProviders: List<FirSymbolProvider>,
+        private val destination: MutableList<FirSymbolProvider>
+    ) {
+        private var remainingSymbolProviders = symbolProviders
 
-        destination.addAll(remainingSymbolProviders2)
+        inline fun <reified A : FirSymbolProvider> merge(create: (List<A>) -> FirSymbolProvider?) {
+            val (specificSymbolProviders, remainingSymbolProviders) = remainingSymbolProviders.partitionIsInstance<_, A>()
+            destination.addIfNotNull(create(specificSymbolProviders))
+            this.remainingSymbolProviders = remainingSymbolProviders
+        }
 
-        // Unfortunately, the functions that an extension synthetic function symbol provider might provide differ between sessions because
+        fun finish() {
-        // they depend on compiler plugins. However, only extension providers that are affected by compiler plugins are added in
+            destination.addAll(remainingSymbolProviders)
-        // `createSourcesSession`. We can still combine these, because the `ClassId` heuristics only need to be checked once.
+            remainingSymbolProviders = emptyList()
-        destination.addIfNotNull(LLFirCombinedSyntheticFunctionSymbolProvider.merge(session, syntheticFunctionSymbolProviders))
+        }
     }
 }