feat(internal): Add method to run code before/after prompts

Signed-off-by: Cezar Craciunoiu <[email protected]>
unikraft · Jul 2, 2024 · 9cd123f · 9cd123f
1 parent 4fb0005
commit 9cd123f
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 6 deletions.
diff --git a/internal/cli/kraft/build/builder_kraftfile_unikraft.go b/internal/cli/kraft/build/builder_kraftfile_unikraft.go
@@ -472,7 +472,7 @@ func (build *builderKraftfileUnikraft) Build(ctx context.Context, opts *BuildOpt
 		configure := true
 
 		if opts.project.IsConfigured(*opts.Target) {
-			configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:")
+			configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:", func() {})
 			if err != nil {
 				return err
 			}

diff --git a/internal/cli/kraft/fetch/fetch.go b/internal/cli/kraft/fetch/fetch.go
@@ -424,7 +424,7 @@ func (opts *FetchOptions) Run(ctx context.Context, _ []string) error {
 			configure := true
 
 			if opts.project.IsConfigured(targ) {
-				configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:")
+				configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:", func() {})
 				if err != nil {
 					return err
 				}

diff --git a/internal/cli/kraft/lib/create/create.go b/internal/cli/kraft/lib/create/create.go
@@ -95,7 +95,7 @@ func (opts *CreateOptions) Run(ctx context.Context, args []string) error {
 
 	if !config.G[config.KraftKit](ctx).NoPrompt {
 		if !opts.GitInit {
-			opts.GitInit, err = confirm.NewConfirm("Do you want to intialise library with git:")
+			opts.GitInit, err = confirm.NewConfirm("Do you want to intialise library with git:", func() {})
 			if err != nil {
 				return err
 			}
@@ -124,7 +124,7 @@ func (opts *CreateOptions) Run(ctx context.Context, args []string) error {
 		}
 
 		if !opts.UpdateRefs {
-			opts.UpdateRefs, err = confirm.NewConfirm("Do you want to package it:")
+			opts.UpdateRefs, err = confirm.NewConfirm("Do you want to package it:", func() {})
 			if err != nil {
 				return err
 			}

diff --git a/internal/cli/kraft/menu/menu.go b/internal/cli/kraft/menu/menu.go
@@ -426,7 +426,7 @@ func (opts *MenuOptions) Run(ctx context.Context, _ []string) error {
 			configure := true
 
 			if opts.project.IsConfigured(targ) {
-				configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:")
+				configure, err = confirm.NewConfirm("project already configured, are you sure you want to rerun the configure step:", func() {})
 				if err != nil {
 					return err
 				}

diff --git a/tui/confirm/confirm.go b/tui/confirm/confirm.go
@@ -13,7 +13,7 @@ import (
 
 // NewConfirm is a utility method used in a CLI context to prompt the user with
 // a yes/no question.
-func NewConfirm(question string) (bool, error) {
+func NewConfirm(question string, beforeAfter func()) (bool, error) {
 	input := confirmation.New(
 		tui.TextWhiteBgBlue("[?]")+" "+
 			question,
@@ -24,5 +24,8 @@ func NewConfirm(question string) (bool, error) {
 	input.KeyMap.SelectYes = append(input.KeyMap.SelectYes, "+")
 	input.KeyMap.SelectNo = append(input.KeyMap.SelectNo, "-")
 
+	beforeAfter()
+	defer beforeAfter()
+
 	return input.RunPrompt()
 }