feat(navigation): add navigation to parent constant definition and vice versa

i582 · i582 · commit 5b4ffdd84b8e · 2025-05-05T13:00:41.000+04:00
Fixes #18
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/psi/TactStorageMembersOwner.kt b/src/main/kotlin/org/tonstudio/tact/lang/psi/TactStorageMembersOwner.kt
@@ -20,6 +20,12 @@ fun TactStorageMembersOwner.fields(): List<TactFieldDefinition> {
     return own + inherited
 }
 
+fun TactStorageMembersOwner.constants(): List<TactConstDefinition> {
+    val own = getConstantsList()
+    val inherited = getInheritedTraits().flatMap { it.traitType.getConstantsList() }
+    return own + inherited
+}
+
 fun TactStorageMembersOwner.name(): String {
     val parent = this.parent
     if (parent is TactNamedElement) {
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/search/TactConstantImplementationsSearch.kt b/src/main/kotlin/org/tonstudio/tact/lang/search/TactConstantImplementationsSearch.kt
@@ -0,0 +1,26 @@
+package org.tonstudio.tact.lang.search
+
+import com.intellij.openapi.application.QueryExecutorBase
+import com.intellij.psi.search.searches.DefinitionsScopedSearch
+import com.intellij.util.Processor
+import org.tonstudio.tact.lang.psi.*
+
+class TactConstantImplementationsSearch : QueryExecutorBase<TactConstDefinition, DefinitionsScopedSearch.SearchParameters>(true) {
+    override fun processQuery(
+        parameters: DefinitionsScopedSearch.SearchParameters,
+        consumer: Processor<in TactConstDefinition>,
+    ) {
+        if (!parameters.isCheckDeep) return
+
+        val constant = parameters.element as? TactConstDefinition ?: return
+        val trait = constant.getOwner() as? TactTraitDeclaration ?: return
+
+        TactImplementationsSearch().processQuery(DefinitionsScopedSearch.SearchParameters(trait), { t ->
+            val implementations = t.getConstantsList().filter { c -> c.name == constant.name }
+            for (implementation in implementations) {
+                consumer.process(implementation)
+            }
+            true
+        })
+    }
+}
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/search/TactGotoSuperHandler.kt b/src/main/kotlin/org/tonstudio/tact/lang/search/TactGotoSuperHandler.kt
@@ -5,6 +5,7 @@ import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
+import org.tonstudio.tact.lang.psi.TactConstDefinition
 import org.tonstudio.tact.lang.psi.TactFile
 import org.tonstudio.tact.lang.psi.TactFunctionDeclaration
 
@@ -23,5 +24,8 @@ class TactGotoSuperHandler : LanguageCodeInsightActionHandler {
         if (parent is TactFunctionDeclaration) {
             showSuperMethodPopup(null, parent)
         }
+        if (parent is TactConstDefinition) {
+            showSuperConstantPopup(null, parent)
+        }
     }
 }
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/search/TactImplementationsProvider.kt b/src/main/kotlin/org/tonstudio/tact/lang/search/TactImplementationsProvider.kt
@@ -9,6 +9,7 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.impl.source.tree.LeafPsiElement
 import com.intellij.psi.search.searches.DefinitionsScopedSearch
 import org.tonstudio.tact.lang.TactTypes
+import org.tonstudio.tact.lang.psi.TactConstDefinition
 import org.tonstudio.tact.lang.psi.TactFunctionDeclaration
 import org.tonstudio.tact.lang.psi.TactNamedElement
 import org.tonstudio.tact.lang.psi.TactTraitDeclaration
@@ -35,11 +36,28 @@ class TactImplementationsProvider : LineMarkerProviderDescriptor() {
     ): Boolean {
         if (element !is LeafPsiElement || element.elementType != TactTypes.IDENTIFIER) return true
 
-        val func = element.parent as? TactFunctionDeclaration ?: return true
+        val func = element.parent as? TactFunctionDeclaration
+        if (func != null) {
+            return processFunction(func, element, result)
+        }
+
+        val constant = element.parent as? TactConstDefinition
+        if (constant != null) {
+            return processConstant(constant, element, result)
+        }
+
+        return false
+    }
+
+    private fun processFunction(
+        func: TactFunctionDeclaration,
+        element: LeafPsiElement,
+        result: MutableCollection<in LineMarkerInfo<*>?>,
+    ): Boolean {
         val owner = func.getOwner() ?: return true
         if (owner !is TactTraitDeclaration) return true
 
-        if (!hasImplementations(func, TactFunctionImplementationsSearch())) return false
+        if (!hasImplementations(func, TactFunctionImplementationsSearch())) return true
 
         val (action, what) = if (!func.isAbstract) "Overrides" to "override" else "Implementations" to "implemented"
 
@@ -59,4 +77,33 @@ class TactImplementationsProvider : LineMarkerProviderDescriptor() {
         result.add(info)
         return false
     }
+
+    private fun processConstant(
+        constant: TactConstDefinition,
+        element: LeafPsiElement,
+        result: MutableCollection<in LineMarkerInfo<*>?>,
+    ): Boolean {
+        val owner = constant.getOwner() ?: return true
+        if (owner !is TactTraitDeclaration) return true
+
+        if (!hasImplementations(constant, TactConstantImplementationsSearch())) return true
+
+        val (action, what) = "Overrides" to "override"
+
+        val info = createInfo(element, { event, elem ->
+            val named = elem.parent as? TactNamedElement ?: return@createInfo
+
+            showPopup(
+                "$action of $name",
+                { size -> "Constant '$name' $what in $size types" },
+                event,
+                DefinitionsScopedSearch.SearchParameters(named),
+                methodRenderer(named),
+                TactConstantImplementationsSearch(),
+            )
+        }, "Go to $action", AllIcons.Gutter.ImplementedMethod, IdeActions.ACTION_GOTO_IMPLEMENTATION)
+
+        result.add(info)
+        return false
+    }
 }
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/search/TactSuperConstantSearch.kt b/src/main/kotlin/org/tonstudio/tact/lang/search/TactSuperConstantSearch.kt
@@ -0,0 +1,44 @@
+package org.tonstudio.tact.lang.search
+
+import com.intellij.openapi.application.QueryExecutorBase
+import com.intellij.psi.search.searches.DefinitionsScopedSearch
+import com.intellij.util.CommonProcessors
+import com.intellij.util.Processor
+import org.tonstudio.tact.lang.psi.TactConstDefinition
+import org.tonstudio.tact.lang.psi.TactContractDeclaration
+import org.tonstudio.tact.lang.psi.TactTraitDeclaration
+import org.tonstudio.tact.lang.psi.constants
+
+fun hasSuperConstant(method: TactConstDefinition): Boolean {
+    val processor = CommonProcessors.FindFirstProcessor<TactConstDefinition>()
+    TactSuperConstantSearch().processQuery(DefinitionsScopedSearch.SearchParameters(method), processor)
+    return processor.isFound
+}
+
+class TactSuperConstantSearch : QueryExecutorBase<TactConstDefinition, DefinitionsScopedSearch.SearchParameters>(true) {
+    override fun processQuery(
+        parameters: DefinitionsScopedSearch.SearchParameters,
+        consumer: Processor<in TactConstDefinition>,
+    ) {
+        if (!parameters.isCheckDeep) return
+
+        val const = parameters.element as? TactConstDefinition ?: return
+        val constName = const.name
+
+        val owner = const.getOwner() ?: return
+        val ownerType = when (owner) {
+            is TactContractDeclaration -> owner.contractType
+            is TactTraitDeclaration    -> owner.traitType
+            else                       -> null
+        } ?: return
+
+        val inheritedTraits = ownerType.getInheritedTraits()
+        if (inheritedTraits.isEmpty()) return
+
+        val superTraitWithConst = inheritedTraits.find { it.traitType.constants().find { const -> const.name == constName } != null }
+        if (superTraitWithConst == null) return
+
+        val superConstant = superTraitWithConst.traitType.constants().find { m -> m.name == constName }
+        consumer.process(superConstant)
+    }
+}
diff --git a/src/main/kotlin/org/tonstudio/tact/lang/search/TactSuperMarkerProvider.kt b/src/main/kotlin/org/tonstudio/tact/lang/search/TactSuperMarkerProvider.kt
@@ -9,6 +9,7 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.impl.source.tree.LeafPsiElement
 import com.intellij.psi.search.searches.DefinitionsScopedSearch
 import org.tonstudio.tact.lang.TactTypes
+import org.tonstudio.tact.lang.psi.TactConstDefinition
 import org.tonstudio.tact.lang.psi.TactFunctionDeclaration
 import java.awt.event.MouseEvent
 
@@ -34,9 +35,9 @@ class TactSuperMarkerProvider : LineMarkerProviderDescriptor() {
     ) {
         if (element !is LeafPsiElement || element.elementType != TactTypes.IDENTIFIER) return
 
-        val func = element.parent as? TactFunctionDeclaration ?: return
+        val func = element.parent as? TactFunctionDeclaration
 
-        if (hasSuperMethod(func)) {
+        if (func != null && hasSuperMethod(func)) {
             result.add(
                 createInfo(
                     element,
@@ -47,6 +48,19 @@ class TactSuperMarkerProvider : LineMarkerProviderDescriptor() {
                 )
             )
         }
+
+        val constant = element.parent as? TactConstDefinition
+        if (constant != null && hasSuperConstant(constant)) {
+            result.add(
+                createInfo(
+                    element,
+                    { e, identifier -> showSuperConstantPopup(e, identifier) },
+                    "Go to Parent Constant",
+                    AllIcons.Gutter.OverridingMethod,
+                    IdeActions.ACTION_GOTO_SUPER,
+                )
+            )
+        }
     }
 }
 
@@ -68,3 +82,22 @@ private fun showSuperMethodPopup(e: MouseEvent?, identifier: PsiElement) {
     val method = identifier.parent as? TactFunctionDeclaration ?: return
     showSuperMethodPopup(e, method)
 }
+
+fun showSuperConstantPopup(event: MouseEvent?, constant: TactConstDefinition) {
+    val name = constant.name
+    showPopup(
+        "Implement constant $name",
+        {
+            "Constant '$name' Overrides Constant of Trait"
+        },
+        event,
+        DefinitionsScopedSearch.SearchParameters(constant),
+        methodRenderer(constant),
+        TactSuperConstantSearch(),
+    )
+}
+
+private fun showSuperConstantPopup(e: MouseEvent?, identifier: PsiElement) {
+    val method = identifier.parent as? TactConstDefinition ?: return
+    showSuperConstantPopup(e, method)
+}
diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
@@ -120,8 +120,10 @@ Provides official support for <a href="https://tact-lang.org/">Tact</a> programm
 
         <!-- region Searches -->
         <definitionsScopedSearch implementation="org.tonstudio.tact.lang.search.TactSuperMethodSearch"/>
+        <definitionsScopedSearch implementation="org.tonstudio.tact.lang.search.TactSuperConstantSearch"/>
         <definitionsScopedSearch implementation="org.tonstudio.tact.lang.search.TactImplementationsSearch"/>
         <definitionsScopedSearch implementation="org.tonstudio.tact.lang.search.TactFunctionImplementationsSearch"/>
+        <definitionsScopedSearch implementation="org.tonstudio.tact.lang.search.TactConstantImplementationsSearch"/>
         <codeInsight.lineMarkerProvider language="tact"
                                         implementationClass="org.tonstudio.tact.lang.search.TactSuperMarkerProvider"/>
         <codeInsight.lineMarkerProvider language="tact"

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ import com.intellij.openapi.editor.Editor`
`5`	`5`	`import com.intellij.openapi.project.Project`
`6`	`6`	`import com.intellij.psi.PsiElement`
`7`	`7`	`import com.intellij.psi.PsiFile`
	`8`	`+import org.tonstudio.tact.lang.psi.TactConstDefinition`
`8`	`9`	`import org.tonstudio.tact.lang.psi.TactFile`
`9`	`10`	`import org.tonstudio.tact.lang.psi.TactFunctionDeclaration`
`10`	`11`
`@@ -23,5 +24,8 @@ class TactGotoSuperHandler : LanguageCodeInsightActionHandler {`
`23`	`24`	`if (parent is TactFunctionDeclaration) {`
`24`	`25`	`showSuperMethodPopup(null, parent)`
`25`	`26`	`}`
	`27`	`+ if (parent is TactConstDefinition) {`
	`28`	`+ showSuperConstantPopup(null, parent)`
	`29`	`+ }`
`26`	`30`	`}`
`27`	`31`	`}`