fix args.go affected from cherry-pick

internal/common/args.go

@@ -1,7 +1,9 @@
 package common
 
 import (
+	"flag"
 	"fmt"
+	"log"
 )
 
 type Args struct {
@@ -20,11 +22,8 @@ const (
 	CommandDebugListMTrace    = "debug-ls-mtrace"
 )
 
-type MainServerCommandValidator struct{}
+var ValidCommands = []string{
-
+	CommandStart,
-func (v MainServerCommandValidator) IsCommandValid(cmd string) bool {
-	switch cmd {
-	case CommandStart,
 	CommandValidate,
 	CommandListConfigs,
 	CommandListRoutes,
@@ -32,8 +31,24 @@ func (v MainServerCommandValidator) IsCommandValid(cmd string) bool {
 	CommandReload,
 	CommandDebugListEntries,
 	CommandDebugListProviders,
-		CommandDebugListMTrace:
+	CommandDebugListMTrace,
-		return true
+}
+
+func validateArg(arg string) error {
+	for _, v := range ValidCommands {
+		if arg == v {
+			return nil
+		}
 	}
 	return fmt.Errorf("invalid command %q", arg)
 }
+
+func GetArgs() Args {
+	var args Args
+	flag.Parse()
+	args.Command = flag.Arg(0)
+	if err := validateArg(args.Command); err != nil {
+		log.Fatalf("invalid command: %s", err)
+	}
+	return args
+}