From ef5d89cc81b71e84a742b2281e35f645e021bd88 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Paul=20B=C3=BCtow?=
 <1224732+snonux@users.noreply.github.com>
Date: Fri, 20 Jun 2025 21:29:02 +0300
Subject: fix fireworks exit

---
 internal/ui/fireworks.go  | 13 +++++++++----
 internal/ui/table_test.go |  2 +-
 2 files changed, 10 insertions(+), 5 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
diff --git a/internal/ui/table_test.go b/internal/ui/table_test.go
index 1161ec2..c2d37f5 100644
--- a/internal/ui/table_test.go
+++ b/internal/ui/table_test.go
@@ -202,7 +202,7 @@ func TestUndoHotkey(t *testing.T) {
 		os.Unsetenv("TASKRC")
 	})
 
-	m, err := New("")
+	m, err := New(nil)
 	if err != nil {
 		t.Fatalf("New: %v", err)
 	}
-- 
cgit v1.2.3