package hprose import ( "github.com/beego/bee/logger/colors" "os" "runtime" "fmt" "path" "strings" "github.com/beego/bee/cmd/commands" "github.com/beego/bee/cmd/commands/api" "github.com/beego/bee/cmd/commands/version" "github.com/beego/bee/generate" beeLogger "github.com/beego/bee/logger" "github.com/beego/bee/utils" ) var CmdHproseapp = &commands.Command{ // CustomFlags: true, UsageLine: "hprose [appname]", Short: "Creates an RPC application based on Hprose and Beego frameworks", Long: ` The command 'hprose' creates an RPC application based on both Beego and Hprose (http://hprose.com/). {{"To scaffold out your application, use:"|bold}} $ bee hprose [appname] [-tables=""] [-driver=mysql] [-conn="root:@tcp(127.0.0.1:3306)/test"] [-module=true] [-beego=v1.12.1] If 'conn' is empty, the command will generate a sample application. Otherwise the command will connect to your database and generate models based on the existing tables. The command 'hprose' creates a folder named [appname] with the following structure: ├── main.go ├── go.mod ├── {{"conf"|foldername}} │ └── app.conf └── {{"models"|foldername}} └── object.go └── user.go `, PreRun: func(cmd *commands.Command, args []string) { version.ShowShortVersionBanner() }, Run: createhprose, } var goMod = ` module %s go %s require github.com/astaxie/beego %s require github.com/smartystreets/goconvey v1.6.4 ` var module utils.DocValue var beegoVersion utils.DocValue func init() { CmdHproseapp.Flag.Var(&generate.Tables, "tables", "List of table names separated by a comma.") CmdHproseapp.Flag.Var(&generate.SQLDriver, "driver", "Database driver. Either mysql, postgres or sqlite.") CmdHproseapp.Flag.Var(&generate.SQLConn, "conn", "Connection string used by the driver to connect to a database instance.") CmdHproseapp.Flag.Var(&module, "module", "Support go modules") CmdHproseapp.Flag.Var(&beegoVersion, "beego", "set beego version,only take effect by -module=true") commands.AvailableCommands = append(commands.AvailableCommands, CmdHproseapp) } func createhprose(cmd *commands.Command, args []string) int { output := cmd.Out() if len(args) == 0 { beeLogger.Log.Fatal("Argument [appname] is missing") } curpath, _ := os.Getwd() if len(args) > 1 { cmd.Flag.Parse(args[1:]) } else { module = "false" } var apppath string var packpath string var err error if module != `true` { beeLogger.Log.Info("generate api project support GOPATH") version.ShowShortVersionBanner() apppath, packpath, err = utils.CheckEnv(args[0]) if err != nil { beeLogger.Log.Fatalf("%s", err) } } else { beeLogger.Log.Info("generate api project support go modules.") apppath = path.Join(utils.GetBeeWorkPath(), args[0]) packpath = args[0] if beegoVersion.String() == `` { beegoVersion.Set(`v1.12.1`) } } if utils.IsExist(apppath) { beeLogger.Log.Errorf(colors.Bold("Application '%s' already exists"), apppath) beeLogger.Log.Warn(colors.Bold("Do you want to overwrite it? [Yes|No] ")) if !utils.AskForConfirmation() { os.Exit(2) } } if generate.SQLDriver == "" { generate.SQLDriver = "mysql" } beeLogger.Log.Info("Creating Hprose application...") os.MkdirAll(apppath, 0755) if module == `true` { //generate first for calc model name fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "go.mod"), "\x1b[0m") utils.WriteToFile(path.Join(apppath, "go.mod"), fmt.Sprintf(goMod, packpath, runtime.Version()[2:], beegoVersion.String())) } fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", apppath, "\x1b[0m") os.Mkdir(path.Join(apppath, "conf"), 0755) fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf"), "\x1b[0m") fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "conf", "app.conf"), "\x1b[0m") utils.WriteToFile(path.Join(apppath, "conf", "app.conf"), strings.Replace(generate.Hproseconf, "{{.Appname}}", args[0], -1)) if generate.SQLConn != "" { beeLogger.Log.Infof("Using '%s' as 'driver'", generate.SQLDriver) beeLogger.Log.Infof("Using '%s' as 'conn'", generate.SQLConn) beeLogger.Log.Infof("Using '%s' as 'tables'", generate.Tables) generate.GenerateHproseAppcode(string(generate.SQLDriver), string(generate.SQLConn), "1", string(generate.Tables), path.Join(curpath, args[0])) fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m") maingoContent := strings.Replace(generate.HproseMainconngo, "{{.Appname}}", packpath, -1) maingoContent = strings.Replace(maingoContent, "{{.DriverName}}", string(generate.SQLDriver), -1) maingoContent = strings.Replace(maingoContent, "{{HproseFunctionList}}", strings.Join(generate.HproseAddFunctions, ""), -1) if generate.SQLDriver == "mysql" { maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1) } else if generate.SQLDriver == "postgres" { maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1) } utils.WriteToFile(path.Join(apppath, "main.go"), strings.Replace( maingoContent, "{{.conn}}", generate.SQLConn.String(), -1, ), ) } else { os.Mkdir(path.Join(apppath, "models"), 0755) fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models"), "\x1b[0m") fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "object.go"), "\x1b[0m") utils.WriteToFile(path.Join(apppath, "models", "object.go"), apiapp.APIModels) fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "models", "user.go"), "\x1b[0m") utils.WriteToFile(path.Join(apppath, "models", "user.go"), apiapp.APIModels2) fmt.Fprintf(output, "\t%s%screate%s\t %s%s\n", "\x1b[32m", "\x1b[1m", "\x1b[21m", path.Join(apppath, "main.go"), "\x1b[0m") utils.WriteToFile(path.Join(apppath, "main.go"), strings.Replace(generate.HproseMaingo, "{{.Appname}}", packpath, -1)) } beeLogger.Log.Success("New Hprose application successfully created!") return 0 }