411 lines
11 KiB
Go
411 lines
11 KiB
Go
|
package tools
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"fmt"
|
|||
|
|
"go-admin/app/admin/service"
|
|||
|
|
"go-admin/app/admin/service/dto"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"text/template"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/api"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/config"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/pkg"
|
|||
|
|
|
|||
|
|
"go-admin/app/other/models/tools"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Gen struct {
|
|||
|
|
api.Api
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) Preview(c *gin.Context) {
|
|||
|
|
e.Context = c
|
|||
|
|
log := e.GetLogger()
|
|||
|
|
table := tools.SysTables{}
|
|||
|
|
id, err := pkg.StringToInt(c.Param("tableId"))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("tableId接收失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
table.TableId = id
|
|||
|
|
t1, err := template.ParseFiles("template/v4/model.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("model模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t2, err := template.ParseFiles("template/v4/no_actions/apis.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("api模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t3, err := template.ParseFiles("template/v4/js.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("js模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t4, err := template.ParseFiles("template/v4/vue.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("vue模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t5, err := template.ParseFiles("template/v4/no_actions/router_check_role.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("路由模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t6, err := template.ParseFiles("template/v4/dto.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("dto模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t7, err := template.ParseFiles("template/v4/no_actions/service.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("service模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db, err := pkg.GetOrm(c)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Errorf("get db connection error, %s", err.Error())
|
|||
|
|
e.Error(500, err, fmt.Sprintf("数据库链接获取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tab, _ := table.Get(db,false)
|
|||
|
|
var b1 bytes.Buffer
|
|||
|
|
err = t1.Execute(&b1, tab)
|
|||
|
|
var b2 bytes.Buffer
|
|||
|
|
err = t2.Execute(&b2, tab)
|
|||
|
|
var b3 bytes.Buffer
|
|||
|
|
err = t3.Execute(&b3, tab)
|
|||
|
|
var b4 bytes.Buffer
|
|||
|
|
err = t4.Execute(&b4, tab)
|
|||
|
|
var b5 bytes.Buffer
|
|||
|
|
err = t5.Execute(&b5, tab)
|
|||
|
|
var b6 bytes.Buffer
|
|||
|
|
err = t6.Execute(&b6, tab)
|
|||
|
|
var b7 bytes.Buffer
|
|||
|
|
err = t7.Execute(&b7, tab)
|
|||
|
|
|
|||
|
|
mp := make(map[string]interface{})
|
|||
|
|
mp["template/model.go.template"] = b1.String()
|
|||
|
|
mp["template/api.go.template"] = b2.String()
|
|||
|
|
mp["template/js.go.template"] = b3.String()
|
|||
|
|
mp["template/vue.go.template"] = b4.String()
|
|||
|
|
mp["template/router.go.template"] = b5.String()
|
|||
|
|
mp["template/dto.go.template"] = b6.String()
|
|||
|
|
mp["template/service.go.template"] = b7.String()
|
|||
|
|
e.OK(mp, "")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) GenCode(c *gin.Context) {
|
|||
|
|
e.Context = c
|
|||
|
|
log := e.GetLogger()
|
|||
|
|
table := tools.SysTables{}
|
|||
|
|
id, err := pkg.StringToInt(c.Param("tableId"))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("tableId参数接收失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db, err := pkg.GetOrm(c)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Errorf("get db connection error, %s", err.Error())
|
|||
|
|
e.Error(500, err, fmt.Sprintf("数据库链接获取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
table.TableId = id
|
|||
|
|
tab, _ := table.Get(db,false)
|
|||
|
|
|
|||
|
|
e.NOActionsGen(c, tab)
|
|||
|
|
|
|||
|
|
e.OK("", "Code generated successfully！")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) GenApiToFile(c *gin.Context) {
|
|||
|
|
e.Context = c
|
|||
|
|
log := e.GetLogger()
|
|||
|
|
table := tools.SysTables{}
|
|||
|
|
id, err := pkg.StringToInt(c.Param("tableId"))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("tableId参数获取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db, err := pkg.GetOrm(c)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Errorf("get db connection error, %s", err.Error())
|
|||
|
|
e.Error(500, err, fmt.Sprintf("数据库链接获取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
table.TableId = id
|
|||
|
|
tab, _ := table.Get(db,false)
|
|||
|
|
e.genApiToFile(c, tab)
|
|||
|
|
|
|||
|
|
e.OK("", "Code generated successfully！")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) NOActionsGen(c *gin.Context, tab tools.SysTables) {
|
|||
|
|
e.Context = c
|
|||
|
|
log := e.GetLogger()
|
|||
|
|
tab.MLTBName = strings.Replace(tab.TBName, "_", "-", -1)
|
|||
|
|
|
|||
|
|
basePath := "template/v4/"
|
|||
|
|
routerFile := basePath + "no_actions/router_check_role.go.template"
|
|||
|
|
|
|||
|
|
if tab.IsAuth == 2 {
|
|||
|
|
routerFile = basePath + "no_actions/router_no_check_role.go.template"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t1, err := template.ParseFiles(basePath + "model.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("model模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t2, err := template.ParseFiles(basePath + "no_actions/apis.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("api模版读取失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t3, err := template.ParseFiles(routerFile)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("路由模版失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t4, err := template.ParseFiles(basePath + "js.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("js模版解析失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t5, err := template.ParseFiles(basePath + "vue.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("vue模版解析失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t6, err := template.ParseFiles(basePath + "dto.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("dto模版解析失败失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
t7, err := template.ParseFiles(basePath + "no_actions/service.go.template")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("service模版失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_ = pkg.PathCreate("./app/" + tab.PackageName + "/apis/")
|
|||
|
|
_ = pkg.PathCreate("./app/" + tab.PackageName + "/models/")
|
|||
|
|
_ = pkg.PathCreate("./app/" + tab.PackageName + "/router/")
|
|||
|
|
_ = pkg.PathCreate("./app/" + tab.PackageName + "/service/dto/")
|
|||
|
|
_ = pkg.PathCreate(config.GenConfig.FrontPath + "/api/" + tab.PackageName + "/")
|
|||
|
|
err = pkg.PathCreate(config.GenConfig.FrontPath + "/views/" + tab.PackageName + "/" + tab.MLTBName + "/")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("views目录创建失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var b1 bytes.Buffer
|
|||
|
|
err = t1.Execute(&b1, tab)
|
|||
|
|
var b2 bytes.Buffer
|
|||
|
|
err = t2.Execute(&b2, tab)
|
|||
|
|
var b3 bytes.Buffer
|
|||
|
|
err = t3.Execute(&b3, tab)
|
|||
|
|
var b4 bytes.Buffer
|
|||
|
|
err = t4.Execute(&b4, tab)
|
|||
|
|
var b5 bytes.Buffer
|
|||
|
|
err = t5.Execute(&b5, tab)
|
|||
|
|
var b6 bytes.Buffer
|
|||
|
|
err = t6.Execute(&b6, tab)
|
|||
|
|
var b7 bytes.Buffer
|
|||
|
|
err = t7.Execute(&b7, tab)
|
|||
|
|
pkg.FileCreate(b1, "./app/"+tab.PackageName+"/models/"+tab.TBName+".go")
|
|||
|
|
pkg.FileCreate(b2, "./app/"+tab.PackageName+"/apis/"+tab.TBName+".go")
|
|||
|
|
pkg.FileCreate(b3, "./app/"+tab.PackageName+"/router/"+tab.TBName+".go")
|
|||
|
|
pkg.FileCreate(b4, config.GenConfig.FrontPath+"/api/"+tab.PackageName+"/"+tab.MLTBName+".js")
|
|||
|
|
pkg.FileCreate(b5, config.GenConfig.FrontPath+"/views/"+tab.PackageName+"/"+tab.MLTBName+"/index.vue")
|
|||
|
|
pkg.FileCreate(b6, "./app/"+tab.PackageName+"/service/dto/"+tab.TBName+".go")
|
|||
|
|
pkg.FileCreate(b7, "./app/"+tab.PackageName+"/service/"+tab.TBName+".go")
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) genApiToFile(c *gin.Context, tab tools.SysTables) {
|
|||
|
|
err := e.MakeContext(c).
|
|||
|
|
MakeOrm().
|
|||
|
|
Errors
|
|||
|
|
if err != nil {
|
|||
|
|
e.Logger.Error(err)
|
|||
|
|
e.Error(500, err, err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := "template/"
|
|||
|
|
|
|||
|
|
t1, err := template.ParseFiles(basePath + "api_migrate.template")
|
|||
|
|
if err != nil {
|
|||
|
|
e.Logger.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("数据迁移模版解析失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
i := strconv.FormatInt(time.Now().UnixNano()/1e6, 10)
|
|||
|
|
var b1 bytes.Buffer
|
|||
|
|
err = t1.Execute(&b1, struct {
|
|||
|
|
tools.SysTables
|
|||
|
|
GenerateTime string
|
|||
|
|
}{tab, i})
|
|||
|
|
|
|||
|
|
pkg.FileCreate(b1, "./cmd/migrate/migration/version-local/"+i+"_migrate.go")
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e Gen) GenMenuAndApi(c *gin.Context) {
|
|||
|
|
s := service.SysMenu{}
|
|||
|
|
err := e.MakeContext(c).
|
|||
|
|
MakeOrm().
|
|||
|
|
MakeService(&s.Service).
|
|||
|
|
Errors
|
|||
|
|
if err != nil {
|
|||
|
|
e.Logger.Error(err)
|
|||
|
|
e.Error(500, err, err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
table := tools.SysTables{}
|
|||
|
|
id, err := pkg.StringToInt(c.Param("tableId"))
|
|||
|
|
if err != nil {
|
|||
|
|
e.Logger.Error(err)
|
|||
|
|
e.Error(500, err, fmt.Sprintf("tableId参数解析失败！错误详情：%s", err.Error()))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
table.TableId = id
|
|||
|
|
tab, _ := table.Get(e.Orm,true)
|
|||
|
|
tab.MLTBName = strings.Replace(tab.TBName, "_", "-", -1)
|
|||
|
|
|
|||
|
|
Mmenu := dto.SysMenuInsertReq{}
|
|||
|
|
Mmenu.Title = tab.TableComment
|
|||
|
|
Mmenu.Icon = "pass"
|
|||
|
|
Mmenu.Path = "/" + tab.MLTBName
|
|||
|
|
Mmenu.MenuType = "M"
|
|||
|
|
Mmenu.Action = "无"
|
|||
|
|
Mmenu.ParentId = 0
|
|||
|
|
Mmenu.NoCache = false
|
|||
|
|
Mmenu.Component = "Layout"
|
|||
|
|
Mmenu.Sort = 0
|
|||
|
|
Mmenu.Visible = "0"
|
|||
|
|
Mmenu.IsFrame = "0"
|
|||
|
|
Mmenu.CreateBy = 1
|
|||
|
|
s.Insert(&Mmenu)
|
|||
|
|
|
|||
|
|
Cmenu := dto.SysMenuInsertReq{}
|
|||
|
|
Cmenu.MenuName = tab.ClassName + "Manage"
|
|||
|
|
Cmenu.Title = tab.TableComment
|
|||
|
|
Cmenu.Icon = "pass"
|
|||
|
|
Cmenu.Path = "/" + tab.PackageName + "/" + tab.MLTBName
|
|||
|
|
Cmenu.MenuType = "C"
|
|||
|
|
Cmenu.Action = "无"
|
|||
|
|
Cmenu.Permission = tab.PackageName + ":" + tab.BusinessName + ":list"
|
|||
|
|
Cmenu.ParentId = Mmenu.MenuId
|
|||
|
|
Cmenu.NoCache = false
|
|||
|
|
Cmenu.Component = "/" + tab.PackageName + "/" + tab.MLTBName + "/index"
|
|||
|
|
Cmenu.Sort = 0
|
|||
|
|
Cmenu.Visible = "0"
|
|||
|
|
Cmenu.IsFrame = "0"
|
|||
|
|
Cmenu.CreateBy = 1
|
|||
|
|
Cmenu.UpdateBy = 1
|
|||
|
|
s.Insert(&Cmenu)
|
|||
|
|
|
|||
|
|
MList := dto.SysMenuInsertReq{}
|
|||
|
|
MList.MenuName = ""
|
|||
|
|
MList.Title = "分页获取" + tab.TableComment
|
|||
|
|
MList.Icon = ""
|
|||
|
|
MList.Path = tab.TBName
|
|||
|
|
MList.MenuType = "F"
|
|||
|
|
MList.Action = "无"
|
|||
|
|
MList.Permission = tab.PackageName + ":" + tab.BusinessName + ":query"
|
|||
|
|
MList.ParentId = Cmenu.MenuId
|
|||
|
|
MList.NoCache = false
|
|||
|
|
MList.Sort = 0
|
|||
|
|
MList.Visible = "0"
|
|||
|
|
MList.IsFrame = "0"
|
|||
|
|
MList.CreateBy = 1
|
|||
|
|
MList.UpdateBy = 1
|
|||
|
|
s.Insert(&MList)
|
|||
|
|
|
|||
|
|
MCreate := dto.SysMenuInsertReq{}
|
|||
|
|
MCreate.MenuName = ""
|
|||
|
|
MCreate.Title = "创建" + tab.TableComment
|
|||
|
|
MCreate.Icon = ""
|
|||
|
|
MCreate.Path = tab.TBName
|
|||
|
|
MCreate.MenuType = "F"
|
|||
|
|
MCreate.Action = "无"
|
|||
|
|
MCreate.Permission = tab.PackageName + ":" + tab.BusinessName + ":add"
|
|||
|
|
MCreate.ParentId = Cmenu.MenuId
|
|||
|
|
MCreate.NoCache = false
|
|||
|
|
MCreate.Sort = 0
|
|||
|
|
MCreate.Visible = "0"
|
|||
|
|
MCreate.IsFrame = "0"
|
|||
|
|
MCreate.CreateBy = 1
|
|||
|
|
MCreate.UpdateBy = 1
|
|||
|
|
s.Insert(&MCreate)
|
|||
|
|
|
|||
|
|
MUpdate := dto.SysMenuInsertReq{}
|
|||
|
|
MUpdate.MenuName = ""
|
|||
|
|
MUpdate.Title = "修改" + tab.TableComment
|
|||
|
|
MUpdate.Icon = ""
|
|||
|
|
MUpdate.Path = tab.TBName
|
|||
|
|
MUpdate.MenuType = "F"
|
|||
|
|
MUpdate.Action = "无"
|
|||
|
|
MUpdate.Permission = tab.PackageName + ":" + tab.BusinessName + ":edit"
|
|||
|
|
MUpdate.ParentId = Cmenu.MenuId
|
|||
|
|
MUpdate.NoCache = false
|
|||
|
|
MUpdate.Sort = 0
|
|||
|
|
MUpdate.Visible = "0"
|
|||
|
|
MUpdate.IsFrame = "0"
|
|||
|
|
MUpdate.CreateBy = 1
|
|||
|
|
MUpdate.UpdateBy = 1
|
|||
|
|
s.Insert(&MUpdate)
|
|||
|
|
|
|||
|
|
MDelete := dto.SysMenuInsertReq{}
|
|||
|
|
MDelete.MenuName = ""
|
|||
|
|
MDelete.Title = "删除" + tab.TableComment
|
|||
|
|
MDelete.Icon = ""
|
|||
|
|
MDelete.Path = tab.TBName
|
|||
|
|
MDelete.MenuType = "F"
|
|||
|
|
MDelete.Action = "无"
|
|||
|
|
MDelete.Permission = tab.PackageName + ":" + tab.BusinessName + ":remove"
|
|||
|
|
MDelete.ParentId = Cmenu.MenuId
|
|||
|
|
MDelete.NoCache = false
|
|||
|
|
MDelete.Sort = 0
|
|||
|
|
MDelete.Visible = "0"
|
|||
|
|
MDelete.IsFrame = "0"
|
|||
|
|
MDelete.CreateBy = 1
|
|||
|
|
MDelete.UpdateBy = 1
|
|||
|
|
s.Insert(&MDelete)
|
|||
|
|
|
|||
|
|
e.OK("", "数据生成成功！")
|
|||
|
|
}
|