fix groceries, todo, calendar

hooks/firebase/useSetUserData.ts

@@ -0,0 +1,30 @@
+import {useAuthContext} from "@/contexts/AuthContext";
+import {useMutation} from "react-query";
+import firestore from "@react-native-firebase/firestore";
+import {UserProfile} from "@/hooks/firebase/types/profileTypes";
+import {FirebaseAuthTypes} from "@react-native-firebase/auth";
+
+export const useSetUserData = () => {
+    const {user: currentUser, setProfileData} = useAuthContext()
+
+    return useMutation({
+        mutationKey: ["updateUserData"],
+        mutationFn: async ({newUserData, customUser}: {newUserData: Partial<UserProfile>, customUser?:  FirebaseAuthTypes.User }) => {
+            const user = currentUser ?? customUser
+
+            if (user) {
+                try {
+                    await firestore()
+                        .collection("Profiles")
+                        .doc(user.uid)
+                        .set(newUserData);
+
+                    const profileData = await firestore().collection("Profiles").doc(user?.uid!).get()
+                    setProfileData(profileData.data() as UserProfile)
+                } catch (e) {
+                    console.error(e)
+                }
+            }
+        }
+    })
+}

hooks/firebase/useSignUp.ts

@@ -1,16 +1,40 @@
-import {useMutation} from "react-query";
+import { useMutation } from "react-query";
 import auth from "@react-native-firebase/auth";
-import {useUpdateUserData} from "@/hooks/firebase/useUpdateUserData";
-import {ProfileType} from "@/contexts/AuthContext";
+import { ProfileType } from "@/contexts/AuthContext";
+import { useSetUserData } from "./useSetUserData";
 
 export const useSignUp = () => {
-    const {mutateAsync: updateUserData} = useUpdateUserData()
+  const { mutateAsync: setUserData } = useSetUserData();
 
-    return useMutation({
-        mutationKey: ["signUp"],
-        mutationFn: async ({email, password, firstName, lastName}: { email: string, password: string, firstName: string, lastName: string }) => {
-            const res = await auth().createUserWithEmailAndPassword(email, password);
-            await updateUserData({newUserData: {userType: ProfileType.PARENT, firstName: firstName, lastName: lastName}, customUser: res.user});
-        }
-    });
-}
+  return useMutation({
+    mutationKey: ["signUp"],
+    mutationFn: async ({
+      email,
+      password,
+      firstName,
+      lastName,
+    }: {
+      email: string;
+      password: string;
+      firstName: string;
+      lastName: string;
+    }) => {
+      await auth()
+        .createUserWithEmailAndPassword(email, password)
+        .then(async (res) => {
+          try {
+            await setUserData({
+              newUserData: {
+                userType: ProfileType.PARENT,
+                firstName: firstName,
+                lastName: lastName,
+              },
+              customUser: res.user,
+            });
+          } catch (error) {
+            console.error(error);
+          }
+        });
+    },
+  });
+};