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 {