app-data-patch

server/resolvers/session.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"time"
+	"os"
 
 	"github.com/google/uuid"
 	log "github.com/sirupsen/logrus"
@@ -14,6 +15,7 @@ import (
 	"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"
 )
@@ -102,6 +104,33 @@ func SessionResolver(ctx context.Context, params *model.SessionQueryInput) (*mod
 	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, _ := 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 {
+            res.User.AppData["follows"] = follows
+        }
+
+        author, err := Provider.GetUserProfile(userID)
+        if err != nil {
+            log.Debug("Failed to get author from Redis: ", err)
+        } else {
+            res.User.AppData["author"] = author
+        }
+    }
+
 	if authToken.RefreshToken != nil {
 		res.RefreshToken = &authToken.RefreshToken.Token
 		memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeRefreshToken+"_"+authToken.FingerPrint, authToken.RefreshToken.Token, authToken.RefreshToken.ExpiresAt)