feat(tui-go): interactive clarification question view

Render stage clarification questions as an AskUserQuestion-style form: header chip, prompt, selectable option chips, and a free-text custom slot. Arrow/hjkl nav, space to select, e for custom, enter to submit. Submit guards on all-answered and sends ClarificationResponse; the parked stage re-runs with the answers in context.
2026-06-14 13:00:59 +04:00
parent 6242b590e4
commit 218a98034e
8 changed files with 584 additions and 3 deletions
@@ -0,0 +1,105 @@
+package app
+
+import (
+	"strings"
+	"testing"
+
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/correx/tui-go/internal/protocol"
+	"github.com/correx/tui-go/internal/ws"
+)
+
+func clarModel() Model {
+	m := NewModel(ws.New("", 0)) // unconnected; Send just buffers
+	m.width, m.height = 120, 40
+	m.theme = NewTheme(SoftBlue)
+	m.selectedID = "s1"
+	m.sessionEntered = true
+	m.applyServer(protocol.ServerMessage{
+		Type:      protocol.TypeClarification,
+		SessionID: "s1",
+		RequestID: "req-1",
+		StageID:   "analyst",
+		Questions: []protocol.ClarificationQuestionDto{
+			{ID: "stack", Prompt: "Which stack?", Options: []string{"React", "Vue"}, Header: "Stack"},
+		},
+	})
+	return m
+}
+
+func TestClarificationEntersStateAndRenders(t *testing.T) {
+	m := clarModel()
+	if m.displayState() != StateClarification {
+		t.Fatalf("want StateClarification, got %v", m.displayState())
+	}
+	out := m.clarificationModal()
+	for _, want := range []string{"clarifying questions", "Which stack?", "React", "Vue", "Stack", "custom"} {
+		if !strings.Contains(out, want) {
+			t.Fatalf("modal missing %q:\n%s", want, out)
+		}
+	}
+}
+
+func TestClarificationSelectAndSubmit(t *testing.T) {
+	m := clarModel()
+	// move the option cursor to "Vue" (index 1) and select it
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRight})
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeySpace})
+	if !m.clarChosenAt(0, 1) {
+		t.Fatalf("expected option 1 chosen, chosen=%v", m.clarChosen)
+	}
+	// submit
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
+	if s := m.session("s1"); s == nil || s.Clar != nil {
+		t.Fatalf("expected clarification cleared after submit")
+	}
+	if m.displayState() != StateInSession {
+		t.Fatalf("want StateInSession after submit, got %v", m.displayState())
+	}
+}
+
+func TestClarificationSubmitGuardWaitsForAllAnswers(t *testing.T) {
+	m := clarModel()
+	m.applyServer(protocol.ServerMessage{
+		Type: protocol.TypeClarification, SessionID: "s1", RequestID: "req-2", StageID: "analyst",
+		Questions: []protocol.ClarificationQuestionDto{
+			{ID: "a", Prompt: "Q A?", Options: []string{"x"}},
+			{ID: "b", Prompt: "Q B?", Options: []string{"y"}},
+		},
+	})
+	// answer only the first question, then attempt submit
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeySpace})
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
+	if s := m.session("s1"); s == nil || s.Clar == nil {
+		t.Fatalf("expected clarification to remain with partial answers")
+	}
+	if m.clarFocus != 1 {
+		t.Fatalf("expected focus nudged to first unanswered (1), got %d", m.clarFocus)
+	}
+}
+
+func TestClarificationCustomTextAnswer(t *testing.T) {
+	m := clarModel()
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("e")})
+	if !m.clarTyping {
+		t.Fatalf("expected typing mode after 'e'")
+	}
+	for _, r := range "Solid" {
+		m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
+	}
+	m = applyClarKey(m, tea.KeyMsg{Type: tea.KeyEnter})
+	if m.clarTyping {
+		t.Fatalf("expected typing committed")
+	}
+	if m.clarTextAt(0) != "Solid" {
+		t.Fatalf("want custom text 'Solid', got %q", m.clarTextAt(0))
+	}
+	if !m.clarAnswered(0, m.session("s1").Clar.Questions[0]) {
+		t.Fatalf("expected question answered via custom text")
+	}
+}
+
+func applyClarKey(m Model, k tea.KeyMsg) Model {
+	updated, _ := m.handleClarificationKey(k)
+	return updated.(Model)
+}