AuthContext.tsx

import React from 'react';
import { AsyncStorage } from 'react-native';
import UUID from 'uuid-random';
import { useApolloClient } from '@apollo/client';
import { useUnregisterDeviceMutation } from '../graphql/generated';

[...]

export const AuthProvider: React.FC = ({ children }) => {
  const [userState, setUserState] = React.useState<User>();
  const [unregisterDeviceMutate] = useUnregisterDeviceMutation();
  const client = useApolloClient();
  
  [...]

  const logout = React.useCallback(async () => {
    try {
      const pushToken = await AsyncStorage.getItem('PushToken');
      if (pushToken) {
        await unregisterDeviceMutate({ 
          fetchPolicy: 'no-cache',
          variables: { token: pushToken }  
        });

        await AsyncStorage.removeItem('PushToken');
      }

      await client.resetStore();
      await clearUserFromStorage();
      setUserState(null);
    } catch (err) {
      console.error(err);
    }
  }, []);

  return (
    <AuthContext.Provider value={{ user: userState, setUser, login, logout }}>
      {children}
    </AuthContext.Provider>
  );
};
[...]