Fixed collection data source with leak collection view.

Author: ivanvorobei

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

@@ -31,11 +31,13 @@ import UIKit
 open class SPDiffableCollectionDataSource: UICollectionViewDiffableDataSource<SPDiffableSection, SPDiffableItem> {
 
     private weak var collectionView: UICollectionView?
-
+    
     // MARK: - Init
 
     public init(collectionView: UICollectionView, cellProviders: [CellProvider], supplementaryViewProviders: [SupplementaryViewProvider] = []) {
 
+        self.collectionView = collectionView
+        
         super.init(collectionView: collectionView) { (collectionView, indexPath, item) -> UICollectionViewCell? in
             for provider in cellProviders {
                 if let cell = provider(collectionView, indexPath, item) {
@@ -91,32 +93,25 @@ open class SPDiffableCollectionDataSource: UICollectionViewDiffableDataSource<SP
             // In this case we shoudn't set header to section snapshot.
             // For this case it condition only.
             let headerAsFirstElement: Bool = {
-                print("collectionView: \(collectionView)")
                 if collectionView?.collectionViewLayout is UICollectionViewFlowLayout {
-                    print("return false")
                     return false
                 }
                 if collectionView?.collectionViewLayout is UICollectionViewCompositionalLayout {
-                    print("return true")
                     return true
                 }
-                print("return true 2")
                 return true
             }()
-            print("headerAsFirstElement \(headerAsFirstElement)")
 
             for section in sections {
                 var sectionSnapshot = SPDiffableSectionSnapshot()
 
                 if headerAsFirstElement {
-                    print("call as headerAsFirstElement")
                     let header = section.header
                     if let header = header {
                         sectionSnapshot.append([header])
                     }
                     sectionSnapshot.append(section.items, to: header)
                 } else {
-                    print("call as DONT headerAsFirstElement")
                     sectionSnapshot.append(section.items)
                 }
 

Sources/SPDiffable/DataSource/SPDiffableCollectionDataSource.swift

@@ -44,6 +44,7 @@ open class SPDiffableTableDataSource: UITableViewDiffableDataSource<SPDiffableSe
     // MARK: - Init
 
     public init(tableView: UITableView, cellProviders: [CellProvider]) {
+        self.tableView = tableView
         super.init(tableView: tableView, cellProvider: { (tableView, indexPath, item) -> UITableViewCell? in
             for provider in cellProviders {
                 if let cell = provider(tableView, indexPath, item) {
@@ -52,7 +53,6 @@ open class SPDiffableTableDataSource: UITableViewDiffableDataSource<SPDiffableSe
             }
             return nil
         })
-        self.tableView = tableView
     }
 
     // MARK: - Apply Content