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

Signed-off-by: Cezar Craciunoiu <[email protected]>
unikraft · Jun 5, 2024 · 8427fa9 · 8427fa9
1 parent 3fe4a52
commit 8427fa9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
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/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()
 }