diff --git a/cmd/bee.go b/cmd/bee.go index 93a75cf..4f72d51 100644 --- a/cmd/bee.go +++ b/cmd/bee.go @@ -29,6 +29,7 @@ import ( _ "github.com/beego/bee/cmd/commands/new" _ "github.com/beego/bee/cmd/commands/pack" _ "github.com/beego/bee/cmd/commands/run" + _ "github.com/beego/bee/cmd/commands/scripts" _ "github.com/beego/bee/cmd/commands/version" "github.com/beego/bee/utils" ) diff --git a/cmd/commands/scripts/scripts.go b/cmd/commands/scripts/scripts.go new file mode 100644 index 0000000..b5576c5 --- /dev/null +++ b/cmd/commands/scripts/scripts.go @@ -0,0 +1,44 @@ +package scripts + +import ( + "os/exec" + + "os" + + "runtime" + + "strings" + + "github.com/beego/bee/cmd/commands" + "github.com/beego/bee/cmd/commands/version" + "github.com/beego/bee/config" + "github.com/beego/bee/logger" +) + +func init() { + for commandName, command := range config.Conf.Scripts { + CmdNew := &commands.Command{ + UsageLine: commandName, + Short: command, + PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() }, + Run: RunScript, + } + commands.AvailableCommands = append(commands.AvailableCommands, CmdNew) + } +} + +func RunScript(cmd *commands.Command, args []string) int { + var c *exec.Cmd + switch runtime.GOOS { + case "darwin", "linux": + c = exec.Command("sh", "-c", cmd.Short+" "+strings.Join(args, " ")) + case "windows": //TODO + } + c.Stdout = os.Stdout + c.Stderr = os.Stderr + err := c.Run() + if err != nil { + beeLogger.Log.Error(err.Error()) + } + return 0 +} diff --git a/config/conf.go b/config/conf.go index 2ed07ae..4277a24 100644 --- a/config/conf.go +++ b/config/conf.go @@ -79,8 +79,9 @@ var Conf struct { Driver string Conn string } - EnableReload bool `json:"enable_reload" yaml:"enable_reload"` - EnableNotification bool `json:"enable_notification" yaml:"enable_notification"` + EnableReload bool `json:"enable_reload" yaml:"enable_reload"` + EnableNotification bool `json:"enable_notification" yaml:"enable_notification"` + Scripts map[string]string `json:"scripts" yaml:"scripts"` } func init() {