internal/cmdlets: Add improved I/O to remap command

Author: the-maldridge

internal/cmdlets/fms_remap.go

@@ -11,6 +11,7 @@ import (
 	"os"
 	"sort"
 	"strconv"
+	"strings"
 
 	"github.com/AlecAivazis/survey/v2"
 	"github.com/spf13/cobra"
@@ -30,6 +31,7 @@ table.  This will disrupt any teams currently on the fms!`
 
 func init() {
 	fmsCmd.AddCommand(fmsRemapCmd)
+	fmsRemapCmd.Flags().Bool("clear", false, "Clear all mappings")
 }
 
 func fmsRemapCmdRun(c *cobra.Command, args []string) {
@@ -38,81 +40,105 @@ func fmsRemapCmdRun(c *cobra.Command, args []string) {
 		fAddr = "localhost:8080"
 	}
 
-	r, err := http.Get("http://" + fAddr + "/admin/cfg/quads")
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error getting quads: %s\n", err)
-		os.Exit(2)
-	}
+	promptQuads := func() map[string]string {
+		r, err := http.Get("http://" + fAddr + "/admin/cfg/quads")
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Error getting quads: %s\n", err)
+			os.Exit(2)
+		}
 
-	quads := []string{}
-	if err := json.NewDecoder(r.Body).Decode(&quads); err != nil {
-		fmt.Fprintf(os.Stderr, "Error getting quads: %s\n", err)
-		os.Exit(2)
-	}
-	sort.Strings(quads)
-	r.Body.Close()
+		quads := []string{}
+		if err := json.NewDecoder(r.Body).Decode(&quads); err != nil {
+			fmt.Fprintf(os.Stderr, "Error getting quads: %s\n", err)
+			os.Exit(2)
+		}
+		sort.Strings(quads)
+		r.Body.Close()
 
-	r, err = http.Get("http://" + fAddr + "/admin/map/current")
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error getting map: %s\n", err)
-		os.Exit(2)
-	}
+		r, err = http.Get("http://" + fAddr + "/admin/map/current")
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Error getting map: %s\n", err)
+			os.Exit(2)
+		}
 
-	cMap := make(map[string]string)
-	if err := json.NewDecoder(r.Body).Decode(&cMap); err != nil {
-		fmt.Fprintf(os.Stderr, "Error getting map: %s\n", err)
-		os.Exit(2)
-	}
-	ccMap := make(map[string]string, len(cMap))
-	r.Body.Close()
-
-	if len(cMap) > 0 {
-		fmt.Println("Current Mapping:")
-		for team, quad := range cMap {
-			fmt.Printf("  %s:\t%s\n", quad, team)
-			ccMap[quad] = team
+		cMap := make(map[string]string)
+		if err := json.NewDecoder(r.Body).Decode(&cMap); err != nil {
+			fmt.Fprintf(os.Stderr, "Error getting map: %s\n", err)
+			os.Exit(2)
 		}
-		fmt.Println()
-	}
-	fmt.Println("Enter new mapping")
+		ccMap := make(map[string]string, len(cMap))
+		r.Body.Close()
+
+		if len(cMap) > 0 {
+			fmt.Println("Current Mapping:")
+			for team, quad := range cMap {
+				fmt.Printf("  %s:\t%s\n", quad, team)
+				ccMap[quad] = team
+			}
+			fmt.Println()
+		}
+		fmt.Println("Enter new mapping")
 
-	tNumValidator := func(a interface{}) error {
-		if a.(string) == "-" {
+		tNumValidator := func(a interface{}) error {
+			if a.(string) == "-" {
+				return nil
+			}
+			if _, err := strconv.Atoi(a.(string)); err != nil {
+
+				return errors.New("team number must be a number")
+			}
 			return nil
 		}
-		if _, err := strconv.Atoi(a.(string)); err != nil {
+		qMap := []*survey.Question{}
+		for _, quad := range quads {
+			qMap = append(qMap, &survey.Question{
+				Name:     quad,
+				Validate: tNumValidator,
+				Prompt: &survey.Input{
+					Message: quad,
+					Default: ccMap[quad],
+				},
+			})
+		}
 
-			return errors.New("team number must be a number")
+		nMap := make(map[string]interface{})
+		if err := survey.Ask(qMap, &nMap); err != nil {
+			fmt.Fprintf(os.Stderr, "Error polling for fields: %s\n", err)
+			os.Exit(2)
+		}
+		nnMap := make(map[string]string, len(nMap))
+		for f, t := range nMap {
+			if t == "-" {
+				continue
+			}
+			nnMap[t.(string)] = f
 		}
-		return nil
-	}
-	qMap := []*survey.Question{}
-	for _, quad := range quads {
-		qMap = append(qMap, &survey.Question{
-			Name:     quad,
-			Validate: tNumValidator,
-			Prompt: &survey.Input{
-				Message: quad,
-				Default: ccMap[quad],
-			},
-		})
-	}
 
-	nMap := make(map[string]interface{})
-	if err := survey.Ask(qMap, &nMap); err != nil {
-		fmt.Fprintf(os.Stderr, "Error polling for fields: %s\n", err)
-		os.Exit(2)
+		return nnMap
 	}
 
-	nnMap := make(map[string]string, len(nMap))
-	for f, t := range nMap {
-		if t == "-" {
-			continue
+	mapping := make(map[string]string)
+	clear, _ := c.Flags().GetBool("clear")
+
+	switch {
+	case clear:
+		// Don't do anything for clear, just return the empty
+		// mapping.
+	case len(args) > 0:
+		// Parse the args as string pairs of the form
+		// fieldN:quad:team
+		for _, a := range args {
+			parts := strings.Split(a, ":")
+			if len(parts) != 3 {
+				continue
+			}
+			mapping[parts[2]] = strings.Join(parts[:2], ":")
 		}
-		nnMap[t.(string)] = f
+	default:
+		mapping = promptQuads()
 	}
 
 	buf := new(bytes.Buffer)
-	json.NewEncoder(buf).Encode(nnMap)
+	json.NewEncoder(buf).Encode(mapping)
 	http.Post("http://"+fAddr+"/admin/map/immediate", "application/json", buf)
 }