Minor/refactoring 🖖

DataSourceKit/CellBinder.swift

@@ -26,7 +26,7 @@ public struct CellBinder {
         self.reuseIdentifier = reuseIdentifier
         self.configureCell = { cell in
             guard let cell = cell as? Cell else {
-                fatalError("Could not cast UICollectionView cell to \(Cell.self)")
+                fatalError("Could not cast cell to \(Cell.self)")
             }
 
             configureCell(cell)

DataSourceKit/CollectionViewDataSource.swift

@@ -9,15 +9,25 @@
 import UIKit
 
 public class CollectionViewDataSource<CellDeclaration>: NSObject, UICollectionViewDataSource {
+
+    // MARK: Public Properties
+
+    public typealias Configurator = (CellDeclaration) -> CellBinder
     public var cellDeclarations = [] as [CellDeclaration]
-
+
+    // MARK: Private Properties
+
     private var registeredReuseIdentifiers = [] as [String]
-    private let binderFromDeclaration: (CellDeclaration) -> CellBinder
-
-    public init(binderFromDeclaration: @escaping (CellDeclaration) -> CellBinder) {
+    private let binderFromDeclaration: Configurator
+
+    // MARK: Init
+
+    public init(binderFromDeclaration: @escaping Configurator) {
         self.binderFromDeclaration = binderFromDeclaration
         super.init()
     }
+
+    // MARK: UICollectionViewDataSource
 
     public func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
         return cellDeclarations.count

DataSourceKit/TableViewDataSource.swift

@@ -9,15 +9,25 @@
 import UIKit
 
 public class TableViewDataSource<CellDeclaration>: NSObject, UITableViewDataSource {
+
+    // MARK: Public Properties
+
+    public typealias Configurator = (CellDeclaration) -> CellBinder
     public var cellDeclarations = [] as [CellDeclaration]
-
+
+    // MARK: Private Properties
+
     private var registeredReuseIdentifiers = [] as [String]
-    private let binderFromDeclaration: (CellDeclaration) -> CellBinder
-
-    public init(binderFromDeclaration: @escaping (CellDeclaration) -> CellBinder) {
+    private let binderFromDeclaration: Configurator
+
+    // MARK: Init
+
+    public init(binderFromDeclaration: @escaping Configurator) {
         self.binderFromDeclaration = binderFromDeclaration
         super.init()
     }
+
+    // MARK: UITableViewDataSource
 
     public func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
         return cellDeclarations.count
@@ -36,8 +46,10 @@ public class TableViewDataSource<CellDeclaration>: NSObject, UITableViewDataSour
             }
             registeredReuseIdentifiers.append(cellBinder.reuseIdentifier)
         }
+
         let cell = tableView.dequeueReusableCell(withIdentifier: cellBinder.reuseIdentifier, for: indexPath)
         cellBinder.configureCell(cell)
+
         return cell
     }
 }

DataSourceKitTests/Example/CTableViewCell.swift

@@ -12,7 +12,7 @@ import DataSourceKit
 class CTableViewCell: UITableViewCell {
     var idLabel = UILabel(frame: .zero)
 
-    override init(style: UITableViewCellStyle, reuseIdentifier: String?) {
+    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
         super.init(style: style, reuseIdentifier: reuseIdentifier)
         contentView.addSubview(idLabel)
         idLabel.leadingAnchor.constraint(equalTo: contentView.leadingAnchor)