[FIR] Add separate session component for FirIdeProvider

idea/idea-analysis/src/org/jetbrains/kotlin/idea/fir/FirIdeJavaModuleBasedSession.kt

@@ -34,10 +34,18 @@ class FirIdeJavaModuleBasedSession(
 
 
     init {
+        val firIdeProvider = FirIdeProvider(project, scope, this, KotlinScopeProvider(::wrapScopeWithJvmMapped))
+
         registerComponent(
             FirProvider::class,
-            FirIdeProvider(project, scope, this, KotlinScopeProvider(::wrapScopeWithJvmMapped))
+            firIdeProvider
         )
+
+        registerComponent(
+            FirIdeProvider::class,
+            firIdeProvider
+        )
+
         registerComponent(
             FirSymbolProvider::class,
             FirCompositeSymbolProvider(

idea/idea-analysis/src/org/jetbrains/kotlin/idea/fir/FirIdeProvider.kt

@@ -150,4 +150,6 @@ class FirIdeProvider(
         getFirClassifierByFqName(classId)
         return cacheProvider.getNestedClassifierScope(classId)
     }
-}
+}
+
+val FirSession.firIdeProvider: FirIdeProvider by FirSession.sessionComponentAccessor()

idea/idea-analysis/src/org/jetbrains/kotlin/idea/fir/FirResolutionApi.kt

@@ -81,7 +81,7 @@ fun KtCallableDeclaration.getOrBuildFir(
     val klassFqName = this.containingClassOrObject?.relativeFqName()
     val declName = this.nameAsSafeName
 
-    val firProvider = session.firProvider as FirIdeProvider
+    val firProvider = session.firIdeProvider
     val firFile = firProvider.getOrBuildFile(file)
     val firMemberSymbol = firFile.findCallableMember(firProvider, this, packageFqName, klassFqName, declName).symbol
     val firMemberDeclaration = firMemberSymbol.fir
@@ -104,7 +104,7 @@ fun KtClassOrObject.getOrBuildFir(
     val packageFqName = file.packageFqName
     val klassFqName = this.relativeFqName()
 
-    val firProvider = session.firProvider as FirIdeProvider
+    val firProvider = session.firIdeProvider
     val firFile = firProvider.getOrBuildFile(file)
 
     val firClassOrEnumEntry = if (this is KtEnumEntry) {
@@ -124,7 +124,7 @@ fun KtClassOrObject.getOrBuildFir(
 
 private fun KtFile.getOrBuildRawFirFile(state: FirModuleResolveState): Pair<FirIdeProvider, FirFile> {
     val session = state.getSession(this)
-    val firProvider = session.firProvider as FirIdeProvider
+    val firProvider = session.firIdeProvider
     return firProvider to firProvider.getOrBuildFile(this)
 }
 

idea/idea-fir/tests/org/jetbrains/kotlin/idea/fir/AbstractFirLazyResolveTest.kt

@@ -102,7 +102,7 @@ abstract class AbstractFirLazyResolveTest : KotlinLightCodeInsightFixtureTestCas
         for (file in files) {
             val psiFile = psiManager.findFile(file) as KtFile
             val session = resolveState.getSession(psiFile)
-            val firProvider = session.firProvider as FirIdeProvider
+            val firProvider = session.firIdeProvider
             val firFile = firProvider.getFile(psiFile) ?: continue
             KotlinTestUtils.assertEqualsToFile(File(expectedTxtPath(file)), firFile.render())
         }