++fixup
diff --git a/idea/idea-fir-fe10-binding/src/org/jetbrains/kotlin/idea/fir/ShortenReferencesByFirImpl.kt b/idea/idea-fir-fe10-binding/src/org/jetbrains/kotlin/idea/fir/ShortenReferencesByFirImpl.kt
new file mode 100644
index 0000000..e27da1f
--- /dev/null
+++ b/idea/idea-fir-fe10-binding/src/org/jetbrains/kotlin/idea/fir/ShortenReferencesByFirImpl.kt
@@ -0,0 +1,50 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.idea.fir
+
+import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.application.runWriteAction
+import com.intellij.openapi.util.TextRange
+import org.jetbrains.kotlin.idea.core.ShortenReferences
+import org.jetbrains.kotlin.idea.util.ActionRunningMode
+import org.jetbrains.kotlin.psi.KtElement
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.idea.frontend.api.analyse
+import org.jetbrains.kotlin.idea.util.runAction
+
+class ShortenReferencesByFirImpl : ShortenReferences {
+
+ override fun process(element: KtElement, actionRunningMode: ActionRunningMode) {
+ val referenceShortenings = runReadAction {
+ analyse(element) {
+ collectPossibleReferenceShortenings(element.containingKtFile, element.textRange)
+ }
+ }
+ runWriteAction {
+ actionRunningMode.runAction { referenceShortenings.invokeShortening() }
+ }
+ }
+
+ override fun process(elements: Collection<KtElement>, actionRunningMode: ActionRunningMode) {
+ runReadAction { elements.groupBy(KtElement::getContainingKtFile) }
+ .forEach { shortenReferencesInFile(it.key, it.value, actionRunningMode) }
+ }
+
+ private fun shortenReferencesInFile(file: KtFile, elements: List<KtElement>, actionRunningMode: ActionRunningMode) {
+ process(file, elements.minOf { it.textRange.startOffset }, elements.maxOf { it.textRange.endOffset }, actionRunningMode)
+ }
+
+ override fun process(file: KtFile, startOffset: Int, endOffset: Int, actionRunningMode: ActionRunningMode) {
+ val referenceShortenings = runReadAction {
+ analyse(file) {
+ collectPossibleReferenceShortenings(file, TextRange(startOffset, endOffset))
+ }
+ }
+ runWriteAction {
+ actionRunningMode.runAction { referenceShortenings.invokeShortening() }
+ }
+ }
+}
\ No newline at end of file
