perms-system-server/internal/logic/user/updateUserStatusLogic.go

package user

import (
	"context"

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

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

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

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

func (l *UpdateUserStatusLogic) UpdateUserStatus(req *types.UpdateUserStatusReq) error {
	if req.Status != consts.StatusEnabled && req.Status != consts.StatusDisabled {
		return response.ErrBadRequest("状态值无效，仅支持 1(启用) 和 2(冻结)")
	}

	callerId := middleware.GetUserId(l.ctx)
	if callerId == req.Id {
		return response.ErrBadRequest("不能修改自己的状态")
	}

	user, err := l.svcCtx.SysUserModel.FindOne(l.ctx, req.Id)
	if err != nil {
		return response.ErrNotFound("用户不存在")
	}

	if user.IsSuperAdmin == consts.IsSuperAdminYes {
		return response.ErrForbidden("不能修改超级管理员的状态")
	}

	productCode := middleware.GetProductCode(l.ctx)
	if productCode != "" {
		caller := middleware.GetUserDetails(l.ctx)
		if caller != nil && !caller.IsSuperAdmin {
			if _, err := l.svcCtx.SysProductMemberModel.FindOneByProductCodeUserId(l.ctx, productCode, req.Id); err != nil {
				return response.ErrBadRequest("目标用户不是当前产品的成员")
			}
		}
	}
	if err := authHelper.CheckManageAccess(l.ctx, l.svcCtx, req.Id, productCode); err != nil {
		return err
	}

	if err := l.svcCtx.SysUserModel.UpdateStatus(l.ctx, req.Id, req.Status); err != nil {
		return err
	}

	l.svcCtx.UserDetailsLoader.Clean(l.ctx, req.Id)
	return nil
}