76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package controller
|
||
|
||
import (
|
||
"SciencesServer/config"
|
||
"SciencesServer/serve/logger"
|
||
"SciencesServer/utils"
|
||
"fmt"
|
||
"os"
|
||
"path"
|
||
"strings"
|
||
|
||
"github.com/spf13/cobra"
|
||
)
|
||
|
||
var ControllerCommand = &cobra.Command{
|
||
Use: "make:controller",
|
||
Short: "quick make controller",
|
||
Example: "ctl make:controller -f sys_user -t sys_user",
|
||
Version: *config.Version,
|
||
}
|
||
|
||
var (
|
||
file string
|
||
address string
|
||
)
|
||
|
||
func init() {
|
||
ControllerCommand.Flags().StringVarP(&file, "file", "f", "", "The file name.")
|
||
ControllerCommand.Flags().StringVarP(&address, "address", "a", "", "The file controller address")
|
||
ControllerCommand.Run = func(cmd *cobra.Command, args []string) {
|
||
utils.TryCatch(func() {
|
||
if file == "" {
|
||
logger.ErrorF("Filename Not Nil")
|
||
return
|
||
}
|
||
output := "../../app/controller"
|
||
|
||
if address == "" {
|
||
logger.ErrorF("Address Not Nil")
|
||
return
|
||
}
|
||
// 解析文件地址
|
||
address = strings.Replace(address, "/", "", -1)
|
||
addresss := strings.Split(address, "/")
|
||
|
||
file := strings.ToLower(file)
|
||
|
||
dir, err := os.Getwd()
|
||
|
||
output = path.Join(dir, output+"/"+address)
|
||
|
||
if err != nil {
|
||
logger.ErrorF("Make Controller Error：%v", err)
|
||
return
|
||
}
|
||
if err = utils.PrepareOutput(output); err != nil {
|
||
logger.ErrorF("Make Controller Error：%v", err)
|
||
return
|
||
}
|
||
fmt.Println(output)
|
||
|
||
modelFile := &ControllerFile{
|
||
Name: addresss[len(addresss)-1],
|
||
StrutName: utils.ToSnake(file, "_"),
|
||
}
|
||
err = modelFile.Execute(path.Join(output, file+".go"), "controller", ControllerTemplate)
|
||
|
||
if err != nil {
|
||
logger.ErrorF("Make Controller Error：%v", err)
|
||
return
|
||
}
|
||
return
|
||
})
|
||
}
|
||
}
|