From b6ee47819077c473fff5e4247fc4cef8d8e02835 Mon Sep 17 00:00:00 2001 From: Roman Golyshev Date: Thu, 6 Aug 2020 00:48:09 +0300 Subject: [PATCH] FIR Completion: Add psi element to the lookups - May be a performance problem later --- .../idea/completion/HighLevelApiLookupElementFactory.kt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/completion/HighLevelApiLookupElementFactory.kt b/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/completion/HighLevelApiLookupElementFactory.kt index 8df4fbfdc03..097478fa7fc 100644 --- a/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/completion/HighLevelApiLookupElementFactory.kt +++ b/idea/idea-frontend-fir/src/org/jetbrains/kotlin/idea/completion/HighLevelApiLookupElementFactory.kt @@ -30,7 +30,9 @@ internal class HighLevelApiLookupElementFactory { else -> throw IllegalArgumentException("Cannot create a lookup element for $symbol") } - return elementBuilder.withIcon(KotlinSymbolIconProvider.getIconFor(symbol)) + return elementBuilder + .withPsiElement(symbol.psi) // TODO check if it is a heavy operation and should be postponed + .withIcon(KotlinSymbolIconProvider.getIconFor(symbol)) } }