perms-system-server/internal/logic/pub/loginService.go

package pub

import (
	"context"
	"errors"
	"fmt"

	"perms-system-server/internal/consts"
	"perms-system-server/internal/loaders"
	authHelper "perms-system-server/internal/logic/auth"
	"perms-system-server/internal/model/user"
	"perms-system-server/internal/svc"

	"github.com/zeromicro/go-zero/core/limit"
	"golang.org/x/crypto/bcrypt"
)

// dummyBcryptHash 用于对不存在的用户名执行等时 bcrypt 比对，防止基于响应时间的用户名枚举
var dummyBcryptHash, _ = bcrypt.GenerateFromPassword([]byte("dummy-anti-timing"), bcrypt.DefaultCost)

type LoginResult struct {
	UserDetails  *loaders.UserDetails
	AccessToken  string
	RefreshToken string
}

type LoginError struct {
	Code    int
	Message string
}

func (e *LoginError) Error() string {
	return e.Message
}

func checkUsernameLimit(svcCtx *svc.ServiceContext, clientIP, username string) bool {
	if svcCtx.UsernameLoginLimit == nil {
		return false
	}
	key := fmt.Sprintf("%s:%s", clientIP, username)
	code, _ := svcCtx.UsernameLoginLimit.Take(key)
	return code == limit.OverQuota
}

func ValidateProductLogin(ctx context.Context, svcCtx *svc.ServiceContext, username, password, productCode, clientIP string) (*LoginResult, error) {
	if checkUsernameLimit(svcCtx, clientIP, username) {
		return nil, &LoginError{Code: 429, Message: "该账号登录尝试过于频繁，请5分钟后再试"}
	}

	u, err := svcCtx.SysUserModel.FindOneByUsername(ctx, username)
	if err != nil {
		if errors.Is(err, user.ErrNotFound) {
			bcrypt.CompareHashAndPassword(dummyBcryptHash, []byte(password))
			return nil, &LoginError{Code: 401, Message: "用户名或密码错误"}
		}
		return nil, err
	}

	if u.Status != consts.StatusEnabled {
		return nil, &LoginError{Code: 403, Message: "账号已被冻结"}
	}

	if err := bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password)); err != nil {
		return nil, &LoginError{Code: 401, Message: "用户名或密码错误"}
	}

	if u.IsSuperAdmin == consts.IsSuperAdminYes {
		return nil, &LoginError{Code: 403, Message: "超级管理员不允许通过产品端登录，请使用管理后台"}
	}

	product, err := svcCtx.SysProductModel.FindOneByCode(ctx, productCode)
	if err != nil {
		return nil, &LoginError{Code: 400, Message: "产品不存在"}
	}
	if product.Status != consts.StatusEnabled {
		return nil, &LoginError{Code: 403, Message: "该产品已被禁用"}
	}

	member, memberErr := svcCtx.SysProductMemberModel.FindOneByProductCodeUserId(ctx, productCode, u.Id)
	if memberErr != nil {
		return nil, &LoginError{Code: 403, Message: "您不是该产品的成员"}
	}
	if member.Status != consts.StatusEnabled {
		return nil, &LoginError{Code: 403, Message: "您在该产品下的成员资格已被禁用"}
	}

	ud := svcCtx.UserDetailsLoader.Load(ctx, u.Id, productCode)

	accessToken, err := authHelper.GenerateAccessToken(
		svcCtx.Config.Auth.AccessSecret,
		svcCtx.Config.Auth.AccessExpire,
		ud.UserId, ud.Username, ud.ProductCode, ud.MemberType, ud.TokenVersion,
	)
	if err != nil {
		return nil, err
	}

	refreshToken, err := authHelper.GenerateRefreshToken(
		svcCtx.Config.Auth.RefreshSecret,
		svcCtx.Config.Auth.RefreshExpire,
		ud.UserId, ud.ProductCode, ud.TokenVersion,
	)
	if err != nil {
		return nil, err
	}

	return &LoginResult{
		UserDetails:  ud,
		AccessToken:  accessToken,
		RefreshToken: refreshToken,
	}, nil
}