From 5861358e9094bea0ecca233feda9fbd8ff1945b2 Mon Sep 17 00:00:00 2001
From: Max Zhang <pipi-max@hotmail.com>
Date: Sat, 15 May 2021 11:49:59 -0700
Subject: [PATCH] Back to Map or Qr after payment success

---
 src/containers/Navbar/NavBarScreen.tsx | 28 +++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/containers/Navbar/NavBarScreen.tsx b/src/containers/Navbar/NavBarScreen.tsx
index c066d5d..42bd764 100644
--- a/src/containers/Navbar/NavBarScreen.tsx
+++ b/src/containers/Navbar/NavBarScreen.tsx
@@ -10,7 +10,9 @@ import Back from '../../assets/back.png';
 import Help from '../../assets/help.png';
 import Logo from '../../assets/logo.png';
 import Menu from '../../assets/menu.png';
-import React from 'react';
+import React, { useContext } from 'react';
+
+import { Ctx } from '../../components/StateProvider';
 
 export const NavCenter = () => <Image style={styles.logo} source={Logo} />;
 
@@ -35,6 +37,8 @@ export const HelpButton = () => {
 };
 
 export const HeaderButton = ({ navigation, screen }: Props) => {
+	const { state } = useContext(Ctx);
+
 	const defaultConfig: ButtonConfig = {
 		action: () => {
 			if (navigation.canGoBack()) {
@@ -72,6 +76,28 @@ export const HeaderButton = ({ navigation, screen }: Props) => {
 				}),
 			icon: Back,
 		},
+		PaymentSuccess: {
+			action: () => {
+				if (state.user && state.isEnterpriseMode) {
+					// @ts-ignore
+					// Navigating inside a nested navigator gives an tsc error.
+					navigation.navigate('BottomBar', {
+						screen: 'Map', // go to Map from PaymentSuccess
+					});
+				} else if (state.user) {
+					// @ts-ignore
+					// Navigating inside a nested navigator gives an tsc error.
+					navigation.navigate('BottomBar', {
+						screen: 'Qr', // go to Map from PaymentSuccess
+					});
+				} else {
+					// @ts-ignore
+					// Navigating inside a nested navigator gives an tsc error.
+					navigation.navigate('Qr');
+				}
+			},
+			icon: Back,
+		},
 	};
 
 	const pressHandler = () => {