Set BottomTab style after TitleBar back button click

android/app/src/main/java/com/reactnativenavigation/controllers/Modal.java

 
     private final AppCompatActivity activity;
     private final OnModalDismissedListener onModalDismissedListener;
+    private final ScreenParams screenParams;
+    private Layout layout;
 
     public void setTopBarVisible(String screenInstanceId, boolean hidden, boolean animated) {
         layout.setTopBarVisible(screenInstanceId, hidden, animated);
         layout.setTitleBarLeftButton(screenInstanceId, navigatorEventId, titleBarLeftButton);
     }
 
+    @Override
+    public void onTitleBarBackPress() {
+        layout.onTitleBarBackPress();
+    }
+
     public interface OnModalDismissedListener {
         void onModalDismissed(Modal modal);
     }
 
-    private final ScreenParams screenParams;
-    private Layout layout;
-
     public Modal(AppCompatActivity activity, OnModalDismissedListener onModalDismissedListener, ScreenParams screenParams) {
         super(activity, R.style.Modal);
         this.activity = activity;

android/app/src/main/java/com/reactnativenavigation/controllers/ModalController.java

             modal.setTitleBarLeftButton(screenInstanceId, navigatorEventId, titleBarLeftButton);
         }
     }
+
+    @Override
+    public void onTitleBarBackPress() {
+        // Do nothing and let the layout handle the back press
+    }
 }

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

     }
 
     private void createAndAddScreenStack(int position) {
-        ScreenStack newStack = new ScreenStack(activity, params.tabParams.get(position));
+        ScreenStack newStack = new ScreenStack(activity, params.tabParams.get(position), this);
         screenStacks[position] = newStack;
         newStack.setVisibility(INVISIBLE);
         LayoutParams lp = new LayoutParams(MATCH_PARENT, MATCH_PARENT);
         currentScreenStack.destroy();
         removeView(currentScreenStack);
 
-        ScreenStack newStack = new ScreenStack(activity, params);
+        ScreenStack newStack = new ScreenStack(activity, params, this);
         screenStacks[currentStackIndex] = newStack;
         addView(newStack, 0, new RelativeLayout.LayoutParams(MATCH_PARENT, MATCH_PARENT));
 
     private void setBottomTabsStyleFromCurrentScreen() {
         bottomTabs.setStyleFromScreen(getCurrentScreenStack().getCurrentScreenStyleParams());
     }
+
+    @Override
+    public void onTitleBarBackPress() {
+        if (getCurrentScreenStack().canPop()) {
+            getCurrentScreenStack().pop();
+            setBottomTabsStyleFromCurrentScreen();
+        }
+    }
 }

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

 package com.reactnativenavigation.layouts;
 
 import com.reactnativenavigation.params.ScreenParams;
+import com.reactnativenavigation.views.TitleBarBackButtonListener;
 
-public interface ScreenStackContainer {
+public interface ScreenStackContainer extends TitleBarBackButtonListener {
     void push(ScreenParams screenParams);
 
     void pop(ScreenParams screenParams);

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

             stack.destroy();
             removeView(stack);
         }
-        stack = new ScreenStack(activity, screenParams);
+        stack = new ScreenStack(activity, screenParams, this);
         addView(stack, new LayoutParams(MATCH_PARENT, MATCH_PARENT));
     }
 
     public void setTitleBarLeftButton(String screenInstanceId, String navigatorEventId, TitleBarLeftButtonParams titleBarLeftButtonParams) {
         stack.setScreenTitleBarLeftButton(screenInstanceId, navigatorEventId, titleBarLeftButtonParams);
     }
+
+    @Override
+    public void onTitleBarBackPress() {
+        if (stack.canPop()) {
+            stack.pop();
+        }
+    }
 }

android/app/src/main/java/com/reactnativenavigation/screens/ScreenStack.java

 import static android.view.ViewGroup.LayoutParams.MATCH_PARENT;
 
 // TODO there's really no reason for ScreenStack to extend FrameLayout. All screens can be added to parent.
-public class ScreenStack extends FrameLayout implements TitleBarBackButtonListener {
+public class ScreenStack extends FrameLayout {
 
     private final AppCompatActivity activity;
+    private TitleBarBackButtonListener titleBarBackButtonListener;
     private Stack<Screen> stack = new Stack<>();
 
-    public ScreenStack(AppCompatActivity activity, ScreenParams initialScreenParams) {
+    public ScreenStack(AppCompatActivity activity, ScreenParams initialScreenParams, TitleBarBackButtonListener titleBarBackButtonListener) {
         super(activity);
         this.activity = activity;
+        this.titleBarBackButtonListener = titleBarBackButtonListener;
         setLayoutTransition(new LayoutTransition());
         pushInitialScreen(initialScreenParams);
     }
     }
 
     private void addScreen(ScreenParams screenParams) {
-        Screen screen = ScreenFactory.create(activity, screenParams, this);
+        Screen screen = ScreenFactory.create(activity, screenParams, titleBarBackButtonListener);
         addView(screen, new FrameLayout.LayoutParams(MATCH_PARENT, MATCH_PARENT));
         stack.push(screen);
     }
         performOnScreen(screenInstanceId, new Task<Screen>() {
             @Override
             public void run(Screen param) {
-                param.setTitleBarLeftButton(navigatorEventId, ScreenStack.this, titleBarLeftButtonParams);
+                param.setTitleBarLeftButton(navigatorEventId, titleBarBackButtonListener, titleBarLeftButtonParams);
             }
         });
     }
 
-    @Override
-    public void onTitleBarBackPress() {
-        if (canPop()) {
-            pop();
-        }
-    }
-
     public StyleParams getCurrentScreenStyleParams() {
         return stack.peek().getStyleParams();
     }