feat(devin-lang): add FileAutoCommand and refactor DevInCompiler to support dynamic file content retrieval. #101

This commit introduces a new class `FileAutoCommand` which is responsible for executing commands that involve dynamic file content. It handles the retrieval of file content from the specified path, and if a range is specified, it will only return the specified portion of the file. The `DevInCompiler` class has been refactored to use this new command, and a fallback text is now provided in case the file content cannot be retrieved. This commit also removes unnecessary imports and simplifies the codebase.

Author: phodal

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/AutoCommand.kt

@@ -0,0 +1,5 @@
+package cc.unitmesh.devti.language.compiler
+
+interface AutoCommand {
+    fun execute(): String?
+}

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/DevInCompiler.kt

@@ -7,17 +7,8 @@ import cc.unitmesh.devti.language.psi.DevInUsed
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
-import com.intellij.openapi.project.guessProjectDir
-import com.intellij.openapi.roots.ProjectFileIndex
 import com.intellij.openapi.util.NlsSafe
-import com.intellij.openapi.util.TextRange
-import com.intellij.openapi.vfs.LocalFileSystem
-import com.intellij.openapi.vfs.VirtualFile
-import com.intellij.openapi.vfs.VirtualFileManager
-import com.intellij.psi.PsiManager
-import com.intellij.psi.search.FilenameIndex
 import com.intellij.psi.util.elementType
-import kotlin.io.path.readText
 
 class DevInCompiler(val myProject: Project, val file: DevInFile, val editor: Editor? = null) {
     private val logger = logger<DevInCompiler>()
@@ -68,7 +59,7 @@ class DevInCompiler(val myProject: Project, val file: DevInFile, val editor: Edi
                     return
                 }
 
-                processingCommand(command, propElement!!.text)
+                processingCommand(command, propElement!!.text, fallbackText = used.text)
             }
 
             DevInTypes.AGENT_START -> {
@@ -90,34 +81,11 @@ class DevInCompiler(val myProject: Project, val file: DevInFile, val editor: Edi
         }
     }
 
-    private fun processingCommand(command: BuiltinCommand, prop: @NlsSafe String) {
+    private fun processingCommand(command: BuiltinCommand, prop: String, fallbackText: String) {
         when (command) {
             BuiltinCommand.FILE -> {
-                val range: TextRange? = if (prop.contains("#")) {
-                    val rangeStr = prop.substringAfter("#")
-                    val start = rangeStr.substringBefore("-").toInt()
-                    val end = rangeStr.substringAfter("-").toInt()
-                    TextRange(start, end)
-                } else {
-                    null
-                }
-
-                val filepath = prop.trim()
-                val projectPath = myProject.guessProjectDir()?.toNioPath()
-                val realpath = projectPath?.resolve(filepath)
-                val content = realpath?.readText()
-
-                content?.let {
-                    val virtualFile: VirtualFile? = VirtualFileManager.getInstance().findFileByNioPath(realpath)
-                    val lang = virtualFile?.let {
-                        PsiManager.getInstance(myProject).findFile(it)?.language?.displayName
-                    } ?: ""
-
-                    output.append("\n```$lang\n")
-                    output.append(content)
-                    output.append("\n```\n")
-                }
-
+                val result = FileAutoCommand(myProject, prop).execute() ?: fallbackText
+                output.append(result)
             }
 
             BuiltinCommand.REV -> {
@@ -136,4 +104,5 @@ class DevInCompiler(val myProject: Project, val file: DevInFile, val editor: Edi
             }
         }
     }
+
 }

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/FileAutoCommand.kt

@@ -0,0 +1,61 @@
+package cc.unitmesh.devti.language.compiler
+
+import com.intellij.openapi.diagnostic.logger
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.project.guessProjectDir
+import com.intellij.openapi.util.TextRange
+import com.intellij.openapi.vfs.VirtualFileManager
+import com.intellij.psi.PsiManager
+
+class FileAutoCommand(val myProject: Project, val prop: String) : AutoCommand {
+    private val logger = logger<FileAutoCommand>()
+    private val output = StringBuilder()
+
+    override fun execute(): String? {
+        val range: TextRange? = if (prop.contains("#")) {
+            val rangeStr = prop.substringAfter("#")
+            val start = rangeStr.substringBefore("-").toInt()
+            val end = rangeStr.substringAfter("-").toInt()
+            TextRange(start, end)
+        } else {
+            null
+        }
+
+        val projectPath = myProject.guessProjectDir()?.toNioPath()
+        val realpath = projectPath?.resolve(prop.trim())
+
+        val virtualFile =
+            VirtualFileManager.getInstance().findFileByUrl("file://${realpath?.toAbsolutePath()}")
+
+        val contentsToByteArray = virtualFile?.contentsToByteArray()
+        if (contentsToByteArray == null) {
+            logger.warn("File not found: $realpath")
+            return null
+        }
+
+        contentsToByteArray.let {
+            val lang = virtualFile.let {
+                PsiManager.getInstance(myProject).findFile(it)?.language?.displayName
+            } ?: ""
+
+            val content = it.toString(Charsets.UTF_8)
+            val fileContent = if (range != null) {
+                val subContent = try {
+                    content.substring(range.startOffset, range.endOffset)
+                } catch (e: StringIndexOutOfBoundsException) {
+                    content
+                }
+
+                subContent
+            } else {
+                content
+            }
+
+            output.append("\n```$lang\n")
+            output.append(fileContent)
+            output.append("\n```\n")
+        }
+
+        return output.toString()
+    }
+}