Amplify, React, Context API, Cognito, useAuth hook, NextJS, apollo

UserContext.js

import React, { useEffect, useContext, useState } from 'react'
import { useRouter } from 'next/router'
import Auth from '@aws-amplify/auth'

const UserContext = React.createContext()

export const initUser = {
  user: null,
}

// user prop not currently used
export const UserProvider = ({ children, user }) => {
  const [currentUserContext, setCurrentUserContext] = useState(user || initUser)

  const setUserContext = values => {
    setCurrentUserContext(values)
  }

  return <UserContext.Provider value={{ userContext: currentUserContext, setUserContext }}>{children}</UserContext.Provider>
}

export const UserConsumer = UserContext.Consumer

export function useAuth(redirect = false, path) {
  const { userContext, setUserContext } = useContext(UserContext)
  const router = useRouter()

  useEffect(() => {
    if (!userContext.user) {
      let updateUser = async nothing => {
        try {
          let user = await Auth.currentAuthenticatedUser()
          setUserContext({ user })
          // optional redirect to protected page (like dashboard)
          if (path) {
            router.push(path)
          }
        } catch {
          // optional redirect to unprotected page (like home)
          if (redirect) {
            router.push('/unprotectedpage')
          }
        }
      }
      updateUser()
    }
  }, [userContext.user])

  return { user: userContext.user }
}

export default UserContext