Add shortcut initializers to data sources

ishkawa · Sep 12, 2018 · 1f72c71 · 1f72c71
1 parent b2d1dcc
commit 1f72c71
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/DataSourceKit/CollectionViewDataSource.swift b/DataSourceKit/CollectionViewDataSource.swift
@@ -36,3 +36,11 @@ public class CollectionViewDataSource<CellDeclaration>: NSObject, UICollectionVi
         return cell
     }
 }
+
+extension CollectionViewDataSource where CellDeclaration == CellBinder {
+    // Add meaningless parameter to avoid error `Members of constrained extensions
+    // cannot be declared @objc`, which is caused by collision to init() of NSObject.
+    public convenience init(_ declarationType: CellDeclaration.Type) {
+        self.init(binderFromDeclaration: { $0 })
+    }
+}
diff --git a/DataSourceKit/TableViewDataSource.swift b/DataSourceKit/TableViewDataSource.swift
@@ -36,3 +36,11 @@ public class TableViewDataSource<CellDeclaration>: NSObject, UITableViewDataSour
         return cell
     }
 }
+
+extension TableViewDataSource where CellDeclaration == CellBinder {
+    // Add meaningless parameter to avoid error `Members of constrained extensions
+    // cannot be declared @objc`, which is caused by collision to init() of NSObject.
+    public convenience init(_ declarationType: CellDeclaration.Type) {
+        self.init(binderFromDeclaration: { $0 })
+    }
+}