chnage navigationModelDelegate

tzopiz · Jul 5, 2024 · f0c7fac · f0c7fac
1 parent 279c6e5
commit f0c7fac
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/Sources/UIComponents/Controllers/BaseViewController/BaseViewController.swift b/Sources/UIComponents/Controllers/BaseViewController/BaseViewController.swift
@@ -81,15 +81,22 @@ extension BaseViewController {
 
 // MARK: - ViewModelNavigationDelegate
 extension BaseViewController: ViewModelNavigationDelegate {
-    public func pushViewController(_ viewController: UIViewController, animated: Bool) {
+    public func pushController(
+        _ viewController: UIViewController,
+        animated: Bool
+    ) {
         guard let navigationController = navigationController else { return }
         navigationController.pushViewController(viewController, animated: animated)
     }
-    public func dismiss(animated: Bool) {
+    public func dismissController(animated: Bool, completion: (() -> Void)? = nil) {
         guard let navigationController = navigationController else { return }
-        navigationController.dismiss(animated: animated, completion: nil)
+        navigationController.dismiss(animated: animated, completion: completion)
     }
-    public func presentViewController(_ viewController: UIViewController, animated: Bool) {
-        present(viewController, animated: animated)
+    public func presentController(
+        _ viewController: UIViewController,
+        animated: Bool,
+        completion: (() -> Void)? = nil
+    ) {
+        present(viewController, animated: animated, completion: completion)
     }
 }
diff --git a/Sources/UIComponents/Controllers/BaseViewController/ViewModelNavigationDelegate.swift b/Sources/UIComponents/Controllers/BaseViewController/ViewModelNavigationDelegate.swift
@@ -8,8 +8,8 @@
 import UIKit
 
 public protocol ViewModelNavigationDelegate: AnyObject {
-    func presentViewController(_ viewController: UIViewController, animated: Bool)
-    func pushViewController(_ viewController: UIViewController, animated: Bool)
-    func dismiss(animated: Bool)
+    func presentController(_ viewController: UIViewController, animated: Bool, completion: (() -> Void)?)
+    func pushController(_ viewController: UIViewController, animated: Bool)
+    func dismissController(animated: Bool, completion: (() -> Void)?)
 }