feat: reworked back to portal button logic (#2375)

Co-authored-by: Alon Peretz <[email protected]>
ballerine-io · May 19, 2024 · c575956 · c575956
1 parent a654561
commit c575956
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 10 deletions.
diff --git a/apps/kyb-app/src/components/layouts/AppShell/Navigation.tsx b/apps/kyb-app/src/components/layouts/AppShell/Navigation.tsx
@@ -15,7 +15,7 @@ export const Navigation = () => {
   const { stateApi } = useStateManagerContext();
   const { currentPage } = usePageResolverContext();
   const { customer } = useCustomer();
-  const exitFromApp = useAppExit();
+  const { exit, isExitAvailable } = useAppExit();
 
   const isFirstStep = currentPage?.number === 1;
   const isDisabled = state.isLoading;
@@ -26,9 +26,11 @@ export const Navigation = () => {
       return;
     }
 
-    exitFromApp();
+    exit();
     return;
-  }, [stateApi, exitFromApp]);
+  }, [stateApi, exit]);
+
+  if (isFirstStep && !isExitAvailable) return null;
 
   return (
     <button

diff --git a/apps/kyb-app/src/hooks/useAppExit/useAppExit.ts b/apps/kyb-app/src/hooks/useAppExit/useAppExit.ts
@@ -25,5 +25,9 @@ export const useAppExit = () => {
     }
   }, [trackExit, customer]);
 
-  return exit;
+  return {
+    exit,
+    isExitAvailable:
+      uiSchema?.config?.kybOnExitAction === 'send-event' ? true : !!customer?.websiteUrl,
+  };
 };
diff --git a/apps/kyb-app/src/pages/CollectionFlow/components/pages/Approved/Approved.tsx b/apps/kyb-app/src/pages/CollectionFlow/components/pages/Approved/Approved.tsx
@@ -9,7 +9,7 @@ import { Button, Card } from '@ballerine/ui';
 export const Approved = withSessionProtected(() => {
   const { t } = useTranslation();
   const { customer } = useCustomer();
-  const exitFromApp = useAppExit();
+  const { exit, isExitAvailable } = useAppExit();
 
   return (
     <div className="flex h-full items-center justify-center">
@@ -29,9 +29,9 @@ export const Approved = withSessionProtected(() => {
             {t('approved.content', { companyName: customer?.displayName })}
           </p>
         </div>
-        {customer && (
+        {customer && isExitAvailable && (
           <div className="flex justify-center">
-            <Button variant="secondary" onClick={exitFromApp}>
+            <Button variant="secondary" onClick={exit}>
               {t('backToPortal', { companyName: customer.displayName })}
             </Button>
           </div>

diff --git a/apps/kyb-app/src/pages/CollectionFlow/components/pages/Success/Success.tsx b/apps/kyb-app/src/pages/CollectionFlow/components/pages/Success/Success.tsx
@@ -10,7 +10,7 @@ export const Success = withSessionProtected(() => {
   const { t } = useTranslation();
   const { customer } = useCustomer();
 
-  const exitFromApp = useAppExit();
+  const { exit, isExitAvailable } = useAppExit();
 
   return (
     <div className="flex h-full items-center justify-center">
@@ -29,9 +29,9 @@ export const Success = withSessionProtected(() => {
             {t('success.content')}
           </h2>
         </div>
-        {customer?.displayName && (
+        {customer?.displayName && isExitAvailable && (
           <div className="flex justify-center">
-            <Button variant="secondary" onClick={exitFromApp}>
+            <Button variant="secondary" onClick={exit}>
               {t('backToPortal', { companyName: customer.displayName })}
             </Button>
           </div>