feat(gui): implement workspace panel for file management #358

phodal · phodal · commit 8b05b456ab79 · 2025-04-03T16:54:57.000+08:00
- Add WorkspacePanel component for managing workspace files
- Integrate workspace functionality into existing components
- Update AutoDevInput and AutoDevInputSection to support workspace
- Modify InputFileToolbar to use workspace for file operations
- Add new string resource for workspace files label JBColor.border()

- Replace UIUtil
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/AutoDevInput.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/AutoDevInput.kt
@@ -60,7 +60,6 @@ class AutoDevInput(
 
         DumbAwareAction.create {
             val editor = editor ?: return@create
-            // Insert a new line
             CommandProcessor.getInstance().executeCommand(project, {
                 val eol = "\n"
                 val document = editor.document
@@ -115,7 +114,6 @@ class AutoDevInput(
         editor.caretModel.moveToOffset(0)
         editor.scrollPane.setBorder(border)
         editor.contentComponent.setOpaque(false)
-
         return editor
     }
 
@@ -160,14 +158,10 @@ class AutoDevInput(
     }
 
     fun appendText(text: String) {
-        WriteCommandAction.runWriteCommandAction(
-            project,
-            "Append text",
-            "intentions.write.action",
-            {
-                val document = this.editor?.document ?: return@runWriteCommandAction
-                InsertUtil.insertStringAndSaveChange(project, text, document, document.textLength, false)
-            })
+        WriteCommandAction.runWriteCommandAction(project, "Append text", "intentions.write.action", {
+            val document = this.editor?.document ?: return@runWriteCommandAction
+            InsertUtil.insertStringAndSaveChange(project, text, document, document.textLength, false)
+        })
     }
 }
 
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/AutoDevInputSection.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/AutoDevInputSection.kt
@@ -70,6 +70,8 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
 
     private val fileListViewModel = FileListViewModel(project)
     private val elementsList = JBList(fileListViewModel.getListModel())
+    
+    private val workspacePanel: WorkspacePanel
 
     private val defaultRag: CustomAgentConfig = CustomAgentConfig("<Select Custom Agent>", "Normal")
     private var customAgent: ComboBox<CustomAgentConfig> = ComboBox(MutableCollectionComboBoxModel(listOf()))
@@ -95,6 +97,8 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
 
     init {
         input = AutoDevInput(project, listOf(), disposable, this)
+        // Initialize WorkspacePanel
+        workspacePanel = WorkspacePanel(project, input)
 
         setupElementsList()
         val sendButtonPresentation = Presentation(AutoDevBundle.message("chat.panel.send"))
@@ -164,6 +168,9 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
 
         inputPanel.add(input, BorderLayout.CENTER)
         inputPanel.addToBottom(layoutPanel)
+        
+        // Add workspace panel to the input panel
+        inputPanel.addToTop(workspacePanel)
 
         val scrollPane = JBScrollPane(elementsList)
         scrollPane.verticalScrollBarPolicy = JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED
@@ -248,7 +255,8 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         val actionType = fileListViewModel.determineFileAction(wrapper, e.point, cellBounds)
         val actionPerformed = fileListViewModel.handleFileAction(wrapper, actionType) { vfile, relativePath ->
             if (relativePath != null) {
-                input.appendText("\n/" + "file" + ":${relativePath}")
+                workspacePanel.addFileToWorkspace(vfile)
+                
                 ApplicationManager.getApplication().invokeLater {
                     if (!vfile.isValid) return@invokeLater
                     val psiFile = PsiManager.getInstance(project).findFile(vfile) ?: return@invokeLater
@@ -361,13 +369,15 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         return customAgent.selectedItem as CustomAgentConfig
     }
 
+    // Reset workspace panel when resetting agent
     fun resetAgent() {
         (customAgent.selectedItem as? CustomAgentConfig)?.let {
             it.state = CustomAgentState.START
         }
 
         customAgent.selectedItem = defaultRag
         text = ""
+        workspacePanel.clear()
     }
 
     fun moveCursorToStart() {
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/InputFileToolbar.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/InputFileToolbar.kt
@@ -10,6 +10,8 @@ import com.intellij.util.ui.JBUI
 import com.intellij.util.ui.UIUtil
 import javax.swing.Box
 import javax.swing.JToolBar
+import java.awt.Component
+import java.awt.Container
 
 object InputFileToolbar {
     fun createToolbar(project: Project, viewModel: FileListViewModel, input: AutoDevInput): JToolBar {
@@ -23,15 +25,39 @@ object InputFileToolbar {
 
         toolbar.add(Box.createHorizontalGlue())
 
+        val findWorkspacePanel: () -> WorkspacePanel? = lambda@{
+            var component: Component? = input.parent
+            while (component != null) {
+                if (component is AutoDevInputSection) {
+                    val workspace = component.findComponentOfType(WorkspacePanel::class.java)
+                    if (workspace != null) {
+                        return@lambda workspace
+                    }
+                }
+
+                component = component.parent
+            }
+            null
+        }
+
         val recentFiles = LinkLabel(AutoDevBundle.message("chat.panel.add.openFiles"), null) { _: LinkLabel<Unit>, _: Unit? ->
             val addedFiles = viewModel.addRecentlyOpenedFiles()
-            val fileReferences = StringBuilder()
-            addedFiles.forEach { vfile ->
-                fileReferences.append("\n/file:${vfile.presentablePath}")
-            }
             
-            if (fileReferences.isNotEmpty()) {
-                input.appendText(fileReferences.toString())
+            val workspace = findWorkspacePanel()
+            if (workspace != null) {
+                addedFiles.forEach { file ->
+                    workspace.addFileToWorkspace(file.virtualFile)
+                }
+            } else {
+                // Fallback to original behavior
+                val fileReferences = StringBuilder()
+                addedFiles.forEach { vfile ->
+                    fileReferences.append("\n/file:${vfile.presentablePath}")
+                }
+                
+                if (fileReferences.isNotEmpty()) {
+                    input.appendText(fileReferences.toString())
+                }
             }
         }
         
@@ -41,6 +67,7 @@ object InputFileToolbar {
 
         val clearAll = LinkLabel(AutoDevBundle.message("chat.panel.clear.all"), null) { _: LinkLabel<Unit>, _: Unit? ->
             viewModel.clearAllFiles()
+            findWorkspacePanel()?.clear()
         }
 
         clearAll.mediumFontFunction()
@@ -51,3 +78,22 @@ object InputFileToolbar {
         return toolbar
     }
 }
+
+fun <T> Component.findComponentOfType(clazz: Class<T>): T? {
+    if (clazz.isInstance(this)) {
+        @Suppress("UNCHECKED_CAST")
+        return this as T
+    }
+    
+    if (this is Container) {
+        for (i in 0 until componentCount) {
+            val component = getComponent(i)
+            val result = component.findComponentOfType(clazz)
+            if (result != null) {
+                return result
+            }
+        }
+    }
+    
+    return null
+}
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/WorkspacePanel.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ui/WorkspacePanel.kt
@@ -0,0 +1,140 @@
+package cc.unitmesh.devti.gui.chat.ui
+
+import cc.unitmesh.devti.AutoDevBundle
+import com.intellij.icons.AllIcons
+import com.intellij.openapi.fileChooser.FileChooser
+import com.intellij.openapi.fileChooser.FileChooserDescriptor
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.ui.JBColor
+import com.intellij.ui.components.JBLabel
+import com.intellij.util.ui.JBUI
+import com.intellij.util.ui.UIUtil
+import java.awt.BorderLayout
+import java.awt.FlowLayout
+import java.awt.Font
+import java.awt.event.MouseAdapter
+import java.awt.event.MouseEvent
+import javax.swing.BorderFactory
+import javax.swing.BoxLayout
+import javax.swing.JPanel
+
+class WorkspacePanel(
+    private val project: Project,
+    private val input: AutoDevInput
+) : JPanel(BorderLayout()) {
+    private val workspaceFiles = mutableListOf<VirtualFile>()
+    private val filesPanel = JPanel()
+    
+    init {
+        border = BorderFactory.createCompoundBorder(
+            BorderFactory.createMatteBorder(1, 0, 0, 0, JBColor.border()),
+            JBUI.Borders.empty(10)
+        )
+        
+        val workspaceLabel = JBLabel(AutoDevBundle.message("chat.panel.workspace.files", "Edit files in your workspace"))
+        workspaceLabel.foreground = UIUtil.getContextHelpForeground()
+        workspaceLabel.font = Font(workspaceLabel.font.family, Font.PLAIN, 12)
+        
+        val addButton = JBLabel(AllIcons.General.Add)
+        addButton.cursor = java.awt.Cursor(java.awt.Cursor.HAND_CURSOR)
+        addButton.toolTipText = "Add files to workspace"
+        addButton.addMouseListener(object : MouseAdapter() {
+            override fun mouseClicked(e: MouseEvent) {
+                addFile()
+            }
+        })
+        
+        filesPanel.layout = BoxLayout(filesPanel, BoxLayout.Y_AXIS)
+        
+        val topPanel = JPanel(BorderLayout())
+        topPanel.isOpaque = false
+        topPanel.add(workspaceLabel, BorderLayout.WEST)
+        topPanel.add(addButton, BorderLayout.EAST)
+        
+        add(topPanel, BorderLayout.NORTH)
+        add(filesPanel, BorderLayout.CENTER)
+        isOpaque = false
+        filesPanel.isOpaque = false
+    }
+    
+    private fun addFile() {
+        val descriptor = FileChooserDescriptor(true, true, false, false, false, true)
+            .withTitle("Select Files for Workspace")
+            .withDescription("Choose files to add to your workspace")
+        
+        FileChooser.chooseFiles(descriptor, project, null) { files ->
+            for (file in files) {
+                addFileToWorkspace(file)
+            }
+        }
+    }
+    
+    fun addFileToWorkspace(file: VirtualFile) {
+        if (!workspaceFiles.contains(file)) {
+            workspaceFiles.add(file)
+            updateFilesPanel()
+            
+            // Add file reference to the input (will be visible on submit)
+            val relativePath = try {
+                project.basePath?.let { basePath ->
+                    file.path.substringAfter(basePath).removePrefix("/")
+                } ?: file.path
+            } catch (e: Exception) {
+                file.path
+            }
+            
+            input.appendText("\n/file:$relativePath")
+        }
+    }
+    
+    private fun updateFilesPanel() {
+        filesPanel.removeAll()
+        
+        for (file in workspaceFiles) {
+            val fileLabel = FileItemPanel(project, file) { 
+                removeFile(file)
+            }
+            filesPanel.add(fileLabel)
+        }
+        
+        filesPanel.revalidate()
+        filesPanel.repaint()
+    }
+    
+    private fun removeFile(file: VirtualFile) {
+        workspaceFiles.remove(file)
+        updateFilesPanel()
+    }
+    
+    fun clear() {
+        workspaceFiles.clear()
+        updateFilesPanel()
+    }
+}
+
+class FileItemPanel(
+    private val project: Project,
+    private val file: VirtualFile,
+    private val onRemove: () -> Unit
+) : JPanel(FlowLayout(FlowLayout.LEFT, 5, 2)) {
+    
+    init {
+        border = JBUI.Borders.empty(2)
+        isOpaque = false
+        
+        val icon = file.fileType.icon
+        val fileLabel = JBLabel(file.name, icon, JBLabel.LEFT)
+        
+        val removeLabel = JBLabel(AllIcons.Actions.Close)
+        removeLabel.cursor = java.awt.Cursor(java.awt.Cursor.HAND_CURSOR)
+        removeLabel.addMouseListener(object : MouseAdapter() {
+            override fun mouseClicked(e: MouseEvent) {
+                onRemove()
+            }
+        })
+        
+        add(fileLabel)
+        add(removeLabel)
+    }
+}
diff --git a/core/src/main/resources/messages/AutoDevBundle_en.properties b/core/src/main/resources/messages/AutoDevBundle_en.properties
@@ -216,3 +216,4 @@ planner.task.status.in_progress=In Progress
 planner.task.status.todo=To Do
 planner.task.execute=Execute
 chat.panel.add.openFiles=Add all open files
+chat.panel.workspace.files=Workspace File
diff --git a/core/src/main/resources/messages/AutoDevBundle_zh.properties b/core/src/main/resources/messages/AutoDevBundle_zh.properties
@@ -207,4 +207,5 @@ planner.task.status.failed=已失败
 planner.task.status.in_progress=进行中
 planner.task.status.todo=待办
 planner.task.execute=执行
-chat.panel.add.openFiles=添加所有打开的文件
+chat.panel.add.openFiles=添加所有打开的文件
+chat.panel.workspace.files=Workspace File
