Assign the coordinator delegate to the view model

mozilla-mobile · Nov 25, 2024 · c17e72d · c17e72d
1 parent f27f864
commit c17e72d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/firefox-ios/Client/Coordinators/Library/BookmarksCoordinator.swift b/firefox-ios/Client/Coordinators/Library/BookmarksCoordinator.swift
@@ -146,6 +146,7 @@ class BookmarksCoordinator: BaseCoordinator,
         viewModel.onBookmarkSaved = { [weak self] in
             self?.reloadLastBookmarksController()
         }
+        viewModel.bookmarkCoordinatorDelegate = self
         setBackBarButtonItemTitle(viewModel.backNavigationButtonTitle())
         let controller = EditBookmarkViewController(viewModel: viewModel,
                                                     windowUUID: windowUUID)

diff --git a/firefox-ios/Client/Frontend/Library/Bookmarks/Edit Bookmark/EditBookmarkViewController.swift b/firefox-ios/Client/Frontend/Library/Bookmarks/Edit Bookmark/EditBookmarkViewController.swift
@@ -262,8 +262,7 @@ class EditBookmarkViewController: UIViewController,
         guard let section = Section(rawValue: indexPath.section) else { return }
         if section == .folder, let folder = viewModel.folderStructures[safe: indexPath.row] {
             viewModel.selectFolder(folder)
-        }
-        else if section == .newFolder {
+        } else if section == .newFolder {
             viewModel.createNewFolder()
         }
     }

diff --git a/firefox-ios/Client/Frontend/Library/Bookmarks/Edit Bookmark/EditBookmarkViewModel.swift b/firefox-ios/Client/Frontend/Library/Bookmarks/Edit Bookmark/EditBookmarkViewModel.swift
@@ -14,6 +14,7 @@ class EditBookmarkViewModel {
     private let profile: Profile
     private let folderFetcher: FolderHierarchyFetcher
     private let bookmarksSaver: BookmarksSaver
+    weak var bookmarkCoordinatorDelegate: BookmarksCoordinatorDelegate?
 
     private var isFolderCollapsed = true
     private(set) var folderStructures: [Folder] = []
@@ -68,7 +69,9 @@ class EditBookmarkViewModel {
     }
 
     func createNewFolder() {
-
+        self.bookmarkCoordinatorDelegate?.showBookmarkDetail(
+            bookmarkType: .folder,
+            parentBookmarkFolder: parentFolder)
     }
 
     private func getFolderStructure(_ selectedFolder: Folder) {