UITableView+Extension.swift

UITableView+Extension.swift

import UIKit

extension UITableViewDataSource where Self: UITableView {
   /**
    * Returns a cells in table
    * - Note: Unlike native cell(for index) or visibleCells calls, this call wont break/complain even when called before the cells are available
    */
   var availableCells: [UITableViewCell] {
      self.subviews.compactMap { $0 as? UITableViewCell }
   }
   /**
    * Returns a cell in table for indexpath
    * - Note: Unlike native cell(for index) or visibleCells calls, this call wont break/complain even when called before the cells are available
    */
   func availableCell(indexPath: IndexPath) -> UITableViewCell? {
      availableCells.first { self.indexPath(for: $0) == indexPath }
   }
}