Fixes #42

okhanokbay · Apr 14, 2020 · df37fd4 · df37fd4
1 parent 11c6881
commit df37fd4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/Example/ExpyTableView/BasicExampleViewController.swift b/Example/ExpyTableView/BasicExampleViewController.swift
@@ -34,6 +34,10 @@ extension BasicExampleViewController: ExpyTableViewDataSource {
 		cell.labelHeader.text = "Section: \(section) Row: 0"
 		return cell
 	}
+
+    func tableView(_ tableView: ExpyTableView, canExpandSection section: Int) -> Bool {
+        return ExpyTableViewDefaultValues.expandableStatus
+    }
 }
 
 //MARK: Basic Table View Implementation, no need to write UITableViewDataSource because ExpyTableViewDataSource is forwarding all the delegate methods of UITableView that are not handled by itself.
@@ -89,6 +93,10 @@ extension BasicExampleViewController: ExpyTableViewDelegate {
 
 		print("DID SELECT row: \(indexPath.row), section: \(indexPath.section)")
 	}
+
+    func tableView(_ tableView: ExpyTableView, expyState state: ExpyState, changeForSection section: Int) {
+        print("Current state: \(state)")
+    }
 }
 
 class FirstTableViewCell: UITableViewCell {

diff --git a/ExpyTableView/Classes/ExpyAbstractions.swift b/ExpyTableView/Classes/ExpyAbstractions.swift
@@ -30,16 +30,6 @@ public enum ExpyActionType {
 	func tableView(_ tableView: ExpyTableView, expandableCellForSection section: Int) -> UITableViewCell
 }
 
-public extension ExpyTableViewDataSource {
-	func tableView(_ tableView: ExpyTableView, canExpandSection section: Int) -> Bool {
-		return ExpyTableViewDefaultValues.expandableStatus
-	}
-}
-
 @objc public protocol ExpyTableViewDelegate: UITableViewDelegate {
 	func tableView(_ tableView: ExpyTableView, expyState state: ExpyState, changeForSection section: Int)
 }
-
-public extension ExpyTableViewDelegate {
-	func tableView(_ tableView: ExpyTableView, expyState state: ExpyState, changeForSection section: Int) {}
-}