fix: session propagation mistakes (#457)

Author: 42atomys

web/ui/src/contexts/currentUser.tsx

@@ -8,6 +8,7 @@ import {
   useMeLazyQuery,
   useUpdateSettingsMutation,
 } from '@graphql.d';
+import { useSession } from 'next-auth/react';
 import React, {
   createContext,
   useCallback,
@@ -35,9 +36,9 @@ const MeContext = createContext<MeContextValue>({
 export const MeProvider: React.FC<MeProviderProps> = ({
   children,
   apolloClient,
-  session,
 }) => {
   const { addNotification } = useNotification();
+  const { status } = useSession();
   const [me, setMe] = useState<MeQuery>({
     me: { settings: {} } as MeQuery['me'],
     myFollowings: [],
@@ -164,7 +165,7 @@ export const MeProvider: React.FC<MeProviderProps> = ({
 
   return (
     <ConditionalWrapper
-      condition={!!session}
+      condition={status === 'authenticated'}
       trueWrapper={(c) => (
         <MeContext.Provider value={value}>{c}</MeContext.Provider>
       )}

web/ui/src/contexts/types.d.ts

@@ -1,5 +1,6 @@
 import { NotificationProps } from '@components/Notification/types';
 import { MeQuery, SettingsInput } from '@graphql.d';
+import { Session } from 'next-auth';
 
 export interface MeContextValue extends Omit<MeQuery, '__typename'> {
   // Function to update the current user in the session storage
@@ -19,7 +20,7 @@ export interface MeContextValue extends Omit<MeQuery, '__typename'> {
 
 type MeProviderProps = React.PropsWithChildren<{
   apolloClient?: ApolloClient<any>;
-  session: Session | null;
+  session?: Session;
 }>;
 
 export interface NotificationContextValue {

web/ui/src/pages/_app.tsx

@@ -20,9 +20,7 @@ const InteractiveApp: React.FC<
   return (
     <SessionProvider session={session}>
       <ApolloProvider client={apolloClient}>
-        <MeProvider apolloClient={apolloClient} session={session}>
-          {children}
-        </MeProvider>
+        <MeProvider apolloClient={apolloClient}>{children}</MeProvider>
       </ApolloProvider>
     </SessionProvider>
   );