add confirmation prompt (#24)

* add confirmation prompt

* update changelog

Author: akerl-unpriv

CHANGELOG.md

@@ -1,3 +1,7 @@
+# 2.1.1 / 2019-08-19
+
+* [FEATURE] Show confirmation prompt before running xargs
+
 # 2.1.0 / 2019-08-19
 
 * [FEATURE] Add xargs subcommand

cmd/xargs.go

@@ -27,6 +27,7 @@ func init() {
 	xargsCmd.Flags().StringP("prompt", "p", "", "Choose prompt to use")
 	xargsCmd.Flags().BoolP("yubikey", "y", false, "Use Yubikey for MFA")
 	xargsCmd.Flags().StringP("command", "c", "", "Command to execute")
+	xargsCmd.Flags().Bool("skipconfirm", false, "Skip confirmation prompt")
 }
 
 // revive:disable-next-line:cyclomatic
@@ -66,6 +67,11 @@ func xargsRunner(cmd *cobra.Command, args []string) error {
 		return fmt.Errorf("command must be provided via --command / -c")
 	}
 
+	skipConfirm, err := flags.GetBool("skipconfirm")
+	if err != nil {
+		return err
+	}
+
 	pack := cartogram.Pack{}
 	if err := pack.Load(); err != nil {
 		return err
@@ -90,6 +96,7 @@ func xargsRunner(cmd *cobra.Command, args []string) error {
 		Args:         args,
 		RoleNames:    []string{flagRole},
 		ProfileNames: []string{flagProfile},
+		SkipConfirm:  skipConfirm,
 	}
 
 	results, err := processor.ExecString(commandStr)

multi/main.go

@@ -1,6 +1,8 @@
 package multi
 
 import (
+	"bufio"
+	"fmt"
 	"os"
 	"strings"
 	"time"
@@ -22,6 +24,7 @@ type Processor struct {
 	Args         []string
 	RoleNames    []string
 	ProfileNames []string
+	SkipConfirm  bool
 }
 
 // ExecString runs a command string against a set of accounts
@@ -39,6 +42,10 @@ func (p Processor) Exec(cmd []string) (map[string]creds.ExecResult, error) {
 		return map[string]creds.ExecResult{}, err
 	}
 
+	if !p.confirm(paths) {
+		return map[string]creds.ExecResult{}, fmt.Errorf("aborted by user")
+	}
+
 	inputCh := make(chan workerInput, len(paths))
 	outputCh := make(chan workerOutput, len(paths))
 	refreshCh := make(chan time.Time)
@@ -82,6 +89,34 @@ func (p Processor) Exec(cmd []string) (map[string]creds.ExecResult, error) {
 	return output, nil
 }
 
+func (p Processor) confirm(paths []travel.Path) bool {
+	if p.SkipConfirm {
+		return true
+	}
+	fmt.Fprintln(os.Stderr, "Will run on the following accounts:")
+	for _, item := range paths {
+		accountID := item[len(item)-1].Account
+		ok, account := p.Grapher.Pack.Lookup(accountID)
+		if !ok {
+			fmt.Fprintf(os.Stderr, "Failed account lookup: %s\n", accountID)
+			return false
+		}
+		fmt.Fprintf(os.Stderr, "  %s -- %s\n", account.Account, account.Tags)
+	}
+	fmt.Fprintln(os.Stderr, "Type 'yes' to confirm")
+	confirmReader := bufio.NewReader(os.Stdin)
+	confirmInput, err := confirmReader.ReadString('\n')
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error reading prompt: %s", err)
+		return false
+	}
+	cleanedInput := strings.TrimSpace(confirmInput)
+	if cleanedInput != "yes" {
+		return false
+	}
+	return true
+}
+
 type workerInput struct {
 	Path    travel.Path
 	Options travel.TraverseOptions