fix: recpie override for session revokes

Author: amirsalarsafaei

recipe/session/getClaimValue_test.go

@@ -5,6 +5,7 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+
 	"github.com/supertokens/supertokens-golang/recipe/session/sessmodels"
 	"github.com/supertokens/supertokens-golang/supertokens"
 	"github.com/supertokens/supertokens-golang/test/unittesting"

recipe/session/recipeImplementation.go

@@ -25,6 +25,7 @@ import (
 	"time"
 
 	"github.com/MicahParks/keyfunc/v2"
+
 	"github.com/supertokens/supertokens-golang/recipe/multitenancy/multitenancymodels"
 	"github.com/supertokens/supertokens-golang/recipe/session/claims"
 	"github.com/supertokens/supertokens-golang/recipe/session/errors"
@@ -283,7 +284,12 @@ func MakeRecipeImplementation(querier supertokens.Querier, config sessmodels.Typ
 		frontToken := BuildFrontToken(response.Session.UserID, response.Session.ExpiryTime, payload)
 		session := response.Session
 
-		sessionContainerInput := makeSessionContainerInput(accessTokenStringForSession, session.Handle, session.UserID, session.TenantId, payload, result, frontToken, antiCsrfToken, nil, nil, !accessTokenNil)
+		recipeInstance, err := getRecipeInstanceOrThrowError()
+		if err != nil {
+			return nil, err
+		}
+
+		sessionContainerInput := makeSessionContainerInput(accessTokenStringForSession, session.Handle, session.UserID, session.TenantId, payload, recipeInstance.RecipeImpl, frontToken, antiCsrfToken, nil, nil, !accessTokenNil)
 		sessionContainer := newSessionContainer(config, &sessionContainerInput)
 
 		return sessionContainer, nil
@@ -314,7 +320,12 @@ func MakeRecipeImplementation(querier supertokens.Querier, config sessmodels.Typ
 		session := response.Session
 		frontToken := BuildFrontToken(session.UserID, response.AccessToken.Expiry, responseToken.Payload)
 
-		sessionContainerInput := makeSessionContainerInput(response.AccessToken.Token, session.Handle, session.UserID, session.TenantId, responseToken.Payload, result, frontToken, response.AntiCsrfToken, nil, &response.RefreshToken, true)
+		recipeInstance, err := getRecipeInstanceOrThrowError()
+		if err != nil {
+			return nil, err
+		}
+
+		sessionContainerInput := makeSessionContainerInput(response.AccessToken.Token, session.Handle, session.UserID, session.TenantId, responseToken.Payload, recipeInstance.RecipeImpl, frontToken, response.AntiCsrfToken, nil, &response.RefreshToken, true)
 		sessionContainer := newSessionContainer(config, &sessionContainerInput)
 
 		return sessionContainer, nil

recipe/session/sessionRequestFunctions.go

@@ -22,7 +22,6 @@ import (
 	"strconv"
 
 	"github.com/supertokens/supertokens-golang/recipe/session/claims"
-
 	"github.com/supertokens/supertokens-golang/recipe/session/errors"
 	"github.com/supertokens/supertokens-golang/recipe/session/sessmodels"
 	"github.com/supertokens/supertokens-golang/supertokens"