From 31000a0b68d314d63bc336dac06c707ebebfba8e Mon Sep 17 00:00:00 2001
From: Dmitry Batrak <Dmitry.Batrak@jetbrains.com>
Date: Tue, 12 Sep 2017 15:56:16 +0300
Subject: [PATCH] IDEA-178801 Multi-line pasting in line comments

---
 .../LineCommentCopyPastePreProcessor.java     | 60 +++++++++++++++++++
 resources/src/META-INF/IdeaPlugin.xml         |  1 +
 2 files changed, 61 insertions(+)
 create mode 100644 java/java-impl/src/com/intellij/codeInsight/editorActions/LineCommentCopyPastePreProcessor.java

diff --git a/java/java-impl/src/com/intellij/codeInsight/editorActions/LineCommentCopyPastePreProcessor.java b/java/java-impl/src/com/intellij/codeInsight/editorActions/LineCommentCopyPastePreProcessor.java
new file mode 100644
index 000000000000..daf96cb13569
--- /dev/null
+++ b/java/java-impl/src/com/intellij/codeInsight/editorActions/LineCommentCopyPastePreProcessor.java
@@ -0,0 +1,60 @@
+/*
+ * Copyright 2000-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 com.intellij.codeInsight.editorActions;
+
+import com.intellij.lang.java.JavaLanguage;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.RawText;
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiJavaFile;
+import com.intellij.psi.codeStyle.CodeStyleSettings;
+import com.intellij.psi.codeStyle.CodeStyleSettingsManager;
+import com.intellij.util.DocumentUtil;
+import com.intellij.util.text.CharArrayUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+public class LineCommentCopyPastePreProcessor implements CopyPastePreProcessor {
+  @Nullable
+  @Override
+  public String preprocessOnCopy(PsiFile file, int[] startOffsets, int[] endOffsets, String text) {
+    return null;
+  }
+
+  @NotNull
+  @Override
+  public String preprocessOnPaste(Project project, PsiFile file, Editor editor, String text, RawText rawText) {
+    if (!(file instanceof PsiJavaFile)) return text;
+
+    Document document = editor.getDocument();
+    int offset = editor.getSelectionModel().getSelectionStart();
+    int lineStartOffset = DocumentUtil.getLineStartOffset(offset, document);
+    CharSequence chars = document.getImmutableCharSequence();
+    int firstNonWsLineOffset = CharArrayUtil.shiftForward(chars, lineStartOffset, " \t");
+    if (offset >= (firstNonWsLineOffset + 2) && 
+        chars.charAt(firstNonWsLineOffset) == '/' && chars.charAt(firstNonWsLineOffset + 1) == '/') {
+      CodeStyleSettings codeStyleSettings = CodeStyleSettingsManager.getSettings(project);
+      String lineStartReplacement = "\n" + chars.subSequence(lineStartOffset, firstNonWsLineOffset + 2) +
+                                    (codeStyleSettings.getCommonSettings(JavaLanguage.INSTANCE).LINE_COMMENT_ADD_SPACE ? " " : "");
+      return text.replace("\n", lineStartReplacement);
+    }
+    else {
+      return text;
+    }
+  }
+}
diff --git a/resources/src/META-INF/IdeaPlugin.xml b/resources/src/META-INF/IdeaPlugin.xml
index 8cfac0e4a9d0..acfed003ed11 100644
--- a/resources/src/META-INF/IdeaPlugin.xml
+++ b/resources/src/META-INF/IdeaPlugin.xml
@@ -1524,6 +1524,7 @@
     <liveTemplateMacro implementation="com.intellij.codeInsight.template.macro.TypeOfVariableMacro"/>
 
 
+    <copyPastePreProcessor implementation="com.intellij.codeInsight.editorActions.LineCommentCopyPastePreProcessor"/>
     <copyPastePreProcessor implementation="com.intellij.codeInsight.editorActions.StringLiteralCopyPasteProcessor"/>
     <copyPastePostProcessor implementation="com.intellij.codeInsight.editorActions.JavaCopyPasteReferenceProcessor"/>