origin/console/command.go

package console

import (
	"fmt"
	"strconv"
	"strings"
)

type CommandFunctionCB func(param interface{})error

var mapRegisterCmd map[string]CommandFunctionCB =  map[string]CommandFunctionCB{}
var programName string

func Run(args []string) error {
	programName = args[0]
	if len(args) <= 1 {
		return fmt.Errorf("command not found, try `%s help` for help",args[0])
	}

	fn,ok := mapRegisterCmd[args[1]]
	if ok == false{
		return fmt.Errorf("command not found, try `%s help` for help",args[0])
	}

	switch  args[1] {
	case "start":
		if len(args)<2 {
			return fmt.Errorf("command not found, try `%s help` for help",args[0])
		}else{
			return start(fn,args[2])
		}
	case "stop":
		return fn(nil)
	}

	return fmt.Errorf("command not found, try `%s help` for help",args[0])
}

func start(fn CommandFunctionCB,param string) error {
	sparam := strings.Split(param,"=")
	if len(sparam) != 2 {
		return fmt.Errorf("invalid option %s",param)
	}
	if sparam[0]!="nodeid" {
		return fmt.Errorf("invalid option %s",param)
	}
	nodeId,err:= strconv.Atoi(sparam[1])
	if err != nil {
		return fmt.Errorf("invalid option %s",param)
	}

	return fn(nodeId)
}


func RegisterCommand(cmd string,fn CommandFunctionCB){
	mapRegisterCmd[cmd] = fn
}