add: use the snapshot from the service and update how the cells are displayed

Author: leandroalonso

WordPress/Classes/ViewRelated/Blog/Blog Dashboard/ViewModel/BlogDashboardViewModel.swift

@@ -17,39 +17,26 @@ class BlogDashboardViewModel {
     private let quickLinks = ["Quick Links"]
     private let posts = ["Posts"]
 
-    typealias DataSource = UICollectionViewDiffableDataSource<Section, String>
-    typealias Snapshot = NSDiffableDataSourceSnapshot<Section, String>
     typealias QuickLinksHostCell = HostCollectionViewCell<QuickLinksView>
 
     private let managedObjectContext: NSManagedObjectContext
     private let blog: Blog
 
-    private lazy var service: DashboardServiceRemote = {
-        let api = WordPressComRestApi.defaultApi(in: managedObjectContext,
-                                                 localeKey: WordPressComRestApi.LocaleKeyV2)
-
-        return DashboardServiceRemote(wordPressComRestApi: api)
+    private lazy var service: BlogDashboardService = {
+        return BlogDashboardService(managedObjectContext: managedObjectContext)
     }()
 
-    private lazy var dataSource: DataSource? = {
+    private lazy var dataSource: DashboardDataSource? = {
         guard let viewController = viewController else {
             return nil
         }
 
-        return DataSource(collectionView: viewController.collectionView) { [unowned self] collectionView, indexPath, identifier in
-            switch identifier {
-            case self.quickLinks.first:
-                let cell = collectionView.dequeueReusableCell(withReuseIdentifier: QuickLinksHostCell.defaultReuseID, for: indexPath) as? QuickLinksHostCell
-                cell?.hostedView = QuickLinksView(title: self.quickLinks[indexPath.item])
-                return cell
-            case self.posts.first:
-                let cell = collectionView.dequeueReusableCell(withReuseIdentifier: DashboardPostsCardCell.defaultReuseID, for: indexPath) as? DashboardPostsCardCell
-                cell?.configure(viewController, blog: blog)
-                return cell
-            default:
-                break
-            }
-            return UICollectionViewCell()
+        return DashboardDataSource(collectionView: viewController.collectionView) { [unowned self] collectionView, indexPath, identifier in
+
+            let cellType = identifier.id.cell
+            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: cellType.defaultReuseID, for: indexPath)
+            return cell
+
         }
     }()
 
@@ -68,15 +55,9 @@ class BlogDashboardViewModel {
         viewController?.showLoading()
         applySnapshotForInitialData()
 
-        let cardsToFetch: [String] = DashboardCard.allCases
-            .filter { $0.isRemote }
-            .map { $0.rawValue }
-
-        service.fetch(cards: cardsToFetch, forBlogID: dotComID, success: { [weak self] _ in
+        service.fetch(wpComID: dotComID, completion: { [weak self] snapshot in
             self?.viewController?.stopLoading()
-            self?.applySnapshotWithMockedData()
-        }, failure: { _ in
-
+            self?.apply(snapshot: snapshot)
         })
     }
 }
@@ -87,15 +68,11 @@ private extension BlogDashboardViewModel {
     // This is necessary when using an IntrinsicCollectionView
     // Otherwise, the collection view will never update its height
     func applySnapshotForInitialData() {
-        let snapshot = Snapshot()
+        let snapshot = DashboardSnapshot()
         dataSource?.apply(snapshot, animatingDifferences: false)
     }
 
-    func applySnapshotWithMockedData() {
-        var snapshot = Snapshot()
-        snapshot.appendSections(Section.allCases)
-        snapshot.appendItems(quickLinks, toSection: Section.quickLinks)
-        snapshot.appendItems(posts, toSection: Section.posts)
+    func apply(snapshot: DashboardSnapshot) {
         dataSource?.apply(snapshot, animatingDifferences: false)
     }
 }