💫 Update: RNIModalSheetViewController

dominicstop · Sep 27, 2024 · 80f7594 · 80f7594
1 parent 2ae97ff
commit 80f7594
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ios/RNIModalSheetView/RNIModalSheetViewController.swift b/ios/RNIModalSheetView/RNIModalSheetViewController.swift
@@ -6,14 +6,15 @@
 //
 
 import UIKit
+import DGSwiftUtilities
 import react_native_ios_utilities
 
 #if !RCT_NEW_ARCH_ENABLED
 import React
 #endif
 
 
-open class RNIModalSheetViewController: UIViewController {
+open class RNIModalSheetViewController: ModalSheetViewControllerLifecycleNotifier {
 
   public var shouldTriggerDefaultCleanup = true;
 
@@ -43,6 +44,7 @@ open class RNIModalSheetViewController: UIViewController {
   // ---------------------------------
 
   public override func viewDidLoad() {
+    super.viewDidLoad();
 
     guard let mainSheetContentParent = self.mainSheetContentParent,
           let mainSheetContent = mainSheetContentParent.contentDelegate as? RNIWrapperViewContent
@@ -90,7 +92,9 @@ open class RNIModalSheetViewController: UIViewController {
     };
   };
 
-  public override func viewDidLayoutSubviews() {
+  public override func viewWillLayoutSubviews() {
+    super.viewWillLayoutSubviews();
+
     guard let mainSheetContentParent = self.mainSheetContentParent else {
       return;
     };