MTP20_IF/controllers/common/menu.go

package common

import (
	"encoding/json"
	"fmt"
	"mtp2_if/global/app"
	"mtp2_if/global/e"
	"mtp2_if/logger"
	"mtp2_if/models"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"
)

// ClientMenu 客户端菜单
type ClientMenu struct {
	Code     string       `json:"code"`     // ID
	Title    string       `json:"title"`    // 标题
	Sort     int          `json:"sort"`     // 排序
	Type     int          `json:"type"`     // 类型，1：菜单 2：按钮
	RuleKey  string       `json:"rulekey"`  // 对应权限主键，对应FUNCMENULIST表RESOURCECODE
	IsShow   bool         `json:"isshow"`   // 是否显示
	URL      string       `json:"url"`      // URL
	Remark   string       `json:"remark"`   // 备注
	Children []ClientMenu `json:"children"` // 子菜单列表
}

// PCWebMenuAuth PC客户端页面（按钮）权限
type PCWebMenuAuth struct {
	Label       string `json:"label"`       // 标题
	Code        string `json:"code"`        // ID，新版本中权限KEY也是这个字段
	RequireAuth bool   `json:"requireauth"` // 是否需要权限判断
	IsShow      bool   `json:"isshow"`      // 是否显示
	Remark      string `json:"remark"`      // 备注
}

// PCWebMenu PC客户端菜单权限
type PCWebMenu struct {
	Title       string          `json:"title"`       // 标题
	Code        string          `json:"code"`        // ID，新版本中权限KEY也是这个字段
	Path        string          `json:"path"`        // 路由路径
	Component   string          `json:"component"`   // 路由组件
	Sort        int             `json:"sort"`        // 排序
	RequireAuth bool            `json:"requireauth"` // 是否需要权限判断
	IsShow      bool            `json:"isshow"`      // 是否显示
	URL         string          `json:"url"`         // URL
	Remark      string          `json:"remark"`      // 备注
	Children    []PCWebMenu     `json:"children"`    // 子菜单列表
	Auth        []PCWebMenuAuth `json:"auth"`        // 页面权限列表
}

// GetClientMenusReq 获取交易端菜单请求参数
type GetClientMenusReq struct {
	LoginID    int    `form:"loginID" binding:"required"`
	ClientType int    `form:"clientType"`
	Name       string `form:"name"`
}

// GetClientMenus 获取交易端菜单
// @Summary 获取交易端菜单
// @Produce json
// @Security ApiKeyAuth
// @Param loginID query int true "登录账号"
// @Param clientType query int false "终端类型，0：PCWEB 1：Mobile，不传默认返回PC端数据"
// @Param name query string false "分支名称，ermcp：风管云平台 pingan：平安期货，不传默认返回风管云平台"
// @Success 200 {array} ClientMenu
// @Failure 500 {object} app.Response
// @Router /Common/GetClientMenus [get]
// @Tags 通用服务
func GetClientMenus(c *gin.Context) {
	appG := app.Gin{C: c}

	// 获取请求参数
	var req GetClientMenusReq
	if err := c.ShouldBind(&req); err != nil {
		logger.GetLogger().Errorf("GetClientMenus parse query req: %v", err)
		appG.Response(http.StatusBadRequest, e.INVALID_PARAMS, nil)
		return
	}

	// 获取菜单数据
	jsonFile := "pc_menu.json"
	if req.ClientType == 1 {
		// Mobile
		jsonFile = "mobile_menu.json"
	}

	// 获取分支名称
	if len(req.Name) > 0 {
		if req.ClientType == 0 {
			// 目前只支持PCWEB
			jsonFile = fmt.Sprintf("pcweb_menu/pcweb_menu_%s.json", req.Name)
		}
	}

	// 读取json文件
	// 21-08-15 增加OEM菜单文件支持功能
	filePtr, err := os.Open(fmt.Sprintf("config/%s", jsonFile))
	if err != nil {
		// 读取文件失败
		logger.GetLogger().Errorf("GetClientMenus failed: %s", err.Error())
		appG.Response(http.StatusBadRequest, e.ERROR_CFG_FILE_FAIL, nil)
		return
	}
	defer filePtr.Close()

	var datas []interface{}
	// 创建Json编码器
	decoder := json.NewDecoder(filePtr)
	if err = decoder.Decode(&datas); err != nil {
		// Json解码失败
		logger.GetLogger().Errorf("GetClientMenus failed: %s", err.Error())
		appG.Response(http.StatusBadRequest, e.ERROR_CFG_JSON_FAIL, nil)
		return
	}

	rst := createMenu(datas, req.LoginID)

	r := make([]ClientMenu, 0)
	if req.ClientType == 0 {
		// PC 去掉第一层没有子项的
		for _, v := range rst {
			if len(v.Children) > 0 {
				r = append(r, v)
			}
		}
	} else {
		r = rst
	}

	// 查询成功
	logger.GetLogger().Debugln("GetPCMenus successed: %v", r)
	appG.Response(http.StatusOK, e.SUCCESS, r)
}

// createMenu 创建菜单数据
func createMenu(datas []interface{}, loginID int) []ClientMenu {
	clientMenus := make([]ClientMenu, 0)
	for _, v := range datas {
		data := v.(map[string]interface{})
		if data["isshow"].(bool) {
			rulekey := data["rulekey"].(string)
			if len(rulekey) != 0 {
				// 判断是否有权限
				funcMenuList, err := models.GetErmcpRoleFuncMenuLists(loginID, rulekey)
				if err != nil {
					continue
				}
				if len(funcMenuList) == 0 {
					continue
				}
			}

			item := ClientMenu{
				Code:     data["code"].(string),
				Title:    data["title"].(string),
				Sort:     int(data["sort"].(float64)),
				Type:     int(data["type"].(float64)),
				RuleKey:  data["rulekey"].(string),
				IsShow:   data["isshow"].(bool),
				Children: createMenu(data["children"].([]interface{}), loginID), // TODO: - 这里为什么可以这样转？
			}
			if data["url"] != nil {
				item.URL = data["url"].(string)
			}
			if data["remark"] != nil {
				item.Remark = data["remark"].(string)
			}
			clientMenus = append(clientMenus, item)
		}
	}

	return clientMenus
}

// GetPCWebMenusReq 获取PCWeb交易端菜单请求参数
type GetPCWebMenusReq struct {
	LoginID int    `form:"loginID" binding:"required"`
	Name    string `form:"name"`
}

// GetPCWebMenus 获取PCWeb交易端菜单
// @Summary 获取PCWeb交易端菜单
// @Produce json
// @Security ApiKeyAuth
// @Param loginID query int true "登录账号"
// @Param name query string false "分支名称，ermcp：风管云平台，不传默认返回ermcp"
// @Success 200 {array} PCWebMenu
// @Failure 500 {object} app.Response
// @Router /Common/GetPCWebMenus [get]
// @Tags 通用服务
func GetPCWebMenus(c *gin.Context) {
	appG := app.Gin{C: c}

	// 获取请求参数
	var req GetPCWebMenusReq
	if err := c.ShouldBind(&req); err != nil {
		logger.GetLogger().Errorf("GetPCWebMenus parse query req: %v", err)
		appG.Response(http.StatusBadRequest, e.INVALID_PARAMS, nil)
		return
	}

	// 获取菜单数据
	jsonFile := "pcweb_menu_ermcp.json"
	// 获取分支名称
	if len(req.Name) > 0 {
		jsonFile = fmt.Sprintf("pcweb_menu/pcweb_menu_%s.json", req.Name)
	}

	// 读取json文件
	filePtr, err := os.Open(fmt.Sprintf("config/%s", jsonFile))
	if err != nil {
		// 读取文件失败
		logger.GetLogger().Errorf("GetPCWebMenus failed: %s", err.Error())
		appG.Response(http.StatusBadRequest, e.ERROR_CFG_FILE_FAIL, nil)
		return
	}
	defer filePtr.Close()

	var datas []interface{}
	// 创建Json编码器
	decoder := json.NewDecoder(filePtr)
	if err = decoder.Decode(&datas); err != nil {
		// Json解码失败
		logger.GetLogger().Errorf("GetPCWebMenus failed: %s", err.Error())
		appG.Response(http.StatusBadRequest, e.ERROR_CFG_JSON_FAIL, nil)
		return
	}

	rst := createPCWebMenu(datas, req.LoginID)

	// 查询成功
	logger.GetLogger().Debugln("GetPCWebMenus successed: %v", rst)
	appG.Response(http.StatusOK, e.SUCCESS, rst)
}

// createPCWebMenu 创建菜单数据
func createPCWebMenu(datas []interface{}, loginID int) []PCWebMenu {
	clientMenus := make([]PCWebMenu, 0)
	for _, v := range datas {
		data := v.(map[string]interface{})
		if data["isshow"].(bool) {
			// 判断权限
			requireAuth := data["requireauth"].(bool)
			// FIXME: - 暂时代码
			requireAuth = false
			if requireAuth {
				code := data["code"].(string) // PC WEB V6版本直接使用CODE来做权限KEY
				// 判断是否有权限
				funcMenuList, err := models.GetErmcpRoleFuncMenuLists(loginID, code)
				if err != nil {
					continue
				}
				if len(funcMenuList) == 0 {
					continue
				}
			}

			// 构建页面（按钮）权限
			auths := make([]PCWebMenuAuth, 0)
			if data["auth"] != nil {
				list := data["auth"].([]interface{})
				for _, a := range list {
					am := a.(map[string]interface{})
					if am["isshow"].(bool) {
						// 判断权限
						requireAuth := data["requireauth"].(bool)
						// FIXME: - 暂时代码
						requireAuth = false
						if requireAuth {
							code := data["code"].(string) // PC WEB V6版本直接使用CODE来做权限KEY
							// 判断是否有权限
							funcMenuList, err := models.GetErmcpRoleFuncMenuLists(loginID, code)
							if err != nil {
								continue
							}
							if len(funcMenuList) == 0 {
								continue
							}
						}

						auth := PCWebMenuAuth{
							Label:       am["label"].(string),
							Code:        am["code"].(string),
							RequireAuth: am["requireauth"].(bool),
							IsShow:      am["isshow"].(bool),
						}
						if am["remark"] != nil {
							auth.Remark = am["remark"].(string)
						}
						auths = append(auths, auth)
					}
				}
			}

			item := PCWebMenu{
				Title:       data["title"].(string),
				Code:        data["code"].(string),
				Path:        data["path"].(string),
				Component:   data["component"].(string),
				Sort:        int(data["sort"].(float64)),
				RequireAuth: data["requireauth"].(bool),
				IsShow:      data["isshow"].(bool),
			}
			if data["url"] != nil {
				item.URL = data["url"].(string)
			}
			if data["remark"] != nil {
				item.Remark = data["remark"].(string)
			}
			if data["children"] != nil {
				item.Children = createPCWebMenu(data["children"].([]interface{}), loginID)
			}
			if len(auths) > 0 {
				item.Auth = auths
			}
			clientMenus = append(clientMenus, item)
		}
	}

	return clientMenus
}