From e34a939d364c9a1b8e21e47f1cfaf050eb66401b Mon Sep 17 00:00:00 2001
From: "Anna.Kozlova" <anna.kozlova@jetbrains.com>
Date: Mon, 16 Apr 2018 14:16:18 +0200
Subject: [PATCH] raw string literal: paste string with leading/trailing tic

---
 .../RawStringLiteralPasteProcessor.java       | 28 +++++++++++++++----
 .../daemon/LightAdvRawStringLiteralsTest.java |  8 ++++++
 2 files changed, 31 insertions(+), 5 deletions(-)

diff --git a/java/java-impl/src/com/intellij/codeInsight/editorActions/RawStringLiteralPasteProcessor.java b/java/java-impl/src/com/intellij/codeInsight/editorActions/RawStringLiteralPasteProcessor.java
index 17ee9539477a..000406ff1184 100644
--- a/java/java-impl/src/com/intellij/codeInsight/editorActions/RawStringLiteralPasteProcessor.java
+++ b/java/java-impl/src/com/intellij/codeInsight/editorActions/RawStringLiteralPasteProcessor.java
@@ -13,6 +13,8 @@ import com.intellij.openapi.editor.EditorModificationUtil;
 import com.intellij.openapi.editor.RangeMarker;
 import com.intellij.openapi.ide.CopyPasteManager;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.TextRange;
+import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.psi.*;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -40,11 +42,27 @@ public class RawStringLiteralPasteProcessor implements PasteProvider {
     PsiElement stringLiteral = findRawStringLiteralAtCaret(file, offset);
     if (stringLiteral == null) return;
 
+    int leadingTicsSequence = PsiRawStringLiteralUtil.getLeadingTicsSequence(text);
+    int tailingTicsSequence = PsiRawStringLiteralUtil.getTailingTicsSequence(text);
+
+    String prefix = "";
+    TextRange textRange = stringLiteral.getTextRange();
+    if (offset == textRange.getStartOffset() + 1 && leadingTicsSequence > 0) {
+      prefix = "\"" + StringUtil.repeat("`", leadingTicsSequence) + "\" + ";
+      text = text.substring(leadingTicsSequence);
+    }
+
+    String suffix = "";
+    if (offset == textRange.getEndOffset() - 1 && tailingTicsSequence > 0) {
+      suffix = "+ \"" + StringUtil.repeat("`", tailingTicsSequence) + "\"";
+      text = text.substring(0, text.length() - tailingTicsSequence);
+    }
+
     String literalText = stringLiteral.getText();
     int ticsLength = PsiRawStringLiteralUtil.getLeadingTicsSequence(literalText);
     String quotes = literalText.substring(0, ticsLength);
     String additionalQuotes = PsiRawStringLiteralUtil.getAdditionalTics(text, quotes);
-    insertAtCaret(text, additionalQuotes, stringLiteral, editor);
+    insertAtCaret(text, additionalQuotes, stringLiteral, editor, prefix, suffix);
   }
 
   @Override
@@ -82,7 +100,7 @@ public class RawStringLiteralPasteProcessor implements PasteProvider {
   private static void insertAtCaret(final String text,
                                     final String additionalQuotes,
                                     final PsiElement element,
-                                    final Editor editor) {
+                                    final Editor editor, String prefix, String suffix) {
     final Project project = editor.getProject();
     if (project == null) return;
 
@@ -98,9 +116,9 @@ public class RawStringLiteralPasteProcessor implements PasteProvider {
 
       EditorModificationUtil.insertStringAtCaret(editor, text);
 
-      if (!additionalQuotes.isEmpty()) {
-        document.insertString(range.getStartOffset(), additionalQuotes);
-        document.insertString(range.getEndOffset(), additionalQuotes);
+      if (!additionalQuotes.isEmpty() || !prefix.isEmpty() || !suffix.isEmpty()) {
+        document.insertString(range.getStartOffset(), prefix + additionalQuotes);
+        document.insertString(range.getEndOffset(), additionalQuotes + suffix);
       }
     });
   }
diff --git a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/LightAdvRawStringLiteralsTest.java b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/LightAdvRawStringLiteralsTest.java
index cb6b7f53f0c9..4d533535c918 100644
--- a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/LightAdvRawStringLiteralsTest.java
+++ b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/LightAdvRawStringLiteralsTest.java
@@ -57,6 +57,14 @@ public class LightAdvRawStringLiteralsTest extends LightCodeInsightFixtureTestCa
     doTestPaste("class A {{String s = `q<caret>`;}}", "a\nb`\nc", "class A {{String s = ``qa\nb`\nc``;}}");
   }
 
+  public void testPasteInRawStringStaringWithTics() {
+    doTestPaste("class A {{String s = `<caret>q`;}}", "``a\n``b", "class A {{String s = \"``\" + `a\n" + "``bq`;}}");
+  }
+
+  public void testPasteInRawStringEndingWithTics() {
+    doTestPaste("class A {{String s = `q<caret>`;}}", "``a\n``b``", "class A {{String s = \"``\" + `a\n" + "``bq`;}}");
+  }
+
   public void testPasteInRawStringLiteralOneAndTwo() {
     doTestPaste("class A {{String s = `q<caret>`;}}", "a\n``b`\nc", "class A {{String s = ```qa\n``b`\nc```;}}");
   }