Rely on original delegate for section row, delegate, and footer heights

Juanpe · Oct 11, 2021 · ee434a0 · ee434a0
1 parent 17fb1b9
commit ee434a0
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/SkeletonViewCore/Sources/Internal/Collections/SkeletonCollectionDelegate.swift b/SkeletonViewCore/Sources/Internal/Collections/SkeletonCollectionDelegate.swift
@@ -43,6 +43,30 @@ extension SkeletonCollectionDelegate: UITableViewDelegate {
         originalTableViewDelegate?.tableView?(tableView, didEndDisplaying: cell, forRowAt: indexPath)
     }
 
+    func tableView(_ tableView: UITableView, estimatedHeightForRowAt indexPath: IndexPath) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, estimatedHeightForRowAt: indexPath) ?? tableView.estimatedRowHeight
+    }
+
+    func tableView(_ tableView: UITableView, estimatedHeightForHeaderInSection section: Int) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, estimatedHeightForHeaderInSection: section) ?? tableView.estimatedSectionHeaderHeight
+    }
+
+    func tableView(_ tableView: UITableView, estimatedHeightForFooterInSection section: Int) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, estimatedHeightForFooterInSection: section) ?? tableView.estimatedSectionFooterHeight
+    }
+
+    func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, heightForRowAt: indexPath) ?? tableView.rowHeight
+    }
+
+    func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, heightForHeaderInSection: section) ?? tableView.sectionHeaderHeight
+    }
+
+    func tableView(_ tableView: UITableView, heightForFooterInSection section: Int) -> CGFloat {
+        return originalTableViewDelegate?.tableView?(tableView, heightForFooterInSection: section) ?? tableView.sectionFooterHeight
+    }
+
     private func headerOrFooterView(_ tableView: UITableView, for viewIdentifier: String? ) -> UIView? {
       guard let viewIdentifier = viewIdentifier, let header = tableView.dequeueReusableHeaderFooterView(withIdentifier: viewIdentifier) else { return nil }
       skeletonViewIfContainerSkeletonIsActive(container: tableView, view: header)