add force flag for update (#101)

* feat: add force flag for update Relates to #95 * feat: rename flag * feat: remove viper dependency * feat: add shorthand for yes * Update cmd/update.go Co-authored-by: Sune Keller <[email protected]> Co-authored-by: Sune Keller <[email protected]>
marcosnils · Jun 23, 2021 · a638c4a · a638c4a
1 parent 83f4e74
commit a638c4a
Showing 1 changed file with 21 additions and 18 deletions.
diff --git a/cmd/update.go b/cmd/update.go
@@ -3,15 +3,14 @@ package cmd
 import (
 	"bufio"
 	"fmt"
-	"os"
-	"strings"
-
 	"github.com/apex/log"
 	"github.com/fatih/color"
 	"github.com/hashicorp/go-version"
 	"github.com/marcosnils/bin/pkg/config"
 	"github.com/marcosnils/bin/pkg/providers"
 	"github.com/spf13/cobra"
+	"os"
+	"strings"
 )
 
 type updateCmd struct {
@@ -20,8 +19,9 @@ type updateCmd struct {
 }
 
 type updateOpts struct {
-	dryRun bool
-	all    bool
+	yesToUpdate bool
+	dryRun      bool
+	all         bool
 }
 
 type updateInfo struct{ version, url string }
@@ -84,21 +84,23 @@ func newUpdateCmd() *updateCmd {
 				return wrapErrorWithCode(fmt.Errorf("Updates found, exit (dry-run mode)."), 3, "")
 			}
 
-			// TODO will have to refactor this prompt to a separate function
-			// so it can be reused in some other places
-			fmt.Printf("\nDo you want to continue? [Y/n] ")
-			reader := bufio.NewReader(os.Stdin)
-			var response string
+			if !root.opts.yesToUpdate {
+				// TODO will have to refactor this prompt to a separate function
+				// so it can be reused in some other places
+				fmt.Printf("\nDo you want to continue? [Y/n] ")
+				reader := bufio.NewReader(os.Stdin)
+				var response string
 
-			response, err := reader.ReadString('\n')
-			if err != nil {
-				return fmt.Errorf("Invalid input")
-			}
+				response, err := reader.ReadString('\n')
+				if err != nil {
+					return fmt.Errorf("Invalid input")
+				}
 
-			switch strings.ToLower(strings.TrimSpace(response)) {
-			case "y", "yes":
-			default:
-				return fmt.Errorf("Command aborted")
+				switch strings.ToLower(strings.TrimSpace(response)) {
+				case "y", "yes":
+				default:
+					return fmt.Errorf("Command aborted")
+				}
 			}
 
 			// TODO 	:S code smell here, this pretty much does
@@ -139,6 +141,7 @@ func newUpdateCmd() *updateCmd {
 
 	root.cmd = cmd
 	root.cmd.Flags().BoolVarP(&root.opts.dryRun, "dry-run", "", false, "Only show status, don't prompt for update")
+	root.cmd.Flags().BoolVarP(&root.opts.yesToUpdate, "yes", "y", false, "Assume yes to update prompt")
 	root.cmd.Flags().BoolVarP(&root.opts.all, "all", "a", false, "Show all possible download options (skip scoring & filtering)")
 	return root
 }