From a2ca3fd2e3352f177838fa79a94f2020e3d17fdf Mon Sep 17 00:00:00 2001
From: Marat Dinmukhametov <marat.dinmukhametov@jetbrains.com>
Date: Wed, 25 Sep 2024 20:46:59 +0300
Subject: [PATCH] [kotlin] IDEA-353900 k2: ShortenReference for uast

This is necessary for the Reactive-streams plugin, which uses shortening

GitOrigin-RevId: fae6588709134846da0c279e372f59505c7530e0
---
 .../uast-kotlin-fir/kotlin.uast.uast-kotlin.k2.iml  |  1 +
 .../kotlin/FirKotlinUastCodeGenerationPlugin.kt     | 13 ++++---------
 2 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/plugins/kotlin/uast/uast-kotlin-fir/kotlin.uast.uast-kotlin.k2.iml b/plugins/kotlin/uast/uast-kotlin-fir/kotlin.uast.uast-kotlin.k2.iml
index 27c1137c03ae..ae15435c02d9 100644
--- a/plugins/kotlin/uast/uast-kotlin-fir/kotlin.uast.uast-kotlin.k2.iml
+++ b/plugins/kotlin/uast/uast-kotlin-fir/kotlin.uast.uast-kotlin.k2.iml
@@ -37,5 +37,6 @@
     <orderEntry type="library" name="kotlinc.low-level-api-fir" level="project" />
     <orderEntry type="library" name="kotlinc.analysis-api-platform-interface" level="project" />
     <orderEntry type="library" name="kotlinc.symbol-light-classes" level="project" />
+    <orderEntry type="module" module-name="kotlin.base.analysis-api.utils" />
   </component>
 </module>
\ No newline at end of file
diff --git a/plugins/kotlin/uast/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinUastCodeGenerationPlugin.kt b/plugins/kotlin/uast/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinUastCodeGenerationPlugin.kt
index fe9b48c62c47..a13cf352d649 100644
--- a/plugins/kotlin/uast/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinUastCodeGenerationPlugin.kt
+++ b/plugins/kotlin/uast/uast-kotlin-fir/src/org/jetbrains/uast/kotlin/FirKotlinUastCodeGenerationPlugin.kt
@@ -2,18 +2,13 @@
 package org.jetbrains.uast.kotlin
 
 import com.intellij.psi.PsiElement
+import org.jetbrains.kotlin.idea.base.analysis.api.utils.shortenReferences
 import org.jetbrains.kotlin.psi.KtElement
 import org.jetbrains.uast.kotlin.generate.KotlinUastBaseCodeGenerationPlugin
-import org.jetbrains.uast.kotlin.internal.analyzeForUast
 
 abstract class FirKotlinUastCodeGenerationPlugin : KotlinUastBaseCodeGenerationPlugin() {
-  override fun shortenReference(sourcePsi: KtElement): PsiElement {
-    val ktFile = sourcePsi.containingKtFile
-    analyzeForUast(ktFile) {
-        collectPossibleReferenceShortenings(ktFile, sourcePsi.textRange)
+    override fun shortenReference(sourcePsi: KtElement): PsiElement? {
+        shortenReferences(sourcePsi)
+        return sourcePsi
     }
-
-    //todo apply shortening
-    return sourcePsi
-  }
 }
\ No newline at end of file