diff --git a/packages/web/src/app/layout/auth-button/AuthButton.tsx b/packages/web/src/app/layout/auth-button/AuthButton.tsx index de5fdf90..254811e2 100644 --- a/packages/web/src/app/layout/auth-button/AuthButton.tsx +++ b/packages/web/src/app/layout/auth-button/AuthButton.tsx @@ -1,5 +1,5 @@ import { useGoogleLogin } from "@react-oauth/google" -import { useState } from "react" +import { useEffect, useState } from "react" import { useAppStore } from "../../../common/storage/app-store" import { trpcNative } from "../../../common/trpc/trpc-native" import { CreateAccount } from "./CreateAccount" @@ -14,6 +14,23 @@ export function AuthButton() { const updateJwt = useAppStore(x => x.updateJwt) const updateUser = useAppStore(x => x.updateUser) + useEffect(() => { + setTimeout(() => { + const jwt = useAppStore.getState().jwt + + if (jwt) { + console.log("jwt", jwt) + setLoading(true) + trpcNative.user.me.query().then(user => { + updateUser(user) + setLoading(false) + }) + } else { + console.log("no jwt") + } + }, 100) + }, [updateUser]) + const login = useGoogleLogin({ onSuccess: async ({ code }) => { try { diff --git a/packages/web/src/app/layout/auth-button/RenameAccount.tsx b/packages/web/src/app/layout/auth-button/RenameAccount.tsx index 29816f0a..a29a1fcc 100644 --- a/packages/web/src/app/layout/auth-button/RenameAccount.tsx +++ b/packages/web/src/app/layout/auth-button/RenameAccount.tsx @@ -68,7 +68,7 @@ export function RenameAccount(props: { open: boolean; onFinished: () => void }) {loading === false && (
- Create Account + Rename
)} {loading && ( diff --git a/packages/web/src/common/storage/app-store.ts b/packages/web/src/common/storage/app-store.ts index 5e90c34f..052ed0e0 100644 --- a/packages/web/src/common/storage/app-store.ts +++ b/packages/web/src/common/storage/app-store.ts @@ -62,9 +62,11 @@ export const useAppStore = create()( storage: createJSONStorage(() => storage), partialize: state => ({ - jwt: state.jwt ?? undefined, + jwt: state.jwt, }), - onRehydrateStorage: () => () => {}, + onRehydrateStorage: () => state => { + console.log("rehydrated", state) + }, }, ), )