[kotlin] ReplaceWithImportAliasInspection: fix psi-tree structure

^KTIJ-20184 GitOrigin-RevId: 5c849164240829b0ed48fffc278ee1f6cd52f6aa

[kotlin] ReplaceWithImportAliasInspection: fix psi-tree structure
^KTIJ-20184 GitOrigin-RevId: 5c849164240829b0ed48fffc278ee1f6cd52f6aa
6d2fcadd · Dmitry Gridin · intellij-monorepo-bot · ff05ef18 · 6d2fcadd
Commit 6d2fcadd authored 3 years ago by Dmitry Gridin Committed by intellij-monorepo-bot 3 years ago
Hide whitespace changes
Inline Side-by-side

Showing

with 8 additions and 1 deletion
+8 -1
--- a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/inspections/ReplaceWithImportAliasInspection.kt
+++ b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/inspections/ReplaceWithImportAliasInspection.kt
@@ -50,7 +50,14 @@ class ReplaceWithImportAliasInspection : AbstractKotlinInspection() {
            val expression = descriptor.psiElement as? KtNameReferenceExpression ?: return
            val aliasNameIdentifier = aliasNameIdentifierPointer.element ?: return
            expression.getIdentifier()?.replace(aliasNameIdentifier.copy())
-            expression.getQualifiedElement().replace(expression.parent.safeAs<KtCallExpression>() ?: expression)
+            expression.getQualifiedElement().let { element ->
+                if (element is KtUserType) {
+                    element.referenceExpression?.replace(expression)
+                    element.deleteQualifier()
+                } else {
+                    element.replace(expression.parent.safeAs<KtCallExpression>() ?: expression)
+                }
+            }
        }
    }
 }