feat: autocomplete brackets

Author: xHeaven

src/main/kotlin/com/github/tempest/framework/views/completion/TemplateBracketBackspaceHandler.kt

@@ -0,0 +1,21 @@
+package com.github.tempest.framework.views.completion
+
+import com.github.tempest.framework.TempestFrameworkUtil
+import com.intellij.codeInsight.editorActions.BackspaceHandlerDelegate
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiFile
+
+class TemplateBracketBackspaceHandler : BackspaceHandlerDelegate() {
+
+    override fun beforeCharDeleted(c: Char, file: PsiFile, editor: Editor) {}
+
+    override fun charDeleted(c: Char, file: PsiFile, editor: Editor): Boolean {
+        if (!file.name.endsWith(TempestFrameworkUtil.TEMPLATE_SUFFIX)) return false
+
+        if (c != '!' && c != '-') return false
+
+        INSTANCE.synchronizeBracketsAfterDeletion(file.project, editor)
+
+        return false
+    }
+}

src/main/kotlin/com/github/tempest/framework/views/completion/TemplateBracketTypedHandler.kt

@@ -0,0 +1,137 @@
+package com.github.tempest.framework.views.completion
+
+import com.github.tempest.framework.TempestFrameworkUtil
+import com.intellij.codeInsight.editorActions.TypedHandlerDelegate
+import com.intellij.openapi.command.WriteCommandAction
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiFile
+
+class TemplateBracketTypedHandler : TypedHandlerDelegate() {
+
+    data class BracketPair(val opening: String, val closing: String)
+
+    override fun charTyped(c: Char, project: Project, editor: Editor, file: PsiFile): Result {
+        if (!file.name.endsWith(TempestFrameworkUtil.TEMPLATE_SUFFIX)) return Result.CONTINUE
+
+        val offset = editor.caretModel.offset
+        val text = editor.document.charsSequence
+
+        if (c == ' ') {
+            return handleAutoComplete(project, editor, text, offset)
+        }
+
+        if (c == '!' || c == '-') {
+            synchronizeBrackets(project, editor, text, offset)
+        }
+
+        return Result.CONTINUE
+    }
+
+    private fun handleAutoComplete(project: Project, editor: Editor, text: CharSequence, offset: Int): Result {
+        if (offset < 2) return Result.CONTINUE
+
+        val textBefore = text.subSequence(0, offset).toString()
+
+        for (pair in AUTO_COMPLETE_PAIRS) {
+            if (textBefore.endsWith(pair.opening + " ")) {
+                if (!hasClosingBracketAhead(text, offset, pair.closing.trim())) {
+                    insertClosingBracket(project, editor, offset, pair.closing)
+                    return Result.STOP
+                }
+            }
+        }
+
+        return Result.CONTINUE
+    }
+
+    private fun synchronizeBrackets(project: Project, editor: Editor, text: CharSequence, offset: Int) {
+        val textBefore = text.subSequence(0, offset).toString()
+
+        val currentPair = BRACKET_PAIRS.find { textBefore.endsWith(it.opening) } ?: return
+
+        if (currentPair.opening == "{{") return
+
+        val textAfter = text.subSequence(offset, text.length).toString()
+
+        for (otherPair in BRACKET_PAIRS) {
+            if (otherPair.closing == currentPair.closing) continue
+
+            val closingIndex = findFirstUnnestedClosing(textAfter, otherPair.closing, currentPair.opening)
+            if (closingIndex != -1) {
+                replaceText(project, editor, offset + closingIndex, otherPair.closing.length, currentPair.closing)
+                return
+            }
+        }
+    }
+
+    fun synchronizeBracketsAfterDeletion(project: Project, editor: Editor) {
+        val offset = editor.caretModel.offset
+        val text = editor.document.charsSequence
+        val textBefore = text.subSequence(0, offset).toString()
+
+        if (!textBefore.endsWith("{{")) return
+
+        if (textBefore.endsWith("{{!!") || textBefore.endsWith("{{--")) return
+
+        val textAfter = text.subSequence(offset, text.length).toString()
+
+        for (pair in BRACKET_PAIRS) {
+            if (pair.closing == "}}") continue
+
+            val closingIndex = findFirstUnnestedClosing(textAfter, pair.closing, "{{")
+            if (closingIndex != -1) {
+                replaceText(project, editor, offset + closingIndex, pair.closing.length, "}}")
+                return
+            }
+        }
+    }
+
+    private fun findFirstUnnestedClosing(textAfter: String, closing: String, opening: String): Int {
+        val closingIndex = textAfter.indexOf(closing)
+        if (closingIndex == -1) return -1
+
+        val textBeforeClosing = textAfter.take(closingIndex)
+        val nextOpenIndex = textBeforeClosing.indexOf(opening)
+
+        return if (nextOpenIndex == -1) closingIndex else -1
+    }
+
+    private fun hasClosingBracketAhead(text: CharSequence, offset: Int, closing: String): Boolean {
+        val textAfter = text.subSequence(offset, text.length).toString()
+
+        val nextClose = textAfter.indexOf(closing)
+        if (nextClose == -1) return false
+
+        val opening = BRACKET_PAIRS.find { it.closing == closing }?.opening ?: return false
+        val nextOpen = textAfter.indexOf(opening)
+
+        return nextOpen == -1 || nextClose < nextOpen
+    }
+
+    private fun insertClosingBracket(project: Project, editor: Editor, offset: Int, closing: String) {
+        WriteCommandAction.runWriteCommandAction(project) {
+            editor.document.insertString(offset, closing)
+        }
+    }
+
+    private fun replaceText(project: Project, editor: Editor, start: Int, length: Int, newText: String) {
+        WriteCommandAction.runWriteCommandAction(project) {
+            editor.document.replaceString(start, start + length, newText)
+        }
+    }
+}
+
+val BRACKET_PAIRS = listOf(
+    TemplateBracketTypedHandler.BracketPair("{{--", "--}}"),
+    TemplateBracketTypedHandler.BracketPair("{{!!", "!!}}"),
+    TemplateBracketTypedHandler.BracketPair("{{", "}}"),
+)
+
+private val AUTO_COMPLETE_PAIRS = listOf(
+    TemplateBracketTypedHandler.BracketPair("{{--", " --}}"),
+    TemplateBracketTypedHandler.BracketPair("{{!!", " !!}}"),
+    TemplateBracketTypedHandler.BracketPair("{{", " }}"),
+)
+
+val INSTANCE = TemplateBracketTypedHandler()

src/main/resources/META-INF/plugin.xml

@@ -67,6 +67,10 @@
         <psi.referenceContributor
                 language="PHP"
                 implementation="com.github.tempest.framework.db.references.QueryBuilderReferenceContributor"/>
+        <typedHandler
+                implementation="com.github.tempest.framework.views.completion.TemplateBracketTypedHandler"/>
+        <backspaceHandlerDelegate
+                implementation="com.github.tempest.framework.views.completion.TemplateBracketBackspaceHandler"/>
     </extensions>
     <extensions defaultExtensionNs="com.jetbrains.php">
         <libraryRoot id="tempest.meta-storm" path="meta-storm" runtime="false" />