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

package member

import (
	"context"
	"time"

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

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

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

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

// UpdateMember 更新产品成员。修改成员类型或启用/禁用状态。降级最后一个 ADMIN 时会被拒绝以保证产品始终有管理员。
func (l *UpdateMemberLogic) UpdateMember(req *types.UpdateMemberReq) error {
	member, err := l.svcCtx.SysProductMemberModel.FindOne(l.ctx, req.Id)
	if err != nil {
		return response.ErrNotFound("成员不存在")
	}

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

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

	if member.MemberType == consts.MemberTypeAdmin && req.MemberType != consts.MemberTypeAdmin {
		adminCount, err := l.svcCtx.SysProductMemberModel.CountActiveAdmins(l.ctx, member.ProductCode)
		if err != nil {
			return err
		}
		if adminCount <= 1 {
			return response.ErrBadRequest("不能降级该产品的最后一个管理员")
		}
	}

	member.MemberType = req.MemberType
	if req.Status != 0 {
		if req.Status != consts.StatusEnabled && req.Status != consts.StatusDisabled {
			return response.ErrBadRequest("状态值无效，仅支持 1(启用) 和 2(禁用)")
		}
		member.Status = req.Status
	}
	member.UpdateTime = time.Now().Unix()

	if err := l.svcCtx.SysProductMemberModel.Update(l.ctx, member); err != nil {
		return err
	}

	l.svcCtx.UserDetailsLoader.Del(l.ctx, member.UserId, member.ProductCode)
	return nil
}