68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package authser
|
|
|
|
import "github.com/gin-gonic/gin"
|
|
|
|
const requestAuthStateKey = "authser.request-auth-state"
|
|
|
|
// requestAuthState 只在单个 Gin 请求内有效，用来避免多个 SSR 服务重复刷新同一会话。
|
|
// freshUser 表示本请求已经从上游取得并写回过的最新用户；suppressAutoAuth
|
|
// 表示本请求明确不应再自动创建游客（例如用户刚退出或正式账号已过期）。
|
|
type requestAuthState struct {
|
|
freshUser *LoginResp
|
|
suppressAutoAuth bool
|
|
}
|
|
|
|
func authState(c *gin.Context) *requestAuthState {
|
|
if c == nil {
|
|
return nil
|
|
}
|
|
if value, ok := c.Get(requestAuthStateKey); ok {
|
|
if state, ok := value.(*requestAuthState); ok && state != nil {
|
|
return state
|
|
}
|
|
}
|
|
state := &requestAuthState{}
|
|
c.Set(requestAuthStateKey, state)
|
|
return state
|
|
}
|
|
|
|
func freshRequestAuthUser(c *gin.Context) *LoginResp {
|
|
state := authState(c)
|
|
if state == nil {
|
|
return nil
|
|
}
|
|
return state.freshUser
|
|
}
|
|
|
|
func rememberRequestAuthUser(c *gin.Context, user *LoginResp) {
|
|
state := authState(c)
|
|
if state == nil {
|
|
return
|
|
}
|
|
state.freshUser = user
|
|
state.suppressAutoAuth = false
|
|
}
|
|
|
|
func forgetRequestAuthUser(c *gin.Context) {
|
|
if state := authState(c); state != nil {
|
|
state.freshUser = nil
|
|
}
|
|
}
|
|
|
|
func suppressRequestAutoAuth(c *gin.Context) {
|
|
if state := authState(c); state != nil {
|
|
state.suppressAutoAuth = true
|
|
}
|
|
}
|
|
|
|
func allowRequestAutoAuth(c *gin.Context) {
|
|
if state := authState(c); state != nil {
|
|
state.suppressAutoAuth = false
|
|
}
|
|
}
|
|
|
|
func requestAutoAuthSuppressed(c *gin.Context) bool {
|
|
state := authState(c)
|
|
return state != nil && state.suppressAutoAuth
|
|
}
|