Added setStackRoot top level api

lib/ios/RNNBridgeModule.m

@@ -48,6 +48,12 @@ RCT_EXPORT_METHOD(pop:(NSString*)componentId options:(NSDictionary*)options reso
 	}];
 }
 
+RCT_EXPORT_METHOD(setStackRoot:(NSString*)componentId layout:(NSDictionary*)layout resolver:(RCTPromiseResolveBlock)resolve rejecter:(RCTPromiseRejectBlock)reject) {
+	[_commandsHandler setStackRoot:componentId layout:layout completion:^{
+		resolve(componentId);
+	}];
+}
+
 RCT_EXPORT_METHOD(popTo:(NSString*)componentId resolver:(RCTPromiseResolveBlock)resolve rejecter:(RCTPromiseRejectBlock)reject) {
 	[_commandsHandler popTo:componentId completion:^{
 		resolve(componentId);

lib/ios/RNNCommandsHandler.h

@@ -22,6 +22,8 @@
 
 -(void) popToRoot:(NSString*)componentId completion:(RNNTransitionCompletionBlock)completion;
 
+-(void)setStackRoot:(NSString*)componentId layout:(NSDictionary*)layout completion:(RNNTransitionCompletionBlock)completion;
+
 -(void) showModal:(NSDictionary*)layout completion:(RNNTransitionCompletionBlock)completion;
 
 -(void) dismissModal:(NSString*)componentId completion:(RNNTransitionCompletionBlock)completion;

lib/ios/RNNCommandsHandler.m

@@ -59,7 +59,7 @@
 	[_controllerFactory setDefaultOptionsDict:optionsDict];
 }
 
--(void) push:(NSString*)componentId layout:(NSDictionary*)layout completion:(RNNTransitionCompletionBlock)completion {
+-(void)push:(NSString*)componentId layout:(NSDictionary*)layout completion:(RNNTransitionCompletionBlock)completion {
 	[self assertReady];
 	
 	UIViewController<RNNRootViewProtocol> *newVc = [_controllerFactory createLayoutAndSaveToStore:layout];
@@ -68,6 +68,15 @@
 	}];
 }
 
+-(void)setStackRoot:(NSString*)componentId layout:(NSDictionary*)layout completion:(RNNTransitionCompletionBlock)completion {
+	[self assertReady];
+	
+	UIViewController<RNNRootViewProtocol> *newVc = [_controllerFactory createLayoutAndSaveToStore:layout];
+	[_navigationStackManager setRoot:newVc fromComponent:componentId completion:^{
+		completion();
+	}];
+}
+
 -(void)pop:(NSString*)componentId options:(NSDictionary*)options completion:(RNNTransitionCompletionBlock)completion {
 	[self assertReady];
 	[CATransaction begin];

lib/ios/RNNNavigationStackManager.h

@@ -14,5 +14,6 @@
 -(void)pop:(NSString*)componentId withTransitionOptions:(RNNAnimationOptions*)transitionOptions;
 -(void)popTo:(NSString*)componentId;
 -(void)popToRoot:(NSString*)componentId;
+-(void)setRoot:(UIViewController<RNNRootViewProtocol> *)newRoot fromComponent:(NSString *)componentId completion:(RNNTransitionCompletionBlock)completion;
 
 @end

lib/ios/RNNNavigationStackManager.m

@@ -96,13 +96,28 @@ dispatch_queue_t RCTGetUIManagerQueue(void);
 	}
 }
 
--(void) popToRoot:(NSString*)componentId {
+-(void)popToRoot:(NSString*)componentId {
 	UIViewController* vc = [_store findComponentForId:componentId];
 	UINavigationController* nvc = [vc navigationController];
 	NSArray* poppedVCs = [nvc popToRootViewControllerAnimated:YES];
 	[self removePopedViewControllers:poppedVCs];
 }
 
+-(void)setRoot:(UIViewController<RNNRootViewProtocol> *)newRoot fromComponent:(NSString *)componentId completion:(RNNTransitionCompletionBlock)completion {
+	UIViewController* vc = [_store findComponentForId:componentId];
+	UINavigationController* nvc = [vc navigationController];
+	[CATransaction begin];
+	[CATransaction setCompletionBlock:^{
+		if (completion) {
+			completion();
+		}
+	}];
+	
+	[nvc setViewControllers:@[newRoot] animated:newRoot.options.animated];
+	
+	[CATransaction commit];
+}
+
 -(void)removePopedViewControllers:(NSArray*)viewControllers {
 	for (UIViewController *popedVC in viewControllers) {
 		[_store removeComponentByViewControllerInstance:popedVC];

lib/src/Navigation.ts

@@ -123,6 +123,13 @@ export class Navigation {
     return this.commands.popToRoot(componentId);
   }
 
+  /**
+   * Sets new root component to stack.
+   */
+  public setStackRoot(componentId: string, layout): Promise<any> {
+    return this.commands.setStackRoot(componentId, layout);
+  }
+
   /**
    * Show overlay on top of the entire app
    */

lib/src/adapters/NativeCommandsSender.ts

@@ -34,6 +34,10 @@ export class NativeCommandsSender {
     return this.nativeCommandsModule.popToRoot(componentId);
   }
 
+  setStackRoot(onComponentId: string, layout: object) {
+    return this.nativeCommandsModule.setStackRoot(onComponentId, layout);
+  }
+
   showModal(layout: object) {
     return this.nativeCommandsModule.showModal(layout);
   }

lib/src/commands/Commands.test.ts

@@ -245,6 +245,22 @@ describe('Commands', () => {
     });
   });
 
+  describe('setStackRoot', () => {
+    it('parses into correct layout node and sends to native', () => {
+      uut.setStackRoot('theComponentId', { component: { name: 'com.example.MyScreen' } });
+      expect(mockCommandsSender.setStackRoot).toHaveBeenCalledTimes(1);
+      expect(mockCommandsSender.setStackRoot).toHaveBeenCalledWith('theComponentId', {
+        type: 'Component',
+        id: 'Component+UNIQUE_ID',
+        data: {
+          name: 'com.example.MyScreen',
+          options: {}
+        },
+        children: []
+      });
+    });
+  });
+
   describe('showOverlay', () => {
     it('sends command to native after parsing into a correct layout tree', () => {
       uut.showOverlay({
@@ -352,6 +368,7 @@ describe('Commands', () => {
         pop: ['id', {}],
         popTo: ['id'],
         popToRoot: ['id'],
+        setStackRoot: ['id', {}],
         showOverlay: [{}],
         dismissOverlay: ['id'],
       };
@@ -366,6 +383,7 @@ describe('Commands', () => {
         pop: { componentId: 'id', options: {} },
         popTo: { componentId: 'id' },
         popToRoot: { componentId: 'id' },
+        setStackRoot: { componentId: 'id', layout: 'parsed' },
         showOverlay: { layout: 'parsed' },
         dismissOverlay: { componentId: 'id' },
       };

lib/src/commands/Commands.ts

@@ -90,6 +90,17 @@ export class Commands {
     return result;
   }
 
+  public setStackRoot(componentId, simpleApi) {
+    const input = _.cloneDeep(simpleApi);
+
+    const layout = this.layoutTreeParser.parse(input);
+    this.layoutTreeCrawler.crawl(layout);
+
+    const result = this.nativeCommandsSender.setStackRoot(componentId, layout);
+    this.commandsObserver.notify('setStackRoot', { componentId, layout });
+    return result;
+  }
+
   public showOverlay(simpleApi) {
     const input = _.cloneDeep(simpleApi);
 

playground/src/screens/PushedScreen.js

@@ -24,6 +24,7 @@ class PushedScreen extends Component {
     this.onClickPopPrevious = this.onClickPopPrevious.bind(this);
     this.onClickPopToFirstPosition = this.onClickPopToFirstPosition.bind(this);
     this.onClickPopToRoot = this.onClickPopToRoot.bind(this);
+    this.onClickSetStackRoot = this.onClickSetStackRoot.bind(this);
   }
 
   render() {
@@ -36,6 +37,7 @@ class PushedScreen extends Component {
         <Button title='Pop' testID={testIDs.POP_BUTTON} onPress={this.onClickPop} />
         <Button title='Pop Previous' testID={testIDs.POP_PREVIOUS_BUTTON} onPress={this.onClickPopPrevious} />
         <Button title='Pop To Root' testID={testIDs.POP_TO_ROOT} onPress={this.onClickPopToRoot} />
+        <Button title='Set Stack Root' testID={testIDs.SET_STACK_ROOT_BUTTON} onPress={this.onClickSetStackRoot} />
         {stackPosition > 2 && <Button title='Pop To Stack Position 1' testID={testIDs.POP_STACK_POSITION_ONE_BUTTON} onPress={this.onClickPopToFirstPosition} />}
         <Text style={styles.footer}>{`this.props.componentId = ${this.props.componentId}`}</Text>
       </View>
@@ -77,6 +79,25 @@ class PushedScreen extends Component {
     await Navigation.popToRoot(this.props.componentId);
   }
 
+  async onClickSetStackRoot() {
+    await Navigation.setStackRoot(this.props.componentId, {
+      component: {
+        name: 'navigation.playground.PushedScreen',
+        passProps: {
+          stackPosition: this.getStackPosition() + 1,
+          previousScreenIds: _.concat([], this.props.previousScreenIds || [], this.props.componentId)
+        },
+        options: {
+          topBar: {
+            title: {
+              text: `Pushed ${this.getStackPosition() + 1}`
+            }
+          }
+        }
+      }
+    });
+  }
+
   getStackPosition() {
     return this.props.stackPosition || 1;
   }

playground/src/testIDs.js

@@ -58,6 +58,7 @@ module.exports = {
   COMPLEX_LAYOUT_BUTTON: `COMPLEX_LAYOUT_BUTTON`,
   EXTERNAL_COMPONENT_IN_STACK: `EXTERNAL_COMPONENT_IN_STACK`,
   EXTERNAL_COMPONENT_IN_DEEP_STACK: `EXTERNAL_COMPONENT_IN_DEEP_STACK`,
+  SET_STACK_ROOT_BUTTON: `SET_STACK_ROOT_BUTTON`,
 
   // Elements
   SCROLLVIEW_ELEMENT: `SCROLLVIEW_ELEMENT`,