Merge pull request #22 from b4b4r07/feature/ask

Add feature to ask if it's ok to run command

Author: b4b4r07

cmd/install.go

@@ -72,6 +72,12 @@ func newInstallCmd() *cobra.Command {
 				pkgs = given
 			}
 
+			yes, _ := c.askRunCommand(*c, getNameInPackages(pkgs))
+			if !yes {
+				fmt.Println("Cancelled")
+				return nil
+			}
+
 			c.Env.AskWhen(map[string]bool{
 				"GITHUB_TOKEN":      config.HasGitHubReleaseBlock(pkgs),
 				"AFX_SUDO_PASSWORD": config.HasSudoInCommandBuildSteps(pkgs),

cmd/meta.go

@@ -7,8 +7,10 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"sort"
 	"strings"
 
+	"github.com/AlecAivazis/survey/v2"
 	"github.com/b4b4r07/afx/pkg/config"
 	"github.com/b4b4r07/afx/pkg/env"
 	"github.com/b4b4r07/afx/pkg/errors"
@@ -191,6 +193,45 @@ func (m *meta) prompt() (config.Package, error) {
 	return nil, errors.New("pkg not found")
 }
 
+func (m *meta) askRunCommand(op interface{}, pkgs []string) (bool, error) {
+	var do string
+	switch op.(type) {
+	case installCmd:
+		do = "install"
+	case uninstallCmd:
+		do = "uninstall"
+	case updateCmd:
+		do = "update"
+	default:
+		return false, errors.New("unsupported command type")
+	}
+
+	length := 3
+	target := strings.Join(pkgs, ", ")
+	if len(pkgs) > length {
+		target = fmt.Sprintf("%s, ... (%d packages)", strings.Join(pkgs[:length], ", "), len(pkgs))
+	}
+
+	yes := false
+	confirm := survey.Confirm{
+		Message: fmt.Sprintf("OK to %s these packages? %s", do, color.YellowString(target)),
+	}
+
+	if len(pkgs) > length {
+		helpMessage := "\n"
+		sort.Strings(pkgs)
+		for _, pkg := range pkgs {
+			helpMessage += fmt.Sprintf("- %s\n", pkg)
+		}
+		confirm.Help = helpMessage
+	}
+
+	if err := survey.AskOne(&confirm, &yes); err != nil {
+		return false, errors.Wrap(err, "failed to catch answer from console")
+	}
+	return yes, nil
+}
+
 func getNameInPackages(pkgs []config.Package) []string {
 	var keys []string
 	for _, pkg := range pkgs {

cmd/uninstall.go

@@ -69,6 +69,12 @@ func newUninstallCmd() *cobra.Command {
 				resources = given
 			}
 
+			yes, _ := c.askRunCommand(*c, getNameInResources(resources))
+			if !yes {
+				fmt.Println("Cancelled")
+				return nil
+			}
+
 			return c.run(resources)
 		},
 		PostRunE: func(cmd *cobra.Command, args []string) error {

cmd/update.go

@@ -72,6 +72,12 @@ func newUpdateCmd() *cobra.Command {
 				pkgs = given
 			}
 
+			yes, _ := c.askRunCommand(*c, getNameInPackages(pkgs))
+			if !yes {
+				fmt.Println("Cancelled")
+				return nil
+			}
+
 			c.Env.AskWhen(map[string]bool{
 				"GITHUB_TOKEN":      config.HasGitHubReleaseBlock(pkgs),
 				"AFX_SUDO_PASSWORD": config.HasSudoInCommandBuildSteps(pkgs),