JVM_IR KT-45697 reference classes from sources before IR generation

compiler/ir/ir.psi2ir/src/org/jetbrains/kotlin/psi2ir/Psi2IrTranslator.kt

@@ -25,7 +25,6 @@ import org.jetbrains.kotlin.ir.linkage.IrDeserializer
 import org.jetbrains.kotlin.ir.linkage.IrProvider
 import org.jetbrains.kotlin.ir.symbols.IrSymbol
 import org.jetbrains.kotlin.ir.util.SymbolTable
-import org.jetbrains.kotlin.ir.util.TypeTranslator
 import org.jetbrains.kotlin.ir.util.noUnboundLeft
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.psi2ir.generators.GeneratorContext
@@ -35,7 +34,9 @@ import org.jetbrains.kotlin.psi2ir.generators.TypeTranslatorImpl
 import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.utils.SmartList
 
-typealias Psi2IrPostprocessingStep = (IrModuleFragment) -> Unit
+fun interface Psi2IrPostprocessingStep {
+    fun invoke(irModuleFragment: IrModuleFragment)
+}
 
 class Psi2IrTranslator(
     val languageVersionSettings: LanguageVersionSettings,

compiler/ir/ir.psi2ir/src/org/jetbrains/kotlin/psi2ir/preprocessing/SourceDeclarationsPreprocessor.kt

@@ -0,0 +1,40 @@
+/*
+ * 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.psi2ir.preprocessing
+
+import org.jetbrains.kotlin.psi.KtClassOrObject
+import org.jetbrains.kotlin.psi.KtDeclaration
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.synthetics.findClassDescriptor
+import org.jetbrains.kotlin.psi2ir.generators.GeneratorContext
+import org.jetbrains.kotlin.resolve.DescriptorUtils
+
+class SourceDeclarationsPreprocessor(private val context: GeneratorContext) {
+
+    fun run(ktFiles: Collection<KtFile>) {
+        for (ktFile in ktFiles.toSet()) {
+            for (ktDeclaration in ktFile.declarations) {
+                processDeclaration(ktDeclaration)
+            }
+        }
+    }
+
+    private fun processDeclaration(ktDeclaration: KtDeclaration) {
+        when (ktDeclaration) {
+            is KtClassOrObject ->
+                processClassOrObject(ktDeclaration)
+        }
+    }
+
+    private fun processClassOrObject(ktClassOrObject: KtClassOrObject) {
+        val classDescriptor = ktClassOrObject.findClassDescriptor(context.bindingContext)
+        if (DescriptorUtils.isEnumEntry(classDescriptor)) return
+        context.symbolTable.referenceClass(classDescriptor)
+        ktClassOrObject.body?.let { ktClassBody ->
+            ktClassBody.declarations.forEach { processDeclaration(it) }
+        }
+    }
+}