DeselectRowsAlongsideTransition Swift

UITableViewExtension.swift

extension UITableView {
  func reloadDataAnimated() {
    UIView.transition(with: self,
      duration: 0.35,
      options: .transitionCrossDissolve,
      animations: {
        self.reloadData()
      })
  }

  /**
   Animates cell deselection alongside transition. Must be called on `viewWillAppear`
   
   - Parameters:
   - viewController: The UIViewController where the table view is located
   */
  func deselectRowsAlongsideTransition(from viewController: UIViewController) {
    guard let coordinator = viewController.transitionCoordinator else {
      indexPathsForSelectedRows?.forEach { deselectRow(at: $0, animated: false) }
      return
    }

    coordinator.animateAlongsideTransition(in: self,
      animation: { [weak self] context in
        self?.indexPathsForSelectedRows?.forEach {
          self?.deselectRow(at: $0, animated: context.isAnimated)
        }
      }, completion: { [weak self] context in
        if context.isCancelled {
          self?.indexPathsForSelectedRows?.forEach {
            self?.selectRow(at: $0, animated: false, scrollPosition: .none)
          }
        }
      })
  }
}