ability to Ctrl+k to kills running test loop as example

Author: gcclinux

internal/ui/aichat.go

@@ -83,6 +83,8 @@ type AIChatPane struct {
 	codeBlockInfos   []dirtracker.CodeBlockInfo // Code blocks with language tags
 	blockDirMappings []string                   // Effective dir per code block index
 	workspaceRoot    string                     // From AppConfig.WorkspaceDir
+	runningCmd       *exec.Cmd                  // Currently running command process (for kill support)
+	processKilled    bool                       // Whether the process was killed by user (Ctrl+K)
 }
 
 // AIResponseMsg is sent when AI response chunk is received.
@@ -134,6 +136,9 @@ type AIAvailabilityMsg struct {
 	Available bool
 }
 
+// ClearStatusMsg is sent to clear the status message in the app.
+type ClearStatusMsg struct{}
+
 // LanguageCheckMsg is sent when a language runtime check is needed.
 type LanguageCheckMsg struct {
 	FileType     string // The file type being checked (e.g., "go", "python")
@@ -582,9 +587,12 @@ func (a *AIChatPane) Update(msg tea.Msg) tea.Cmd {
 	case AINotificationMsg:
 		a.DisplayNotification(msg.Content)
 	case TerminalOutputMsg:
-		a.terminalOutput = append(a.terminalOutput, msg.Line)
-		if a.terminalMode {
-			a.viewModeScroll = len(a.terminalOutput)
+		// Ignore output if process was killed by user
+		if !a.processKilled {
+			a.terminalOutput = append(a.terminalOutput, msg.Line)
+			if a.terminalMode {
+				a.viewModeScroll = len(a.terminalOutput)
+			}
 		}
 		return func() tea.Msg {
 			return <-msg.Output
@@ -593,12 +601,20 @@ func (a *AIChatPane) Update(msg tea.Msg) tea.Cmd {
 		a.cmdRunning = false
 		a.stdinWriter = nil
 		a.terminalInput = ""
-		a.terminalOutput = append(a.terminalOutput, "")
-		if msg.Err != nil {
-			a.terminalOutput = append(a.terminalOutput, fmt.Sprintf("[Process failed: %v]", msg.Err))
-		} else {
-			a.terminalOutput = append(a.terminalOutput, fmt.Sprintf("[Process exited with code %d]", msg.ExitCode))
+
+		// Only show exit message if process wasn't killed by user
+		if !a.processKilled {
+			a.terminalOutput = append(a.terminalOutput, "")
+			if msg.Err != nil {
+				a.terminalOutput = append(a.terminalOutput, fmt.Sprintf("[Process failed: %v]", msg.Err))
+			} else {
+				a.terminalOutput = append(a.terminalOutput, fmt.Sprintf("[Process exited with code %d]", msg.ExitCode))
+			}
 		}
+
+		// Reset the killed flag for next execution
+		a.processKilled = false
+
 		if a.terminalMode {
 			a.viewModeScroll = len(a.terminalOutput)
 		}
@@ -618,6 +634,9 @@ func (a *AIChatPane) Update(msg tea.Msg) tea.Cmd {
 func (a *AIChatPane) executeCommand(script string, cwd string) tea.Cmd {
 	outChan := make(chan tea.Msg)
 
+	// Reset the killed flag for new execution
+	a.processKilled = false
+
 	// Determine effective working directory
 	effectiveDir := cwd
 	if effectiveDir == "" {
@@ -669,13 +688,17 @@ func (a *AIChatPane) executeCommand(script string, cwd string) tea.Cmd {
 			cmd.Dir = effectiveDir
 		}
 
+		// Store the running command so it can be killed with Ctrl+K
+		a.runningCmd = cmd
+
 		stdin, _ := cmd.StdinPipe()
 		stdout, _ := cmd.StdoutPipe()
 		stderr, _ := cmd.StderrPipe()
 
 		a.stdinWriter = stdin
 
 		if err := cmd.Start(); err != nil {
+			a.runningCmd = nil
 			outChan <- TerminalDoneMsg{Err: err}
 			return
 		}
@@ -718,6 +741,7 @@ func (a *AIChatPane) executeCommand(script string, cwd string) tea.Cmd {
 			}
 		}
 		a.stdinWriter = nil
+		a.runningCmd = nil
 		outChan <- TerminalDoneMsg{ExitCode: exitCode, Err: err}
 	}()
 
@@ -858,6 +882,38 @@ func (a *AIChatPane) handleKeyPress(msg tea.KeyMsg) tea.Cmd {
 		// When a command is running in terminal mode, forward input to the process
 		if a.terminalMode && a.cmdRunning && a.stdinWriter != nil {
 			switch keyStr {
+			case "ctrl+k":
+				// Kill the running process
+				if a.runningCmd != nil && a.runningCmd.Process != nil {
+					// Set flag to ignore further output
+					a.processKilled = true
+					// Close stdin first
+					if a.stdinWriter != nil {
+						a.stdinWriter.Close()
+					}
+
+					// Kill the process tree (important on Windows to kill child processes)
+					pid := a.runningCmd.Process.Pid
+					if runtime.GOOS == "windows" {
+						// On Windows, use taskkill to kill the entire process tree
+						killCmd := exec.Command("taskkill", "/F", "/T", "/PID", fmt.Sprintf("%d", pid))
+						killCmd.Run()
+					} else {
+						// On Unix, just kill the process
+						a.runningCmd.Process.Kill()
+					}
+
+					// Mark as not running
+					a.cmdRunning = false
+					a.terminalOutput = append(a.terminalOutput, "", "[Process killed by user (Ctrl+K)]")
+					a.viewModeScroll = len(a.terminalOutput)
+
+					// Return a command to clear the status message in the app
+					return func() tea.Msg {
+						return ClearStatusMsg{}
+					}
+				}
+				return nil
 			case "esc":
 				// Esc always exits — close stdin and let the process finish
 				a.stdinWriter.Close()
@@ -1626,7 +1682,7 @@ func (a *AIChatPane) renderViewMode() string {
 				MarginBottom(1).
 				Padding(0, 1).
 				Width(a.width - 4).
-				Render(" ⚙  [Enter] Send Input  |  [Esc] Close Stdin  |  [↑↓/PgUp/PgDn] Scroll ")
+				Render(" ⚙  [Enter] Send Input  |  [Ctrl+K] Kill Process  |  [Esc] Close Stdin  |  [↑↓/PgUp/PgDn] Scroll ")
 		} else {
 			instructions = lipgloss.NewStyle().
 				Foreground(lipgloss.Color("15")).
@@ -1918,7 +1974,7 @@ func (a *AIChatPane) GetLastAssistantResponse() string {
 
 // RunScript enters terminal mode and executes the given command, streaming
 // output into the AI chat pane. Returns a tea.Cmd to start the execution.
-func (a *AIChatPane) RunScript(command string, label string) tea.Cmd {
+func (a *AIChatPane) RunScript(command string, label string, workingDir string) tea.Cmd {
 	if a.cmdRunning {
 		return nil
 	}
@@ -1937,7 +1993,7 @@ func (a *AIChatPane) RunScript(command string, label string) tea.Cmd {
 	}
 	a.viewModeScroll = 0
 
-	return a.executeCommand(command, "")
+	return a.executeCommand(command, workingDir)
 }
 
 // EnterConfigMode enters the configuration editor mode.

internal/ui/app.go

@@ -474,7 +474,11 @@ func (a *App) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 		return a, nil
 
-	case TerminalOutputMsg, TerminalDoneMsg, AIResponseMsg, AINotificationMsg, AIAvailabilityMsg:
+	case TerminalOutputMsg, TerminalDoneMsg, AIResponseMsg, AINotificationMsg, AIAvailabilityMsg, ClearStatusMsg:
+		// Handle ClearStatusMsg
+		if _, ok := msg.(ClearStatusMsg); ok {
+			a.statusMessage = ""
+		}
 		cmd := a.aiPane.Update(msg)
 		cmds = append(cmds, cmd)
 		return a, tea.Batch(cmds...)
@@ -1154,9 +1158,10 @@ func (a *App) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			a.aiPane.focused = true
 
 			fileName := filepath.Base(filePath)
+			fileDir := filepath.Dir(filePath)
 			a.statusMessage = "Running " + fileName + "..."
 
-			cmd := a.aiPane.RunScript(runCmd, fileName)
+			cmd := a.aiPane.RunScript(runCmd, fileName, fileDir)
 			cmds = append(cmds, cmd)
 			return a, tea.Batch(cmds...)
 
@@ -1751,6 +1756,8 @@ func (a *App) handleAIMessage(message string) tea.Cmd {
 		helpText += "  Ctrl+N    New file\n"
 		helpText += "  Ctrl+S    Save file\n"
 		helpText += "  Ctrl+X    Close file\n"
+		helpText += "  Ctrl+R    Run current script\n"
+		helpText += "  Ctrl+K    Kill running process (in terminal mode)\n"
 		helpText += "  Ctrl+B    Backup Picker (Restore previous versions)\n"
 		helpText += "  Ctrl+Q    Quit\n\n"
 		helpText += "AI\n"

internal/ui/help.go

@@ -38,6 +38,7 @@ func (a *App) renderHelpDialog() string {
 	helpText += keyStyle.Render("  Ctrl+S") + descStyle.Render("    Save file") + "\n"
 	helpText += keyStyle.Render("  Ctrl+X") + descStyle.Render("    Close file") + "\n"
 	helpText += keyStyle.Render("  Ctrl+R") + descStyle.Render("    Run current script") + "\n"
+	helpText += keyStyle.Render("  Ctrl+K") + descStyle.Render("    Kill running process (in terminal mode)") + "\n"
 	helpText += keyStyle.Render("  Ctrl+B") + descStyle.Render("    Backup Picker (Restore previous versions)") + "\n"
 	helpText += keyStyle.Render("  Ctrl+Q") + descStyle.Render("    Quit") + "\n"
 	helpText += "\n"

main.go

@@ -54,6 +54,7 @@ func main() {
 		fmt.Println("  Tab         - Switch between editor and AI pane")
 		fmt.Println("  Ctrl+S      - Save current file")
 		fmt.Println("  Ctrl+R      - Execute current script")
+		fmt.Println("  Ctrl+K      - Kill running process (in terminal mode)")
 		fmt.Println("  Ctrl+Enter  - Send message to AI")
 		fmt.Println("  Ctrl+C      - Copy selected line or AI block")
 		fmt.Println("  Ctrl+Q      - Quit application")