feat(cli): add Bubble Tea TUI for watch, status, trace, init, and workspace commands (#143)

* feat(cli): add TUI command flows and stabilize watch/status behavior

* feat: introduce dynamic watch supervisor to manage multiple worktree sessions with TUI integration and event observation

* feat: improve TUI responsiveness for narrow terminals and refactor status file loading with dedicated tests.

* fix: Ensure dynamic watch supervisor gracefully shuts down on context cancellation and add a test for this behavior.

* fix: resolve TUI lint findings

* test: normalize dynamic watch roots across platforms

* refactor: introduce dedicated TUI components for ledger and progress using `charmbracelet/bubbles` and refactor `tui_watch` to integrate them.

* feat: Enable direct number input for selecting options in the TUI init wizard and add a corresponding test.

* feat: introduce progress observers for initial scan and embedding, updating watch TUI with real-time status.

* fix watch tui stats baseline and stabilize dynamic watch CI test

* style: gofmt dynamic watch retry test

* feat(tui): add provider, backend, and RPG config steps to init wizard

Extend the TUI init wizard with text-input based configuration steps
for embedding provider endpoints/models, storage backend connection
details, and RPG (Repository Planning Graph) enablement with optional
LLM configuration. Replaces hardcoded defaults with interactive inputs
using charmbracelet/bubbles textinput components.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

* fix(cli): address PR review items for TUI watch/init

* feat: Enhance watch daemon stopping logic to support multiple log directories and worktrees, and improve TUI statistics handling with per-project snapshots.

* fix: Prevent double-counting of incremental stats when re-basing snapshots in watch UI.

* feat: enhance trace TUI to display call graph edges without nodes and add file indexing statistics to watch TUI

* refactor: Remove `filesSkipped` metric from the TUI and clear project statistics when sessions are removed.

---------

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

Author: tinker495

cli/init.go

@@ -18,6 +18,7 @@ var (
 	initBackend        string
 	initNonInteractive bool
 	initInherit        bool
+	initUI             bool
 )
 
 const (
@@ -43,6 +44,7 @@ func init() {
 	initCmd.Flags().StringVarP(&initBackend, "backend", "b", "", "Storage backend (gob, postgres, or qdrant)")
 	initCmd.Flags().BoolVar(&initNonInteractive, "yes", false, "Use defaults without prompting")
 	initCmd.Flags().BoolVar(&initInherit, "inherit", false, "Inherit configuration from main worktree (for git worktrees)")
+	initCmd.Flags().BoolVar(&initUI, "ui", false, "Run interactive Bubble Tea UI wizard")
 }
 
 func runInit(cmd *cobra.Command, args []string) error {
@@ -60,19 +62,23 @@ func runInit(cmd *cobra.Command, args []string) error {
 
 	cfg := config.DefaultConfig()
 	skipPrompts := false
+	var detectedGitInfo *git.DetectInfo
+	var detectedMainCfg *config.Config
 
 	// Detect git worktree and offer config inheritance
 	gitInfo, gitErr := git.Detect(cwd)
 	if gitErr == nil && gitInfo.IsWorktree && config.Exists(gitInfo.MainWorktree) {
 		mainCfg, loadErr := config.Load(gitInfo.MainWorktree)
 		if loadErr == nil {
+			detectedGitInfo = gitInfo
+			detectedMainCfg = mainCfg
 			fmt.Printf("\nGit worktree detected.\n")
 			fmt.Printf("  Main worktree: %s\n", gitInfo.MainWorktree)
 			fmt.Printf("  Worktree ID:   %s\n", gitInfo.WorktreeID)
 			fmt.Printf("  Backend:       %s\n", mainCfg.Store.Backend)
 
 			shouldInherit := initInherit
-			if !shouldInherit && !initNonInteractive {
+			if shouldPromptInheritChoice(shouldInherit, initNonInteractive, initUI) {
 				reader := bufio.NewReader(os.Stdin)
 				fmt.Print("\nInherit configuration from main worktree? [Y/n]: ")
 				input, _ := reader.ReadString('\n')
@@ -96,6 +102,15 @@ func runInit(cmd *cobra.Command, args []string) error {
 		}
 	}
 
+	if initUI && !initNonInteractive {
+		uiCfg, uiErr := runInitWizardUI(cwd, cfg, detectedGitInfo, detectedMainCfg, initInherit)
+		if uiErr != nil {
+			return uiErr
+		}
+		cfg = uiCfg
+		skipPrompts = true
+	}
+
 	// Interactive mode
 	if !skipPrompts && !initNonInteractive {
 		reader := bufio.NewReader(os.Stdin)
@@ -334,3 +349,7 @@ func runInit(cmd *cobra.Command, args []string) error {
 
 	return nil
 }
+
+func shouldPromptInheritChoice(shouldInherit, nonInteractive, uiMode bool) bool {
+	return !shouldInherit && !nonInteractive && !uiMode
+}

cli/init_ui_prompt_test.go

@@ -0,0 +1,18 @@
+package cli
+
+import "testing"
+
+func TestShouldPromptInheritChoice(t *testing.T) {
+	if shouldPromptInheritChoice(false, false, true) {
+		t.Fatal("UI mode should skip legacy inherit prompt")
+	}
+	if shouldPromptInheritChoice(false, true, false) {
+		t.Fatal("non-interactive mode should skip inherit prompt")
+	}
+	if shouldPromptInheritChoice(true, false, false) {
+		t.Fatal("already-selected inherit should skip prompt")
+	}
+	if !shouldPromptInheritChoice(false, false, false) {
+		t.Fatal("interactive non-UI mode should prompt for inherit choice")
+	}
+}

cli/status.go

@@ -3,16 +3,22 @@ package cli
 import (
 	"context"
 	"fmt"
+	"os"
+	"path/filepath"
 	"sort"
 	"strings"
 
 	"github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 	"github.com/spf13/cobra"
 	"github.com/yoanbernabeu/grepai/config"
+	"github.com/yoanbernabeu/grepai/daemon"
+	"github.com/yoanbernabeu/grepai/git"
 	"github.com/yoanbernabeu/grepai/store"
 )
 
+var statusNoUI bool
+
 var statusCmd = &cobra.Command{
 	Use:   "status",
 	Short: "Display index status and browse indexed files",
@@ -45,9 +51,18 @@ type model struct {
 	selectedChunk int
 	width         int
 	height        int
+	watchRunning  bool
+	watchPID      int
+	watchLogDir   string
+	watchLogFile  string
+	worktreeID    string
 	err           error
 }
 
+func init() {
+	statusCmd.Flags().BoolVar(&statusNoUI, "no-ui", false, "Print plain text summary instead of interactive UI")
+}
+
 // Styles
 var (
 	titleStyle = lipgloss.NewStyle().
@@ -185,6 +200,19 @@ func (m model) viewStats() string {
 	sb.WriteString(normalStyle.Render("Provider:         "))
 	sb.WriteString(fmt.Sprintf("%s (%s)\n", m.cfg.Embedder.Provider, m.cfg.Embedder.Model))
 
+	sb.WriteString(normalStyle.Render("Watcher status:   "))
+	if m.watchRunning {
+		sb.WriteString(fmt.Sprintf("running (PID %d)\n", m.watchPID))
+	} else {
+		sb.WriteString("not running\n")
+	}
+	sb.WriteString(normalStyle.Render("Watcher logs:     "))
+	if m.watchLogFile == "" {
+		sb.WriteString("N/A\n")
+	} else {
+		sb.WriteString(fmt.Sprintf("%s\n", m.watchLogFile))
+	}
+
 	sb.WriteString("\n")
 	sb.WriteString(helpStyle.Render("[Enter] Browse files  [q] Quit"))
 
@@ -349,24 +377,31 @@ func runStatus(cmd *cobra.Command, args []string) error {
 		return fmt.Errorf("failed to get stats: %w", err)
 	}
 
-	// Get files
-	files, err := st.ListFilesWithStats(ctx)
+	watchStatus := resolveWatcherRuntimeStatus(projectRoot)
+	useUI := shouldUseStatusUI(isInteractiveTerminal(), statusNoUI)
+
+	if !useUI {
+		fmt.Print(renderStatusSummary(cfg, stats, watchStatus))
+		return nil
+	}
+
+	files, err := loadStatusFiles(ctx, useUI, st.ListFilesWithStats)
 	if err != nil {
 		return fmt.Errorf("failed to list files: %w", err)
 	}
 
-	// Sort files by path
-	sort.Slice(files, func(i, j int) bool {
-		return files[i].Path < files[j].Path
-	})
-
 	// Create model
 	m := model{
-		st:    st,
-		cfg:   cfg,
-		state: viewStats,
-		stats: stats,
-		files: files,
+		st:           st,
+		cfg:          cfg,
+		state:        viewStats,
+		stats:        stats,
+		files:        files,
+		watchRunning: watchStatus.running,
+		watchPID:     watchStatus.pid,
+		watchLogDir:  watchStatus.logDir,
+		watchLogFile: watchStatus.logFile,
+		worktreeID:   watchStatus.worktreeID,
 	}
 
 	// Run TUI
@@ -397,3 +432,124 @@ func truncatePath(path string, maxLen int) string {
 	}
 	return "..." + path[len(path)-maxLen+3:]
 }
+
+type watcherRuntimeStatus struct {
+	running    bool
+	pid        int
+	logDir     string
+	logFile    string
+	worktreeID string
+}
+
+func resolveWatcherRuntimeStatus(projectRoot string) watcherRuntimeStatus {
+	status := watcherRuntimeStatus{}
+
+	logDirs, err := resolveWatcherCandidateLogDirs(projectRoot)
+	if err != nil {
+		return status
+	}
+	if len(logDirs) == 0 {
+		return status
+	}
+
+	cwd, err := os.Getwd()
+	var worktreeID string
+	if err == nil {
+		gitInfo, gitErr := git.Detect(cwd)
+		if gitErr == nil && gitInfo.WorktreeID != "" {
+			worktreeID = gitInfo.WorktreeID
+		}
+	}
+
+	for idx, logDir := range logDirs {
+		status.logDir = logDir
+		status.worktreeID = worktreeID
+		if worktreeID != "" {
+			pid, _ := daemon.GetRunningWorktreePID(logDir, worktreeID)
+			logFile := daemon.GetWorktreeLogFile(logDir, worktreeID)
+			if pid == 0 {
+				legacyPID, _ := daemon.GetRunningPID(logDir)
+				if legacyPID > 0 {
+					pid = legacyPID
+					logFile = filepath.Join(logDir, "grepai-watch.log")
+				}
+			}
+			status.pid = pid
+			status.running = pid > 0
+			status.logFile = logFile
+		} else {
+			pid, _ := daemon.GetRunningPID(logDir)
+			status.pid = pid
+			status.running = pid > 0
+			status.logFile = filepath.Join(logDir, "grepai-watch.log")
+		}
+		if status.running || idx == len(logDirs)-1 {
+			return status
+		}
+	}
+
+	return status
+}
+
+func resolveWatcherCandidateLogDirs(projectRoot string) ([]string, error) {
+	defaultLogDir, err := daemon.GetDefaultLogDir()
+	if err != nil {
+		return nil, err
+	}
+
+	logDirs := make([]string, 0, 2)
+	if projectRoot != "" {
+		hintedLogDir, readErr := readWatchLogDirHint(projectRoot)
+		if readErr == nil && hintedLogDir != "" {
+			hintedLogDir = filepath.Clean(hintedLogDir)
+			if hintedLogDir != filepath.Clean(defaultLogDir) {
+				logDirs = append(logDirs, hintedLogDir)
+			}
+		}
+	}
+
+	logDirs = append(logDirs, defaultLogDir)
+	return logDirs, nil
+}
+
+func renderStatusSummary(cfg *config.Config, stats *store.IndexStats, watch watcherRuntimeStatus) string {
+	var sb strings.Builder
+	sb.WriteString("grepai index status\n")
+	sb.WriteString(fmt.Sprintf("Files indexed: %d\n", stats.TotalFiles))
+	sb.WriteString(fmt.Sprintf("Total chunks: %d\n", stats.TotalChunks))
+	sb.WriteString(fmt.Sprintf("Index size: %s\n", formatBytes(stats.IndexSize)))
+	if stats.LastUpdated.IsZero() {
+		sb.WriteString("Last updated: Never\n")
+	} else {
+		sb.WriteString(fmt.Sprintf("Last updated: %s\n", stats.LastUpdated.Format("2006-01-02 15:04:05")))
+	}
+	sb.WriteString(fmt.Sprintf("Provider: %s (%s)\n", cfg.Embedder.Provider, cfg.Embedder.Model))
+	if watch.running {
+		sb.WriteString(fmt.Sprintf("Watcher: running (PID %d)\n", watch.pid))
+	} else {
+		sb.WriteString("Watcher: not running\n")
+	}
+	if watch.logFile != "" {
+		sb.WriteString(fmt.Sprintf("Watcher log: %s\n", watch.logFile))
+	}
+	return sb.String()
+}
+
+func loadStatusFiles(
+	ctx context.Context,
+	useUI bool,
+	listFn func(context.Context) ([]store.FileStats, error),
+) ([]store.FileStats, error) {
+	if !useUI {
+		return nil, nil
+	}
+
+	files, err := listFn(ctx)
+	if err != nil {
+		return nil, err
+	}
+	sort.Slice(files, func(i, j int) bool {
+		return files[i].Path < files[j].Path
+	})
+	return files, nil
+}

cli/status_files_test.go

@@ -0,0 +1,45 @@
+package cli
+
+import (
+	"context"
+	"testing"
+
+	"github.com/yoanbernabeu/grepai/store"
+)
+
+func TestLoadStatusFiles_SkipsWhenUIOff(t *testing.T) {
+	called := 0
+
+	files, err := loadStatusFiles(context.Background(), false, func(context.Context) ([]store.FileStats, error) {
+		called++
+		return []store.FileStats{{Path: "b"}, {Path: "a"}}, nil
+	})
+	if err != nil {
+		t.Fatalf("loadStatusFiles() error: %v", err)
+	}
+	if called != 0 {
+		t.Fatalf("list function called %d times, want 0", called)
+	}
+	if len(files) != 0 {
+		t.Fatalf("files length = %d, want 0", len(files))
+	}
+}
+
+func TestLoadStatusFiles_LoadsAndSortsWhenUIOn(t *testing.T) {
+	files, err := loadStatusFiles(context.Background(), true, func(context.Context) ([]store.FileStats, error) {
+		return []store.FileStats{
+			{Path: "z-last.go"},
+			{Path: "a-first.go"},
+			{Path: "m-mid.go"},
+		}, nil
+	})
+	if err != nil {
+		t.Fatalf("loadStatusFiles() error: %v", err)
+	}
+	if len(files) != 3 {
+		t.Fatalf("files length = %d, want 3", len(files))
+	}
+	if files[0].Path != "a-first.go" || files[1].Path != "m-mid.go" || files[2].Path != "z-last.go" {
+		t.Fatalf("files not sorted by path: %+v", files)
+	}
+}