Added supplementary view providers logic.

Author: ivanvorobei

Example Apps/SPDiffable.xcodeproj/project.xcworkspace/xcuserdata/ivanvorobei.xcuserdatad/UserInterfaceState.xcuserstate

@@ -1,7 +1,7 @@
 Pod::Spec.new do |s|
 
   s.name = 'SPDiffable'
-  s.version = '1.2.5'
+  s.version = '1.2.6'
   s.summary = 'Extenshion of Diffable API which allow not duplicate code and use less models.'
   s.homepage = 'https://github.com/ivanvorobei/SPDiffable'
   s.source = { :git => 'https://github.com/ivanvorobei/SPDiffable.git', :tag => s.version }

SPDiffable.podspec

@@ -43,8 +43,16 @@ open class SPDiffableCollectionController: UICollectionViewController {
      - parameter providers: Cell Providers with valid order for processing.
      - parameter sections: Content as array of `SPDiffableSection`.
      */
-    open func setCellProviders( _ providers: [SPDiffableCollectionCellProvider], sections: [SPDiffableSection]) {
-        diffableDataSource = SPDiffableCollectionDataSource(collectionView: collectionView, cellProviders: providers)
+    open func setCellProviders(
+        _ cellProviders: [SPDiffableCollectionCellProvider],
+        supplementaryViewProviders: [SPDiffableCollectionSupplementaryViewProvider] = [],
+        sections: [SPDiffableSection])
+    {
+        diffableDataSource = SPDiffableCollectionDataSource(
+            collectionView: collectionView,
+            cellProviders: cellProviders,
+            supplementaryViewProviders: supplementaryViewProviders
+        )
         diffableDataSource?.apply(sections, animated: false)
     }
 

Sources/SPDiffable/Collection/SPDiffableCollectionController.swift

@@ -59,8 +59,16 @@ open class SPDiffableCollectionView: UICollectionView, UICollectionViewDelegate
      - parameter providers: Cell Providers with valid order for processing.
      - parameter sections: Content as array of `SPDiffableSection`.
      */
-    open func setCellProviders( _ providers: [SPDiffableCollectionCellProvider], sections: [SPDiffableSection]) {
-        diffableDataSource = SPDiffableCollectionDataSource(collectionView: self, cellProviders: providers)
+    open func setCellProviders(
+        _ cellProviders: [SPDiffableCollectionCellProvider],
+        supplementaryViewProviders: [SPDiffableCollectionSupplementaryViewProvider] = [],
+        sections: [SPDiffableSection])
+    {
+        diffableDataSource = SPDiffableCollectionDataSource(
+            collectionView: self,
+            cellProviders: cellProviders,
+            supplementaryViewProviders: supplementaryViewProviders
+        )
         diffableDataSource?.apply(sections, animated: false)
     }
 

Sources/SPDiffable/Collection/SPDiffableCollectionView.swift

@@ -34,7 +34,8 @@ open class SPDiffableCollectionDataSource: UICollectionViewDiffableDataSource<SP
 
     // MARK: - Init
 
-    public init(collectionView: UICollectionView, cellProviders: [CellProvider]) {
+    public init(collectionView: UICollectionView, cellProviders: [CellProvider], supplementaryViewProviders: [SupplementaryViewProvider]) {
+        
         super.init(collectionView: collectionView) { (collectionView, indexPath, item) -> UICollectionViewCell? in
             for provider in cellProviders {
                 if let cell = provider(collectionView, indexPath, item) {
@@ -43,6 +44,17 @@ open class SPDiffableCollectionDataSource: UICollectionViewDiffableDataSource<SP
             }
             return nil
         }
+        
+        if !supplementaryViewProviders.isEmpty {
+            supplementaryViewProvider = { (collectionView, kind, indexPath) -> UICollectionReusableView? in
+                for provider in supplementaryViewProviders {
+                    if let view = provider(collectionView, kind, indexPath) {
+                        return view
+                    }
+                }
+                return nil
+            }
+        }
     }
 
     // MARK: - Apply Content
@@ -134,3 +146,9 @@ open class SPDiffableCollectionDataSource: UICollectionViewDiffableDataSource<SP
  */
 @available(iOS 13.0, *)
 public typealias SPDiffableCollectionCellProvider = SPDiffableCollectionDataSource.CellProvider
+
+/**
+ SPDiffable: Wrapper of collection supplementary view provider.
+ */
+@available(iOS 13.0, *)
+public typealias SPDiffableCollectionSupplementaryViewProvider = SPDiffableCollectionDataSource.SupplementaryViewProvider