golib/gconsole/console.go

//
// cli.go
// Copyright (C) 2022 tiglog <me@tiglog.com>
//
// Distributed under terms of the MIT license.
//

package gconsole

import (
	"fmt"
	"os"
)

type App struct {
	name    string
	version string
	desc    string
	cmds    map[string]ICommand
	about   ActionHandler
}

func New(name, desc string) *App {
	app := &App{
		name:    name,
		version: "v0.1.0",
		desc:    desc,
		cmds:    make(map[string]ICommand),
	}
	app.AddCmd(NewAboutCmd(app))
	app.AddCmd(NewAirCmd(app))
	app.AddCmd(NewGenCmd(app))
	app.AddCmd(NewListCmd(app))
	app.AddCmd(NewHelpCmd(app))
	return app
}

func (s *App) GetCmds() map[string]ICommand {
	return s.cmds
}

func (s *App) GetName() string {
	return s.name
}

func (s *App) GetDesc() string {
	return s.desc
}

func (s *App) GetVersion() string {
	return s.version
}

func (s *App) AddCmd(cmd ICommand) {
	s.cmds[cmd.GetName()] = cmd
}

func (s *App) HasCmd(cmd string) bool {
	_, ok := s.cmds[cmd]
	return ok
}

func (s *App) SetExtraAbout(about ActionHandler) {
	s.about = about
}

func (s *App) GetExtraAbout() ActionHandler {
	return s.about
}

func (s *App) 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)
	}
}

type ICommand interface {
	Init(args []string)
	GetName() string
	GetDesc() string
	GetAction() ActionHandler
	GetHelp() string
}

type IConsole interface {
	GetCmds() map[string]ICommand
	GetName() string
	GetDesc() string
	GetVersion() string
	GetExtraAbout() ActionHandler
}
type ActionHandler func() error