58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
// Copyright 2022 E99p1ant. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package context
|
|
|
|
import (
|
|
"github.com/flamego/flamego"
|
|
"github.com/flamego/session"
|
|
|
|
"github.com/tamakyi/TamaBox/internal/db"
|
|
"github.com/tamakyi/TamaBox/internal/i18n"
|
|
)
|
|
|
|
// authenticatedUser returns the user object of the authenticated user.
|
|
func authenticatedUser(ctx flamego.Context, sess session.Session) *db.User {
|
|
uid, ok := sess.Get("uid").(uint)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
user, err := db.Users.GetByID(ctx.Request().Context(), uid)
|
|
if err != nil || user == nil {
|
|
return nil
|
|
}
|
|
|
|
// 密码变更后，旧会话保存的版本号不匹配，强制重新登录。
|
|
ver, _ := sess.Get("pwd_ver").(uint)
|
|
if user.PasswordVersion != ver {
|
|
return nil
|
|
}
|
|
return user
|
|
}
|
|
|
|
type ToggleOptions struct {
|
|
UserSignInRequired bool
|
|
UserSignOutRequired bool
|
|
}
|
|
|
|
func Toggle(options *ToggleOptions) flamego.Handler {
|
|
return func(ctx Context, endpoint EndpointType) error {
|
|
if options.UserSignOutRequired && ctx.IsLogged {
|
|
ctx.Redirect("/")
|
|
return nil
|
|
}
|
|
|
|
if options.UserSignInRequired && !ctx.IsLogged {
|
|
if endpoint.IsAPI() {
|
|
return ctx.JSONError(40100, i18n.T(ctx.Lang(), "auth.errLoginRequired"))
|
|
}
|
|
ctx.SetErrorFlash(i18n.T(ctx.Lang(), "auth.errLoginRequired"))
|
|
ctx.Redirect("/login")
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
}
|