perms-system-server/internal/logic/member/addMemberLogic.go

package member

import (
	"context"
	"strings"
	"time"

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

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

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

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

func (l *AddMemberLogic) AddMember(req *types.AddMemberReq) (resp *types.IdResp, err error) {
	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 _, err := l.svcCtx.SysUserModel.FindOne(l.ctx, req.UserId); err != nil {
		return nil, response.ErrNotFound("用户不存在")
	}

	if req.MemberType != consts.MemberTypeAdmin &&
		req.MemberType != consts.MemberTypeDeveloper &&
		req.MemberType != consts.MemberTypeMember {
		return nil, response.ErrBadRequest("无效的成员类型")
	}

	if err := authHelper.RequireProductAdminFor(l.ctx, req.ProductCode); err != nil {
		return nil, err
	}
	if err := authHelper.CheckMemberTypeAssignment(l.ctx, req.MemberType); err != nil {
		return nil, err
	}

	_, findErr := l.svcCtx.SysProductMemberModel.FindOneByProductCodeUserId(l.ctx, req.ProductCode, req.UserId)
	if findErr == nil {
		return nil, response.ErrConflict("该用户已是该产品成员")
	}

	now := time.Now().Unix()
	result, err := l.svcCtx.SysProductMemberModel.Insert(l.ctx, &productmember.SysProductMember{
		ProductCode: req.ProductCode,
		UserId:      req.UserId,
		MemberType:  req.MemberType,
		Status:      consts.StatusEnabled,
		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
	}

	l.svcCtx.UserDetailsLoader.Del(l.ctx, req.UserId, req.ProductCode)

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