97 lines
1.5 KiB
Go
97 lines
1.5 KiB
Go
|
//
|
|||
|
|
// cli.go
|
|||
|
|
// Copyright (C) 2022 tiglog <me@tiglog.com>
|
|||
|
|
//
|
|||
|
|
// Distributed under terms of the MIT license.
|
|||
|
|
//
|
|||
|
|
|
|||
|
|
package console
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type sApp struct {
|
|||
|
|
name string
|
|||
|
|
version string
|
|||
|
|
desc string
|
|||
|
|
cmds map[string]ICommand
|
|||
|
|
about ActionHandler
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func New(name, desc string) *sApp {
|
|||
|
|
app := &sApp{
|
|||
|
|
name: name,
|
|||
|
|
version: "v0.1.0",
|
|||
|
|
desc: desc,
|
|||
|
|
cmds: make(map[string]ICommand),
|
|||
|
|
}
|
|||
|
|
app.AddCmd(NewAboutCmd(app))
|
|||
|
|
app.AddCmd(NewAirCmd(app))
|
|||
|
|
app.AddCmd(NewListCmd(app))
|
|||
|
|
app.AddCmd(NewHelpCmd(app))
|
|||
|
|
return app
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) GetCmds() map[string]ICommand {
|
|||
|
|
return s.cmds
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) GetName() string {
|
|||
|
|
return s.name
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) GetDesc() string {
|
|||
|
|
return s.desc
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) GetVersion() string {
|
|||
|
|
return s.version
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) AddCmd(cmd ICommand) {
|
|||
|
|
s.cmds[cmd.GetName()] = cmd
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) HasCmd(cmd string) bool {
|
|||
|
|
_, ok := s.cmds[cmd]
|
|||
|
|
return ok
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) SetExtraAbout(about ActionHandler) {
|
|||
|
|
s.about = about
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) GetExtraAbout() ActionHandler {
|
|||
|
|
return s.about
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *sApp) Run(args []string) {
|
|||
|
|
cmd := "list"
|
|||
|
|
if len(args) == 1 {
|
|||
|
|
args = []string{cmd}
|
|||
|
|
} else {
|
|||
|
|
cmd = args[1]
|
|||
|
|
args = args[2:]
|
|||
|
|
}
|
|||
|
|
if !s.HasCmd(cmd) {
|
|||
|
|
fmt.Printf("%q is not valid command.\n", cmd)
|
|||
|
|
os.Exit(2)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
scmd := s.cmds[cmd]
|
|||
|
|
scmd.Init(args)
|
|||
|
|
act := scmd.GetAction()
|
|||
|
|
if act == nil {
|
|||
|
|
// fmt.Println("未定义 Action，无法执行 Action.")
|
|||
|
|
fmt.Println(scmd.GetHelp())
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
err := act()
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Printf("执行异常：%v\n", err)
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
}
|