72 lines
1.4 KiB
Go
72 lines
1.4 KiB
Go
package userctx
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"git.kealoha.me/lks/lenslocked/models"
|
|
)
|
|
|
|
type key string
|
|
|
|
const userKey key = "User"
|
|
|
|
func WithUser(ctx context.Context, user *models.User) context.Context {
|
|
return context.WithValue(ctx, userKey, user)
|
|
}
|
|
|
|
func User(ctx context.Context) *models.User {
|
|
val := ctx.Value(userKey)
|
|
user, ok := val.(*models.User)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
return user
|
|
}
|
|
|
|
type UserMiddleware struct {
|
|
SS *models.SessionService
|
|
}
|
|
|
|
func (umw UserMiddleware) SetUser(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
seshCookie, err := r.Cookie("session")
|
|
if err != nil {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
user, err := umw.SS.User(seshCookie.Value)
|
|
if err != nil {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
ctx := r.Context()
|
|
ctx = WithUser(ctx, user)
|
|
r = r.WithContext(ctx)
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func (umw UserMiddleware) RequireUserfn(next http.HandlerFunc) http.HandlerFunc {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
user := User(r.Context())
|
|
if user == nil {
|
|
http.Redirect(w, r, "/signin", http.StatusFound)
|
|
return
|
|
}
|
|
next(w, r)
|
|
})
|
|
}
|
|
|
|
func (umw UserMiddleware) RequireUser(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
user := User(r.Context())
|
|
if user == nil {
|
|
http.Redirect(w, r, "/signin", http.StatusFound)
|
|
return
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|