fix: auth key errors (#411)

Author: rutajdash

client/src/config/ApolloClient.js

@@ -9,6 +9,7 @@ import {
 } from '@apollo/client';
 import { setContext } from '@apollo/client/link/context';
 import { onError } from '@apollo/client/link/error';
+import { parseCookies } from 'nookies';
 
 const cache = new InMemoryCache();
 
@@ -43,10 +44,11 @@ const link = from([
 ]);
 
 const getApolloLink = (token) => {
+  const cookies = parseCookies();
   const authLink = setContext((_, { headers }) => ({
     headers: {
       ...headers,
-      Authorization: token ? token : '',
+      Authorization: token || cookies.firebaseToken || '',
     },
   }));
   return authLink.concat(link);

client/src/context/auth/AuthState.js

@@ -45,17 +45,19 @@ const AuthState = ({ children }) => {
       sameSite: true,
       maxAge: 3600,
     });
-  }, [firebaseToken]);
+  }, [user, firebaseToken, refreshToken]);
 
   const loginWithToken = async (_token) => {
     const credential = GoogleAuthProvider.credential(_token);
 
     const res = await signInWithCredential(auth, credential);
 
     const { user, _tokenResponse } = res;
-    setUser(user);
+
+    GraphClient.setLink(getApolloLink(user.accessToken));
     setFirebaseToken(user.accessToken);
     setRefreshToken(user.refreshToken);
+    setUser(user);
 
     if (getAdditionalUserInfo(res).isNewUser) {
       try {
@@ -65,6 +67,15 @@ const AuthState = ({ children }) => {
           authenticationEndpoint: `${process.env.NEXT_PUBLIC_SERVER_ADDRESS}${process.env.NEXT_PUBLIC_IMAGEKIT_AUTHENTICATION_ENDPOINT}`,
         });
 
+        const newAccount = await GraphClient.mutate({
+          mutation: registerUser,
+          variables: {
+            fullName: user.displayName,
+            email: user.email,
+          },
+        });
+        console.log('Account Created ', newAccount);
+
         const userPicture = await (await fetch(user.photoURL)).blob();
 
         if (!['image/png', 'image/jpeg'].includes(userPicture.type)) {
@@ -73,25 +84,16 @@ const AuthState = ({ children }) => {
 
         const imageUpload = await imagekit
           .upload({
-            file: userPicture,
+            file: user.photoURL,
             folder: '/user',
-            fileName: `${user.uid}.${
+            fileName: `${newAccount.data.registerUser.id}.${
               userPicture.type.toString().split('/')[1]
             }`,
             tags: [user.uid, 'user', 'profilePicture'],
           })
           .then((result) => {
             console.log('Upload Success', result);
           });
-
-        const newAccount = await GraphClient.mutate({
-          mutation: registerUser,
-          variables: {
-            fullName: user.displayName,
-            email: user.email,
-          },
-        });
-        console.log('Account Created ', newAccount);
       } catch (err) {
         console.log(err);
       }