yao/cmd/start.go

128 lines
3.6 KiB
Go

package cmd
import (
"fmt"
"os"
"path/filepath"
"strings"
"github.com/fatih/color"
"github.com/spf13/cobra"
"github.com/yaoapp/gou"
"github.com/yaoapp/yao/config"
"github.com/yaoapp/yao/engine"
"github.com/yaoapp/yao/service"
"github.com/yaoapp/yao/share"
)
var startDebug = false
var startAlpha = false
var startCmd = &cobra.Command{
Use: "start",
Short: L("Start Engine"),
Long: L("Start Engine"),
Run: func(cmd *cobra.Command, args []string) {
defer service.Stop(func() { fmt.Println(L("Service stopped")) })
Boot()
if startDebug { // 强制 debug 模式启动
config.Development()
}
mode := config.Conf.Mode
err := engine.Load(config.Conf) // 加载脚本等
if err != nil {
fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
os.Exit(1)
}
port := fmt.Sprintf(":%d", config.Conf.Port)
if port == ":80" {
port = ""
}
host := config.Conf.Host
if host == "0.0.0.0" {
host = "127.0.0.1"
}
if mode == "development" {
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("API List")))
fmt.Println(color.WhiteString("---------------------------------"))
for _, api := range gou.APIs { // API信息
if len(api.HTTP.Paths) <= 0 {
continue
}
fmt.Printf(color.CyanString("\n%s(%d)\n", api.Name, len(api.HTTP.Paths)))
for _, p := range api.HTTP.Paths {
fmt.Println(
colorMehtod(p.Method),
color.WhiteString(filepath.Join("/api", api.HTTP.Group, p.Path)),
"\tprocess:", p.Process)
}
}
}
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(share.App.Name), color.WhiteString(share.App.Version), mode)
fmt.Println(color.WhiteString("---------------------------------"))
if !share.BUILDIN {
root, _ := filepath.Abs(config.Conf.Root)
fmt.Println(color.WhiteString(L("Root")), color.GreenString(" %s", root))
}
fmt.Println(color.WhiteString(L("Frontend")), color.GreenString(" http://%s%s/", host, port))
fmt.Println(color.WhiteString(L("Dashboard")), color.GreenString(" http://%s%s/xiang/login/admin", host, port))
fmt.Println(color.WhiteString(L("API")), color.GreenString(" http://%s%s/api", host, port))
fmt.Println(color.WhiteString(L("SessionPort")), color.GreenString(" %d", share.SessionPort))
fmt.Println(color.WhiteString(L("Listening")), color.GreenString(" %s:%d", config.Conf.Host, config.Conf.Port))
fmt.Println("")
// 调试模式
if mode == "development" {
service.Watch(config.Conf)
}
// with the alpha features
if startAlpha {
for _, sock := range gou.Sockets {
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(sock.Name))
fmt.Println(color.WhiteString("---------------------------------"))
fmt.Println(color.GreenString("Mode: %s", sock.Mode))
fmt.Println(color.GreenString("Host: %s://%s", sock.Protocol, sock.Host))
fmt.Println(color.GreenString("Port: %s\n\n", sock.Port))
if sock.Mode == "server" {
go sock.Start()
} else if sock.Mode == "client" {
go sock.Connect()
}
}
}
fmt.Println(color.GreenString(L("✨LISTENING✨")))
service.Start()
},
}
func colorMehtod(method string) string {
method = strings.ToUpper(method)
switch method {
case "GET":
return color.GreenString("GET")
case "POST":
return color.YellowString("POST")
default:
return color.WhiteString(method)
}
}
func init() {
startCmd.PersistentFlags().BoolVarP(&startDebug, "debug", "", false, L("Development mode"))
startCmd.PersistentFlags().BoolVarP(&startAlpha, "alpha", "", false, L("Enabled unstable features"))
}