add count of all bookmarks in domain to view model

Package.resolved

@@ -6,8 +6,8 @@
         "repositoryURL": "https://github.com/duckduckgo/content-scope-scripts",
         "state": {
           "branch": null,
-          "revision": "ed10de3485f454ca59259266a0b3942bd21b43db",
-          "version": "3.4.1"
+          "revision": "801b7f23476f797c6eaa72b070e6c80abb82801a",
+          "version": "4.4.4"
         }
       },
       {

Sources/Bookmarks/BookmarkListViewModel.swift

@@ -149,15 +149,32 @@ public class BookmarkListViewModel: BookmarkListInteracting, ObservableObject {
     }
 
     // MARK: - Read
-
+
+    public func countBookmarksForDomain(_ domain: String) -> Int {
+        let count = countBookmarksForDomain(domain, inFolder: fetchRootBookmarksFolder())
+        return count
+    }
+
+    private func countBookmarksForDomain(_ domain: String, inFolder folder: BookmarkEntity) -> Int {
+        var count = 0
+        folder.childrenArray.forEach { child in
+            if child.isFolder {
+                count += countBookmarksForDomain(domain, inFolder: child)
+            } else if child.urlObject?.isPart(ofDomain: domain) == true {
+                count += 1
+            }
+        }
+        return count
+    }
+
     public var totalBookmarksCount: Int {
         let countRequest = BookmarkEntity.fetchRequest()
         countRequest.predicate = NSPredicate(format: "%K == false", #keyPath(BookmarkEntity.isFolder))
 
         return (try? context.count(for: countRequest)) ?? 0
     }
 
-    public func fetchRootBookmarksFolder() -> BookmarkEntity {
+    private func fetchRootBookmarksFolder() -> BookmarkEntity {
         return BookmarkUtils.fetchRootFolder(context)!
     }
 

Sources/Bookmarks/BookmarksModel.swift

@@ -43,6 +43,8 @@ public protocol BookmarkListInteracting: BookmarkStoring {
     func moveBookmark(_ bookmark: BookmarkEntity,
                       fromIndex: Int,
                       toIndex: Int)
+
+    func countBookmarksForDomain(_ domain: String) -> Int 
 }
 
 public protocol FavoritesListInteracting: BookmarkStoring {