From 9b7849f6a608ae5d7a6a7d212ba15954ed34c861 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=BF=97=E8=B1=AA?= <izhangzhihao@hotmail.com>
Date: Sat, 8 Aug 2020 22:40:09 +0800
Subject: [PATCH] #27 Some experiment support for token analysis

---
 .../rainbow/fart/RainbowFartTypedHandler.kt   | 60 +++++++++++++++----
 1 file changed, 50 insertions(+), 10 deletions(-)

diff --git a/src/main/kotlin/com/github/izhangzhihao/rainbow/fart/RainbowFartTypedHandler.kt b/src/main/kotlin/com/github/izhangzhihao/rainbow/fart/RainbowFartTypedHandler.kt
index aa8e4e5..29a7b50 100644
--- a/src/main/kotlin/com/github/izhangzhihao/rainbow/fart/RainbowFartTypedHandler.kt
+++ b/src/main/kotlin/com/github/izhangzhihao/rainbow/fart/RainbowFartTypedHandler.kt
@@ -1,6 +1,6 @@
 package com.github.izhangzhihao.rainbow.fart
 
-import com.github.izhangzhihao.rainbow.fart.RainbowFartTypedHandler.FartTypedHandler.isTypingInsideComment
+import com.github.izhangzhihao.rainbow.fart.RainbowFartTypedHandler.FartTypedHandler.handle
 import com.github.izhangzhihao.rainbow.fart.RainbowFartTypedHandler.FartTypedHandler.releaseFart
 import com.github.izhangzhihao.rainbow.fart.settings.RainbowFartSettings
 import com.intellij.codeInsight.template.impl.editorActions.TypedActionHandlerBase
@@ -20,6 +20,10 @@ import kotlinx.coroutines.launch
 import java.io.File
 
 
+/***
+ *  Also should read https://github.com/JetBrains/kotlin/blob/master/idea/src/org/jetbrains/kotlin/idea/editor/KotlinTypedHandler.java
+ *  and [com.intellij.codeInsight.editorActions.JavaTypedHandler]
+ */
 class RainbowFartTypedHandler(originalHandler: TypedActionHandler) : TypedActionHandlerBase(originalHandler) {
 
     private var candidates: MutableList<Char> = mutableListOf()
@@ -36,12 +40,11 @@ class RainbowFartTypedHandler(originalHandler: TypedActionHandler) : TypedAction
                 val virtualFile = FileDocumentManager.getInstance().getFile(editor.document)
                 if (virtualFile != null) {
                     val file = PsiManager.getInstance(project).findFile(virtualFile)
-                    if (file != null && isTypingInsideComment(editor, file)) {
+                    if (file != null && handle(editor, file)) {
                         return
                     }
                 }
             }
-
             candidates.add(charTyped)
             val str = candidates.joinToString("")
             BuildInContributes.buildInContributesSeq
@@ -86,7 +89,7 @@ class RainbowFartTypedHandler(originalHandler: TypedActionHandler) : TypedAction
             player.close()
         }
 
-        fun isTypingInsideComment(editor: Editor, file: PsiFile): Boolean {
+        fun handle(editor: Editor, file: PsiFile): Boolean {
             val provider = file.viewProvider
             val offset = editor.caretModel.offset
             val elementAtCaret: PsiElement?
@@ -95,17 +98,54 @@ class RainbowFartTypedHandler(originalHandler: TypedActionHandler) : TypedAction
             } else {
                 provider.findElementAt(editor.document.textLength - 1)
             }
+            if (
+                    elementAtCaret?.language?.id.equals("kotlin", ignoreCase = true) ||
+                    elementAtCaret?.language?.id.equals("java", ignoreCase = true) ||
+                    elementAtCaret?.language?.id.equals("javascript", ignoreCase = true) ||
+                    elementAtCaret?.language?.id == "ECMAScript 6" ||
+                    elementAtCaret?.language?.id.equals("typescript", ignoreCase = true)
+            ) {
+                triggerForSpecificLang(elementAtCaret)
+                return true
+            }
+
             var element = elementAtCaret
             while (element is PsiWhiteSpace) {
                 element = element.getPrevSibling()
             }
-
             return element.elementType.toString().contains("comment", true)
-//            if (element == null) {
-//                return false
-//            }
-//            val node: ASTNode? = element.node
-//            return node != null //&& JavaDocTokenType.ALL_JAVADOC_TOKENS.contains(node.getElementType())
+        }
+
+        private fun triggerForSpecificLang(elementAtCaret: PsiElement?) {
+            val str =
+                    when {
+                        elementAtCaret !is PsiWhiteSpace -> {
+                            elementAtCaret.toString()
+                        }
+                        elementAtCaret.language.id.equals("java", ignoreCase = true) -> { // && element.prevSibling !is PsiWhiteSpace
+                            elementAtCaret.prevSibling.toString()
+                        }
+                        elementAtCaret.language.id == "ECMAScript 6" ||
+                                elementAtCaret.language.id.equals("javascript", ignoreCase = true) ||
+                                elementAtCaret.language.id.equals("typescript", ignoreCase = true)
+                        -> { // && element.parent?.firstChild !is PsiWhiteSpace
+                            elementAtCaret.parent?.firstChild.toString()
+                        }
+                        elementAtCaret.language.id.equals("kotlin", ignoreCase = true) -> {
+                            elementAtCaret.parent?.firstChild.toString()
+                        }
+                        else -> {
+                            elementAtCaret.prevSibling.toString()
+                        }
+                    }
+            BuildInContributes.buildInContributesSeq
+                    .firstOrNull { (keyword, _) ->
+                        str.contains(keyword, true)
+                    }?.let { (_, voices) ->
+                        GlobalScope.launch(Dispatchers.Default) {
+                            releaseFart(voices)
+                        }
+                    }
         }
     }
 }
\ No newline at end of file