azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Analysis API: introduce annotation value for KClass

Browse Source
This commit is contained in:
Ilya Kirillov
2021-11-24 12:38:03 +01:00
parent f722a54c78
commit 673459580c
16 changed files with 86 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/evaluate/FirAnnotationValueConverter.kt
+12 -3
View File
@@ -78,6 +78,7 @@ internal object FirAnnotationValueConverter {
private fun FirExpression.convertConstantExpression(
session: FirSession,
): KtAnnotationValue? {
val sourcePsi = psi as? KtElement
return when (this) {
is FirConstExpression<*> -> convertConstantExpression()
is FirNamedArgumentExpression -> {
@@ -87,11 +88,11 @@ internal object FirAnnotationValueConverter {
expression.convertConstantExpression(session)
}
is FirVarargArgumentsExpression -> {
arguments.convertConstantExpression(session).toArrayConstantValueIfNecessary(psi as? KtElement)
arguments.convertConstantExpression(session).toArrayConstantValueIfNecessary(sourcePsi)
}
is FirArrayOfCall -> {
// Desugared collection literals.
KtArrayAnnotationValue(argumentList.arguments.convertConstantExpression(session), psi as? KtElement)
KtArrayAnnotationValue(argumentList.arguments.convertConstantExpression(session), sourcePsi)
}
is FirFunctionCall -> {
val reference = calleeReference as? FirResolvedNamedReference ?: return null
@@ -126,11 +127,19 @@ internal object FirAnnotationValueConverter {
val reference = calleeReference as? FirResolvedNamedReference ?: return null
when (val resolvedSymbol = reference.resolvedSymbol) {
is FirEnumEntrySymbol -> {
KtEnumEntryAnnotationValue(resolvedSymbol.callableId, psi as? KtElement)
KtEnumEntryAnnotationValue(resolvedSymbol.callableId, sourcePsi)
}
else -> null
}
}
is FirGetClassCall -> {
val symbol = (argument as FirResolvedQualifier).symbol
when {
symbol == null -> KtErrorClassAnnotationValue(sourcePsi)
symbol.classId.isLocal -> KtLocalKClassAnnotationValue(symbol.fir.psi as KtClassOrObject, sourcePsi)
else -> KtNonLocalKClassAnnotationValue(symbol.classId, sourcePsi)
}
}
else -> null
} ?: FirCompileTimeConstantEvaluator.evaluate(this)?.convertConstantExpression()
}