perms-system-server/internal/logic/role/roleListLogic.go

package role

import (
	"context"

	"perms-system-server/internal/middleware"
	roleModel "perms-system-server/internal/model/role"
	"perms-system-server/internal/response"
	"perms-system-server/internal/svc"
	"perms-system-server/internal/types"
	"perms-system-server/internal/util"

	"github.com/zeromicro/go-zero/core/logx"
)

type RoleListLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewRoleListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RoleListLogic {
	return &RoleListLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

// RoleList 角色列表。按产品分页查询角色信息，用于角色管理和用户角色分配的选择列表。
func (l *RoleListLogic) RoleList(req *types.RoleListReq) (resp *types.PageResp, err error) {
	page, pageSize := util.NormalizePage(req.Page, req.PageSize)

	caller := middleware.GetUserDetails(l.ctx)
	if caller == nil {
		return nil, response.ErrUnauthorized("未登录")
	}

	var productCode string
	if caller.IsSuperAdmin {
		productCode = req.ProductCode
	} else {
		productCode = middleware.GetProductCode(l.ctx)
		if productCode == "" {
			return nil, response.ErrForbidden("缺少产品上下文")
		}
	}

	var list []*roleModel.SysRole
	var total int64
	if productCode != "" {
		list, total, err = l.svcCtx.SysRoleModel.FindListByProductCode(l.ctx, productCode, page, pageSize)
	} else {
		list, total, err = l.svcCtx.SysRoleModel.FindListByPage(l.ctx, page, pageSize)
	}
	if err != nil {
		return nil, err
	}

	items := make([]types.RoleItem, 0, len(list))
	for _, r := range list {
		items = append(items, types.RoleItem{
			Id:          r.Id,
			ProductCode: r.ProductCode,
			Name:        r.Name,
			Remark:      r.Remark,
			Status:      r.Status,
			PermsLevel:  r.PermsLevel,
			CreateTime:  r.CreateTime,
		})
	}

	return &types.PageResp{
		Total: total,
		List:  items,
	}, nil
}