org.jetbrains.kotlin.fir.builder.ConversionUtils.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation
Show all versions of kotlin-compiler-embeddable Show documentation
the Kotlin compiler embeddable
/*
* Copyright 2010-2024 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.builder
import com.intellij.psi.tree.IElementType
import org.jetbrains.kotlin.*
import org.jetbrains.kotlin.contracts.description.LogicOperationKind
import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.descriptors.Visibilities
import org.jetbrains.kotlin.descriptors.annotations.AnnotationUseSiteTarget
import org.jetbrains.kotlin.fir.*
import org.jetbrains.kotlin.fir.contracts.FirContractDescription
import org.jetbrains.kotlin.fir.contracts.FirLegacyRawContractDescription
import org.jetbrains.kotlin.fir.contracts.builder.buildLegacyRawContractDescription
import org.jetbrains.kotlin.fir.declarations.*
import org.jetbrains.kotlin.fir.declarations.builder.*
import org.jetbrains.kotlin.fir.declarations.impl.FirDeclarationStatusImpl
import org.jetbrains.kotlin.fir.declarations.impl.FirDefaultPropertyAccessor
import org.jetbrains.kotlin.fir.diagnostics.ConeDiagnostic
import org.jetbrains.kotlin.fir.diagnostics.DiagnosticKind
import org.jetbrains.kotlin.fir.expressions.*
import org.jetbrains.kotlin.fir.expressions.builder.*
import org.jetbrains.kotlin.fir.expressions.impl.FirContractCallBlock
import org.jetbrains.kotlin.fir.expressions.impl.FirSingleExpressionBlock
import org.jetbrains.kotlin.fir.references.FirNamedReference
import org.jetbrains.kotlin.fir.references.builder.buildDelegateFieldReference
import org.jetbrains.kotlin.fir.references.builder.buildImplicitThisReference
import org.jetbrains.kotlin.fir.references.builder.buildResolvedNamedReference
import org.jetbrains.kotlin.fir.references.builder.buildSimpleNamedReference
import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirDelegateFieldSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirPropertyAccessorSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirValueParameterSymbol
import org.jetbrains.kotlin.fir.types.*
import org.jetbrains.kotlin.fir.types.builder.buildResolvedTypeRef
import org.jetbrains.kotlin.fir.types.builder.buildTypeProjectionWithVariance
import org.jetbrains.kotlin.fir.types.impl.ConeTypeParameterTypeImpl
import org.jetbrains.kotlin.fir.types.impl.FirImplicitTypeRefImplWithoutSource
import org.jetbrains.kotlin.lexer.KtTokens
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.name.SpecialNames
import org.jetbrains.kotlin.name.StandardClassIds
import org.jetbrains.kotlin.types.ConstantValueKind
import org.jetbrains.kotlin.types.Variance
import org.jetbrains.kotlin.types.expressions.OperatorConventions
import org.jetbrains.kotlin.util.OperatorNameConventions
import kotlin.contracts.ExperimentalContracts
import kotlin.contracts.contract
fun String.parseCharacter(): CharacterWithDiagnostic {
// Strip the quotes
if (length < 2 || this[0] != '\'' || this[length - 1] != '\'') {
return CharacterWithDiagnostic(DiagnosticKind.IncorrectCharacterLiteral)
}
val text = substring(1, length - 1) // now there is no quotes
if (text.isEmpty()) {
return CharacterWithDiagnostic(DiagnosticKind.EmptyCharacterLiteral)
}
return if (text[0] != '\\') {
// No escape
if (text.length == 1) {
CharacterWithDiagnostic(text[0])
} else {
CharacterWithDiagnostic(DiagnosticKind.TooManyCharactersInCharacterLiteral)
}
} else {
escapedStringToCharacter(text)
}
}
fun escapedStringToCharacter(text: String): CharacterWithDiagnostic {
assert(text.isNotEmpty() && text[0] == '\\') {
"Only escaped sequences must be passed to this routine: $text"
}
// Escape
val escape = text.substring(1) // strip the slash
when (escape.length) {
0 -> {
// bare slash
return CharacterWithDiagnostic(DiagnosticKind.IllegalEscape)
}
1 -> {
// one-char escape
return translateEscape(escape[0])
}
5 -> {
// unicode escape
if (escape[0] == 'u') {
val intValue = escape.substring(1).toIntOrNull(16)
// If error occurs it will be reported below
if (intValue != null) {
return CharacterWithDiagnostic(intValue.toChar())
}
}
}
}
return CharacterWithDiagnostic(DiagnosticKind.IllegalEscape)
}
internal fun translateEscape(c: Char): CharacterWithDiagnostic =
when (c) {
't' -> CharacterWithDiagnostic('\t')
'b' -> CharacterWithDiagnostic('\b')
'n' -> CharacterWithDiagnostic('\n')
'r' -> CharacterWithDiagnostic('\r')
'\'' -> CharacterWithDiagnostic('\'')
'\"' -> CharacterWithDiagnostic('\"')
'\\' -> CharacterWithDiagnostic('\\')
'$' -> CharacterWithDiagnostic('$')
else -> CharacterWithDiagnostic(DiagnosticKind.IllegalEscape)
}
class CharacterWithDiagnostic {
private val diagnostic: DiagnosticKind?
val value: Char?
constructor(diagnostic: DiagnosticKind) {
this.diagnostic = diagnostic
this.value = null
}
constructor(value: Char) {
this.diagnostic = null
this.value = value
}
fun getDiagnostic(): DiagnosticKind? {
return diagnostic
}
}
fun IElementType.toBinaryName(): Name? {
return OperatorConventions.BINARY_OPERATION_NAMES[this]
}
fun IElementType.toUnaryName(): Name? {
return OperatorConventions.UNARY_OPERATION_NAMES[this]
}
fun IElementType.toFirOperation(): FirOperation =
toFirOperationOrNull() ?: error("Cannot convert element type to FIR operation: $this")
fun IElementType.toFirOperationOrNull(): FirOperation? =
when (this) {
KtTokens.LT -> FirOperation.LT
KtTokens.GT -> FirOperation.GT
KtTokens.LTEQ -> FirOperation.LT_EQ
KtTokens.GTEQ -> FirOperation.GT_EQ
KtTokens.EQEQ -> FirOperation.EQ
KtTokens.EXCLEQ -> FirOperation.NOT_EQ
KtTokens.EQEQEQ -> FirOperation.IDENTITY
KtTokens.EXCLEQEQEQ -> FirOperation.NOT_IDENTITY
KtTokens.EQ -> FirOperation.ASSIGN
KtTokens.PLUSEQ -> FirOperation.PLUS_ASSIGN
KtTokens.MINUSEQ -> FirOperation.MINUS_ASSIGN
KtTokens.MULTEQ -> FirOperation.TIMES_ASSIGN
KtTokens.DIVEQ -> FirOperation.DIV_ASSIGN
KtTokens.PERCEQ -> FirOperation.REM_ASSIGN
KtTokens.AS_KEYWORD -> FirOperation.AS
KtTokens.AS_SAFE -> FirOperation.SAFE_AS
else -> null
}
fun FirExpression.generateNotNullOrOther(
other: FirExpression, baseSource: KtSourceElement?,
): FirElvisExpression {
return buildElvisExpression {
source = baseSource
lhs = this@generateNotNullOrOther
rhs = other
}
}
fun FirExpression.generateLazyLogicalOperation(
other: FirExpression, isAnd: Boolean, baseSource: KtSourceElement?,
): FirBooleanOperatorExpression {
return buildBooleanOperatorExpression {
source = baseSource
leftOperand = this@generateLazyLogicalOperation
rightOperand = other
kind = if (isAnd) LogicOperationKind.AND else LogicOperationKind.OR
}
}
fun FirExpression.generateContainsOperation(
argument: FirExpression,
inverted: Boolean,
baseSource: KtSourceElement?,
operationReferenceSource: KtSourceElement?
): FirFunctionCall {
val resultReferenceSource = if (inverted)
operationReferenceSource?.fakeElement(KtFakeSourceElementKind.DesugaredInvertedContains)
else
operationReferenceSource
val containsCall = createConventionCall(resultReferenceSource, baseSource, argument, OperatorNameConventions.CONTAINS)
if (!inverted) return containsCall
return buildFunctionCall {
source = baseSource?.fakeElement(KtFakeSourceElementKind.DesugaredInvertedContains)
calleeReference = buildSimpleNamedReference {
source = resultReferenceSource
name = OperatorNameConventions.NOT
}
explicitReceiver = containsCall
origin = FirFunctionCallOrigin.Operator
}
}
fun FirExpression.generateComparisonExpression(
argument: FirExpression,
operatorToken: IElementType,
baseSource: KtSourceElement?,
operationReferenceSource: KtSourceElement?,
): FirComparisonExpression {
require(operatorToken in OperatorConventions.COMPARISON_OPERATIONS) {
"$operatorToken is not in ${OperatorConventions.COMPARISON_OPERATIONS}"
}
val compareToCall = createConventionCall(
operationReferenceSource,
baseSource?.fakeElement(KtFakeSourceElementKind.GeneratedComparisonExpression),
argument,
OperatorNameConventions.COMPARE_TO
)
val firOperation = when (operatorToken) {
KtTokens.LT -> FirOperation.LT
KtTokens.GT -> FirOperation.GT
KtTokens.LTEQ -> FirOperation.LT_EQ
KtTokens.GTEQ -> FirOperation.GT_EQ
else -> error("Unknown $operatorToken")
}
return buildComparisonExpression {
this.source = baseSource
this.operation = firOperation
this.compareToCall = compareToCall
}
}
private fun FirExpression.createConventionCall(
operationReferenceSource: KtSourceElement?,
baseSource: KtSourceElement?,
argument: FirExpression,
conventionName: Name
): FirFunctionCall {
return buildFunctionCall {
source = baseSource
calleeReference = buildSimpleNamedReference {
source = operationReferenceSource
name = conventionName
}
explicitReceiver = this@createConventionCall
argumentList = buildUnaryArgumentList(argument)
origin = FirFunctionCallOrigin.Operator
}
}
fun generateAccessExpression(
qualifiedSource: KtSourceElement?,
calleeReferenceSource: KtSourceElement?,
name: Name,
diagnostic: ConeDiagnostic? = null
): FirQualifiedAccessExpression =
buildPropertyAccessExpression {
this.source = qualifiedSource
calleeReference = buildSimpleNamedReference {
this.source = if (calleeReferenceSource == qualifiedSource)
calleeReferenceSource?.fakeElement(KtFakeSourceElementKind.ReferenceInAtomicQualifiedAccess)
else
calleeReferenceSource
this.name = name
}
if (diagnostic != null) {
this.nonFatalDiagnostics.add(diagnostic)
}
}
fun generateResolvedAccessExpression(source: KtSourceElement?, variable: FirVariable): FirQualifiedAccessExpression =
buildPropertyAccessExpression {
this.source = source
calleeReference = buildResolvedNamedReference {
this.source = source
name = variable.name
resolvedSymbol = variable.symbol
}
}
fun FirVariable.toComponentCall(
entrySource: KtSourceElement?,
index: Int,
): FirComponentCall {
return buildComponentCall {
val componentCallSource = entrySource?.fakeElement(KtFakeSourceElementKind.DesugaredComponentFunctionCall)
source = componentCallSource
explicitReceiver = generateResolvedAccessExpression(componentCallSource, this@toComponentCall)
componentIndex = index + 1
}
}
val FirClassBuilder.ownerRegularOrAnonymousObjectSymbol: FirClassSymbol
get() = when (this) {
is FirAnonymousObjectBuilder -> symbol
is FirRegularClassBuilder -> symbol
}
fun FirPropertyBuilder.generateAccessorsByDelegate(
delegateBuilder: FirWrappedDelegateExpressionBuilder?,
moduleData: FirModuleData,
ownerRegularOrAnonymousObjectSymbol: FirClassSymbol<*>?,
context: Context,
isExtension: Boolean,
lazyDelegateExpression: FirLazyExpression? = null,
lazyBodyForGeneratedAccessors: FirLazyBlock? = null,
bindFunction: (target: FirFunctionTarget, function: FirFunction) -> Unit = FirFunctionTarget::bind,
) {
if (delegateBuilder == null) return
val delegateFieldSymbol = FirDelegateFieldSymbol(symbol.callableId).also {
this.delegateFieldSymbol = it
}
val isMember = ownerRegularOrAnonymousObjectSymbol != null
val fakeSource = delegateBuilder.source?.fakeElement(KtFakeSourceElementKind.DelegatedPropertyAccessor)
/*
* If we have delegation with provide delegate then we generate call like
* `delegateExpression.provideDelegate(this, ::prop)`
* Note that `this` is always reference for dispatch receiver
* unlike other `this` references in `getValue` `setValue` calls, where
* `this` is reference to closest receiver (extension, then dispatch)
*
* So for top-level extension properties we should generate
* val A.prop by delegateExpression.provideDelegate(null, ::prop)
* get() = delegate.getValue(this@prop, ::prop)
*
* And for this case we can pass isForDelegateProviderCall to this reference
* generator function
*/
fun thisRef(forDispatchReceiver: Boolean = false): FirExpression =
when {
isExtension && !forDispatchReceiver -> buildThisReceiverExpression {
source = fakeSource
calleeReference = buildImplicitThisReference {
boundSymbol = [email protected]
}
}
ownerRegularOrAnonymousObjectSymbol != null -> buildThisReceiverExpression {
source = fakeSource
calleeReference = buildImplicitThisReference {
boundSymbol = ownerRegularOrAnonymousObjectSymbol
}
coneTypeOrNull = context.dispatchReceiverTypesStack.last()
}
else -> buildLiteralExpression(null, ConstantValueKind.Null, null, setType = false)
}
fun delegateAccess() = buildPropertyAccessExpression {
source = fakeSource
calleeReference = buildDelegateFieldReference {
source = fakeSource
resolvedSymbol = delegateFieldSymbol
}
if (ownerRegularOrAnonymousObjectSymbol != null) {
dispatchReceiver = thisRef(forDispatchReceiver = true)
}
}
val isVar = [email protected]
fun propertyRef() = buildCallableReferenceAccess {
source = fakeSource
calleeReference = buildResolvedNamedReference {
source = fakeSource
name = [email protected]
resolvedSymbol = [email protected]
}
coneTypeOrNull = when {
!isMember && !isExtension -> if (isVar) {
StandardClassIds.KMutableProperty0.constructClassLikeType(arrayOf(ConeStarProjection))
} else {
StandardClassIds.KProperty0.constructClassLikeType(arrayOf(ConeStarProjection))
}
isMember && isExtension -> if (isVar) {
StandardClassIds.KMutableProperty2.constructClassLikeType(
arrayOf(
ConeStarProjection,
ConeStarProjection,
ConeStarProjection
)
)
} else {
StandardClassIds.KProperty2.constructClassLikeType(arrayOf(ConeStarProjection, ConeStarProjection, ConeStarProjection))
}
else -> if (isVar) {
StandardClassIds.KMutableProperty1.constructClassLikeType(arrayOf(ConeStarProjection, ConeStarProjection))
} else {
StandardClassIds.KProperty1.constructClassLikeType(arrayOf(ConeStarProjection, ConeStarProjection))
}
}
[email protected](typeArguments) {
buildTypeProjectionWithVariance {
source = fakeSource
variance = Variance.INVARIANT
typeRef = buildResolvedTypeRef {
coneType = ConeTypeParameterTypeImpl(it.symbol.toLookupTag(), false)
}
}
}
}
delegate = lazyDelegateExpression ?: run {
delegateBuilder.provideDelegateCall = buildFunctionCall {
explicitReceiver = delegateBuilder.expression
calleeReference = buildSimpleNamedReference {
source = fakeSource
name = OperatorNameConventions.PROVIDE_DELEGATE
}
argumentList = buildBinaryArgumentList(thisRef(forDispatchReceiver = true), propertyRef())
origin = FirFunctionCallOrigin.Operator
source = fakeSource
}
delegateBuilder.build()
}
if (getter == null || getter is FirDefaultPropertyAccessor) {
val annotations = getter?.annotations
val returnTarget = FirFunctionTarget(null, isLambda = false)
val getterStatus = getter?.status
val getterElement = getter?.source?.takeIf {
it.kind == KtRealSourceElementKind
}?.fakeElement(KtFakeSourceElementKind.DelegatedPropertyAccessor) ?: fakeSource
getter = buildPropertyAccessor {
this.source = getterElement
this.moduleData = moduleData
origin = FirDeclarationOrigin.Source
returnTypeRef = FirImplicitTypeRefImplWithoutSource
isGetter = true
status = FirDeclarationStatusImpl(getterStatus?.visibility ?: Visibilities.Unknown, Modality.FINAL).apply {
isInline = getterStatus?.isInline ?: isInline
}
symbol = FirPropertyAccessorSymbol()
body = lazyBodyForGeneratedAccessors ?: FirSingleExpressionBlock(
buildReturnExpression {
result = buildFunctionCall {
source = fakeSource
explicitReceiver = delegateAccess()
calleeReference = buildSimpleNamedReference {
source = fakeSource
name = OperatorNameConventions.GET_VALUE
}
argumentList = buildBinaryArgumentList(thisRef(), propertyRef())
origin = FirFunctionCallOrigin.Operator
}
target = returnTarget
source = fakeSource
}
)
if (annotations != null) {
this.annotations.addAll(annotations)
}
propertySymbol = [email protected]
}.also {
bindFunction(returnTarget, it)
it.initContainingClassAttr(context)
}
}
if (isVar && (setter == null || setter is FirDefaultPropertyAccessor)) {
val annotations = setter?.annotations
val returnTarget = FirFunctionTarget(null, isLambda = false)
val parameterAnnotations = setter?.valueParameters?.firstOrNull()?.annotations
val setterStatus = setter?.status
val setterElement = setter?.source?.fakeElement(KtFakeSourceElementKind.DelegatedPropertyAccessor) ?: fakeSource
setter = buildPropertyAccessor {
this.source = setterElement
this.moduleData = moduleData
origin = FirDeclarationOrigin.Source
returnTypeRef = moduleData.session.builtinTypes.unitType
isGetter = false
status = FirDeclarationStatusImpl(setterStatus?.visibility ?: Visibilities.Unknown, Modality.FINAL).apply {
isInline = setterStatus?.isInline ?: isInline
}
symbol = FirPropertyAccessorSymbol()
val parameter = buildValueParameter {
source = fakeSource
containingFunctionSymbol = [email protected]
this.moduleData = moduleData
origin = FirDeclarationOrigin.Source
returnTypeRef = FirImplicitTypeRefImplWithoutSource
name = SpecialNames.IMPLICIT_SET_PARAMETER
symbol = FirValueParameterSymbol([email protected])
isCrossinline = false
isNoinline = false
isVararg = false
if (parameterAnnotations != null) {
this.annotations.addAll(parameterAnnotations)
}
}
valueParameters += parameter
body = lazyBodyForGeneratedAccessors ?: FirSingleExpressionBlock(
buildReturnExpression {
result = buildFunctionCall {
source = fakeSource
explicitReceiver = delegateAccess()
calleeReference = buildSimpleNamedReference {
source = fakeSource
name = OperatorNameConventions.SET_VALUE
}
argumentList = buildArgumentList {
arguments += thisRef()
arguments += propertyRef()
arguments += buildPropertyAccessExpression {
source = fakeSource
calleeReference = buildResolvedNamedReference {
source = fakeSource
name = SpecialNames.IMPLICIT_SET_PARAMETER
resolvedSymbol = parameter.symbol
}
}
}
origin = FirFunctionCallOrigin.Operator
}
target = returnTarget
source = fakeSource
}
)
if (annotations != null) {
this.annotations.addAll(annotations)
}
propertySymbol = [email protected]
}.also {
bindFunction(returnTarget, it)
it.initContainingClassAttr(context)
}
}
}
fun processLegacyContractDescription(block: FirBlock, diagnostic: ConeDiagnostic?): FirContractDescription? {
if (block.isContractPresentFirCheck()) {
val contractCall = block.replaceFirstStatement { FirContractCallBlock(it) }
return contractCall.toLegacyRawContractDescription(diagnostic)
}
return null
}
fun FirFunctionCall.toLegacyRawContractDescription(diagnostic: ConeDiagnostic? = null): FirLegacyRawContractDescription {
return buildLegacyRawContractDescription {
this.source = [email protected]
this.contractCall = this@toLegacyRawContractDescription
this.diagnostic = diagnostic
}
}
fun FirBlock.isContractPresentFirCheck(): Boolean {
val firstStatement = statements.firstOrNull() ?: return false
return firstStatement.isContractBlockFirCheck()
}
@OptIn(ExperimentalContracts::class)
fun FirStatement.isContractBlockFirCheck(): Boolean {
contract { returns(true) implies (this@isContractBlockFirCheck is FirFunctionCall) }
val contractCall = this as? FirFunctionCall ?: return false
if (contractCall.calleeReference.name.asString() != "contract") return false
val receiver = contractCall.explicitReceiver as? FirQualifiedAccessExpression ?: return true
if (!contractCall.checkReceiver("contracts")) return false
if (!receiver.checkReceiver("kotlin")) return false
val receiverOfReceiver = receiver.explicitReceiver as? FirQualifiedAccessExpression ?: return false
if (receiverOfReceiver.explicitReceiver != null) return false
return true
}
private fun FirExpression.checkReceiver(name: String?): Boolean {
if (this !is FirQualifiedAccessExpression) return false
val receiver = explicitReceiver as? FirQualifiedAccessExpression ?: return false
val receiverName = (receiver.calleeReference as? FirNamedReference)?.name?.asString() ?: return false
return receiverName == name
}
// this = .f(...)
// receiver =
// Returns safe call ?.{ f(...) }
fun FirQualifiedAccessExpression.createSafeCall(receiver: FirExpression, source: KtSourceElement): FirSafeCallExpression {
val checkedSafeCallSubject = buildCheckedSafeCallSubject {
@OptIn(FirContractViolation::class)
this.originalReceiverRef = FirExpressionRef().apply {
bind(receiver)
}
this.source = receiver.source?.fakeElement(KtFakeSourceElementKind.CheckedSafeCallSubject)
}
// If an `invoke` function from a functional type expects a
// receiver, it's still defined as the first value parameter.
// A construction like `1.(fun Int.() = 1)()` means we're calling
// `Function1.invoke(Int)`.
if (this is FirImplicitInvokeCall) {
val newArguments = buildArgumentList {
arguments.add(checkedSafeCallSubject)
arguments.addAll([email protected])
}
replaceArgumentList(newArguments)
} else {
replaceExplicitReceiver(checkedSafeCallSubject)
}
return buildSafeCallExpression {
this.receiver = receiver
@OptIn(FirContractViolation::class)
this.checkedSubjectRef = FirExpressionRef().apply {
bind(checkedSafeCallSubject)
}
this.selector = this@createSafeCall
this.source = source
}
}
fun FirQualifiedAccessExpression.pullUpSafeCallIfNecessary(): FirExpression =
pullUpSafeCallIfNecessary(
FirQualifiedAccessExpression::explicitReceiver,
FirQualifiedAccessExpression::replaceExplicitReceiver
)
// Turns a?.b.f(...) to a?.{ b.f(...) ) -- for any qualified access `.f(...)`
// Other patterns remain unchanged
fun F.pullUpSafeCallIfNecessary(
obtainReceiver: F.() -> FirExpression?,
replaceReceiver: F.(FirExpression) -> Unit,
): FirExpression {
val safeCall = obtainReceiver() as? FirSafeCallExpression ?: return this
val safeCallSelector = safeCall.selector as? FirExpression ?: return this
// (a?.b).f and `(a?.b)[3]` should be left as is
if (safeCall.isChildInParentheses()) return this
replaceReceiver(safeCallSelector)
safeCall.replaceSelector(this)
return safeCall
}
fun FirStatement.isChildInParentheses(): Boolean {
val sourceElement = source ?: error("Nullable source")
return sourceElement.isChildInParentheses()
}
fun KtSourceElement.isChildInParentheses() =
treeStructure.getParent(lighterASTNode)?.tokenType == KtNodeTypes.PARENTHESIZED
fun List.filterUseSiteTarget(target: AnnotationUseSiteTarget): List =
mapNotNull {
if (it.useSiteTarget != target) null
else buildAnnotationCallCopy(it) {
source = it.source?.fakeElement(KtFakeSourceElementKind.FromUseSiteTarget)
}
}
// TODO: avoid mutability KT-55002
fun FirTypeRef.convertToReceiverParameter(): FirReceiverParameter {
val typeRef = this
@Suppress("UNCHECKED_CAST")
return buildReceiverParameter {
source = typeRef.source?.fakeElement(KtFakeSourceElementKind.ReceiverFromType)
annotations += (typeRef.annotations as List).filterUseSiteTarget(AnnotationUseSiteTarget.RECEIVER)
val filteredTypeRefAnnotations = typeRef.annotations.filterNot { it.useSiteTarget == AnnotationUseSiteTarget.RECEIVER }
if (filteredTypeRefAnnotations.size != typeRef.annotations.size) {
typeRef.replaceAnnotations(filteredTypeRefAnnotations)
}
this.typeRef = typeRef
}
}
fun KtSourceElement.asReceiverParameter(): FirReceiverParameter = buildReceiverParameter {
source = [email protected](KtFakeSourceElementKind.ReceiverFromType)
typeRef = FirImplicitTypeRefImplWithoutSource
}
fun FirCallableDeclaration.initContainingClassAttr(context: Context) {
containingClassForStaticMemberAttr = currentDispatchReceiverType(context)?.lookupTag ?: return
}
fun currentDispatchReceiverType(context: Context): ConeClassLikeType? {
return context.dispatchReceiverTypesStack.lastOrNull()
}
val CharSequence.isUnderscore: Boolean
get() = all { it == '_' }
data class CalleeAndReceiver(
val reference: FirNamedReference,
val receiverForInvoke: FirExpression? = null,
)
/**
* Creates balanced tree of OR expressions for given set of conditions
* We do so, to avoid too deep OR-expression structures, that can cause running out of stack while processing
* [conditions] should contain at least one element, otherwise it will cause StackOverflow
*/
fun buildBalancedOrExpressionTree(conditions: List, lower: Int = 0, upper: Int = conditions.lastIndex): FirExpression {
val size = upper - lower + 1
val middle = size / 2 + lower
if (lower == upper) {
return conditions[middle]
}
val leftNode = buildBalancedOrExpressionTree(conditions, lower, middle - 1)
val rightNode = buildBalancedOrExpressionTree(conditions, middle, upper)
return leftNode.generateLazyLogicalOperation(
rightNode,
isAnd = false,
(leftNode.source ?: rightNode.source)?.fakeElement(KtFakeSourceElementKind.WhenCondition)
)
}
fun FirExpression.guardedBy(
guard: FirExpression?,
): FirExpression = when (guard) {
null -> this
else -> this.generateLazyLogicalOperation(
guard,
isAnd = true,
(this.source ?: guard.source)?.fakeElement(KtFakeSourceElementKind.WhenCondition)
)
}
fun AnnotationUseSiteTarget?.appliesToPrimaryConstructorParameter() = this == null ||
this == AnnotationUseSiteTarget.CONSTRUCTOR_PARAMETER ||
this == AnnotationUseSiteTarget.RECEIVER ||
this == AnnotationUseSiteTarget.FILE
fun FirErrorTypeRef.wrapIntoArray(): FirResolvedTypeRef {
val typeRef = this
return buildResolvedTypeRef {
source = typeRef.source
coneType = StandardClassIds.Array.constructClassLikeType(arrayOf(ConeKotlinTypeProjectionOut(typeRef.coneType)))
delegatedTypeRef = typeRef.copyWithNewSourceKind(KtFakeSourceElementKind.ArrayTypeFromVarargParameter)
}
}
fun shouldGenerateDelegatedSuperCall(
isAnySuperCall: Boolean,
isExpectClass: Boolean,
isEnumEntry: Boolean,
hasExplicitDelegatedCalls: Boolean
): Boolean {
if (isAnySuperCall) {
return false
}
if (isExpectClass) {
// Generally, an `expect` class cannot inherit from other expect class.
// However, for the IDE resolution purposes, we keep invalid explicit delegate calls.
return !isEnumEntry && hasExplicitDelegatedCalls
}
return true
}