refactor: extract parseExtraArgs

cmd/actions/down.go

@@ -17,31 +17,16 @@ var (
 		Example: "  squadron down --namespace demo",
 		Args:    cobra.MinimumNArgs(0),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			var extraArgs []string
-			var units []string
-			for i, arg := range args {
-				if arg == "--" {
-					extraArgs = args[i:]
-					units = args[:i]
-				}
-			}
-			return down(log, units, cwd, flagNamespace, extraArgs)
+			_, helmArgs := parseExtraArgs(args)
+			return down(log, cwd, flagNamespace, helmArgs)
 		},
 	}
 )
 
-func down(l *logrus.Entry, unitNames []string, cwd, namespace string, extraArgs []string) error {
+func down(l *logrus.Entry, cwd, namespace string, extraArgs []string) error {
 	sq, err := squadron.New(l, cwd, namespace, nil)
 	if err != nil {
 		return err
 	}
-
-	var units map[string]squadron.Unit
-	if len(unitNames) == 0 {
-		units = sq.Units()
-	}
-	for _, un := range unitNames {
-		units[un] = sq.Units()[un]
-	}
 	return sq.Down(extraArgs)
 }

cmd/actions/root.go

@@ -53,3 +53,13 @@ func newLogger(verbose bool) *logrus.Entry {
 	}
 	return logrus.NewEntry(logger)
 }
+
+func parseExtraArgs(args []string) (out []string, extraArgs []string) {
+	for i, arg := range args {
+		if arg == "--" {
+			out, extraArgs = args[:i], args[i+1:]
+			break
+		}
+	}
+	return
+}

cmd/actions/up.go

@@ -19,16 +19,8 @@ var (
 		Example: "  squadron up frontend backend --build --push --namespace demo -- --dry-run",
 		Args:    cobra.MinimumNArgs(0),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			var extraArgs []string
-			units := args
-			for i, arg := range args {
-				if arg == "--" {
-					extraArgs = args[i+1:]
-					units = args[:i]
-					break
-				}
-			}
-			return up(log, units, cwd, flagNamespace, flagBuild, flagPush, flagFiles, extraArgs)
+			unitNames, helmArgs := parseExtraArgs(args)
+			return up(log, unitNames, cwd, flagNamespace, flagBuild, flagPush, flagFiles, helmArgs)
 		},
 	}
 )