From a4769e3979585143337378187547c668c14ec42f Mon Sep 17 00:00:00 2001
From: LlamaLad7 <l3gomindstorms@gmail.com>
Date: Thu, 4 Jul 2024 15:57:49 +0100
Subject: [PATCH] Expressions: Confidently suggest completions after `::`.

---
 .../MEExpressionTypedHandlerDelegate.kt       | 42 +++++++++++++++++++
 src/main/resources/META-INF/plugin.xml        |  1 +
 2 files changed, 43 insertions(+)
 create mode 100644 src/main/kotlin/platform/mixin/expression/MEExpressionTypedHandlerDelegate.kt

diff --git a/src/main/kotlin/platform/mixin/expression/MEExpressionTypedHandlerDelegate.kt b/src/main/kotlin/platform/mixin/expression/MEExpressionTypedHandlerDelegate.kt
new file mode 100644
index 000000000..6d714b43b
--- /dev/null
+++ b/src/main/kotlin/platform/mixin/expression/MEExpressionTypedHandlerDelegate.kt
@@ -0,0 +1,42 @@
+/*
+ * Minecraft Development for IntelliJ
+ *
+ * https://mcdev.io/
+ *
+ * Copyright (C) 2024 minecraft-dev
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published
+ * by the Free Software Foundation, version 3.0 only.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package com.demonwav.mcdev.platform.mixin.expression
+
+import com.demonwav.mcdev.platform.mixin.expression.gen.psi.MEExpressionTypes
+import com.intellij.codeInsight.AutoPopupController
+import com.intellij.codeInsight.editorActions.TypedHandlerDelegate
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiFile
+import com.intellij.psi.util.elementType
+
+class MEExpressionTypedHandlerDelegate : TypedHandlerDelegate() {
+    override fun checkAutoPopup(charTyped: Char, project: Project, editor: Editor, file: PsiFile): Result {
+        if (charTyped == ':' && file.language == MEExpressionLanguage) {
+            AutoPopupController.getInstance(project).autoPopupMemberLookup(editor) {
+                val offset = editor.caretModel.offset
+                it.findElementAt(offset - 1).elementType == MEExpressionTypes.TOKEN_METHOD_REF
+            }
+            return Result.STOP
+        }
+        return Result.CONTINUE
+    }
+}
diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
index 307909a6e..273df3975 100644
--- a/src/main/resources/META-INF/plugin.xml
+++ b/src/main/resources/META-INF/plugin.xml
@@ -514,6 +514,7 @@
         <completion.contributor language="MEExpression"
                                 implementationClass="com.demonwav.mcdev.platform.mixin.expression.MEExpressionCompletionContributor"/>
         <multiHostInjector implementation="com.demonwav.mcdev.platform.mixin.expression.MEExpressionInjector"/>
+        <typedHandler implementation="com.demonwav.mcdev.platform.mixin.expression.MEExpressionTypedHandlerDelegate"/>
         <!--endregion-->
 
         <!--endregion-->