[FIR] Move FirPhaseManager to :compiler:fir:tree

2021-07-07 11:34:45 +03:00
parent 51fc2e453f
commit 9f161f16b3
19 changed files with 55 additions and 59 deletions
@@ -0,0 +1,38 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * 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.symbols
+
+import org.jetbrains.kotlin.fir.FirSession
+import org.jetbrains.kotlin.fir.FirSessionComponent
+import org.jetbrains.kotlin.fir.declarations.FirDeclaration
+import org.jetbrains.kotlin.fir.declarations.FirResolvePhase
+
+abstract class FirPhaseManager : FirSessionComponent {
+    abstract fun ensureResolved(
+        symbol: FirBasedSymbol<*>,
+        requiredPhase: FirResolvePhase = FirResolvePhase.DECLARATIONS
+    )
+}
+
+val FirSession.phaseManager: FirPhaseManager by FirSession.sessionComponentAccessor()
+
+fun FirBasedSymbol<*>.ensureResolved(
+    requiredPhase: FirResolvePhase,
+    // TODO: Currently, the parameter is unused but it's needed to guarantee that all call-sites are able to supply use-site session
+    // TODO: Decide which one session should be used and probably get rid of the parameter if use-site session is not needed
+    @Suppress("UNUSED_PARAMETER") useSiteSession: FirSession,
+) {
+    val session = fir.moduleData.session
+    val phaseManager = session.phaseManager
+    phaseManager.ensureResolved(this, requiredPhase)
+}
+
+fun FirDeclaration.ensureResolved(
+    requiredPhase: FirResolvePhase,
+    useSiteSession: FirSession,
+) {
+    symbol.ensureResolved(requiredPhase, useSiteSession)
+}