Compare commits

...

2 Commits

Author SHA1 Message Date
4bad13510a Add update method to auth uservalue 2025-08-22 08:46:42 -05:00
32116905b7 add missing deleteflash method 2025-08-21 10:05:39 -05:00
3 changed files with 29 additions and 6 deletions

View File

@ -7,12 +7,22 @@ import (
const UserCtxKey = "user"
// Middleware adds authentication handling
func Middleware(userLookup func(int) any) sushi.Middleware {
// Auth holds the authentication middleware and user lookup function
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()) {
sess := sushi.GetCurrentSession(ctx)
if sess != nil && sess.UserID > 0 && userLookup != nil {
user := userLookup(sess.UserID)
if sess != nil && sess.UserID > 0 && a.userLookup != nil {
user := a.userLookup(sess.UserID)
if user != nil {
ctx.SetUserValue(UserCtxKey, user)
} 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
func RequireAuth(redirectPath ...string) sushi.Middleware {
redirect := "/login"

View File

@ -101,10 +101,10 @@ func (f FormValue) IsEmpty() bool {
// GetFormArray gets multiple form values as string slice
func (ctx Ctx) GetFormArray(key string) []string {
var values []string
ctx.PostArgs().VisitAll(func(k, v []byte) {
for k, v := range ctx.PostArgs().All() {
if string(k) == key {
values = append(values, string(v))
}
})
}
return values
}

View File

@ -104,6 +104,10 @@ func (s *Session) GetFlash(key string) (any, bool) {
return value, exists
}
func (s *Session) DeleteFlash(key string) {
s.Delete("flash_" + key)
}
func (s *Session) GetFlashMessage(key string) string {
if flash, exists := s.GetFlash(key); exists {
if msg, ok := flash.(string); ok {