feat: proper snapshot when having multiple modals (#2121)

PR handling setting a proper snapshot of a modal when it is dismissed from JS on new arch. Still does not work with Test1829
software-mansion · May 14, 2024 · cfcd5c3 · cfcd5c3
1 parent 56f7bb5
commit cfcd5c3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ios/RNSScreenStack.mm b/ios/RNSScreenStack.mm
@@ -1175,7 +1175,11 @@ - (void)unmountChildComponentView:(UIView<RCTComponentViewProtocol> *)childCompo
 
 - (void)takeSnapshot
 {
-  _snapshot = [_controller.visibleViewController.view snapshotViewAfterScreenUpdates:NO];
+  if (_presentedModals.count < 2) {
+    _snapshot = [_controller.visibleViewController.view snapshotViewAfterScreenUpdates:NO];
+  } else {
+    _snapshot = [[_presentedModals.lastObject view] snapshotViewAfterScreenUpdates:NO];
+  }
 }
 
 - (void)mountingTransactionWillMount:(react::MountingTransaction const &)transaction