FIR: Add components storage to FirSession

compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSession.kt

@@ -5,4 +5,16 @@
 
 package org.jetbrains.kotlin.fir
 
-interface FirSession
+import kotlin.reflect.KClass
+
+interface FirSession {
+
+
+    val components: Map<KClass<*>, Any>
+
+    fun <T : Any> getService(kclass: KClass<T>): T =
+        components[kclass] as T
+}
+
+inline fun <reified T : Any> FirSession.service(): T =
+    getService(T::class)

compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSessionBase.kt

@@ -0,0 +1,19 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir
+
+import kotlin.reflect.KClass
+
+abstract class FirSessionBase : FirSession {
+
+    protected fun <T : Any> registerComponent(tClass: KClass<T>, t: T) {
+        assert(tClass !in components) { "Already registered component" }
+        components[tClass] = t
+    }
+
+    override val components: MutableMap<KClass<*>, Any> = mutableMapOf()
+
+}

compiler/tests-common/tests/org/jetbrains/kotlin/fir/builder/AbstractRawFirBuilderTestCase.kt

@@ -16,7 +16,7 @@ import com.intellij.util.PathUtil
 import org.jetbrains.kotlin.KtNodeTypes
 import org.jetbrains.kotlin.cli.common.script.CliScriptDefinitionProvider
 import org.jetbrains.kotlin.fir.FirRenderer
-import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.FirSessionBase
 import org.jetbrains.kotlin.parsing.KotlinParserDefinition
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.psi.KtPsiFactory
@@ -58,7 +58,7 @@ abstract class AbstractRawFirBuilderTestCase : KtParsingTestCase(
         val file = createFile(filePath, fileType) as KtFile
         myFile = file
 
-        val firFile = RawFirBuilder(object : FirSession {}).buildFirFile(file)
+        val firFile = RawFirBuilder(object : FirSessionBase() {}).buildFirFile(file)
         val firFileDump = StringBuilder().also { FirRenderer(it).visitFile(firFile) }.toString()
         val expectedPath = filePath.replace(".kt", ".txt")
         KotlinTestUtils.assertEqualsToFile(File(expectedPath), firFileDump)