forked from zelo72/mastodon-ios
39 lines
1.5 KiB
Swift
39 lines
1.5 KiB
Swift
//
|
|
// TableViewCellHeightCacheableContainer.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-3-3.
|
|
//
|
|
|
|
import UIKit
|
|
|
|
protocol TableViewCellHeightCacheableContainer {
|
|
var cellFrameCache: NSCache<NSNumber, NSValue> { get }
|
|
func cacheTableView(_ tableView: UITableView, didEndDisplaying cell: UITableViewCell, forRowAt indexPath: IndexPath)
|
|
func handleTableView(_ tableView: UITableView, estimatedHeightForRowAt indexPath: IndexPath) -> CGFloat
|
|
}
|
|
|
|
extension TableViewCellHeightCacheableContainer where Self: StatusProvider {
|
|
|
|
func cacheTableView(_ tableView: UITableView, didEndDisplaying cell: UITableViewCell, forRowAt indexPath: IndexPath) {
|
|
guard let item = item(for: nil, indexPath: indexPath) else { return }
|
|
|
|
let key = item.hashValue
|
|
let frame = cell.frame
|
|
cellFrameCache.setObject(NSValue(cgRect: frame), forKey: NSNumber(value: key))
|
|
}
|
|
|
|
func handleTableView(_ tableView: UITableView, estimatedHeightForRowAt indexPath: IndexPath) -> CGFloat {
|
|
guard let item = item(for: nil, indexPath: indexPath) else { return UITableView.automaticDimension }
|
|
guard let frame = cellFrameCache.object(forKey: NSNumber(value: item.hashValue))?.cgRectValue else {
|
|
if case .bottomLoader = item {
|
|
return TimelineLoaderTableViewCell.cellHeight
|
|
} else {
|
|
return UITableView.automaticDimension
|
|
}
|
|
}
|
|
|
|
return ceil(frame.height)
|
|
}
|
|
}
|