Add auth storage

phisn · Apr 3, 2024 · 42c31f8 · 42c31f8
1 parent 1852373
commit 42c31f8
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 4 deletions.
diff --git 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
@@ -68,7 +68,7 @@ export function RenameAccount(props: { open: boolean; onFinished: () => void })
                         </label>
                         {loading === false && (
                             <div className="btn w-44" onClick={onRename}>
-                                Create Account
+                                Rename
                             </div>
                         )}
                         {loading && (

diff --git 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<AppStore>()(
             storage: createJSONStorage(() => storage),
 
             partialize: state => ({
-                jwt: state.jwt ?? undefined,
+                jwt: state.jwt,
             }),
-            onRehydrateStorage: () => () => {},
+            onRehydrateStorage: () => state => {
+                console.log("rehydrated", state)
+            },
         },
     ),
 )