feat(ext-container): enhance Docker context provider with error handling #306

- Add try-catch block to handle exceptions during Docker context collection
- Include

Author: phodal

exts/ext-container/src/233/main/kotlin/cc/unitmesh/container/provider/DockerContextProvider.kt

@@ -3,10 +3,10 @@ package cc.unitmesh.container.provider
 import cc.unitmesh.devti.provider.context.ChatContextItem
 import cc.unitmesh.devti.provider.context.ChatContextProvider
 import cc.unitmesh.devti.provider.context.ChatCreationContext
-import com.github.dockerjava.core.dockerfile.Dockerfile
 import com.intellij.docker.DockerFileSearch
 import com.intellij.docker.dockerFile.DockerPsiFile
 import com.intellij.docker.dockerFile.parser.psi.DockerFileFromCommand
+import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiManager
 import com.intellij.psi.util.PsiTreeUtil
@@ -19,23 +19,32 @@ class DockerContextProvider : ChatContextProvider {
         project: Project,
         creationContext: ChatCreationContext
     ): List<ChatContextItem> {
-        val dockerFiles = DockerFileSearch.getInstance().getDockerFiles(project).mapNotNull {
-            PsiManager.getInstance(project).findFile(it) as? DockerPsiFile
-        }
+        try {
+            val dockerFiles = DockerFileSearch.getInstance().getDockerFiles(project).mapNotNull {
+                PsiManager.getInstance(project).findFile(it) as? DockerPsiFile
+            }
+
+            if (dockerFiles.isEmpty()) return emptyList()
 
-        if (dockerFiles.isEmpty()) return emptyList()
+            val virtualFile = dockerFiles.firstOrNull()?.virtualFile
+            val defaultContent = "This project use Docker to run in server, path: ${virtualFile?.path}"
+            val defaultContext = listOf(ChatContextItem(DockerContextProvider::class, defaultContent))
 
-        val fromCommands = dockerFiles.mapNotNull {
-            PsiTreeUtil.getChildrenOfType(it, DockerFileFromCommand::class.java)?.toList()
-        }.flatten()
+            val fromCommands = dockerFiles.mapNotNull {
+                PsiTreeUtil.getChildrenOfType(it, DockerFileFromCommand::class.java)?.toList()
+            }.flatten()
 
-        if (fromCommands.isEmpty()) return listOf(
-            ChatContextItem(DockerContextProvider::class, "This project use Docker to run in server")
-        )
+            if (fromCommands.isEmpty()) {
+                return defaultContext
+            }
 
-        val text = "This project use Docker to run in server. Here is related info:\n" +
-                fromCommands.joinToString("\n") { it.text }
+            val text = "This project use Docker to run in server. Here is related info:\n" +
+                    fromCommands.joinToString("\n") { it.text }
 
-        return listOf(ChatContextItem(DockerContextProvider::class, text))
+            return listOf(ChatContextItem(DockerContextProvider::class, text))
+        } catch (e: Exception) {
+            logger<DockerContextProvider>().warn("Failed to collect Docker context", e)
+            return listOf(ChatContextItem(DockerContextProvider::class, "This project use Docker."))
+        }
     }
 }