Return updated element after reference shortening

Fresh new expression obtained after qualified expression binding is
invalidated after shortening.
This commit is contained in:
Nikolay Krasko
2018-09-25 13:24:33 +03:00
parent c4283de9cb
commit 18c181641f
@@ -178,16 +178,18 @@ class KtSimpleNameReference(expression: KtSimpleNameExpression) : KtSimpleRefere
if (shorteningMode == ShorteningMode.NO_SHORTENING) return newExpression
val needToShorten =
PsiTreeUtil.getParentOfType(expression, KtImportDirective::class.java, KtPackageDirective::class.java) == null
if (needToShorten) {
when (shorteningMode) {
ShorteningMode.FORCED_SHORTENING -> ShortenReferences.DEFAULT.process(newQualifiedElement)
else -> newQualifiedElement.addToShorteningWaitSet()
}
val needToShorten = PsiTreeUtil.getParentOfType(expression, KtImportDirective::class.java, KtPackageDirective::class.java) == null
if (!needToShorten) {
return newExpression
}
return newExpression
return when (shorteningMode) {
ShorteningMode.FORCED_SHORTENING -> ShortenReferences.DEFAULT.process(newQualifiedElement)
else -> {
newQualifiedElement.addToShorteningWaitSet()
newExpression
}
}
}
/**