feat: highlight exit points when cursor on fun or receive/external/bounced/init

Fixes #71

Author: i582

src/main/kotlin/org/tonstudio/tact/ide/highlight/exitpoint/TactFunctionExitPointHandler.kt

@@ -0,0 +1,69 @@
+package org.tonstudio.tact.ide.highlight.exitpoint
+
+import com.intellij.codeInsight.highlighting.HighlightUsagesHandlerBase
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.util.parentOfTypes
+import com.intellij.util.Consumer
+import org.tonstudio.tact.lang.psi.*
+
+class TactFunctionExitPointHandler(
+    editor: Editor,
+    file: PsiFile,
+    private val target: PsiElement,
+    private val function: TactTypeOwner,
+) : HighlightUsagesHandlerBase<PsiElement>(editor, file) {
+
+    override fun getTargets() = listOf(target)
+
+    override fun selectTargets(targets: List<PsiElement>, selectionConsumer: Consumer<in MutableList<out PsiElement>?>) {
+        selectionConsumer.consume(targets.toMutableList())
+    }
+
+    override fun computeUsages(targets: List<PsiElement>) {
+        if (function is TactFunctionOrMethodDeclaration) {
+            val identifier = function.getIdentifier()
+            if (identifier != null) {
+                addOccurrence(identifier)
+            }
+        }
+
+        if (function is TactMessageFunctionDeclaration) {
+            val identifier = function.getIdentifier()
+            if (identifier != null) {
+                addOccurrence(identifier)
+            }
+        }
+
+        val visitor = object : TactRecursiveElementVisitor() {
+            override fun visitReturnStatement(expr: TactReturnStatement) {
+                addOccurrence(expr)
+            }
+
+            override fun visitCallExpr(o: TactCallExpr) {
+                val callerName = o.expression?.text
+                if (callerName == "throw") {
+                    addOccurrence(o)
+                }
+            }
+
+            override fun visitAssertNotNullExpr(o: TactAssertNotNullExpr) {
+                addOccurrence(o.assertOp)
+            }
+        }
+
+        visitor.visitTypeOwner(function)
+    }
+
+    companion object {
+        fun createForElement(editor: Editor, file: PsiFile, element: PsiElement): TactFunctionExitPointHandler? {
+            val function = element.parentOfTypes(
+                TactFunctionOrMethodDeclaration::class,
+                TactMessageFunctionDeclaration::class,
+                TactContractInitDeclaration::class
+            )
+            return function?.let { TactFunctionExitPointHandler(editor, file, element, it) }
+        }
+    }
+}

src/main/kotlin/org/tonstudio/tact/ide/highlight/exitpoint/TactHighlightExitPointsHandlerFactory.kt

@@ -0,0 +1,53 @@
+package org.tonstudio.tact.ide.highlight.exitpoint
+
+import com.intellij.codeInsight.highlighting.HighlightUsagesHandlerBase
+import com.intellij.codeInsight.highlighting.HighlightUsagesHandlerFactoryBase
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.impl.source.tree.LeafPsiElement
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.psi.util.elementType
+import org.tonstudio.tact.lang.TactTypes
+import org.tonstudio.tact.lang.psi.TactAssertNotNullExpr
+import org.tonstudio.tact.lang.psi.TactContractInitDeclaration
+
+class TactHighlightExitPointsHandlerFactory : HighlightUsagesHandlerFactoryBase() {
+    override fun createHighlightUsagesHandler(editor: Editor, file: PsiFile, target: PsiElement): HighlightUsagesHandlerBase<*>? {
+        if (target !is LeafPsiElement) return null
+
+        val prevToken = PsiTreeUtil.prevLeaf(target) ?: return null
+        return createHandler(editor, file, target) ?: createHandler(editor, file, prevToken)
+    }
+
+    private fun createHandler(editor: Editor, file: PsiFile, target: PsiElement): HighlightUsagesHandlerBase<*>? {
+        val elementType = target.elementType
+
+        if (
+            elementType == TactTypes.IDENTIFIER &&
+            target.parent is TactContractInitDeclaration &&
+            target.textMatches("init")
+        ) {
+            return TactFunctionExitPointHandler.createForElement(editor, file, target)
+        }
+
+        if (
+            elementType == TactTypes.RETURN ||
+            elementType == TactTypes.FUN ||
+            elementType == TactTypes.RECEIVE ||
+            elementType == TactTypes.EXTERNAL ||
+            elementType == TactTypes.BOUNCED
+        ) {
+            return TactFunctionExitPointHandler.createForElement(editor, file, target)
+        }
+
+        if (elementType == TactTypes.ASSERT_OP) {
+            if (target.parent is TactAssertNotNullExpr) {
+                return TactFunctionExitPointHandler.createForElement(editor, file, target)
+            }
+        }
+
+        return null
+    }
+
+}

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

@@ -136,6 +136,8 @@ Provides official support for <a href="https://tact-lang.org/">Tact</a> programm
                                    implementationClass="org.tonstudio.tact.ide.codeInsight.TactParameterInfoHandler"/>
         <codeInsight.lineMarkerProvider language="tact"
                                         implementationClass="org.tonstudio.tact.ide.highlight.TactRecursiveCallMarkerProvider"/>
+        <highlightUsagesHandlerFactory
+                implementation="org.tonstudio.tact.ide.highlight.exitpoint.TactHighlightExitPointsHandlerFactory"/>
         <!-- endregion Code Insight -->
 
         <!-- region Documentation -->