mirror of
https://github.com/raseels-repos/golang-saas-starter-kit.git
synced 2025-06-06 23:46:29 +02:00
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package webcontext
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gorilla/sessions"
|
|
)
|
|
|
|
// ctxKeySession represents the type of value for the context key.
|
|
type ctxKeySession int
|
|
|
|
// KeySession is used to store/retrieve a Session from a context.Context.
|
|
const KeySession ctxKeySession = 1
|
|
|
|
// KeyAccessToken is used to store the access token for the user in their session.
|
|
const KeyAccessToken = "AccessToken"
|
|
|
|
// ContextWithSession appends a universal translator to a context.
|
|
func ContextWithSession(ctx context.Context, session *sessions.Session) context.Context {
|
|
return context.WithValue(ctx, KeySession, session)
|
|
}
|
|
|
|
// ContextSession returns the session from a context.
|
|
func ContextSession(ctx context.Context) *sessions.Session {
|
|
return ctx.Value(KeySession).(*sessions.Session)
|
|
}
|
|
|
|
func ContextAccessToken(ctx context.Context) (string, bool) {
|
|
session := ContextSession(ctx)
|
|
|
|
return SessionAccessToken(session)
|
|
}
|
|
|
|
func SessionAccessToken(session *sessions.Session) (string, bool) {
|
|
if sv, ok := session.Values[KeyAccessToken].(string); ok {
|
|
return sv, true
|
|
}
|
|
|
|
return "", false
|
|
}
|
|
|
|
func SessionWithAccessToken(session *sessions.Session, accessToken string) *sessions.Session {
|
|
|
|
if accessToken != "" {
|
|
session.Values[KeyAccessToken] = accessToken
|
|
} else {
|
|
delete(session.Values, KeyAccessToken)
|
|
}
|
|
|
|
return session
|
|
}
|