feat: android orientation management

README.md

@@ -272,6 +272,21 @@ Customize the weight of the font to be used for the large title.
 
 Boolean that allows for disabling drop shadow under navigation header when the edge of any scrollable content reaches the matching edge of the navigation bar.
 
+#### `screenOrientation`
+
+Sets the current screen's available orientations and forces rotation if current orientation is not included. Possible values:
+
+- `default` - on iOS, it resolves to [UIInterfaceOrientationMaskAllButUpsideDown](https://developer.apple.com/documentation/uikit/uiinterfaceorientationmask/uiinterfaceorientationmaskallbutupsidedown?language=objc). On Android, this lets the system decide the best orientation.
+- `all`
+- `portrait`
+- `portrait_up`
+- `portrait_down`
+- `landscape`
+- `landscape_left`
+- `landscape_right`
+
+Defaults to `default`.
+
 #### `statusBarAnimation` (iOS only)
 
 Sets the status bar animation (similar to the `StatusBar` component). Requires enabling (or deleting) `View controller-based status bar appearance` in your Info.plist file. Defaults to `fade`.

TestsExample/src/Test42.js

@@ -4,6 +4,7 @@ import React from 'react';
 import {ScrollView, StyleSheet, View, Button, Text} from 'react-native';
 import {createNativeStackNavigator} from 'react-native-screens/native-stack';
 import {createBottomTabNavigator} from '@react-navigation/bottom-tabs';
+import {createStackNavigator} from '@react-navigation/stack';
 
 const Stack = createNativeStackNavigator();
 
@@ -22,8 +23,8 @@ export default function NativeNavigation() {
           }}
         />
         <Stack.Screen
-          name="TabNavigator"
-          component={TabNavigator}
+          name="NestedNavigator"
+          component={NestedNavigator}
           options={{
             screenOrientation: 'landscape',
           }}
@@ -33,13 +34,14 @@ export default function NativeNavigation() {
   );
 }
 
+// change to createStackNavigator to test with stack in the middle
 const Tab = createBottomTabNavigator();
 
-const TabNavigator = (props) => (
+const NestedNavigator = (props) => (
   <Tab.Navigator screensEnabled={true}>
-    <Tab.Screen name="Tab1" component={Home} />
-    <Tab.Screen name="Tab2" component={Inner} />
-    <Tab.Screen name="Tab3" component={Home} />
+    <Tab.Screen name="Screen1" component={Home} />
+    <Tab.Screen name="Screen2" component={Inner} />
+    <Tab.Screen name="Screen3" component={Home} />
   </Tab.Navigator>
 );
 
@@ -63,9 +65,15 @@ function Home({navigation}) {
       >
       <View style={styles.leftTop} />
       <Button
-        title="TabNavigator"
+        title="NestedNavigator"
         onPress={() => {
-          navigation.push('TabNavigator');
+          navigation.push('NestedNavigator');
+        }}
+      />
+      <Button
+        title="Screen2"
+        onPress={() => {
+          navigation.navigate('Screen2');
         }}
       />
       <Button

android/src/main/java/com/swmansion/rnscreens/Screen.java

@@ -150,6 +150,14 @@ protected void onAttachedToWindow() {
     }
   }
 
+  protected ScreenStackHeaderConfig getHeaderConfig() {
+    View child = getChildAt(0);
+    if (child instanceof ScreenStackHeaderConfig) {
+      return (ScreenStackHeaderConfig) child;
+    }
+    return null;
+  }
+
   /**
    * While transitioning this property allows to optimize rendering behavior on Android and provide
    * a correct blending options for the animated screen. It is turned on automatically by the container

android/src/main/java/com/swmansion/rnscreens/ScreenContainer.java

@@ -136,8 +136,8 @@ protected void removeScreenAt(int index) {
   }
 
   protected void removeAllScreens() {
-    for (int i = 0, size = mScreenFragments.size(); i < size; i++) {
-      mScreenFragments.get(i).getScreen().setContainer(null);
+    for (ScreenFragment screenFragment: mScreenFragments) {
+      screenFragment.getScreen().setContainer(null);
     }
     mScreenFragments.clear();
     markUpdated();
@@ -151,6 +151,15 @@ protected Screen getScreenAt(int index) {
     return mScreenFragments.get(index).getScreen();
   }
 
+  public @Nullable Screen getTopScreen() {
+    for (ScreenFragment screenFragment: mScreenFragments) {
+      if (getActivityState(screenFragment) == Screen.ActivityState.ON_TOP) {
+        return screenFragment.getScreen();
+      }
+    }
+    return null;
+  }
+
   private void setFragmentManager(FragmentManager fm) {
     mFragmentManager = fm;
     updateIfNeeded();
@@ -328,13 +337,13 @@ private final void onUpdate() {
     mFragmentManager.executePendingTransactions();
 
     performUpdate();
+    notifyContainerUpdate();
   }
 
   protected void performUpdate() {
     // detach screens that are no longer active
     Set<Fragment> orphaned = new HashSet<>(mFragmentManager.getFragments());
-    for (int i = 0, size = mScreenFragments.size(); i < size; i++) {
-      ScreenFragment screenFragment = mScreenFragments.get(i);
+    for (ScreenFragment screenFragment: mScreenFragments) {
       if (getActivityState(screenFragment) == Screen.ActivityState.INACTIVE && screenFragment.isAdded()) {
         detachScreen(screenFragment);
       }
@@ -353,18 +362,16 @@ protected void performUpdate() {
 
     boolean transitioning = true;
 
-    for (int i = 0, size = mScreenFragments.size(); i < size; i++) {
-      ScreenFragment screenFragment = mScreenFragments.get(i);
-      if (getActivityState(screenFragment) == Screen.ActivityState.ON_TOP) {
-        // if there is an "onTop" screen it means the transition has ended
-        transitioning = false;
-      }
+
+    Screen topScreen = getTopScreen();
+    if (topScreen != null) {
+      // if there is an "onTop" screen it means the transition has ended
+      transitioning = false;
     }
 
     // attach newly activated screens
     boolean addedBefore = false;
-    for (int i = 0, size = mScreenFragments.size(); i < size; i++) {
-      ScreenFragment screenFragment = mScreenFragments.get(i);
+    for (ScreenFragment screenFragment: mScreenFragments) {
       Screen.ActivityState activityState = getActivityState(screenFragment);
       if (activityState != Screen.ActivityState.INACTIVE && !screenFragment.isAdded()) {
         addedBefore = true;
@@ -377,4 +384,11 @@ protected void performUpdate() {
 
     tryCommitTransaction();
   }
+
+  protected void notifyContainerUpdate() {
+    Screen topScreen = getTopScreen();
+    if (topScreen != null) {
+      topScreen.getFragment().onContainerUpdate();
+    }
+  }
 }

android/src/main/java/com/swmansion/rnscreens/ScreenFragment.java

@@ -1,13 +1,11 @@
 package com.swmansion.rnscreens;
 
 import android.annotation.SuppressLint;
-import android.content.Context;
 import android.os.Bundle;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
 import android.view.ViewParent;
-import android.view.inputmethod.InputMethodManager;
 import android.widget.FrameLayout;
 
 import com.facebook.react.bridge.ReactContext;
@@ -38,6 +36,7 @@ protected static View recycleView(View view) {
   }
 
   protected Screen mScreenView;
+  protected boolean mHasChildWithConfig = false;
   private List<ScreenContainer> mChildScreenContainers = new ArrayList<>();
 
   public ScreenFragment() {
@@ -66,6 +65,68 @@ public Screen getScreen() {
     return mScreenView;
   }
 
+  public void onContainerUpdate() {
+    mHasChildWithConfig = hasChildScreenWithConfig(getScreen());
+    markParentFragments(mHasChildWithConfig);
+    if (!mHasChildWithConfig) {
+      // if there is no child with config, we look for a parent with config to set the orientation
+      ScreenStackHeaderConfig config = findHeaderConfig();
+      if (config != null && config.getScreenFragment().getActivity() != null) {
+        config.getScreenFragment().getActivity().setRequestedOrientation(config.getScreenOrientation());
+      }
+    }
+  }
+
+  private @Nullable ScreenStackHeaderConfig findHeaderConfig() {
+    ViewParent parent = getScreen().getContainer();
+    while (parent != null) {
+      if (parent instanceof Screen) {
+        ScreenStackHeaderConfig headerConfig = ((Screen) parent).getHeaderConfig();
+        if (headerConfig != null) {
+          return headerConfig;
+        }
+        parent = ((Screen) parent).getContainer();
+      } else {
+        parent = parent.getParent();
+      }
+    }
+    return null;
+  }
+
+  protected boolean hasChildScreenWithConfig(Screen screen) {
+    if (screen == null) {
+      return false;
+    }
+    for (ScreenContainer sc : screen.getFragment().getChildScreenContainers()) {
+      // we check only the top screen for header config
+      Screen topScreen = sc.getTopScreen();
+      ScreenStackHeaderConfig headerConfig = topScreen != null ? topScreen.getHeaderConfig(): null;
+      if (headerConfig != null) {
+        return true;
+      }
+      if (topScreen.getFragment().mHasChildWithConfig) {
+        return true;
+      }
+    }
+    return false;
+  }
+
+  protected void markParentFragments(boolean hasChildWithConfig) {
+    ViewParent parent = getScreen().getContainer();
+    while (parent != null) {
+      if (parent instanceof Screen) {
+        ((Screen) parent).getFragment().mHasChildWithConfig = hasChildWithConfig;
+        parent = ((Screen) parent).getContainer();
+      } else {
+        parent = parent.getParent();
+      }
+    }
+  }
+
+  public List<ScreenContainer> getChildScreenContainers() {
+    return mChildScreenContainers;
+  }
+
   protected void dispatchOnWillAppear() {
     ((ReactContext) mScreenView.getContext())
         .getNativeModule(UIManagerModule.class)

android/src/main/java/com/swmansion/rnscreens/ScreenStack.java

@@ -3,6 +3,7 @@
 import android.content.Context;
 import android.view.View;
 
+import androidx.annotation.Nullable;
 import androidx.fragment.app.Fragment;
 import androidx.fragment.app.FragmentManager;
 import androidx.fragment.app.FragmentTransaction;
@@ -54,7 +55,8 @@ public void dismiss(ScreenStackFragment screenFragment) {
     markUpdated();
   }
 
-  public Screen getTopScreen() {
+  @Override
+  public @Nullable Screen getTopScreen() {
     return mTopScreen != null ? mTopScreen.getScreen() : null;
   }
 
@@ -261,9 +263,12 @@ public void run() {
     if (mTopScreen != null) {
       setupBackHandlerIfNeeded(mTopScreen);
     }
+  }
 
+  @Override
+  protected void notifyContainerUpdate() {
     for (ScreenStackFragment screen : mStack) {
-      screen.onStackUpdate();
+      screen.onContainerUpdate();
     }
   }
 

android/src/main/java/com/swmansion/rnscreens/ScreenStackFragment.java

@@ -109,10 +109,11 @@ public void setToolbarTranslucent(boolean translucent) {
     }
   }
 
-  public void onStackUpdate() {
-    View child = mScreenView.getChildAt(0);
-    if (child instanceof ScreenStackHeaderConfig) {
-      ((ScreenStackHeaderConfig) child).onUpdate();
+  @Override
+  public void onContainerUpdate() {
+    ScreenStackHeaderConfig headerConfig = getScreen().getHeaderConfig();
+    if (headerConfig != null) {
+      headerConfig.onUpdate();
     }
   }