From 5863ed7f4df221a864a7377ac9248abfb6539aca Mon Sep 17 00:00:00 2001
From: Milan Paunovic <milan.paunovic@visit.com>
Date: Fri, 23 Aug 2024 18:04:30 +0200
Subject: [PATCH] Fix

---
 hooks/firebase/useSignUp.ts         | 4 ++--
 hooks/firebase/useUpdateUserData.ts | 7 +++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/hooks/firebase/useSignUp.ts b/hooks/firebase/useSignUp.ts
index 0f79b41..d119798 100644
--- a/hooks/firebase/useSignUp.ts
+++ b/hooks/firebase/useSignUp.ts
@@ -9,8 +9,8 @@ export const useSignUp = () => {
     return useMutation({
         mutationKey: ["signUp"],
         mutationFn: async ({email, password}: { email: string, password: string }) => {
-            await auth().createUserWithEmailAndPassword(email, password);
-            await updateUserData({userType: ProfileType.PARENT, email, password});
+            const res=  await auth().createUserWithEmailAndPassword(email, password);
+            await updateUserData({userType: ProfileType.PARENT, email, password, customUser: res.user});
         }
     });
 }
\ No newline at end of file
diff --git a/hooks/firebase/useUpdateUserData.ts b/hooks/firebase/useUpdateUserData.ts
index 2a4bd81..a7527d2 100644
--- a/hooks/firebase/useUpdateUserData.ts
+++ b/hooks/firebase/useUpdateUserData.ts
@@ -2,13 +2,16 @@ 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 useUpdateUserData = () => {
-    const {user, setProfileData} = useAuthContext()
+    const {user: currentUser, setProfileData} = useAuthContext()
 
     return useMutation({
         mutationKey: ["updateUserData"],
-        mutationFn: async (newProfileData: Partial<UserProfile>) => {
+        mutationFn: async (newProfileData: Partial<UserProfile & { customUser?:  FirebaseAuthTypes.User }>) => {
+            const user = currentUser ?? newProfileData.customUser
+
             if (user) {
                 try {
                     console.log("yall don't even");