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,20 +22,33 @@ const (
 	CommandDebugListMTrace    = "debug-ls-mtrace"
 )
 
-type MainServerCommandValidator struct{}
+var ValidCommands = []string{
+	CommandStart,
+	CommandValidate,
+	CommandListConfigs,
+	CommandListRoutes,
+	CommandListIcons,
+	CommandReload,
+	CommandDebugListEntries,
+	CommandDebugListProviders,
+	CommandDebugListMTrace,
+}
 
-func (v MainServerCommandValidator) IsCommandValid(cmd string) bool {
+func validateArg(arg string) error {
-	switch cmd {
+	for _, v := range ValidCommands {
-	case CommandStart,
+		if arg == v {
-		CommandValidate,
+			return nil
-		CommandListConfigs,
+		}
-		CommandListRoutes,
-		CommandListIcons,
-		CommandReload,
-		CommandDebugListEntries,
-		CommandDebugListProviders,
-		CommandDebugListMTrace:
-		return true
 	}
 	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
+}