feat(docker): implement DockerContextProvider and RunDockerfileService #306

- Added DockerContextProvider and RunDockerfileService implementations.
- Moved extensions to core/src/233/main/resources/META-INF/docker.xml.
- Optimized DockerContextProvider to use runReadAction for command extraction.

Author: phodal

core/src/223/main/resources/META-INF/docker.xml

@@ -1,3 +1,8 @@
 <!-- Defines IDEA IDE-specific contributions and implementations. -->
 <idea-plugin>
+    <extensions defaultExtensionNs="cc.unitmesh">
+        <runService implementation="cc.unitmesh.container.RunDockerfileService"/>
+
+        <chatContextProvider implementation="cc.unitmesh.container.provider.DockerContextProvider"/>
+    </extensions>
 </idea-plugin>

core/src/233/main/resources/META-INF/docker.xml

@@ -1,3 +1,8 @@
 <!-- Defines IDEA IDE-specific contributions and implementations. -->
 <idea-plugin>
+    <extensions defaultExtensionNs="cc.unitmesh">
+        <runService implementation="cc.unitmesh.container.RunDockerfileService"/>
+
+        <chatContextProvider implementation="cc.unitmesh.container.provider.DockerContextProvider"/>
+    </extensions>
 </idea-plugin>

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

@@ -12,7 +12,10 @@ class RunDockerfileService : RunService {
 
     fun isApplicable(element: PsiElement) = element.containingFile?.language == DockerLanguage.INSTANCE
 
-    override fun runConfigurationClass(project: Project): Class<out RunProfile>? {
-        return null
+    override fun runConfigurationClass(project: Project): Class<out RunProfile>? = null
+
+    override fun runFile(project: Project, virtualFile: VirtualFile, psiElement: PsiElement?, isFromToolAction: Boolean)
+            : String? {
+        return super.runFile(project, virtualFile, psiElement, isFromToolAction)
     }
 }

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

@@ -3,18 +3,14 @@ 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 cc.unitmesh.devti.sketch.ui.patch.readText
 import com.intellij.docker.DockerFileSearch
 import com.intellij.docker.dockerFile.parser.psi.DockerFileFromCommand
 import com.intellij.openapi.application.runReadAction
-import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiManager
 import com.intellij.psi.impl.source.PsiFileImpl
 
 class DockerContextProvider : ChatContextProvider {
-    private val fromRegex = Regex("FROM\\s+((?:--platform=[^\\s]+\\s+)?[^\\s]+)(?:\\s+AS\\s+([^\\s]+))?")
-
     override fun isApplicable(project: Project, creationContext: ChatCreationContext): Boolean =
         DockerFileSearch.getInstance().getDockerFiles(project).isNotEmpty()
 
@@ -36,26 +32,15 @@ class DockerContextProvider : ChatContextProvider {
         context = "This project use Docker, path: ${virtualFile.path}"
 
         var additionalCtx = ""
-        try {
-            val fromCommands = dockerFiles.map {
+        val fromCommands = runReadAction {
+            dockerFiles.map {
                 (it as PsiFileImpl).findChildrenByClass(DockerFileFromCommand::class.java).toList()
             }.flatten()
+        }
 
-            if (fromCommands.isEmpty()) return listOf(ChatContextItem(DockerContextProvider::class, context))
-            additionalCtx = fromCommands.joinToString("\n") {
-                runReadAction { it.text }
-            }
-        } catch (e: Exception) {
-            logger<DockerContextProvider>().warn("Failed to collect Docker context", e)
-            val fromMatch = fromRegex.find(virtualFile.readText())
-
-            if (fromMatch != null) {
-                additionalCtx = fromMatch.groupValues[1]
-            }
-
-            if (additionalCtx.isEmpty()) {
-                return listOf(ChatContextItem(DockerContextProvider::class, context))
-            }
+        if (fromCommands.isEmpty()) return listOf(ChatContextItem(DockerContextProvider::class, context))
+        additionalCtx = fromCommands.joinToString("\n") {
+            runReadAction { it.text }
         }
 
         val text = "This project use Docker to run in server. Here is related info:\n$additionalCtx"

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

@@ -5,9 +5,5 @@
         <plugin id="Docker"/>
     </dependencies>
 
-    <extensions defaultExtensionNs="cc.unitmesh">
-        <!--        <runService implementation="cc.unitmesh.container.RunDockerfileService"/>-->
-
-        <chatContextProvider implementation="cc.unitmesh.container.provider.DockerContextProvider"/>
-    </extensions>
+    <!--  Since limit of Docker, we move implementation to docker-contrib.xml  -->
 </idea-plugin>