Unmount previous root before resolving setRoot promise (#6197)

* Unmount previous root before resolving setRoot promise

* Unmount hirerchy on setRoot

* Fix tests

Co-authored-by: yogevbd <yogev132@gmail.com>

e2e/StaticLifecycleEvents.test.js

     await elementById(TestIDs.PUSH_BTN).tap();
     await expect(elementByLabel('componentDidDisappear | ReactTitleView | TopBarTitle')).toBeVisible();
   });
+
+  it('unmounts previous root before resolving setRoot promise', async () => {
+    await elementById(TestIDs.SET_ROOT_BTN).tap();
+    await elementById(TestIDs.SET_ROOT_BTN).tap();
+
+    await expect(elementByLabel('setRoot complete - previous root is unmounted')).toBeVisible();
+  });
 });

lib/android/app/src/main/java/com/reactnativenavigation/viewcontrollers/navigator/Navigator.java

             @Override
             public void onSuccess(String childId) {
                 if (removeSplashView) contentLayout.removeViewAt(0);
-                super.onSuccess(childId);
                 destroyPreviousRoot();
+                super.onSuccess(childId);
             }
         }, reactInstanceManager);
     }

lib/ios/RNNCommandsHandler.m

 	}
 	
 	[_modalManager dismissAllModalsAnimated:NO completion:nil];
-	
-	UIViewController *vc = [_controllerFactory createLayout:layout[@"root"]];
+    
+    UIViewController *vc = [_controllerFactory createLayout:layout[@"root"]];
     vc.waitForRender = [vc.resolveOptionsWithDefault.animations.setRoot.waitForRender getWithDefaultValue:NO];
     __weak UIViewController* weakVC = vc;
     [vc setReactViewReadyCallback:^{
+        [self->_mainWindow.rootViewController destroy];
         self->_mainWindow.rootViewController = weakVC;
         [self->_eventEmitter sendOnNavigationCommandCompletion:setRoot commandId:commandId params:@{@"layout": layout}];
         completion();
     }];
     
-	[vc render];
+    [vc render];
 }
 
 - (void)mergeOptions:(NSString*)componentId options:(NSDictionary*)mergeOptions completion:(RNNTransitionCompletionBlock)completion {

lib/ios/RNNComponentViewController.h

 
 - (void)onButtonPress:(RNNUIBarButtonItem *)barButtonItem;
 
+- (void)destroyReactView;
+
 @end

lib/ios/RNNComponentViewController.m

     [self renderReactViewIfNeeded];
 }
 
+- (void)destroyReactView {
+    if ([self.view isKindOfClass: [RNNReactView class]]) {
+        [((RNNReactView *)self.view) invalidate];
+    }
+}
+
 - (void)renderReactViewIfNeeded {
     if (!self.isViewLoaded) {
         self.view = [self.creator createRootView:self.layoutInfo.name

lib/ios/RNNReactView.h

 
 - (void)componentDidDisappear;
 
+- (void)invalidate;
+
 @end

lib/ios/RNNReactView.m

 #import "RNNReactView.h"
+#import <React/RCTRootContentView.h>
 
 @implementation RNNReactView {
     BOOL _isAppeared;
     _isAppeared = NO;
 }
 
+- (void)invalidate {
+    [((RCTRootContentView *)self.contentView) invalidate];
+}
+
 - (NSString *)componentId {
 	return self.appProperties[@"componentId"];
 }

lib/ios/UIViewController+LayoutProtocol.h

 
 - (UIViewController *)topMostViewController;
 
+- (void)destroy;
+
 - (void)mergeOptions:(RNNNavigationOptions *)options;
 
 - (void)mergeChildOptions:(RNNNavigationOptions *)options child:(UIViewController *)child;

lib/ios/UIViewController+LayoutProtocol.m

     return nil;
 }
 
+- (void)destroy {
+    [self destroyReactView];
+    [self.presentedViewController destroy];
+    
+    for (UIViewController* child in self.childViewControllers) {
+        [child destroy];
+    }
+}
+
+- (void)destroyReactView {
+    
+}
+
 - (UIViewController *)presentedComponentViewController {
     UIViewController* currentChild = self.getCurrentChild;
     return currentChild ? currentChild.presentedComponentViewController : self;

playground/src/screens/SetRootScreen.js

 const Navigation = require('./../services/Navigation');
 const {
   NAVIGATION_TAB,
-  SET_MULTIPLE_ROOTS_BTN
+  SET_MULTIPLE_ROOTS_BTN,
+  SET_ROOT_BTN
 } = require('../testIDs');
 const Screens = require('./Screens');
+const { logLifecycleEvent } = require('./StaticLifecycleOverlay');
+let unmounted;
 
 class SetRootScreen  extends React.Component {
   static options() {
     };
   }
 
+  constructor(props) {
+    super(props);
+    unmounted = false;
+  }
+
   render() {
     return (
       <Root componentId={this.props.componentId}>
+        <Button label='Set Root' testID={SET_ROOT_BTN} onPress={this.setSingleRoot} />
         <Button label='Set Multiple Roots' testID={SET_MULTIPLE_ROOTS_BTN} onPress={this.setMultipleRoot} />
       </Root>
     );
   }
 
+  componentWillUnmount() {
+    unmounted = true;
+  }
+
+  setSingleRoot = async () => {
+    await this.setRoot();
+    logLifecycleEvent({text: `setRoot complete - previous root is${unmounted ? '' : ' not'} unmounted`});
+  }
+
   setMultipleRoot = async () => {
     await this.setRoot();
     await this.setRoot();

playground/src/screens/StaticEventsScreen.js

   PUSH_BTN,
   POP_BTN,
   STATIC_EVENTS_OVERLAY_BTN,
-  MODAL_BTN
+  MODAL_BTN,
+  SET_ROOT_BTN
+
 } = require('../testIDs');
 const Screens = require('./Screens');
 
         <Button label='Push' testID={PUSH_BTN} onPress={this.push} />
         <Button label='Pop' testID={POP_BTN} onPress={this.pop} />
         <Button label='Show Modal' testID={MODAL_BTN} onPress={this.showModal} />
+        <Button label='Set Root' testID={SET_ROOT_BTN} onPress={this.setRoot} />
       </Root>
     );
   }
   });
   push = () => Navigation.push(this, Screens.Pushed);
   pop = () => Navigation.pop(this);
+  setRoot = () => Navigation.setRoot(Screens.SetRoot);
 }
 
 module.exports = StaticEventsScreen;

playground/src/screens/StaticLifecycleOverlay.js

 const { Navigation } = require('react-native-navigation');
 const TestIDs = require('../testIDs');
 
+let _overlayInstance;
+const logLifecycleEvent = (event) => {
+  _overlayInstance.setState({
+    events: [..._overlayInstance.state.events, event]
+  });
+}
+
 class StaticLifecycleOverlay extends Component {
   static options() {
     return {
       }
     }
   }
+
+  componentDidMount() {
+    _overlayInstance = this;
+  }
+
+  componentWillUnmount() {
+    _overlayInstance = null;
+  }
+
   constructor(props) {
     super(props);
     this.state = {
       return <Text style={styles.h2}>{`${event.event} | ${event.componentName} | ${event.componentType}`}</Text>;
     } else if (event.buttonId) {
       return <Text style={styles.h2}>{`${event.event} | ${event.buttonId}`}</Text>;
+    } else if (event.text){
+      return <Text style={styles.h2}>{`${event.text}`}</Text>;
     } else {
       return <Text style={styles.h2}>{`${event.event} | ${event.componentId}`}</Text>;
     }
     );
   }
 }
-module.exports = StaticLifecycleOverlay;
+module.exports = {
+  StaticLifecycleOverlay,
+  logLifecycleEvent
+}
 
 const styles = {
   root: {

playground/src/screens/index.js

   Navigation.registerComponent(Screens.CocktailDetailsScreen, () => require('./sharedElementTransition/CocktailDetailsScreen'));
   Navigation.registerComponent(Screens.CocktailsListScreen, () => require('./sharedElementTransition/CocktailsListScreen'));
   Navigation.registerComponent(Screens.CocktailsListMasterScreen, () => require('./splitView/CocktailsListMasterScreen'));
-  Navigation.registerComponent(Screens.EventsOverlay, () => require('./StaticLifecycleOverlay'));
+  Navigation.registerComponent(Screens.EventsOverlay, () => require('./StaticLifecycleOverlay').StaticLifecycleOverlay);
   Navigation.registerComponent(Screens.EventsScreen, () => require('./StaticEventsScreen'));
   Navigation.registerComponent(Screens.ExternalComponent, () => require('./ExternalComponentScreen'));
   Navigation.registerComponent(Screens.FirstBottomTabsScreen, () => require('./FirstBottomTabScreen'));