feat(container): add DockerContextProvider for chat context #306

Introduce DockerContextProvider to collect Docker-related context for chat interactions. Update plugin configuration to register the provider.

Author: phodal

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

@@ -0,0 +1,41 @@
+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.project.Project
+import com.intellij.psi.PsiManager
+import com.intellij.psi.util.PsiTreeUtil
+
+class DockerContextProvider : ChatContextProvider {
+    override fun isApplicable(project: Project, creationContext: ChatCreationContext): Boolean =
+        DockerFileSearch.getInstance().getDockerFiles(project).isNotEmpty()
+
+    override suspend fun collect(
+        project: Project,
+        creationContext: ChatCreationContext
+    ): List<ChatContextItem> {
+        val dockerFiles = DockerFileSearch.getInstance().getDockerFiles(project).mapNotNull {
+            PsiManager.getInstance(project).findFile(it) as? DockerPsiFile
+        }
+
+        if (dockerFiles.isEmpty()) return emptyList()
+
+        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")
+        )
+
+        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))
+    }
+}

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

@@ -6,6 +6,8 @@
     </dependencies>
 
     <extensions defaultExtensionNs="cc.unitmesh">
-<!--        <runService implementation="cc.unitmesh.container.RunDockerfileService"/>-->
+        <!--        <runService implementation="cc.unitmesh.container.RunDockerfileService"/>-->
+
+        <chatContextProvider implementation="cc.unitmesh.container.provider.DockerContextProvider"/>
     </extensions>
 </idea-plugin>