Add open method that is being called every time sections are updated

adam-leitgeb · Apr 23, 2020 · d833b43 · d833b43
1 parent 7524c40
commit d833b43
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/Sources/DataSource/DataSource.swift b/Sources/DataSource/DataSource.swift
@@ -6,7 +6,7 @@ open class DataSource: NSObject {
 
     public var sections: [Section] = [] {
         didSet {
-            canCallScrollToBottom = true
+            sectionsDidUpdate(sections)
         }
     }
 
@@ -25,8 +25,16 @@ open class DataSource: NSObject {
         self.sections = sections
     }
 
-    // MARK: - Scroll view delegate
+    // MARK: - Utilities
 
+    open func sectionsDidUpdate(_ sections: [Section]) {
+        canCallScrollToBottom = true
+    }
+}
+
+// MARK: - UIScrollViewDelegate
+
+extension DataSource {
     open func scrollViewDidScroll(_ scrollView: UIScrollView) {
         let offsetY = scrollView.contentOffset.y
         let contentHeight = scrollView.contentSize.height