automatically map to AnyCellViewModel

Author: jessesquires

Example/Sources/Grid/GridViewController.swift

@@ -57,14 +57,12 @@ final class GridViewController: UICollectionViewController {
 
     func makeViewModel(from model: Model) -> CollectionViewModel {
         let peopleCellViewModels = model.people.map { GridPersonCellViewModel(person: $0) }
-        let anyPeopleModels = peopleCellViewModels.map { $0.toAnyViewModel() }
         let peopleSection = SectionViewModel(id: "section_0_people",
-                                             cells: anyPeopleModels)
+                                             cells: peopleCellViewModels)
 
         let colorCellViewModels = model.colors.map { GridColorCellViewModel(color: $0) }
-        let anyColorModels = colorCellViewModels.map { $0.toAnyViewModel() }
         let colorSection = SectionViewModel(id: "section_1_colors",
-                                            cells: anyColorModels)
+                                            cells: colorCellViewModels)
 
         return CollectionViewModel(sections: [peopleSection, colorSection])
     }

Example/Sources/List/ListViewController.swift

@@ -41,14 +41,12 @@ final class ListViewController: UICollectionViewController {
 
     func makeViewModel(from model: Model) -> CollectionViewModel {
         let peopleCellViewModels = model.people.map { ListPersonCellViewModel(person: $0) }
-        let anyPeopleModels = peopleCellViewModels.map { $0.toAnyViewModel() }
         let peopleSection = SectionViewModel(id: "section_0_people",
-                                             cells: anyPeopleModels)
+                                             cells: peopleCellViewModels)
 
         let colorCellViewModels = model.colors.map { ListColorCellViewModel(color: $0) }
-        let anyColorModels = colorCellViewModels.map { $0.toAnyViewModel() }
         let colorSection = SectionViewModel(id: "section_1_colors",
-                                            cells: anyColorModels)
+                                            cells: colorCellViewModels)
 
         return CollectionViewModel(sections: [peopleSection, colorSection])
     }

Sources/CellViewModel.swift

@@ -54,10 +54,6 @@ extension CellViewModel {
     }
 
     public func didSelect(with controller: UIViewController) { }
-
-    public func toAnyViewModel() -> AnyCellViewModel {
-        AnyCellViewModel(self)
-    }
 }
 
 /// A type-erased cell view model.

Sources/SectionViewModel.swift

@@ -19,10 +19,9 @@ public struct SectionViewModel: DiffableViewModel {
 
     public let cellViewModels: [AnyCellViewModel]
 
-    public init(id: UniqueIdentifier,
-                cells: [AnyCellViewModel]) {
+    public init<T: CellViewModel>(id: UniqueIdentifier, cells: [T]) {
         self.id = id
-        self.cellViewModels = cells
+        self.cellViewModels = cells.map { AnyCellViewModel($0) }
     }
 }