feat(compiler): add DIR command for listing files and directories #257

- Introduce DIR command to list files and directories in a tree-like structure.
- Implement DirInsCommand to handle directory listing logic.
- Update BuiltinCommand enum to include DIR command.

Author: phodal

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

@@ -231,6 +231,11 @@ class DevInsCompiler(
                 RefactorInsCommand(myProject, prop, nextTextSegment)
             }
 
+            BuiltinCommand.DIR -> {
+                result.isLocalCommand = true
+                DirInsCommand(myProject, prop)
+            }
+
             else -> {
                 PrintInsCommand("/" + commandNode.commandName + ":" + prop)
             }

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

@@ -0,0 +1,46 @@
+package cc.unitmesh.devti.language.compiler.exec
+
+import cc.unitmesh.devti.language.utils.lookupFile
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiDirectory
+import com.intellij.psi.PsiManager
+
+/**
+ *  Dir List files and directories in a tree-like structure
+ */
+class DirInsCommand(private val myProject: Project, private val dir: String) : InsCommand {
+    private val output = StringBuilder()
+
+    override suspend fun execute(): String? {
+        val virtualFile = myProject.lookupFile(dir) ?: return "File not found: $dir"
+        val psiDirectory = PsiManager.getInstance(myProject).findDirectory(virtualFile) ?: return null
+
+        output.appendLine("$dir/")
+        listDirectory(psiDirectory, 1)
+
+        return output.toString()
+    }
+
+    private fun listDirectory(directory: PsiDirectory, depth: Int) {
+        val files = directory.files
+        val subdirectories = directory.subdirectories
+
+        for ((index, file) in files.withIndex()) {
+            if (index == files.size - 1) {
+                output.appendLine("${"  ".repeat(depth)}└── ${file.name}")
+            } else {
+                output.appendLine("${"  ".repeat(depth)}├── ${file.name}")
+            }
+        }
+
+        for ((index, subdirectory) in subdirectories.withIndex()) {
+            if (index == subdirectories.size - 1) {
+                output.appendLine("${"  ".repeat(depth)}└── ${subdirectory.name}/")
+            } else {
+                output.appendLine("${"  ".repeat(depth)}├── ${subdirectory.name}/")
+            }
+            listDirectory(subdirectory, depth + 1)
+        }
+    }
+}
+

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

@@ -41,20 +41,9 @@ enum class BuiltinCommand(
         true,
         true
     ),
-    BROWSE(
-        "browse",
-        "Get the content of a given URL",
-        AllIcons.Toolwindows.WebToolWindow,
-        true,
-        true
-    ),
-    Refactor(
-        "refactor",
-        "Refactor the content of a file",
-        AutoDevIcons.Idea,
-        true,
-        true
-    ),
+    BROWSE("browse", "Get the content of a given URL", AllIcons.Toolwindows.WebToolWindow, true, true),
+    Refactor("refactor", "Refactor the content of a file", AutoDevIcons.Idea, true, true),
+    DIR("dir", "List files and directories in a tree-like structure", AllIcons.Actions.ProjectDirectory, true, true),
     ;
 
     companion object {