Merge pull request #44 from nathfavour/master

Enhance TUI with intelligent suggestions and split-view workspace

Author: nathfavour

cmd/vibeaura/chat.go

@@ -3,7 +3,10 @@ package main
 import (
 	"context"
 	"fmt"
+	"os"
+	"path/filepath"
 	"runtime"
+	"sort"
 	"strings"
 
 	"github.com/charmbracelet/bubbles/textarea"
@@ -15,14 +18,19 @@ import (
 )
 
 type model struct {
-	viewport    viewport.Model
-	messages    []string
-	textarea    textarea.Model
-	err         error
-	brain       *brain.Brain
-	width       int
-	height      int
-	initialized bool
+	viewport      viewport.Model
+	messages      []string
+	textarea      textarea.Model
+	err           error
+	brain         *brain.Brain
+	width         int
+	height        int
+	initialized   bool
+	showTree      bool
+	treeView      string
+	suggestions   []string
+	suggestionIdx int
+	triggerChar   string // '/' or '#'
 }
 
 var (
@@ -51,13 +59,36 @@ var (
 
 	helpStyle = lipgloss.NewStyle().
 			Foreground(lipgloss.Color("#626262"))
+
+	tagStyle = lipgloss.NewStyle().
+			Foreground(lipgloss.Color("#FFD700")).
+			Bold(true).
+			Italic(true)
+
+	suggestionStyle = lipgloss.NewStyle().
+			Foreground(lipgloss.Color("#7D56F4")).
+			Background(lipgloss.Color("#222222"))
+
+	selectedSuggestionStyle = lipgloss.NewStyle().
+				Foreground(lipgloss.Color("#FAFAFA")).
+				Background(lipgloss.Color("#7D56F4")).
+				Bold(true)
+
+	treeStyle = lipgloss.NewStyle().
+			Border(lipgloss.NormalBorder(), false, false, false, true).
+			BorderForeground(lipgloss.Color("#444444")).
+			PaddingLeft(2)
 )
 
 type chatState struct {
 	Messages []string `json:"messages"`
 	Input    string   `json:"input"`
 }
 
+var allCommands = []string{
+	"/help", "/status", "/cwd", "/version", "/clear", "/exit", "/show-tree",
+}
+
 func initialModel(b *brain.Brain) *model {
 	ta := textarea.New()
 	ta.Placeholder = "Send a message or type / for commands..."
@@ -128,9 +159,28 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.width = msg.Width
 		m.height = msg.Height
 		m.viewport.Width = msg.Width
-		m.textarea.SetWidth(msg.Width)
+		if m.showTree {
+			m.viewport.Width = msg.Width / 2
+		}
+		m.textarea.SetWidth(m.viewport.Width)
 		m.viewport.Height = msg.Height - m.textarea.Height() - 6
 	case tea.KeyMsg:
+		// Suggestion navigation
+		if len(m.suggestions) > 0 {
+			switch msg.String() {
+			case "tab", "down":
+				m.suggestionIdx = (m.suggestionIdx + 1) % len(m.suggestions)
+				return m, nil
+			case "shift+tab", "up":
+				m.suggestionIdx = (m.suggestionIdx - 1 + len(m.suggestions)) % len(m.suggestions)
+				return m, nil
+			case "enter":
+				// Accept suggestion
+				m.applySuggestion()
+				return m, nil
+			}
+		}
+
 		switch msg.String() {
 		case "ctrl+c", "esc":
 			m.saveState()
@@ -149,14 +199,18 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			// Add user message via a response update
 			m.messages = append(m.messages, userStyle.Render("You: ")+v)
 			m.textarea.Reset()
+			m.updateSuggestions("") // Clear suggestions
 			m.viewport.SetContent(strings.Join(m.messages, "\n\n"))
 			m.viewport.GotoBottom()
 
 			m.saveState()
 
 			// Process via brain
 			return m, m.processRequest(v)
-
+		default:
+			// After normal keypress, update suggestions
+			m.updateSuggestions(m.textarea.Value())
+			m.updateDynamicPreview()
 		}
 	case brain.Response:
 		if msg.Error != nil {
@@ -172,6 +226,160 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	return m, tea.Batch(tiCmd, vpCmd)
 }
 
+func (m *model) updateSuggestions(val string) {
+	m.suggestions = nil
+	m.suggestionIdx = 0
+	m.triggerChar = ""
+
+	if val == "" {
+		return
+	}
+
+	// Find the word being typed
+	words := strings.Fields(val)
+	if len(words) == 0 {
+		if strings.HasSuffix(val, "/") {
+			m.triggerChar = "/"
+			m.suggestions = append([]string{}, allCommands...)
+			sort.Strings(m.suggestions)
+		} else if strings.HasSuffix(val, "#") {
+			m.triggerChar = "#"
+			m.suggestions = m.getFileSuggestions("")
+		}
+		return
+	}
+
+	lastWord := words[len(words)-1]
+	if strings.HasPrefix(lastWord, "/") {
+		m.triggerChar = "/"
+		for _, cmd := range allCommands {
+			if strings.HasPrefix(cmd, lastWord) {
+				m.suggestions = append(m.suggestions, cmd)
+			}
+		}
+		sort.Strings(m.suggestions)
+	} else if strings.HasPrefix(lastWord, "#") {
+		m.triggerChar = "#"
+		m.suggestions = m.getFileSuggestions(lastWord[1:])
+	}
+}
+
+func (m *model) getFileSuggestions(prefix string) []string {
+	var suggestions []string
+	root, _ := os.Getwd()
+	
+	filepath.WalkDir(root, func(path string, d os.DirEntry, err error) error {
+		if err != nil || len(suggestions) > 30 {
+			return nil
+		}
+		
+		name := d.Name()
+		if d.IsDir() {
+			if name == ".git" || name == "node_modules" || name == "vendor" || name == "bin" || name == "dist" {
+				return filepath.SkipDir
+			}
+			if prefix != "" && !strings.HasPrefix(name, prefix) && !strings.HasPrefix(path, prefix) {
+				return nil
+			}
+		}
+
+		rel, _ := filepath.Rel(root, path)
+		if rel == "." {
+			return nil
+		}
+
+		if prefix == "" || strings.HasPrefix(rel, prefix) || strings.HasPrefix(name, prefix) {
+			suggestions = append(suggestions, rel)
+		}
+		
+		return nil
+	})
+
+	sort.Strings(suggestions)
+	return suggestions
+}
+
+func (m *model) applySuggestion() {
+	if len(m.suggestions) == 0 {
+		return
+	}
+	
+	val := m.textarea.Value()
+	words := strings.Fields(val)
+	if len(words) == 0 {
+		m.textarea.SetValue(m.suggestions[m.suggestionIdx] + " ")
+	} else {
+		words[len(words)-1] = m.triggerChar + m.suggestions[m.suggestionIdx]
+		m.textarea.SetValue(strings.Join(words, " ") + " ")
+	}
+	m.textarea.SetCursor(len(m.textarea.Value()))
+	m.suggestions = nil
+	m.updateDynamicPreview()
+}
+
+func (m *model) updateDynamicPreview() {
+	if !m.showTree {
+		return
+	}
+
+	val := m.textarea.Value()
+	tags := m.extractTags(val)
+	
+	if len(tags) > 0 {
+		// Show the last tag's content
+		lastTag := tags[len(tags)-1]
+		content, err := os.ReadFile(lastTag)
+		if err == nil {
+			m.treeView = string(content)
+			return
+		}
+		
+		// If it's a directory, show tree
+		info, err := os.Stat(lastTag)
+		if err == nil && info.IsDir() {
+			m.treeView = m.renderTree(lastTag)
+			return
+		}
+	}
+
+	// Default to workspace tree
+	m.treeView = m.renderTree(".")
+}
+
+func (m *model) extractTags(val string) []string {
+	var tags []string
+	words := strings.Fields(val)
+	for _, w := range words {
+		if strings.HasPrefix(w, "#") {
+			path := strings.TrimPrefix(w, "#")
+			if _, err := os.Stat(path); err == nil {
+				tags = append(tags, path)
+			}
+		}
+	}
+	return tags
+}
+
+func (m *model) renderTree(root string) string {
+	var sb strings.Builder
+	sb.WriteString(systemStyle.Render(" EXPLORER: " + root) + "\n\n")
+	
+	entries, _ := os.ReadDir(root)
+	for _, entry := range entries {
+		name := entry.Name()
+		if strings.HasPrefix(name, ".") && name != ".env" {
+			continue
+		}
+		
+		icon := "📄 "
+		if entry.IsDir() {
+			icon = "📁 "
+		}
+		sb.WriteString(icon + name + "\n")
+	}
+	return sb.String()
+}
+
 func (m *model) processRequest(content string) tea.Cmd {
 	return func() tea.Msg {
 		ctx := context.Background()
@@ -200,6 +408,16 @@ func (m *model) handleSlashCommand(cmd string) (tea.Model, tea.Cmd) {
 		m.messages = append(m.messages, systemStyle.Render(" CWD ") + " " + helpStyle.Render(snapshot.WorkingDir))
 	case "/version":
 		m.messages = append(m.messages, systemStyle.Render(" VERSION ") + "\n" + helpStyle.Render(fmt.Sprintf("App: %s\nCommit: %s\nCompiler: %s", Version, Commit, runtime.Version())))
+	case "/show-tree":
+		m.showTree = !m.showTree
+		if m.showTree {
+			m.viewport.Width = m.width / 2
+			m.treeView = m.renderTree(".")
+		} else {
+			m.viewport.Width = m.width
+		}
+		m.textarea.SetWidth(m.viewport.Width)
+		m.messages = append(m.messages, systemStyle.Render(" Sideview Toggled "))
 	case "/clear":
 		m.messages = []string{}
 		m.viewport.SetContent(systemStyle.Render(" Session Cleared "))
@@ -216,19 +434,66 @@ func (m *model) handleSlashCommand(cmd string) (tea.Model, tea.Cmd) {
 }
 
 func (m *model) View() string {
-	header := titleStyle.Render(" vibeauracle ") + " " + helpStyle.Render("v" + Version)
+	header := titleStyle.Render(" vibeauracle ") + " " + helpStyle.Render("v"+Version)
 	border := strings.Repeat("─", m.width)
 	if m.width > 20 {
 		border = strings.Repeat("─", m.width-1)
 	}
 
-	return fmt.Sprintf(
+	mainContent := m.viewport.View()
+	if m.showTree {
+		mainContent = lipgloss.JoinHorizontal(lipgloss.Top,
+			m.viewport.View(),
+			treeStyle.Render(m.treeView),
+		)
+	}
+
+	suggestionView := ""
+	if len(m.suggestions) > 0 {
+		var sbs []string
+		for i, s := range m.suggestions {
+			style := suggestionStyle
+			if i == m.suggestionIdx {
+				style = selectedSuggestionStyle
+			}
+			
+			// Format: name      path (truncated)
+			name := filepath.Base(s)
+			if m.triggerChar == "/" {
+				name = s
+			}
+			dir := filepath.Dir(s)
+			if dir == "." {
+				dir = ""
+			} else {
+				dir = "  " + dir
+				if len(dir) > 20 {
+					dir = dir[:17] + "..."
+				}
+			}
+			
+			sbs = append(sbs, style.Render(fmt.Sprintf(" %-15s %s ", name, dir)))
+		}
+		suggestionView = lipgloss.NewStyle().
+			Border(lipgloss.RoundedBorder()).
+			BorderForeground(lipgloss.Color("#7D56F4")).
+			Render(strings.Join(sbs, "\n"))
+	}
+
+	view := fmt.Sprintf(
 		"%s\n%s\n%s\n%s\n%s",
 		header,
 		lipgloss.NewStyle().Foreground(lipgloss.Color("#444444")).Render(border),
-		m.viewport.View(),
+		mainContent,
 		lipgloss.NewStyle().Foreground(lipgloss.Color("#444444")).Render(border),
 		m.textarea.View(),
-	) + "\n"
+	)
+
+	if suggestionView != "" {
+		// Overlay logic: simplified for TUI
+		view += "\n" + suggestionView
+	}
+
+	return view + "\n"
 }