FIR: replace resolve phase in FirFile properly

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirAbstractPhaseTransformer.kt

@@ -32,6 +32,13 @@ abstract class FirAbstractPhaseTransformer<D>(
             return phasedFir(session, requiredPhase)
         }
 
+    override fun transformFile(file: FirFile, data: D): CompositeTransformResult<FirFile> {
+        file.replaceResolvePhase(transformerPhase)
+
+        @Suppress("UNCHECKED_CAST")
+        return super.transformFile(file, data) as CompositeTransformResult<FirFile>
+    }
+
     override fun transformDeclaration(declaration: FirDeclaration, data: D): CompositeTransformResult<FirDeclaration> {
         declaration.replaceResolvePhase(transformerPhase)
 

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirImportResolveTransformer.kt

@@ -33,6 +33,7 @@ class FirImportResolveTransformer() : FirAbstractTreeTransformer<Nothing?>(phase
     }
 
     override fun transformFile(file: FirFile, data: Nothing?): CompositeTransformResult<FirFile> {
+        file.replaceResolvePhase(transformerPhase)
         session = file.session
         symbolProvider = FirSymbolProvider.getInstance(file.session)
         return file.also { it.transformChildren(this, null) }.compose()

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirStatusResolveTransformer.kt

@@ -26,7 +26,7 @@ class FirStatusResolveTransformer : FirAbstractTreeTransformer<FirDeclarationSta
 
     override fun transformFile(file: FirFile, data: FirDeclarationStatus?): CompositeTransformResult<FirFile> {
         session = file.session
-        return transformElement(file, data)
+        return super.transformFile(file, data)
     }
 
     override fun transformDeclarationStatus(

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirSupertypesResolution.kt

@@ -52,6 +52,8 @@ private class FirApplySupertypesTransformer(
     }
 
     override fun transformFile(file: FirFile, data: Nothing?): CompositeTransformResult<FirDeclaration> {
+        file.replaceResolvePhase(FirResolvePhase.SUPER_TYPES)
+
         return (file.transformChildren(this, null) as FirFile).compose()
     }
 

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirTypeResolveTransformer.kt

@@ -30,7 +30,7 @@ class FirTypeResolveTransformer : FirAbstractTreeTransformerWithSuperTypes(
         return withScopeCleanup {
             towerScope.addImportingScopes(file, session, scopeSession)
             typeResolverTransformer = FirSpecificTypeResolverTransformer(towerScope, session)
-            transformElement(file, data)
+            super.transformFile(file, data)
         }
     }
 

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirBodyResolveTransformer.kt

@@ -41,8 +41,7 @@ open class FirBodyResolveTransformer(
         components.file = file
         return withScopeCleanup(components.topLevelScopes) {
             components.topLevelScopes.addImportingScopes(file, session, components.scopeSession)
-            @Suppress("UNCHECKED_CAST")
+            super.transformFile(file, data)
-            super.transformFile(file, data) as CompositeTransformResult<FirFile>
         }
     }