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

package role

import (
	"context"
	"strings"
	"time"

	"perms-system-server/internal/consts"
	authHelper "perms-system-server/internal/logic/auth"
	roleModel "perms-system-server/internal/model/role"
	"perms-system-server/internal/response"
	"perms-system-server/internal/svc"
	"perms-system-server/internal/types"

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

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

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

func (l *CreateRoleLogic) CreateRole(req *types.CreateRoleReq) (resp *types.IdResp, err error) {
	if err := authHelper.RequireProductAdminFor(l.ctx, req.ProductCode); err != nil {
		return nil, err
	}

	product, err := l.svcCtx.SysProductModel.FindOneByCode(l.ctx, req.ProductCode)
	if err != nil {
		return nil, response.ErrNotFound("产品不存在")
	}
	if product.Status != consts.StatusEnabled {
		return nil, response.ErrBadRequest("产品已被禁用，无法创建角色")
	}

	if len(req.Name) > 64 {
		return nil, response.ErrBadRequest("角色名长度不能超过64个字符")
	}
	if len(req.Remark) > 255 {
		return nil, response.ErrBadRequest("备注长度不能超过255个字符")
	}

	if req.PermsLevel < 1 || req.PermsLevel > 999 {
		return nil, response.ErrBadRequest("权限级别必须在 1-999 之间")
	}

	now := time.Now().Unix()
	result, err := l.svcCtx.SysRoleModel.Insert(l.ctx, &roleModel.SysRole{
		ProductCode: req.ProductCode,
		Name:        req.Name,
		Remark:      req.Remark,
		Status:      consts.StatusEnabled,
		PermsLevel:  req.PermsLevel,
		CreateTime:  now,
		UpdateTime:  now,
	})
	if err != nil {
		if strings.Contains(err.Error(), "1062") || strings.Contains(err.Error(), "Duplicate entry") {
			return nil, response.ErrConflict("该产品下角色名已存在")
		}
		return nil, err
	}

	id, _ := result.LastInsertId()
	return &types.IdResp{Id: id}, nil
}