watchr/internal/ui/keys.go

package ui

import (
	tea "github.com/charmbracelet/bubbletea"
)

func (m *model) handleKeyPress(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	if m.showHelp {
		return m.handleHelpMode(msg)
	}
	if m.confirmMode {
		return m.handleConfirmMode(msg)
	}
	if m.cmdPaletteMode {
		return m.handleCmdPaletteMode(msg)
	}
	if m.filterMode {
		return m.handleFilterMode(msg)
	}
	return m.handleNormalMode(msg)
}

func (m *model) handleHelpMode(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	switch msg.String() {
	case "?", "esc", "q", "enter":
		m.showHelp = false
	}
	return m, nil
}

func (m *model) handleConfirmMode(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	switch msg.String() {
	case "y", "Y":
		m.confirmMode = false
		if m.confirmAction != nil {
			return m.confirmAction(m)
		}
	default:
		m.confirmMode = false
	}
	return m, nil
}

func (m *model) handleCmdPaletteMode(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	switch msg.Type {
	case tea.KeyEsc:
		m.cmdPaletteMode = false
		m.cmdPaletteFilter = ""
		m.cmdPaletteCursor = 0
		m.cmdPaletteSelected = 0
		return m, nil
	case tea.KeyEnter:
		filtered := m.filteredCommands()
		if len(filtered) > 0 && m.cmdPaletteSelected < len(filtered) {
			m.cmdPaletteMode = false
			cmd := filtered[m.cmdPaletteSelected]
			m.cmdPaletteFilter = ""
			m.cmdPaletteCursor = 0
			m.cmdPaletteSelected = 0
			return cmd.action(m)
		}
		return m, nil
	case tea.KeyUp:
		if m.cmdPaletteSelected > 0 {
			m.cmdPaletteSelected--
		}
		return m, nil
	case tea.KeyDown:
		filtered := m.filteredCommands()
		if m.cmdPaletteSelected < len(filtered)-1 {
			m.cmdPaletteSelected++
		}
		return m, nil
	case tea.KeyLeft:
		if msg.Alt {
			m.cmdPaletteCursor = wordBoundaryLeft(m.cmdPaletteFilter, m.cmdPaletteCursor)
		} else if m.cmdPaletteCursor > 0 {
			m.cmdPaletteCursor--
		}
		return m, nil
	case tea.KeyRight:
		if msg.Alt {
			m.cmdPaletteCursor = wordBoundaryRight(m.cmdPaletteFilter, m.cmdPaletteCursor)
		} else if m.cmdPaletteCursor < len(m.cmdPaletteFilter) {
			m.cmdPaletteCursor++
		}
		return m, nil
	case tea.KeyBackspace:
		if msg.Alt {
			m.cmdPaletteFilter, m.cmdPaletteCursor = textBackspaceWord(m.cmdPaletteFilter, m.cmdPaletteCursor)
		} else {
			m.cmdPaletteFilter, m.cmdPaletteCursor = textBackspace(m.cmdPaletteFilter, m.cmdPaletteCursor)
		}
		m.cmdPaletteSelected = 0
		return m, nil
	default:
		if len(msg.Runes) > 0 {
			m.cmdPaletteFilter, m.cmdPaletteCursor = textInsert(m.cmdPaletteFilter, string(msg.Runes), m.cmdPaletteCursor)
			m.cmdPaletteSelected = 0
		}
		return m, nil
	}
}

func (m *model) handleFilterMode(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	switch msg.Type {
	case tea.KeyEsc:
		m.filterMode = false
		m.filter = ""
		m.filterCursor = 0
		m.filterRegex = false
		m.filterRegexErr = nil
		m.updateFiltered()
		return m, nil
	case tea.KeyEnter:
		m.filterMode = false
		return m, nil
	case tea.KeyLeft:
		if msg.Alt {
			m.filterCursor = wordBoundaryLeft(m.filter, m.filterCursor)
		} else if m.filterCursor > 0 {
			m.filterCursor--
		}
		return m, nil
	case tea.KeyRight:
		if msg.Alt {
			m.filterCursor = wordBoundaryRight(m.filter, m.filterCursor)
		} else if m.filterCursor < len(m.filter) {
			m.filterCursor++
		}
		return m, nil
	case tea.KeyBackspace:
		if msg.Alt {
			m.filter, m.filterCursor = textBackspaceWord(m.filter, m.filterCursor)
		} else {
			m.filter, m.filterCursor = textBackspace(m.filter, m.filterCursor)
		}
		m.updateFiltered()
		return m, nil
	default:
		if len(msg.Runes) > 0 {
			s := string(msg.Runes)
			if s == "/" && m.filter == "" {
				m.filterRegex = !m.filterRegex
				m.filterRegexErr = nil
			} else {
				m.filter, m.filterCursor = textInsert(m.filter, s, m.filterCursor)
			}
			m.updateFiltered()
		}
		return m, nil
	}
}

func (m *model) handleNormalMode(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
	switch msg.String() {
	case "q", "ctrl+c":
		return m.actionQuit()
	case "esc":
		if m.filter != "" || m.filterRegex {
			m.filter = ""
			m.filterCursor = 0
			m.filterRegex = false
			m.filterRegexErr = nil
			m.updateFiltered()
			return m, nil
		}
		return m.actionQuit()

	case "j", "down", "ctrl+n":
		m.userScrolled = true
		m.moveCursor(1)
	case "k", "up", "ctrl+p":
		m.userScrolled = true
		m.moveCursor(-1)
	case "g", "home":
		return m.actionGoToFirst()
	case "G", "end":
		return m.actionGoToLast()
	case "ctrl+d":
		m.userScrolled = true
		m.moveCursor(m.visibleLines() / 2)
	case "ctrl+u":
		m.userScrolled = true
		m.moveCursor(-m.visibleLines() / 2)
	case "J":
		if m.showPreview {
			m.previewOffset++
			m.clampPreviewOffset()
		}
	case "K":
		if m.showPreview && m.previewOffset > 0 {
			m.previewOffset--
		}
	case "pgdown", "ctrl+f":
		m.userScrolled = true
		m.moveCursor(m.visibleLines())
	case "pgup", "ctrl+b":
		m.userScrolled = true
		m.moveCursor(-m.visibleLines())
	case "p":
		return m.actionTogglePreview()
	case "+", "=":
		return m.actionIncreasePreview()
	case "-":
		return m.actionDecreasePreview()
	case "r", "ctrl+r":
		return m.actionReload()
	case "R":
		return m.actionReloadClear()
	case "d", "delete":
		return m.actionDeleteLine()
	case "D":
		return m.actionClearAllLines()
	case "c":
		return m.actionStopCommand()
	case "/":
		return m.actionEnterFilter()
	case ":":
		return m.actionOpenPalette()
	case "?":
		return m.actionShowHelp()
	case "y":
		return m.actionCopyLine(false)
	case "Y":
		return m.actionCopyLine(true)
	}

	return m, nil
}