From b0caa50e7cb6a38e748b5edf51690ead59237a72 Mon Sep 17 00:00:00 2001
From: Joseph D'Souza <joseph.souza@trya.it>
Date: Wed, 11 Feb 2026 13:13:23 +0100
Subject: [PATCH] refactor and update app navigator

---
 mobile/src/navigation/AppNavigator.tsx | 38 +++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 7 deletions(-)

diff --git a/mobile/src/navigation/AppNavigator.tsx b/mobile/src/navigation/AppNavigator.tsx
index 6c8ab58..a39a3e5 100644
--- a/mobile/src/navigation/AppNavigator.tsx
+++ b/mobile/src/navigation/AppNavigator.tsx
@@ -2,16 +2,20 @@ import React from 'react';
 import { View, ActivityIndicator } from 'react-native';
 import { createNativeStackNavigator } from '@react-navigation/native-stack';
 import AuthScreen from '../screens/AuthScreen';
-import HomeScreen from '../screens/HomeScreen';
 import TabNavigator from './TabNavigator';
+import EditProfileScreen from '../screens/EditProfileScreen';
+import DocumentViewScreen from '../screens/DocumentViewScreen';
 import ForgotPasswordScreen from '../screens/ForgotPasswordScreen';
 import { useAuthStore } from '../store/useAuthStore';
 
-// Define the types for your stack
+// Define the types for the stack
 export type RootStackParamList = {
   Auth: undefined;
   Home: undefined; // This is now a nested navigator
-  Main: undefined; // The tab navigator 'Main'
+  Main: undefined; // The tab navigator
+  EditProfile: undefined;
+  DocumentView: { uri: string };
+  ForgotPassword: undefined;
 };
 
 const Stack = createNativeStackNavigator<RootStackParamList>();
@@ -29,20 +33,40 @@ export default function AppNavigator() {
   }
 
   return (
-    <Stack.Navigator screenOptions={{ headerShown: false }}>
+    <Stack.Navigator id="RootStack" screenOptions={{ headerShown: false }}>
       {user ? (
         // Screens for logged-in users
-                      <Stack.Screen name="Main" component={TabNavigator} options={{ headerShown: false }} />
+        <>
+          <Stack.Screen name="Main" component={TabNavigator} options={{ headerShown: false }} />
+          <Stack.Screen
+            name="EditProfile"
+            component={EditProfileScreen}
+            options={{
+              title: 'Edit Profile',
+              headerBackTitle: '',
+              headerShown: true,
+            }}
+          />
+          <Stack.Screen
+            name="DocumentView"
+            component={DocumentViewScreen}
+            options={{
+              title: 'Document',
+              headerBackTitle: '',
+              headerShown: true,
+            }}
+          />
+        </>
       ) : (
         // Screens for logged-out users
-        <Stack.Group>
+        <>
           <Stack.Screen name="Auth" component={AuthScreen} />
           <Stack.Screen 
             name="ForgotPassword" 
             component={ForgotPasswordScreen}
             options={{ presentation: 'fullScreenModal' }}
           />
-        </Stack.Group>
+        </>
       )}
     </Stack.Navigator>
   );