Added delegate as wrapper optionals data source methods

Author: ivanvorobei

SPDiffable.xcodeproj/project.pbxproj

@@ -7,6 +7,7 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
+		F4632035249EBA7100AF7413 /* SPTableDiffableDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = F4632034249EBA7100AF7413 /* SPTableDiffableDelegate.swift */; };
 		F4954BE2249B788A007072FA /* SPDiffable.h in Headers */ = {isa = PBXBuildFile; fileRef = F4954BE0249B788A007072FA /* SPDiffable.h */; settings = {ATTRIBUTES = (Public, ); }; };
 		F4954BEE249B95F2007072FA /* FUNDING.yml in Resources */ = {isa = PBXBuildFile; fileRef = F4954BED249B95F2007072FA /* FUNDING.yml */; };
 		F4954BF7249B962C007072FA /* Readme.md in Resources */ = {isa = PBXBuildFile; fileRef = F4954BF2249B962C007072FA /* Readme.md */; };
@@ -21,6 +22,7 @@
 /* End PBXBuildFile section */
 
 /* Begin PBXFileReference section */
+		F4632034249EBA7100AF7413 /* SPTableDiffableDelegate.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SPTableDiffableDelegate.swift; sourceTree = "<group>"; };
 		F4954BDD249B788A007072FA /* SPDiffable.framework */ = {isa = PBXFileReference; explicitFileType = wrapper.framework; includeInIndex = 0; path = SPDiffable.framework; sourceTree = BUILT_PRODUCTS_DIR; };
 		F4954BE0249B788A007072FA /* SPDiffable.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = SPDiffable.h; sourceTree = "<group>"; };
 		F4954BE1249B788A007072FA /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = Info.plist; sourceTree = "<group>"; };
@@ -50,6 +52,15 @@
 /* End PBXFrameworksBuildPhase section */
 
 /* Begin PBXGroup section */
+		F4632036249EBB9200AF7413 /* Views */ = {
+			isa = PBXGroup;
+			children = (
+				F4954C01249B9715007072FA /* SPDiffableTableController.swift */,
+				F4954BFF249B9715007072FA /* SPDiffableTableView.swift */,
+			);
+			path = Views;
+			sourceTree = "<group>";
+		};
 		F4954BD3249B788A007072FA = {
 			isa = PBXGroup;
 			children = (
@@ -129,9 +140,9 @@
 		F4954BFE249B9715007072FA /* Table */ = {
 			isa = PBXGroup;
 			children = (
-				F4954BFF249B9715007072FA /* SPDiffableTableView.swift */,
+				F4632036249EBB9200AF7413 /* Views */,
 				F4954C00249B9715007072FA /* SPTableDiffableDataSource.swift */,
-				F4954C01249B9715007072FA /* SPDiffableTableController.swift */,
+				F4632034249EBA7100AF7413 /* SPTableDiffableDelegate.swift */,
 			);
 			path = Table;
 			sourceTree = "<group>";
@@ -225,6 +236,7 @@
 				F4954C03249B9716007072FA /* SPDiffableFooter.swift in Sources */,
 				F4954C06249B9716007072FA /* SPDiffableSection.swift in Sources */,
 				F4954C07249B9716007072FA /* SPDiffableTableView.swift in Sources */,
+				F4632035249EBA7100AF7413 /* SPTableDiffableDelegate.swift in Sources */,
 				F4954C08249B9716007072FA /* SPTableDiffableDataSource.swift in Sources */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;

Source/SPDiffable/Table/SPTableDiffableDataSource.swift

@@ -23,6 +23,8 @@ import UIKit
 
 open class SPTableDiffableDataSource: UITableViewDiffableDataSource<SPDiffableSection, SPDiffableItem> {
 
+    public weak var diffableDelegate: SPTableDiffableDelegate?
+    
     public init(tableView: UITableView, cellProviders: [CellProvider]) {
         super.init(tableView: tableView, cellProvider: { (tableView, indexPath, model) -> UITableViewCell? in
             for provider in cellProviders {
@@ -45,21 +47,27 @@ open class SPTableDiffableDataSource: UITableViewDiffableDataSource<SPDiffableSe
     }
 
     public override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {
+        if let title = diffableDelegate?.tableView(tableView, titleForHeaderInSection: section) {
+            return title
+        }
         if let header = snapshot().sectionIdentifiers[section].header as? SPDiffableTextHeader {
             return header.text
         }
         return nil
     }
 
     public override func tableView(_ tableView: UITableView, titleForFooterInSection section: Int) -> String? {
+        if let title = diffableDelegate?.tableView(tableView, titleForFooterInSection: section) {
+            return title
+        }
         if let footer = snapshot().sectionIdentifiers[section].footer as? SPDiffableTextFooter {
             return footer.text
         }
         return nil
     }
 
     public override func tableView(_ tableView: UITableView, canEditRowAt indexPath: IndexPath) -> Bool {
-        return true
+        return diffableDelegate?.tableView(tableView, canEditRowAt: indexPath) ?? false
     }
 }
 

Source/SPDiffable/Table/SPTableDiffableDelegate.swift

@@ -0,0 +1,31 @@
+// The MIT License (MIT)
+// Copyright © 2020 Ivan Varabei (varabeis@icloud.com)
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+import UIKit
+
+public protocol SPTableDiffableDelegate: class {
+    
+    func tableView(_ tableView: UITableView, canEditRowAt indexPath: IndexPath) -> Bool
+    
+    func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String?
+    
+    func tableView(_ tableView: UITableView, titleForFooterInSection section: Int) -> String?
+}