57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package sessions
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/sessions"
|
|
"github.com/kirsle/blog/src/log"
|
|
"github.com/kirsle/blog/src/types"
|
|
)
|
|
|
|
// Store holds your cookie store information.
|
|
var Store sessions.Store
|
|
|
|
// SetSecretKey initializes a session cookie store with the secret key.
|
|
func SetSecretKey(keyPairs ...[]byte) {
|
|
Store = sessions.NewCookieStore(keyPairs...)
|
|
}
|
|
|
|
// Middleware gets the Gorilla session store and makes it available on the
|
|
// Request context.
|
|
//
|
|
// Middleware is the first custom middleware applied, so it takes the current
|
|
// datetime to make available later in the request and stores it on the request
|
|
// context.
|
|
func Middleware(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
|
|
// Store the current datetime on the request context.
|
|
ctx := context.WithValue(r.Context(), types.StartTimeKey, time.Now())
|
|
|
|
// Get the Gorilla session and make it available in the request context.
|
|
session, _ := Store.Get(r, "session")
|
|
ctx = context.WithValue(ctx, types.SessionKey, session)
|
|
|
|
next(w, r.WithContext(ctx))
|
|
}
|
|
|
|
// Get returns the current request's session.
|
|
func Get(r *http.Request) *sessions.Session {
|
|
if r == nil {
|
|
panic("Session(*http.Request) with a nil argument!?")
|
|
}
|
|
|
|
ctx := r.Context()
|
|
if session, ok := ctx.Value(types.SessionKey).(*sessions.Session); ok {
|
|
return session
|
|
}
|
|
|
|
// If the session wasn't on the request, it means I broke something.
|
|
log.Error(
|
|
"Session(): didn't find session in request context! Getting it " +
|
|
"from the session store instead.",
|
|
)
|
|
session, _ := Store.Get(r, "session")
|
|
return session
|
|
}
|