ui: add ultra mode entry and exit hotkeys (vl vm)

1 files changed, 70 insertions, 0 deletions

diff --git a/internal/ui/table_test.go b/internal/ui/table_test.go
index 25eea72..ec2b9bc 100644
--- a/internal/ui/table_test.go
+++ b/internal/ui/table_test.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"regexp"
 	"strings"
 	"testing"
 	"time"
@@ -659,6 +660,75 @@ func TestEscClosesHelp(t *testing.T) {
 	}
 }
 
+func TestUltraExitHotkeysClearUltraState(t *testing.T) {
+	tmp := t.TempDir()
+	taskPath := setupBasicTask(t, tmp)
+	setupEnv(t, taskPath)
+
+	m, err := New(nil, "firefox")
+	if err != nil {
+		t.Fatalf("New: %v", err)
+	}
+
+	mv, cmd := (&m).Update(tea.KeyPressMsg{Code: 'u', Text: "u"})
+	if cmd != nil {
+		t.Fatalf("u unexpectedly returned a command")
+	}
+	m = *mv.(*Model)
+	if !m.showUltra {
+		t.Fatalf("u did not enter ultra mode")
+	}
+
+	m.ultraSearchRegex = regexp.MustCompile("alpha")
+	m.ultraFiltered = []int{0, 1}
+	m.ultraSearchInput.SetValue("ultra needle")
+
+	mv, cmd = (&m).Update(tea.KeyPressMsg{Code: 'q', Text: "q"})
+	if cmd != nil {
+		t.Fatalf("q in ultra mode unexpectedly returned a command")
+	}
+	m = *mv.(*Model)
+	if m.showUltra {
+		t.Fatalf("q did not exit ultra mode")
+	}
+	if m.ultraSearchRegex != nil {
+		t.Fatalf("q did not clear ultraSearchRegex")
+	}
+	if m.ultraFiltered != nil {
+		t.Fatalf("q did not clear ultraFiltered")
+	}
+	if got := m.ultraSearchInput.Value(); got != "" {
+		t.Fatalf("q did not clear ultraSearchInput, got %q", got)
+	}
+
+	mv, cmd = (&m).Update(tea.KeyPressMsg{Code: 'u', Text: "u"})
+	if cmd != nil {
+		t.Fatalf("u unexpectedly returned a command on re-entry")
+	}
+	m = *mv.(*Model)
+	m.ultraSearchRegex = regexp.MustCompile("beta")
+	m.ultraFiltered = []int{2}
+	m.ultraSearchInput.SetValue("second needle")
+
+	mv, cmd = (&m).Update(tea.KeyPressMsg{Code: tea.KeyEsc})
+	if cmd != nil {
+		t.Fatalf("esc in ultra mode unexpectedly returned a command")
+	}
+	m = *mv.(*Model)
+	if m.showUltra {
+		t.Fatalf("esc did not exit ultra mode")
+	}
+	if m.ultraSearchRegex != nil {
+		t.Fatalf("esc did not clear ultraSearchRegex")
+	}
+	if m.ultraFiltered != nil {
+		t.Fatalf("esc did not clear ultraFiltered")
+	}
+	if got := m.ultraSearchInput.Value(); got != "" {
+		t.Fatalf("esc did not clear ultraSearchInput, got %q", got)
+	}
+}
+
 // TestExpandedCellViewNoDoubleRender is a regression test for a bug where
 // expandedCellView() was appended to the layout unconditionally AND again
 // inside the cellExpanded guard, producing a duplicate line when expanded.