提交

modules/base/service/base_sys_role.go

@@ -0,0 +1,174 @@
+package service
+
+import (
+	"context"
+
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/cool-team-official/cool-admin-go/modules/base/model"
+	"github.com/gogf/gf/v2/container/garray"
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+var baseSysUserRole = model.NewBaseSysUserRole()
+
+type BaseSysRoleService struct {
+	*cool.Service
+}
+
+// ModifyAfter modify after
+func (s *BaseSysRoleService) ModifyAfter(ctx context.Context, method string, param g.MapStrAny) (err error) {
+	if param["id"] != nil {
+		err = s.updatePerms(ctx, gconv.Uint(param["id"]), gconv.SliceUint(param["menuIdList"]), gconv.SliceUint(param["departmentIdList"]))
+	}
+	return
+}
+
+// updatePerms(roleId, menuIdList?, departmentIds = [])
+func (s *BaseSysRoleService) updatePerms(ctx context.Context, roleId uint, menuIdList, departmentIds []uint) (err error) {
+	// 更新菜单权限
+	cool.DBM(model.NewBaseSysRoleMenu()).Where("roleId = ?", roleId).Delete()
+	if len(menuIdList) > 0 {
+		roleMenuList := make([]g.MapStrAny, len(menuIdList))
+		for i, menuId := range menuIdList {
+			roleMenuList[i] = g.MapStrAny{
+				"roleId": roleId,
+				"menuId": menuId,
+			}
+		}
+		cool.DBM(model.NewBaseSysRoleMenu()).Data(roleMenuList).Insert()
+	}
+	// 更新部门权限
+	cool.DBM(model.NewBaseSysRoleDepartment()).Where("roleId = ?", roleId).Delete()
+	if len(departmentIds) > 0 {
+		roleDepartmentList := make([]g.MapStrAny, len(departmentIds))
+		for i, departmentId := range departmentIds {
+			roleDepartmentList[i] = g.MapStrAny{
+				"roleId":       roleId,
+				"departmentId": departmentId,
+			}
+		}
+		cool.DBM(model.NewBaseSysRoleDepartment()).Data(roleDepartmentList).Insert()
+	}
+	// 刷新权限
+	userRoles, err := cool.DBM(model.NewBaseSysUserRole()).Where("roleId = ?", roleId).All()
+	if err != nil {
+		return
+	}
+	baseSysPermsService := NewBaseSysPermsService()
+	for _, v := range userRoles {
+		vmap := v.Map()
+		if vmap["userId"] != nil {
+			baseSysPermsService.RefreshPerms(ctx, gconv.Uint(vmap["userId"]))
+		}
+	}
+
+	return
+
+}
+
+// GetByUser get array  roleId by userId
+func (s *BaseSysRoleService) GetByUser(userId uint) []string {
+	var (
+		roles []string
+	)
+	res, _ := cool.DBM(baseSysUserRole).Where("userId = ?", userId).Array("roleId")
+	for _, v := range res {
+		roles = append(roles, gconv.String(v))
+	}
+	return roles
+}
+
+// BaseSysRoleService Info 方法重构
+func (s *BaseSysRoleService) ServiceInfo(ctx context.Context, req *cool.InfoReq) (data interface{}, err error) {
+	info, err := cool.DBM(s.Model).Where("id = ?", req.Id).One()
+	if err != nil {
+		return nil, err
+	}
+	if !info.IsEmpty() {
+		var menus gdb.Result
+		if req.Id == 1 {
+			menus, err = cool.DBM(model.NewBaseSysMenu()).All()
+			if err != nil {
+				return nil, err
+			}
+		} else {
+			menus, err = cool.DBM(model.NewBaseSysRoleMenu()).Where("roleId = ?", req.Id).All()
+			if err != nil {
+				return nil, err
+			}
+		}
+		menuIdList := garray.NewIntArray()
+		for _, v := range menus {
+			menuIdList.Append(gconv.Int(v["menuId"]))
+		}
+		var departments gdb.Result
+		if req.Id == 1 {
+			departments, err = cool.DBM(model.NewBaseSysRoleDepartment()).All()
+			if err != nil {
+				return nil, err
+			}
+		} else {
+			departments, err = cool.DBM(model.NewBaseSysRoleDepartment()).Where("roleId = ?", req.Id).All()
+			if err != nil {
+				return nil, err
+			}
+		}
+		departmentIdList := garray.NewIntArray()
+		for _, v := range departments {
+			departmentIdList.Append(gconv.Int(v["departmentId"]))
+		}
+		result := gconv.Map(info)
+		result["menuIdList"] = menuIdList.Slice()
+		result["departmentIdList"] = departmentIdList.Slice()
+		data = result
+		return
+	}
+	data = g.Map{}
+	return
+}
+
+// NewBaseSysRoleService create a new BaseSysRoleService
+func NewBaseSysRoleService() *BaseSysRoleService {
+	return &BaseSysRoleService{
+		Service: &cool.Service{
+			Model: model.NewBaseSysRole(),
+			ListQueryOp: &cool.QueryOp{
+				Where: func(ctx context.Context) [][]interface{} {
+					var (
+						admin   = cool.GetAdmin(ctx)
+						userId  = admin.UserId
+						roleIds = garray.NewIntArrayFromCopy(gconv.Ints(admin.RoleIds))
+					)
+					return [][]interface{}{
+						{"label != ?", g.Slice{"admin"}, true},
+						{"(userId=? or id in (?))", g.Slice{userId, admin.RoleIds}, !roleIds.Contains(1)},
+					}
+				},
+			},
+			PageQueryOp: &cool.QueryOp{
+				KeyWordField: []string{"name", "label"},
+				AddOrderby:   map[string]string{},
+				Where: func(ctx context.Context) [][]interface{} {
+					var (
+						admin   = cool.GetAdmin(ctx)
+						userId  = admin.UserId
+						roleIds = garray.NewIntArrayFromCopy(gconv.Ints(admin.RoleIds))
+					)
+					return [][]interface{}{
+						{"label != ?", g.Slice{"admin"}, true},
+						{"(userId=? or id in (?))", g.Slice{userId, admin.RoleIds}, !roleIds.Contains(1)},
+					}
+				},
+			},
+			InsertParam: func(ctx context.Context) map[string]interface{} {
+				return g.Map{"userId": cool.GetAdmin(ctx).UserId}
+			},
+			UniqueKey: map[string]string{
+				"name":  "角色名称不能重复",
+				"label": "角色标识不能重复",
+			},
+		},
+	}
+}