diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt index 8beae165279..79fa884442a 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt @@ -62,7 +62,7 @@ abstract class KotlinUAnnotationBase( 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") diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt.182 b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt.182 index 08113a6d2d3..974cd127163 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt.182 +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUAnnotation.kt.182 @@ -60,7 +60,7 @@ abstract class KotlinUAnnotationBase( 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") diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUImportStatement.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUImportStatement.kt index fd8b911b262..a6c8b8ec3ca 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUImportStatement.kt +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/declarations/KotlinUImportStatement.kt @@ -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() } } } \ No newline at end of file diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt index fd722e7fbc2..9a2dcebe3af 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt @@ -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 -> diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt.191 b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt.191 index 5e621b471a9..6ce62734c92 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt.191 +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/expressions/KotlinUSimpleReferenceExpression.kt.191 @@ -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 -> diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinInternalUastUtils.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinInternalUastUtils.kt index b767ff74e77..06b0b9dd73f 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinInternalUastUtils.kt +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinInternalUastUtils.kt @@ -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 { diff --git a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinUastMultiresolve.kt b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinUastMultiresolve.kt index b8c9ef57de8..d0bfeed6550 100644 --- a/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinUastMultiresolve.kt +++ b/plugins/uast-kotlin/src/org/jetbrains/uast/kotlin/internal/kotlinUastMultiresolve.kt @@ -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() }