feat(terminal): add shell tool detection and context-aware command suggestions #135

Author: phodal

exts/ext-terminal/src/222/main/kotlin/cc/unitmesh/terminal/TerminalUtil.kt

@@ -8,6 +8,7 @@ import org.jetbrains.plugins.terminal.TerminalView
 
 object TerminalUtil {
     fun getCurrentTerminalWidget(project: Project): JBTerminalWidget? {
+        // TODO: test with `e.dataContext.getData(JBTerminalWidget.TERMINAL_DATA_KEY) ?: return`
         val toolWindow = ToolWindowManager.getInstance(project)
             .getToolWindow(TerminalToolWindowFactory.TOOL_WINDOW_ID) ?: return null
         val content = toolWindow.contentManager.selectedContent ?: return null

exts/ext-terminal/src/main/kotlin/cc/unitmesh/terminal/ShellCommandSuggestAction.kt

@@ -13,6 +13,7 @@ import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.AnActionHolder
 import com.intellij.openapi.actionSystem.PlatformCoreDataKeys
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.project.guessProjectDir
 import com.intellij.openapi.ui.popup.Balloon
 import com.intellij.openapi.ui.popup.JBPopupFactory
 import com.intellij.openapi.ui.popup.JBPopupListener
@@ -27,6 +28,7 @@ import com.intellij.util.ui.SwingHelper
 import com.intellij.util.ui.UIUtil
 import kotlinx.coroutines.flow.*
 import kotlinx.coroutines.launch
+import org.jetbrains.plugins.terminal.TerminalProjectOptionsProvider
 import java.awt.Component
 import java.awt.Font
 import java.awt.Point
@@ -42,23 +44,24 @@ private const val ERROR_VALUE = "error"
 class ShellCommandSuggestAction : AnAction() {
     override fun actionPerformed(e: AnActionEvent) {
         val project = e.project ?: return
-
-        val popupPoint = getPreferredPopupPoint(e)
-
         val contextComponent = e.getData(PlatformCoreDataKeys.CONTEXT_COMPONENT) ?: return
 
-        showContentRenamePopup(contextComponent, popupPoint) { data ->
+        showContentRenamePopup(contextComponent, getPreferredPopupPoint(e)) { data ->
             val widget = TerminalUtil.getCurrentTerminalWidget(project) ?: return@showContentRenamePopup
             suggestCommand(widget, data, project)
         }
     }
 
-    data class ShellSuggestions(val question: String)
-
     private fun suggestCommand(widget: JBTerminalWidget, data: String, project: Project) {
         val templateRender = TemplateRender(GENIUS_PRACTISES)
         val template = templateRender.getTemplate("shell-suggest.vm")
-        templateRender.context = ShellSuggestions(data)
+
+        val options = TerminalProjectOptionsProvider.getInstance(project)
+
+        templateRender.context = ShellSuggestContext(
+            data, options.shellPath, options.startingDirectory
+                ?: project.guessProjectDir()?.path ?: System.getProperty("user.home")
+        )
         val promptText = templateRender.renderTemplate(template)
 
         val llm = LlmFactory.instance.create(project)
@@ -156,3 +159,4 @@ class ShellCommandSuggestAction : AnAction() {
     }
 }
 
+data class ShellSuggestContext(val question: String, val shellPath: String, val cwd: String)

src/main/resources/genius/en/practises/shell-suggest.vm

@@ -2,6 +2,8 @@ Return only the command to be executed as a raw string, no string delimiters
 wrapping it, no yapping, no markdown, no fenced code blocks, what you return
 will be passed to subprocess.check_output() directly.
 
+User current directory is: ${context.cwd}, user use is: ${context.shellPath}, according the tool to create the command.
+
 For example, if the user asks: undo last git commit
 
 You return only line command: git reset --soft HEAD~1

src/main/resources/genius/zh/practises/shell-suggest.vm

@@ -2,6 +2,8 @@ Return only the command to be executed as a raw string, no string delimiters
 wrapping it, no yapping, no markdown, no fenced code blocks, what you return
 will be passed to subprocess.check_output() directly.
 
+User current directory is: ${context.cwd}, user use is: ${context.shellPath}, according the tool to create the command.
+
 For example, if the user asks: undo last git commit
 
 You return only line command: git reset --soft HEAD~1