feat(devins-lang): add SHELL command and related functionality #101

Add a new SHELL command to the DevIns language, along with
supporting functionality such as command execution and
completion provider for code insight.

Author: phodal

build.gradle.kts

@@ -591,7 +591,7 @@ project(":exts:devins-lang") {
 
     intellij {
         version.set(ideaVersion)
-        plugins.set((ideaPlugins + "org.intellij.plugins.markdown"))
+        plugins.set((ideaPlugins + "org.intellij.plugins.markdown" + "com.jetbrains.sh"))
     }
 
     dependencies {

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

@@ -183,6 +183,11 @@ class DevInsCompiler(
                 result.isLocalCommand = true
                 FileFuncInsCommand(myProject, prop)
             }
+
+            BuiltinCommand.SHELL -> {
+                result.isLocalCommand = true
+                ShellInsCommand(myProject, prop)
+            }
         }
 
         val execResult = command.execute()

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

@@ -0,0 +1,71 @@
+package cc.unitmesh.devti.language.compiler.exec
+
+import cc.unitmesh.devti.language.utils.lookupFile
+import com.intellij.execution.DefaultExecutionResult
+import com.intellij.execution.ExecutionException
+import com.intellij.execution.configurations.GeneralCommandLine
+import com.intellij.execution.configurations.PtyCommandLine
+import com.intellij.execution.process.KillableProcessHandler
+import com.intellij.execution.process.ProcessHandler
+import com.intellij.execution.process.ProcessTerminatedListener
+import com.intellij.execution.ui.ConsoleView
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.project.Project
+import com.intellij.sh.run.ShConfigurationType
+import com.intellij.sh.run.ShRunner
+import com.intellij.terminal.TerminalExecutionConsole
+import com.intellij.util.io.BaseOutputReader
+
+class ShellInsCommand(val myProject: Project, val prop: String) : InsCommand {
+    override fun execute(): String? {
+        val virtualFile = myProject.lookupFile(prop.trim()) ?: return "<DevInsError>: File not found: $prop"
+
+        val workingDirectory = virtualFile.parent.path
+        val shRunner = ApplicationManager.getApplication().getService(
+            ShRunner::class.java
+        )
+        if (shRunner != null && shRunner.isAvailable(myProject)) {
+            shRunner.run(myProject, virtualFile.path, workingDirectory, "RunDevInsShell", true)
+        }
+
+        // TODO: after run done
+//        runInTerminal(virtualFile.path, workingDirectory, myProject)
+
+        return ""
+    }
+
+    @Throws(ExecutionException::class)
+    private fun runInTerminal(command: String, workingDirectory: String, project: Project): DefaultExecutionResult {
+        val commandLine = createCommandLineForScript(project, workingDirectory, command)
+        val processHandler = createProcessHandler(commandLine)
+        ProcessTerminatedListener.attach(processHandler)
+        val console: ConsoleView = TerminalExecutionConsole(project, processHandler)
+        console.attachToProcess(processHandler)
+        return DefaultExecutionResult(console, processHandler)
+    }
+
+    private fun createCommandLineForScript(
+        project: Project,
+        workingDirectory: String,
+        command: String
+    ): GeneralCommandLine {
+        val commandLine = PtyCommandLine()
+        commandLine.withConsoleMode(false)
+        commandLine.withInitialColumns(120)
+        commandLine.withParentEnvironmentType(GeneralCommandLine.ParentEnvironmentType.CONSOLE)
+        commandLine.setWorkDirectory(workingDirectory)
+        commandLine.withExePath(ShConfigurationType.getDefaultShell(project))
+        commandLine.withParameters("-c")
+        commandLine.withParameters(command)
+        return commandLine
+    }
+
+    @Throws(ExecutionException::class)
+    private fun createProcessHandler(commandLine: GeneralCommandLine): ProcessHandler {
+        return object : KillableProcessHandler(commandLine) {
+            override fun readerOptions(): BaseOutputReader.Options {
+                return BaseOutputReader.Options.forTerminalPtyProcess()
+            }
+        }
+    }
+}

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/provider/BuiltinCommandCompletion.kt

@@ -5,7 +5,6 @@ import com.intellij.codeInsight.AutoPopupController
 import com.intellij.codeInsight.completion.*
 import com.intellij.codeInsight.lookup.LookupElementBuilder
 import com.intellij.util.ProcessingContext
-import org.jetbrains.kotlin.idea.completion.weighers.VariableOrParameterNameWithTypeWeigher.nameWithTypePriority
 
 class BuiltinCommandCompletion : CompletionProvider<CompletionParameters>() {
     override fun addCompletions(

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

@@ -23,8 +23,9 @@ enum class BuiltinCommand(
     WRITE("write", "Write content to a file, /write:path/to/file:L1-L2", AllIcons.Actions.Edit, true, true),
     PATCH("patch", "Apply patch to a file, /patch:path/to/file", AllIcons.Vcs.Patch_file, false),
     RUN("run", "Run the content of a file", AllIcons.Actions.Execute, true, true),
+    SHELL("shell", "Run shell command", AllIcons.Actions.Execute, true, true),
+    COMMIT("commit", "Commit the content of a file", AllIcons.Vcs.CommitNode, false),
     FILE_FUNC("file-func", "Read the name of a file", AllIcons.Actions.GroupByFile, true, true),
-    COMMIT("commit", "Commit the content of a file", AllIcons.Vcs.CommitNode, false)
     ;
 
     companion object {

exts/devins-lang/src/main/resources/cc.unitmesh.devti.language.xml

@@ -1,6 +1,7 @@
 <idea-plugin package="cc.unitmesh.devti.language">
     <dependencies>
         <plugin id="org.intellij.plugins.markdown"/>
+        <plugin id="com.jetbrains.sh"/>
     </dependencies>
 
     <extensions defaultExtensionNs="com.intellij">