feat(build): add support for detecting package files in build systems #316

Add `isDeclarePackageFile` method to `BuildSystemProvider` to identify package files (e.g., `requirements.txt`, `package.json`, `go.mod`). This enables proper UI handling and tooling for dependency management across various build systems.

Author: phodal

core/src/223/main/resources/META-INF/autodev-core.xml

@@ -336,6 +336,11 @@
             <add-to-group group-id="EditorPopupMenu" anchor="first"/>
         </action>
 
+        <group id="AutoDev.ToolWindow.Snippet.DependenciesToolbar">
+            <action id="AutoDev.ToolWindow.Snippet.DependenciesLabelAction"
+                    class="cc.unitmesh.devti.gui.snippet.AutoDevDependenciesLabelAction"/>
+        </group>
+
         <group id="AutoDev.ToolWindow.Snippet.Toolbar">
             <action id="AutoDev.ToolWindow.Snippet.LanguageLabelAction"
                     class="cc.unitmesh.devti.gui.snippet.AutoDevLanguageLabelAction"/>

core/src/233/main/resources/META-INF/autodev-core.xml

@@ -336,6 +336,10 @@
 
             <add-to-group group-id="EditorPopupMenu" anchor="first"/>
         </action>
+        <group id="AutoDev.ToolWindow.Snippet.DependenciesToolbar">
+            <action id="AutoDev.ToolWindow.Snippet.DependenciesLabelAction"
+                    class="cc.unitmesh.devti.gui.snippet.AutoDevDependenciesLabelAction"/>
+        </group>
 
         <group id="AutoDev.ToolWindow.Snippet.Toolbar">
             <action id="AutoDev.ToolWindow.Snippet.LanguageLabelAction"

core/src/main/kotlin/cc/unitmesh/devti/gui/snippet/AutoDevDependenciesLabelAction.kt

@@ -0,0 +1,49 @@
+package cc.unitmesh.devti.gui.snippet
+
+import com.intellij.openapi.actionSystem.ActionUpdateThread
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.CommonDataKeys
+import com.intellij.openapi.actionSystem.Presentation
+import com.intellij.openapi.actionSystem.ex.CustomComponentAction
+import com.intellij.openapi.fileEditor.FileDocumentManager
+import com.intellij.openapi.project.DumbAwareAction
+import com.intellij.openapi.util.Key
+import com.intellij.testFramework.LightVirtualFile
+import com.intellij.ui.components.JBLabel
+import com.intellij.util.ui.UIUtil
+import javax.swing.JComponent
+
+class AutoDevDependenciesLabelAction : DumbAwareAction(), CustomComponentAction {
+    override fun getActionUpdateThread(): ActionUpdateThread = ActionUpdateThread.BGT
+
+    override fun createCustomComponent(presentation: Presentation, place: String): JComponent {
+        val languageId = presentation.getClientProperty(DEPENDENCIES_PRESENTATION_KEY) ?: ""
+        val jBLabel: JComponent = JBLabel(languageId)
+        jBLabel.setOpaque(false)
+        jBLabel.setForeground(UIUtil.getLabelInfoForeground())
+        return jBLabel
+    }
+
+    override fun updateCustomComponent(component: JComponent, presentation: Presentation) {
+        if (component !is JBLabel) return
+
+        val filename = presentation.getClientProperty(DEPENDENCIES_PRESENTATION_KEY) ?: ""
+        if (filename.isNotBlank()) {
+            component.text = "SCA: $filename"
+        }
+    }
+
+    override fun actionPerformed(e: AnActionEvent) {
+
+    }
+
+    override fun update(e: AnActionEvent) {
+        val editor = e.dataContext.getData(CommonDataKeys.EDITOR) ?: return
+        val lightVirtualFile = FileDocumentManager.getInstance().getFile(editor.document) as? LightVirtualFile ?: return
+        e.presentation.putClientProperty(DEPENDENCIES_PRESENTATION_KEY, lightVirtualFile.name)
+    }
+
+    companion object {
+        val DEPENDENCIES_PRESENTATION_KEY: Key<String> = Key.create("DependenciesPresentationKey")
+    }
+}

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

@@ -16,6 +16,11 @@ import com.intellij.util.xmlb.annotations.Attribute
 abstract class BuildSystemProvider : LazyExtensionInstance<BuildSystemProvider>() {
     abstract fun collect(project: Project): DockerfileContext?
 
+    /**
+     * the DeclarePackageFile means `build.gradle`, `pom.xml`, `build.sbt`, `package.json` etc.
+     */
+    abstract fun isDeclarePackageFile(filename: String): Boolean
+
     /**
      * For PsiFile only for resolve in Sketch and Bridge mode
      * the BuildFilePsiFile means `build.gradle`, `pom.xml`, `build.sbt`, `package.json` etc.
@@ -40,6 +45,14 @@ abstract class BuildSystemProvider : LazyExtensionInstance<BuildSystemProvider>(
                 it.collect(project)
             }
         }
+
+        fun isDeclarePackageFile(filename: String?): Boolean {
+            if (filename == null) return false
+
+            return EP_NAME.extensionList.any {
+                it.isDeclarePackageFile(filename)
+            }
+        }
     }
 }
 

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/LangSketch.kt

@@ -6,6 +6,7 @@ import com.intellij.openapi.actionSystem.ActionGroup
 import com.intellij.openapi.actionSystem.ActionManager
 import com.intellij.openapi.actionSystem.ActionPlaces
 import com.intellij.openapi.actionSystem.ActionToolbar
+import com.intellij.openapi.actionSystem.DefaultActionGroup
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.editor.colors.EditorColorsListener
 import com.intellij.openapi.editor.colors.EditorColorsManager
@@ -25,12 +26,13 @@ interface LangSketch : Disposable {
      */
     fun onDoneStream(allText: String) {}
 
-    fun setupActionBar(project: Project, editor: Editor) {
-        val toolbar = actionToolbar() ?: return
+    fun setupActionBar(project: Project, editor: Editor, isDeclarePackageFile: Boolean) {
+        val toolbar = collectActionBar(isDeclarePackageFile) ?: return
 
         if (editor is EditorEx) {
             toolbar.component.setBackground(editor.backgroundColor)
         }
+
         toolbar.component.setOpaque(true)
         toolbar.targetComponent = editor.contentComponent
         editor.headerComponent = toolbar.component
@@ -44,9 +46,14 @@ interface LangSketch : Disposable {
         })
     }
 
-    fun actionToolbar(): ActionToolbar? {
-        val toolbarActionGroup = ActionManager.getInstance().getAction("AutoDev.ToolWindow.Snippet.Toolbar") as? ActionGroup
-            ?: return null
+    fun collectActionBar(isDeclarePackageFile: Boolean): ActionToolbar? {
+        val toolbarActionGroup = if (isDeclarePackageFile) {
+            ActionManager.getInstance().getAction("AutoDev.ToolWindow.Snippet.DependenciesToolbar") as? ActionGroup
+                ?: return null
+        } else {
+            ActionManager.getInstance().getAction("AutoDev.ToolWindow.Snippet.Toolbar") as? ActionGroup
+                ?: return null
+        }
 
         val toolbar = ActionManager.getInstance()
             .createActionToolbar(ActionPlaces.MAIN_TOOLBAR, toolbarActionGroup, true)

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/code/CodeHighlightSketch.kt

@@ -2,8 +2,12 @@ package cc.unitmesh.devti.sketch.ui.code
 
 import cc.unitmesh.devti.AutoDevNotifications
 import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
+import cc.unitmesh.devti.provider.BuildSystemProvider
 import cc.unitmesh.devti.provider.RunService
+import cc.unitmesh.devti.sketch.ui.LangSketch
+import cc.unitmesh.devti.sketch.ui.LanguageSketchProvider
 import cc.unitmesh.devti.util.parser.CodeFence
+import com.intellij.ide.scratch.ScratchRootType
 import com.intellij.lang.Language
 import com.intellij.openapi.Disposable
 import com.intellij.openapi.actionSystem.DataProvider
@@ -20,24 +24,21 @@ import com.intellij.openapi.editor.ex.FocusChangeListener
 import com.intellij.openapi.editor.ex.MarkupModelEx
 import com.intellij.openapi.editor.highlighter.EditorHighlighterFactory
 import com.intellij.openapi.fileEditor.FileDocumentManager
+import com.intellij.openapi.fileEditor.FileEditor
+import com.intellij.openapi.fileEditor.FileEditorProvider
+import com.intellij.openapi.fileEditor.TextEditorWithPreview
 import com.intellij.openapi.fileTypes.PlainTextLanguage
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.util.text.StringUtil
 import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.psi.PsiManager
+import com.intellij.temporary.gui.block.whenDisposed
 import com.intellij.testFramework.LightVirtualFile
 import com.intellij.ui.JBColor
 import com.intellij.ui.components.JBPanel
 import com.intellij.util.concurrency.annotations.RequiresReadLock
-import com.intellij.util.ui.JBUI
-import cc.unitmesh.devti.sketch.ui.LangSketch
-import cc.unitmesh.devti.sketch.ui.LanguageSketchProvider
-import com.intellij.ide.scratch.ScratchRootType
-import com.intellij.openapi.fileEditor.FileEditor
-import com.intellij.openapi.fileEditor.FileEditorProvider
-import com.intellij.openapi.fileEditor.TextEditorWithPreview
-import com.intellij.openapi.util.text.StringUtil
-import com.intellij.psi.PsiManager
-import com.intellij.temporary.gui.block.whenDisposed
 import com.intellij.util.ui.JBEmptyBorder
+import com.intellij.util.ui.JBUI
 import java.awt.BorderLayout
 import javax.swing.BoxLayout
 import javax.swing.JButton
@@ -95,8 +96,9 @@ open class CodeHighlightSketch(
 
         add(editorFragment!!.getContent(), BorderLayout.CENTER)
 
+        val isDeclarePackageFile = BuildSystemProvider.isDeclarePackageFile(fileName)
         if (textLanguage != null && textLanguage?.lowercase() != "markdown" && ideaLanguage != PlainTextLanguage.INSTANCE) {
-            setupActionBar(project, editor)
+            setupActionBar(project, editor, isDeclarePackageFile)
         } else {
             editor.backgroundColor = JBColor.PanelBackground
         }
@@ -161,15 +163,17 @@ open class CodeHighlightSketch(
 
         val currentText = getViewText()
         if (currentText.startsWith("/" + BuiltinCommand.WRITE.commandName + ":")) {
-            /// get fileName after : and before \n
             processWriteCommand(currentText)
+            /// get fileName after : and before \n
             val fileName = currentText.lines().firstOrNull()?.substringAfter(":")
-            val ext = fileName?.substringAfterLast(".")
-            val parse = CodeFence.parse(editorFragment!!.editor.document.text)
-            val language = if (ext != null) CodeFence.findLanguage(ext) else ideaLanguage
-            val sketch = CodeHighlightSketch(project, parse.text, language, editorLineThreshold, fileName)
-            add(sketch, BorderLayout.SOUTH)
-            return
+            if (BuildSystemProvider.isDeclarePackageFile(fileName)) {
+                val ext = fileName?.substringAfterLast(".")
+                val parse = CodeFence.parse(editorFragment!!.editor.document.text)
+                val language = if (ext != null) CodeFence.findLanguage(ext) else ideaLanguage
+                val sketch = CodeHighlightSketch(project, parse.text, language, editorLineThreshold, fileName)
+                add(sketch, BorderLayout.SOUTH)
+                return
+            }
         }
 
         val parse = CodeFence.parse(editorFragment!!.editor.document.text)

exts/ext-dependencies/src/233/main/resources/cc.unitmesh.dependencies.xml

@@ -9,7 +9,7 @@
                 icon="AllIcons.Actions.DependencyAnalyzer"
                 class="cc.unitmesh.dependencies.AutoDevDependenciesCheck">
 
-            <add-to-group group-id="AutoDev.ToolWindow.Snippet.Toolbar" />
+            <add-to-group group-id="AutoDev.ToolWindow.Snippet.DependenciesToolbar" />
         </action>
     </actions>
 

goland/src/main/kotlin/cc/unitmesh/go/provider/GoBuildSystemProvider.kt

@@ -20,6 +20,10 @@ class GoBuildSystemProvider : BuildSystemProvider() {
         return null
     }
 
+    override fun isDeclarePackageFile(filename: String): Boolean {
+        return filename == "go.mod"
+    }
+
     override fun collectDependencies(
         project: Project,
         buildFilePsi: PsiFile

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

@@ -55,6 +55,10 @@ open class JavaBuildSystemProvider : BuildSystemProvider() {
         )
     }
 
+    override fun isDeclarePackageFile(filename: String): Boolean {
+        return filename == "build.gradle" || filename == "pom.xml" || filename == "build.gradle.kts"
+    }
+
     override fun collectDependencies(
         project: Project,
         buildFilePsi: PsiFile

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/JavaScriptBuildSystemProvider.kt

@@ -87,6 +87,10 @@ class JavaScriptBuildSystemProvider : BuildSystemProvider() {
         )
     }
 
+    override fun isDeclarePackageFile(filename: String): Boolean {
+        return filename == "package.json"
+    }
+
     override fun collectDependencies(project: Project, buildFilePsi: PsiFile): List<DevPackage> {
         val packageJson = buildFilePsi as? JsonFile ?: return emptyList()
         return PackageJsonUtil.getDependencies(packageJson, PackageJsonUtil.PROD_DEV_DEPENDENCIES)

pycharm/src/main/kotlin/cc/unitmesh/python/provider/PythonBuildSystemProvider.kt

@@ -11,6 +11,10 @@ import com.jetbrains.python.packaging.PyRequirementParser
 class PythonBuildSystemProvider : BuildSystemProvider() {
     override fun collect(project: Project): DockerfileContext? = null
 
+    override fun isDeclarePackageFile(filename: String): Boolean {
+        return filename == "requirements.txt" || filename == "Pipfile" || filename == "pyproject.toml"
+    }
+
     override fun collectDependencies(project: Project, buildFilePsi: PsiFile): List<DevPackage> {
         val reqs: List<PyRequirement> = PyRequirementParser.fromFile(buildFilePsi.virtualFile)