origin/console/command.go

package console

import (
	"flag"
	"fmt"
	"os"
)

type valueType int
type CommandFunctionCB func(args interface{}) error

var commandList []*command
var programName string

const (
	boolType   valueType = 0
	stringType valueType = 1
	intType    valueType = 2
)

type command struct {
	valType  valueType
	name     string
	bValue   bool
	strValue string
	intValue int
	usage    string
	fn       CommandFunctionCB
}

func (cmd *command) execute() error {
	if cmd.valType == boolType {
		return cmd.fn(cmd.bValue)
	} else if cmd.valType == stringType {
		return cmd.fn(cmd.strValue)
	} else if cmd.valType == intType {
		return cmd.fn(cmd.intValue)
	}

	return fmt.Errorf("unknow command type")
}

func Run(args []string) error {
	flag.Parse()
	programName = args[0]
	if flag.NFlag() <= 0 {
		return fmt.Errorf("Command input parameter error,try `%s -help` for help", args[0])
	}

	// 只收集命令行中实际传入的 flag
	setFlags := make(map[string]bool)
	flag.Visit(func(f *flag.Flag) {
		setFlags[f.Name] = true
	})

	var startCmd *command
	for _, val := range commandList {
		if !setFlags[val.name] {
			continue // 未传入的命令跳过
		}
		if val.name == "start" {
			startCmd = val
			continue
		}
		err := val.execute()
		if err != nil {
			return err
		}
	}

	if startCmd != nil {
		return startCmd.execute()
	}

	return fmt.Errorf("Command input parameter error,try `%s -help` for help", args[0])
}

func RegisterCommandBool(cmdName string, defaultValue bool, usage string, fn CommandFunctionCB) {
	var cmd command
	cmd.valType = boolType
	cmd.name = cmdName
	cmd.fn = fn
	cmd.usage = usage
	flag.BoolVar(&cmd.bValue, cmdName, defaultValue, usage)
	commandList = append(commandList, &cmd)
}

func RegisterCommandInt(cmdName string, defaultValue int, usage string, fn CommandFunctionCB) {
	var cmd command
	cmd.valType = intType
	cmd.name = cmdName
	cmd.fn = fn
	cmd.usage = usage
	flag.IntVar(&cmd.intValue, cmdName, defaultValue, usage)
	commandList = append(commandList, &cmd)
}

func RegisterCommandString(cmdName string, defaultValue string, usage string, fn CommandFunctionCB) {
	var cmd command
	cmd.valType = stringType
	cmd.name = cmdName
	cmd.fn = fn
	cmd.usage = usage
	flag.StringVar(&cmd.strValue, cmdName, defaultValue, usage)
	commandList = append(commandList, &cmd)
}

func PrintDefaults() {
	fmt.Fprintf(os.Stderr, "Options:\n")

	for _, val := range commandList {
		fmt.Fprintf(os.Stderr, "  -%-10s%10s\n", val.name, val.usage)
	}
}

func GetParamStringVal(paramName string) string {
	for _, cmd := range commandList {
		if cmd.name == paramName {
			return cmd.strValue
		}
	}
	return ""
}

func GetParamBoolVal(paramName string) bool {
	for _, cmd := range commandList {
		if cmd.name == paramName {
			return cmd.bValue
		}
	}

	return false
}