wlw/ai-agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(auth): simplify admin session token flow

Browse Source
This commit is contained in:
mlogclub
2026-04-30 17:42:24 +08:00
parent e763a75f78
commit fbafb13a8a
12 changed files with 411 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config/config.example.yaml
+1 -2
View File
@@ -15,8 +15,7 @@ logger:
addSource: false
auth:
accessTokenTTLHours: 12
refreshTokenTTLDays: 7
tokenTTLHours: 12
maxFailedAttempts: 5
credentialLockMinute: 15
internal/controllers/api/auth_controller.go
+1 -22
View File
@@ -30,20 +30,6 @@ func (c *AuthController) PostLogin() *web.JsonResult {
return web.JsonData(ret)
}
func (c *AuthController) PostRefresh_token() *web.JsonResult {
cfg := config.Current()
req := request.RefreshTokenRequest{}
if err := params.ReadJSON(c.Ctx, &req); err != nil {
return web.JsonError(err)
}
ret, err := services.AuthService.RefreshToken(req.RefreshToken, cfg.Auth, c.Ctx.RemoteAddr(), c.Ctx.GetHeader("User-Agent"))
if err != nil {
return web.JsonError(err)
}
return web.JsonData(ret)
}
func (c *AuthController) GetWxwork_login() {
loginURL, err := services.WxWorkLoginService.BuildWxWorkLoginURL(c.Ctx.URLParam("next"))
if err != nil {
@@ -91,14 +77,7 @@ func (c *AuthController) PostWxwork_exchange() *web.JsonResult {
}
func (c *AuthController) PostLogout() *web.JsonResult {
req := request.LogoutRequest{}
if c.Ctx.GetContentLength() > 0 {
if err := params.ReadJSON(c.Ctx, &req); err != nil {
return web.JsonError(err)
}
}
if err := services.AuthService.Logout(c.Ctx.GetHeader("Authorization"), req.RefreshToken); err != nil {
if err := services.AuthService.Logout(c.Ctx.GetHeader("Authorization")); err != nil {
return web.JsonError(err)
}
return web.JsonSuccess()
internal/controllers/dashboard/session_controller.go
-2
View File
@@ -24,7 +24,6 @@ func (c *SessionController) AnyList() *web.JsonResult {
cnd := params.NewPagedSqlCnd(c.Ctx,
params.QueryFilter{ParamName: "userId"},
params.QueryFilter{ParamName: "tokenType"},
params.QueryFilter{ParamName: "clientType"},
).Desc("id")
list, paging := services.LoginSessionService.FindPageByCnd(cnd)
@@ -38,7 +37,6 @@ func (c *SessionController) AnyList() *web.JsonResult {
ID: item.ID,
UserID: item.UserID,
Username: username,
TokenType: item.TokenType,
ClientType: item.ClientType,
ClientIP: item.ClientIP,
UserAgent: item.UserAgent,
internal/models/models.go
+19 -20
View File
@@ -287,31 +287,30 @@ type UserPermission struct {
AuditFields
}
// LoginSession 登录会话或刷新令牌记录。
// LoginSession 表示一次后台登录会话。
type LoginSession struct {
ID int64 `gorm:"primaryKey;autoIncrement"`
UserID int64 `gorm:"type:bigint;not null;index"`
TokenID string `gorm:"type:varchar(128);not null;uniqueIndex"`
TokenType string `gorm:"type:varchar(20);not null;default:'';index"`
ClientType string `gorm:"type:varchar(50);not null;default:'';index"`
ClientIP string `gorm:"type:varchar(64);not null;default:''"`
UserAgent string `gorm:"type:varchar(255);not null;default:''"`
ExpiredAt time.Time `gorm:"type:datetime;not null;index"`
RevokedAt *time.Time `gorm:"type:datetime;index"`
LastSeenAt *time.Time `gorm:"type:datetime"`
ID int64 `gorm:"primaryKey;autoIncrement"` // ID 为登录会话主键。
UserID int64 `gorm:"type:bigint;not null;index"` // UserID 为登录用户 ID。
Token string `gorm:"type:varchar(128);not null;uniqueIndex"` // Token 为随机不透明登录凭证，使用 ak_ 前缀。
ClientType string `gorm:"type:varchar(50);not null;default:'';index"` // ClientType 为客户端类型，后台 Web 端固定为 admin_web。
ClientIP string `gorm:"type:varchar(64);not null;default:''"` // ClientIP 为登录请求来源 IP。
UserAgent string `gorm:"type:varchar(255);not null;default:''"` // UserAgent 为登录请求浏览器或客户端 UA。
ExpiredAt time.Time `gorm:"type:datetime;not null;index"` // ExpiredAt 为 token 过期时间。
RevokedAt *time.Time `gorm:"type:datetime;index"` // RevokedAt 为主动注销或踢下线时间，非空表示已失效。
LastSeenAt *time.Time `gorm:"type:datetime"` // LastSeenAt 为最近一次成功鉴权时间。
AuditFields
}
// LoginCredentialLog 登录凭证校验日志。
// LoginCredentialLog 记录一次后台登录凭证校验结果。
type LoginCredentialLog struct {
ID int64 `gorm:"primaryKey;autoIncrement"`
Principal string `gorm:"type:varchar(100);not null;default:'';index"`
UserID int64 `gorm:"type:bigint;not null;default:0;index"`
Success bool `gorm:"not null;default:false;index"`
ClientIP string `gorm:"type:varchar(64);not null;default:''"`
UserAgent string `gorm:"type:varchar(255);not null;default:''"`
Reason string `gorm:"type:varchar(255);not null;default:''"`
CreatedAt time.Time `gorm:"type:datetime;not null;index"`
ID int64 `gorm:"primaryKey;autoIncrement"` // ID 为登录凭证日志主键。
Principal string `gorm:"type:varchar(100);not null;default:'';index"` // Principal 为用户输入的登录名。
UserID int64 `gorm:"type:bigint;not null;default:0;index"` // UserID 为匹配到的用户 ID，未匹配时为 0。
Success bool `gorm:"not null;default:false;index"` // Success 表示本次凭证校验是否成功。
ClientIP string `gorm:"type:varchar(64);not null;default:''"` // ClientIP 为登录请求来源 IP。
UserAgent string `gorm:"type:varchar(255);not null;default:''"` // UserAgent 为登录请求浏览器或客户端 UA。
Reason string `gorm:"type:varchar(255);not null;default:''"` // Reason 为校验结果原因。
CreatedAt time.Time `gorm:"type:datetime;not null;index"` // CreatedAt 为日志创建时间。
}
// Asset 存储的文件资源，如上传的附件等。
internal/pkg/config/config.go
+1 -2
View File
@@ -55,8 +55,7 @@ type LoggerConfig struct {
}
type AuthConfig struct {
AccessTokenTTLHours int `yaml:"accessTokenTTLHours"`
RefreshTokenTTLDays int `yaml:"refreshTokenTTLDays"`
TokenTTLHours int `yaml:"tokenTTLHours"`
MaxFailedAttempts int `yaml:"maxFailedAttempts"`
CredentialLockMinute int `yaml:"credentialLockMinute"`
}
internal/pkg/constants/auth.go
+1 -7
View File
@@ -8,13 +8,7 @@ const (
)
const (
AccessTokenPrefix = "atk_"
RefreshTokenPrefix = "rtk_"
)
const (
TokenTypeAccess = "access"
TokenTypeRefresh = "refresh"
AuthTokenPrefix = "ak_"
)
const (
internal/pkg/dto/request/auth_request.go
-8
View File
@@ -5,14 +5,6 @@ type LoginRequest struct {
Password string `json:"password"`
}
type RefreshTokenRequest struct {
RefreshToken string `json:"refreshToken"`
}
type LogoutRequest struct {
RefreshToken string `json:"refreshToken"`
}
type WxWorkExchangeRequest struct {
Ticket string `json:"ticket"`
}
internal/pkg/dto/response/admin_response.go
+3 -4
View File
@@ -47,12 +47,11 @@ type CreateUserResultResponse struct {
type SessionResponse struct {
ID int64 `json:"id"`
UserID int64 `json:"userId"`
Username string `json:"username,omitempty"`
TokenType string `json:"tokenType"`
Username string `json:"username"`
ClientType string `json:"clientType"`
ClientIP string `json:"clientIp"`
UserAgent string `json:"userAgent"`
ExpiredAt string `json:"expiredAt"`
RevokedAt string `json:"revokedAt,omitempty"`
LastSeenAt string `json:"lastSeenAt,omitempty"`
RevokedAt string `json:"revokedAt"`
LastSeenAt string `json:"lastSeenAt"`
}
internal/pkg/dto/response/auth_response.go
-1
View File
@@ -13,7 +13,6 @@ type AuthUserResponse struct {
type LoginResponse struct {
AccessToken string `json:"accessToken"`
RefreshToken string `json:"refreshToken"`
ExpiresAt string `json:"expiresAt"`
User *AuthUserResponse `json:"user"`
Permissions []string `json:"permissions"`
internal/pkg/errorsx/errors.go
+5
View File
@@ -9,6 +9,7 @@ const (
CodeAuthForbidden = 3001
CodeAuthInvalidToken = 3002
CodeAuthInvalidAccount = 3003
CodeAuthCredentialLocked = 3004
)
func InvalidParam(message string) error {
@@ -34,3 +35,7 @@ func InvalidToken(message string) error {
func InvalidAccount(message string) error {
return web.NewError(CodeAuthInvalidAccount, message)
}
func CredentialLocked(message string) error {
return web.NewError(CodeAuthCredentialLocked, message)
}
internal/services/auth_service.go
+43 -98
View File
@@ -87,6 +87,14 @@ func (s *authService) Login(req request.LoginRequest, authCfg config.AuthConfig,
}
user := UserService.GetByUsername(username)
if s.isCredentialLocked(username, authCfg) {
userID := int64(0)
if user != nil {
userID = user.ID
}
_ = s.createLoginCredentialLog(username, userID, false, clientIP, userAgent, "credential locked")
return nil, errorsx.CredentialLocked("登录失败次数过多，请稍后再试")
}
if user == nil || user.Status != enums.StatusOk {
_ = s.createLoginCredentialLog(username, 0, false, clientIP, userAgent, "user not found")
return nil, errorsx.InvalidAccount("用户名或密码错误")
@@ -121,56 +129,11 @@ func (s *authService) Login(req request.LoginRequest, authCfg config.AuthConfig,
return ret, nil
}
func (s *authService) RefreshToken(refreshToken string, authCfg config.AuthConfig, clientIP, userAgent string) (*response.LoginResponse, error) {
session, err := s.validateSessionToken(refreshToken, constants.TokenTypeRefresh)
if err != nil {
return nil, err
}
if session.RevokedAt != nil {
return nil, errorsx.InvalidToken("refresh token 已失效")
}
user := UserService.Get(session.UserID)
if user == nil || user.Status != enums.StatusOk {
return nil, errorsx.Unauthorized("用户不存在或已被禁用")
}
var ret *response.LoginResponse
if err := sqls.WithTransaction(func(ctx *sqls.TxContext) error {
var dbErr error
if dbErr = repositories.LoginSessionRepository.Updates(ctx.Tx, session.ID, map[string]any{
"revoked_at": time.Now(),
"update_user_id": user.ID,
"update_user_name": user.Username,
"updated_at": time.Now(),
}); dbErr != nil {
return dbErr
}
if ret, dbErr = s.issueTokens(ctx, user, clientIP, userAgent, authCfg); dbErr != nil {
return dbErr
}
return nil
}); err != nil {
return nil, err
}
return ret, nil
}
func (s *authService) Logout(accessToken, refreshToken string) error {
func (s *authService) Logout(accessToken string) error {
accessToken = s.extractBearerToken(accessToken)
now := time.Now()
if accessToken != "" {
if session := LoginSessionService.FindOne(sqls.NewCnd().Eq("token_id", accessToken).Eq("token_type", constants.TokenTypeAccess)); session != nil && session.RevokedAt == nil {
if err := LoginSessionService.Updates(session.ID, map[string]any{
"revoked_at": now,
"updated_at": now,
}); err != nil {
return err
}
}
}
if refreshToken != "" {
if session := LoginSessionService.FindOne(sqls.NewCnd().Eq("token_id", refreshToken).Eq("token_type", constants.TokenTypeRefresh)); session != nil && session.RevokedAt == nil {
if session := LoginSessionService.FindOne(sqls.NewCnd().Eq("token", accessToken)); session != nil && session.RevokedAt == nil {
if err := LoginSessionService.Updates(session.ID, map[string]any{
"revoked_at": now,
"updated_at": now,
@@ -195,7 +158,7 @@ func (s *authService) Authenticate(ctx iris.Context) (*dto.AuthPrincipal, error)
return nil, errorsx.Unauthorized("未登录或登录已过期")
}
session, err := s.validateSessionToken(token, constants.TokenTypeAccess)
session, err := s.validateSessionToken(token)
if err != nil {
return nil, err
}
@@ -262,48 +225,20 @@ func (s *authService) issueTokens(ctx *sqls.TxContext, user *models.User, client
return nil, err
}
accessTTL, refreshTTL := s.resolveTokenTTL(authCfg)
accessToken, err := randomToken(constants.AccessTokenPrefix)
if err != nil {
return nil, err
}
refreshToken, err := randomToken(constants.RefreshTokenPrefix)
tokenTTL := s.resolveTokenTTL(authCfg)
accessToken, err := randomToken(constants.AuthTokenPrefix)
if err != nil {
return nil, err
}
now := time.Now()
// accessSession
if err := repositories.LoginSessionRepository.Create(ctx.Tx, &models.LoginSession{
UserID: user.ID,
TokenID: accessToken,
TokenType: constants.TokenTypeAccess,
Token: accessToken,
ClientType: constants.ClientTypeAdminWeb,
ClientIP: clientIP,
UserAgent: userAgent,
ExpiredAt: now.Add(accessTTL),
LastSeenAt: &now,
AuditFields: models.AuditFields{
CreatedAt: now,
CreateUserID: user.ID,
CreateUserName: user.Username,
UpdatedAt: now,
UpdateUserID: user.ID,
UpdateUserName: user.Username,
},
}); err != nil {
return nil, err
}
// refreshSession
if err := repositories.LoginSessionRepository.Create(ctx.Tx, &models.LoginSession{
UserID: user.ID,
TokenID: refreshToken,
TokenType: constants.TokenTypeRefresh,
ClientType: constants.ClientTypeAdminWeb,
ClientIP: clientIP,
UserAgent: userAgent,
ExpiredAt: now.Add(refreshTTL),
ExpiredAt: now.Add(tokenTTL),
LastSeenAt: &now,
AuditFields: models.AuditFields{
CreatedAt: now,
@@ -319,8 +254,7 @@ func (s *authService) issueTokens(ctx *sqls.TxContext, user *models.User, client
return &response.LoginResponse{
AccessToken: accessToken,
RefreshToken: refreshToken,
ExpiresAt: now.Add(accessTTL).Format(time.DateTime),
ExpiresAt: now.Add(tokenTTL).Format(time.DateTime),
User: &response.AuthUserResponse{
ID: user.ID,
Username: user.Username,
@@ -334,33 +268,27 @@ func (s *authService) issueTokens(ctx *sqls.TxContext, user *models.User, client
}, nil
}
func (s *authService) resolveTokenTTL(authCfg config.AuthConfig) (time.Duration, time.Duration) {
accessTTL := 12 * time.Hour
refreshTTL := 7 * 24 * time.Hour
if authCfg.AccessTokenTTLHours > 0 {
accessTTL = time.Duration(authCfg.AccessTokenTTLHours) * time.Hour
func (s *authService) resolveTokenTTL(authCfg config.AuthConfig) time.Duration {
tokenTTL := 12 * time.Hour
if authCfg.TokenTTLHours > 0 {
tokenTTL = time.Duration(authCfg.TokenTTLHours) * time.Hour
}
if authCfg.RefreshTokenTTLDays > 0 {
refreshTTL = time.Duration(authCfg.RefreshTokenTTLDays) * 24 * time.Hour
}
return accessTTL, refreshTTL
return tokenTTL
}
func (s *authService) validateSessionToken(token, tokenType string) (*models.LoginSession, error) {
func (s *authService) validateSessionToken(token string) (*models.LoginSession, error) {
if strings.TrimSpace(token) == "" {
return nil, errorsx.InvalidToken("token 不能为空")
}
session := LoginSessionService.FindOne(sqls.NewCnd().
Eq("token_id", token).
Eq("token_type", tokenType))
session := LoginSessionService.FindOne(sqls.NewCnd().Eq("token", token))
if session == nil {
return nil, errorsx.InvalidToken("token 无效")
return nil, errorsx.InvalidToken("登录凭证无效")
}
if session.RevokedAt != nil {
return nil, errorsx.InvalidToken("token 已失效")
return nil, errorsx.InvalidToken("登录凭证已失效")
}
if time.Now().After(session.ExpiredAt) {
return nil, errorsx.InvalidToken("token 已过期")
return nil, errorsx.InvalidToken("登录凭证已过期")
}
return session, nil
}
@@ -480,6 +408,23 @@ func (s *authService) createLoginCredentialLog(principal string, userID int64, s
})
}
func (s *authService) isCredentialLocked(username string, authCfg config.AuthConfig) bool {
maxFailedAttempts := authCfg.MaxFailedAttempts
if maxFailedAttempts <= 0 {
maxFailedAttempts = 5
}
lockMinute := authCfg.CredentialLockMinute
if lockMinute <= 0 {
lockMinute = 15
}
since := time.Now().Add(-time.Duration(lockMinute) * time.Minute)
return LoginCredentialLogService.Count(sqls.NewCnd().
Eq("principal", username).
Eq("success", false).
NotEq("reason", "credential locked").
Where("created_at >= ?", since)) >= int64(maxFailedAttempts)
}
func randomToken(prefix string) (string, error) {
buf := make([]byte, 24)
if _, err := rand.Read(buf); err != nil {
internal/services/auth_service_test.go
+325 -1
View File
@@ -1,6 +1,24 @@
package services
import "testing"
import (
"errors"
"strings"
"testing"
"time"
"cs-agent/internal/models"
"cs-agent/internal/pkg/config"
"cs-agent/internal/pkg/dto/request"
"cs-agent/internal/pkg/enums"
"cs-agent/internal/pkg/errorsx"
"github.com/glebarez/sqlite"
"github.com/mlogclub/simple/sqls"
"github.com/mlogclub/simple/web"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
func TestExtractBearerToken(t *testing.T) {
svc := newAuthService()
@@ -13,3 +31,309 @@ func TestExtractBearerToken(t *testing.T) {
t.Fatalf("expected raw token to be rejected by bearer extractor, got %q", got)
}
}
func TestAuthServiceLoginCreatesSingleAccessSession(t *testing.T) {
db := setupAuthServiceTestDB(t)
user := createAuthTestUser(t, db, "admin", "secret")
svc := newAuthService()
ret, err := svc.Login(request.LoginRequest{
Username: " admin ",
Password: "secret",
}, config.AuthConfig{TokenTTLHours: 2, MaxFailedAttempts: 5, CredentialLockMinute: 15}, "127.0.0.1", "go-test")
if err != nil {
t.Fatalf("login failed: %v", err)
}
if ret.AccessToken == "" || !strings.HasPrefix(ret.AccessToken, "ak_") {
t.Fatalf("expected ak_ access token, got %q", ret.AccessToken)
}
if ret.ExpiresAt == "" {
t.Fatal("expected expiresAt to be returned")
}
var sessions []models.LoginSession
if err := db.Find(&sessions).Error; err != nil {
t.Fatalf("query login sessions: %v", err)
}
if len(sessions) != 1 {
t.Fatalf("expected exactly one session, got %d", len(sessions))
}
if sessions[0].Token != ret.AccessToken {
t.Fatalf("expected session token %q, got %q", ret.AccessToken, sessions[0].Token)
}
if sessions[0].UserID != user.ID {
t.Fatalf("expected session user %d, got %d", user.ID, sessions[0].UserID)
}
if sessions[0].ClientType != "admin_web" {
t.Fatalf("expected admin_web client type, got %q", sessions[0].ClientType)
}
logs := findCredentialLogs(t, db)
if len(logs) != 1 {
t.Fatalf("expected one credential log, got %d", len(logs))
}
if !logs[0].Success || logs[0].Principal != "admin" || logs[0].UserID != user.ID {
t.Fatalf("unexpected success credential log: %+v", logs[0])
}
}
func TestAuthServiceLoginFailureWritesCredentialLogs(t *testing.T) {
db := setupAuthServiceTestDB(t)
createAuthTestUser(t, db, "admin", "secret")
svc := newAuthService()
authCfg := config.AuthConfig{TokenTTLHours: 2, MaxFailedAttempts: 5, CredentialLockMinute: 15}
if _, err := svc.Login(request.LoginRequest{Username: "missing", Password: "secret"}, authCfg, "127.0.0.1", "go-test"); !hasCode(err, errorsx.CodeAuthInvalidAccount) {
t.Fatalf("expected invalid account for missing user, got %v", err)
}
if _, err := svc.Login(request.LoginRequest{Username: "admin", Password: "wrong"}, authCfg, "127.0.0.1", "go-test"); !hasCode(err, errorsx.CodeAuthInvalidAccount) {
t.Fatalf("expected invalid account for password mismatch, got %v", err)
}
logs := findCredentialLogs(t, db)
if len(logs) != 2 {
t.Fatalf("expected two credential logs, got %d", len(logs))
}
if logs[0].Reason != "user not found" || logs[0].Success {
t.Fatalf("unexpected missing-user log: %+v", logs[0])
}
if logs[1].Reason != "password mismatch" || logs[1].Success {
t.Fatalf("unexpected password-mismatch log: %+v", logs[1])
}
}
func TestAuthServiceLoginCredentialLockout(t *testing.T) {
db := setupAuthServiceTestDB(t)
user := createAuthTestUser(t, db, "admin", "secret")
now := time.Now()
for i := 0; i < 2; i++ {
if err := db.Create(&models.LoginCredentialLog{
Principal: "admin",
UserID: user.ID,
Success: false,
Reason: "password mismatch",
CreatedAt: now.Add(-time.Duration(i+1) * time.Minute),
}).Error; err != nil {
t.Fatalf("seed credential log: %v", err)
}
}
if err := db.Create(&models.LoginCredentialLog{
Principal: "admin",
UserID: user.ID,
Success: false,
Reason: "password mismatch",
CreatedAt: now.Add(-30 * time.Minute),
}).Error; err != nil {
t.Fatalf("seed old credential log: %v", err)
}
svc := newAuthService()
_, err := svc.Login(request.LoginRequest{Username: "admin", Password: "secret"}, config.AuthConfig{
TokenTTLHours: 2,
MaxFailedAttempts: 2,
CredentialLockMinute: 15,
}, "127.0.0.1", "go-test")
if !hasCode(err, errorsx.CodeAuthCredentialLocked) {
t.Fatalf("expected credential locked error, got %v", err)
}
var lockedLog models.LoginCredentialLog
if err := db.Order("id DESC").Take(&lockedLog).Error; err != nil {
t.Fatalf("query latest credential log: %v", err)
}
if lockedLog.Reason != "credential locked" || lockedLog.Success {
t.Fatalf("unexpected locked credential log: %+v", lockedLog)
}
var sessionCount int64
if err := db.Model(&models.LoginSession{}).Count(&sessionCount).Error; err != nil {
t.Fatalf("count sessions: %v", err)
}
if sessionCount != 0 {
t.Fatalf("expected no session while credential locked, got %d", sessionCount)
}
}
func TestValidateSessionTokenStates(t *testing.T) {
db := setupAuthServiceTestDB(t)
svc := newAuthService()
now := time.Now()
if _, err := svc.validateSessionToken("missing"); !hasCode(err, errorsx.CodeAuthInvalidToken) {
t.Fatalf("expected invalid token for missing session, got %v", err)
}
revokedAt := now
if err := db.Create(&models.LoginSession{
UserID: 1,
Token: "ak_revoked",
ClientType: "admin_web",
ExpiredAt: now.Add(time.Hour),
RevokedAt: &revokedAt,
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
}).Error; err != nil {
t.Fatalf("seed revoked session: %v", err)
}
if _, err := svc.validateSessionToken("ak_revoked"); !hasCode(err, errorsx.CodeAuthInvalidToken) {
t.Fatalf("expected invalid token for revoked session, got %v", err)
}
if err := db.Create(&models.LoginSession{
UserID: 1,
Token: "ak_expired",
ClientType: "admin_web",
ExpiredAt: now.Add(-time.Hour),
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
}).Error; err != nil {
t.Fatalf("seed expired session: %v", err)
}
if _, err := svc.validateSessionToken("ak_expired"); !hasCode(err, errorsx.CodeAuthInvalidToken) {
t.Fatalf("expected invalid token for expired session, got %v", err)
}
if err := db.Create(&models.LoginSession{
UserID: 1,
Token: "ak_valid",
ClientType: "admin_web",
ExpiredAt: now.Add(time.Hour),
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
}).Error; err != nil {
t.Fatalf("seed valid session: %v", err)
}
session, err := svc.validateSessionToken("ak_valid")
if err != nil {
t.Fatalf("expected valid session token, got %v", err)
}
if session.Token != "ak_valid" {
t.Fatalf("expected valid session token ak_valid, got %q", session.Token)
}
}
func TestAuthServiceLogoutRevokesCurrentTokenOnly(t *testing.T) {
db := setupAuthServiceTestDB(t)
now := time.Now()
sessions := []models.LoginSession{
{
UserID: 1,
Token: "ak_current",
ClientType: "admin_web",
ExpiredAt: now.Add(time.Hour),
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
},
{
UserID: 1,
Token: "ak_other",
ClientType: "admin_web",
ExpiredAt: now.Add(time.Hour),
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
},
}
if err := db.Create(&sessions).Error; err != nil {
t.Fatalf("seed sessions: %v", err)
}
if err := newAuthService().Logout("Bearer ak_current"); err != nil {
t.Fatalf("logout failed: %v", err)
}
var current models.LoginSession
if err := db.Take(&current, "token = ?", "ak_current").Error; err != nil {
t.Fatalf("query current session: %v", err)
}
if current.RevokedAt == nil {
t.Fatal("expected current session to be revoked")
}
var other models.LoginSession
if err := db.Take(&other, "token = ?", "ak_other").Error; err != nil {
t.Fatalf("query other session: %v", err)
}
if other.RevokedAt != nil {
t.Fatal("expected other session to remain active")
}
}
func setupAuthServiceTestDB(t *testing.T) *gorm.DB {
t.Helper()
db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{
NamingStrategy: schema.NamingStrategy{
TablePrefix: "t_",
SingularTable: true,
},
})
if err != nil {
t.Fatalf("open sqlite db: %v", err)
}
if err := db.AutoMigrate(
&models.User{},
&models.Role{},
&models.Permission{},
&models.UserRole{},
&models.RolePermission{},
&models.UserPermission{},
&models.LoginSession{},
&models.LoginCredentialLog{},
); err != nil {
t.Fatalf("migrate auth tables: %v", err)
}
sqls.SetDB(db)
return db
}
func createAuthTestUser(t *testing.T, db *gorm.DB, username, password string) *models.User {
t.Helper()
passwordHash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
if err != nil {
t.Fatalf("hash password: %v", err)
}
now := time.Now()
user := &models.User{
Username: username,
Nickname: username,
Password: string(passwordHash),
Status: enums.StatusOk,
AuditFields: models.AuditFields{
CreatedAt: now,
UpdatedAt: now,
},
}
if err := db.Create(user).Error; err != nil {
t.Fatalf("create auth test user: %v", err)
}
return user
}
func findCredentialLogs(t *testing.T, db *gorm.DB) []models.LoginCredentialLog {
t.Helper()
var logs []models.LoginCredentialLog
if err := db.Order("id ASC").Find(&logs).Error; err != nil {
t.Fatalf("query credential logs: %v", err)
}
return logs
}
func hasCode(err error, code int) bool {
if err == nil {
return false
}
var codeErr *web.CodeError
if errors.As(err, &codeErr) {
return codeErr.Code == code
}
return false
}