feature:自动创建基本菜单

Author: piexlMax

server/api/v1/system/sys_auto_code.go

@@ -63,6 +63,7 @@ func (autoApi *AutoCodeApi) CreateTemp(c *gin.Context) {
 }
 a.Pretreatment()
 var apiIds []uint
+var menuId uint
 if a.AutoCreateApiToSql {
 if ids, err := autoCodeService.AutoCreateApi(&a); err != nil {
 global.GVA_LOG.Error("自动化创建失败!请自行清空垃圾数据!", zap.Error(err))
@@ -73,8 +74,17 @@ func (autoApi *AutoCodeApi) CreateTemp(c *gin.Context) {
 apiIds = ids
 }
 }
+if a.AutoCreateApiToSql {
+if id, err := autoCodeService.AutoCreateMenu(&a); err != nil {
+global.GVA_LOG.Error("自动化创建失败!请自行清空垃圾数据!", zap.Error(err))
+c.Writer.Header().Add("success", "false")
+c.Writer.Header().Add("msg", url.QueryEscape("自动化创建失败!请自行清空垃圾数据!"))
+} else {
+menuId = id
+}
+}
 a.PackageT = utils.FirstUpper(a.Package)
-err := autoCodeService.CreateTemp(a, apiIds...)
+err := autoCodeService.CreateTemp(a, menuId, apiIds...)
 if err != nil {
 if errors.Is(err, system.ErrAutoMove) {
 c.Writer.Header().Add("success", "true")

server/model/system/sys_auto_code.go

@@ -10,29 +10,30 @@ import (
 
 // AutoCodeStruct 初始版本自动化代码工具
 type AutoCodeStruct struct {
-StructName string `json:"structName"` // Struct名称
-TableName string `json:"tableName"` // 表名
-PackageName string `json:"packageName"` // 文件名称
-HumpPackageName string `json:"humpPackageName"` // go文件名称
-Abbreviation string `json:"abbreviation"` // Struct简称
-Description string `json:"description"` // Struct中文名称
-AutoCreateApiToSql bool `json:"autoCreateApiToSql"` // 是否自动创建api
-AutoCreateResource bool `json:"autoCreateResource"` // 是否自动创建资源标识
-AutoMoveFile bool `json:"autoMoveFile"` // 是否自动移动文件
-BusinessDB string `json:"businessDB"` // 业务数据库
-GvaModel bool `json:"gvaModel"` // 是否使用gva默认Model
-Fields []*Field `json:"fields"`
-PrimaryField *Field `json:"primaryField"`
-HasTimer bool `json:"-"`
-HasSearchTimer bool `json:"-"`
-DictTypes []string `json:"-"`
-Package string `json:"package"`
-PackageT string `json:"-"`
-NeedSort bool `json:"-"`
-HasPic bool `json:"-"`
-HasRichText bool `json:"-"`
-HasFile bool `json:"-"`
-NeedJSON bool `json:"-"`
+StructName string `json:"structName"` // Struct名称
+TableName string `json:"tableName"` // 表名
+PackageName string `json:"packageName"` // 文件名称
+HumpPackageName string `json:"humpPackageName"` // go文件名称
+Abbreviation string `json:"abbreviation"` // Struct简称
+Description string `json:"description"` // Struct中文名称
+AutoCreateApiToSql bool `json:"autoCreateApiToSql"` // 是否自动创建api
+AutoCreateMenuToSql bool `json:"autoCreateMenuToSql"` // 是否自动创建menu
+AutoCreateResource bool `json:"autoCreateResource"` // 是否自动创建资源标识
+AutoMoveFile bool `json:"autoMoveFile"` // 是否自动移动文件
+BusinessDB string `json:"businessDB"` // 业务数据库
+GvaModel bool `json:"gvaModel"` // 是否使用gva默认Model
+Fields []*Field `json:"fields"`
+PrimaryField *Field `json:"primaryField"`
+HasTimer bool `json:"-"`
+HasSearchTimer bool `json:"-"`
+DictTypes []string `json:"-"`
+Package string `json:"package"`
+PackageT string `json:"-"`
+NeedSort bool `json:"-"`
+HasPic bool `json:"-"`
+HasRichText bool `json:"-"`
+HasFile bool `json:"-"`
+NeedJSON bool `json:"-"`
 }
 
 func (a *AutoCodeStruct) Pretreatment() {

server/model/system/sys_autocode_history.go

@@ -14,6 +14,7 @@ type SysAutoCodeHistory struct {
 Package string `json:"package"`
 BusinessDB string `json:"businessDB"`
 TableName string `json:"tableName"`
+MenuID uint `json:"menuID"`
 RequestMeta string `gorm:"type:text" json:"requestMeta,omitempty"` // 前端传入的结构化信息
 AutoCodePath string `gorm:"type:text" json:"autoCodePath,omitempty"` // 其他meta信息 path;path
 InjectionMeta string `gorm:"type:text" json:"injectionMeta,omitempty"` // 注入的内容 RouterPath@functionName@RouterString;

server/service/system/sys_auto_code.go

@@ -253,7 +253,7 @@ func makeDictTypes(autoCode *system.AutoCodeStruct) {
 // @param: model.AutoCodeStruct
 // @return: err error
 
-func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruct, ids ...uint) (err error) {
+func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruct, menuID uint, ids ...uint) (err error) {
 makeDictTypes(&autoCode)
 for i := range autoCode.Fields {
 if autoCode.Fields[i].FieldType == "time.Time" {
@@ -401,6 +401,7 @@ func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruc
 idBf.String(),
 autoCode.Package,
 autoCode.BusinessDB,
+menuID,
 )
 } else {
 err = AutoCodeHistoryServiceApp.CreateAutoCodeHistory(
@@ -413,6 +414,7 @@ func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruc
 idBf.String(),
 autoCode.Package,
 autoCode.BusinessDB,
+menuID,
 )
 }
 }
@@ -569,6 +571,20 @@ func (autoCodeService *AutoCodeService) AutoCreateApi(a *system.AutoCodeStruct)
 return ids, err
 }
 
+func (autoCodeService *AutoCodeService) AutoCreateMenu(a *system.AutoCodeStruct) (id uint, err error) {
+var menu system.SysBaseMenu
+err = global.GVA_DB.First(&menu, "name = ?", menu.Name).Error
+if err == nil {
+return 0, errors.New("存在相同的菜单路由，请关闭自动创建菜单功能")
+}
+menu.Name = a.Abbreviation
+menu.Path = a.Abbreviation
+menu.Meta.Title = a.Description
+menu.Component = fmt.Sprintf("view/%s/%s.vue", a.PackageName, a.PackageName)
+err = global.GVA_DB.Create(&menu).Error
+return menu.ID, err
+}
+
 func (autoCodeService *AutoCodeService) getNeedList(autoCode *system.AutoCodeStruct) (dataList []tplData, fileList []string, needMkdir []string, err error) {
 // 去除所有空格
 utils.TrimSpace(autoCode)

server/service/system/sys_autocode_history.go

@@ -30,7 +30,7 @@ var AutoCodeHistoryServiceApp = new(AutoCodeHistoryService)
 // RouterPath : RouterPath@RouterString;RouterPath2@RouterString2
 // Author [SliverHorn](https://github.com/SliverHorn)
 // Author [songzhibin97](https://github.com/songzhibin97)
-func (autoCodeHistoryService *AutoCodeHistoryService) CreateAutoCodeHistory(meta, structName, structCNName, autoCodePath string, injectionMeta string, tableName string, apiIds string, Package string, BusinessDB string) error {
+func (autoCodeHistoryService *AutoCodeHistoryService) CreateAutoCodeHistory(meta, structName, structCNName, autoCodePath string, injectionMeta string, tableName string, apiIds string, Package string, BusinessDB string, menuID uint) error {
 return global.GVA_DB.Create(&system.SysAutoCodeHistory{
 Package: Package,
 RequestMeta: meta,
@@ -81,9 +81,17 @@ func (autoCodeHistoryService *AutoCodeHistoryService) RollBack(info *systemReq.R
 ids.Ids = append(ids.Ids, id)
 }
 err := ApiServiceApp.DeleteApisByIds(ids)
+
 if err != nil {
 global.GVA_LOG.Error("ClearTag DeleteApiByIds:", zap.Error(err))
 }
+
+err = BaseMenuServiceApp.DeleteBaseMenu(int(md.MenuID))
+
+if err != nil {
+global.GVA_LOG.Error("ClearTag DeleteBaseMenu:", zap.Error(err))
+}
+
 // 删除表
 if info.DeleteTable {
 if err = AutoCodeServiceApp.DropTable(md.BusinessDB, md.TableName); err != nil {

server/service/system/sys_base_menu.go

@@ -16,6 +16,8 @@ type BaseMenuService struct{}
 //@param: id float64
 //@return: err error
 
+var BaseMenuServiceApp = new(BaseMenuService)
+
 func (baseMenuService *BaseMenuService) DeleteBaseMenu(id int) (err error) {
 err = global.GVA_DB.Preload("MenuBtn").Preload("Parameters").Where("parent_id = ?", id).First(&system.SysBaseMenu{}).Error
 if err != nil {