projectx/internal/router/manage/system/sys_role.go

package system

import (
	"fmt"
	"net/http"
	"strings"
	"time"

	"management/internal/db/model/dto"
	db "management/internal/db/sqlc"
	"management/internal/pkg/convertor"
	"management/internal/router/manage/util"
	systemservice "management/internal/service/system"
	"management/internal/tpl"
)

type SysRoleHandler struct{}

func NewSysRoleHandler() *SysRoleHandler {
	return &SysRoleHandler{}
}

func (h *SysRoleHandler) List(w http.ResponseWriter, r *http.Request) {
	tpl.HTML(w, r, "role/list.tmpl", nil)
}

func (h *SysRoleHandler) PostList(w http.ResponseWriter, r *http.Request) {
	var q dto.SearchDto
	q.SearchStatus = convertor.ConvertInt(r.PostFormValue("status"), 9999)
	q.SearchParentID = convertor.ConvertInt(r.PostFormValue("parentId"), 0)
	q.SearchName = r.PostFormValue("name")
	q.SearchID = convertor.ConvertInt[int64](r.PostFormValue("id"), 0)
	q.Page = convertor.ConvertInt(r.PostFormValue("page"), 1)
	q.Rows = convertor.ConvertInt(r.PostFormValue("rows"), 10)
	res, count, err := systemservice.ListSysRoleCondition(r.Context(), q)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	data := tpl.ResponseList{
		Code:    0,
		Message: "ok",
		Count:   count,
		Data:    res,
	}
	tpl.JSON(w, data)
}

func (h *SysRoleHandler) Add(w http.ResponseWriter, r *http.Request) {
	vm := &db.SysRole{Sort: 6666}
	tpl.HTML(w, r, "role/edit.tmpl", map[string]any{
		"Item": vm,
	})
}

func (h *SysRoleHandler) Edit(w http.ResponseWriter, r *http.Request) {
	vars := r.URL.Query()
	id := util.DefaultInt(vars, "id", 0)
	vm := &db.SysRole{Sort: 6666}
	if id > 0 {
		ctx := r.Context()
		vm, _ = systemservice.GetSysRole(ctx, int32(id))
	}
	tpl.HTML(w, r, "role/edit.tmpl", map[string]any{
		"Item": vm,
	})
}

func (h *SysRoleHandler) Save(w http.ResponseWriter, r *http.Request) {
	id := util.ConvertInt(r.PostFormValue("ID"), 0)
	name := r.PostFormValue("Name")
	parentID := util.ConvertInt(r.PostFormValue("ParentID"), 0)
	displayName := r.PostFormValue("DisplayName")
	sort := util.ConvertInt(r.PostFormValue("Sort"), 6666)
	status := util.ConvertInt(r.PostFormValue("Status"), 9999)

	ctx := r.Context()
	var parent *db.SysRole
	if parentID > 0 {
		var err error
		parent, err = systemservice.GetSysRole(ctx, int32(parentID))
		if err != nil {
			tpl.JSON(w, tpl.Response{Success: false, Message: "父级节点错误"})
			return
		}
	} else {
		parent = &db.SysRole{
			ID:         0,
			ParentID:   0,
			ParentPath: ",0,",
		}
	}

	if id == 0 {
		arg := &db.CreateSysRoleParams{
			Name:        name,
			DisplayName: displayName,
			Vip:         false,
			ParentID:    parent.ID,
			ParentPath:  fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
			Status:      int32(status),
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		}
		_, err := systemservice.CreateSysRole(ctx, arg)
		if err != nil {
			if db.IsUniqueViolation(err) {
				tpl.JSON(w, tpl.Response{Success: false, Message: "角色名称已存在"})
				return
			}
			tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
			return
		}

		tpl.JSON(w, tpl.Response{Success: true, Message: "添加成功"})
	} else {
		res, err := systemservice.GetSysRole(ctx, int32(id))
		if err != nil {
			tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
			return
		}
		arg := &db.UpdateSysRoleParams{
			ID:          res.ID,
			DisplayName: displayName,
			Sort:        int32(sort),
			Status:      int32(status),
			ParentID:    parent.ID,
			ParentPath:  fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
			UpdatedAt:   time.Now(),
		}
		_, err = systemservice.UpdateSysRole(ctx, arg)
		if err != nil {
			tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
			return
		}

		tpl.JSON(w, tpl.Response{Success: true, Message: "更新成功"})
	}
}

func (h *SysRoleHandler) XmSelect(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	res, err := systemservice.XmSelectSysRole(ctx, 0)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	tpl.JSON(w, res)
}

func (h *SysRoleHandler) SetMenu(w http.ResponseWriter, r *http.Request) {
	vars := r.URL.Query()
	id := util.DefaultInt(vars, "id", 0)
	vm := struct {
		Role  *db.SysRole
		Menus []*dto.SetMenuDto
	}{}
	if id > 0 {
		ctx := r.Context()
		var err error
		vm.Role, err = systemservice.GetSysRole(ctx, int32(id))
		if err == nil {
			vm.Menus, _ = systemservice.SetMenuViewData(ctx, vm.Role.ID)
		}
	}

	tpl.HTML(w, r, "role/set_menu.tmpl", map[string]any{
		"Item": vm,
	})
}

func (h *SysRoleHandler) PostSetMenu(w http.ResponseWriter, r *http.Request) {
	id := util.ConvertInt(r.PostFormValue("ID"), 0)
	menus := r.PostFormValue("roleMenu")

	if id == 0 {
		tpl.JSON(w, tpl.Response{Success: false, Message: "角色异常, 请刷新重试"})
		return
	}

	if len(menus) == 0 {
		tpl.JSON(w, tpl.Response{Success: false, Message: "请选择菜单"})
		return
	}

	ctx := r.Context()
	_, err := systemservice.GetSysRole(ctx, int32(id))
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	menuArr := strings.Split(menus, ",")
	if len(menuArr) == 0 {
		tpl.JSON(w, tpl.Response{Success: false, Message: "请选择菜单"})
		return
	}

	var menuList []*db.SysMenu
	for _, v := range menuArr {
		menuID := util.ConvertInt(v, 0)
		if menuID > 0 {
			menu, err := systemservice.GetSysMenu(ctx, int32(menuID))
			if err != nil {
				tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
				return
			}

			menuList = append(menuList, menu)
		}
	}

	if len(menuList) == 0 {
		tpl.JSON(w, tpl.Response{Success: false, Message: "请选择正确的菜单"})
		return
	}

	err = systemservice.SetMenu(ctx, int32(id), menuList)
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	tpl.JSON(w, tpl.Response{Success: true, Message: "设置成功"})
}

func (h *SysRoleHandler) DTree(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	res, err := systemservice.DTreeSysRole(ctx, 0)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	rsp := tpl.ResponseDtree{
		Status: tpl.ResponseDtreeStatus{
			Code:    200,
			Message: "OK",
		},
		Data: res,
	}
	tpl.JSON(w, rsp)
}

func (h *SysRoleHandler) Refresh(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	err := systemservice.RefreshSysRole(ctx)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	tpl.JSON(w, tpl.Response{Success: true, Message: "刷新成功"})
}

func (h *SysRoleHandler) RebuildParentPath(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	err := systemservice.RebuildSysRoleParentPath(ctx)
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	tpl.JSON(w, tpl.Response{Success: true, Message: "重建成功"})
}

func (h *SysRoleHandler) RefreshRoleMenus(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()

	// 获取需要刷新的角色ID
	roleID := util.ConvertInt(r.PostFormValue("roleID"), 0)
	sysRole, err := systemservice.GetSysRole(ctx, int32(roleID))
	if err != nil || sysRole == nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	// 刷新角色菜单 (角色所拥有的菜单集合)
	_, err = systemservice.SetOwnerListMenuByRoleID(ctx, sysRole.ID)
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	// 刷新角色菜单 (角色所拥有的菜单集合)
	_, err = systemservice.SetOwnerMapMenuByRoleID(ctx, sysRole.ID)
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	// 刷新角色菜单树 (pear admin layui 使用的格式)
	_, err = systemservice.SetRecursiveSysMenus(ctx, sysRole.ID)
	if err != nil {
		tpl.JSON(w, tpl.Response{Success: false, Message: err.Error()})
		return
	}

	tpl.JSON(w, tpl.Response{Success: true, Message: "刷新成功"})
}