easygoadmin/app/controller/menu.go

/**
 * 菜单管理-控制器
 * @author
 * @since 2021/9/13
 * @File : menu
 */
package controller

import (
	"easygoadmin/app/dto"
	"easygoadmin/app/model"
	"easygoadmin/app/service"
	"easygoadmin/utils"
	"easygoadmin/utils/common"
	"easygoadmin/utils/gconv"
	"easygoadmin/utils/response"
	"github.com/gin-gonic/gin"
	"net/http"
)

var Menu = new(menuCtl)

type menuCtl struct{}

func (c *menuCtl) Index(ctx *gin.Context) {
	// 渲染模板
	response.BuildTpl(ctx, "menu_index.html").WriteTpl()
}

func (c *menuCtl) List(ctx *gin.Context) {
	//// 参数绑定
	//var req *dto.MenuQueryReq
	//if err := ctx.Bind(&req); err != nil {
	//	ctx.JSON(http.StatusOK, common.JsonResult{
	//		Code: -1,
	//		Msg:  err.Error(),
	//	})
	//	return
	//}

	// 调用查询列表方法
	list, err := service.Menu.GetList(nil)
	if err != nil {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 返回结果
	ctx.JSON(http.StatusOK, common.JsonResult{
		Code: 0,
		Msg:  "查询成功",
		Data: list,
	})
}

func (c *menuCtl) Edit(ctx *gin.Context) {
	// 获取菜单列表
	menuTreeList, _ := service.Menu.GetTreeList()
	// 数据源转换
	menuList := service.Menu.MakeList(menuTreeList)

	// 记录ID
	id := gconv.Int(ctx.Query("id"))
	if id > 0 {
		// 编辑
		info := &model.Menu{Id: id}
		has, err := info.Get()
		if err != nil || !has {
			ctx.JSON(http.StatusOK, common.JsonResult{
				Code: -1,
				Msg:  err.Error(),
			})
			return
		}

		// 获取节点
		funcList := make([]model.Menu, 0)
		utils.XormDb.Where("pid=? and type=1 and mark=1", id).Find(&funcList)
		sortList := make([]interface{}, 0)
		for _, v := range funcList {
			sortList = append(sortList, v.Sort)
		}

		// 渲染模板
		response.BuildTpl(ctx, "menu_edit.html").WriteTpl(gin.H{
			"info":     info,
			"typeList": common.MENU_TYPE_LIST,
			"funcList": sortList,
			"menuList": menuList,
		})
	} else {
		// 添加

		pid := ctx.GetInt("pid")
		var info model.Menu
		info.Pid = pid
		info.Status = 1
		info.Target = 1

		// 渲染模板
		response.BuildTpl(ctx, "menu_edit.html").WriteTpl(gin.H{
			"info":     info,
			"typeList": common.MENU_TYPE_LIST,
			"funcList": make([]interface{}, 0),
			"menuList": menuList,
		})
	}
}

func (c *menuCtl) Detail(ctx *gin.Context) {
	// 记录ID
	id := ctx.Query("id")
	if id == "" {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  "记录ID不能为空",
		})
		return
	}
	info := model.Menu{Id: gconv.Int(id)}
	has, err := info.Get()
	if err != nil || !has {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 菜单信息
	menu := dto.MenuInfoVo{}
	menu.Menu = info
	// 获取权限节点
	if info.Type == 0 {
		// 获取角色菜单权限列表
		var menuList []model.Menu
		utils.XormDb.Where("pid=?", info.Id).Where("type=1").Where("mark=1").Find(&menuList)
		checkedList := make([]int, 0)
		for _, v := range menuList {
			checkedList = append(checkedList, v.Sort)
		}
		if len(checkedList) > 0 {
			menu.CheckedList = checkedList
		} else {
			menu.CheckedList = make([]int, 0)
		}
	}

	// 返回结果
	ctx.JSON(http.StatusOK, common.JsonResult{
		Code: 0,
		Msg:  "查询成功",
		Data: menu,
	})
}

func (c *menuCtl) Add(ctx *gin.Context) {
	// 参数绑定
	var req *dto.MenuAddReq
	if err := ctx.ShouldBind(&req); err != nil {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 调用添加方法
	rows, err := service.Menu.Add(req, utils.Uid(ctx))
	if err != nil || rows == 0 {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 返回结果
	ctx.JSON(http.StatusOK, common.JsonResult{
		Code: 0,
		Msg:  "添加成功",
	})
}

func (c *menuCtl) Update(ctx *gin.Context) {
	// 参数绑定
	var req *dto.MenuUpdateReq
	if err := ctx.ShouldBind(&req); err != nil {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 调用更新方法
	rows, err := service.Menu.Update(req, utils.Uid(ctx))
	if err != nil || rows == 0 {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 返回结果
	ctx.JSON(http.StatusOK, common.JsonResult{
		Code: 0,
		Msg:  "更新成功",
	})
}

func (c *menuCtl) Delete(ctx *gin.Context) {
	// 记录ID
	ids := ctx.Param("ids")
	// 调用删除方法
	rows, err := service.Menu.Delete(ids)
	if err != nil || rows == 0 {
		ctx.JSON(http.StatusOK, common.JsonResult{
			Code: -1,
			Msg:  err.Error(),
		})
		return
	}

	// 返回结果
	ctx.JSON(http.StatusOK, common.JsonResult{
		Code: 0,
		Msg:  "删除成功",
	})
}