projectx/internal/erpserver/handler/system/role/role.go

package role

import (
	"net/http"
	"strings"

	"management/internal/erpserver/model/dto"
	"management/internal/erpserver/model/form"
	"management/internal/erpserver/model/system"
	"management/internal/erpserver/model/view"
	v1 "management/internal/erpserver/service/v1"
	"management/internal/erpserver/templ/system/role"
	"management/internal/pkg/binding"
	"management/internal/pkg/convertor"
	"management/internal/pkg/render"
)

type App struct {
	render      render.Renderer
	roleService v1.RoleService
	menuService v1.MenuService
}

func NewApp(render render.Renderer, roleService v1.RoleService, menuService v1.MenuService) *App {
	return &App{
		render:      render,
		roleService: roleService,
		menuService: menuService,
	}
}

func (a *App) list(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case http.MethodGet:
		ctx := r.Context()
		a.render.Render(ctx, w, role.List(ctx))
	case http.MethodPost:
		var q dto.SearchDto
		q.SearchTimeBegin, q.SearchTimeEnd = convertor.DefaultStartTimeAndEndTime(r.PostFormValue("timeBegin"), r.PostFormValue("timeEnd"))
		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 := a.roleService.List(r.Context(), q)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}

		data := render.ResponseList{
			Code:    0,
			Message: "ok",
			Count:   count,
			Data:    res,
		}
		a.render.JSON(w, data)
	default:
		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
	}
}

func (a *App) add(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	a.render.Render(ctx, w, role.Edit(ctx, &system.Role{Sort: 6666}))
}

func (a *App) addChildren(w http.ResponseWriter, r *http.Request) {
	vars := r.URL.Query()
	parentID := convertor.QueryInt[int32](vars, "parentID", 0)
	vm := &system.Role{ParentID: parentID, Sort: 6666}
	ctx := r.Context()
	a.render.Render(ctx, w, role.Edit(ctx, vm))
}

func (a *App) edit(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	vars := r.URL.Query()
	id := convertor.QueryInt[int32](vars, "id", 0)
	vm := &system.Role{Sort: 6666}
	if id > 0 {
		vm, _ = a.roleService.Get(ctx, id)
	}
	a.render.Render(ctx, w, role.Edit(ctx, vm))
}

func (a *App) save(w http.ResponseWriter, r *http.Request) {
	var req form.Role
	if err := binding.Form.Bind(r, &req); err != nil {
		a.render.JSONErr(w, binding.ValidatorErrors(err))
		return
	}

	ctx := r.Context()
	if *req.ID == 0 {
		err := a.roleService.Create(ctx, &req)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}
		a.render.JSONOk(w, "添加成功")
	} else {
		err := a.roleService.Update(ctx, &req)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}
		a.render.JSONOk(w, "更新成功")
	}
}

func (a *App) data(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	vars := r.URL.Query()
	t := vars.Get("type")
	if t == "tree" {
		res, err := a.roleService.Tree(ctx, 0)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}
		a.render.JSON(w, res)
	} else if t == "xm_select_tree" {
		res, err := a.roleService.XmSelectTree(ctx, 0)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}
		a.render.JSON(w, res)
	}
}

func (a *App) refreshCache(w http.ResponseWriter, r *http.Request) {
	err := a.roleService.RefreshCache(r.Context())
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}

	a.render.JSONOk(w, "缓存刷新成功")
}

func (a *App) rebuildParentPath(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	err := a.roleService.RebuildParentPath(ctx)
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}

	a.render.JSONOk(w, "重建成功")
}

func (a *App) refreshRoleMenus(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()

	// 获取需要刷新的角色ID
	roleID := convertor.ConvertInt[int32](r.PostFormValue("roleID"), 0)
	roleModel, err := a.roleService.Get(ctx, roleID)
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}
	if roleModel == nil {
		a.render.JSONErr(w, "角色不存在")
		return
	}

	// 刷新角色菜单 (角色所拥有的菜单集合)
	_, err = a.menuService.SetListByRoleID(ctx, roleModel.ID)
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}

	// 刷新角色菜单 (角色所拥有的菜单集合)
	_, err = a.menuService.SetListByRoleIDToMap(ctx, roleModel.ID)
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}

	// 刷新角色菜单树 (pear admin layui 使用的格式)
	_, err = a.menuService.SetOwerMenus(ctx, roleModel.ID)
	if err != nil {
		a.render.JSONErr(w, err.Error())
		return
	}

	a.render.JSONOk(w, "刷新成功")
}

func (a *App) setMenu(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case http.MethodGet:
		vars := r.URL.Query()
		id := convertor.QueryInt[int32](vars, "id", 0)
		vm := &view.SetMenuView{
			Role:  &system.Role{},
			Menus: []*dto.SetMenuDto{},
		}
		ctx := r.Context()
		if id > 0 {
			var err error
			vm.Role, err = a.roleService.Get(ctx, id)
			if err == nil {
				vm.Menus, _ = a.menuService.MenuViewData(ctx, vm.Role.ID)
			}
		}
		a.render.Render(ctx, w, role.SetMenu(ctx, vm))
	case http.MethodPost:
		ctx := r.Context()
		id := convertor.ConvertInt[int32](r.PostFormValue("ID"), 0)
		if id == 0 {
			a.render.JSONErr(w, "角色异常, 请刷新重试")
			return
		}
		roleModel, err := a.roleService.Get(ctx, id)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}

		menus := r.PostFormValue("roleMenu")
		if len(menus) == 0 {
			a.render.JSONErr(w, "请选择菜单")
			return
		}
		menuArr := strings.Split(menus, ",")
		if len(menuArr) == 0 {
			a.render.JSONErr(w, "请选择菜单")
			return
		}

		var rms []*system.RoleMenu
		for _, v := range menuArr {
			menuID := convertor.ConvertInt(v, 0)
			if menuID > 0 {
				menu, err := a.menuService.Get(ctx, int32(menuID))
				if err != nil {
					a.render.JSONErr(w, err.Error())
					return
				}

				rms = append(rms, &system.RoleMenu{
					RoleID: roleModel.ID,
					MenuID: menu.ID,
				})
			}
		}

		if len(rms) == 0 {
			a.render.JSONErr(w, "请选择正确的菜单")
			return
		}

		err = a.menuService.SetRoleMenu(ctx, id, rms)
		if err != nil {
			a.render.JSONErr(w, err.Error())
			return
		}

		a.render.JSONOk(w, "设置成功")
	default:
		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
	}
}