From 0099b7b3b149d1c564f39f376b51862c6f4e15b7 Mon Sep 17 00:00:00 2001
From: Toshiaki Kameyama <kmymtsak@gmail.com>
Date: Wed, 8 Aug 2018 19:58:57 +0900
Subject: [PATCH] "Remove explicit type specification" intention: Add type
 argument to initializer if need #KT-13343 Fixed

---
 .../idea/intentions/RemoveExplicitTypeIntention.kt     |  5 +++++
 .../intentions/removeExplicitType/needTypeArgument.kt  |  2 ++
 .../removeExplicitType/needTypeArgument.kt.after       |  2 ++
 .../intentions/removeExplicitType/needTypeArgument2.kt |  2 ++
 .../removeExplicitType/needTypeArgument2.kt.after      |  2 ++
 .../kotlin/idea/intentions/IntentionTestGenerated.java | 10 ++++++++++
 6 files changed, 23 insertions(+)
 create mode 100644 idea/testData/intentions/removeExplicitType/needTypeArgument.kt
 create mode 100644 idea/testData/intentions/removeExplicitType/needTypeArgument.kt.after
 create mode 100644 idea/testData/intentions/removeExplicitType/needTypeArgument2.kt
 create mode 100644 idea/testData/intentions/removeExplicitType/needTypeArgument2.kt.after

diff --git a/idea/src/org/jetbrains/kotlin/idea/intentions/RemoveExplicitTypeIntention.kt b/idea/src/org/jetbrains/kotlin/idea/intentions/RemoveExplicitTypeIntention.kt
index bac14c6550f..a60fb4d974a 100644
--- a/idea/src/org/jetbrains/kotlin/idea/intentions/RemoveExplicitTypeIntention.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/intentions/RemoveExplicitTypeIntention.kt
@@ -19,6 +19,8 @@ package org.jetbrains.kotlin.idea.intentions
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.util.TextRange
 import org.jetbrains.kotlin.descriptors.FunctionDescriptor
+import org.jetbrains.kotlin.idea.refactoring.addTypeArgumentsIfNeeded
+import org.jetbrains.kotlin.idea.refactoring.getQualifiedTypeArgumentList
 import org.jetbrains.kotlin.idea.search.usagesSearch.descriptor
 import org.jetbrains.kotlin.psi.*
 import org.jetbrains.kotlin.psi.psiUtil.endOffset
@@ -35,7 +37,10 @@ class RemoveExplicitTypeIntention : SelfTargetingRangeIntention<KtCallableDeclar
     }
 
     override fun applyTo(element: KtCallableDeclaration, editor: Editor?) {
+        val initializer = (element as? KtProperty)?.initializer
+        val typeArgumentList = initializer?.let { getQualifiedTypeArgumentList(it) }
         element.typeReference = null
+        if (typeArgumentList != null) addTypeArgumentsIfNeeded(initializer, typeArgumentList)
     }
 
     companion object {
diff --git a/idea/testData/intentions/removeExplicitType/needTypeArgument.kt b/idea/testData/intentions/removeExplicitType/needTypeArgument.kt
new file mode 100644
index 00000000000..05b0f655f52
--- /dev/null
+++ b/idea/testData/intentions/removeExplicitType/needTypeArgument.kt
@@ -0,0 +1,2 @@
+// WITH_RUNTIME
+val x: <caret>Set<Int> = setOf()
\ No newline at end of file
diff --git a/idea/testData/intentions/removeExplicitType/needTypeArgument.kt.after b/idea/testData/intentions/removeExplicitType/needTypeArgument.kt.after
new file mode 100644
index 00000000000..55344e969ab
--- /dev/null
+++ b/idea/testData/intentions/removeExplicitType/needTypeArgument.kt.after
@@ -0,0 +1,2 @@
+// WITH_RUNTIME
+val x = setOf<Int>()
\ No newline at end of file
diff --git a/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt b/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt
new file mode 100644
index 00000000000..a64653649f0
--- /dev/null
+++ b/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt
@@ -0,0 +1,2 @@
+class Foo<T, U>
+val foo: <caret>Foo<Int, String> = Foo()
\ No newline at end of file
diff --git a/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt.after b/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt.after
new file mode 100644
index 00000000000..5d1dc11e6b5
--- /dev/null
+++ b/idea/testData/intentions/removeExplicitType/needTypeArgument2.kt.after
@@ -0,0 +1,2 @@
+class Foo<T, U>
+val foo = Foo<Int, String>()
\ No newline at end of file
diff --git a/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java b/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java
index 6528eb35547..64c02d9b4cf 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java
+++ b/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java
@@ -13146,6 +13146,16 @@ public class IntentionTestGenerated extends AbstractIntentionTest {
             runTest("idea/testData/intentions/removeExplicitType/lambdaInitializer5.kt");
         }
 
+        @TestMetadata("needTypeArgument.kt")
+        public void testNeedTypeArgument() throws Exception {
+            runTest("idea/testData/intentions/removeExplicitType/needTypeArgument.kt");
+        }
+
+        @TestMetadata("needTypeArgument2.kt")
+        public void testNeedTypeArgument2() throws Exception {
+            runTest("idea/testData/intentions/removeExplicitType/needTypeArgument2.kt");
+        }
+
         @TestMetadata("notOnParameterOfFunctionType.kt")
         public void testNotOnParameterOfFunctionType() throws Exception {
             runTest("idea/testData/intentions/removeExplicitType/notOnParameterOfFunctionType.kt");