feat: TUI improvements

- Make --tui default for scan command (use --no-tui for text output)
- Rescan items after deletion instead of exiting (press q to quit)
- Remember cursor position when navigating tree (cursorStack)
- Update help text to show rescan/quit options

Author: thanhdevapp

cmd/root/scan.go

@@ -25,14 +25,13 @@ var scanCmd = &cobra.Command{
 	Short: "Scan for development artifacts",
 	Long: `Scan your system for development artifacts that can be cleaned.
 
-By default, scans all categories (iOS, Android, Node.js).
-Use flags to scan specific categories only.
+By default, opens interactive TUI for selection.
+Use --no-tui for simple text output.
 
 Examples:
-  dev-cleaner scan              # Scan all categories
-  dev-cleaner scan --ios        # Scan iOS/Xcode only
-  dev-cleaner scan --android    # Scan Android/Gradle only
-  dev-cleaner scan --node       # Scan Node.js only`,
+  dev-cleaner scan              # Scan + TUI (default)
+  dev-cleaner scan --no-tui     # Scan + text output
+  dev-cleaner scan --ios        # Scan iOS/Xcode only`,
 	Run: runScan,
 }
 
@@ -43,7 +42,8 @@ func init() {
 	scanCmd.Flags().BoolVar(&scanAndroid, "android", false, "Scan Android/Gradle artifacts only")
 	scanCmd.Flags().BoolVar(&scanNode, "node", false, "Scan Node.js artifacts only")
 	scanCmd.Flags().BoolVar(&scanAll, "all", true, "Scan all categories (default)")
-	scanCmd.Flags().BoolVar(&scanTUI, "tui", false, "Launch interactive TUI for selection")
+	scanCmd.Flags().BoolVar(&scanTUI, "tui", true, "Launch interactive TUI (default)")
+	scanCmd.Flags().BoolP("no-tui", "T", false, "Disable TUI, show text output")
 }
 
 func runScan(cmd *cobra.Command, args []string) {
@@ -86,7 +86,13 @@ func runScan(cmd *cobra.Command, args []string) {
 	// Sort by size (largest first)
 	sortBySize(results)
 
-	// Launch TUI if --tui flag is set
+	// Check for --no-tui flag
+	noTUI, _ := cmd.Flags().GetBool("no-tui")
+	if noTUI {
+		scanTUI = false
+	}
+
+	// Launch TUI by default
 	if scanTUI {
 		if err := tui.Run(results, false); err != nil {
 			fmt.Fprintf(os.Stderr, "Error running TUI: %v\n", err)

internal/tui/tui.go

@@ -155,6 +155,7 @@ type Model struct {
 	treeMode     bool              // True when in tree view
 	currentNode  *types.TreeNode   // Current tree node
 	nodeStack    []*types.TreeNode // Breadcrumb trail
+	cursorStack  []int             // Cursor positions for each level
 	maxDepth     int               // Max depth limit
 	treeSelected map[string]bool   // Selected items in tree
 	scanning     bool              // True while scanning
@@ -217,7 +218,13 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		// Handle based on current state
 		switch m.state {
 		case StateDone:
-			return m, tea.Quit
+			// 'q' to quit, any other key to rescan and continue
+			if msg.String() == "q" || msg.String() == "ctrl+c" {
+				m.quitting = true
+				return m, tea.Quit
+			}
+			// Rescan and return to selection
+			return m, m.rescanItems()
 
 		case StateConfirming:
 			switch msg.String() {
@@ -346,6 +353,20 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		}
 		m.cursor = 0
 		return m, nil
+
+	case rescanItemsMsg:
+		if msg.err != nil {
+			m.err = msg.err
+			return m, nil
+		}
+		// Reset state and show new items
+		m.items = msg.items
+		m.selected = make(map[int]bool)
+		m.cursor = 0
+		m.state = StateSelecting
+		m.results = nil
+		m.err = nil
+		return m, nil
 	}
 
 	return m, nil
@@ -368,6 +389,45 @@ type scanNodeMsg struct {
 	err  error
 }
 
+// rescanItemsMsg is sent when items rescan completes
+type rescanItemsMsg struct {
+	items []types.ScanResult
+	err   error
+}
+
+// rescanItems rescans all items and returns to selection
+func (m Model) rescanItems() tea.Cmd {
+	return func() tea.Msg {
+		s, err := scanner.New()
+		if err != nil {
+			return rescanItemsMsg{err: err}
+		}
+
+		opts := types.ScanOptions{
+			MaxDepth:       3,
+			IncludeXcode:   true,
+			IncludeAndroid: true,
+			IncludeNode:    true,
+		}
+
+		results, err := s.ScanAll(opts)
+		if err != nil {
+			return rescanItemsMsg{err: err}
+		}
+
+		// Sort by size
+		for i := 0; i < len(results)-1; i++ {
+			for j := i + 1; j < len(results); j++ {
+				if results[j].Size > results[i].Size {
+					results[i], results[j] = results[j], results[i]
+				}
+			}
+		}
+
+		return rescanItemsMsg{items: results}
+	}
+}
+
 // enterTreeMode transitions from flat list to tree view
 func (m Model) enterTreeMode() tea.Cmd {
 	return func() tea.Msg {
@@ -409,9 +469,17 @@ func (m *Model) goBackInTree() {
 		return
 	}
 
+	// Pop node and cursor from stacks
 	m.currentNode = m.nodeStack[len(m.nodeStack)-1]
 	m.nodeStack = m.nodeStack[:len(m.nodeStack)-1]
-	m.cursor = 0
+
+	// Restore cursor position
+	if len(m.cursorStack) > 0 {
+		m.cursor = m.cursorStack[len(m.cursorStack)-1]
+		m.cursorStack = m.cursorStack[:len(m.cursorStack)-1]
+	} else {
+		m.cursor = 0
+	}
 }
 
 // drillDownInTree navigates into child node
@@ -436,9 +504,13 @@ func (m *Model) drillDownInTree() tea.Cmd {
 
 	if selectedNode.NeedsScanning() {
 		m.scanning = true
+		// Save cursor position before scanning
+		m.cursorStack = append(m.cursorStack, m.cursor)
 		return m.scanNode(selectedNode)
 	}
 
+	// Save cursor position before navigating
+	m.cursorStack = append(m.cursorStack, m.cursor)
 	m.nodeStack = append(m.nodeStack, m.currentNode)
 	m.currentNode = selectedNode
 	m.cursor = 0
@@ -759,7 +831,7 @@ func (m Model) renderSelection(b *strings.Builder) string {
 func (m Model) renderResults(b *strings.Builder) string {
 	if m.err != nil {
 		b.WriteString(errorStyle.Render(fmt.Sprintf("Error: %v", m.err)))
-		b.WriteString("\n\nPress any key to exit.")
+		b.WriteString("\n\nPress any key to rescan, q to quit.")
 		return b.String()
 	}
 
@@ -786,7 +858,7 @@ func (m Model) renderResults(b *strings.Builder) string {
 		summary += fmt.Sprintf(" (%s freed)", ui.FormatSize(freedSize))
 	}
 	b.WriteString(successStyle.Render(summary))
-	b.WriteString("\n\nPress any key to exit.")
+	b.WriteString("\n\nPress any key to rescan, q to quit.")
 
 	return b.String()
 }