feat(devins): add usage command to get class, method, or field usage in the project #358

- Add USAGE command to BuiltinCommand enum
- Implement UsageInsCommand to handle usage retrieval
- Update DevInsCompiler to support the new command
- Extend completion contributor for USAGE command support
- Add example for usage command in toolExamples

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/devin/dataprovider/BuiltinCommand.kt

@@ -121,6 +121,13 @@ enum class BuiltinCommand(
         true,
         true
     ),
+    USAGE(
+        "usage",
+        "Get the usage of a class, method, or field in the project",
+        AllIcons.Actions.DynamicUsages,
+        true,
+        true
+    ),
     TOOLCHAIN_COMMAND(
         "x",
         "Execute custom toolchain command",

core/src/main/kotlin/cc/unitmesh/devti/sketch/AutoSketchMode.kt

@@ -85,6 +85,7 @@ class AutoSketchMode(val project: Project) {
             RELATED,
             RIPGREP_SEARCH,
             RULE,
+            USAGE,
             BROWSE,
         )
 

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

@@ -310,6 +310,11 @@ class DevInsCompiler(
             RuleInsCommand(myProject, prop)
         }
 
+        BuiltinCommand.USAGE -> {
+            result.isLocalCommand = true
+            UsageInsCommand(myProject, prop)
+        }
+
         BuiltinCommand.OPEN -> {
             result.isLocalCommand = true
             OpenInsCommand(myProject, prop)

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

@@ -0,0 +1,29 @@
+package cc.unitmesh.devti.language.compiler.exec
+
+import cc.unitmesh.devti.devin.InsCommand
+import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
+import cc.unitmesh.devti.language.compiler.error.DEVINS_ERROR
+import cc.unitmesh.devti.provider.RelatedClassesProvider
+import cc.unitmesh.devti.provider.devins.DevInsSymbolProvider
+import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.project.Project
+
+class UsageInsCommand(val myProject: Project, private val symbol: String) : InsCommand {
+    override val commandName: BuiltinCommand = BuiltinCommand.RELATED
+
+    override suspend fun execute(): String? {
+        val elements = DevInsSymbolProvider.all().map {
+            runReadAction { it.resolveElement(myProject, symbol) }
+        }.flatten()
+
+        if (elements.isEmpty()) return "$DEVINS_ERROR: No symbol found for $symbol"
+
+        val psiElements = elements.mapNotNull {
+            RelatedClassesProvider.provide(it.language)?.lookupCallee(myProject, it)
+        }.flatten()
+
+        if (psiElements.isEmpty()) return "$DEVINS_ERROR: No usage found for $symbol"
+
+        return psiElements.joinToString("\n") { it.text }
+    }
+}

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/DevInCompletionContributor.kt

@@ -38,7 +38,7 @@ class DevInCompletionContributor : CompletionContributor() {
         )
         extend(
             CompletionType.BASIC,
-            (valuePatterns(listOf(BuiltinCommand.SYMBOL, BuiltinCommand.RELATED))),
+            (valuePatterns(listOf(BuiltinCommand.SYMBOL, BuiltinCommand.RELATED, BuiltinCommand.USAGE))),
             SymbolReferenceLanguageProvider()
         )
         extend(

exts/devins-lang/src/main/resources/agent/toolExamples/usage.devin

@@ -0,0 +1 @@
+/usage:cc.unitmesh.untitled.demo.service.BlogService.createBlog