Introduce Variable: Move caret to the end of declaration when replacing "statement" expression

#KT-10663 Fixed

idea/src/org/jetbrains/kotlin/idea/refactoring/introduce/introduceVariable/KotlinIntroduceVariableHandler.kt

@@ -500,16 +500,6 @@ object KotlinIntroduceVariableHandler : RefactoringActionHandler {
 
         val allOccurrences = occurrencesToReplace ?: expression.findOccurrences(occurrenceContainer)
 
-        fun postProcess(declaration: KtDeclaration) {
-            if (typeArgumentList == null) return
-            val initializer = when (declaration) {
-                is KtProperty -> declaration.initializer
-                is KtDestructuringDeclaration -> declaration.initializer
-                else -> null
-            } ?: return
-            runWriteAction { addTypeArgumentsIfNeeded(initializer, typeArgumentList) }
-        }
-
         val callback = Pass<OccurrencesChooser.ReplaceChoice> { replaceChoice ->
             val allReplaces = when (replaceChoice) {
                 OccurrencesChooser.ReplaceChoice.ALL -> allOccurrences
@@ -525,6 +515,21 @@ object KotlinIntroduceVariableHandler : RefactoringActionHandler {
                 commonContainer = container
             }
 
+            fun postProcess(declaration: KtDeclaration) {
+                if (typeArgumentList != null) {
+                    val initializer = when (declaration) {
+                                          is KtProperty -> declaration.initializer
+                                          is KtDestructuringDeclaration -> declaration.initializer
+                                          else -> null
+                                      } ?: return
+                    runWriteAction { addTypeArgumentsIfNeeded(initializer, typeArgumentList) }
+                }
+
+                if (editor != null && !replaceOccurrence) {
+                    editor.caretModel.moveToOffset(declaration.endOffset)
+                }
+            }
+
             physicalExpression.chooseApplicableComponentFunctionsForVariableDeclaration(replaceOccurrence, editor) { componentFunctions ->
                 val validator = NewDeclarationNameValidator(
                         commonContainer,

idea/src/org/jetbrains/kotlin/idea/refactoring/introduce/introduceVariable/KotlinVariableInplaceIntroducer.kt

@@ -155,6 +155,12 @@ class KotlinVariableInplaceIntroducer(
                 it.replace(replacement)
             }
         }
-        postProcess(addedVariable)
+    }
+
+    override fun moveOffsetAfter(success: Boolean) {
+        super.moveOffsetAfter(success)
+        if (success) {
+            postProcess(addedVariable)
+        }
     }
 }