From 29e12c1aa6e592e00fff83aac624d368bba40613 Mon Sep 17 00:00:00 2001 From: tiglog Date: Sun, 15 Oct 2023 00:49:22 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20Console=20App=20=E9=9C=80=E8=A6=81?= =?UTF-8?q?=E6=9A=B4=E9=9C=B2=E5=87=BA=E5=8E=BB,=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=B3=A8=E5=86=8C=20gen=20cmd?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- console/console.go | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/console/console.go b/console/console.go index 291dcae..0a1ff2a 100644 --- a/console/console.go +++ b/console/console.go @@ -12,7 +12,7 @@ import ( "os" ) -type sApp struct { +type ConsoleApp struct { name string version string desc string @@ -20,8 +20,8 @@ type sApp struct { about ActionHandler } -func New(name, desc string) *sApp { - app := &sApp{ +func New(name, desc string) *ConsoleApp { + app := &ConsoleApp{ name: name, version: "v0.1.0", desc: desc, @@ -29,45 +29,46 @@ func New(name, desc string) *sApp { } app.AddCmd(NewAboutCmd(app)) app.AddCmd(NewAirCmd(app)) + app.AddCmd(NewGenCmd(app)) app.AddCmd(NewListCmd(app)) app.AddCmd(NewHelpCmd(app)) return app } -func (s *sApp) GetCmds() map[string]ICommand { +func (s *ConsoleApp) GetCmds() map[string]ICommand { return s.cmds } -func (s *sApp) GetName() string { +func (s *ConsoleApp) GetName() string { return s.name } -func (s *sApp) GetDesc() string { +func (s *ConsoleApp) GetDesc() string { return s.desc } -func (s *sApp) GetVersion() string { +func (s *ConsoleApp) GetVersion() string { return s.version } -func (s *sApp) AddCmd(cmd ICommand) { +func (s *ConsoleApp) AddCmd(cmd ICommand) { s.cmds[cmd.GetName()] = cmd } -func (s *sApp) HasCmd(cmd string) bool { +func (s *ConsoleApp) HasCmd(cmd string) bool { _, ok := s.cmds[cmd] return ok } -func (s *sApp) SetExtraAbout(about ActionHandler) { +func (s *ConsoleApp) SetExtraAbout(about ActionHandler) { s.about = about } -func (s *sApp) GetExtraAbout() ActionHandler { +func (s *ConsoleApp) GetExtraAbout() ActionHandler { return s.about } -func (s *sApp) Run(args []string) { +func (s *ConsoleApp) Run(args []string) { cmd := "list" if len(args) == 1 { args = []string{cmd}