magic_admin/model/system/sys_apis.go

package system

import (
	"go_server/model/common"
	"gorm.io/gorm"
	"strings"
)

// generateApiName 根据路径自动生成有意义的中文接口名称
// 路径格式: /admin/api/{group}/{module}/{action}
func generateApiName(path string) string {
	groupMap := map[string]string{
		"sys":     "系统",
		"app":     "应用",
		"daytask": "日任务",
		"login":   "登录",
	}

	moduleMap := map[string]string{
		// sys 模块
		"user":       "管理员",
		"role":       "角色",
		"menu":       "菜单",
		"apis":       "API权限",
		"dictionary": "字典",
		"file":       "文件",
		"sign":       "签名",
		"auto":       "代码生成",
		"db":         "数据库",
		// app 模块
		"magic_asset":                        "资产",
		"magic_asset_bill":                   "资产账单",
		"magic_asset_rw_callback_log":         "充提回调日志",
		"magic_asset_rw_record":              "充提记录",
		"magic_stake_product":                "质押产品",
		"magic_stake_queue_info":             "质押队列",
		"magic_stake_user_current_ops_record": "质押操作记录",
		"magic_stake_user_current_order":     "质押订单",
		"magic_user":                         "用户",
		"magic_user_action_log":              "用户行为日志",
		"magic_user_logs":                    "用户日志",
		"magic_user_profit":                  "用户收益",
		"magic_user_profit_record":           "收益记录",
		"magic_user_quota":                   "用户配额",
		"node_banner":                        "节点Banner",
		"node_info":                          "节点信息",
		"node_order":                         "节点订单",
		"node_order_payments":                "节点支付",
		"promotion_upgrade_level":            "晋级推广",
		"sys_coin":                           "币种配置",
		"sys_i18n":                           "国际化",
		"sys_job":                            "定时任务",
		"sys_level_config":                   "等级配置",
		"sys_sign_config":                    "签到配置",
		"sys_stake_period_job":               "质押周期",
		"magic_extra_stake_order":            "额外质押订单",
		"magic_extra_stake_order_ops_record": "额外质押操作",
		// daytask 模块
		"user_social":      "社交账号",
		"user_payment":     "收款方式",
		"user_level":       "用户等级",
		"recharge_order":   "充值订单",
		"payment_channel":  "支付渠道",
		"task_category":    "任务分类",
		"task":             "任务",
		"task_step":        "任务步骤",
		"task_example":     "审核样例",
		"user_task":        "用户任务",
		"withdraw_order":   "提现订单",
		"balance_log":      "资金流水",
		"notice":           "公告",
		"banner":           "Banner",
		"help_category":    "帮助分类",
		"help":             "帮助中心",
		"customer_service": "客服配置",
		"material":         "素材",
		"material_group":   "素材分组",
		"music_group":      "歌单分组",
		"music":            "歌曲",
		"admin":            "管理员",
		"config":           "系统配置",
		"operation_log":    "操作日志",
		"daily_stats":      "每日统计",
	}

	actionMap := map[string]string{
		"get":              "获取详情",
		"find":             "查询列表",
		"create":           "创建",
		"update":           "修改",
		"delete":           "删除",
		"set":              "修改配置",
		"setUser":          "修改用户",
		"tree":             "树形结构",
		"info":             "个人信息",
		"logs":             "操作日志",
		"audit":            "审核",
		"pay":              "支付",
		"comment":          "评论",
		"cancel":           "取消",
		"dbs":              "数据库列表",
		"tbs":              "表列表",
		"cols":             "字段列表",
		"upload":           "上传文件",
		"oss/auth":         "OSS授权",
		"server/code":      "生成代码",
		"server/rollback":  "回滚代码",
		"model/auto":       "同步模型",
		"getGoogleKey":     "获取Google密钥",
		"cancelGoogleKey":  "取消Google密钥",
		"replaceGoogleKey": "重置Google密钥",
		"generateCaptcha":  "生成验证码",
		"check":            "验证登录",
		"in":               "登录",
		"wallet/info":      "钱包信息",
		"get_by_group":     "按分组查询",
		"summary":          "数据概览",
	}

	// 路径格式: /admin/api/{group}/{module}/{action}
	// Trim("/") 后 Split 得到: ["admin", "api", "group", "module", "action"]
	// 索引:                         0        1       2        3        4
	parts := strings.Split(strings.Trim(path, "/"), "/")

	groupName := ""
	moduleName := ""
	actionName := ""

	if len(parts) >= 3 {
		groupName = groupMap[parts[2]]
	}
	if len(parts) >= 4 {
		moduleName = moduleMap[parts[3]]
		if moduleName == "" {
			moduleName = parts[3]
		}
	}
	if len(parts) >= 5 {
		// 优先尝试双段 action，如 oss/auth、server/code
		if len(parts) >= 6 {
			doubleKey := parts[4] + "/" + parts[5]
			if v, ok := actionMap[doubleKey]; ok {
				actionName = v
			}
		}
		if actionName == "" {
			actionName = actionMap[parts[4]]
		}
		if actionName == "" {
			actionName = parts[4]
		}
	}

	if groupName != "" && moduleName != "" && actionName != "" {
		return groupName + "-" + moduleName + "-" + actionName
	}
	if moduleName != "" && actionName != "" {
		return moduleName + "-" + actionName
	}
	return "待设置"
}

// BatchUpdateApiNames 批量将 name="待设置" 的接口更新为有意义的中文名称
func (s *Apis) BatchUpdateApiNames(db *gorm.DB) error {
	var apis []Apis
	if err := db.Where("name = ?", "待设置").Find(&apis).Error; err != nil {
		return err
	}
	for _, api := range apis {
		name := generateApiName(api.Path)
		if name == "待设置" {
			continue
		}
		if err := db.Exec(
			"UPDATE "+s.TableName()+" SET name = ? WHERE id = ? AND deleted_at IS NULL",
			name, api.ID,
		).Error; err != nil {
			return err
		}
	}
	return nil
}

type Apis struct {
	common.GormFullModel
	ParentId int64  `json:"parentId" gorm:"column:parent_id;comment:上级接口Id"`
	Group    string `json:"group" gorm:"column:group;type:varchar(50);comment:分组名称"`
	Name     string `json:"name" gorm:"column:name;type:varchar(50);comment:接口名称"`
	Method   string `json:"method" gorm:"column:method;type:varchar(20);comment:请求方法"`
	Path     string `json:"path" gorm:"column:path;type:varchar(80);comment:接口全路径;unique"`
}

func (*Apis) TableName() string {
	return common.ModelPrefix + "apis"
}

func NewApis() *Apis {
	return &Apis{}
}

func (*Apis) Comment() string {
	return "api权限表"
}

func (s *Apis) FindOrCreateAuth(db *gorm.DB, path, method string) (*Apis, error) {
	if err := db.Where("path", path).First(&s).Error; err != nil {
		if !strings.Contains(path, "*") {
			pathList := strings.Split(path, "/")
			if len(pathList) > 2 {
				s.Group = pathList[2]
			}
			s.ParentId = 1
			s.Path = path
			s.Method = method
			s.Name = generateApiName(path)
			if err := db.Create(&s).Error; err != nil {
				return s, err
			}
		}
		return s, err
	}
	return s, nil
}

type ApiNode struct {
	Id       int64      `json:"id"`
	ParentId int64      `json:"parentId"`
	Group    string     `json:"group"`
	Name     string     `json:"name"`
	Method   string     `json:"method"`
	Path     string     `json:"path"`
	Children []*ApiNode `json:"children"`
}

func (s *Apis) GetUserTree(db *gorm.DB, ids []string) ([]ApiNode, error) {
	menusNodeTree := make([]ApiNode, 0) // 根节点
	// 所有节点
	allMenus := make([]Apis, 0)
	if err := db.Model(&Apis{}).
		Where("id in (?)", ids).
		Find(&allMenus).Error; err != nil {
		return menusNodeTree, err
	}
	for _, m := range allMenus {
		childMenus := make([]*ApiNode, 0)
		if m.ParentId == 0 {
			rootNode := ApiNode{
				Id:       m.ID,
				ParentId: m.ParentId,
				Group:    m.Group,
				Name:     m.Name,
				Method:   m.Method,
				Path:     m.Path,
				Children: childMenus,
			}
			menusNodeTree = append(menusNodeTree, rootNode)
		}
	}
	for i, _ := range menusNodeTree {
		s.walk(allMenus, &menusNodeTree[i])
	}
	return menusNodeTree, nil
}

// MenuTree 系统递归树[满足无限层级递归]

func (s *Apis) SysTree(db *gorm.DB) ([]ApiNode, error) {
	menusNodeTree := make([]ApiNode, 0) // 根节点
	// 所有节点
	allMenus := make([]Apis, 0)
	if err := db.Model(&Apis{}).Where("1=1").Find(&allMenus).Error; err != nil {
		return menusNodeTree, err
	}
	for _, m := range allMenus {
		childMenus := make([]*ApiNode, 0)
		if m.ParentId == 0 {
			rootNode := ApiNode{
				Id:       m.ID,
				ParentId: m.ParentId,
				Name:     m.Name,
				Group:    m.Group,
				Method:   m.Method,
				Path:     m.Path,
				Children: childMenus,
			}
			menusNodeTree = append(menusNodeTree, rootNode)
		}
	}
	for i, _ := range menusNodeTree {
		s.walk(allMenus, &menusNodeTree[i])
	}
	return menusNodeTree, nil
}

// 递归组装菜单树（根节点）
func (s *Apis) walk(allMenus []Apis, rootNode *ApiNode) {
	// 列出所有下级子目录
	nodes := s.childrenList(allMenus, rootNode.Id)
	if len(nodes) == 0 {
		return
	}
	// 遍历这些文件
	for _, m := range nodes {
		newNode := ApiNode{
			Id:       m.Id,
			ParentId: m.ParentId,
			Group:    m.Group,
			Name:     m.Name,
			Method:   m.Method,
			Path:     m.Path,
			Children: nil,
		}
		s.walk(allMenus, &newNode)
		rootNode.Children = append(rootNode.Children, &newNode)
	}
	return
}

// 获得子节点列表
func (s *Apis) childrenList(allMenus []Apis, pId int64) (menusNodeTree []ApiNode) {
	for _, m := range allMenus {
		if m.ParentId == pId {
			rootNode := ApiNode{
				Id:       m.ID,
				ParentId: m.ParentId,
				Name:     m.Name,
				Group:    m.Group,
				Method:   m.Method,
				Path:     m.Path,
				Children: nil,
			}
			menusNodeTree = append(menusNodeTree, rootNode)
		}
	}
	return
}