From d230afe726849f68ee0db815d8e9ac48f4ea1a3d Mon Sep 17 00:00:00 2001
From: Kirill Rakhman <kirill.rakhman@gmail.com>
Date: Wed, 23 Mar 2016 15:24:17 +0100
Subject: [PATCH] Reuse BindingContext and perform quickfix directly on PSI
 instead of the editor

---
 .../kotlin/idea/quickfix/ConvertClassToKClassFix.kt    | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/quickfix/ConvertClassToKClassFix.kt b/idea/src/org/jetbrains/kotlin/idea/quickfix/ConvertClassToKClassFix.kt
index ee634bda1da..63846702af0 100644
--- a/idea/src/org/jetbrains/kotlin/idea/quickfix/ConvertClassToKClassFix.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/quickfix/ConvertClassToKClassFix.kt
@@ -29,8 +29,6 @@ import org.jetbrains.kotlin.name.FqName
 import org.jetbrains.kotlin.psi.KtDotQualifiedExpression
 import org.jetbrains.kotlin.psi.KtExpression
 import org.jetbrains.kotlin.psi.KtFile
-import org.jetbrains.kotlin.psi.psiUtil.endOffset
-import org.jetbrains.kotlin.psi.psiUtil.startOffset
 import org.jetbrains.kotlin.resolve.descriptorUtil.fqNameSafe
 import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
 import org.jetbrains.kotlin.types.KotlinType
@@ -51,21 +49,21 @@ class ConvertClassToKClassFix(element: KtDotQualifiedExpression, private val typ
     override fun isAvailable(project: Project, editor: Editor?, file: PsiFile): Boolean {
         if (!super.isAvailable(project, editor, file)) return false
 
-        val expressionType = element.analyze(BodyResolveMode.PARTIAL).getType(element) ?: return false
+        val bindingContext = element.analyze(BodyResolveMode.PARTIAL)
+        val expressionType = bindingContext.getType(element) ?: return false
         if (!expressionType.isJClass()) return false
 
         val children = element.children
         if (children.size != 2) return false
 
         val firstChild = children.first() as? KtExpression ?: return false
-        val firstChildType = firstChild.analyze(BodyResolveMode.PARTIAL).getType(firstChild) ?: return false
+        val firstChildType = bindingContext.getType(firstChild) ?: return false
 
         return firstChildType.isSubtypeOf(type)
     }
 
     override fun invoke(project: Project, editor: Editor?, file: KtFile) {
-        val lastChild = element.children.lastOrNull() ?: return
-        editor?.document?.replaceString(lastChild.startOffset - 1, lastChild.endOffset, "")
+        element.replace(element.firstChild)
     }
 
     object Factory : KotlinIntentionActionsFactory() {