Change implementation of headerConfig prop on Android

software-mansion · Sep 6, 2023 · 0545a7e · 0545a7e
1 parent e330f36
commit 0545a7e
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/android/src/main/java/com/swmansion/rnscreens/Screen.kt b/android/src/main/java/com/swmansion/rnscreens/Screen.kt
@@ -8,6 +8,7 @@ import android.util.SparseArray
 import android.view.ViewGroup
 import android.view.WindowManager
 import android.webkit.WebView
+import androidx.core.view.children
 import com.facebook.react.bridge.GuardedRunnable
 import com.facebook.react.bridge.ReactContext
 import com.facebook.react.uimanager.UIManagerModule
@@ -84,7 +85,15 @@ class Screen constructor(context: ReactContext?) : FabricEnabledViewGroup(contex
     }
 
     val headerConfig: ScreenStackHeaderConfig?
-        get() = getChildAt(0) as? ScreenStackHeaderConfig
+        get() {
+            for (child in children) {
+                if (child is ScreenStackHeaderConfig) {
+                    return child
+                }
+            }
+
+            return null
+        }
 
     /**
      * While transitioning this property allows to optimize rendering behavior on Android and provide