mirror of
https://github.com/beego/bee.git
synced 2024-10-31 18:50:54 +00:00
161 lines
3.2 KiB
Go
161 lines
3.2 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"html/template"
|
||
|
"io"
|
||
|
"log"
|
||
|
"os"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
type Command struct {
|
||
|
// Run runs the command.
|
||
|
// The args are the arguments after the command name.
|
||
|
Run func(cmd *Command, args []string)
|
||
|
|
||
|
// UsageLine is the one-line usage message.
|
||
|
// The first word in the line is taken to be the command name.
|
||
|
UsageLine string
|
||
|
|
||
|
// Short is the short description shown in the 'go help' output.
|
||
|
Short string
|
||
|
|
||
|
// Long is the long message shown in the 'go help <this-command>' output.
|
||
|
Long string
|
||
|
|
||
|
// Flag is a set of flags specific to this command.
|
||
|
Flag flag.FlagSet
|
||
|
|
||
|
// CustomFlags indicates that the command will do its own
|
||
|
// flag parsing.
|
||
|
CustomFlags bool
|
||
|
}
|
||
|
|
||
|
// Name returns the command's name: the first word in the usage line.
|
||
|
func (c *Command) Name() string {
|
||
|
name := c.UsageLine
|
||
|
i := strings.Index(name, " ")
|
||
|
if i >= 0 {
|
||
|
name = name[:i]
|
||
|
}
|
||
|
return name
|
||
|
}
|
||
|
|
||
|
func (c *Command) Usage() {
|
||
|
fmt.Fprintf(os.Stderr, "usage: %s\n\n", c.UsageLine)
|
||
|
fmt.Fprintf(os.Stderr, "%s\n", strings.TrimSpace(c.Long))
|
||
|
os.Exit(2)
|
||
|
}
|
||
|
|
||
|
// Runnable reports whether the command can be run; otherwise
|
||
|
// it is a documentation pseudo-command such as importpath.
|
||
|
func (c *Command) Runnable() bool {
|
||
|
return c.Run != nil
|
||
|
}
|
||
|
|
||
|
var commands = []*Command{
|
||
|
cmdCreate,
|
||
|
cmdStart,
|
||
|
//cmdReStart,
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
flag.Usage = usage
|
||
|
flag.Parse()
|
||
|
log.SetFlags(0)
|
||
|
|
||
|
args := flag.Args()
|
||
|
if len(args) < 1 {
|
||
|
usage()
|
||
|
}
|
||
|
|
||
|
if args[0] == "help" {
|
||
|
help(args[1:])
|
||
|
return
|
||
|
}
|
||
|
|
||
|
for _, cmd := range commands {
|
||
|
if cmd.Name() == args[0] && cmd.Run != nil {
|
||
|
cmd.Flag.Usage = func() { cmd.Usage() }
|
||
|
if cmd.CustomFlags {
|
||
|
args = args[1:]
|
||
|
} else {
|
||
|
cmd.Flag.Parse(args[1:])
|
||
|
args = cmd.Flag.Args()
|
||
|
}
|
||
|
cmd.Run(cmd, args)
|
||
|
os.Exit(2)
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fmt.Fprintf(os.Stderr, "go: unknown subcommand %q\nRun 'go help' for usage.\n", args[0])
|
||
|
os.Exit(2)
|
||
|
}
|
||
|
|
||
|
var usageTemplate = `Bee is a tool for managing beego framework.
|
||
|
|
||
|
Usage:
|
||
|
|
||
|
bee command [arguments]
|
||
|
|
||
|
The commands are:
|
||
|
{{range .}}{{if .Runnable}}
|
||
|
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
|
||
|
|
||
|
Use "bee help [command]" for more information about a command.
|
||
|
|
||
|
Additional help topics:
|
||
|
{{range .}}{{if not .Runnable}}
|
||
|
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
|
||
|
|
||
|
Use "bee help [topic]" for more information about that topic.
|
||
|
|
||
|
`
|
||
|
|
||
|
var helpTemplate = `{{if .Runnable}}usage: bee {{.UsageLine}}
|
||
|
|
||
|
{{end}}{{.Long | trim}}
|
||
|
`
|
||
|
|
||
|
func usage() {
|
||
|
tmpl(os.Stdout, usageTemplate, commands)
|
||
|
os.Exit(2)
|
||
|
}
|
||
|
|
||
|
func tmpl(w io.Writer, text string, data interface{}) {
|
||
|
t := template.New("top")
|
||
|
t.Funcs(template.FuncMap{"trim": strings.TrimSpace})
|
||
|
template.Must(t.Parse(text))
|
||
|
if err := t.Execute(w, data); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func help(args []string) {
|
||
|
if len(args) == 0 {
|
||
|
usage()
|
||
|
// not exit 2: succeeded at 'go help'.
|
||
|
return
|
||
|
}
|
||
|
if len(args) != 1 {
|
||
|
fmt.Fprintf(os.Stdout, "usage: bee help command\n\nToo many arguments given.\n")
|
||
|
os.Exit(2) // failed at 'bee help'
|
||
|
}
|
||
|
|
||
|
arg := args[0]
|
||
|
|
||
|
for _, cmd := range commands {
|
||
|
if cmd.Name() == arg {
|
||
|
tmpl(os.Stdout, helpTemplate, cmd)
|
||
|
// not exit 2: succeeded at 'go help cmd'.
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fmt.Fprintf(os.Stdout, "Unknown help topic %#q. Run 'bee help'.\n", arg)
|
||
|
os.Exit(2) // failed at 'bee help cmd'
|
||
|
}
|