Make start on session management
This commit is contained in:
parent
352d9555ba
commit
17dd20478d
4 changed files with 65 additions and 7 deletions
|
|
@ -54,11 +54,12 @@ func (asset *Asset) fetchIfNotExists(staticPath string) {
|
|||
}
|
||||
|
||||
type Config struct {
|
||||
Address string
|
||||
Protocol string
|
||||
Extension string
|
||||
NotesDir string
|
||||
LogAccess bool
|
||||
Address string
|
||||
Protocol string
|
||||
Extension string
|
||||
NotesDir string
|
||||
LogAccess bool
|
||||
Production bool
|
||||
}
|
||||
|
||||
var (
|
||||
|
|
|
|||
34
internal/middleware/session.go
Normal file
34
internal/middleware/session.go
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
// Middleware to deal with sessions
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func SessionMiddleware(cache map[string]string, next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
sessionCookie, err := r.Cookie("id")
|
||||
// No session yet
|
||||
if err != nil {
|
||||
http.Redirect(w, r, "/login/", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
user, ok := cache[sessionCookie.Value]
|
||||
|
||||
// Session expired
|
||||
if !ok {
|
||||
http.Redirect(w, r, "/login/", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("User is %s", user)
|
||||
|
||||
ctx := r.Context()
|
||||
ctx = context.WithValue(ctx, "user", user)
|
||||
|
||||
next.ServeHTTP(w, r.WithContext(ctx))
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue