From d82d4de0d64d0940f7b3c377aa1c331bdd643db1 Mon Sep 17 00:00:00 2001
From: Daniil Ovchinnikov <daniil.ovchinnikov@jetbrains.com>
Date: Mon, 27 Mar 2017 19:41:27 +0300
Subject: [PATCH] [groovy] delete closing triple single/double quote
 (IDEA-169046)

---
 plugins/groovy/src/META-INF/plugin.xml        |  1 +
 ...oovyTripleQuoteBackspaceHandlerDelegate.kt | 50 +++++++++++++++++++
 .../groovy/lang/GroovyEditingTest.groovy      | 11 +++-
 3 files changed, 61 insertions(+), 1 deletion(-)
 create mode 100644 plugins/groovy/src/org/jetbrains/plugins/groovy/editor/actions/GroovyTripleQuoteBackspaceHandlerDelegate.kt

diff --git a/plugins/groovy/src/META-INF/plugin.xml b/plugins/groovy/src/META-INF/plugin.xml
index eb607f96ad77..11896cc55fb9 100644
--- a/plugins/groovy/src/META-INF/plugin.xml
+++ b/plugins/groovy/src/META-INF/plugin.xml
@@ -399,6 +399,7 @@
                           order="before EnterBetweenBracesHandler"/>
     <typedHandler implementation="org.jetbrains.plugins.groovy.editor.actions.GStringTypedActionHandler"/>
     <backspaceHandlerDelegate implementation="org.jetbrains.plugins.groovy.editor.actions.GStringBackspaceHandlerDelegate"/>
+    <backspaceHandlerDelegate implementation="org.jetbrains.plugins.groovy.editor.actions.GroovyTripleQuoteBackspaceHandlerDelegate"/>
 
     <typedHandler implementation="org.jetbrains.plugins.groovy.editor.actions.GroovyTypedHandler"/>
     <backspaceHandlerDelegate implementation="org.jetbrains.plugins.groovy.editor.actions.GroovyBackspaceHandler"/>
diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/editor/actions/GroovyTripleQuoteBackspaceHandlerDelegate.kt b/plugins/groovy/src/org/jetbrains/plugins/groovy/editor/actions/GroovyTripleQuoteBackspaceHandlerDelegate.kt
new file mode 100644
index 000000000000..4da2346249dc
--- /dev/null
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/editor/actions/GroovyTripleQuoteBackspaceHandlerDelegate.kt
@@ -0,0 +1,50 @@
+/*
+ * 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 org.jetbrains.plugins.groovy.editor.actions
+
+import com.intellij.codeInsight.CodeInsightSettings
+import com.intellij.codeInsight.editorActions.BackspaceHandlerDelegate
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.editor.ex.EditorEx
+import com.intellij.openapi.editor.highlighter.HighlighterIterator
+import com.intellij.psi.PsiFile
+import org.jetbrains.plugins.groovy.lang.lexer.GroovyTokenTypes.mGSTRING_LITERAL
+import org.jetbrains.plugins.groovy.lang.lexer.GroovyTokenTypes.mSTRING_LITERAL
+
+class GroovyTripleQuoteBackspaceHandlerDelegate : BackspaceHandlerDelegate() {
+
+  private var myWithinTripleQuoted: Boolean = false
+
+  override fun beforeCharDeleted(c: Char, file: PsiFile?, editor: Editor?) {
+    myWithinTripleQuoted = false
+    if (!CodeInsightSettings.getInstance().AUTOINSERT_PAIR_QUOTE) return
+    if (c != '\'' && c != '"') return
+    editor as? EditorEx ?: return
+    val offset = editor.caretModel.offset
+    val iterator: HighlighterIterator = editor.highlighter.createIterator(offset)
+    val tokenType = iterator.tokenType
+    if (tokenType == mSTRING_LITERAL || tokenType == mGSTRING_LITERAL) {
+      myWithinTripleQuoted = iterator.start + 3 == offset && iterator.end - 3 == offset
+    }
+  }
+
+  override fun charDeleted(c: Char, file: PsiFile?, editor: Editor?): Boolean {
+    if (!myWithinTripleQuoted || editor == null) return false
+    val offset = editor.caretModel.offset
+    editor.document.deleteString(offset, offset + 3)
+    return true
+  }
+}
\ No newline at end of file
diff --git a/plugins/groovy/test/org/jetbrains/plugins/groovy/lang/GroovyEditingTest.groovy b/plugins/groovy/test/org/jetbrains/plugins/groovy/lang/GroovyEditingTest.groovy
index 9a198e069429..d343d30c348a 100644
--- a/plugins/groovy/test/org/jetbrains/plugins/groovy/lang/GroovyEditingTest.groovy
+++ b/plugins/groovy/test/org/jetbrains/plugins/groovy/lang/GroovyEditingTest.groovy
@@ -1,5 +1,5 @@
 /*
- * Copyright 2000-2016 JetBrains s.r.o.
+ * 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.
@@ -234,6 +234,15 @@ class A {}
 ''')
   }
 
+  void testBackspace() {
+    doTest(/'<caret>'/, '\b', /<caret>/)
+    doTest(/''<caret>/, '\b', /'<caret>/)
+    doTest(/"<caret>"/, '\b', /<caret>/)
+    doTest(/""<caret>/, '\b', /"<caret>/)
+    doTest(/'''<caret>'''/, '\b', /''<caret>/)
+    doTest(/"""<caret>"""/, '\b', /""<caret>/)
+  }
+
   /*
   todo uncomment when implemented
   void testRegex1() {