Add prepareForPresentation

Author: t0maboro

ios/gamma/modals/form-sheet/RNSFormSheetContentController.h

@@ -18,6 +18,8 @@ NS_ASSUME_NONNULL_BEGIN
 
 @property (nonatomic, readonly, nonnull) RNSFormSheetContentView *contentView;
 
+- (void)prepareForPresentation;
+
 @end
 
 NS_ASSUME_NONNULL_END

ios/gamma/modals/form-sheet/RNSFormSheetContentController.mm

@@ -30,19 +30,13 @@ - (void)loadView
   self.view = [RNSFormSheetContentView new];
 }
 
-- (void)viewWillAppear:(BOOL)animated
-{
-  [super viewWillAppear:animated];
-  // Recreate that delegate on each dismiss/present cycle, because
-  // UIKit isn't triggering presentationControllerDidDismiss many times for
-  // a single initialization.
-  self.presentationController.delegate = self;
-}
+#pragma mark - Presentation Setup
 
-- (void)viewDidDisappear:(BOOL)animated
+- (void)prepareForPresentation
 {
-  [super viewDidDisappear:animated];
-  self.presentationController.delegate = nil;
+  // The presentation controller is recreated by UIKit on every present/dismiss cycle.
+  // We must assign this delegate before actual presentation
+  self.presentationController.delegate = self;
 }
 
 - (void)viewDidLayoutSubviews

ios/gamma/modals/form-sheet/RNSFormSheetHostComponentView.mm

@@ -101,6 +101,9 @@ - (void)updatePresentationState
       return;
     }
 
+    // TODO: @t0maboro - this log definitely requires refactor now and it should be removed outside Host in a followup
+    // PR
+    [_controller prepareForPresentation];
     [presentationSourceViewController presentViewController:_controller animated:YES completion:nil];
   } else if (!_isOpen && isPresented) {
     [_controller dismissViewControllerAnimated:YES completion:nil];