update session

auth/auth.go

@@ -22,6 +22,7 @@ import (
 	"context"
 	"fmt"
 	legacyauth "github.com/dapperlabs/flow-playground-api/auth/legacy"
+	playgroundErrors "github.com/dapperlabs/flow-playground-api/middleware/errors"
 	"github.com/dapperlabs/flow-playground-api/middleware/sessions"
 	"github.com/dapperlabs/flow-playground-api/model"
 	"github.com/dapperlabs/flow-playground-api/storage"
@@ -67,10 +68,13 @@ func (a *Authenticator) GetOrCreateUser(ctx context.Context) (*model.User, error
 		// Try to load existing user
 		if session.Values[userIDKey] != nil {
 			user, err = a.getCurrentUser(session.Values[userIDKey].(string))
-			if err == nil {
+			if err != nil {
 				fmt.Printf("Failed to load user id %s from session\n", session.Values[userIDKey].(string))
+			} else {
 				userLoaded = true
 			}
+		} else {
+			return nil, playgroundErrors.NewAuthorizationError("userID is missing from existing session")
 		}
 	}
 
@@ -86,6 +90,7 @@ func (a *Authenticator) GetOrCreateUser(ctx context.Context) (*model.User, error
 
 	err = sessions.Save(ctx, session)
 	if err != nil {
+		fmt.Println("FAILED TO UPDATE SESSION")
 		return nil, errors.Wrap(err, "failed to update session")
 	}
 

middleware/sessions/sessions.go

@@ -40,6 +40,8 @@ func Middleware(store sessions.Store) func(http.Handler) http.Handler {
 		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			ctx := context.WithValue(r.Context(), sessionCtxKeySession, store)
 
+			fmt.Println("Request: ", r.Header)
+
 			fmt.Println("Sessions Middleware Handler:")
 			session, err := store.Get(r, "flow-playground")
 			if err != nil {