301 lines
7.7 KiB
Go
301 lines
7.7 KiB
Go
package system
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"management/internal/db/model/dto"
|
|
db "management/internal/db/sqlc"
|
|
"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 = util.ConvertInt(r.PostFormValue("SearchStatus"), 9999)
|
|
q.SearchParentID = util.ConvertInt(r.PostFormValue("SearchParentID"), 0)
|
|
q.SearchName = r.PostFormValue("SearchName")
|
|
q.SearchKey = r.PostFormValue("SearchKey")
|
|
q.Page = util.ConvertInt(r.PostFormValue("page"), 1)
|
|
q.Rows = util.ConvertInt(r.PostFormValue("rows"), 10)
|
|
ctx := r.Context()
|
|
res, count, err := systemservice.ListSysRoleCondition(ctx, 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)
|
|
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: "刷新成功"})
|
|
}
|