feat(devti): implement caller lookup functionality #358

- Add lookupCaller function to RelatedClassesProvider interface
- Implement caller lookup for Java and Kotlin
- Update UsageInsCommand to use lookupCaller instead of lookupCallee
- Improve output formatting for UsageInsCommand

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/provider/RelatedClassesProvider.kt

@@ -2,10 +2,15 @@ package cc.unitmesh.devti.provider
 
 import com.intellij.lang.Language
 import com.intellij.lang.LanguageExtension
+import com.intellij.openapi.progress.ProgressManager
+import com.intellij.openapi.progress.util.ProgressIndicatorBase
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
+import com.intellij.psi.PsiMethod
 import com.intellij.psi.PsiNamedElement
+import com.intellij.psi.search.searches.MethodReferencesSearch
+import com.intellij.psi.util.PsiTreeUtil
 
 /**
  * The `RelatedClassesProvider` interface is used to provide related classes for a given element.
@@ -33,6 +38,8 @@ interface RelatedClassesProvider {
 
     fun lookupCallee(project: Project, element: PsiElement): List<PsiNamedElement> = emptyList()
 
+    fun lookupCaller(project: Project, element: PsiElement): List<PsiNamedElement> = emptyList()
+
     companion object {
         private val languageExtension: LanguageExtension<RelatedClassesProvider> =
             LanguageExtension("cc.unitmesh.relatedClassProvider")

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/exec/UsageInsCommand.kt

@@ -19,11 +19,13 @@ class UsageInsCommand(val myProject: Project, private val symbol: String) : InsC
         if (elements.isEmpty()) return "$DEVINS_ERROR: No symbol found for $symbol"
 
         val psiElements = elements.mapNotNull {
-            RelatedClassesProvider.provide(it.language)?.lookupCallee(myProject, it)
+            RelatedClassesProvider.provide(it.language)?.lookupCaller(myProject, it)
         }.flatten()
 
         if (psiElements.isEmpty()) return "$DEVINS_ERROR: No usage found for $symbol"
 
-        return psiElements.joinToString("\n") { it.text }
+        return "Here is related to $symbol usage" + psiElements.joinToString("\n\n") {
+            runReadAction { it.text }
+        }
     }
 }

java/src/main/kotlin/cc/unitmesh/idea/provider/JavaRelatedClassesProvider.kt

@@ -3,7 +3,7 @@ package cc.unitmesh.idea.provider
 import cc.unitmesh.devti.provider.RelatedClassesProvider
 import cc.unitmesh.idea.context.JavaContextCollection
 import cc.unitmesh.idea.service.JavaTypeUtil.resolveByType
-import cc.unitmesh.idea.util.JavaCallHelper.findCallees
+import cc.unitmesh.idea.util.JavaCallHelper
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.project.Project
@@ -32,7 +32,17 @@ class JavaRelatedClassesProvider : RelatedClassesProvider {
 
     override fun lookupCallee(project: Project, element: PsiElement): List<PsiNamedElement> {
         return when (element) {
-            is PsiMethod -> runReadAction { findCallees(project, element) }
+            is PsiMethod -> runReadAction { JavaCallHelper.findCallees(project, element) }
+            else -> emptyList()
+        }
+    }
+
+    override fun lookupCaller(
+        project: Project,
+        element: PsiElement
+    ): List<PsiNamedElement> {
+        return when (element) {
+            is PsiMethod -> runReadAction { JavaCallHelper.findCallers(project, element) }
             else -> emptyList()
         }
     }

kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinRelatedClassesProvider.kt

@@ -6,6 +6,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 import com.intellij.psi.PsiNamedElement
+import com.intellij.psi.search.searches.ReferencesSearch
 import org.jetbrains.kotlin.idea.structuralsearch.visitor.KotlinRecursiveElementWalkingVisitor
 import org.jetbrains.kotlin.psi.KtCallExpression
 import org.jetbrains.kotlin.psi.KtNamedFunction
@@ -26,6 +27,16 @@ class KotlinRelatedClassProvider : RelatedClassesProvider {
         }
     }
 
+    override fun lookupCaller(
+        project: Project,
+        element: PsiElement
+    ): List<PsiNamedElement> {
+        return when (element) {
+            is KtNamedFunction -> findCallers(project, element)
+            else -> emptyList()
+        }
+    }
+
     fun findCallees(project: Project, method: KtNamedFunction): List<KtNamedFunction> {
         val calledMethods = mutableSetOf<KtNamedFunction>()
         method.accept(object : KotlinRecursiveElementWalkingVisitor() {
@@ -39,4 +50,30 @@ class KotlinRelatedClassProvider : RelatedClassesProvider {
 
         return calledMethods.toList()
     }
+    
+    fun findCallers(project: Project, method: KtNamedFunction): List<KtNamedFunction> {
+        val callers = mutableSetOf<KtNamedFunction>()
+        val references = ReferencesSearch.search(method).findAll()
+        
+        for (reference in references) {
+            val element = reference.element
+            
+            var parentFunction: KtNamedFunction? = null
+            var parent = element.parent
+            
+            while (parent != null) {
+                if (parent is KtNamedFunction) {
+                    parentFunction = parent
+                    break
+                }
+                parent = parent.parent
+            }
+            
+            if (parentFunction != null && parentFunction != method) {
+                callers.add(parentFunction)
+            }
+        }
+        
+        return callers.toList()
+    }
 }