130 lines
3.8 KiB
Go
130 lines
3.8 KiB
Go
package auth
|
|
|
|
import (
|
|
"backend/config"
|
|
"backend/internal/dto"
|
|
"backend/internal/model"
|
|
"backend/utils"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// AuthService 认证业务逻辑层 // Logan.Li
|
|
type AuthService struct {
|
|
db *gorm.DB
|
|
cfg *config.Config
|
|
}
|
|
|
|
// 登录防护参数：连续失败 5 次锁定 15 分钟（账号级，与 IP 限流互补） // Logan.Li
|
|
const (
|
|
maxLoginAttempts = 5
|
|
loginLockDuration = 15 * time.Minute
|
|
)
|
|
|
|
// NewAuthService 创建 service 实例 // Logan.Li
|
|
func NewAuthService(db *gorm.DB, cfg *config.Config) *AuthService {
|
|
return &AuthService{db: db, cfg: cfg}
|
|
}
|
|
|
|
// Login 校验账号密码并签发 JWT。
|
|
// 含账号级防护：连续失败达上限后锁定一段时间；成功登录清零计数。 // Logan.Li
|
|
func (s *AuthService) Login(req dto.LoginRequest) (*dto.LoginResponse, error) {
|
|
var user model.User
|
|
if err := s.db.Where("username = ?", req.Username).First(&user).Error; err != nil {
|
|
return nil, errors.New("用户名或密码错误")
|
|
}
|
|
|
|
if !user.IsActive {
|
|
return nil, errors.New("账号已被禁用")
|
|
}
|
|
|
|
// 账号锁定检查（到点自动解锁） // Logan.Li
|
|
if user.LockedUntil != nil && user.LockedUntil.After(time.Now()) {
|
|
remaining := int(time.Until(*user.LockedUntil).Minutes()) + 1
|
|
return nil, fmt.Errorf("失败次数过多，账号已锁定，请约 %d 分钟后再试", remaining)
|
|
}
|
|
|
|
if !utils.CheckPassword(req.Password, user.Password) {
|
|
s.registerLoginFailure(&user)
|
|
return nil, errors.New("用户名或密码错误")
|
|
}
|
|
|
|
// 登录成功：清零失败计数与锁定 // Logan.Li
|
|
if user.FailedLoginAttempts > 0 || user.LockedUntil != nil {
|
|
s.db.Model(&user).Updates(map[string]interface{}{
|
|
"failed_login_attempts": 0,
|
|
"locked_until": nil,
|
|
})
|
|
}
|
|
|
|
// 更新最后登录时间 // Logan.Li
|
|
now := time.Now()
|
|
s.db.Model(&user).Update("last_login", now)
|
|
|
|
// 签发 JWT // Logan.Li
|
|
token, err := utils.GenerateToken(user.ID, user.Username, user.Role, s.cfg.JWTSecret)
|
|
if err != nil {
|
|
return nil, errors.New("令牌生成失败")
|
|
}
|
|
|
|
return &dto.LoginResponse{
|
|
Token: token,
|
|
UserInfo: dto.UserInfoResponse{
|
|
ID: user.ID,
|
|
Username: user.Username,
|
|
Email: user.Email,
|
|
Avatar: user.Avatar,
|
|
Role: user.Role,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// registerLoginFailure 记录一次登录失败，达到上限则锁定账号 // Logan.Li
|
|
func (s *AuthService) registerLoginFailure(user *model.User) {
|
|
attempts := user.FailedLoginAttempts + 1
|
|
updates := map[string]interface{}{"failed_login_attempts": attempts}
|
|
if attempts >= maxLoginAttempts {
|
|
lockedUntil := time.Now().Add(loginLockDuration)
|
|
updates["locked_until"] = &lockedUntil
|
|
updates["failed_login_attempts"] = 0 // 锁定期结束后重新计数 // Logan.Li
|
|
}
|
|
s.db.Model(user).Updates(updates)
|
|
}
|
|
|
|
// GetCurrentUser 根据 ID 获取当前用户信息 // Logan.Li
|
|
func (s *AuthService) GetCurrentUser(userID uint) (*dto.UserInfoResponse, error) {
|
|
var user model.User
|
|
if err := s.db.First(&user, userID).Error; err != nil {
|
|
return nil, errors.New("用户不存在")
|
|
}
|
|
|
|
return &dto.UserInfoResponse{
|
|
ID: user.ID,
|
|
Username: user.Username,
|
|
Email: user.Email,
|
|
Avatar: user.Avatar,
|
|
Role: user.Role,
|
|
}, nil
|
|
}
|
|
|
|
// ChangePassword 修改当前用户密码；同时刷新会话失效基线，使所有旧令牌立即作废 // Logan.Li
|
|
func (s *AuthService) ChangePassword(userID uint, req dto.ChangePasswordRequest) error {
|
|
var user model.User
|
|
if err := s.db.First(&user, userID).Error; err != nil {
|
|
return errors.New("用户不存在")
|
|
}
|
|
|
|
if !utils.CheckPassword(req.OldPassword, user.Password) {
|
|
return errors.New("原密码错误")
|
|
}
|
|
|
|
user.Password = utils.HashPassword(req.NewPassword)
|
|
user.TokenValidAfter = time.Now() // 旧 JWT 全部失效，需重新登录 // Logan.Li
|
|
user.FailedLoginAttempts = 0
|
|
user.LockedUntil = nil
|
|
return s.db.Save(&user).Error
|
|
}
|