From 47c0405012e2110c3f05a648bb20a2b1d51fbaeb Mon Sep 17 00:00:00 2001
From: Robert Lucas <100799838+Robert-M-Lucas@users.noreply.github.com>
Date: Thu, 25 Apr 2024 12:23:41 +0100
Subject: [PATCH] Added a test page for UserPrefs

---
 src/pages/test firestore/TestFirestore.tsx | 64 ++++++++++++++++------
 src/utils/user_prefs.ts                    |  3 +-
 2 files changed, 47 insertions(+), 20 deletions(-)

diff --git a/src/pages/test firestore/TestFirestore.tsx b/src/pages/test firestore/TestFirestore.tsx
index a92d27f..434b0f4 100644
--- a/src/pages/test firestore/TestFirestore.tsx	
+++ b/src/pages/test firestore/TestFirestore.tsx	
@@ -13,6 +13,8 @@ import {getCurrentBalance} from "../../utils/transaction_utils.ts";
 import {signInWithGoogle} from "../../utils/authentication.ts";
 import {Header} from "../../components/Header.tsx";
 import { orderBy } from "firebase/firestore";
+import {User} from "firebase/auth";
+import {getUserPrefs, setUserPrefs, UserPrefs} from "../../utils/user_prefs.ts";
 
 function writeSampleData() {
     if (auth.currentUser === null) {
@@ -43,6 +45,8 @@ export  function TestFirestorePage() {
     const [transactions, setTransactions] = useState<Transaction[]>([]);
     const [balance, setBalance] = useState(0);
     const [update ,setUpdate] = useState(0);
+    const [user, setUser] = useState<string | undefined>(undefined);
+    const [userPrefs, _setUserPrefs] = useState<UserPrefs | null>(null);
 
     if (!authResolved) {
         auth.authStateReady().then(() => setAuthResolved(true));
@@ -56,6 +60,11 @@ export  function TestFirestorePage() {
     }
 
     if (auth.currentUser === null) {
+        auth.onAuthStateChanged((new_user: User | null) => {
+            if (new_user !== null && new_user.uid !== user) {
+                setUser(new_user.uid)
+            }
+        });
         return <>
                 <Header/>
                 <FullscreenCenter>
@@ -71,6 +80,7 @@ export  function TestFirestorePage() {
 
     getTransactionsFilterOrderBy(auth.currentUser, orderBy("dateTime", "desc")).then((t) => setTransactions(t));
     getCurrentBalance(auth.currentUser).then((b) => setBalance(b));
+    getUserPrefs(auth.currentUser).then((up) => _setUserPrefs(up));
 
     return (
         <>
@@ -81,30 +91,48 @@ export  function TestFirestorePage() {
                     Sign in with Google
                 </button>
             </div>
-            <button onClick={() => {
+            <div>
+                <h4>UserPrefs</h4>
+                {
+                    userPrefs ? <>
+                    <p>Goal: {userPrefs.goal}</p>
+                    <button className="mb-4" onClick={() => {
+                        userPrefs!.goal += 100;
+                        setUserPrefs(auth.currentUser!, userPrefs).then(() => setUpdate(update + 1));
+                    }}>Increment
+                    </button>
+                </> : <p>Loading</p>
+                }
+            </div>
+            <h1>
+                Transactions
+            </h1>
+            <p>
+                Balance: {balance}
+            </p>
+            <button className="mb-4" onClick={() => {
                 writeSampleData();
                 setUpdate(update + 1);
             }}>Add Sample Transaction
             </button>
-            <p>
-                Balance: {balance}
-            </p>
-            <p>
-                Transactions: <br/>
-            </p>
             {
                 transactions.map((t) => <div className="mb-4" key={t.forceGetDocName()}>
-                    <p className="m-0">{new Date(t.dateTime).toISOString()}</p>
-                    <textarea readOnly={true} style={{width: "100%", height: "320px"}} value={JSON.stringify(t, null, 4)}/>
-                    <button onClick={() => {deleteTransaction(t.forceGetDocName()).then(() => setUpdate(update + 1))}}>Delete</button>
-                    <button onClick={() => {
-                        if (auth.currentUser != null) {
-                            t.dateTime = new Date(Date.now()).valueOf();
-                            overwriteTransaction(auth.currentUser, t.forceGetDocName(), t).then(() => setUpdate(update + 1));
-                        }
-                    }}>Set Time To Now</button>
-                </div>
-            )}
+                        <p className="m-0">{new Date(t.dateTime).toISOString()}</p>
+                        <textarea readOnly={true} style={{width: "100%", height: "320px"}}
+                                  value={JSON.stringify(t, null, 4)}/>
+                        <button onClick={() => {
+                            deleteTransaction(t.forceGetDocName()).then(() => setUpdate(update + 1))
+                        }}>Delete
+                        </button>
+                        <button onClick={() => {
+                            if (auth.currentUser != null) {
+                                t.dateTime = new Date(Date.now()).valueOf();
+                                overwriteTransaction(auth.currentUser, t.forceGetDocName(), t).then(() => setUpdate(update + 1));
+                            }
+                        }}>Set Time To Now
+                        </button>
+                    </div>
+                )}
         </>
     )
 }
diff --git a/src/utils/user_prefs.ts b/src/utils/user_prefs.ts
index c2da073..2aa65b5 100644
--- a/src/utils/user_prefs.ts
+++ b/src/utils/user_prefs.ts
@@ -20,8 +20,7 @@ export class UserPrefs {
         if (!data) {
             throw Error("No data returned for snapshot!");
         }
-        const u = new UserPrefs(data.goal);
-        return u;
+        return new UserPrefs(data.goal);
     }
 
     toSendObject(): object {