fix: adds global GracefulPanic and checks to deploy TUI (#542)

main.go

@@ -8,13 +8,15 @@ import (
 	"embed"
 
 	"github.com/defenseunicorns/uds-cli/src/cmd"
+	"github.com/defenseunicorns/uds-cli/src/pkg/utils"
 	"github.com/defenseunicorns/zarf/src/pkg/packager/lint"
 )
 
 //go:embed zarf.schema.json
 var zarfSchema embed.FS
 
 func main() {
+	defer utils.GracefulPanic()
 	lint.ZarfSchema = zarfSchema
 	cmd.Execute()
 }

src/cmd/uds.go

@@ -112,26 +112,6 @@ var deployCmd = &cobra.Command{
 	},
 }
 
-func deployWithoutTea(bndlClient *bundle.Bundle) {
-	_, _, _, err := bndlClient.PreDeployValidation()
-	if err != nil {
-		message.Fatalf(err, "Failed to validate bundle: %s", err.Error())
-	}
-	// confirm deployment
-	if ok := bndlClient.ConfirmBundleDeploy(); !ok {
-		message.Fatal(nil, "bundle deployment cancelled")
-	}
-	// create an empty program and kill it, this makes Program.Send a no-op
-	deploy.Program = tea.NewProgram(nil)
-	deploy.Program.Kill()
-
-	// deploy the bundle
-	if err := bndlClient.Deploy(); err != nil {
-		bndlClient.ClearPaths()
-		message.Fatalf(err, "Failed to deploy bundle: %s", err.Error())
-	}
-}
-
 var inspectCmd = &cobra.Command{
 	Use:     "inspect [BUNDLE_TARBALL|OCI_REF]",
 	Aliases: []string{"i"},
@@ -357,3 +337,23 @@ func chooseBundle(args []string) string {
 
 	return path
 }
+
+func deployWithoutTea(bndlClient *bundle.Bundle) {
+	_, _, _, err := bndlClient.PreDeployValidation()
+	if err != nil {
+		message.Fatalf(err, "Failed to validate bundle: %s", err.Error())
+	}
+	// confirm deployment
+	if ok := bndlClient.ConfirmBundleDeploy(); !ok {
+		message.Fatal(nil, "bundle deployment cancelled")
+	}
+	// create an empty program and kill it, this makes Program.Send a no-op
+	deploy.Program = tea.NewProgram(nil)
+	deploy.Program.Kill()
+
+	// deploy the bundle
+	if err := bndlClient.Deploy(); err != nil {
+		bndlClient.ClearPaths()
+		message.Fatalf(err, "Failed to deploy bundle: %s", err.Error())
+	}
+}

src/pkg/bundle/tui/deploy/model.go

@@ -192,7 +192,7 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		case tea.KeyMsg:
 			switch msg.String() {
 			case "y", "Y":
-				if !m.confirmed {
+				if !m.validatingBundle && !m.confirmed {
 					m.confirmed = true
 					m.inProgress = true
 				}
@@ -201,7 +201,7 @@ func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 				}
 
 			case "n", "N":
-				if !m.confirmed && !m.inProgress {
+				if !m.validatingBundle && !m.confirmed && !m.inProgress {
 					m.done = true
 					quitMsg := tea.Println(tui.IndentStyle.Render("\n👋 Deployment cancelled"))
 					return m, tea.Sequence(quitMsg, tea.Println(), tea.Quit)