From 9df896bc25e4558b8dae3ad83fc07d74d017cf9a Mon Sep 17 00:00:00 2001
From: Kashif Khan <70996046+kashifkhan0771@users.noreply.github.com>
Date: Thu, 31 Jul 2025 22:21:21 +0500
Subject: [PATCH] Do not quit TUI, if 'q' is pressed on specific pages (#4356)

* Do not quit TUI, if 'q' is pressed on Analyze Form page

* quit for specific pages only
---
 pkg/tui/tui.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/tui/tui.go b/pkg/tui/tui.go
index 9e833bffc..0faffaf35 100644
--- a/pkg/tui/tui.go
+++ b/pkg/tui/tui.go
@@ -142,7 +142,9 @@ func (ui *TUI) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		case tea.KeyMsg:
 			switch {
 			case key.Matches(msg, ui.common.KeyMap.Help):
-			case key.Matches(msg, ui.common.KeyMap.CmdQuit) && ui.activePage() != sourceConfigurePage:
+			case key.Matches(msg, ui.common.KeyMap.CmdQuit) &&
+				(ui.activePage() == wizardIntroPage || ui.activePage() == analyzeKeysPage || ui.activePage() == sourceSelectPage):
+
 				ui.args = nil
 				return ui, tea.Quit
 			case key.Matches(msg, ui.common.KeyMap.Quit):