Add update method to auth uservalue
This commit is contained in:
parent
32116905b7
commit
4bad13510a
@ -7,12 +7,22 @@ import (
|
|||||||
|
|
||||||
const UserCtxKey = "user"
|
const UserCtxKey = "user"
|
||||||
|
|
||||||
// Middleware adds authentication handling
|
// Auth holds the authentication middleware and user lookup function
|
||||||
func Middleware(userLookup func(int) any) sushi.Middleware {
|
type Auth struct {
|
||||||
|
userLookup func(int) any
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new Auth instance
|
||||||
|
func New(userLookup func(int) any) *Auth {
|
||||||
|
return &Auth{userLookup: userLookup}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Middleware returns the authentication middleware function
|
||||||
|
func (a *Auth) Middleware() sushi.Middleware {
|
||||||
return func(ctx sushi.Ctx, next func()) {
|
return func(ctx sushi.Ctx, next func()) {
|
||||||
sess := sushi.GetCurrentSession(ctx)
|
sess := sushi.GetCurrentSession(ctx)
|
||||||
if sess != nil && sess.UserID > 0 && userLookup != nil {
|
if sess != nil && sess.UserID > 0 && a.userLookup != nil {
|
||||||
user := userLookup(sess.UserID)
|
user := a.userLookup(sess.UserID)
|
||||||
if user != nil {
|
if user != nil {
|
||||||
ctx.SetUserValue(UserCtxKey, user)
|
ctx.SetUserValue(UserCtxKey, user)
|
||||||
} else {
|
} else {
|
||||||
@ -24,6 +34,15 @@ func Middleware(userLookup func(int) any) sushi.Middleware {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Update refreshes the current user data in the context
|
||||||
|
func (a *Auth) Update(ctx sushi.Ctx) {
|
||||||
|
sess := sushi.GetCurrentSession(ctx)
|
||||||
|
if sess != nil && sess.UserID > 0 && a.userLookup != nil {
|
||||||
|
user := a.userLookup(sess.UserID)
|
||||||
|
ctx.SetUserValue(UserCtxKey, user)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// RequireAuth middleware that redirects unauthenticated users
|
// RequireAuth middleware that redirects unauthenticated users
|
||||||
func RequireAuth(redirectPath ...string) sushi.Middleware {
|
func RequireAuth(redirectPath ...string) sushi.Middleware {
|
||||||
redirect := "/login"
|
redirect := "/login"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user