Uast: getMaybeLightElement cleanup

This commit is contained in:
Nicolay Mitropolsky
2019-03-05 13:37:09 +03:00
parent ec1badf60d
commit f664499708
7 changed files with 8 additions and 8 deletions
@@ -62,7 +62,7 @@ abstract class KotlinUAnnotationBase<T : KtCallElement>(
protected abstract fun computeClassDescriptor(): ClassDescriptor?
override fun resolve(): PsiClass? = computeClassDescriptor()?.toSource()?.getMaybeLightElement(this) as? PsiClass
override fun resolve(): PsiClass? = computeClassDescriptor()?.toSource()?.getMaybeLightElement() as? PsiClass
override fun findAttributeValue(name: String?): UExpression? =
findDeclaredAttributeValue(name) ?: findAttributeDefaultValue(name ?: "value")
@@ -60,7 +60,7 @@ abstract class KotlinUAnnotationBase<T : KtCallElement>(
protected abstract fun computeClassDescriptor(): ClassDescriptor?
override fun resolve(): PsiClass? = computeClassDescriptor()?.toSource()?.getMaybeLightElement(this) as? PsiClass
override fun resolve(): PsiClass? = computeClassDescriptor()?.toSource()?.getMaybeLightElement() as? PsiClass
override fun findAttributeValue(name: String?): UExpression? =
findDeclaredAttributeValue(name) ?: findAttributeDefaultValue(name ?: "value")
@@ -65,7 +65,7 @@ class KotlinUImportStatement(
val reference = psi.getQualifiedElementSelector() as? KtReferenceExpression ?: return null
val bindingContext = reference.analyze()
val referenceTarget = bindingContext[BindingContext.REFERENCE_TARGET, reference] ?: return null
return referenceTarget.toSource()?.getMaybeLightElement(this)
return referenceTarget.toSource()?.getMaybeLightElement()
}
}
}
@@ -216,7 +216,7 @@ class KotlinClassViaConstructorUSimpleReferenceExpression(
private val resolved by lazy {
when (val resultingDescriptor = psi.getResolvedCall(psi.analyze())?.resultingDescriptor) {
is ConstructorDescriptor -> {
resultingDescriptor.constructedClass.toSource()?.getMaybeLightElement(this)
resultingDescriptor.constructedClass.toSource()?.getMaybeLightElement()
?: (resultingDescriptor as? DeserializedCallableMemberDescriptor)?.let { resolveContainingDeserializedClass(psi, it) }
}
is SamConstructorDescriptor ->
@@ -218,7 +218,7 @@ class KotlinClassViaConstructorUSimpleReferenceExpression(
private val resolved by lazy {
when (val resultingDescriptor = psi.getResolvedCall(psi.analyze())?.resultingDescriptor) {
is ConstructorDescriptor -> {
resultingDescriptor.constructedClass.toSource()?.getMaybeLightElement(this)
resultingDescriptor.constructedClass.toSource()?.getMaybeLightElement()
?: (resultingDescriptor as? DeserializedCallableMemberDescriptor)?.let { resolveContainingDeserializedClass(psi, it) }
}
is SamConstructorDescriptor ->
@@ -252,7 +252,7 @@ internal fun KtClassOrObject.toPsiType(): PsiType {
return PsiTypesUtil.getClassType(lightClass)
}
internal fun PsiElement.getMaybeLightElement(context: UElement? = null /* TODO: remove */): PsiElement? {
internal fun PsiElement.getMaybeLightElement(): PsiElement? {
return when (this) {
is KtDeclaration -> {
val lightElement = toLightElements().firstOrNull()
@@ -279,7 +279,7 @@ internal fun KtElement.resolveCallToDeclaration(
resolvedCall.resultingDescriptor
}
return descriptor.toSource()?.getMaybeLightElement(context)
return descriptor.toSource()?.getMaybeLightElement()
}
internal fun KtExpression.unwrapBlockOrParenthesis(): KtExpression {
@@ -34,7 +34,7 @@ internal fun UElement.getResolveResultVariants(ktExpression: KtExpression?): Ite
val referenceVariants = getReferenceVariants(ktExpression, ktExpression.name ?: ktExpression.text)
fun asCandidateInfo(descriptor: DeclarationDescriptor): CandidateInfo? =
descriptor.toSource()?.getMaybeLightElement(this)?.let { CandidateInfo(it, PsiSubstitutor.EMPTY) }
descriptor.toSource()?.getMaybeLightElement()?.let { CandidateInfo(it, PsiSubstitutor.EMPTY) }
return referenceVariants.mapNotNull(::asCandidateInfo).asIterable()
}