Fix bug in PrintCommandLineWithCursor when flag is a slice of strings.

VirusTotal · Aug 2, 2018 · 2cdae17 · 2cdae17
1 parent cddffd6
commit 2cdae17
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -132,7 +132,15 @@ func PrintCommandLineWithCursor(cmd *cobra.Command, it *vt.Iterator) {
 		flags := make([]string, 0)
 		cmd.Flags().Visit(func(flag *pflag.Flag) {
 			if flag.Name != "cursor" {
-				flags = append(flags, fmt.Sprintf("--%s=%v", flag.Name, flag.Value.String()))
+				var f string
+				switch flag.Value.Type() {
+				case "stringSlice":
+					ss, _ := cmd.Flags().GetStringSlice(flag.Name)
+					f = fmt.Sprintf("--%s='%s'", flag.Name, strings.Join(ss, ","))
+				default:
+					f = fmt.Sprintf("--%s=%v", flag.Name, flag.Value.String())
+				}
+				flags = append(flags, f)
 			}
 		})
 		flags = append(flags, fmt.Sprintf("--cursor=%s", cursor))