mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 18:17:52 -05:00
55 lines
858 B
Go
55 lines
858 B
Go
package control
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type Description struct {
|
|
Short string
|
|
Usage []string
|
|
}
|
|
|
|
type Command interface {
|
|
Name() string
|
|
Description() Description
|
|
Execute(args []string) error
|
|
}
|
|
|
|
var (
|
|
commandRegistry = make(map[string]Command)
|
|
ctllog = log.New(os.Stderr, "v2ctl> ", 0)
|
|
)
|
|
|
|
func RegisterCommand(cmd Command) error {
|
|
entry := strings.ToLower(cmd.Name())
|
|
if entry == "" {
|
|
return newError("empty command name")
|
|
}
|
|
commandRegistry[entry] = cmd
|
|
return nil
|
|
}
|
|
|
|
func GetCommand(name string) Command {
|
|
cmd, found := commandRegistry[name]
|
|
if !found {
|
|
return nil
|
|
}
|
|
return cmd
|
|
}
|
|
|
|
type hiddenCommand interface {
|
|
Hidden() bool
|
|
}
|
|
|
|
func PrintUsage() {
|
|
for name, cmd := range commandRegistry {
|
|
if _, ok := cmd.(hiddenCommand); ok {
|
|
continue
|
|
}
|
|
fmt.Println(" ", name, "\t\t\t", cmd.Description())
|
|
}
|
|
}
|