[IR] Introduce & use declaration-based SymbolTable.withReferenceScope

compiler/ir/ir.tree/src/org/jetbrains/kotlin/ir/util/SymbolTable.kt

@@ -63,8 +63,10 @@ interface ReferenceSymbolTable {
     fun referenceTypeAliasFromLinker(descriptor: TypeAliasDescriptor, sig: IdSignature): IrTypeAliasSymbol
 
     fun enterScope(owner: DeclarationDescriptor)
+    fun enterScope(owner: IrDeclaration)
 
     fun leaveScope(owner: DeclarationDescriptor)
+    fun leaveScope(owner: IrDeclaration)
 }
 
 open class SymbolTable(
@@ -955,7 +957,7 @@ open class SymbolTable(
         scopedSymbolTables.forEach { it.enterScope(owner) }
     }
 
-    fun enterScope(owner: IrDeclaration) {
+    override fun enterScope(owner: IrDeclaration) {
         enterScope(owner.descriptor)
     }
 
@@ -963,7 +965,7 @@ open class SymbolTable(
         scopedSymbolTables.forEach { it.leaveScope(owner) }
     }
 
-    fun leaveScope(owner: IrDeclaration) {
+    override fun leaveScope(owner: IrDeclaration) {
         leaveScope(owner.descriptor)
     }
 
@@ -1026,6 +1028,13 @@ inline fun <T, D : DeclarationDescriptor> ReferenceSymbolTable.withReferenceScop
     return result
 }
 
+inline fun <T, D : IrDeclaration> ReferenceSymbolTable.withReferenceScope(owner: D, block: ReferenceSymbolTable.(D) -> T): T {
+    enterScope(owner)
+    val result = block(owner)
+    leaveScope(owner)
+    return result
+}
+
 val SymbolTable.allUnbound: Set<IrSymbol>
     get() {
         val r = mutableSetOf<IrSymbol>()

plugins/kotlin-serialization/kotlin-serialization-compiler/src/org/jetbrains/kotlinx/serialization/compiler/backend/ir/GeneratorHelpers.kt

@@ -136,7 +136,7 @@ interface IrBuilderExtension {
 
     fun <T : IrDeclaration> T.buildWithScope(builder: (T) -> Unit): T =
         also { irDeclaration ->
-            compilerContext.symbolTable.withReferenceScope(irDeclaration.descriptor) {
+            compilerContext.symbolTable.withReferenceScope(irDeclaration) {
                 builder(irDeclaration)
             }
         }