handle snapshots on iOS

Author: bartlomiejbloniarz

packages/react-native-reanimated/Common/cpp/reanimated/LayoutAnimations/LayoutAnimationsProxy_Experimental.cpp

@@ -57,7 +57,12 @@ std::optional<MountingTransaction> LayoutAnimationsProxy_Experimental::pullTrans
       ReanimatedSystraceSection s("find after elements");
       findSharedElementsOnScreen(afterTopScreen, AFTER, propsParserContext);
 #ifdef __APPLE__
-      forceScreenSnapshot_(afterTopScreen->current.tag);
+      // this is a temporary workaround for RNScreens on iOS, which takes
+      // the snapshot of the popped screen before we hide the shared element,
+      // the issue should be gone with the new stack implementation
+      if (auto screen = findParentRNSScreen(afterTopScreen)) {
+        forceScreenSnapshot_(screen->current.tag);
+      }
 #endif
     }
     const bool hasScreenChanged = beforeTopScreen && afterTopScreen && beforeTopScreen != afterTopScreen;

packages/react-native-reanimated/Common/cpp/reanimated/LayoutAnimations/LayoutAnimationsUtils.h

@@ -135,6 +135,14 @@ static inline bool isRNSScreen(const std::shared_ptr<LightNode> &node) {
   return !std::strcmp(componentName, "RNSScreen") || !std::strcmp(componentName, "RNSModalScreen");
 }
 
+static inline std::shared_ptr<LightNode> findParentRNSScreen(const std::shared_ptr<LightNode> &node) {
+  auto current = node->parent.lock();
+  while (current && !isRNSScreen(current)) {
+    current = current->parent.lock();
+  }
+  return current;
+}
+
 static inline bool isSETBoundary(const std::shared_ptr<LightNode> &node) {
   return !std::strcmp(node->current.componentName, RNReanimatedSharedTransitionBoundaryComponentName);
 }