Catch ScreenStackNotFoundException

Perform all stack related commands in performOnStack which
doesn't crash when requested stack is no found

android/app/src/main/java/com/reactnativenavigation/layouts/BottomTabsLayout.java

@@ -32,6 +32,7 @@ import com.reactnativenavigation.params.TitleBarLeftButtonParams;
 import com.reactnativenavigation.screens.NavigationType;
 import com.reactnativenavigation.screens.Screen;
 import com.reactnativenavigation.screens.ScreenStack;
+import com.reactnativenavigation.utils.Task;
 import com.reactnativenavigation.utils.ViewUtils;
 import com.reactnativenavigation.views.BottomTabs;
 import com.reactnativenavigation.views.LightBox;
@@ -333,14 +334,13 @@ public class BottomTabsLayout extends BaseLayout implements AHBottomNavigation.O
         bottomTabs.setCurrentItem(index);
     }
 
-    public void selectBottomTabByNavigatorId(String navigatorId) {
-        try {
-            bottomTabs.setCurrentItem(getScreenStackIndex(navigatorId));
-        } catch (ScreenStackNotFoundException e) {
-            Log.e("BottomTabsLayout", "Could not select bottom tab by navigatorId [" + navigatorId + "]." +
-                                      "This should not have happened, it probably means navigator.switchToTab()" +
-                                      "was called from an unmounted tab.");
-        }
+    public void selectBottomTabByNavigatorId(final String navigatorId) {
+        performOnStack(navigatorId, new Task<ScreenStack>() {
+            @Override
+            public void run(ScreenStack param) {
+                bottomTabs.setCurrentItem(getScreenStackIndex(navigatorId));
+            }
+        });
     }
 
     private boolean hasBackgroundColor(StyleParams params) {
@@ -358,14 +358,18 @@ public class BottomTabsLayout extends BaseLayout implements AHBottomNavigation.O
     }
 
     @Override
-    public void push(ScreenParams params) {
-        ScreenStack screenStack = getScreenStack(params.getNavigatorId());
-        screenStack.push(params, createScreenLayoutParams(params));
-        setStyleFromScreen(params.styleParams);
-        if (isCurrentStack(screenStack)) {
-            alignSnackbarContainerWithBottomTabs((LayoutParams) snackbarAndFabContainer.getLayoutParams(), params.styleParams);
-            EventBus.instance.post(new ScreenChangedEvent(params));
-        }
+    public void push(final ScreenParams params) {
+        performOnStack(params.getNavigatorId(), new Task<ScreenStack>() {
+            @Override
+            public void run(ScreenStack screenStack) {
+                screenStack.push(params, createScreenLayoutParams(params));
+                setStyleFromScreen(params.styleParams);
+                if (isCurrentStack(screenStack)) {
+                    alignSnackbarContainerWithBottomTabs((LayoutParams) snackbarAndFabContainer.getLayoutParams(), params.styleParams);
+                    EventBus.instance.post(new ScreenChangedEvent(params));
+                }
+            }
+        });
     }
 
     @Override
@@ -394,13 +398,17 @@ public class BottomTabsLayout extends BaseLayout implements AHBottomNavigation.O
 
     @Override
     public void newStack(final ScreenParams params) {
-        ScreenStack screenStack = getScreenStack(params.getNavigatorId());
-        screenStack.newStack(params, createScreenLayoutParams(params));
-        if (isCurrentStack(screenStack)) {
-            setStyleFromScreen(params.styleParams);
-            alignSnackbarContainerWithBottomTabs((LayoutParams) snackbarAndFabContainer.getLayoutParams(), params.styleParams);
-            EventBus.instance.post(new ScreenChangedEvent(params));
-        }
+        performOnStack(params.getNavigatorId(), new Task<ScreenStack>() {
+            @Override
+            public void run(ScreenStack screenStack) {
+                screenStack.newStack(params, createScreenLayoutParams(params));
+                if (isCurrentStack(screenStack)) {
+                    setStyleFromScreen(params.styleParams);
+                    alignSnackbarContainerWithBottomTabs((LayoutParams) snackbarAndFabContainer.getLayoutParams(), params.styleParams);
+                    EventBus.instance.post(new ScreenChangedEvent(params));
+                }
+            }
+        });
     }
 
     private void alignSnackbarContainerWithBottomTabs(LayoutParams lp, StyleParams styleParams) {
@@ -411,6 +419,17 @@ public class BottomTabsLayout extends BaseLayout implements AHBottomNavigation.O
         }
     }
 
+    private void performOnStack(String navigatorId, Task<ScreenStack> task) {
+        try {
+            ScreenStack screenStack = getScreenStack(navigatorId);
+            task.run(screenStack);
+        } catch (ScreenStackNotFoundException e) {
+            Log.e("Navigation", "Could not perform action on stack [" + navigatorId + "]." +
+                                      "This should not have happened, it probably means a navigator action" +
+                                      "was called from an unmounted tab.");
+        }
+    }
+
     @Override
     public void destroy() {
         snackbarAndFabContainer.destroy();
@@ -485,7 +504,7 @@ public class BottomTabsLayout extends BaseLayout implements AHBottomNavigation.O
 
     private
     @NonNull
-    ScreenStack getScreenStack(String navigatorId) {
+    ScreenStack getScreenStack(String navigatorId) throws ScreenStackNotFoundException {
         int index = getScreenStackIndex(navigatorId);
         return screenStacks[index];
     }