use the right delegate method for footers (#394)

Juanpe · May 11, 2021 · 74b5172 · 74b5172
1 parent 318e629
commit 74b5172
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/Sources/Collections/CollectionViews/UICollectionView+CollectionSkeleton.swift b/Sources/Collections/CollectionViews/UICollectionView+CollectionSkeleton.swift
@@ -7,11 +7,10 @@
 //
 
 import UIKit
-
+ 
 extension UICollectionView: CollectionSkeleton {
     var estimatedNumberOfRows: Int {
         guard let flowlayout = collectionViewLayout as? UICollectionViewFlowLayout else { return 0 }
-        return Int(ceil(frame.height / flowlayout.itemSize.height))
         switch flowlayout.scrollDirection {
         case .vertical:
             return Int(ceil(frame.height / flowlayout.itemSize.height))

diff --git a/Sources/Collections/SkeletonCollectionDelegate.swift b/Sources/Collections/SkeletonCollectionDelegate.swift
@@ -25,7 +25,7 @@ extension SkeletonCollectionDelegate: UITableViewDelegate {
     }
 
     func tableView(_ tableView: UITableView, viewForFooterInSection section: Int) -> UIView? {
-        headerOrFooterView(tableView, for: originalTableViewDelegate?.collectionSkeletonView(tableView, identifierForHeaderInSection: section))
+        headerOrFooterView(tableView, for: originalTableViewDelegate?.collectionSkeletonView(tableView, identifierForFooterInSection: section))
     }
 
     func tableView(_ tableView: UITableView, didEndDisplayingHeaderView view: UIView, forSection section: Int) {