Files
authorizer/server/resolvers/session.go

104 lines
2.9 KiB
Go
Raw Normal View History

package resolvers
import (
"context"
2022-03-15 09:57:09 +05:30
"errors"
"fmt"
"time"
2022-05-24 12:42:29 +05:30
log "github.com/sirupsen/logrus"
"github.com/authorizerdev/authorizer/server/cookie"
2021-07-23 21:57:44 +05:30
"github.com/authorizerdev/authorizer/server/db"
"github.com/authorizerdev/authorizer/server/graph/model"
2022-05-27 23:20:38 +05:30
"github.com/authorizerdev/authorizer/server/memorystore"
"github.com/authorizerdev/authorizer/server/token"
2021-07-23 21:57:44 +05:30
"github.com/authorizerdev/authorizer/server/utils"
)
2022-01-17 11:32:13 +05:30
// SessionResolver is a resolver for session query
2022-03-02 17:42:31 +05:30
// TODO allow validating with code and code verifier instead of cookie (PKCE flow)
2022-01-24 00:32:06 +05:30
func SessionResolver(ctx context.Context, params *model.SessionQueryInput) (*model.AuthResponse, error) {
var res *model.AuthResponse
gc, err := utils.GinContextFromContext(ctx)
if err != nil {
2022-05-25 12:30:22 +05:30
log.Debug("Failed to get GinContext: ", err)
return res, err
}
2022-03-02 17:42:31 +05:30
sessionToken, err := cookie.GetSession(gc)
if err != nil {
2022-05-24 12:42:29 +05:30
log.Debug("Failed to get session token", err)
2022-03-15 09:57:09 +05:30
return res, errors.New("unauthorized")
}
2022-03-02 17:42:31 +05:30
// get session from cookie
claims, err := token.ValidateBrowserSession(gc, sessionToken)
if err != nil {
2022-05-24 12:42:29 +05:30
log.Debug("Failed to validate session token", err)
2022-03-15 09:57:09 +05:30
return res, errors.New("unauthorized")
}
2022-03-02 17:42:31 +05:30
userID := claims.Subject
2022-05-24 12:42:29 +05:30
log := log.WithFields(log.Fields{
"user_id": userID,
})
user, err := db.Provider.GetUserByID(userID)
if err != nil {
return res, err
}
// refresh token has "roles" as claim
2022-03-02 17:42:31 +05:30
claimRoleInterface := claims.Roles
claimRoles := []string{}
2022-05-24 12:42:29 +05:30
claimRoles = append(claimRoles, claimRoleInterface...)
2022-01-24 00:32:06 +05:30
if params != nil && params.Roles != nil && len(params.Roles) > 0 {
for _, v := range params.Roles {
2021-10-19 12:57:59 +05:30
if !utils.StringSliceContains(claimRoles, v) {
2022-05-25 12:30:22 +05:30
log.Debug("User does not have required role: ", claimRoles, v)
return res, fmt.Errorf(`unauthorized`)
}
}
}
2022-03-02 17:42:31 +05:30
scope := []string{"openid", "email", "profile"}
if params != nil && params.Scope != nil && len(scope) > 0 {
scope = params.Scope
}
2022-03-02 17:42:31 +05:30
authToken, err := token.CreateAuthToken(gc, user, claimRoles, scope)
if err != nil {
2022-05-25 12:30:22 +05:30
log.Debug("Failed to create auth token: ", err)
return res, err
}
2022-03-02 17:42:31 +05:30
// rollover the session for security
2022-05-27 23:20:38 +05:30
memorystore.Provider.RemoveState(sessionToken)
memorystore.Provider.SetState(authToken.FingerPrintHash, authToken.FingerPrint+"@"+user.ID)
memorystore.Provider.SetState(authToken.AccessToken.Token, authToken.FingerPrint+"@"+user.ID)
2022-03-02 17:42:31 +05:30
cookie.SetSession(gc, authToken.FingerPrintHash)
expiresIn := authToken.AccessToken.ExpiresAt - time.Now().Unix()
if expiresIn <= 0 {
expiresIn = 1
}
res = &model.AuthResponse{
Message: `Session token refreshed`,
AccessToken: &authToken.AccessToken.Token,
2022-03-02 17:42:31 +05:30
ExpiresIn: &expiresIn,
IDToken: &authToken.IDToken.Token,
User: user.AsAPIUser(),
}
2022-03-02 17:42:31 +05:30
if authToken.RefreshToken != nil {
res.RefreshToken = &authToken.RefreshToken.Token
2022-05-27 23:20:38 +05:30
memorystore.Provider.SetState(authToken.RefreshToken.Token, authToken.FingerPrint+"@"+user.ID)
2022-03-02 17:42:31 +05:30
}
return res, nil
}