conditionally navigating after uploading picture

1 files changed, 9 insertions, 2 deletions

diff --git a/src/screens/suggestedPeopleOnboarding/SuggestedPeopleUploadPictureScreen.tsx b/src/screens/suggestedPeopleOnboarding/SuggestedPeopleUploadPictureScreen.tsx
index 1b30c72f..5b616c4a 100644
--- a/src/screens/suggestedPeopleOnboarding/SuggestedPeopleUploadPictureScreen.tsx
+++ b/src/screens/suggestedPeopleOnboarding/SuggestedPeopleUploadPictureScreen.tsx
@@ -1,3 +1,4 @@
+import { useNavigation } from '@react-navigation/native';
 import React, {useState} from 'react';
 import {
   Alert,
@@ -19,10 +20,12 @@ import {sendSuggestedPeoplePhoto} from '../../services';
 import {uploadedSuggestedPeoplePhoto} from '../../store/actions';
 import {normalize, SCREEN_HEIGHT, SCREEN_WIDTH} from '../../utils';
 
-const SuggestedPeopleUploadPictureScreen: React.FC = () => {
+const SuggestedPeopleUploadPictureScreen: React.FC = ({route}) => {
+  const {goTo} = route.params;
   const [image, setImage] = useState<string | undefined>(undefined);
   const [loading, setLoading] = useState(false);
   const dispatch = useDispatch();
+  const navigation = useNavigation();
 
   const openImagePicker = () => {
     ImagePicker.openPicker({
@@ -52,12 +55,16 @@ const SuggestedPeopleUploadPictureScreen: React.FC = () => {
     if (image) {
       const success = await sendSuggestedPeoplePhoto(image);
       if (success) {
-        dispatch(uploadedSuggestedPeoplePhoto());
+        const suggested_people_linked = 1;
+        dispatch(uploadedSuggestedPeoplePhoto(suggested_people_linked));
       } else {
         Alert.alert(ERROR_UPLOAD);
       }
     }
     setLoading(false);
+    if (goTo === 'Profile') {
+      navigation.goBack();
+    }
   };
 
   return (