add: generate

1 files changed, 12 insertions, 0 deletions

diff --git a/internal/cli/cli.go b/internal/cli/cli.go
index 6609a7e..fecf64b 100644
--- a/internal/cli/cli.go
+++ b/internal/cli/cli.go
@@ -32,6 +32,7 @@ type model struct {
 	commands []command
 	busy     bool
 	quitting bool
+	events   chan tea.Msg
 
 	width int
 }
@@ -100,6 +101,17 @@ func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		m.busy = false
 		return m, tea.Println(msg.text)
 
+	case streamMsg:
+		return m, tea.Sequence(tea.Println(msg.text), waitForEvent(m.events))
+
+	case streamDoneMsg:
+		m.busy = false
+		m.events = nil
+		if msg.text == "" {
+			return m, nil
+		}
+		return m, tea.Println(msg.text)
+
 	case spinner.TickMsg:
 		if !m.busy {
 			return m, nil