Set cookies in auth handler before looking up user

This commit is contained in:
Marshall Polaris 2022-07-21 01:28:49 -07:00
parent 23b704ffe0
commit e84088ba75

View File

@ -41,6 +41,7 @@ export function AuthProvider({ children }: any) {
useEffect(() => { useEffect(() => {
return onIdTokenChanged(auth, async (fbUser) => { return onIdTokenChanged(auth, async (fbUser) => {
if (fbUser) { if (fbUser) {
setAuthCookies(await fbUser.getIdToken(), fbUser.refreshToken)
let user = await getUser(fbUser.uid) let user = await getUser(fbUser.uid)
if (!user) { if (!user) {
const deviceToken = ensureDeviceToken() const deviceToken = ensureDeviceToken()
@ -51,12 +52,11 @@ export function AuthProvider({ children }: any) {
// Note: Cap on localStorage size is ~5mb // Note: Cap on localStorage size is ~5mb
localStorage.setItem(CACHED_USER_KEY, JSON.stringify(user)) localStorage.setItem(CACHED_USER_KEY, JSON.stringify(user))
setCachedReferralInfoForUser(user) setCachedReferralInfoForUser(user)
setAuthCookies(await fbUser.getIdToken(), fbUser.refreshToken)
} else { } else {
// User logged out; reset to null // User logged out; reset to null
deleteAuthCookies()
setAuthUser(null) setAuthUser(null)
localStorage.removeItem(CACHED_USER_KEY) localStorage.removeItem(CACHED_USER_KEY)
deleteAuthCookies()
} }
}) })
}, [setAuthUser]) }, [setAuthUser])