Merge pull request #46 from snonux/codex/fix-fireworks-on-exit-with-q-key

Fix fireworks exit by ignoring leftover key

1 files changed, 9 insertions, 4 deletions

diff --git a/internal/ui/fireworks.go b/internal/ui/fireworks.go
index c902a67..8442541 100644
--- a/internal/ui/fireworks.go
+++ b/internal/ui/fireworks.go
@@ -9,10 +9,11 @@ import (
 )
 
 type fwModel struct {
-	width  int
-	height int
-	start  time.Time
-	fws    []firework
+	width          int
+	height         int
+	start          time.Time
+	fws            []firework
+	ignoreFirstKey bool
 }
 
 type firework struct {
@@ -66,6 +67,10 @@ func (m fwModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		}
 		return m, tick()
 	case tea.KeyMsg:
+		if !m.ignoreFirstKey {
+			m.ignoreFirstKey = true
+			return m, nil
+		}
 		return m, tea.Quit
 	}
 	return m, nil