Merge pull request #612 from Deepam02/feat/agent-commands-in-tui

feat: add agent commands support in TUI

Author: dgageot

pkg/app/app.go

@@ -3,6 +3,7 @@ package app
 import (
 	"context"
 	"os/exec"
+	"strings"
 	"time"
 
 	tea "github.com/charmbracelet/bubbletea/v2"
@@ -49,6 +50,47 @@ func (a *App) Title() string {
 	return a.title
 }
 
+// CurrentAgentCommands returns the commands for the active agent
+func (a *App) CurrentAgentCommands() map[string]string {
+	if a.runtime == nil {
+		return nil
+	}
+
+	agent := a.runtime.CurrentAgent()
+	if agent == nil {
+		return nil
+	}
+
+	return agent.Commands()
+}
+
+// ResolveCommand converts /command to its prompt text
+func (a *App) ResolveCommand(input string) string {
+	if !strings.HasPrefix(input, "/") {
+		return input
+	}
+
+	trimmed := strings.TrimSpace(input)
+	parts := strings.Fields(trimmed)
+	if len(parts) == 0 {
+		return input
+	}
+
+	cmdName := strings.TrimPrefix(parts[0], "/")
+	commands := a.CurrentAgentCommands()
+
+	if prompt, ok := commands[cmdName]; ok {
+		// If there are additional arguments, append them to the prompt
+		if len(parts) > 1 {
+			args := strings.Join(parts[1:], " ")
+			return prompt + " " + args
+		}
+		return prompt
+	}
+
+	return input
+}
+
 // Run one agent loop
 func (a *App) Run(ctx context.Context, cancel context.CancelFunc, message string) {
 	a.cancel = cancel

pkg/tui/components/editor/editor.go

@@ -28,13 +28,14 @@ type Editor interface {
 // editor implements Editor
 type editor struct {
 	textarea *textarea.Model
+	resolver func(string) string
 	width    int
 	height   int
 	working  bool
 }
 
 // New creates a new editor component
-func New() Editor {
+func New(resolver func(string) string) Editor {
 	ta := textarea.New()
 	ta.SetStyles(styles.InputStyle)
 	ta.Placeholder = "Type your message here..."
@@ -48,6 +49,7 @@ func New() Editor {
 
 	return &editor{
 		textarea: ta,
+		resolver: resolver,
 	}
 }
 
@@ -71,6 +73,10 @@ func (e *editor) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			value := e.textarea.Value()
 			if value != "" && !e.working {
 				e.textarea.Reset()
+				// Resolve command before sending
+				if e.resolver != nil {
+					value = e.resolver(value)
+				}
 				return e, core.CmdHandler(SendMsg{Content: value})
 			}
 			return e, nil

pkg/tui/page/chat/chat.go

@@ -39,6 +39,7 @@ type Page interface {
 	layout.Help
 	CompactSession() tea.Cmd
 	CopySessionToClipboard() tea.Cmd
+	ExecuteCommand(commandText string) tea.Cmd
 	Cleanup()
 }
 
@@ -95,7 +96,7 @@ func New(a *app.App) Page {
 		title:        a.Title(),
 		sidebar:      sidebar.New(),
 		messages:     messages.New(a),
-		editor:       editor.New(),
+		editor:       editor.New(a.ResolveCommand),
 		focusedPanel: PanelEditor,
 		app:          a,
 		keyMap:       defaultKeyMap(),
@@ -511,6 +512,13 @@ func (p *chatPage) CompactSession() tea.Cmd {
 	return p.messages.ScrollToBottom()
 }
 
+// ExecuteCommand sends a command text as a message (used by command palette)
+func (p *chatPage) ExecuteCommand(commandText string) tea.Cmd {
+	return func() tea.Msg {
+		return editor.SendMsg{Content: commandText}
+	}
+}
+
 func (p *chatPage) Cleanup() {
 	p.stopProgressBar()
 }

pkg/tui/tui.go

@@ -321,7 +321,7 @@ func toFullscreenView(content string) tea.View {
 
 // buildCommandCategories builds the list of command categories for the command palette
 func (a *appModel) buildCommandCategories() []dialog.CommandCategory {
-	return []dialog.CommandCategory{
+	categories := []dialog.CommandCategory{
 		{
 			Name: "Session",
 			Commands: []dialog.Command{
@@ -370,4 +370,38 @@ func (a *appModel) buildCommandCategories() []dialog.CommandCategory {
 			},
 		},
 	}
+
+	// Add agent commands if available
+	agentCommands := a.application.CurrentAgentCommands()
+	if len(agentCommands) > 0 {
+		commands := make([]dialog.Command, 0, len(agentCommands))
+		for name, prompt := range agentCommands {
+			cmdText := "/" + name
+
+			// Truncate long descriptions to fit on one line
+			description := prompt
+			if len(description) > 60 {
+				description = description[:57] + "..."
+			}
+
+			// Capture cmdText in closure properly
+			commandText := cmdText
+			commands = append(commands, dialog.Command{
+				ID:          "agent.command." + name,
+				Label:       commandText,
+				Description: description,
+				Category:    "Agent Commands",
+				Execute: func() tea.Cmd {
+					return a.chatPage.ExecuteCommand(commandText)
+				},
+			})
+		}
+
+		categories = append(categories, dialog.CommandCategory{
+			Name:     "Agent Commands",
+			Commands: commands,
+		})
+	}
+
+	return categories
 }