diff --git a/compiler/fir/fir2ir/src/org/jetbrains/kotlin/fir/symbols/Fir2IrBindableSymbol.kt b/compiler/fir/fir2ir/src/org/jetbrains/kotlin/fir/symbols/Fir2IrBindableSymbol.kt new file mode 100644 index 00000000000..85a9d6f65b9 --- /dev/null +++ b/compiler/fir/fir2ir/src/org/jetbrains/kotlin/fir/symbols/Fir2IrBindableSymbol.kt @@ -0,0 +1,69 @@ +/* + * Copyright 2010-2020 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.descriptors.DeclarationDescriptor +import org.jetbrains.kotlin.descriptors.SourceElement +import org.jetbrains.kotlin.descriptors.annotations.Annotations +import org.jetbrains.kotlin.ir.declarations.* +import org.jetbrains.kotlin.ir.descriptors.* +import org.jetbrains.kotlin.ir.symbols.IrBindableSymbol +import org.jetbrains.kotlin.ir.util.IdSignature +import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedContainerSource + +abstract class Fir2IrBindableSymbol( + private val containerSource: DeserializedContainerSource? = null +) : IrBindableSymbol { + + private var _owner: B? = null + override val owner: B + get() = _owner ?: throw IllegalStateException("Symbol is unbound") + + override fun bind(owner: B) { + if (_owner == null) { + _owner = owner + } else { + throw IllegalStateException("${javaClass.simpleName} is already bound") + } + } + + override val isPublicApi: Boolean = false + + override val signature: IdSignature + get() = error("IdSignature is allowed only for PublicApi symbols") + + override val isBound: Boolean + get() = _owner != null + + override val descriptor: D by lazy { + when (val owner = owner) { + is IrEnumEntry -> WrappedEnumEntryDescriptor().apply { bind(owner) } + is IrClass -> WrappedClassDescriptor().apply { bind(owner) } + is IrConstructor -> WrappedClassConstructorDescriptor().apply { bind(owner) } + is IrSimpleFunction -> when { + owner.name.isSpecial && owner.name.asString().startsWith("get") -> + WrappedPropertyGetterDescriptor(Annotations.EMPTY, SourceElement.NO_SOURCE) + owner.name.isSpecial && owner.name.asString().startsWith("set") -> + WrappedPropertySetterDescriptor(Annotations.EMPTY, SourceElement.NO_SOURCE) + else -> if (containerSource != null) { + WrappedFunctionDescriptorWithContainerSource(containerSource) + } else { + WrappedSimpleFunctionDescriptor() + } + }.apply { bind(owner) } + is IrVariable -> WrappedVariableDescriptor().apply { bind(owner) } + is IrValueParameter -> WrappedValueParameterDescriptor().apply { bind(owner) } + is IrTypeParameter -> WrappedTypeParameterDescriptor().apply { bind(owner) } + is IrProperty -> if (containerSource != null) { + WrappedPropertyDescriptorWithContainerSource(containerSource) + } else { + WrappedPropertyDescriptor() + }.apply { bind(owner) } + is IrField -> WrappedFieldDescriptor().apply { bind(owner) } + else -> throw IllegalStateException("Unsupported owner in Fir2IrBindableSymbol: $owner") + } as D + } +}