mastodon-ios/Mastodon/Protocol/TableViewCellHeightCacheabl...

37 lines
1.2 KiB
Swift
Raw Normal View History

//
// TableViewCellHeightCacheableContainer.swift
// Mastodon
//
// Created by MainasuK Cirno on 2021-3-3.
//
import UIKit
2021-04-07 08:24:28 +02:00
protocol TableViewCellHeightCacheableContainer: StatusProvider {
var cellFrameCache: NSCache<NSNumber, NSValue> { get }
}
extension TableViewCellHeightCacheableContainer {
func cacheTableView(_ tableView: UITableView, didEndDisplaying cell: UITableViewCell, forRowAt indexPath: IndexPath) {
2021-04-07 08:24:28 +02:00
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 {
2021-04-08 05:26:15 +02:00
guard let item = item(for: nil, indexPath: indexPath) else { return UITableView.automaticDimension }
2021-04-07 08:24:28 +02:00
guard let frame = cellFrameCache.object(forKey: NSNumber(value: item.hashValue))?.cgRectValue else {
if case .bottomLoader = item {
return TimelineLoaderTableViewCell.cellHeight
} else {
return UITableView.automaticDimension
2021-04-07 08:24:28 +02:00
}
}
return ceil(frame.height)
}
}