From eb12cfd444934a6a7be03b036fb195e3cef697f8 Mon Sep 17 00:00:00 2001
From: Toshiaki Kameyama <kmymtsak@gmail.com>
Date: Wed, 3 Jan 2018 19:27:16 +0900
Subject: [PATCH] KT-18674 Join Lines should join strings (#1305)

* Join Lines should join strings #KT-18674 Fixed

* #KT-18674 Fixed
---
 idea/src/META-INF/plugin.xml                  |  1 +
 .../ConvertToStringTemplateIntention.kt       |  4 +-
 .../joinLines/JoinToStringTemplateHandler.kt  | 50 +++++++++++++++++++
 .../stringTemplate/firstLineVariable.kt       |  5 ++
 .../stringTemplate/firstLineVariable.kt.after |  4 ++
 .../stringTemplate/secondLineVariable.kt      |  5 ++
 .../secondLineVariable.kt.after               |  4 ++
 .../joinLines/stringTemplate/simple.kt        |  4 ++
 .../joinLines/stringTemplate/simple.kt.after  |  3 ++
 .../joinLines/stringTemplate/threeLines.kt    |  5 ++
 .../stringTemplate/threeLines.kt.after        |  4 ++
 .../declarations/JoinLinesTestGenerated.java  | 33 ++++++++++++
 12 files changed, 120 insertions(+), 2 deletions(-)
 create mode 100644 idea/src/org/jetbrains/kotlin/idea/joinLines/JoinToStringTemplateHandler.kt
 create mode 100644 idea/testData/joinLines/stringTemplate/firstLineVariable.kt
 create mode 100644 idea/testData/joinLines/stringTemplate/firstLineVariable.kt.after
 create mode 100644 idea/testData/joinLines/stringTemplate/secondLineVariable.kt
 create mode 100644 idea/testData/joinLines/stringTemplate/secondLineVariable.kt.after
 create mode 100644 idea/testData/joinLines/stringTemplate/simple.kt
 create mode 100644 idea/testData/joinLines/stringTemplate/simple.kt.after
 create mode 100644 idea/testData/joinLines/stringTemplate/threeLines.kt
 create mode 100644 idea/testData/joinLines/stringTemplate/threeLines.kt.after

diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml
index b5bc7f9af0f..7d3ee5a2907 100644
--- a/idea/src/META-INF/plugin.xml
+++ b/idea/src/META-INF/plugin.xml
@@ -727,6 +727,7 @@
     <joinLinesHandler implementation="org.jetbrains.kotlin.idea.joinLines.JoinDeclarationAndAssignmentHandler"/>
     <joinLinesHandler implementation="org.jetbrains.kotlin.idea.joinLines.JoinBlockIntoSingleStatementHandler"/>
     <joinLinesHandler implementation="org.jetbrains.kotlin.idea.joinLines.JoinStatementsAddSemicolonHandler"/>
+    <joinLinesHandler implementation="org.jetbrains.kotlin.idea.joinLines.JoinToStringTemplateHandler"/>
 
     <targetElementEvaluator
         language="kotlin"
diff --git a/idea/src/org/jetbrains/kotlin/idea/intentions/ConvertToStringTemplateIntention.kt b/idea/src/org/jetbrains/kotlin/idea/intentions/ConvertToStringTemplateIntention.kt
index bcd5b25c916..cbbc98082f1 100644
--- a/idea/src/org/jetbrains/kotlin/idea/intentions/ConvertToStringTemplateIntention.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/intentions/ConvertToStringTemplateIntention.kt
@@ -57,7 +57,7 @@ open class ConvertToStringTemplateIntention : SelfTargetingOffsetIndependentInte
         }
 
         @JvmStatic
-        protected fun buildReplacement(expression: KtBinaryExpression): KtStringTemplateExpression {
+        fun buildReplacement(expression: KtBinaryExpression): KtStringTemplateExpression {
             val rightText = buildText(expression.right, false)
             return fold(expression.left, rightText, KtPsiFactory(expression))
         }
@@ -130,7 +130,7 @@ open class ConvertToStringTemplateIntention : SelfTargetingOffsetIndependentInte
             return "\${$expressionText}"
         }
 
-        private fun isApplicableToNoParentCheck(expression: KtBinaryExpression): Boolean {
+        fun isApplicableToNoParentCheck(expression: KtBinaryExpression): Boolean {
             if (expression.operationToken != KtTokens.PLUS) return false
             val expressionType = expression.analyze(BodyResolveMode.PARTIAL).getType(expression)
             if (!KotlinBuiltIns.isString(expressionType)) return false
diff --git a/idea/src/org/jetbrains/kotlin/idea/joinLines/JoinToStringTemplateHandler.kt b/idea/src/org/jetbrains/kotlin/idea/joinLines/JoinToStringTemplateHandler.kt
new file mode 100644
index 00000000000..c38581b068e
--- /dev/null
+++ b/idea/src/org/jetbrains/kotlin/idea/joinLines/JoinToStringTemplateHandler.kt
@@ -0,0 +1,50 @@
+/*
+ * Copyright 2010-2017 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.idea.joinLines
+
+import com.intellij.codeInsight.editorActions.JoinRawLinesHandlerDelegate
+import com.intellij.openapi.editor.Document
+import com.intellij.psi.PsiFile
+import org.jetbrains.kotlin.idea.intentions.ConvertToStringTemplateIntention
+import org.jetbrains.kotlin.lexer.KtTokens
+import org.jetbrains.kotlin.psi.KtBinaryExpression
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.KtStringTemplateExpression
+import org.jetbrains.kotlin.psi.psiUtil.endOffset
+
+class JoinToStringTemplateHandler : JoinRawLinesHandlerDelegate {
+    override fun tryJoinRawLines(document: Document, file: PsiFile, start: Int, end: Int): Int {
+        if (file !is KtFile) return -1
+
+        if (start == 0) return -1
+        val c = document.charsSequence[start]
+        val index = if (c == '\n') start - 1 else start
+
+        val plus = file.findElementAt(index)?.takeIf { it.node?.elementType == KtTokens.PLUS } ?: return -1
+        val expression = (plus.parent.parent as? KtBinaryExpression) ?: return -1
+        val left = expression.left ?: return -1
+        val right = expression.right ?: return -1
+        if (left !is KtStringTemplateExpression || right !is KtStringTemplateExpression) return -1
+        if (!ConvertToStringTemplateIntention.isApplicableToNoParentCheck(expression)) return -1
+
+        val offset = left.endOffset - 1
+        expression.replace(ConvertToStringTemplateIntention.buildReplacement(expression))
+        return offset
+    }
+
+    override fun tryJoinLines(document: Document, file: PsiFile, start: Int, end: Int): Int = -1
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/firstLineVariable.kt b/idea/testData/joinLines/stringTemplate/firstLineVariable.kt
new file mode 100644
index 00000000000..bbf2e2c568e
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/firstLineVariable.kt
@@ -0,0 +1,5 @@
+fun test() {
+    val foo = "1"
+    val s = foo +<caret>
+            "bar"
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/firstLineVariable.kt.after b/idea/testData/joinLines/stringTemplate/firstLineVariable.kt.after
new file mode 100644
index 00000000000..588d2b2e799
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/firstLineVariable.kt.after
@@ -0,0 +1,4 @@
+fun test() {
+    val foo = "1"
+    val s = foo +<caret> "bar"
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/secondLineVariable.kt b/idea/testData/joinLines/stringTemplate/secondLineVariable.kt
new file mode 100644
index 00000000000..188610ddb4d
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/secondLineVariable.kt
@@ -0,0 +1,5 @@
+fun test() {
+    val foo = "1"
+    val s = "bar" +<caret>
+            foo
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/secondLineVariable.kt.after b/idea/testData/joinLines/stringTemplate/secondLineVariable.kt.after
new file mode 100644
index 00000000000..68df4130f30
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/secondLineVariable.kt.after
@@ -0,0 +1,4 @@
+fun test() {
+    val foo = "1"
+    val s = "bar" +<caret> foo
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/simple.kt b/idea/testData/joinLines/stringTemplate/simple.kt
new file mode 100644
index 00000000000..30709391ee6
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/simple.kt
@@ -0,0 +1,4 @@
+fun test() {
+    val s = "foo" +<caret>
+            "bar"
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/simple.kt.after b/idea/testData/joinLines/stringTemplate/simple.kt.after
new file mode 100644
index 00000000000..53f0a7490a2
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/simple.kt.after
@@ -0,0 +1,3 @@
+fun test() {
+    val s = "foo<caret>bar"
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/threeLines.kt b/idea/testData/joinLines/stringTemplate/threeLines.kt
new file mode 100644
index 00000000000..260502316c3
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/threeLines.kt
@@ -0,0 +1,5 @@
+fun test() {
+    val s = "foo" +<caret>
+            "bar" +
+            "baz"
+}
\ No newline at end of file
diff --git a/idea/testData/joinLines/stringTemplate/threeLines.kt.after b/idea/testData/joinLines/stringTemplate/threeLines.kt.after
new file mode 100644
index 00000000000..ae29b808428
--- /dev/null
+++ b/idea/testData/joinLines/stringTemplate/threeLines.kt.after
@@ -0,0 +1,4 @@
+fun test() {
+    val s = "foo<caret>bar" +
+            "baz"
+}
\ No newline at end of file
diff --git a/idea/tests/org/jetbrains/kotlin/idea/intentions/declarations/JoinLinesTestGenerated.java b/idea/tests/org/jetbrains/kotlin/idea/intentions/declarations/JoinLinesTestGenerated.java
index d17ebbc20a1..120efc43ab3 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/intentions/declarations/JoinLinesTestGenerated.java
+++ b/idea/tests/org/jetbrains/kotlin/idea/intentions/declarations/JoinLinesTestGenerated.java
@@ -425,4 +425,37 @@ public class JoinLinesTestGenerated extends AbstractJoinLinesTest {
             doTest(fileName);
         }
     }
+
+    @TestMetadata("idea/testData/joinLines/stringTemplate")
+    @TestDataPath("$PROJECT_ROOT")
+    @RunWith(JUnit3RunnerWithInners.class)
+    public static class StringTemplate extends AbstractJoinLinesTest {
+        public void testAllFilesPresentInStringTemplate() throws Exception {
+            KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/joinLines/stringTemplate"), Pattern.compile("^(.+)\\.kt$"), TargetBackend.ANY, true);
+        }
+
+        @TestMetadata("firstLineVariable.kt")
+        public void testFirstLineVariable() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/joinLines/stringTemplate/firstLineVariable.kt");
+            doTest(fileName);
+        }
+
+        @TestMetadata("secondLineVariable.kt")
+        public void testSecondLineVariable() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/joinLines/stringTemplate/secondLineVariable.kt");
+            doTest(fileName);
+        }
+
+        @TestMetadata("simple.kt")
+        public void testSimple() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/joinLines/stringTemplate/simple.kt");
+            doTest(fileName);
+        }
+
+        @TestMetadata("threeLines.kt")
+        public void testThreeLines() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/joinLines/stringTemplate/threeLines.kt");
+            doTest(fileName);
+        }
+    }
 }