authorizer/server/resolvers/session.go

package resolvers

import (
	"context"
	"errors"
	"fmt"
	"time"
	"os"

	"github.com/google/uuid"
	log "github.com/sirupsen/logrus"

	"github.com/authorizerdev/authorizer/server/constants"
	"github.com/authorizerdev/authorizer/server/cookie"
	"github.com/authorizerdev/authorizer/server/db"
	"github.com/authorizerdev/authorizer/server/graph/model"
	"github.com/authorizerdev/authorizer/server/memorystore"
	"github.com/authorizerdev/authorizer/server/memorystore/providers/redis"
	"github.com/authorizerdev/authorizer/server/token"
	"github.com/authorizerdev/authorizer/server/utils"
)

// SessionResolver is a resolver for session query
// TODO allow validating with code and code verifier instead of cookie (PKCE flow)
func SessionResolver(ctx context.Context, params *model.SessionQueryInput) (*model.AuthResponse, error) {
	var res *model.AuthResponse

	gc, err := utils.GinContextFromContext(ctx)
	if err != nil {
		log.Debug("Failed to get GinContext: ", err)
		return res, err
	}

	sessionToken, err := cookie.GetSession(gc)
	if err != nil {
		log.Debug("Failed to get session token: ", err)
		return res, errors.New("unauthorized")
	}

	// get session from cookie
	claims, err := token.ValidateBrowserSession(gc, sessionToken)
	if err != nil {
		log.Debug("Failed to validate session token: ", err)
		return res, errors.New("unauthorized")
	}
	userID := claims.Subject

	log := log.WithFields(log.Fields{
		"user_id": userID,
	})

	user, err := db.Provider.GetUserByID(ctx, userID)
	if err != nil {
		return res, err
	}

	// refresh token has "roles" as claim
	claimRoleInterface := claims.Roles
	claimRoles := []string{}
	claimRoles = append(claimRoles, claimRoleInterface...)

	if params != nil && params.Roles != nil && len(params.Roles) > 0 {
		for _, v := range params.Roles {
			if !utils.StringSliceContains(claimRoles, v) {
				log.Debug("User does not have required role: ", claimRoles, v)
				return res, fmt.Errorf(`unauthorized`)
			}
		}
	}

	scope := []string{"openid", "email", "profile"}
	if params != nil && params.Scope != nil && len(scope) > 0 {
		scope = params.Scope
	}

	nonce := uuid.New().String()
	authToken, err := token.CreateAuthToken(gc, user, claimRoles, scope, claims.LoginMethod, nonce, "")
	if err != nil {
		log.Debug("Failed to create auth token: ", err)
		return res, err
	}

	// rollover the session for security
	sessionKey := userID
	if claims.LoginMethod != "" {
		sessionKey = claims.LoginMethod + ":" + userID
	}
	go memorystore.Provider.DeleteUserSession(sessionKey, claims.Nonce)

	expiresIn := authToken.AccessToken.ExpiresAt - time.Now().Unix()
	if expiresIn <= 0 {
		expiresIn = 1
	}

	res = &model.AuthResponse{
		Message:     `Session token refreshed`,
		AccessToken: &authToken.AccessToken.Token,
		ExpiresIn:   &expiresIn,
		IDToken:     &authToken.IDToken.Token,
		User:        user.AsAPIUser(),
	}

	cookie.SetSession(gc, authToken.FingerPrintHash)
	memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeSessionToken+"_"+authToken.FingerPrint, authToken.FingerPrintHash, authToken.SessionTokenExpiresAt)
	memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeAccessToken+"_"+authToken.FingerPrint, authToken.AccessToken.Token, authToken.AccessToken.ExpiresAt)

    redisURL := os.Getenv(constants.EnvKeyRedisURL)
    if redisURL != "" {
        log.Info("Initializing Redis provider")
        Provider, err := redis.NewRedisProvider(redisURL)
        if err != nil {
            log.Debug("Failed to init Redis: ", err)
        }

        if res.User.AppData == nil {
            res.User.AppData = make(map[string]interface{})
        }

        follows, err := Provider.GetUserFollows(userID)
        if err != nil {
            log.Debug("Failed to get follows from Redis: ", err)
        } else {
            if follows != "" {
                res.User.AppData["follows"] = follows
            } else {
                log.Debug("Follows data from Redis is empty")
            }
        }

        author, err := Provider.GetUserProfile(userID)
        if err != nil {
            log.Debug("Failed to get author from Redis: ", err)
        } else {
            if author != "" {
                res.User.AppData["author"] = author
            } else {
                log.Debug("Author data from Redis is empty")
            }
        }
    }

	if authToken.RefreshToken != nil {
		res.RefreshToken = &authToken.RefreshToken.Token
		memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeRefreshToken+"_"+authToken.FingerPrint, authToken.RefreshToken.Token, authToken.RefreshToken.ExpiresAt)
	}
	return res, nil
}