fix: set provider as default

Signed-off-by: Eray Ates <[email protected]>
worldline-go · Mar 6, 2025 · 05ba6cf · 05ba6cf
1 parent da27f1c
commit 05ba6cf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/pkg/server/http/middleware/session/action.go b/pkg/server/http/middleware/session/action.go
@@ -405,6 +405,10 @@ func (m *Session) GetToken(r *http.Request) (*TokenData, *Oauth2, error) {
 		return nil, nil, fmt.Errorf("cookie not found")
 	}
 
+	if m.SetProvider != "" {
+		providerName = m.SetProvider
+	}
+
 	// check if token is valid
 	token, err := ParseToken64(v64)
 	if err != nil {
@@ -439,6 +443,10 @@ func (m *Session) IsLogged(w http.ResponseWriter, r *http.Request) (*claims.Cust
 		return nil, false, nil
 	}
 
+	if m.SetProvider != "" {
+		providerName = m.SetProvider
+	}
+
 	// check if token is valid
 	token, err := ParseToken64(v64)
 	if err != nil {

diff --git a/pkg/server/http/middleware/session/store.go b/pkg/server/http/middleware/session/store.go
@@ -100,6 +100,10 @@ func (m *Session) SetStore(ctx context.Context) error {
 func (m *Session) SetToken(w http.ResponseWriter, r *http.Request, token []byte, providerName string) error {
 	cookieValue := base64.StdEncoding.EncodeToString(token)
 
+	if m.SetProvider != "" {
+		providerName = m.SetProvider
+	}
+
 	// set the cookie
 	session, _ := m.store.Get(r, m.GetCookieName(r))
 	session.Values[TokenKey] = cookieValue