/
command.go
47 lines (41 loc) · 1.06 KB
/
command.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package commands
import (
"fmt"
"github.com/zhsyourai/URCF-engine/commands/account"
"github.com/zhsyourai/URCF-engine/commands/kill"
"github.com/zhsyourai/URCF-engine/commands/serve"
"github.com/zhsyourai/URCF-engine/commands/version"
"gopkg.in/alecthomas/kingpin.v2"
"os"
)
var app = kingpin.New("urcf", "Universal Remote Config Framework Engine")
var registry = make(map[string]func() error)
func register(processors map[string]func() error) {
for command, processor := range processors {
if registry[command] != nil {
panic(fmt.Errorf("command %q is already registered", command))
}
registry[command] = processor
}
}
func init() {
register(version.Prepare(app))
register(serve.Prepare(app))
register(kill.Prepare(app))
register(account.Prepare(app))
}
func Run() int {
selected := kingpin.MustParse(app.Parse(os.Args[1:]))
if selected == "" {
app.Usage(os.Args[1:])
return -1
}
processor := registry[selected]
if processor == nil {
panic(fmt.Errorf("command %q not found", selected))
}
if processor() != nil {
return -1
}
return 0
}